pydantic-ai 0.2.13__tar.gz → 0.2.15__tar.gz

{pydantic_ai-0.2.13 → pydantic_ai-0.2.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.2.13
+Version: 0.2.15
 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,9 +28,9 @@ 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[a2a,anthropic,bedrock,cli,cohere,evals,google,groq,mcp,mistral,openai,vertexai]==0.2.13
+Requires-Dist: pydantic-ai-slim[a2a,anthropic,bedrock,cli,cohere,evals,google,groq,mcp,mistral,openai,vertexai]==0.2.15
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==0.2.13; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==0.2.15; extra == 'examples'
 Provides-Extra: logfire
 Requires-Dist: logfire>=3.11.0; extra == 'logfire'
 Description-Content-Type: text/markdown

pydantic_ai-0.2.15/tests/models/cassettes/test_google/test_google_model_empty_user_prompt.yaml

@@ -0,0 +1,68 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '173'
+      content-type:
+      - application/json
+      host:
+      - generativelanguage.googleapis.com
+    method: POST
+    parsed_body:
+      contents:
+      - parts:
+        - text: ''
+        role: user
+      generationConfig: {}
+      systemInstruction:
+        parts:
+        - text: You are a helpful assistant.
+        role: user
+    uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent
+  response:
+    headers:
+      alt-svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      content-length:
+      - '736'
+      content-type:
+      - application/json; charset=UTF-8
+      server-timing:
+      - gfet4t7; dur=361
+      transfer-encoding:
+      - chunked
+      vary:
+      - Origin
+      - X-Origin
+      - Referer
+    parsed_body:
+      candidates:
+      - avgLogprobs: -0.20726837430681502
+        content:
+          parts:
+          - text: |
+              Please provide me with a question or task. I need some information to be able to help you.
+          role: model
+        finishReason: STOP
+      modelVersion: gemini-1.5-flash
+      responseId: y5BBaPCQELGqmecP68OBiAc
+      usageMetadata:
+        candidatesTokenCount: 21
+        candidatesTokensDetails:
+        - modality: TEXT
+          tokenCount: 21
+        promptTokenCount: 7
+        promptTokensDetails:
+        - modality: TEXT
+          tokenCount: 7
+        totalTokenCount: 28
+    status:
+      code: 200
+      message: OK
+version: 1

{pydantic_ai-0.2.13 → pydantic_ai-0.2.15}/tests/models/test_google.py

@@ -581,3 +581,13 @@ async def test_google_model_safety_settings(allow_model_requests: None, google_p
 
     with pytest.raises(UnexpectedModelBehavior, match='Safety settings triggered'):
         await agent.run('Tell me a joke about a Brazilians.')
+
+
+async def test_google_model_empty_user_prompt(allow_model_requests: None, google_provider: GoogleProvider):
+    m = GoogleModel('gemini-1.5-flash', provider=google_provider)
+    agent = Agent(m, instructions='You are a helpful assistant.')
+
+    result = await agent.run()
+    assert result.output == snapshot(
+        'Please provide me with a question or task. I need some information to be able to help you.\n'
+    )

{pydantic_ai-0.2.13 → pydantic_ai-0.2.15}/tests/test_mcp.py

@@ -1,7 +1,6 @@
 """Tests for the MCP (Model Context Protocol) server implementation."""
 
 import re
-from datetime import timedelta
 from pathlib import Path
 
 import pytest
@@ -71,40 +70,25 @@ async def test_stdio_server_with_cwd():
         assert len(tools) == 10
 
 
-def test_http_server():
-    http_server = MCPServerHTTP(url='http://localhost:8000/sse')
-    assert http_server.url == 'http://localhost:8000/sse'
-    assert http_server._get_log_level() is None  # pyright: ignore[reportPrivateUsage]
+def test_sse_server():
+    sse_server = MCPServerHTTP(url='http://localhost:8000/sse')
+    assert sse_server.url == 'http://localhost:8000/sse'
+    assert sse_server._get_log_level() is None  # pyright: ignore[reportPrivateUsage]
 
 
-def test_http_server_with_header_and_timeout():
-    http_server = MCPServerHTTP(
+def test_sse_server_with_header_and_timeout():
+    sse_server = MCPServerHTTP(
         url='http://localhost:8000/sse',
         headers={'my-custom-header': 'my-header-value'},
         timeout=10,
         sse_read_timeout=100,
         log_level='info',
     )
-    assert http_server.url == 'http://localhost:8000/sse'
-    assert http_server.headers is not None and http_server.headers['my-custom-header'] == 'my-header-value'
-    assert http_server.timeout == 10
-    assert http_server.sse_read_timeout == 100
-    assert http_server._get_log_level() == 'info'  # pyright: ignore[reportPrivateUsage]
-
-
-def test_http_server_with_timedelta_arguments():
-    http_server = MCPServerHTTP(
-        url='http://localhost:8000/sse',
-        headers={'my-custom-header': 'my-header-value'},
-        timeout=timedelta(seconds=10),  # type: ignore[arg-type]
-        sse_read_timeout=timedelta(seconds=100),  # type: ignore[arg-type]
-        log_level='info',
-    )
-    assert http_server.url == 'http://localhost:8000/sse'
-    assert http_server.headers is not None and http_server.headers['my-custom-header'] == 'my-header-value'
-    assert http_server.timeout == 10
-    assert http_server.sse_read_timeout == 100
-    assert http_server._get_log_level() == 'info'  # pyright: ignore[reportPrivateUsage]
+    assert sse_server.url == 'http://localhost:8000/sse'
+    assert sse_server.headers is not None and sse_server.headers['my-custom-header'] == 'my-header-value'
+    assert sse_server.timeout == 10
+    assert sse_server.sse_read_timeout == 100
+    assert sse_server._get_log_level() == 'info'  # pyright: ignore[reportPrivateUsage]
 
 
 @pytest.mark.vcr()