updated ValueError statement for gemini

defog-ai · Nov 15, 2024 · 2917db8 · 2917db8
1 parent 8412c5b
commit 2917db8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/defog_utils/utils_multi_llm.py b/defog_utils/utils_multi_llm.py
@@ -39,7 +39,7 @@ def map_model_to_chat_fn_async(model: str) -> Callable:
     if model.startswith("claude"):
         return chat_anthropic_async
     if model.startswith("gemini"):
-        return ValueError("Gemini does not support async chat")
+        raise ValueError("Gemini does not support async chat")
     if model.startswith("gpt") or model in ["o1", "o1-mini", "o1-preview"]:
         return chat_openai_async
     if (