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

goose/_internal/agent.py

@@ -37,13 +37,13 @@ class Agent:
         self.run_id = run_id
         self.logger = logger
 
-    async def generate[R: Result](
+    async def generate[R: Result, M: LLMModelAlias](
         self,
         *,
         messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
-        model: LLMModelAlias,
+        model: M,
         task_name: str,
-        router: LLMRouter[LLMModelAlias],
+        router: LLMRouter[M],
         response_model: type[R] = TextResult,
     ) -> R:
         start_time = datetime.now()
@@ -88,13 +88,13 @@ class Agent:
 
         return parsed_response
 
-    async def ask(
+    async def ask[M: LLMModelAlias](
         self,
         *,
         messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
-        model: LLMModelAlias,
+        model: M,
         task_name: str,
-        router: LLMRouter[LLMModelAlias],
+        router: LLMRouter[M],
     ) -> str:
         start_time = datetime.now()
         typed_messages: list[ExpectedMessage] = [*messages]
@@ -129,12 +129,12 @@ class Agent:
 
         return response.text
 
-    async def refine[R: Result](
+    async def refine[R: Result, M: LLMModelAlias](
         self,
         *,
         messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
-        model: LLMModelAlias,
-        router: LLMRouter[LLMModelAlias],
+        model: M,
+        router: LLMRouter[M],
         task_name: str,
         response_model: type[R],
     ) -> R:
@@ -181,58 +181,58 @@ class Agent:
         return refined_response
 
     @overload
-    async def __call__[R: Result](
+    async def __call__[R: Result, M: LLMModelAlias](
         self,
         *,
         messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
-        model: LLMModelAlias,
-        router: LLMRouter[LLMModelAlias],
+        model: M,
+        router: LLMRouter[M],
         task_name: str,
         mode: Literal["generate"],
         response_model: type[R],
     ) -> R: ...
 
     @overload
-    async def __call__[R: Result](
+    async def __call__[R: Result, M: LLMModelAlias](
         self,
         *,
         messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
-        model: LLMModelAlias,
-        router: LLMRouter[LLMModelAlias],
+        model: M,
+        router: LLMRouter[M],
         task_name: str,
         mode: Literal["ask"],
         response_model: type[R] = TextResult,
     ) -> str: ...
 
     @overload
-    async def __call__[R: Result](
+    async def __call__[R: Result, M: LLMModelAlias](
         self,
         *,
         messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
-        model: LLMModelAlias,
-        router: LLMRouter[LLMModelAlias],
+        model: M,
+        router: LLMRouter[M],
         task_name: str,
         response_model: type[R],
         mode: Literal["refine"],
     ) -> R: ...
 
     @overload
-    async def __call__[R: Result](
+    async def __call__[R: Result, M: LLMModelAlias](
         self,
         *,
         messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
-        model: LLMModelAlias,
-        router: LLMRouter[LLMModelAlias],
+        model: M,
+        router: LLMRouter[M],
         task_name: str,
         response_model: type[R],
     ) -> R: ...
 
-    async def __call__[R: Result](
+    async def __call__[R: Result, M: LLMModelAlias](
         self,
         *,
         messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
-        model: LLMModelAlias,
-        router: LLMRouter[LLMModelAlias],
+        model: M,
+        router: LLMRouter[M],
         task_name: str,
         response_model: type[R] = TextResult,
         mode: Literal["generate", "ask", "refine"] = "generate",

{goose_py-0.11.22.dist-info → goose_py-0.11.23.dist-info}/METADATA

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

@@ -4,7 +4,7 @@ goose/flow.py,sha256=YsZLBa5I1W27_P6LYGWbtFX8ZYx9vJG3KtENYChHm5E,111
 goose/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 goose/runs.py,sha256=ub-r_gzbUbaIzWXX-jc-dncNxEh6zTfzIkmnDfCSbRI,160
 goose/task.py,sha256=95rspdxETJoY12IHBl3KjnVIdqQnf1jDKlnGWNWOTvQ,53
-goose/_internal/agent.py,sha256=Wn-bWCG14D60XBmlZoyp2OFX7khw8tdtYv5zsPC5mwA,9235
+goose/_internal/agent.py,sha256=9Hz15SOrhaJWgOXhfQYYW4nrolB2xbMG5G3RwHX_k90,9187
 goose/_internal/conversation.py,sha256=vhJwe1pHk2lV60DaB9Tz9KbpzQo7_thRYInPjbIoUTE,1437
 goose/_internal/flow.py,sha256=8MJxlhHYSAzUHZefpF_sRJc37o532OF0X7l3KRopDmc,4115
 goose/_internal/result.py,sha256=vtJMfBxb9skfl8st2tn4hBmEq6qmXiJTme_B5QTgu2M,538
@@ -13,6 +13,6 @@ goose/_internal/store.py,sha256=tWmKfa1-yq1jU6lT3l6kSOmVt2m3H7I1xLMTrxnUDI8,889
 goose/_internal/task.py,sha256=27xzbD_4E_j3Cn7JCIxXFiYMgLzVomCKF5I1QInRb2U,6627
 goose/_internal/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 goose/_internal/types/telemetry.py,sha256=xpfkhx7zCdZjjKU8rPuUJ2UHgqmZ084ZupzbTU36gSM,3791
-goose_py-0.11.22.dist-info/METADATA,sha256=FwtMXDWnfQrb4Z2zmBCAalMgGAnqtNc1wxwCGQlx6uM,444
-goose_py-0.11.22.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-goose_py-0.11.22.dist-info/RECORD,,
+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,,