Fix mypy errors: SpeakV1Text type param, turbopuffer RowParam typing

Co-authored-by: Neevash Ramdial (Nash) <mail@neevash.com>

Author: cursoragent

plugins/deepgram/vision_agents/plugins/deepgram/tts.py

@@ -102,13 +102,13 @@ async def stream_audio(self, text: str, *_, **__) -> AsyncIterator[PcmData]:
         self._stop_event.clear()
 
         try:
-            await socket.send_text(SpeakV1Text(text=text))
+            await socket.send_text(SpeakV1Text(type="Speak", text=text))
             await socket.send_flush()
         except (websockets.exceptions.ConnectionClosed, ConnectionError):
             logger.warning("Deepgram TTS websocket dropped, reconnecting")
             await self._reset_connection()
             socket = await self._ensure_connection()
-            await socket.send_text(SpeakV1Text(text=text))
+            await socket.send_text(SpeakV1Text(type="Speak", text=text))
             await socket.send_flush()
 
         self._generation += 1

plugins/turbopuffer/vision_agents/plugins/turbopuffer/turbopuffer_rag.py

@@ -46,6 +46,7 @@
 from langchain_google_genai import GoogleGenerativeAIEmbeddings
 from langchain_text_splitters import RecursiveCharacterTextSplitter
 from turbopuffer import AsyncTurbopuffer, NotFoundError
+from turbopuffer.types import RowParam
 
 from vision_agents.core.rag import RAG, Document
 
@@ -181,19 +182,18 @@ async def add_documents(self, documents: list[Document]) -> int:
             None, self._embeddings.embed_documents, all_chunks
         )
 
-        rows = []
+        rows: list[RowParam] = []
         for chunk, embedding, (source, idx) in zip(
             all_chunks, embeddings, chunk_sources
         ):
-            rows.append(
-                {
-                    "id": f"{source}_{idx}",
-                    "vector": embedding,
-                    "text": chunk,
-                    "source": source,
-                    "chunk_index": idx,
-                }
-            )
+            row: RowParam = {
+                "id": f"{source}_{idx}",
+                "vector": embedding,
+                "text": chunk,
+                "source": source,
+                "chunk_index": idx,
+            }
+            rows.append(row)
 
         ns = self._client.namespace(self._namespace_name)
         await ns.write(