Imports cleanup

Author: dangusev

agents-core/vision_agents/core/agents/agent_session.py

@@ -1,7 +1,9 @@
 import asyncio
 import contextvars
+import typing
 
-from vision_agents.core.agents import Agent
+if typing.TYPE_CHECKING:
+    from .agents import Agent
 
 
 class AgentSessionContextManager:
@@ -27,7 +29,7 @@ class AgentSessionContextManager:
             returned by the edge transport (kept open during the context).
     """
 
-    def __init__(self, agent: Agent, connection_cm=None):
+    def __init__(self, agent: "Agent", connection_cm=None):
         self.agent = agent
         self._connection_cm = connection_cm
 

agents-core/vision_agents/core/agents/agents.py

@@ -60,12 +60,11 @@
 from opentelemetry import trace, context as otel_context
 from opentelemetry.trace import Tracer
 from opentelemetry.context import Token
+from .agent_session import AgentSessionContextManager
 
 if TYPE_CHECKING:
     from vision_agents.plugins.getstream.stream_edge_transport import StreamEdge
 
-    from .agent_session import AgentSessionContextManager
-
 logger = logging.getLogger(__name__)
 
 tracer: Tracer = trace.get_tracer("agents")
@@ -539,7 +538,6 @@ async def join(
                 await self.edge.publish_tracks(audio_track, video_track)
 
 
-        from .agent_session import AgentSessionContextManager
 
         # wait for conversation creation coro at the very end of the join flow
         self.conversation = await create_conversation_coro