cleanup

sonichi · sonichi · commit 6f5463170787 · 2023-09-21T02:48:20.000Z
diff --git a/autogen/agentchat/contrib/retrieve_assistant_agent.py b/autogen/agentchat/contrib/retrieve_assistant_agent.py
@@ -1,5 +1,5 @@
-from agentchat.agent import Agent
-from agentchat.assistant_agent import AssistantAgent
+from autogen.agentchat.agent import Agent
+from autogen.agentchat.assistant_agent import AssistantAgent
 from typing import Dict, Optional, Union, List, Tuple, Any
 
 
