Fix failing test_retrievechat tests

ag2ai · Jan 21, 2025 · 76eea6a · 76eea6a
1 parent 5761ac4
commit 76eea6a
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/test/agentchat/contrib/retrievechat/test_retrievechat.py b/test/agentchat/contrib/retrievechat/test_retrievechat.py
@@ -10,19 +10,20 @@
 
 import pytest
 
+from autogen import AssistantAgent
+from autogen.agentchat.contrib.retrieve_user_proxy_agent import (
+    RetrieveUserProxyAgent,
+)
 from autogen.import_utils import optional_import_block
 
 from ....conftest import Credentials, reason
 
 with optional_import_block() as result:
     import chromadb
     import openai  # noqa: F401
+    from IPython import get_ipython  # noqa: F401
     from chromadb.utils import embedding_functions as ef
 
-    from autogen import AssistantAgent
-    from autogen.agentchat.contrib.retrieve_user_proxy_agent import (
-        RetrieveUserProxyAgent,
-    )
 
 skip = not result.is_successful