From 752cf88f9224d456210abce72e5f0e00b6ca9aac Mon Sep 17 00:00:00 2001 From: Wei <21039366+Mai0313@users.noreply.github.com> Date: Thu, 30 May 2024 23:45:09 +0800 Subject: [PATCH] Fix initialization of client in retrieve_docs() function (#2830) --- autogen/agentchat/contrib/retrieve_user_proxy_agent.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/autogen/agentchat/contrib/retrieve_user_proxy_agent.py b/autogen/agentchat/contrib/retrieve_user_proxy_agent.py index f8cc205570fa..8d4dbc0d50ac 100644 --- a/autogen/agentchat/contrib/retrieve_user_proxy_agent.py +++ b/autogen/agentchat/contrib/retrieve_user_proxy_agent.py @@ -253,7 +253,9 @@ def retrieve_docs(self, problem: str, n_results: int = 20, search_string: str = self._task = self._retrieve_config.get("task", "default") self._vector_db = self._retrieve_config.get("vector_db", "chroma") self._db_config = self._retrieve_config.get("db_config", {}) - self._client = self._retrieve_config.get("client", chromadb.Client()) + self._client = self._retrieve_config.get("client", None) + if self._client is None: + self._client = chromadb.Client() self._docs_path = self._retrieve_config.get("docs_path", None) self._extra_docs = self._retrieve_config.get("extra_docs", False) self._new_docs = self._retrieve_config.get("new_docs", True)