phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(feat) - Add seed to Cohere Chat.

Browse source
This commit is contained in:
David Manouchehri 2024-04-18 20:49:44 +00:00
parent f610061a79
commit f65c02d43a
No known key found for this signature in database
2 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/utils.py
View file

@ -4739,6 +4739,8 @@ def get_optional_params(
optional_params["stop_sequences"] = stop
if tools is not None:
optional_params["tools"] = tools
if seed is not None:
optional_params["seed"] = seed
elif custom_llm_provider == "maritalk":
## check if unsupported param passed in
supported_params = get_supported_openai_params(
@ -5517,6 +5519,7 @@ def get_supported_openai_params(model: str, custom_llm_provider: str):
"n",
"tools",
"tool_choice",
"seed",
]
elif custom_llm_provider == "maritalk":
return [