Skip to content

Commit 8f7c248

Browse files
committed
refactor: clean up test examples
1 parent 8b5ce73 commit 8f7c248

File tree

2 files changed

+1
-5
lines changed

2 files changed

+1
-5
lines changed

tests/models/test_anthropic.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -669,10 +669,7 @@ async def my_tool(first: str, second: str) -> int:
669669
input_tokens=20,
670670
output_tokens=5,
671671
tool_calls=1,
672-
details={
673-
'input_tokens': 20,
674-
'output_tokens': 5,
675-
},
672+
details={'input_tokens': 20, 'output_tokens': 5},
676673
)
677674
)
678675
assert tool_called

tests/test_examples.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -628,7 +628,6 @@ async def model_logic( # noqa: C901
628628
parts=[ToolCallPart(tool_name='final_result', args=args, tool_call_id='pyd_ai_tool_call_id')]
629629
)
630630
elif isinstance(m, ToolReturnPart) and m.tool_name == 'do_work':
631-
# Trigger a second tool call to cause tool_calls_limit to be exceeded in the docs example
632631
return ModelResponse(parts=[ToolCallPart(tool_name='do_work', args={}, tool_call_id='pyd_ai_tool_call_id')])
633632
elif isinstance(m, RetryPromptPart) and m.tool_name == 'calc_volume':
634633
return ModelResponse(

0 commit comments

Comments
 (0)