pydantic-ai 0.3.0__tar.gz → 0.3.1__tar.gz

{pydantic_ai-0.3.0 → pydantic_ai-0.3.1}/PKG-INFO

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

pydantic_ai-0.3.1/tests/test_thinking_part.py

@@ -0,0 +1,69 @@
+from __future__ import annotations as _annotations
+
+import pytest
+from inline_snapshot import snapshot
+
+from pydantic_ai._thinking_part import split_content_into_text_and_thinking
+from pydantic_ai.messages import ModelResponsePart, TextPart, ThinkingPart, ThinkingPartDelta
+
+
+@pytest.mark.parametrize(
+    ('content', 'parts'),
+    [
+        ('foo bar', [TextPart(content='foo bar')]),
+        (
+            'foo bar<think>thinking</think>',
+            [TextPart(content='foo bar'), ThinkingPart(content='thinking')],
+        ),
+        (
+            'foo bar<think>thinking</think>baz',
+            [TextPart(content='foo bar'), ThinkingPart(content='thinking'), TextPart(content='baz')],
+        ),
+        (
+            'foo bar<think>thinking',
+            [TextPart(content='foo bar'), TextPart(content='thinking')],
+        ),
+    ],
+)
+def test_split_content_into_text_and_thinking(content: str, parts: list[ModelResponsePart]):
+    assert split_content_into_text_and_thinking(content) == parts
+
+
+def test_thinking_part_delta_applies_both_content_and_signature():
+    thinking_part = ThinkingPart(content='Initial content', signature='initial_sig')
+    delta = ThinkingPartDelta(content_delta=' added', signature_delta='new_sig')
+
+    result = delta.apply(thinking_part)
+
+    # The content is appended, and the signature is updated.
+    assert result == snapshot(ThinkingPart(content='Initial content added', signature='new_sig'))
+
+
+def test_thinking_part_delta_applies_signature_only():
+    thinking_part = ThinkingPart(content='Initial content', signature='initial_sig')
+    delta_sig_only = ThinkingPartDelta(content_delta=None, signature_delta='sig_only')
+
+    result_sig_only = delta_sig_only.apply(thinking_part)
+
+    # The content is unchanged, and the signature is updated.
+    assert result_sig_only == snapshot(ThinkingPart(content='Initial content', signature='sig_only'))
+
+
+def test_thinking_part_delta_applies_content_only_preserves_signature():
+    thinking_part = ThinkingPart(content='Initial content', signature='initial_sig')
+    delta_content_only = ThinkingPartDelta(content_delta=' more', signature_delta=None)
+
+    result_content_only = delta_content_only.apply(thinking_part)
+
+    # The content is appended, and the signature is preserved.
+    assert result_content_only == snapshot(ThinkingPart(content='Initial content more', signature='initial_sig'))
+
+
+def test_thinking_part_delta_applies_to_part_with_none_signature():
+    thinking_part_no_sig = ThinkingPart(content='No sig content', signature=None)
+    delta_to_none_sig = ThinkingPartDelta(content_delta=' extra', signature_delta='added_sig')
+
+    result_none_sig = delta_to_none_sig.apply(thinking_part_no_sig)
+
+    # The content is appended, and the signature is updated.
+    assert result_none_sig == snapshot(ThinkingPart(content='No sig content extra', signature='added_sig'))

pydantic_ai-0.3.0/tests/test_thinking_part.py

@@ -1,28 +0,0 @@
-from __future__ import annotations as _annotations
-
-import pytest
-
-from pydantic_ai._thinking_part import split_content_into_text_and_thinking
-from pydantic_ai.messages import ModelResponsePart, TextPart, ThinkingPart
-
-
-@pytest.mark.parametrize(
-    ('content', 'parts'),
-    [
-        ('foo bar', [TextPart(content='foo bar')]),
-        (
-            'foo bar<think>thinking</think>',
-            [TextPart(content='foo bar'), ThinkingPart(content='thinking')],
-        ),
-        (
-            'foo bar<think>thinking</think>baz',
-            [TextPart(content='foo bar'), ThinkingPart(content='thinking'), TextPart(content='baz')],
-        ),
-        (
-            'foo bar<think>thinking',
-            [TextPart(content='foo bar'), TextPart(content='thinking')],
-        ),
-    ],
-)
-def test_split_content_into_text_and_thinking(content: str, parts: list[ModelResponsePart]):
-    assert split_content_into_text_and_thinking(content) == parts