tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-14 20:22:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_completion_cohere_command_r_plus_function_call

Browse Source
This commit is contained in:
Ishaan Jaffer
2025-10-16 20:33:41 -07:00
parent 64876634ef
commit 39b0e76a33
1 changed files with 0 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/llm_translation/test_cohere.py
-24
View File
@@ -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