goose-py 0.3.10__tar.gz → 0.3.12__tar.gz

{goose_py-0.3.10 → goose_py-0.3.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: goose-py
-Version: 0.3.10
+Version: 0.3.12
 Summary: A tool for AI workflows based on human-computer collaboration and structured output.
 Home-page: https://github.com/chelle-ai/goose
 Keywords: ai,yaml,configuration,llm

{goose_py-0.3.10 → goose_py-0.3.12}/goose/agent.py

@@ -10,10 +10,9 @@ from pydantic import BaseModel, computed_field
 
 
 class GeminiModel(StrEnum):
-    EXP = "gemini/gemini-exp-1121"
-    PRO = "gemini/gemini-1.5-pro"
-    FLASH = "gemini/gemini-1.5-flash"
-    FLASH_8B = "gemini/gemini-1.5-flash-8b"
+    PRO = "vertex_ai/gemini-1.5-pro"
+    FLASH = "vertex_ai/gemini-1.5-flash"
+    FLASH_8B = "vertex_ai/gemini-1.5-flash-8b"
 
 
 class UserMediaContentType(StrEnum):
@@ -23,7 +22,7 @@ class UserMediaContentType(StrEnum):
     WEBP = "image/webp"
 
     # audio
-    MP3 = "audio/mpeg"
+    MP3 = "audio/mp3"
     WAV = "audio/wav"
 
     # files
@@ -121,13 +120,11 @@ class AgentResponse[R: BaseModel](BaseModel):
         GeminiModel.FLASH_8B: 30,
         GeminiModel.FLASH: 15,
         GeminiModel.PRO: 500,
-        GeminiModel.EXP: 0,
     }
     OUTPUT_CENTS_PER_MILLION_TOKENS: ClassVar[dict[GeminiModel, float]] = {
         GeminiModel.FLASH_8B: 30,
         GeminiModel.FLASH: 15,
         GeminiModel.PRO: 500,
-        GeminiModel.EXP: 0,
     }
 
     response: R

{goose_py-0.3.10 → goose_py-0.3.12}/goose/flow.py

@@ -159,7 +159,7 @@ class FlowRun:
         else:
             return NodeState[task.result_type](
                 task_name=task.name,
-                index=index or 0,
+                index=index,
                 conversation=Conversation[task.result_type](
                     user_messages=[], result_messages=[]
                 ),
@@ -186,6 +186,11 @@ class FlowRun:
         self._id = ""
         self._agent = None
 
+    def clear_node(self, *, task: "Task[Any, Result]", index: int) -> None:
+        key = (task.name, index)
+        if key in self._node_states:
+            del self._node_states[key]
+
     def dump(self) -> SerializedFlowRun:
         return SerializedFlowRun(
             json.dumps(

{goose_py-0.3.10 → goose_py-0.3.12}/goose/store.py

@@ -10,6 +10,7 @@ class IFlowRunStore(Protocol):
     def __init__(self, *, flow_name: str) -> None: ...
     async def get(self, *, run_id: str) -> FlowRun | None: ...
     async def save(self, *, run: FlowRun) -> None: ...
+    async def delete(self, *, run_id: str) -> None: ...
 
 
 class InMemoryFlowRunStore(IFlowRunStore):
@@ -22,3 +23,6 @@ class InMemoryFlowRunStore(IFlowRunStore):
 
     async def save(self, *, run: FlowRun) -> None:
         self._runs[run.id] = run
+
+    async def delete(self, *, run_id: str) -> None:
+        self._runs.pop(run_id, None)

{goose_py-0.3.10 → goose_py-0.3.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "goose-py"
-version = "0.3.10"
+version = "0.3.12"
 description = "A tool for AI workflows based on human-computer collaboration and structured output."
 authors = [
     "Nash Taylor <nash@chelle.ai>",