andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-27 09:08:14 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix: change github action to test different pipeline

Browse Source
This commit is contained in:
Eduardo Chiarotti
2024-07-04 19:25:30 -03:00
parent 2945ddd7e9
commit a34bc87163
2 changed files with 45 additions and 881 deletions
Download Patch File Download Diff File Expand all files Collapse all files

923
tests/cassettes/test_output_pydantic.yaml
View File

File diff suppressed because it is too large Load Diff

3
tests/task_test.py
View File

@@ -200,6 +200,8 @@ def test_multiple_output_type_error():
@pytest.mark.vcr(filter_headers=["authorization"]) @pytest.mark.vcr(filter_headers=["authorization"])
def test_output_pydantic(): def test_output_pydantic():
from langchain_openai import ChatOpenAI
class ScoreOutput(BaseModel): class ScoreOutput(BaseModel):
score: int score: int
@@ -215,6 +217,7 @@ def test_output_pydantic():
expected_output="The score of the title.", expected_output="The score of the title.",
output_pydantic=ScoreOutput, output_pydantic=ScoreOutput,
agent=scorer, agent=scorer,
llm=ChatOpenAI(model="gpt-4o"),
) )
crew = Crew(agents=[scorer], tasks=[task]) crew = Crew(agents=[scorer], tasks=[task])