From 76eea6a95db5cca326b5848bc9ccc7fce205f474 Mon Sep 17 00:00:00 2001 From: Kumaran Rajendhiran Date: Tue, 21 Jan 2025 16:11:22 +0000 Subject: [PATCH] Fix failing test_retrievechat tests --- test/agentchat/contrib/retrievechat/test_retrievechat.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/test/agentchat/contrib/retrievechat/test_retrievechat.py b/test/agentchat/contrib/retrievechat/test_retrievechat.py index e40e97d16f..e9eb32701c 100755 --- a/test/agentchat/contrib/retrievechat/test_retrievechat.py +++ b/test/agentchat/contrib/retrievechat/test_retrievechat.py @@ -10,6 +10,10 @@ 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 @@ -17,12 +21,9 @@ 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