goose-py 0.9.8__py3-none-any.whl → 0.9.10__py3-none-any.whl

goose/_internal/types/agent.py

@@ -1,7 +1,19 @@
+import base64
 from enum import StrEnum
-from typing import Literal, NotRequired, TypedDict
+from typing import Any, Literal, NotRequired, TypedDict
 
-from pydantic import BaseModel
+from pydantic import BaseModel, GetCoreSchemaHandler
+from pydantic_core import CoreSchema, core_schema
+
+
+class Base64MediaContent(str):
+    @classmethod
+    def __get_pydantic_core_schema__(cls, source_type: Any, handler: GetCoreSchemaHandler) -> CoreSchema:
+        return core_schema.no_info_after_validator_function(cls, handler(str))
+
+    @classmethod
+    def from_bytes(cls, content: bytes, /) -> "Base64MediaContent":
+        return cls(base64.b64encode(content).decode())
 
 
 class AIModel(StrEnum):
@@ -59,7 +71,7 @@ class TextMessagePart(BaseModel):
 
 class MediaMessagePart(BaseModel):
     content_type: UserMediaContentType
-    content: str
+    content: Base64MediaContent
 
     def render(self) -> LLMMediaMessagePart:
         return {

goose/agent.py

@@ -2,6 +2,7 @@ from ._internal.agent import AgentResponse, IAgentLogger
 from ._internal.types.agent import (
     AIModel,
     AssistantMessage,
+    Base64MediaContent,
     LLMMediaMessagePart,
     LLMMessage,
     LLMTextMessagePart,
@@ -17,6 +18,7 @@ __all__ = [
     "AIModel",
     "IAgentLogger",
     "AssistantMessage",
+    "Base64MediaContent",
     "LLMMediaMessagePart",
     "LLMMessage",
     "LLMTextMessagePart",

goose/task.py

@@ -0,0 +1,3 @@
+from ._internal.task import Task
+
+__all__ = ["Task"]

{goose_py-0.9.8.dist-info → goose_py-0.9.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: goose-py
-Version: 0.9.8
+Version: 0.9.10
 Summary: A tool for AI workflows based on human-computer collaboration and structured output.
 Author-email: Nash Taylor <nash@chelle.ai>, Joshua Cook <joshua@chelle.ai>, Michael Sankur <michael@chelle.ai>
 Requires-Python: >=3.12

{goose_py-0.9.8.dist-info → goose_py-0.9.10.dist-info}/RECORD

@@ -1,9 +1,10 @@
 goose/__init__.py,sha256=wjGDgWzKcD6S8loVr0n-rLCpRwg-ZKAixcUaw1wobMc,243
-goose/agent.py,sha256=2iFjsZtXXNLXu1grCp3S_eJotGJnfP2190GYKEFQdTg,582
+goose/agent.py,sha256=3T5hobYzro2tmYaUd_kNfIxpmnioLOYy9HaYT0XttIA,632
 goose/errors.py,sha256=-0OyZQJWYTRw5YgnCB2_uorVaUsL6Z0QYQO2FqzCiyg,32
 goose/flow.py,sha256=YsZLBa5I1W27_P6LYGWbtFX8ZYx9vJG3KtENYChHm5E,111
 goose/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 goose/runs.py,sha256=ub-r_gzbUbaIzWXX-jc-dncNxEh6zTfzIkmnDfCSbRI,160
+goose/task.py,sha256=95rspdxETJoY12IHBl3KjnVIdqQnf1jDKlnGWNWOTvQ,53
 goose/_internal/agent.py,sha256=VRDEhBDTpKZS09B6xnWsOqLsxP7D5NHKVGlxwayPdoo,6030
 goose/_internal/conversation.py,sha256=zbMvP4oxhKAzATVEXZfGVKXWfEjh472MYKhmyJzSLgI,1172
 goose/_internal/flow.py,sha256=RShMsxgt49g1fZJ3rlwDHtI1j39lZzewx8hZ7DGN5kg,4124
@@ -12,7 +13,7 @@ goose/_internal/state.py,sha256=pI-C37Ybazo7EJPbZklxbiCYFy3u4I031NKBr8Jm_CI,6534
 goose/_internal/store.py,sha256=tWmKfa1-yq1jU6lT3l6kSOmVt2m3H7I1xLMTrxnUDI8,889
 goose/_internal/task.py,sha256=Ka3tm5Ymu8sE6_GCbkj3JDHK683ZET5HlkcgrQgnzEo,6028
 goose/_internal/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-goose/_internal/types/agent.py,sha256=QRYux_vbOe_wRtYBMqFpt4y6v9fKN70mwcsqaT0e1wE,2430
-goose_py-0.9.8.dist-info/METADATA,sha256=UpVDqVUdVPwynJJtQwQUbyva7qThIBdAPwaTOXpjUPs,441
-goose_py-0.9.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-goose_py-0.9.8.dist-info/RECORD,,
+goose/_internal/types/agent.py,sha256=zgS_OYi22rrPEBS95n-PGxPO86b6ci7aPsgHFJ2m0pc,2912
+goose_py-0.9.10.dist-info/METADATA,sha256=Vg_6i2_JVMvjwbK6QY3gCiLEk2lMiFL2BaOVkk1soPk,442
+goose_py-0.9.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+goose_py-0.9.10.dist-info/RECORD,,