goose-py 0.7.1__tar.gz → 0.7.2__tar.gz

{goose_py-0.7.1 → goose_py-0.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: goose-py
-Version: 0.7.1
+Version: 0.7.2
 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.7.1 → goose_py-0.7.2}/goose/_internal/conversation.py

@@ -1,12 +1,9 @@
+from typing import Self
+
 from pydantic import BaseModel
 
 from goose._internal.result import Result
-from goose._internal.types.agent import (
-    AssistantMessage,
-    LLMMessage,
-    SystemMessage,
-    UserMessage,
-)
+from goose._internal.types.agent import AssistantMessage, LLMMessage, SystemMessage, UserMessage
 
 
 class Conversation[R: Result](BaseModel):
@@ -31,3 +28,8 @@ class Conversation[R: Result](BaseModel):
             messages.append(AssistantMessage(text=self.result_messages[-1].model_dump_json()).render())
 
         return messages
+
+    def undo(self) -> Self:
+        self.user_messages.pop()
+        self.result_messages.pop()
+        return self

{goose_py-0.7.1 → goose_py-0.7.2}/goose/_internal/state.py

@@ -61,6 +61,10 @@ class NodeState[ResultT: Result](BaseModel):
         self.conversation.user_messages.append(message)
         return self
 
+    def undo(self) -> Self:
+        self.conversation.undo()
+        return self
+
 
 class FlowRun:
     def __init__(self) -> None:

{goose_py-0.7.1 → goose_py-0.7.2}/goose/_internal/task.py

@@ -63,6 +63,12 @@ class Task[**P, R: Result]:
 
         return result
 
+    def undo(self, *, index: int = 0) -> None:
+        flow_run = self.__get_current_flow_run()
+        node_state = flow_run.get(task=self, index=index)
+        node_state.undo()
+        flow_run.add_node_state(node_state)
+
     async def __call__(self, *args: P.args, **kwargs: P.kwargs) -> R:
         flow_run = self.__get_current_flow_run()
         node_state = flow_run.get_next(task=self)

{goose_py-0.7.1 → goose_py-0.7.2}/pyproject.toml

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

{goose_py-0.7.1 → goose_py-0.7.2}/tests/test_state.py

@@ -1,9 +1,12 @@
 import random
 import string
+from unittest.mock import Mock
 
 import pytest
+from pytest_mock import MockerFixture
 
 from goose import Result, flow, task
+from goose._internal.types.agent import SystemMessage, TextMessagePart, UserMessage
 from goose.errors import Honk
 
 
@@ -20,6 +23,15 @@ async def generate_random_word(*, n_characters: int) -> GeneratedWord:
     return GeneratedWord(word="".join(random.sample(string.ascii_lowercase, n_characters)))
 
 
+@pytest.fixture
+def generate_random_word_adapter(mocker: MockerFixture) -> Mock:
+    return mocker.patch.object(
+        generate_random_word,
+        "_Task__adapt",
+        return_value=GeneratedWord(word="__ADAPTED__"),
+    )
+
+
 @task
 async def make_sentence(*, words: list[GeneratedWord]) -> GeneratedSentence:
     return GeneratedSentence(sentence=" ".join([word.word for word in words]))
@@ -47,3 +59,22 @@ async def test_state_causes_caching() -> None:
     new_random_word = new_run.get(task=generate_random_word).result.word
 
     assert random_word == new_random_word  # unchanged node is not re-generated
+
+
+@pytest.mark.asyncio
+@pytest.mark.usefixtures("generate_random_word_adapter")
+async def test_state_undo() -> None:
+    async with with_state.start_run(run_id="2"):
+        await with_state.generate()
+
+    async with with_state.start_run(run_id="2"):
+        await generate_random_word.jam(
+            index=0,
+            user_message=UserMessage(parts=[TextMessagePart(text="Change it")]),
+            context=SystemMessage(parts=[TextMessagePart(text="Extra info")]),
+        )
+
+    async with with_state.start_run(run_id="2") as run:
+        generate_random_word.undo()
+
+    assert run.get(task=generate_random_word).result.word != "__ADAPTED__"

{goose_py-0.7.1 → goose_py-0.7.2}/uv.lock

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