fix(cohere_chat.py): support passing 'extra_headers'

Fixes https://github.com/BerriAI/litellm/issues/4709
This commit is contained in:
Krrish Dholakia 2024-08-22 10:16:43 -07:00
parent 36c021b309
commit 11bfc1dca7
5 changed files with 50 additions and 16 deletions

View file

@ -4219,6 +4219,7 @@ def get_supported_openai_params(
"presence_penalty",
"stop",
"n",
"extra_headers",
]
elif custom_llm_provider == "cohere_chat":
return [
@ -4233,6 +4234,7 @@ def get_supported_openai_params(
"tools",
"tool_choice",
"seed",
"extra_headers",
]
elif custom_llm_provider == "maritalk":
return [