fix: tests

This commit is contained in:
Eduardo Chiarotti
2024-06-27 10:06:24 -07:00
parent e3d93dbd90
commit 62679d2fec
2 changed files with 112 additions and 109 deletions

View File

@@ -367,6 +367,8 @@ def test_save_task_json_output():
@pytest.mark.vcr(filter_headers=["authorization"])
def test_save_task_pydantic_output():
from langchain_openai import ChatOpenAI
class ScoreOutput(BaseModel):
score: int
@@ -375,6 +377,7 @@ def test_save_task_pydantic_output():
goal="Score the title",
backstory="You're an expert scorer, specialized in scoring titles.",
allow_delegation=False,
llm=ChatOpenAI(model="gpt-4o"),
)
task = Task(