Skip to content

Commit a78dbe9

Browse files
author
Dylan Huang
committed
increase number of result retries
1 parent b6e1d7a commit a78dbe9

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/chinook/test_pydantic_chinook.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,7 @@ class Response(BaseModel):
136136
system_prompt=LLM_JUDGE_PROMPT,
137137
output_type=Response,
138138
model=model,
139+
result_retries=3,
139140
)
140141
result = await comparison_agent.run(
141142
f"Expected answer: {row.ground_truth}\nResponse: {last_assistant_message.content}"

0 commit comments

Comments
 (0)