pydantic-ai 0.7.3__tar.gz → 0.7.4__tar.gz

{pydantic_ai-0.7.3 → pydantic_ai-0.7.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.7.3
+Version: 0.7.4
 Summary: Agent Framework / shim to use Pydantic with LLMs
 Project-URL: Homepage, https://ai.pydantic.dev
 Project-URL: Source, https://github.com/pydantic/pydantic-ai
@@ -28,11 +28,11 @@ Classifier: Topic :: Internet
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.9
-Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,bedrock,cli,cohere,evals,google,groq,huggingface,mcp,mistral,openai,retries,temporal,vertexai]==0.7.3
+Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,bedrock,cli,cohere,evals,google,groq,huggingface,mcp,mistral,openai,retries,temporal,vertexai]==0.7.4
 Provides-Extra: a2a
 Requires-Dist: fasta2a>=0.4.1; extra == 'a2a'
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==0.7.3; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==0.7.4; extra == 'examples'
 Provides-Extra: logfire
 Requires-Dist: logfire>=3.14.1; extra == 'logfire'
 Description-Content-Type: text/markdown

pydantic_ai-0.7.4/tests/models/cassettes/test_google/test_google_model_url_context_tool.yaml

@@ -0,0 +1,76 @@
+interactions:
+- request:
+    body: '{"contents": [{"parts": [{"text": "What is the first sentence on the page
+      https://ai.pydantic.dev? Reply with only the sentence."}], "role": "user"}],
+      "systemInstruction": {"parts": [{"text": "You are a helpful chatbot."}], "role":
+      "user"}, "tools": [{"urlContext": {}}], "generationConfig": {}}'
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '295'
+      content-type:
+      - application/json
+      host:
+      - generativelanguage.googleapis.com
+    method: POST
+    uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent
+  response:
+    body:
+      string: "{\n  \"candidates\": [\n    {\n      \"content\": {\n        \"parts\"\
+        : [\n          {\n            \"text\": \"Pydantic AI is a Python agent framework\
+        \ designed to make it less painful to build production grade applications\
+        \ with Generative AI.\"\n          }\n        ],\n        \"role\": \"model\"\
+        \n      },\n      \"finishReason\": \"STOP\",\n      \"index\": 0,\n     \
+        \ \"groundingMetadata\": {\n        \"groundingChunks\": [\n          {\n\
+        \            \"web\": {\n              \"uri\": \"https://ai.pydantic.dev\"\
+        ,\n              \"title\": \"Pydantic AI\"\n            }\n          }\n\
+        \        ],\n        \"groundingSupports\": [\n          {\n            \"\
+        segment\": {\n              \"endIndex\": 131,\n              \"text\": \"\
+        Pydantic AI is a Python agent framework designed to make it less painful to\
+        \ build production grade applications with Generative AI.\"\n            },\n\
+        \            \"groundingChunkIndices\": [\n              0\n            ]\n\
+        \          }\n        ]\n      },\n      \"urlContextMetadata\": {\n     \
+        \   \"urlMetadata\": [\n          {\n            \"retrievedUrl\": \"https://ai.pydantic.dev\"\
+        ,\n            \"urlRetrievalStatus\": \"URL_RETRIEVAL_STATUS_SUCCESS\"\n\
+        \          }\n        ]\n      }\n    }\n  ],\n  \"usageMetadata\": {\n  \
+        \  \"promptTokenCount\": 32,\n    \"candidatesTokenCount\": 41,\n    \"totalTokenCount\"\
+        : 2515,\n    \"promptTokensDetails\": [\n      {\n        \"modality\": \"\
+        TEXT\",\n        \"tokenCount\": 32\n      }\n    ],\n    \"toolUsePromptTokenCount\"\
+        : 2395,\n    \"toolUsePromptTokensDetails\": [\n      {\n        \"modality\"\
+        : \"TEXT\",\n        \"tokenCount\": 2395\n      }\n    ],\n    \"thoughtsTokenCount\"\
+        : 47\n  },\n  \"modelVersion\": \"gemini-2.5-flash\",\n  \"responseId\": \"\
+        qgqkaI-iDLrTjMcP0bP24A4\"\n}\n"
+    headers:
+      Alt-Svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      Content-Type:
+      - application/json; charset=UTF-8
+      Date:
+      - Tue, 19 Aug 2025 05:24:58 GMT
+      Server:
+      - scaffolding on HTTPServer2
+      Server-Timing:
+      - gfet4t7; dur=9182
+      Transfer-Encoding:
+      - chunked
+      Vary:
+      - Origin
+      - X-Origin
+      - Referer
+      X-Content-Type-Options:
+      - nosniff
+      X-Frame-Options:
+      - SAMEORIGIN
+      X-XSS-Protection:
+      - '0'
+      content-length:
+      - '1627'
+    status:
+      code: 200
+      message: OK
+version: 1

{pydantic_ai-0.7.3 → pydantic_ai-0.7.4}/tests/models/cassettes/test_model_names/test_known_model_names.yaml

@@ -12,8 +12,6 @@ interactions:
     uri: https://us.inference.heroku.com/available-models
   response:
     headers:
-      connection:
-      - keep-alive
       content-length:
       - '760'
       content-security-policy:

{pydantic_ai-0.7.3 → pydantic_ai-0.7.4}/tests/models/test_google.py

@@ -12,7 +12,7 @@ from typing_extensions import TypedDict
 
 from pydantic_ai import UsageLimitExceeded
 from pydantic_ai.agent import Agent
-from pydantic_ai.builtin_tools import CodeExecutionTool, WebSearchTool
+from pydantic_ai.builtin_tools import CodeExecutionTool, UrlContextTool, WebSearchTool
 from pydantic_ai.exceptions import ModelRetry, UnexpectedModelBehavior, UserError
 from pydantic_ai.messages import (
     AudioUrl,
@@ -600,6 +600,19 @@ async def test_google_model_web_search_tool(allow_model_requests: None, google_p
     assert result.output == snapshot('Today is Wednesday, May 28, 2025, in Utrecht.\n')
 
 
+async def test_google_model_url_context_tool(allow_model_requests: None, google_provider: GoogleProvider):
+    m = GoogleModel('gemini-2.5-flash', provider=google_provider)
+    agent = Agent(m, system_prompt='You are a helpful chatbot.', builtin_tools=[UrlContextTool()])
+
+    result = await agent.run(
+        'What is the first sentence on the page https://ai.pydantic.dev? Reply with only the sentence.'
+    )
+
+    assert result.output == snapshot(
+        'Pydantic AI is a Python agent framework designed to make it less painful to build production grade applications with Generative AI.'
+    )
+
+
 async def test_google_model_code_execution_tool(allow_model_requests: None, google_provider: GoogleProvider):
     m = GoogleModel('gemini-2.0-flash', provider=google_provider)
     agent = Agent(m, system_prompt='You are a helpful chatbot.', builtin_tools=[CodeExecutionTool()])
@@ -1657,7 +1670,9 @@ Just a small note, it's typically written as "lazy dog" (two words) and usually
 """)
 
 
-async def test_google_vertexai_model_usage_limit_exceeded(allow_model_requests: None, vertex_provider: GoogleProvider):
+async def test_google_vertexai_model_usage_limit_exceeded(
+    allow_model_requests: None, vertex_provider: GoogleProvider
+):  # pragma: lax no cover
     model = GoogleModel('gemini-2.0-flash', provider=vertex_provider, settings=ModelSettings(max_tokens=100))
 
     agent = Agent(model, system_prompt='You are a chatbot.')

{pydantic_ai-0.7.3 → pydantic_ai-0.7.4}/tests/test_examples.py

@@ -304,6 +304,7 @@ text_responses: dict[str, str | ToolCallPart] = {
     'Who was Albert Einstein?': 'Albert Einstein was a German-born theoretical physicist.',
     'What was his most famous equation?': "Albert Einstein's most famous equation is (E = mc^2).",
     'What is the date?': 'Hello Frank, the date today is 2032-01-02.',
+    'What is this? https://ai.pydantic.dev': 'A Python agent framework for building Generative AI applications.',
     'Put my money on square eighteen': ToolCallPart(
         tool_name='roulette_wheel', args={'square': 18}, tool_call_id='pyd_ai_tool_call_id'
     ),

{pydantic_ai-0.7.3 → pydantic_ai-0.7.4}/tests/test_tools.py

@@ -1150,6 +1150,35 @@ def test_function_tool_consistent_with_schema():
     assert agent._function_toolset.tools['foobar'].max_retries == 0
 
 
+def test_function_tool_from_schema_with_ctx():
+    def function(ctx: RunContext[str], *args: Any, **kwargs: Any) -> str:
+        assert len(args) == 0
+        assert set(kwargs) == {'one', 'two'}
+        return ctx.deps + 'I like being called like this'
+
+    json_schema = {
+        'type': 'object',
+        'additionalProperties': False,
+        'properties': {
+            'one': {'description': 'first argument', 'type': 'string'},
+            'two': {'description': 'second argument', 'type': 'object'},
+        },
+        'required': ['one', 'two'],
+    }
+    pydantic_tool = Tool[str].from_schema(
+        function, name='foobar', description='does foobar stuff', json_schema=json_schema, takes_ctx=True
+    )
+
+    assert pydantic_tool.takes_ctx is True
+    assert pydantic_tool.function_schema.takes_ctx is True
+
+    agent = Agent('test', tools=[pydantic_tool], retries=0, deps_type=str)
+    result = agent.run_sync('foobar', deps='Hello, ')
+    assert result.output == snapshot('{"foobar":"Hello, I like being called like this"}')
+    assert agent._function_toolset.tools['foobar'].takes_ctx is True
+    assert agent._function_toolset.tools['foobar'].max_retries == 0
+
+
 def test_function_tool_inconsistent_with_schema():
     def function(three: str, four: int) -> str:
         return 'Coverage made me call this'