NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-12-19 03:42:22 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix ollama issues with ChatCompletionAssistantToolCall

Browse Source 
PiperOrigin-RevId: 761732316
This commit is contained in:
Google Team Member
2025-05-21 17:26:31 -07:00
committed by Copybara-Service
parent 505d936007
commit 4542af5650
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/unittests/models/test_litellm.py
View File

@@ -700,13 +700,12 @@ def test_content_to_message_param_function_call():
message = _content_to_message_param(content)
assert message["role"] == "assistant"
assert message["content"] == None
assert message["tool_calls"][0].type == "function"
assert message["tool_calls"][0].id == "test_tool_call_id"
assert message["tool_calls"][0].function.name == "test_function"
assert (
message["tool_calls"][0].function.arguments
== '{"test_arg": "test_value"}'
)
tool_call = message["tool_calls"][0]
assert tool_call["type"] == "function"
assert tool_call["id"] == "test_tool_call_id"
assert tool_call["function"]["name"] == "test_function"
assert tool_call["function"]["arguments"] == '{"test_arg": "test_value"}'
def test_message_to_generate_content_response_text():