diff --git a/autogen/agentchat/contrib/retrieve_user_proxy_agent.py b/autogen/agentchat/contrib/retrieve_user_proxy_agent.py index 611ee08514..73eb885629 100644 --- 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