perf: pyinstrument integration

Author: yarikdevcom

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

@@ -531,14 +531,13 @@ async def finish(self):
         Subscribes to the edge transport's `call_ended` event and awaits it. If
         no connection is active, returns immediately.
         """
-        # If connection is None or already closed, return immediately
         if not self._connection:
             self.logger.info(
                 "🔚 Agent connection is already closed, finishing immediately"
             )
             return
 
-
+        running_event = asyncio.Event()
         with self.span("agent.finish"):
             # If connection is None or already closed, return immediately
             if not self._connection:
@@ -549,15 +548,16 @@ async def finish(self):
 
             @self.edge.events.subscribe
             async def on_ended(event: CallEndedEvent):
+                running_event.set()
                 self._is_running = False
+        # TODO: add members count check (particiapnts left + count = 1 timeout 2 minutes)
 
-            while self._is_running:
-                try:
-                    await asyncio.sleep(0.0001)
-                except asyncio.CancelledError:
-                    self._is_running = False
+        try:
+            await running_event.wait()
+        except asyncio.CancelledError:
+            running_event.clear()
 
-            await asyncio.shield(self.close())
+        await asyncio.shield(self.close())
 
     async def close(self):
         """Clean up all connections and resources.

examples/01_simple_agent_example/pyproject.toml

@@ -3,7 +3,7 @@ name = "simple-agents-example"
 version = "0.0.0"
 requires-python = ">=3.13"
 
-# put only what this example needs
+# Dependencies that the example actually uses
 dependencies = [
   "python-dotenv>=1.0",
   "vision-agents-plugins-deepgram",
@@ -19,6 +19,7 @@ dependencies = [
   "anthropic>=0.66.0",
   "google-genai>=1.33.0",
   "fal-client>=0.5.3",
+  "pyinstrument>=5.1.1",
 ]
 
 [tool.uv.sources]
@@ -30,5 +31,4 @@ dependencies = [
 "vision-agents-plugins-smart-turn" = {path = "../../plugins/smart_turn", editable=true}
 "vision-agents-plugins-cartesia" = {path = "../../plugins/cartesia", editable=true}
 "vision-agents-plugins-gemini" = {path = "../../plugins/gemini", editable=true}
-
 "vision-agents" = {path = "../../agents-core", editable=true}

examples/01_simple_agent_example/simple_agent_example.py

@@ -1,6 +1,7 @@
 import asyncio
 from uuid import uuid4
 from dotenv import load_dotenv
+import pyinstrument
 
 from vision_agents.core import User, Agent
 from vision_agents.plugins import cartesia, deepgram, getstream, gemini, vogent
@@ -9,6 +10,8 @@
 
 
 async def start_agent() -> None:
+    profiler = pyinstrument.Profiler()
+    profiler.start()
     llm = gemini.LLM("gemini-2.0-flash")
     # create an agent to run with Stream's edge, openAI llm
     agent = Agent(
@@ -56,6 +59,10 @@ async def start_agent() -> None:
         # run till the call ends
         await agent.finish()
 
+    profiler.stop()
+    with open('profiled.html', 'w') as f:
+        f.write(profiler.output_html())
+
 
 def setup_telemetry():
     import atexit