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

Update test_lite_agent.py (#3040)

Browse Source 
Co-authored-by: Lucas Gomide <lucaslg200@gmail.com>
This commit is contained in:
kilavvy
2025-06-26 15:55:53 +02:00
committed by GitHub
parent cdf8388b18
commit 94a62d84e1
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_lite_agent.py
View File

@@ -192,7 +192,7 @@ def test_lite_agent_structured_output():
) )
result = agent.kickoff( result = agent.kickoff(
"What is the population of Tokyo? Return your strucutred output in JSON format with the following fields: summary, confidence", "What is the population of Tokyo? Return your structured output in JSON format with the following fields: summary, confidence",
response_format=SimpleOutput, response_format=SimpleOutput,
) )
@@ -230,7 +230,7 @@ def test_lite_agent_returns_usage_metrics():
) )
result = agent.kickoff( result = agent.kickoff(
"What is the population of Tokyo? Return your strucutred output in JSON format with the following fields: summary, confidence" "What is the population of Tokyo? Return your structured output in JSON format with the following fields: summary, confidence"
) )
assert result.usage_metrics is not None assert result.usage_metrics is not None
@@ -252,7 +252,7 @@ async def test_lite_agent_returns_usage_metrics_async():
) )
result = await agent.kickoff_async( result = await agent.kickoff_async(
"What is the population of Tokyo? Return your strucutred output in JSON format with the following fields: summary, confidence" "What is the population of Tokyo? Return your structured output in JSON format with the following fields: summary, confidence"
) )
assert isinstance(result, LiteAgentOutput) assert isinstance(result, LiteAgentOutput)
assert "21 million" in result.raw or "37 million" in result.raw assert "21 million" in result.raw or "37 million" in result.raw