phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

reaf:moved tools parameter handling to map_openai_params

Browse source
This commit is contained in:
sajdkabir 2025-04-11 23:11:10 +05:30
parent 6783e4765e
commit 565329c4ac
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/cohere/chat/transformation_v2.py
View file

@ -181,7 +181,9 @@ class CohereChatConfigV2(BaseConfig):
optional_params["presence_penalty"] = value optional_params["presence_penalty"] = value
if param == "stop": if param == "stop":
optional_params["stop_sequences"] = value optional_params["stop_sequences"] = value
# The transform_request method will handle the tools parameter with _construct_cohere_tool if param == "tools":
cohere_tools = self._construct_cohere_tool(tools=value)
optional_params["tools"] = cohere_tools
if param == "seed": if param == "seed":
optional_params["seed"] = value optional_params["seed"] = value
return optional_params return optional_params
@ -211,10 +213,7 @@ class CohereChatConfigV2(BaseConfig):
optional_params["messages"] = cohere_messages optional_params["messages"] = cohere_messages
optional_params["model"] = model optional_params["model"] = model
## Handle Tool Calling ## Tool Calling is now handled in map_openai_params
if "tools" in optional_params:
cohere_tools = self._construct_cohere_tool(tools=optional_params["tools"])
optional_params["tools"] = cohere_tools
# Handle tool results if present # Handle tool results if present
if "tool_results" in optional_params and isinstance(optional_params["tool_results"], list): if "tool_results" in optional_params and isinstance(optional_params["tool_results"], list):