goose-py 0.11.23__tar.gz → 0.11.24__tar.gz

{goose_py-0.11.23 → goose_py-0.11.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: goose-py
-Version: 0.11.23
+Version: 0.11.24
 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.11.23 → goose_py-0.11.24}/goose/_internal/task.py

@@ -18,11 +18,9 @@ class Task[**P, R: Result]:
         /,
         *,
         retries: int = 0,
-        refinement_model: LLMModelAlias = "gemini-2.0-flash-lite",
     ) -> None:
         self._generator = generator
         self._retries = retries
-        self._refinement_model: LLMModelAlias = refinement_model
 
     @property
     def result_type(self) -> type[R]:
@@ -44,11 +42,12 @@ class Task[**P, R: Result]:
         else:
             return self.result_type.model_validate_json(state.raw_result)
 
-    async def ask(
+    async def ask[M: LLMModelAlias](
         self,
         *,
         user_message: LLMUserMessage,
-        router: LLMRouter[LLMModelAlias],
+        router: LLMRouter[M],
+        model: M,
         context: LLMSystemMessage | None = None,
         index: int = 0,
     ) -> str:
@@ -64,7 +63,7 @@ class Task[**P, R: Result]:
 
         answer = await flow_run.agent(
             messages=node_state.conversation.render(),
-            model=self._refinement_model,
+            model=model,
             task_name=f"ask--{self.name}",
             mode="ask",
             router=router,
@@ -74,11 +73,12 @@ class Task[**P, R: Result]:
 
         return answer
 
-    async def refine(
+    async def refine[M: LLMModelAlias](
         self,
         *,
         user_message: LLMUserMessage,
-        router: LLMRouter[LLMModelAlias],
+        router: LLMRouter[M],
+        model: M,
         context: LLMSystemMessage | None = None,
         index: int = 0,
     ) -> R:
@@ -94,7 +94,7 @@ class Task[**P, R: Result]:
 
         result = await flow_run.agent(
             messages=node_state.conversation.render(),
-            model=self._refinement_model,
+            model=model,
             task_name=f"refine--{self.name}",
             response_model=self.result_type,
             mode="refine",
@@ -161,21 +161,18 @@ class Task[**P, R: Result]:
 @overload
 def task[**P, R: Result](generator: Callable[P, Awaitable[R]], /) -> Task[P, R]: ...
 @overload
-def task[**P, R: Result](
-    *, retries: int = 0, refinement_model: LLMModelAlias = "gemini-2.0-flash-lite"
-) -> Callable[[Callable[P, Awaitable[R]]], Task[P, R]]: ...
+def task[**P, R: Result](*, retries: int = 0) -> Callable[[Callable[P, Awaitable[R]]], Task[P, R]]: ...
 def task[**P, R: Result](
     generator: Callable[P, Awaitable[R]] | None = None,
     /,
     *,
     retries: int = 0,
-    refinement_model: LLMModelAlias = "gemini-2.0-flash-lite",
 ) -> Task[P, R] | Callable[[Callable[P, Awaitable[R]]], Task[P, R]]:
     if generator is None:
 
         def decorator(fn: Callable[P, Awaitable[R]]) -> Task[P, R]:
-            return Task(fn, retries=retries, refinement_model=refinement_model)
+            return Task(fn, retries=retries)
 
         return decorator
 
-    return Task(generator, retries=retries, refinement_model=refinement_model)
+    return Task(generator, retries=retries)

{goose_py-0.11.23 → goose_py-0.11.24}/pyproject.toml

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

{goose_py-0.11.23 → goose_py-0.11.24}/tests/test_ask.py

@@ -58,6 +58,7 @@ async def test_ask_adds_to_conversation():
         response = await basic_task.ask(
             user_message=LLMUserMessage(parts=[LLMMessagePart(content="Can you explain how you got that?")]),
             router=ROUTER,
+            model="gemini-2.0-flash-lite",
         )
 
         # Verify the response exists and makes sense
@@ -94,6 +95,7 @@ async def test_ask_requires_completed_task():
             await basic_task.ask(
                 user_message=LLMUserMessage(parts=[LLMMessagePart(content="Can you explain?")]),
                 router=ROUTER,
+                model="gemini-2.0-flash-lite",
             )
 
 
@@ -113,6 +115,7 @@ async def test_ask_multiple_questions():
             response = await basic_task.ask(
                 user_message=LLMUserMessage(parts=[LLMMessagePart(content=question)]),
                 router=ROUTER,
+                model="gemini-2.0-flash-lite",
             )
             responses.append(response)
 

{goose_py-0.11.23 → goose_py-0.11.24}/tests/test_refining.py

@@ -69,6 +69,7 @@ async def test_refining() -> None:
             user_message=LLMUserMessage(parts=[LLMMessagePart(content="Change it")]),
             context=LLMSystemMessage(parts=[LLMMessagePart(content="Extra info")]),
             router=ROUTER,
+            model="gemini-2.0-flash-lite",
         )
 
     initial_random_words = first_run.get_all_results(task=generate_random_word)
@@ -80,6 +81,7 @@ async def test_refining() -> None:
             user_message=LLMUserMessage(parts=[LLMMessagePart(content="Change it")]),
             context=LLMSystemMessage(parts=[LLMMessagePart(content="Extra info")]),
             router=ROUTER,
+            model="gemini-2.0-flash-lite",
         )
         # Since refine now directly returns the result from the agent call
         assert isinstance(result, GeneratedWord)
@@ -99,4 +101,5 @@ async def test_refining_before_generate_fails() -> None:
                 user_message=LLMUserMessage(parts=[LLMMessagePart(content="Change it")]),
                 context=LLMSystemMessage(parts=[LLMMessagePart(content="Extra info")]),
                 router=ROUTER,
+                model="gemini-2.0-flash-lite",
             )

{goose_py-0.11.23 → goose_py-0.11.24}/tests/test_state.py

@@ -76,6 +76,7 @@ async def test_state_undo() -> None:
             user_message=LLMUserMessage(parts=[LLMMessagePart(content="Change it")]),
             context=LLMSystemMessage(parts=[LLMMessagePart(content="Extra info")]),
             router=Mock(),
+            model="gemini-2.0-flash-lite",
         )
 
     async with with_state.start_run(run_id="2") as run:

{goose_py-0.11.23 → goose_py-0.11.24}/uv.lock

@@ -345,7 +345,7 @@ wheels = [
 
 [[package]]
 name = "goose-py"
-version = "0.11.23"
+version = "0.11.24"
 source = { editable = "." }
 dependencies = [
     { name = "aikernel" },