support Pydantic v2.10 (#79)

Author: samuelcolvin

docs/install.md

@@ -9,9 +9,6 @@ pip/uv-add pydantic-ai
 
 It requires Python 3.9+.
 
-!!! note "Use `--prerelease=allow` with uv"
-    Until Pydantic v2.10 is released (very soon), you'll need to use `--prerelease=allow` with `uv` to install PydanticAI.
-
 ## Use with Pydantic Logfire
 
 PydanticAI has an excellent (but completely optional) integration with [Pydantic Logfire](https://pydantic.dev/logfire) to help you view and understand agent runs.

pydantic_ai/_result.py

@@ -5,7 +5,7 @@
 import types
 from collections.abc import Awaitable
 from dataclasses import dataclass, field
-from typing import Any, Callable, Generic, Union, cast, get_args, get_origin
+from typing import Any, Callable, Generic, Literal, Union, cast, get_args, get_origin
 
 from pydantic import TypeAdapter, ValidationError
 from typing_extensions import Self, TypeAliasType, TypedDict
@@ -184,13 +184,14 @@ def validate(
             Either the validated result data (left) or a retry message (right).
         """
         try:
+            pyd_allow_partial: Literal['off', 'trailing-strings'] = 'trailing-strings' if allow_partial else 'off'
             if isinstance(tool_call.args, messages.ArgsJson):
                 result = self.type_adapter.validate_json(
-                    tool_call.args.args_json or '', experimental_allow_partial=allow_partial
+                    tool_call.args.args_json or '', experimental_allow_partial=pyd_allow_partial
                 )
             else:
                 result = self.type_adapter.validate_python(
-                    tool_call.args.args_object, experimental_allow_partial=allow_partial
+                    tool_call.args.args_object, experimental_allow_partial=pyd_allow_partial
                 )
         except ValidationError as e:
             if wrap_validation_errors:

pydantic_ai/models/gemini.py

@@ -213,8 +213,8 @@ async def _process_streamed_response(http_response: HTTPResponse) -> EitherStrea
         async for chunk in aiter_bytes:
             content.extend(chunk)
             responses = _gemini_streamed_response_ta.validate_json(
-                content,  # type: ignore # see https://github.com/pydantic/pydantic/pull/10802
-                experimental_allow_partial=True,
+                content,
+                experimental_allow_partial='trailing-strings',
             )
             if responses:
                 last = responses[-1]
@@ -279,7 +279,9 @@ def get(self, *, final: bool = False) -> Iterable[str]:
             all_items = pydantic_core.from_json(self._json_content, allow_partial=True)
             new_items = all_items[self._position : -1]
             self._position = len(all_items) - 1
-            new_responses = _gemini_streamed_response_ta.validate_python(new_items, experimental_allow_partial=True)
+            new_responses = _gemini_streamed_response_ta.validate_python(
+                new_items, experimental_allow_partial='trailing-strings'
+            )
         for r in new_responses:
             self._cost += _metadata_as_cost(r['usage_metadata'])
             parts = r['candidates'][0]['content']['parts']
@@ -321,8 +323,8 @@ def get(self, *, final: bool = False) -> ModelStructuredResponse:
         separate parts.
         """
         responses = _gemini_streamed_response_ta.validate_json(
-            self._content,  # type: ignore # see https://github.com/pydantic/pydantic/pull/10802
-            experimental_allow_partial=not final,
+            self._content,
+            experimental_allow_partial='off' if final else 'trailing-strings',
         )
         combined_parts: list[_GeminiFunctionCallPart] = []
         self._cost = result.Cost()

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pydantic-ai"
-version = "0.0.4"
+version = "0.0.5"
 description = "Agent Framework / shim to use Pydantic with LLMs"
 authors = [
     { name = "Samuel Colvin", email = "[email protected]" },
@@ -39,7 +39,7 @@ dependencies = [
     "httpx>=0.27.2",
     "logfire-api>=1.2.0",
     "openai>=1.54.3",
-    "pydantic>=2.10.0b1",
+    "pydantic>=2.10",
 ]
 
 [project.optional-dependencies]