Fix typo in require_optional_import decorator

ag2ai · Jan 21, 2025 · 3a76d53 · 3a76d53
1 parent 76eea6a
commit 3a76d53
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autogen/agentchat/contrib/retrieve_user_proxy_agent.py b/autogen/agentchat/contrib/retrieve_user_proxy_agent.py
@@ -89,7 +89,7 @@
 UPDATE_CONTEXT_IN_PROMPT = "you should reply exactly `UPDATE CONTEXT`"
 
 
-@require_optional_import(["chromadb", "ipython"], "retrievechat")
+@require_optional_import(["chromadb", "IPython"], "retrievechat")
 class RetrieveUserProxyAgent(UserProxyAgent):
     """(In preview) The Retrieval-Augmented User Proxy retrieves document chunks based on the embedding
     similarity, and sends them along with the question to the Retrieval-Augmented Assistant