fix type errors in FileSearchTool integration tests

- Add type assertions for store.name (cannot be None)
- Initialize file/vector_store/store to None before try blocks
- Update cleanup checks from 'in locals()' to 'is not None'
- Simplify streaming tests to use stream_text() instead of raw events
- All pyright type checks now pass on changed files

Author: gorkachea

tests/models/test_google.py

@@ -3209,38 +3209,39 @@ async def test_google_model_file_search_tool(allow_model_requests: None, google_
     from pydantic_ai.builtin_tools import FileSearchTool
 
     client = google_provider.client
-    
+
     import tempfile
+
     with tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False) as f:
         f.write('Paris is the capital of France. The Eiffel Tower is a famous landmark in Paris.')
         test_file_path = f.name
-    
+
+    store = None
     try:
-        store = await client.aio.file_search_stores.create(
-            config={'display_name': 'test-file-search-store'}
-        )
-        
+        store = await client.aio.file_search_stores.create(config={'display_name': 'test-file-search-store'})
+        assert store.name is not None
+
         with open(test_file_path, 'rb') as f:
             await client.aio.file_search_stores.upload_to_file_search_store(
-                file_search_store_name=store.name,
-                file=f,
-                config={'mime_type': 'text/plain'}
+                file_search_store_name=store.name, file=f, config={'mime_type': 'text/plain'}
             )
-        
+
         import asyncio
+
         await asyncio.sleep(3)
-        
+
         model = GoogleModel('gemini-2.5-pro', provider=google_provider)
         agent = Agent(model=model, builtin_tools=[FileSearchTool(vector_store_ids=[store.name])])
-        
+
         result = await agent.run('What is the capital of France according to my files?')
-        
+
         assert 'Paris' in result.output or 'paris' in result.output.lower()
-        
+
     finally:
         import os
+
         os.unlink(test_file_path)
-        if 'store' in locals():
+        if store is not None and store.name is not None:
             await client.aio.file_search_stores.delete(name=store.name, config={'force': True})
 
 
@@ -3250,46 +3251,44 @@ async def test_google_model_file_search_tool_stream(allow_model_requests: None,
     from pydantic_ai.builtin_tools import FileSearchTool
 
     client = google_provider.client
-    
+
     import tempfile
+
     with tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False) as f:
         f.write('The Louvre Museum is located in Paris, France. It houses the Mona Lisa.')
         test_file_path = f.name
-    
+
+    store = None
     try:
-        store = await client.aio.file_search_stores.create(
-            config={'display_name': 'test-file-search-stream'}
-        )
-        
+        store = await client.aio.file_search_stores.create(config={'display_name': 'test-file-search-stream'})
+        assert store.name is not None
+
         with open(test_file_path, 'rb') as f:
             await client.aio.file_search_stores.upload_to_file_search_store(
-                file_search_store_name=store.name,
-                file=f,
-                config={'mime_type': 'text/plain'}
+                file_search_store_name=store.name, file=f, config={'mime_type': 'text/plain'}
             )
-        
+
         import asyncio
+
         await asyncio.sleep(3)
-        
+
         model = GoogleModel('gemini-2.5-pro', provider=google_provider)
         agent = Agent(model=model, builtin_tools=[FileSearchTool(vector_store_ids=[store.name])])
-        
-        events = []
-        async with agent.iter(user_prompt='Where is the Louvre Museum according to my files?') as agent_run:
-            async for node in agent_run:
-                if Agent.is_model_request_node(node):
-                    async with node.stream(agent_run.ctx) as stream:
-                        async for event in stream:
-                            events.append(event)
-        
-        assert len(events) > 0
-        output = ''.join(str(e.part.content) if hasattr(e, 'part') and hasattr(e.part, 'content') else '' for e in events)
+
+        result_text: list[str] = []
+        async with agent.run_stream('Where is the Louvre Museum according to my files?') as result:
+            async for text in result.stream_text(delta=False):
+                result_text.append(text)
+            output = await result.get_output()
+
+        assert len(result_text) > 0
         assert 'Paris' in output or 'France' in output or 'Louvre' in output or 'paris' in output.lower()
-        
+
     finally:
         import os
+
         os.unlink(test_file_path)
-        if 'store' in locals():
+        if store is not None and store.name is not None:
             await client.aio.file_search_stores.delete(name=store.name, config={'force': True})
 
 

tests/models/test_openai_responses.py

@@ -7445,35 +7445,40 @@ async def test_openai_responses_model_file_search_tool(allow_model_requests: Non
     from pydantic_ai.providers.openai import OpenAIProvider
 
     async_client = AsyncOpenAI(api_key=openai_api_key)
-    
+
     import tempfile
+
     with tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False) as f:
         f.write('Paris is the capital of France. It is known for the Eiffel Tower.')
         test_file_path = f.name
-    
+
+    file = None
+    vector_store = None
     try:
         with open(test_file_path, 'rb') as f:
             file = await async_client.files.create(file=f, purpose='assistants')
-        
+
         vector_store = await async_client.vector_stores.create(name='test-file-search')
         await async_client.vector_stores.files.create(vector_store_id=vector_store.id, file_id=file.id)
-        
+
         import asyncio
+
         await asyncio.sleep(2)
-        
+
         model = OpenAIResponsesModel('gpt-4o', provider=OpenAIProvider(openai_client=async_client))
         agent = Agent(model=model, builtin_tools=[FileSearchTool(vector_store_ids=[vector_store.id])])
-        
+
         result = await agent.run('What is the capital of France according to my files?')
-        
+
         assert 'Paris' in result.output or 'paris' in result.output.lower()
-        
+
     finally:
         import os
+
         os.unlink(test_file_path)
-        if 'file' in locals():
+        if file is not None:
             await async_client.files.delete(file.id)
-        if 'vector_store' in locals():
+        if vector_store is not None:
             await async_client.vector_stores.delete(vector_store.id)
         await async_client.close()
 
@@ -7487,40 +7492,45 @@ async def test_openai_responses_model_file_search_tool_stream(allow_model_reques
     from pydantic_ai.providers.openai import OpenAIProvider
 
     async_client = AsyncOpenAI(api_key=openai_api_key)
-    
+
     import tempfile
+
     with tempfile.NamedTemporaryFile(mode='w', suffix='.txt', delete=False) as f:
         f.write('The Eiffel Tower is located in Paris, France.')
         test_file_path = f.name
-    
+
+    file = None
+    vector_store = None
     try:
         with open(test_file_path, 'rb') as f:
             file = await async_client.files.create(file=f, purpose='assistants')
-        
+
         vector_store = await async_client.vector_stores.create(name='test-file-search-stream')
         await async_client.vector_stores.files.create(vector_store_id=vector_store.id, file_id=file.id)
-        
+
         import asyncio
+
         await asyncio.sleep(2)
-        
+
         model = OpenAIResponsesModel('gpt-4o', provider=OpenAIProvider(openai_client=async_client))
         agent = Agent(model=model, builtin_tools=[FileSearchTool(vector_store_ids=[vector_store.id])])
-        
-        parts = []
+
+        result_text: list[str] = []
         async with agent.run_stream('Where is the Eiffel Tower according to my files?') as result:
-            async for part in result.stream_responses(debounce_by=None):
-                parts.append(part)
+            async for text in result.stream_text(delta=False):
+                result_text.append(text)
             output = await result.get_output()
-        
-        assert len(parts) > 0
+
+        assert len(result_text) > 0
         assert 'Paris' in output or 'France' in output or 'paris' in output.lower()
-        
+
     finally:
         import os
+
         os.unlink(test_file_path)
-        if 'file' in locals():
+        if file is not None:
             await async_client.files.delete(file.id)
-        if 'vector_store' in locals():
+        if vector_store is not None:
             await async_client.vector_stores.delete(vector_store.id)
         await async_client.close()