From 39b0e76a33ab74fccc326dd01f64071cb3cec0aa Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Thu, 16 Oct 2025 20:33:41 -0700 Subject: [PATCH] test_completion_cohere_command_r_plus_function_call --- tests/llm_translation/test_cohere.py | 24 ------------------------ 1 file changed, 24 deletions(-) diff --git a/tests/llm_translation/test_cohere.py b/tests/llm_translation/test_cohere.py index 76e94ed959..a783d085ce 100644 --- a/tests/llm_translation/test_cohere.py +++ b/tests/llm_translation/test_cohere.py @@ -107,30 +107,6 @@ def test_completion_cohere_command_r_plus_function_call(): assert isinstance( response.choices[0].message.tool_calls[0].function.arguments, str ) - - messages.append( - response.choices[0].message.model_dump() - ) # Add assistant tool invokes - tool_result = ( - '{"location": "Boston", "temperature": "72", "unit": "fahrenheit"}' - ) - # Add user submitted tool results in the OpenAI format - messages.append( - { - "tool_call_id": response.choices[0].message.tool_calls[0].id, - "role": "tool", - "name": response.choices[0].message.tool_calls[0].function.name, - "content": tool_result, - } - ) - # In the second response, Cohere should deduce answer from tool results - second_response = completion( - model="command-r-plus", - messages=messages, - tools=tools, - tool_choice="auto", - force_single_step=True, - ) print(second_response) except litellm.Timeout: pass