From 19d6051dba59f6742b45269ea84ded1ffbb9dfd3 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 21 Mar 2025 17:48:16 -0700 Subject: [PATCH] test mcp agent --- tests/mcp_tests/test_mcp_litellm_client.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/mcp_tests/test_mcp_litellm_client.py b/tests/mcp_tests/test_mcp_litellm_client.py index 448f0c3ce5..e69aec8553 100644 --- a/tests/mcp_tests/test_mcp_litellm_client.py +++ b/tests/mcp_tests/test_mcp_litellm_client.py @@ -42,11 +42,12 @@ async def test_mcp_agent(): api_key=os.getenv("OPENAI_API_KEY"), messages=messages, tools=tools, + tool_choice="required", ) print("LLM RESPONSE: ", json.dumps(llm_response, indent=4, default=str)) - # Add assertions to verify the response assert llm_response["choices"][0]["message"]["tool_calls"] is not None + assert ( llm_response["choices"][0]["message"]["tool_calls"][0]["function"][ "name"