From e734568b5a254fd28c4f8626e20e75086b3e6b85 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 9 Aug 2024 12:10:02 -0700
Subject: [PATCH] fix cohere / cohere_chat when timeout is None

---
 litellm/main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/main.py b/litellm/main.py
index 1840a900a60..6200d8cb18a 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -3387,7 +3387,7 @@ def embedding(
                 client=client,
                 aembedding=aembedding,
             )
-        elif custom_llm_provider == "cohere":
+        elif custom_llm_provider == "cohere" or custom_llm_provider == "cohere_chat":
             cohere_key = (
                 api_key
                 or litellm.cohere_key
@@ -3404,7 +3404,7 @@ def embedding(
                 logging_obj=logging,
                 model_response=EmbeddingResponse(),
                 aembedding=aembedding,
-                timeout=float(timeout),
+                timeout=timeout,
                 client=client,
             )
         elif custom_llm_provider == "huggingface":