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

fix cohere / cohere_chat when timeout is None

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-09 12:10:02 -07:00
parent fac610beab
commit 0561c06bcf
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/main.py
View file

@ -3387,7 +3387,7 @@ def embedding(
client=client, client=client,
aembedding=aembedding, aembedding=aembedding,
) )
elif custom_llm_provider == "cohere": elif custom_llm_provider == "cohere" or custom_llm_provider == "cohere_chat":
cohere_key = ( cohere_key = (
api_key api_key
or litellm.cohere_key or litellm.cohere_key
@ -3404,7 +3404,7 @@ def embedding(
logging_obj=logging, logging_obj=logging,
model_response=EmbeddingResponse(), model_response=EmbeddingResponse(),
aembedding=aembedding, aembedding=aembedding,
timeout=float(timeout), timeout=timeout,
client=client, client=client,
) )
elif custom_llm_provider == "huggingface": elif custom_llm_provider == "huggingface":