Update error messages

ag2ai · Jan 29, 2025 · 27cb917 · 27cb917
1 parent 666bb62
commit 27cb917
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/autogen/oai/client.py b/autogen/oai/client.py
@@ -82,7 +82,7 @@
     gemini_import_exception: Optional[ImportError] = None
 else:
     gemini_InternalServerError = gemini_ResourceExhausted = Exception  # noqa: N816
-    gemini_import_exception = ImportError("google-generativeai not found")
+    gemini_import_exception = ImportError("google-genai not found")
 
 with optional_import_block() as anthropic_result:
     from anthropic import (  # noqa
@@ -756,7 +756,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` and 'vertexai' to use Google's API.")
+                    raise ImportError("Please install `google-genai` and 'vertexai' to use Google's API.")
                 client = GeminiClient(response_format=response_format, **openai_config)
                 self._clients.append(client)
             elif api_type is not None and api_type.startswith("anthropic"):