pyagentic-core 2.3.0a2__tar.gz → 2.3.0a4__tar.gz

{pyagentic_core-2.3.0a2/pyagentic_core.egg-info → pyagentic_core-2.3.0a4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyagentic-core
-Version: 2.3.0a2
+Version: 2.3.0a4
 Summary: Build LLM Agents in a Pythonic way
 Author-email: Ryan Mikulec <rmikulec.dev@gmail.com>
 License: MIT

{pyagentic_core-2.3.0a2 → pyagentic_core-2.3.0a4}/pyagentic/_base/_agent/_agent.py

@@ -38,7 +38,7 @@ from pyagentic.llm._provider import LLMProvider
 from pyagentic.llm import LLMProviders
 from pyagentic.tracing._tracer import AgentTracer, traced
 from pyagentic.tracing import BasicTracer
-
+from pyagentic._utils._image import _encode_image
 
 logger = get_logger(__name__)
 
@@ -437,8 +437,11 @@ class BaseAgent(metaclass=AgentMeta):
             stringified_result = f"[Image returned by {tool_call.name}]"
             # Add tool result message
             self.state._messages.append(
-                self.provider.to_tool_call_result_message(result=stringified_result, id_=tool_call.id)
+                self.provider.to_tool_call_result_message(
+                    result=stringified_result, id_=tool_call.id
+                )
             )
+            result = _encode_image(result)
             # Add the image as a user message so the LLM can see it
             self.state._messages.append(
                 Message(role="user", content=f"Image from {tool_call.name}:", image=result)
@@ -451,7 +454,9 @@ class BaseAgent(metaclass=AgentMeta):
             )
             # Add tool result to conversation history for LLM
             self.state._messages.append(
-                self.provider.to_tool_call_result_message(result=stringified_result, id_=tool_call.id)
+                self.provider.to_tool_call_result_message(
+                    result=stringified_result, id_=tool_call.id
+                )
             )
 
         if self.phases:

{pyagentic_core-2.3.0a2 → pyagentic_core-2.3.0a4}/pyagentic/_utils/_image.py

@@ -17,4 +17,4 @@ def _encode_image(image: Image) -> str:
     image.save(buffer, format="PNG")
     image_bytes = buffer.getvalue()
     base64_image = base64.b64encode(image_bytes).decode("utf-8")
-    return f"data:image/png;base64,{base64_image}"
+    return base64_image

{pyagentic_core-2.3.0a2 → pyagentic_core-2.3.0a4}/pyagentic/llm/_openai.py

@@ -106,13 +106,16 @@ class OpenAIProvider(LLMProvider):
 
             # If message has an image, convert to OpenAI's content array format
             if message.image is not None:
-                image_url = _encode_image(message.image)
-
                 # Build content array with text and image
                 content = []
                 if message.content:
                     content.append({"type": "text", "text": message.content})
-                content.append({"type": "image_url", "image_url": {"url": image_url}})
+                content.append(
+                    {
+                        "type": "image_url",
+                        "image_url": {"url": f"data:image/png;base64,{message.image}"},
+                    }
+                )
 
                 msg_dict["content"] = content
                 # Remove the image field as it's now in content

{pyagentic_core-2.3.0a2 → pyagentic_core-2.3.0a4}/pyagentic/models/llm.py

@@ -1,8 +1,7 @@
-from typing import Any, List, Optional, TYPE_CHECKING
+from typing import Any, List, Optional
 from pydantic import BaseModel, ConfigDict
 
-if TYPE_CHECKING:
-    from PIL.Image import Image
+from PIL.Image import Image
 
 
 class Message(BaseModel):
@@ -18,7 +17,7 @@ class Message(BaseModel):
     type: Optional[str] = None
     role: Optional[str] = None
     content: Optional[str] = None
-    image: Optional["Image"] = None
+    image: Optional[str] = None
 
     def to_dict(self, exclude_none: bool = True):
         """

{pyagentic_core-2.3.0a2 → pyagentic_core-2.3.0a4}/pyagentic/models/response.py

@@ -69,8 +69,10 @@ class ToolResponse(BaseModel):
                 case _:
                     raise Exception(f"Unsupported type: {param_type}")
 
+        return_type = str if tool_def.return_type == Image else tool_def.return_type
+
         return create_model(
-            f"ToolResponse[{tool_def.name}]", __base__=cls, output=tool_def.return_type, **fields
+            f"ToolResponse[{tool_def.name}]", __base__=cls, output=return_type, **fields
         )
 
 

{pyagentic_core-2.3.0a2 → pyagentic_core-2.3.0a4/pyagentic_core.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyagentic-core
-Version: 2.3.0a2
+Version: 2.3.0a4
 Summary: Build LLM Agents in a Pythonic way
 Author-email: Ryan Mikulec <rmikulec.dev@gmail.com>
 License: MIT

{pyagentic_core-2.3.0a2 → pyagentic_core-2.3.0a4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pyagentic-core"
-version = "2.3.0-a.2"
+version = "2.3.0-a.4"
 description = "Build LLM Agents in a Pythonic way"
 readme = "README.md"
 requires-python = ">=3.13"