Pyright

Author: sarth6

pydantic_ai_slim/pydantic_ai/__init__.py

@@ -14,7 +14,7 @@
     ImageGenerationTool,
     MCPServerTool,
     MemoryTool,
-    UrlContextTool,
+    UrlContextTool,  # pyright: ignore[reportDeprecated]
     WebFetchTool,
     WebSearchTool,
     WebSearchUserLocation,

tests/models/test_anthropic.py

@@ -4431,12 +4431,12 @@ async def test_anthropic_web_fetch_tool_with_parameters():
     assert web_fetch_tool_param is not None
 
     # Verify all parameters are passed correctly
-    assert web_fetch_tool_param['type'] == 'web_fetch_20250910'
-    assert web_fetch_tool_param['max_uses'] == 5
-    assert web_fetch_tool_param['allowed_domains'] == ['example.com', 'ai.pydantic.dev']
-    assert web_fetch_tool_param['blocked_domains'] is None
-    assert web_fetch_tool_param['citations'] == {'enabled': True}
-    assert web_fetch_tool_param['max_content_tokens'] == 50000
+    assert web_fetch_tool_param.get('type') == 'web_fetch_20250910'
+    assert web_fetch_tool_param.get('max_uses') == 5
+    assert web_fetch_tool_param.get('allowed_domains') == ['example.com', 'ai.pydantic.dev']
+    assert web_fetch_tool_param.get('blocked_domains') is None
+    assert web_fetch_tool_param.get('citations') == {'enabled': True}
+    assert web_fetch_tool_param.get('max_content_tokens') == 50000
 
 
 async def test_anthropic_web_fetch_tool_domain_filtering():
@@ -4464,8 +4464,8 @@ async def test_anthropic_web_fetch_tool_domain_filtering():
     assert web_fetch_tool_param is not None
 
     # Verify blocked_domains is passed correctly
-    assert web_fetch_tool_param['blocked_domains'] == ['private.example.com', 'internal.example.com']
-    assert web_fetch_tool_param['allowed_domains'] is None
+    assert web_fetch_tool_param.get('blocked_domains') == ['private.example.com', 'internal.example.com']
+    assert web_fetch_tool_param.get('allowed_domains') is None
 
 
 @pytest.mark.vcr()

tests/models/test_google.py

@@ -45,7 +45,7 @@
 from pydantic_ai.builtin_tools import (
     CodeExecutionTool,
     ImageGenerationTool,
-    UrlContextTool,
+    UrlContextTool,  # pyright: ignore[reportDeprecated]
     WebFetchTool,
     WebSearchTool,
 )
@@ -1356,7 +1356,7 @@ async def test_google_model_web_fetch_tool(
 
     if use_deprecated_url_context_tool:
         with pytest.warns(DeprecationWarning, match='Use `WebFetchTool` instead.'):
-            tool = UrlContextTool()
+            tool = UrlContextTool()  # pyright: ignore[reportDeprecated]
     else:
         tool = WebFetchTool()
 

tests/test_builtin_tools.py

@@ -3,7 +3,11 @@
 import pytest
 
 from pydantic_ai.agent import Agent
-from pydantic_ai.builtin_tools import CodeExecutionTool, UrlContextTool, WebSearchTool
+from pydantic_ai.builtin_tools import (  # pyright: ignore[reportDeprecated]
+    CodeExecutionTool,
+    UrlContextTool,
+    WebSearchTool,
+)
 from pydantic_ai.exceptions import UserError
 from pydantic_ai.models import Model
 
@@ -45,4 +49,4 @@ async def test_builtin_tools_not_supported_code_execution_stream(model: Model, a
 def test_url_context_tool_is_deprecated():
     """Test that UrlContextTool is deprecated and warns users to use WebFetchTool instead."""
     with pytest.warns(DeprecationWarning, match='Use `WebFetchTool` instead.'):
-        UrlContextTool()
+        UrlContextTool()  # pyright: ignore[reportDeprecated]