Cerebras exception correction, added Gemini installation advice (#66)

autogenhub · Oct 18, 2024 · 66f195b · 66f195b
1 parent e318461
commit 66f195b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/autogen/oai/client.py b/autogen/oai/client.py
@@ -59,6 +59,7 @@
 
     cerebras_import_exception: Optional[ImportError] = None
 except ImportError as e:
+    cerebras_AuthenticationError = cerebras_InternalServerError = cerebras_RateLimitError = Exception
     cerebras_import_exception = e
 
 try:
@@ -577,7 +578,7 @@ def _register_default_client(self, config: Dict[str, Any], openai_config: Dict[s
                 self._clients.append(client)
             elif api_type is not None and api_type.startswith("google"):
                 if gemini_import_exception:
-                    raise ImportError("Please install `google-generativeai` to use Google OpenAI API.")
+                    raise ImportError("Please install `google-generativeai` and 'vertexai' to use Google's API.")
                 client = GeminiClient(**openai_config)
                 self._clients.append(client)
             elif api_type is not None and api_type.startswith("anthropic"):