scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 81

fix llm responses

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-21 10:50:55 -07:00
parent 1a56bb5bdd
commit bbf1962540
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/mcp_tests/test_mcp_litellm_client.py
View file

@ -44,5 +44,10 @@ async def test_mcp_agent():
print("LLM RESPONSE: ", json.dumps(llm_response, indent=4, default=str))
# Add assertions to verify the response
assert isinstance(llm_response, dict)
print(llm_response)
assert llm_response["choices"][0]["message"]["tool_calls"] is not None
assert (
llm_response["choices"][0]["message"]["tool_calls"][0]["function"][
"name"
]
== "add"
)