goose-py 0.11.23__py3-none-any.whl → 0.11.25__py3-none-any.whl

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.dist-info → goose_py-0.11.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: goose-py
-Version: 0.11.23
+Version: 0.11.25
 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.dist-info → goose_py-0.11.25.dist-info}/RECORD

@@ -10,9 +10,9 @@ goose/_internal/flow.py,sha256=8MJxlhHYSAzUHZefpF_sRJc37o532OF0X7l3KRopDmc,4115
 goose/_internal/result.py,sha256=vtJMfBxb9skfl8st2tn4hBmEq6qmXiJTme_B5QTgu2M,538
 goose/_internal/state.py,sha256=kA116MpsetsQz6nYodsXOqE3uYz37OTgjC9Vcy_3Qvg,8065
 goose/_internal/store.py,sha256=tWmKfa1-yq1jU6lT3l6kSOmVt2m3H7I1xLMTrxnUDI8,889
-goose/_internal/task.py,sha256=27xzbD_4E_j3Cn7JCIxXFiYMgLzVomCKF5I1QInRb2U,6627
+goose/_internal/task.py,sha256=A13TsBQJlo1yuAqbUpDY8k15JAd5WZxqgrz32F18H2U,6311
 goose/_internal/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 goose/_internal/types/telemetry.py,sha256=xpfkhx7zCdZjjKU8rPuUJ2UHgqmZ084ZupzbTU36gSM,3791
-goose_py-0.11.23.dist-info/METADATA,sha256=k-jLSjvNxBP1aao0ZymSjCiTYXU4PR0OZgbBCIRcKYk,444
-goose_py-0.11.23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-goose_py-0.11.23.dist-info/RECORD,,
+goose_py-0.11.25.dist-info/METADATA,sha256=varvcbhfLhYcbexYd9BFx_Usp2PMxsgboBWspwXOc1I,444
+goose_py-0.11.25.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+goose_py-0.11.25.dist-info/RECORD,,