goose-py 0.11.10__tar.gz → 0.11.12__tar.gz

{goose_py-0.11.10 → goose_py-0.11.12}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: goose-py
-Version: 0.11.10
+Version: 0.11.12
 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
-Requires-Dist: aikernel==0.1.30
+Requires-Dist: aikernel==0.1.32
 Requires-Dist: jsonpath-ng>=1.7.0
 Requires-Dist: pydantic>=2.8.2
 Description-Content-Type: text/markdown

{goose_py-0.11.10 → goose_py-0.11.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "goose-py"
-version = "0.11.10"
+version = "0.11.12"
 description = "A tool for AI workflows based on human-computer collaboration and structured output."
 readme = "README.md"
 authors = [
@@ -11,7 +11,7 @@ authors = [
 requires-python = ">=3.12"
 dependencies = [
     "jsonpath-ng>=1.7.0",
-    "aikernel==0.1.30",
+    "aikernel==0.1.32",
     "pydantic>=2.8.2",
 ]
 

{goose_py-0.11.10 → goose_py-0.11.12}/tests/test_refining.py

@@ -1,10 +1,13 @@
 import random
 import string
+from unittest.mock import Mock
 
 import pytest
 from aikernel import LLMMessagePart, LLMSystemMessage, LLMUserMessage
+from pytest_mock import MockerFixture
 
 from goose import Agent, FlowArguments, Result, flow, task
+from goose._internal.result import FindReplaceResponse, Replacement
 from goose.errors import Honk
 
 
@@ -36,11 +39,32 @@ async def sentence(*, flow_arguments: MyFlowArguments, agent: Agent) -> None:
     await make_sentence(words=words)
 
 
+@pytest.fixture
+def mock_llm_structured(mocker: MockerFixture) -> Mock:
+    return mocker.patch(
+        "goose._internal.agent.llm_structured",
+        return_value=Mock(
+            structured_response=FindReplaceResponse(
+                replacements=[Replacement(find="a", replace="b")],
+            ),
+            usage=Mock(input_tokens=10, output_tokens=10),
+        ),
+    )
+
+
 @pytest.mark.asyncio
+@pytest.mark.usefixtures("mock_llm_structured")
 async def test_refining() -> None:
-    async with sentence.start_run(run_id="1") as first_run:
+    async with sentence.start_run(run_id="1"):
         await sentence.generate(MyFlowArguments())
 
+    async with sentence.start_run(run_id="1") as first_run:
+        await generate_random_word.refine(
+            index=0,
+            user_message=LLMUserMessage(parts=[LLMMessagePart(content="Change it")]),
+            context=LLMSystemMessage(parts=[LLMMessagePart(content="Extra info")]),
+        )
+
     initial_random_words = first_run.get_all_results(task=generate_random_word)
     assert len(initial_random_words) == 3
 

{goose_py-0.11.10 → goose_py-0.11.12}/uv.lock

@@ -3,16 +3,16 @@ requires-python = ">=3.12"
 
 [[package]]
 name = "aikernel"
-version = "0.1.30"
+version = "0.1.32"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "litellm" },
     { name = "openai" },
     { name = "pydantic" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/63/37/4588449d03f71080da26d83d82be2732165a26d03560754ba1b5d112be4b/aikernel-0.1.30.tar.gz", hash = "sha256:560d2c55d2a800ed6cec108e213f4f59373eddcb0b92c134c006a18cfcd6b090", size = 71528 }
+sdist = { url = "https://files.pythonhosted.org/packages/7f/8a/f372bb5457ea3b19b8e84f49fa15288abdf56ca30af57ba15de7b0678a97/aikernel-0.1.32.tar.gz", hash = "sha256:8581034882ade6d18d1a78237101c8a3972004feb08ca2f212623ce63f6ebc86", size = 71876 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/7a/3a/e4811df7cef5f9e94afd1065c45152815d80621eb2c9b64a6f5239275a7e/aikernel-0.1.30-py3-none-any.whl", hash = "sha256:1c62b2af6725ec996345def23bce87fba884250d91f958051374cc6b78451a50", size = 8812 },
+    { url = "https://files.pythonhosted.org/packages/d1/90/3ef733ae865525a14d6bf12747d16312628edbf800b01e8944eb6b52084a/aikernel-0.1.32-py3-none-any.whl", hash = "sha256:6bdc9983602613aeb73e08f68180e857cc3df92d60029fb2f3c86fd2ae8f1206", size = 9262 },
 ]
 
 [[package]]
@@ -346,7 +346,7 @@ wheels = [
 
 [[package]]
 name = "goose-py"
-version = "0.11.10"
+version = "0.11.12"
 source = { editable = "." }
 dependencies = [
     { name = "aikernel" },
@@ -366,7 +366,7 @@ dev = [
 
 [package.metadata]
 requires-dist = [
-    { name = "aikernel", specifier = "==0.1.30" },
+    { name = "aikernel", specifier = "==0.1.32" },
     { name = "jsonpath-ng", specifier = ">=1.7.0" },
     { name = "pydantic", specifier = ">=2.8.2" },
 ]