Skip to content

Commit

Permalink
Fix vcr issues
Browse files Browse the repository at this point in the history
  • Loading branch information
bhancockio committed Jan 15, 2025
1 parent 611b4a0 commit 5d78eb4
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/utilities/test_converter.py
Original file line number Diff line number Diff line change
Expand Up @@ -324,7 +324,7 @@ class ModelWithDictField(BaseModel):
assert description == expected_description


@pytest.mark.vcr(filter_headers=["authorization"])
@pytest.mark.vcr(filter_headers=["authorization"], record_mode="new_episodes")
def test_convert_with_instructions():
llm = LLM(model="gpt-4o-mini")
sample_text = "Name: Alice, Age: 30"
Expand All @@ -346,7 +346,7 @@ def test_convert_with_instructions():
assert output.age == 30


@pytest.mark.vcr(filter_headers=["authorization"])
@pytest.mark.vcr(filter_headers=["authorization"], record_mode="new_episodes")
def test_converter_with_llama3_2_model():
llm = LLM(model="ollama/llama3.2:3b", base_url="http://localhost:11434")

Expand All @@ -367,7 +367,7 @@ def test_converter_with_llama3_2_model():
assert output.age == 30


@pytest.mark.vcr(filter_headers=["authorization"])
@pytest.mark.vcr(filter_headers=["authorization"], record_mode="new_episodes")
def test_converter_with_llama3_1_model():
llm = LLM(model="ollama/llama3.1", base_url="http://localhost:11434")
sample_text = "Name: Alice Llama, Age: 30"
Expand All @@ -387,7 +387,7 @@ def test_converter_with_llama3_1_model():
assert output.age == 30


@pytest.mark.vcr(filter_headers=["authorization"])
@pytest.mark.vcr(filter_headers=["authorization"], record_mode="new_episodes")
def test_converter_with_nested_model():
llm = LLM(model="gpt-4o-mini")
sample_text = "Name: John Doe\nAge: 30\nAddress: 123 Main St, Anytown, 12345"
Expand Down

0 comments on commit 5d78eb4

Please sign in to comment.