Better error message

microsoft · Oct 1, 2024 · 00d2a58 · 00d2a58
1 parent 1291d92
commit 00d2a58
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autogen/oai/client.py b/autogen/oai/client.py
@@ -558,7 +558,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("ollama"):
                 if ollama_import_exception:
-                    raise ImportError("Please install `ollama` to use the Ollama API.")
+                    raise ImportError("Please install with `[ollama]` option to use the Ollama API.")
                 client = OllamaClient(**openai_config)
                 self._clients.append(client)
             elif api_type is not None and api_type.startswith("bedrock"):