NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-13 15:14:50 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: autoformat the test_litellm.py

Browse Source 
PiperOrigin-RevId: 764937761
This commit is contained in:
Xiang (Sean) Zhou 2025-05-29 16:48:33 -07:00 committed by Copybara-Service
parent 2b41824e46
commit ae7d3a79c8
1 changed files with 62 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -13,6 +13,7 @@
# limitations under the License.
import json
from unittest.mock import AsyncMock
from unittest.mock import Mock
@ -38,7 +39,6 @@ from litellm.types.utils import Delta
from litellm.types.utils import ModelResponse
from litellm.types.utils import StreamingChoices
import pytest
import json
LLM_REQUEST_WITH_FUNCTION_DECLARATION = LlmRequest(
contents=[
@ -1251,13 +1251,9 @@ async def test_generate_content_async_multiple_function_calls(
# Verify first function call
assert final_response.content.parts[0].function_call.name == "function_1"
assert final_response.content.parts[0].function_call.id == "call_1"
assert final_response.content.parts[0].function_call.args == {
"arg": "value1"
}
assert final_response.content.parts[0].function_call.args == {"arg": "value1"}
# Verify second function call
assert final_response.content.parts[1].function_call.name == "function_2"
assert final_response.content.parts[1].function_call.id == "call_2"
assert final_response.content.parts[1].function_call.args == {
"arg": "value2"
}
assert final_response.content.parts[1].function_call.args == {"arg": "value2"}