Fix initialization of client in retrieve_docs() function (#2830)

microsoft · May 30, 2024 · 752cf88 · 752cf88
1 parent 47f9052
commit 752cf88
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)