From 00d2a58eca0cf8f7ba879b3c3fa435ebd86ca72b Mon Sep 17 00:00:00 2001 From: Eric Zhu Date: Mon, 30 Sep 2024 22:48:25 -0700 Subject: [PATCH] Better error message --- autogen/oai/client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"):