pydantic-ai-slim 0.2.8__tar.gz → 0.2.10__tar.gz

{pydantic_ai_slim-0.2.8 → pydantic_ai_slim-0.2.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai-slim
-Version: 0.2.8
+Version: 0.2.10
 Summary: Agent Framework / shim to use Pydantic with LLMs, slim package
 Author-email: Samuel Colvin <samuel@pydantic.dev>, Marcelo Trylesinski <marcelotryle@gmail.com>, David Montague <david@pydantic.dev>, Alex Hall <alex@pydantic.dev>
 License-Expression: MIT
@@ -30,13 +30,13 @@ Requires-Dist: exceptiongroup; python_version < '3.11'
 Requires-Dist: griffe>=1.3.2
 Requires-Dist: httpx>=0.27
 Requires-Dist: opentelemetry-api>=1.28.0
-Requires-Dist: pydantic-graph==0.2.8
+Requires-Dist: pydantic-graph==0.2.10
 Requires-Dist: pydantic>=2.10
 Requires-Dist: typing-inspection>=0.4.0
 Provides-Extra: a2a
-Requires-Dist: fasta2a==0.2.8; extra == 'a2a'
+Requires-Dist: fasta2a==0.2.10; extra == 'a2a'
 Provides-Extra: anthropic
-Requires-Dist: anthropic>=0.49.0; extra == 'anthropic'
+Requires-Dist: anthropic>=0.52.0; extra == 'anthropic'
 Provides-Extra: bedrock
 Requires-Dist: boto3>=1.35.74; extra == 'bedrock'
 Provides-Extra: cli
@@ -48,7 +48,7 @@ Requires-Dist: cohere>=5.13.11; (platform_system != 'Emscripten') and extra == '
 Provides-Extra: duckduckgo
 Requires-Dist: duckduckgo-search>=7.0.0; extra == 'duckduckgo'
 Provides-Extra: evals
-Requires-Dist: pydantic-evals==0.2.8; extra == 'evals'
+Requires-Dist: pydantic-evals==0.2.10; extra == 'evals'
 Provides-Extra: google
 Requires-Dist: google-genai>=1.15.0; extra == 'google'
 Provides-Extra: groq
@@ -56,7 +56,7 @@ Requires-Dist: groq>=0.15.0; extra == 'groq'
 Provides-Extra: logfire
 Requires-Dist: logfire>=3.11.0; extra == 'logfire'
 Provides-Extra: mcp
-Requires-Dist: mcp>=1.6.0; (python_version >= '3.10') and extra == 'mcp'
+Requires-Dist: mcp>=1.8.0; (python_version >= '3.10') and extra == 'mcp'
 Provides-Extra: mistral
 Requires-Dist: mistralai>=1.2.5; extra == 'mistral'
 Provides-Extra: openai

{pydantic_ai_slim-0.2.8 → pydantic_ai_slim-0.2.10}/pydantic_ai/_cli.py

@@ -284,7 +284,7 @@ async def ask_agent(
                         stack.enter_context(live)  # entering multiple times is idempotent
 
                         async for content in handle_stream.stream_output(debounce_by=None):
-                            live.update(Markdown(content, code_theme=code_theme))
+                            live.update(Markdown(str(content), code_theme=code_theme))
 
         assert agent_run.result is not None
         return agent_run.result.all_messages()

{pydantic_ai_slim-0.2.8 → pydantic_ai_slim-0.2.10}/pydantic_ai/mcp.py

@@ -6,16 +6,18 @@ from abc import ABC, abstractmethod
 from collections.abc import AsyncIterator, Sequence
 from contextlib import AsyncExitStack, asynccontextmanager
 from dataclasses import dataclass
+from datetime import timedelta
 from pathlib import Path
 from types import TracebackType
 from typing import Any
 
+import anyio
 from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
+from mcp.shared.message import SessionMessage
 from mcp.types import (
     BlobResourceContents,
     EmbeddedResource,
     ImageContent,
-    JSONRPCMessage,
     LoggingLevel,
     TextContent,
     TextResourceContents,
@@ -28,8 +30,8 @@ from pydantic_ai.tools import ToolDefinition
 
 try:
     from mcp.client.session import ClientSession
-    from mcp.client.sse import sse_client
     from mcp.client.stdio import StdioServerParameters, stdio_client
+    from mcp.client.streamable_http import streamablehttp_client
 except ImportError as _import_error:
     raise ImportError(
         'Please install the `mcp` package to use the MCP server, '
@@ -55,8 +57,8 @@ class MCPServer(ABC):
     """
 
     _client: ClientSession
-    _read_stream: MemoryObjectReceiveStream[JSONRPCMessage | Exception]
-    _write_stream: MemoryObjectSendStream[JSONRPCMessage]
+    _read_stream: MemoryObjectReceiveStream[SessionMessage | Exception]
+    _write_stream: MemoryObjectSendStream[SessionMessage]
     _exit_stack: AsyncExitStack
 
     @abstractmethod
@@ -64,10 +66,7 @@ class MCPServer(ABC):
     async def client_streams(
         self,
     ) -> AsyncIterator[
-        tuple[
-            MemoryObjectReceiveStream[JSONRPCMessage | Exception],
-            MemoryObjectSendStream[JSONRPCMessage],
-        ]
+        tuple[MemoryObjectReceiveStream[SessionMessage | Exception], MemoryObjectSendStream[SessionMessage]]
     ]:
         """Create the streams for the MCP server."""
         raise NotImplementedError('MCP Server subclasses must implement this method.')
@@ -78,6 +77,9 @@ class MCPServer(ABC):
         """Get the log level for the MCP server."""
         raise NotImplementedError('MCP Server subclasses must implement this method.')
 
+    def _get_client_initialize_timeout(self) -> float:
+        return 5  # pragma: no cover
+
     def get_prefixed_tool_name(self, tool_name: str) -> str:
         """Get the tool name with prefix if `tool_prefix` is set."""
         return f'{self.tool_prefix}_{tool_name}' if self.tool_prefix else tool_name
@@ -137,7 +139,9 @@ class MCPServer(ABC):
         client = ClientSession(read_stream=self._read_stream, write_stream=self._write_stream)
         self._client = await self._exit_stack.enter_async_context(client)
 
-        await self._client.initialize()
+        with anyio.fail_after(self._get_client_initialize_timeout()):
+            await self._client.initialize()
+
         if log_level := self._get_log_level():
             await self._client.set_logging_level(log_level)
         self.is_running = True
@@ -252,14 +256,14 @@ class MCPServerStdio(MCPServer):
     e.g. if `tool_prefix='foo'`, then a tool named `bar` will be registered as `foo_bar`
     """
 
+    timeout: float = 5
+    """ The timeout in seconds to wait for the client to initialize."""
+
     @asynccontextmanager
     async def client_streams(
         self,
     ) -> AsyncIterator[
-        tuple[
-            MemoryObjectReceiveStream[JSONRPCMessage | Exception],
-            MemoryObjectSendStream[JSONRPCMessage],
-        ]
+        tuple[MemoryObjectReceiveStream[SessionMessage | Exception], MemoryObjectSendStream[SessionMessage]]
     ]:
         server = StdioServerParameters(command=self.command, args=list(self.args), env=self.env, cwd=self.cwd)
         async with stdio_client(server=server) as (read_stream, write_stream):
@@ -271,16 +275,19 @@ class MCPServerStdio(MCPServer):
     def __repr__(self) -> str:
         return f'MCPServerStdio(command={self.command!r}, args={self.args!r}, tool_prefix={self.tool_prefix!r})'
 
+    def _get_client_initialize_timeout(self) -> float:
+        return self.timeout
+
 
 @dataclass
 class MCPServerHTTP(MCPServer):
     """An MCP server that connects over streamable HTTP connections.
 
-    This class implements the SSE transport from the MCP specification.
-    See <https://spec.modelcontextprotocol.io/specification/2024-11-05/basic/transports/#http-with-sse> for more information.
+    This class implements the Streamable HTTP transport from the MCP specification.
+    See <https://modelcontextprotocol.io/specification/2025-03-26/basic/transports#streamable-http> for more information.
 
-    The name "HTTP" is used since this implemented will be adapted in future to use the new
-    [Streamable HTTP](https://github.com/modelcontextprotocol/specification/pull/206) currently in development.
+    The Streamable HTTP transport is intended to replace the SSE transport from the previous protocol, but it is fully
+    backwards compatible with SSE-based servers.
 
     !!! note
         Using this class as an async context manager will create a new pool of HTTP connections to connect
@@ -291,7 +298,7 @@ class MCPServerHTTP(MCPServer):
     from pydantic_ai import Agent
     from pydantic_ai.mcp import MCPServerHTTP
 
-    server = MCPServerHTTP('http://localhost:3001/sse')  # (1)!
+    server = MCPServerHTTP('http://localhost:3001/mcp')  # (1)!
     agent = Agent('openai:gpt-4o', mcp_servers=[server])
 
     async def main():
@@ -304,27 +311,27 @@ class MCPServerHTTP(MCPServer):
     """
 
     url: str
-    """The URL of the SSE endpoint on the MCP server.
+    """The URL of the SSE or MCP endpoint on the MCP server.
 
-    For example for a server running locally, this might be `http://localhost:3001/sse`.
+    For example for a server running locally, this might be `http://localhost:3001/mcp`.
     """
 
     headers: dict[str, Any] | None = None
-    """Optional HTTP headers to be sent with each request to the SSE endpoint.
+    """Optional HTTP headers to be sent with each request to the endpoint.
 
     These headers will be passed directly to the underlying `httpx.AsyncClient`.
     Useful for authentication, custom headers, or other HTTP-specific configurations.
     """
 
     timeout: float = 5
-    """Initial connection timeout in seconds for establishing the SSE connection.
+    """Initial connection timeout in seconds for establishing the connection.
 
     This timeout applies to the initial connection setup and handshake.
     If the connection cannot be established within this time, the operation will fail.
     """
 
-    sse_read_timeout: float = 60 * 5
-    """Maximum time in seconds to wait for new SSE messages before timing out.
+    sse_read_timeout: float = 300
+    """Maximum time as in seconds to wait for new SSE messages before timing out.
 
     This timeout applies to the long-lived SSE connection after it's established.
     If no new messages are received within this time, the connection will be considered stale
@@ -346,21 +353,28 @@ class MCPServerHTTP(MCPServer):
     For example, if `tool_prefix='foo'`, then a tool named `bar` will be registered as `foo_bar`
     """
 
+    def __post_init__(self):
+        # streamablehttp_client expects timedeltas, so we accept them too to match,
+        # but primarily work with floats for a simpler user API.
+
+        if isinstance(self.timeout, timedelta):
+            self.timeout = self.timeout.total_seconds()
+
+        if isinstance(self.sse_read_timeout, timedelta):
+            self.sse_read_timeout = self.sse_read_timeout.total_seconds()
+
     @asynccontextmanager
     async def client_streams(
         self,
     ) -> AsyncIterator[
-        tuple[
-            MemoryObjectReceiveStream[JSONRPCMessage | Exception],
-            MemoryObjectSendStream[JSONRPCMessage],
-        ]
+        tuple[MemoryObjectReceiveStream[SessionMessage | Exception], MemoryObjectSendStream[SessionMessage]]
     ]:  # pragma: no cover
-        async with sse_client(
+        async with streamablehttp_client(
             url=self.url,
             headers=self.headers,
-            timeout=self.timeout,
-            sse_read_timeout=self.sse_read_timeout,
-        ) as (read_stream, write_stream):
+            timeout=timedelta(seconds=self.timeout),
+            sse_read_timeout=timedelta(self.sse_read_timeout),
+        ) as (read_stream, write_stream, _):
             yield read_stream, write_stream
 
     def _get_log_level(self) -> LoggingLevel | None:
@@ -368,3 +382,6 @@ class MCPServerHTTP(MCPServer):
 
     def __repr__(self) -> str:  # pragma: no cover
         return f'MCPServerHTTP(url={self.url!r}, tool_prefix={self.tool_prefix!r})'
+
+    def _get_client_initialize_timeout(self) -> float:  # pragma: no cover
+        return self.timeout

{pydantic_ai_slim-0.2.8 → pydantic_ai_slim-0.2.10}/pydantic_ai/models/__init__.py

@@ -26,12 +26,25 @@ from ..usage import Usage
 KnownModelName = TypeAliasType(
     'KnownModelName',
     Literal[
-        'anthropic:claude-3-7-sonnet-latest',
+        'anthropic:claude-2.0',
+        'anthropic:claude-2.1',
+        'anthropic:claude-3-5-haiku-20241022',
         'anthropic:claude-3-5-haiku-latest',
+        'anthropic:claude-3-5-sonnet-20240620',
+        'anthropic:claude-3-5-sonnet-20241022',
         'anthropic:claude-3-5-sonnet-latest',
+        'anthropic:claude-3-7-sonnet-20250219',
+        'anthropic:claude-3-7-sonnet-latest',
+        'anthropic:claude-3-haiku-20240307',
+        'anthropic:claude-3-opus-20240229',
         'anthropic:claude-3-opus-latest',
-        'claude-3-7-sonnet-latest',
-        'claude-3-5-haiku-latest',
+        'anthropic:claude-3-sonnet-20240229',
+        'anthropic:claude-4-opus-20250514',
+        'anthropic:claude-4-sonnet-20250514',
+        'anthropic:claude-opus-4-0',
+        'anthropic:claude-opus-4-20250514',
+        'anthropic:claude-sonnet-4-0',
+        'anthropic:claude-sonnet-4-20250514',
         'bedrock:amazon.titan-tg1-large',
         'bedrock:amazon.titan-text-lite-v1',
         'bedrock:amazon.titan-text-express-v1',
@@ -75,8 +88,25 @@ KnownModelName = TypeAliasType(
         'bedrock:mistral.mixtral-8x7b-instruct-v0:1',
         'bedrock:mistral.mistral-large-2402-v1:0',
         'bedrock:mistral.mistral-large-2407-v1:0',
+        'claude-2.0',
+        'claude-2.1',
+        'claude-3-5-haiku-20241022',
+        'claude-3-5-haiku-latest',
+        'claude-3-5-sonnet-20240620',
+        'claude-3-5-sonnet-20241022',
         'claude-3-5-sonnet-latest',
+        'claude-3-7-sonnet-20250219',
+        'claude-3-7-sonnet-latest',
+        'claude-3-haiku-20240307',
+        'claude-3-opus-20240229',
         'claude-3-opus-latest',
+        'claude-3-sonnet-20240229',
+        'claude-4-opus-20250514',
+        'claude-4-sonnet-20250514',
+        'claude-opus-4-0',
+        'claude-opus-4-20250514',
+        'claude-sonnet-4-0',
+        'claude-sonnet-4-20250514',
         'cohere:c4ai-aya-expanse-32b',
         'cohere:c4ai-aya-expanse-8b',
         'cohere:command',
@@ -92,32 +122,26 @@ KnownModelName = TypeAliasType(
         'cohere:command-r7b-12-2024',
         'deepseek:deepseek-chat',
         'deepseek:deepseek-reasoner',
-        'google-gla:gemini-1.0-pro',
         'google-gla:gemini-1.5-flash',
         'google-gla:gemini-1.5-flash-8b',
         'google-gla:gemini-1.5-pro',
-        'google-gla:gemini-2.0-flash-exp',
-        'google-gla:gemini-2.0-flash-thinking-exp-01-21',
-        'google-gla:gemini-exp-1206',
+        'google-gla:gemini-1.0-pro',
         'google-gla:gemini-2.0-flash',
         'google-gla:gemini-2.0-flash-lite-preview-02-05',
         'google-gla:gemini-2.0-pro-exp-02-05',
-        'google-gla:gemini-2.5-flash-preview-04-17',
+        'google-gla:gemini-2.5-flash-preview-05-20',
         'google-gla:gemini-2.5-pro-exp-03-25',
-        'google-gla:gemini-2.5-pro-preview-03-25',
-        'google-vertex:gemini-1.0-pro',
+        'google-gla:gemini-2.5-pro-preview-05-06',
         'google-vertex:gemini-1.5-flash',
         'google-vertex:gemini-1.5-flash-8b',
         'google-vertex:gemini-1.5-pro',
-        'google-vertex:gemini-2.0-flash-exp',
-        'google-vertex:gemini-2.0-flash-thinking-exp-01-21',
-        'google-vertex:gemini-exp-1206',
+        'google-vertex:gemini-1.0-pro',
         'google-vertex:gemini-2.0-flash',
         'google-vertex:gemini-2.0-flash-lite-preview-02-05',
         'google-vertex:gemini-2.0-pro-exp-02-05',
-        'google-vertex:gemini-2.5-flash-preview-04-17',
+        'google-vertex:gemini-2.5-flash-preview-05-20',
         'google-vertex:gemini-2.5-pro-exp-03-25',
-        'google-vertex:gemini-2.5-pro-preview-03-25',
+        'google-vertex:gemini-2.5-pro-preview-05-06',
         'gpt-3.5-turbo',
         'gpt-3.5-turbo-0125',
         'gpt-3.5-turbo-0301',

{pydantic_ai_slim-0.2.8 → pydantic_ai_slim-0.2.10}/pydantic_ai/models/anthropic.py

@@ -7,7 +7,6 @@ from dataclasses import dataclass, field
 from datetime import datetime, timezone
 from typing import Any, Literal, Union, cast, overload
 
-from anthropic import AsyncAnthropic
 from typing_extensions import assert_never
 
 from .. import ModelHTTPError, UnexpectedModelBehavior, _utils, usage
@@ -41,7 +40,7 @@ from . import (
 )
 
 try:
-    from anthropic import NOT_GIVEN, APIStatusError, AsyncStream
+    from anthropic import NOT_GIVEN, APIStatusError, AsyncAnthropic, AsyncStream
     from anthropic.types.beta import (
         BetaBase64PDFBlockParam,
         BetaBase64PDFSourceParam,
@@ -68,6 +67,7 @@ try:
         BetaToolUseBlock,
         BetaToolUseBlockParam,
     )
+    from anthropic.types.model_param import ModelParam
 
 except ImportError as _import_error:
     raise ImportError(
@@ -75,12 +75,7 @@ except ImportError as _import_error:
         'you can use the `anthropic` optional group — `pip install "pydantic-ai-slim[anthropic]"`'
     ) from _import_error
 
-LatestAnthropicModelNames = Literal[
-    'claude-3-7-sonnet-latest',
-    'claude-3-5-haiku-latest',
-    'claude-3-5-sonnet-latest',
-    'claude-3-opus-latest',
-]
+LatestAnthropicModelNames = ModelParam
 """Latest Anthropic models."""
 
 AnthropicModelName = Union[str, LatestAnthropicModelNames]

{pydantic_ai_slim-0.2.8 → pydantic_ai_slim-0.2.10}/pydantic_ai/models/gemini.py

@@ -52,15 +52,12 @@ LatestGeminiModelNames = Literal[
     'gemini-1.5-flash-8b',
     'gemini-1.5-pro',
     'gemini-1.0-pro',
-    'gemini-2.0-flash-exp',
-    'gemini-2.0-flash-thinking-exp-01-21',
-    'gemini-exp-1206',
     'gemini-2.0-flash',
     'gemini-2.0-flash-lite-preview-02-05',
     'gemini-2.0-pro-exp-02-05',
-    'gemini-2.5-flash-preview-04-17',
+    'gemini-2.5-flash-preview-05-20',
     'gemini-2.5-pro-exp-03-25',
-    'gemini-2.5-pro-preview-03-25',
+    'gemini-2.5-pro-preview-05-06',
 ]
 """Latest Gemini models."""
 
@@ -80,6 +77,7 @@ class GeminiModelSettings(ModelSettings, total=False):
     """
 
     gemini_safety_settings: list[GeminiSafetySettings]
+    """Safety settings options for Gemini model request."""
 
     gemini_thinking_config: ThinkingConfig
     """Thinking is "on" by default in both the API and AI Studio.
@@ -93,6 +91,12 @@ class GeminiModelSettings(ModelSettings, total=False):
     See more about it on <https://ai.google.dev/gemini-api/docs/thinking>.
     """
 
+    gemini_labels: dict[str, str]
+    """User-defined metadata to break down billed charges. Only supported by the Vertex AI provider.
+
+    See the [Gemini API docs](https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/add-labels-to-api-calls) for use cases and limitations.
+    """
+
 
 @dataclass(init=False)
 class GeminiModel(Model):
@@ -223,25 +227,17 @@ class GeminiModel(Model):
         if tool_config is not None:
             request_data['toolConfig'] = tool_config
 
-        generation_config: _GeminiGenerationConfig = {}
-        if model_settings:
-            if (max_tokens := model_settings.get('max_tokens')) is not None:
-                generation_config['max_output_tokens'] = max_tokens
-            if (temperature := model_settings.get('temperature')) is not None:
-                generation_config['temperature'] = temperature
-            if (top_p := model_settings.get('top_p')) is not None:
-                generation_config['top_p'] = top_p
-            if (presence_penalty := model_settings.get('presence_penalty')) is not None:
-                generation_config['presence_penalty'] = presence_penalty
-            if (frequency_penalty := model_settings.get('frequency_penalty')) is not None:
-                generation_config['frequency_penalty'] = frequency_penalty
-            if (thinkingConfig := model_settings.get('gemini_thinking_config')) is not None:
-                generation_config['thinking_config'] = thinkingConfig  # pragma: no cover
-            if (gemini_safety_settings := model_settings.get('gemini_safety_settings')) is not None:
-                request_data['safetySettings'] = gemini_safety_settings
+        generation_config = _settings_to_generation_config(model_settings)
         if generation_config:
             request_data['generationConfig'] = generation_config
 
+        if gemini_safety_settings := model_settings.get('gemini_safety_settings'):
+            request_data['safetySettings'] = gemini_safety_settings
+
+        if gemini_labels := model_settings.get('gemini_labels'):
+            if self._system == 'google-vertex':
+                request_data['labels'] = gemini_labels
+
         headers = {'Content-Type': 'application/json', 'User-Agent': get_user_agent()}
         url = f'/{self._model_name}:{"streamGenerateContent" if streamed else "generateContent"}'
 
@@ -362,6 +358,23 @@ class GeminiModel(Model):
         return content
 
 
+def _settings_to_generation_config(model_settings: GeminiModelSettings) -> _GeminiGenerationConfig:
+    config: _GeminiGenerationConfig = {}
+    if (max_tokens := model_settings.get('max_tokens')) is not None:
+        config['max_output_tokens'] = max_tokens
+    if (temperature := model_settings.get('temperature')) is not None:
+        config['temperature'] = temperature
+    if (top_p := model_settings.get('top_p')) is not None:
+        config['top_p'] = top_p
+    if (presence_penalty := model_settings.get('presence_penalty')) is not None:
+        config['presence_penalty'] = presence_penalty
+    if (frequency_penalty := model_settings.get('frequency_penalty')) is not None:
+        config['frequency_penalty'] = frequency_penalty
+    if (thinkingConfig := model_settings.get('gemini_thinking_config')) is not None:
+        config['thinking_config'] = thinkingConfig  # pragma: no cover
+    return config
+
+
 class AuthProtocol(Protocol):
     """Abstract definition for Gemini authentication."""
 
@@ -483,6 +496,7 @@ class _GeminiRequest(TypedDict):
     <https://ai.google.dev/gemini-api/docs/system-instructions?lang=rest>
     """
     generationConfig: NotRequired[_GeminiGenerationConfig]
+    labels: NotRequired[dict[str, str]]
 
 
 class GeminiSafetySettings(TypedDict):

{pydantic_ai_slim-0.2.8 → pydantic_ai_slim-0.2.10}/pydantic_ai/models/google.py

@@ -76,15 +76,12 @@ LatestGoogleModelNames = Literal[
     'gemini-1.5-flash-8b',
     'gemini-1.5-pro',
     'gemini-1.0-pro',
-    'gemini-2.0-flash-exp',
-    'gemini-2.0-flash-thinking-exp-01-21',
-    'gemini-exp-1206',
     'gemini-2.0-flash',
     'gemini-2.0-flash-lite-preview-02-05',
     'gemini-2.0-pro-exp-02-05',
-    'gemini-2.5-flash-preview-04-17',
+    'gemini-2.5-flash-preview-05-20',
     'gemini-2.5-pro-exp-03-25',
-    'gemini-2.5-pro-preview-03-25',
+    'gemini-2.5-pro-preview-05-06',
 ]
 """Latest Gemini models."""
 
@@ -115,6 +112,12 @@ class GoogleModelSettings(ModelSettings, total=False):
     See <https://ai.google.dev/gemini-api/docs/thinking> for more information.
     """
 
+    google_labels: dict[str, str]
+    """User-defined metadata to break down billed charges. Only supported by the Vertex AI API.
+
+    See the [Gemini API docs](https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/add-labels-to-api-calls) for use cases and limitations.
+    """
+
 
 @dataclass(init=False)
 class GoogleModel(Model):
@@ -269,6 +272,7 @@ class GoogleModel(Model):
             frequency_penalty=model_settings.get('frequency_penalty'),
             safety_settings=model_settings.get('google_safety_settings'),
             thinking_config=model_settings.get('google_thinking_config'),
+            labels=model_settings.get('google_labels'),
             tools=cast(ToolListUnionDict, tools),
             tool_config=tool_config,
         )

{pydantic_ai_slim-0.2.8 → pydantic_ai_slim-0.2.10}/pyproject.toml

@@ -65,7 +65,7 @@ openai = ["openai>=1.75.0"]
 cohere = ["cohere>=5.13.11; platform_system != 'Emscripten'"]
 vertexai = ["google-auth>=2.36.0", "requests>=2.32.2"]
 google = ["google-genai>=1.15.0"]
-anthropic = ["anthropic>=0.49.0"]
+anthropic = ["anthropic>=0.52.0"]
 groq = ["groq>=0.15.0"]
 mistral = ["mistralai>=1.2.5"]
 bedrock = ["boto3>=1.35.74"]
@@ -75,7 +75,7 @@ tavily = ["tavily-python>=0.5.0"]
 # CLI
 cli = ["rich>=13", "prompt-toolkit>=3", "argcomplete>=3.5.0"]
 # MCP
-mcp = ["mcp>=1.6.0; python_version >= '3.10'"]
+mcp = ["mcp>=1.8.0; python_version >= '3.10'"]
 # Evals
 evals = ["pydantic-evals=={{ version }}"]
 # A2A