goose-py 0.9.12__tar.gz → 0.9.14__tar.gz

{goose_py-0.9.12 → goose_py-0.9.14}/.stubs/litellm/__init__.pyi

@@ -4,9 +4,11 @@ _LiteLLMGeminiModel = Literal[
     "vertex_ai/gemini-1.5-flash",
     "vertex_ai/gemini-1.5-pro",
     "vertex_ai/gemini-1.5-flash-8b",
+    "vertex_ai/gemini-2.0-flash",
     "gemini/gemini-1.5-flash",
     "gemini/gemini-1.5-pro",
     "gemini/gemini-1.5-flash-8b",
+    "gemini/gemini-2.0-flash",
 ]
 _MessageRole = Literal["system", "user", "assistant"]
 

{goose_py-0.9.12 → goose_py-0.9.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: goose-py
-Version: 0.9.12
+Version: 0.9.14
 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.12 → goose_py-0.9.14}/goose/_internal/types/agent.py

@@ -1,19 +1,8 @@
 import base64
 from enum import StrEnum
-from typing import Any, Literal, NotRequired, TypedDict
+from typing import Literal, NotRequired, TypedDict
 
-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())
+from pydantic import BaseModel
 
 
 class AIModel(StrEnum):
@@ -21,14 +10,19 @@ class AIModel(StrEnum):
     VERTEX_PRO = "vertex_ai/gemini-1.5-pro"
     VERTEX_FLASH = "vertex_ai/gemini-1.5-flash"
     VERTEX_FLASH_8B = "vertex_ai/gemini-1.5-flash-8b"
+    VERTEX_FLASH_2_0 = "vertex_ai/gemini-2.0-flash"
 
     # gemini (publicly available, no GCP environment required)
     GEMINI_PRO = "gemini/gemini-1.5-pro"
     GEMINI_FLASH = "gemini/gemini-1.5-flash"
     GEMINI_FLASH_8B = "gemini/gemini-1.5-flash-8b"
+    GEMINI_FLASH_2_0 = "gemini/gemini-2.0-flash"
+
 
+class ContentType(StrEnum):
+    # text
+    TEXT = "text/plain"
 
-class UserMediaContentType(StrEnum):
     # images
     JPEG = "image/jpeg"
     PNG = "image/png"
@@ -62,33 +56,30 @@ class LLMMessage(TypedDict):
     cache_control: NotRequired[CacheControl]
 
 
-class TextMessagePart(BaseModel):
-    text: str
-
-    def render(self) -> LLMTextMessagePart:
-        return {"type": "text", "text": self.text}
-
+class MessagePart(BaseModel):
+    content: str
+    content_type: ContentType = ContentType.TEXT
 
-class MediaMessagePart(BaseModel):
-    content_type: UserMediaContentType
-    content: Base64MediaContent
+    @classmethod
+    def from_media(cls, *, content: bytes, content_type: ContentType) -> "MessagePart":
+        return cls(content=base64.b64encode(content).decode(), content_type=content_type)
 
-    def render(self) -> LLMMediaMessagePart:
-        return {
-            "type": "image_url",
-            "image_url": f"data:{self.content_type};base64,{self.content}",
-        }
+    def render(self) -> LLMTextMessagePart | LLMMediaMessagePart:
+        if self.content_type == ContentType.TEXT:
+            return {"type": "text", "text": self.content}
+        else:
+            return {"type": "image_url", "image_url": f"data:{self.content_type};base64,{self.content}"}
 
 
 class UserMessage(BaseModel):
-    parts: list[TextMessagePart | MediaMessagePart]
+    parts: list[MessagePart]
 
     def render(self) -> LLMMessage:
         content: LLMMessage = {
             "role": "user",
             "content": [part.render() for part in self.parts],
         }
-        if any(isinstance(part, MediaMessagePart) for part in self.parts):
+        if any(part.content_type != ContentType.TEXT for part in self.parts):
             content["cache_control"] = {"type": "ephemeral"}
         return content
 
@@ -101,7 +92,7 @@ class AssistantMessage(BaseModel):
 
 
 class SystemMessage(BaseModel):
-    parts: list[TextMessagePart | MediaMessagePart]
+    parts: list[MessagePart]
 
     def render(self) -> LLMMessage:
         return {

{goose_py-0.9.12 → goose_py-0.9.14}/goose/agent.py

@@ -2,14 +2,12 @@ from ._internal.agent import AgentResponse, IAgentLogger
 from ._internal.types.agent import (
     AIModel,
     AssistantMessage,
-    Base64MediaContent,
+    ContentType,
     LLMMediaMessagePart,
     LLMMessage,
     LLMTextMessagePart,
-    MediaMessagePart,
+    MessagePart,
     SystemMessage,
-    TextMessagePart,
-    UserMediaContentType,
     UserMessage,
 )
 
@@ -18,13 +16,11 @@ __all__ = [
     "AIModel",
     "IAgentLogger",
     "AssistantMessage",
-    "Base64MediaContent",
     "LLMMediaMessagePart",
     "LLMMessage",
     "LLMTextMessagePart",
-    "MediaMessagePart",
     "SystemMessage",
-    "TextMessagePart",
-    "UserMediaContentType",
+    "MessagePart",
+    "ContentType",
     "UserMessage",
 ]

{goose_py-0.9.12 → goose_py-0.9.14}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "goose-py"
-version = "0.9.12"
+version = "0.9.14"
 description = "A tool for AI workflows based on human-computer collaboration and structured output."
 readme = "README.md"
 authors = [

{goose_py-0.9.12 → goose_py-0.9.14}/tests/test_agent.py

@@ -4,7 +4,8 @@ import pytest
 from pytest_mock import MockerFixture
 
 from goose import Agent, FlowArguments, TextResult, flow, task
-from goose.agent import AgentResponse, AIModel, IAgentLogger, TextMessagePart, UserMessage
+from goose._internal.types.agent import MessagePart
+from goose.agent import AgentResponse, AIModel, IAgentLogger, UserMessage
 
 
 class TestFlowArguments(FlowArguments):
@@ -28,7 +29,7 @@ def mock_litellm(mocker: MockerFixture) -> Mock:
 @task
 async def use_agent(*, agent: Agent) -> TextResult:
     return await agent(
-        messages=[UserMessage(parts=[TextMessagePart(text="Hello")])],
+        messages=[UserMessage(parts=[MessagePart(content="Hello")])],
         model=AIModel.GEMINI_FLASH_8B,
         task_name="greet",
     )

{goose_py-0.9.12 → goose_py-0.9.14}/tests/test_refining.py

@@ -6,7 +6,8 @@ import pytest
 from pytest_mock import MockerFixture
 
 from goose import Agent, FlowArguments, Result, flow, task
-from goose.agent import SystemMessage, TextMessagePart, UserMessage
+from goose._internal.types.agent import MessagePart
+from goose.agent import SystemMessage, UserMessage
 from goose.errors import Honk
 
 
@@ -59,8 +60,8 @@ async def test_refining() -> None:
     # imagine this is a new process
     async with sentence.start_run(run_id="1") as second_run:
         await generate_random_word.refine(
-            user_message=UserMessage(parts=[TextMessagePart(text="Change it")]),
-            context=SystemMessage(parts=[TextMessagePart(text="Extra info")]),
+            user_message=UserMessage(parts=[MessagePart(content="Change it")]),
+            context=SystemMessage(parts=[MessagePart(content="Extra info")]),
         )
 
     random_words = second_run.get_all(task=generate_random_word)
@@ -76,6 +77,6 @@ async def test_refining_before_generate_fails() -> None:
     with pytest.raises(Honk):
         async with sentence.start_run(run_id="2"):
             await generate_random_word.refine(
-                user_message=UserMessage(parts=[TextMessagePart(text="Change it")]),
-                context=SystemMessage(parts=[TextMessagePart(text="Extra info")]),
+                user_message=UserMessage(parts=[MessagePart(content="Change it")]),
+                context=SystemMessage(parts=[MessagePart(content="Extra info")]),
             )

{goose_py-0.9.12 → goose_py-0.9.14}/tests/test_state.py

@@ -6,7 +6,7 @@ import pytest
 from pytest_mock import MockerFixture
 
 from goose import Agent, FlowArguments, Result, flow, task
-from goose._internal.types.agent import SystemMessage, TextMessagePart, UserMessage
+from goose._internal.types.agent import MessagePart, SystemMessage, UserMessage
 from goose.errors import Honk
 
 
@@ -74,8 +74,8 @@ async def test_state_undo() -> None:
     async with with_state.start_run(run_id="2"):
         await generate_random_word.refine(
             index=0,
-            user_message=UserMessage(parts=[TextMessagePart(text="Change it")]),
-            context=SystemMessage(parts=[TextMessagePart(text="Extra info")]),
+            user_message=UserMessage(parts=[MessagePart(content="Change it")]),
+            context=SystemMessage(parts=[MessagePart(content="Extra info")]),
         )
 
     async with with_state.start_run(run_id="2") as run:

{goose_py-0.9.12 → goose_py-0.9.14}/uv.lock

@@ -234,7 +234,7 @@ wheels = [
 
 [[package]]
 name = "goose-py"
-version = "0.9.12"
+version = "0.9.14"
 source = { editable = "." }
 dependencies = [
     { name = "jsonpath-ng" },