pydantic-ai 0.1.4__tar.gz → 0.1.6__tar.gz

{pydantic_ai-0.1.4 → pydantic_ai-0.1.6}/PKG-INFO

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

{pydantic_ai-0.1.4 → pydantic_ai-0.1.6}/tests/models/test_instrumented.py

@@ -12,6 +12,10 @@ from opentelemetry._events import NoOpEventLoggerProvider
 from opentelemetry.trace import NoOpTracerProvider
 
 from pydantic_ai.messages import (
+    AudioUrl,
+    BinaryContent,
+    DocumentUrl,
+    ImageUrl,
     ModelMessage,
     ModelRequest,
     ModelResponse,
@@ -25,6 +29,7 @@ from pydantic_ai.messages import (
     ToolCallPart,
     ToolReturnPart,
     UserPromptPart,
+    VideoUrl,
 )
 from pydantic_ai.models import Model, ModelRequestParameters, StreamedResponse
 from pydantic_ai.models.instrumented import InstrumentationSettings, InstrumentedModel
@@ -736,3 +741,81 @@ def test_messages_to_otel_events_instructions_multiple_messages():
             {'content': 'user_prompt2', 'role': 'user', 'gen_ai.message.index': 2, 'event.name': 'gen_ai.user.message'},
         ]
     )
+
+
+def test_messages_to_otel_events_image_url(document_content: BinaryContent):
+    messages = [
+        ModelRequest(parts=[UserPromptPart(content=['user_prompt', ImageUrl('https://example.com/image.png')])]),
+        ModelRequest(parts=[UserPromptPart(content=['user_prompt2', AudioUrl('https://example.com/audio.mp3')])]),
+        ModelRequest(parts=[UserPromptPart(content=['user_prompt3', DocumentUrl('https://example.com/document.pdf')])]),
+        ModelRequest(parts=[UserPromptPart(content=['user_prompt4', VideoUrl('https://example.com/video.mp4')])]),
+        ModelRequest(
+            parts=[
+                UserPromptPart(
+                    content=[
+                        'user_prompt5',
+                        ImageUrl('https://example.com/image2.png'),
+                        AudioUrl('https://example.com/audio2.mp3'),
+                        DocumentUrl('https://example.com/document2.pdf'),
+                        VideoUrl('https://example.com/video2.mp4'),
+                    ]
+                )
+            ]
+        ),
+        ModelRequest(parts=[UserPromptPart(content=['user_prompt6', document_content])]),
+        ModelResponse(parts=[TextPart('text1')]),
+    ]
+    assert [
+        InstrumentedModel.event_to_dict(e) for e in InstrumentedModel.messages_to_otel_events(messages)
+    ] == snapshot(
+        [
+            {
+                'content': ['user_prompt', {'kind': 'image-url', 'url': 'https://example.com/image.png'}],
+                'role': 'user',
+                'gen_ai.message.index': 0,
+                'event.name': 'gen_ai.user.message',
+            },
+            {
+                'content': ['user_prompt2', {'kind': 'audio-url', 'url': 'https://example.com/audio.mp3'}],
+                'role': 'user',
+                'gen_ai.message.index': 1,
+                'event.name': 'gen_ai.user.message',
+            },
+            {
+                'content': ['user_prompt3', {'kind': 'document-url', 'url': 'https://example.com/document.pdf'}],
+                'role': 'user',
+                'gen_ai.message.index': 2,
+                'event.name': 'gen_ai.user.message',
+            },
+            {
+                'content': ['user_prompt4', {'kind': 'video-url', 'url': 'https://example.com/video.mp4'}],
+                'role': 'user',
+                'gen_ai.message.index': 3,
+                'event.name': 'gen_ai.user.message',
+            },
+            {
+                'content': [
+                    'user_prompt5',
+                    {'kind': 'image-url', 'url': 'https://example.com/image2.png'},
+                    {'kind': 'audio-url', 'url': 'https://example.com/audio2.mp3'},
+                    {'kind': 'document-url', 'url': 'https://example.com/document2.pdf'},
+                    {'kind': 'video-url', 'url': 'https://example.com/video2.mp4'},
+                ],
+                'role': 'user',
+                'gen_ai.message.index': 4,
+                'event.name': 'gen_ai.user.message',
+            },
+            {
+                'content': ['user_prompt6', {'kind': 'binary'}],
+                'role': 'user',
+                'gen_ai.message.index': 5,
+                'event.name': 'gen_ai.user.message',
+            },
+            {
+                'role': 'assistant',
+                'content': 'text1',
+                'gen_ai.message.index': 6,
+                'event.name': 'gen_ai.assistant.message',
+            },
+        ]
+    )