perf: pyinstrument integration

Author: yarikdevcom

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

@@ -424,20 +424,22 @@ 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:
             logging.info("🔚 Agent connection already closed, finishing immediately")
             return
 
+        running_event = asyncio.Event()
+
         @self.edge.events.subscribe
         async def on_ended(event: CallEndedEvent):
-            self._is_running = False
+            running_event.set()
 
-        while self._is_running:
-            try:
-                await asyncio.sleep(0.0001)
-            except asyncio.CancelledError:
-                self._is_running = False
+        # TODO: add members count check (particiapnts left + count = 1 timeout 2 minutes)
+
+        try:
+            await running_event.wait()
+        except asyncio.CancelledError:
+            running_event.clear()
 
         await asyncio.shield(self.close())
 

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
@@ -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())
+
 
 if __name__ == "__main__":
     asyncio.run(start_agent())