some cleanup

Author: tschellenbach

examples/03_phone_and_rag_example/conftest.py

@@ -1,5 +1,4 @@
 """Pytest configuration for phone example tests."""
 
-import pytest
 
 # Override root conftest - we don't need blockbuster for these tests

examples/03_phone_and_rag_example/test_rag_turbopuffer.py

@@ -1,5 +1,6 @@
 """Tests for TurboPuffer Hybrid RAG."""
 
+import logging
 from pathlib import Path
 
 import pytest
@@ -8,7 +9,6 @@
 from rag_turbopuffer import TurboPufferRAG
 
 load_dotenv()
-import logging
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)
@@ -31,6 +31,5 @@ async def test_index_and_search(knowledge_dir: Path):
     result = await rag.search("chat messaging")
     assert "Chat" in result
     logger.info("result %s", result)
-    import pdb; pdb.set_trace()
 
     await rag.close()

plugins/gemini/vision_agents/plugins/gemini/file_search.py

@@ -86,6 +86,7 @@ async def create(self) -> str:
         )
         self._store_name = store.name
         logger.info(f"Created FileSearchStore '{self.name}': {self._store_name}")
+        assert self._store_name is not None
         return self._store_name
 
     async def upload_file(self, file_path: str | Path, display_name: Optional[str] = None) -> None:
@@ -99,6 +100,8 @@ async def upload_file(self, file_path: str | Path, display_name: Optional[str] =
         if not self._store_name:
             raise ValueError("Store not created. Call create() first.")
 
+        store_name = self._store_name
+        
         file_path = Path(file_path)
         if not file_path.exists():
             raise FileNotFoundError(f"File not found: {file_path}")
@@ -112,7 +115,7 @@ async def upload_file(self, file_path: str | Path, display_name: Optional[str] =
             None,
             lambda: self._client.file_search_stores.upload_to_file_search_store(
                 file=str(file_path),
-                file_search_store_name=self._store_name,
+                file_search_store_name=store_name,
                 config={"display_name": display_name}
             )
         )

plugins/gemini/vision_agents/plugins/gemini/gemini_realtime.py

@@ -353,7 +353,7 @@ def get_config(self) -> LiveConnectConfigDict:
         if self.file_search_store and self.file_search_store.is_created:
             file_search_config = self.file_search_store.get_tool_config()
             tools.append(file_search_config)
-            logger.info(f"Adding file search tool to Gemini Live config")
+            logger.info("Adding file search tool to Gemini Live config")
 
         if tools:
             config["tools"] = tools  # type: ignore[typeddict-item]

plugins/twilio/pyproject.toml

@@ -13,6 +13,7 @@ dependencies = [
     "vision-agents",
     "twilio>=9.0.0",
     "numpy>=1.24.0",
+    "fastapi>=0.100.0",
 ]
 
 [project.urls]

plugins/twilio/tests/test_audio.py

@@ -1,7 +1,6 @@
 """Comprehensive tests for mulaw/PCM audio conversion."""
 
 import numpy as np
-import pytest
 from getstream.video.rtc.track_util import PcmData, AudioFormat
 
 from vision_agents.plugins.twilio.audio import (
@@ -131,7 +130,7 @@ def test_encode_matches_reference(self):
                     f"PCM {pcm_val}: ours={our_mulaw:#04x}, expected={expected_mulaw:#04x}"
                 )
 
-        assert not mismatches, f"Encoding mismatches:\n" + "\n".join(mismatches)
+        assert not mismatches, "Encoding mismatches:\n" + "\n".join(mismatches)
 
     def test_encode_symmetry(self):
         """Positive and negative values should encode symmetrically."""

plugins/twilio/tests/test_twilio.py

@@ -1,4 +1,3 @@
-import pytest
 from datetime import datetime
 from dotenv import load_dotenv
 
@@ -90,7 +89,6 @@ class TestAudioConversion:
     def test_mulaw_to_pcm(self):
         """Test mulaw to PCM conversion."""
         # Create some test mulaw bytes
-        import numpy as np
         mulaw_bytes = bytes([0xFF, 0x7F, 0x00, 0x80])  # Various mulaw values
 
         pcm = twilio.mulaw_to_pcm(mulaw_bytes)

plugins/twilio/vision_agents/plugins/twilio/audio.py

@@ -64,11 +64,11 @@ def _build_mulaw_encode_table() -> np.ndarray:
 
     # Fallback: Try audioop-lts (available for Python >= 3.13)
     try:
-        import audioop_lts as audioop
+        import audioop_lts
 
         for i in range(65536):
             pcm_bytes = i.to_bytes(2, byteorder="little", signed=False)
-            table[i] = audioop.lin2ulaw(pcm_bytes, 2)[0]
+            table[i] = audioop_lts.lin2ulaw(pcm_bytes, 2)[0]
         return table
 
     except ImportError:

plugins/twilio/vision_agents/plugins/twilio/call_registry.py

@@ -5,7 +5,7 @@
 import secrets
 from dataclasses import dataclass, field
 from datetime import datetime
-from typing import Any, Awaitable, Callable, Optional, TYPE_CHECKING, TypeVar
+from typing import Any, Callable, Coroutine, Optional, TYPE_CHECKING, TypeVar
 
 if TYPE_CHECKING:
     from .media_stream import TwilioMediaStream
@@ -119,7 +119,7 @@ def create(
         self,
         call_id: str,
         webhook_data: Optional["CallWebhookInput"] = None,
-        prepare: Optional[Callable[[], Awaitable[Any]]] = None,
+        prepare: Optional[Callable[[], Coroutine[Any, Any, Any]]] = None,
     ) -> TwilioCall:
         """
         Create and register a new TwilioCall.