pydantic-ai-slim 0.0.7__py3-none-any.whl → 0.0.8__py3-none-any.whl

pydantic_ai/agent.py

@@ -284,6 +284,7 @@ class Agent(Generic[AgentDeps, ResultData]):
                                     self._result_schema,
                                     deps,
                                     self._result_validators,
+                                    lambda m: run_span.set_attribute('all_messages', messages),
                                 )
                                 return
                             else:

pydantic_ai/messages.py

@@ -139,12 +139,16 @@ class ModelTextResponse:
 
 @dataclass
 class ArgsJson:
+    """Tool arguments as a JSON string."""
+
     args_json: str
     """A JSON string of arguments."""
 
 
 @dataclass
 class ArgsDict:
+    """Tool arguments as a Python dictionary."""
+
     args_dict: dict[str, Any]
     """A python dictionary of arguments."""
 

pydantic_ai/models/gemini.py

@@ -109,11 +109,15 @@ class GeminiModel(Model):
 
 
 class AuthProtocol(Protocol):
+    """Abstract definition for Gemini authentication."""
+
     async def headers(self) -> dict[str, str]: ...
 
 
 @dataclass
 class ApiKeyAuth:
+    """Authentication using an API key for the `X-Goog-Api-Key` header."""
+
     api_key: str
 
     async def headers(self) -> dict[str, str]:

pydantic_ai/models/test.py

@@ -31,14 +31,6 @@ from . import (
 )
 
 
-class UnSetType:
-    def __repr__(self):
-        return 'UnSet'
-
-
-UnSet = UnSetType()
-
-
 @dataclass
 class TestModel(Model):
     """A model specifically for testing purposes.
@@ -186,6 +178,8 @@ class TestAgentModel(AgentModel):
 
 @dataclass
 class TestStreamTextResponse(StreamTextResponse):
+    """A text response that streams test data."""
+
     _text: str
     _cost: Cost
     _iter: Iterator[str] = field(init=False)
@@ -217,6 +211,8 @@ class TestStreamTextResponse(StreamTextResponse):
 
 @dataclass
 class TestStreamStructuredResponse(StreamStructuredResponse):
+    """A structured response that streams test data."""
+
     _structured_response: ModelStructuredResponse
     _cost: Cost
     _iter: Iterator[None] = field(default_factory=lambda: iter([None]))

pydantic_ai/models/vertexai.py

@@ -182,6 +182,8 @@ MAX_TOKEN_AGE = timedelta(seconds=3000)
 
 @dataclass
 class BearerTokenAuth:
+    """Authentication using a bearer token generated by google-auth."""
+
     credentials: BaseCredentials | ServiceAccountCredentials
     token_created: datetime | None = field(default=None, init=False)
 

pydantic_ai/result.py

@@ -1,8 +1,8 @@
 from __future__ import annotations as _annotations
 
 from abc import ABC, abstractmethod
-from collections.abc import AsyncIterator
-from dataclasses import dataclass
+from collections.abc import AsyncIterator, Callable
+from dataclasses import dataclass, field
 from datetime import datetime
 from typing import Generic, TypeVar, cast
 
@@ -49,11 +49,11 @@ class Cost:
         This is provided so it's trivial to sum costs from multiple requests and runs.
         """
         counts: dict[str, int] = {}
-        for field in 'request_tokens', 'response_tokens', 'total_tokens':
-            self_value = getattr(self, field)
-            other_value = getattr(other, field)
+        for f in 'request_tokens', 'response_tokens', 'total_tokens':
+            self_value = getattr(self, f)
+            other_value = getattr(other, f)
             if self_value is not None or other_value is not None:
-                counts[field] = (self_value or 0) + (other_value or 0)
+                counts[f] = (self_value or 0) + (other_value or 0)
 
         details = self.details.copy() if self.details is not None else None
         if other.details is not None:
@@ -122,7 +122,8 @@ class StreamedRunResult(_BaseRunResult[ResultData], Generic[AgentDeps, ResultDat
     _result_schema: _result.ResultSchema[ResultData] | None
     _deps: AgentDeps
     _result_validators: list[_result.ResultValidator[AgentDeps, ResultData]]
-    is_complete: bool = False
+    _on_complete: Callable[[list[messages.Message]], None]
+    is_complete: bool = field(default=False, init=False)
     """Whether the stream has all been received.
 
     This is set to `True` when one of
@@ -312,3 +313,4 @@ class StreamedRunResult(_BaseRunResult[ResultData], Generic[AgentDeps, ResultDat
         else:
             assert structured_message is not None, 'Either text or structured_message should provided, not both'
             self._all_messages.append(structured_message)
+        self._on_complete(self._all_messages)

{pydantic_ai_slim-0.0.7.dist-info → pydantic_ai_slim-0.0.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pydantic-ai-slim
-Version: 0.0.7
+Version: 0.0.8
 Summary: Agent Framework / shim to use Pydantic with LLMs, slim package
 Author-email: Samuel Colvin <samuel@pydantic.dev>
 License: MIT

{pydantic_ai_slim-0.0.7.dist-info → pydantic_ai_slim-0.0.8.dist-info}/RECORD

@@ -5,19 +5,19 @@ pydantic_ai/_result.py,sha256=cAqfPipK39cz-p-ftlJ83Q5_LI1TRb3-HH-iivb5rEM,9674
 pydantic_ai/_system_prompt.py,sha256=63egOej8zHsDVOInPayn0EEEDXKd0HVAbbrqXUTV96s,1092
 pydantic_ai/_tool.py,sha256=5Q9XaGOEXbyOLS644osB1AA5EMoJkr4eYK60MVZo0Z8,4528
 pydantic_ai/_utils.py,sha256=eNb7f3-ZQC8WDEa87iUcXGQ-lyuutFQG-5yBCMD4Vvs,8227
-pydantic_ai/agent.py,sha256=RhLl6tf6kogtauqQr3U1ufg712k-g002YPcio9-BPB4,34242
+pydantic_ai/agent.py,sha256=r5DI4ZBqYE67GOMEEu-LXrTa5ty2AchW4szotwm5Qis,34338
 pydantic_ai/dependencies.py,sha256=EHvD68AFkItxMnfHzJLG7T_AD1RGI2MZOfzm1v89hGQ,2399
 pydantic_ai/exceptions.py,sha256=ko_47M0k6Rhg9mUC9P1cj7N4LCH6cC0pEsF65A2vL-U,1561
-pydantic_ai/messages.py,sha256=tyBs2ucqIrKjjMIX2bvjMaZ6BMxim4VBkmCzTbeH-AI,7493
+pydantic_ai/messages.py,sha256=FFTQ9Bo2Ct4bLuyJF-M9xkeraw05I--NC_ieR6oGtTM,7587
 pydantic_ai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pydantic_ai/result.py,sha256=Gs2ZjuFJgONGJm8M5R2mGph5-lUFLBg7FxVrs2CVDPs,13525
+pydantic_ai/result.py,sha256=qsanb7v4qJ4pJdkdsqpy68kuZ1WNCpQB5jcXeTwGpe0,13658
 pydantic_ai/models/__init__.py,sha256=Cx8PjEsi5gkNOVQic32sf4CmM-A3pRu1LcjpM6poiBI,10138
 pydantic_ai/models/function.py,sha256=Mzc-zXnb2RayWAA8N9NS7KGF49do1S-VW3U9fkc661o,10045
-pydantic_ai/models/gemini.py,sha256=2mD5MJT7qaiMAWrySLjeUFSVoMYhYOS4V7ueTOxzkdA,26472
+pydantic_ai/models/gemini.py,sha256=ruO4tnnpDDuHThg7jUOphs8I_KXBJH7gfDMluliED8E,26606
 pydantic_ai/models/groq.py,sha256=Tx2yU3ysmPLBmWGsjzES-XcumzrsoBtB7spCnJBlLiM,14947
 pydantic_ai/models/openai.py,sha256=5ihH25CrS0tnZNW-BZw4GyPe8V-IxIHWw3B9ulPVjQE,14931
-pydantic_ai/models/test.py,sha256=HOS3_u0n3nYWJmBFPsAFpZ9gfHiZLauz24o825q3e9M,14443
-pydantic_ai/models/vertexai.py,sha256=xHatvwRn7_vyqmp3aDtHqryYAq8NoeXgXVATkj7yHuw,9088
-pydantic_ai_slim-0.0.7.dist-info/METADATA,sha256=UeP3i7vUZv1MCKG-VGLZUwEO8kbcCzzLpD6RsAN_Vbs,2561
-pydantic_ai_slim-0.0.7.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-pydantic_ai_slim-0.0.7.dist-info/RECORD,,
+pydantic_ai/models/test.py,sha256=q1wch_E7TSb4qx9PCcP1YyBGZx567MGlAQhlAlON0S8,14463
+pydantic_ai/models/vertexai.py,sha256=5wI8y2YjeRgSE51uKy5OtevQkks65uEbxIUAs5EGBaI,9161
+pydantic_ai_slim-0.0.8.dist-info/METADATA,sha256=CmpvlEAUyaWaPbdRCbFuypVLJ8yNC3TwZ0jgvlR9yps,2561
+pydantic_ai_slim-0.0.8.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+pydantic_ai_slim-0.0.8.dist-info/RECORD,,