diff --git a/autogen/oai/client.py b/autogen/oai/client.py index 0a0ff0c90779..f8d1731cdb04 100644 --- 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"):