From 27cb917114046cfdba617f73d1a60034c146d3fe Mon Sep 17 00:00:00 2001 From: Kumaran Rajendhiran Date: Wed, 29 Jan 2025 05:15:29 +0000 Subject: [PATCH] Update error messages --- autogen/oai/client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/autogen/oai/client.py b/autogen/oai/client.py index 0f7e78a4ed..bc3f8c1b77 100644 --- 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"):