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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (5) hide show
  1. goose/_internal/agent.py +9 -9
  2. goose/_internal/task.py +8 -3
  3. {goose_py-0.11.21.dist-info → goose_py-0.11.22.dist-info}/METADATA +2 -2
  4. {goose_py-0.11.21.dist-info → goose_py-0.11.22.dist-info}/RECORD +5 -5
  5. {goose_py-0.11.21.dist-info → goose_py-0.11.22.dist-info}/WHEEL +0 -0
goose/_internal/agent.py CHANGED
@@ -5,10 +5,10 @@ from typing import Any, Literal, Protocol, overload
5
5
  from aikernel import (
6
6
  LLMAssistantMessage,
7
7
  LLMModelAlias,
8
+ LLMRouter,
8
9
  LLMSystemMessage,
9
10
  LLMToolMessage,
10
11
  LLMUserMessage,
11
- Router,
12
12
  llm_structured,
13
13
  llm_unstructured,
14
14
  )
@@ -43,7 +43,7 @@ class Agent:
43
43
  messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
44
44
  model: LLMModelAlias,
45
45
  task_name: str,
46
- router: Router,
46
+ router: LLMRouter[LLMModelAlias],
47
47
  response_model: type[R] = TextResult,
48
48
  ) -> R:
49
49
  start_time = datetime.now()
@@ -94,7 +94,7 @@ class Agent:
94
94
  messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
95
95
  model: LLMModelAlias,
96
96
  task_name: str,
97
- router: Router,
97
+ router: LLMRouter[LLMModelAlias],
98
98
  ) -> str:
99
99
  start_time = datetime.now()
100
100
  typed_messages: list[ExpectedMessage] = [*messages]
@@ -134,7 +134,7 @@ class Agent:
134
134
  *,
135
135
  messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
136
136
  model: LLMModelAlias,
137
- router: Router,
137
+ router: LLMRouter[LLMModelAlias],
138
138
  task_name: str,
139
139
  response_model: type[R],
140
140
  ) -> R:
@@ -186,7 +186,7 @@ class Agent:
186
186
  *,
187
187
  messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
188
188
  model: LLMModelAlias,
189
- router: Router,
189
+ router: LLMRouter[LLMModelAlias],
190
190
  task_name: str,
191
191
  mode: Literal["generate"],
192
192
  response_model: type[R],
@@ -198,7 +198,7 @@ class Agent:
198
198
  *,
199
199
  messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
200
200
  model: LLMModelAlias,
201
- router: Router,
201
+ router: LLMRouter[LLMModelAlias],
202
202
  task_name: str,
203
203
  mode: Literal["ask"],
204
204
  response_model: type[R] = TextResult,
@@ -210,7 +210,7 @@ class Agent:
210
210
  *,
211
211
  messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
212
212
  model: LLMModelAlias,
213
- router: Router,
213
+ router: LLMRouter[LLMModelAlias],
214
214
  task_name: str,
215
215
  response_model: type[R],
216
216
  mode: Literal["refine"],
@@ -222,7 +222,7 @@ class Agent:
222
222
  *,
223
223
  messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
224
224
  model: LLMModelAlias,
225
- router: Router,
225
+ router: LLMRouter[LLMModelAlias],
226
226
  task_name: str,
227
227
  response_model: type[R],
228
228
  ) -> R: ...
@@ -232,7 +232,7 @@ class Agent:
232
232
  *,
233
233
  messages: list[LLMUserMessage | LLMAssistantMessage | LLMSystemMessage],
234
234
  model: LLMModelAlias,
235
- router: Router,
235
+ router: LLMRouter[LLMModelAlias],
236
236
  task_name: str,
237
237
  response_model: type[R] = TextResult,
238
238
  mode: Literal["generate", "ask", "refine"] = "generate",
goose/_internal/task.py CHANGED
@@ -2,7 +2,7 @@ import hashlib
2
2
  from collections.abc import Awaitable, Callable
3
3
  from typing import Any, overload
4
4
 
5
- from aikernel import LLMModelAlias, LLMSystemMessage, LLMUserMessage, Router
5
+ from aikernel import LLMModelAlias, LLMRouter, LLMSystemMessage, LLMUserMessage
6
6
  from pydantic import BaseModel
7
7
 
8
8
  from goose._internal.agent import Agent
@@ -45,7 +45,12 @@ class Task[**P, R: Result]:
45
45
  return self.result_type.model_validate_json(state.raw_result)
46
46
 
47
47
  async def ask(
48
- self, *, user_message: LLMUserMessage, router: Router, context: LLMSystemMessage | None = None, index: int = 0
48
+ self,
49
+ *,
50
+ user_message: LLMUserMessage,
51
+ router: LLMRouter[LLMModelAlias],
52
+ context: LLMSystemMessage | None = None,
53
+ index: int = 0,
49
54
  ) -> str:
50
55
  flow_run = self.__get_current_flow_run()
51
56
  node_state = flow_run.get_state(task=self, index=index)
@@ -73,7 +78,7 @@ class Task[**P, R: Result]:
73
78
  self,
74
79
  *,
75
80
  user_message: LLMUserMessage,
76
- router: Router,
81
+ router: LLMRouter[LLMModelAlias],
77
82
  context: LLMSystemMessage | None = None,
78
83
  index: int = 0,
79
84
  ) -> R:
{goose_py-0.11.21.dist-info → goose_py-0.11.22.dist-info}/METADATA RENAMED
@@ -1,10 +1,10 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: goose-py
3
- Version: 0.11.21
3
+ Version: 0.11.22
4
4
  Summary: A tool for AI workflows based on human-computer collaboration and structured output.
5
5
  Author-email: Nash Taylor <nash@chelle.ai>, Joshua Cook <joshua@chelle.ai>, Michael Sankur <michael@chelle.ai>
6
6
  Requires-Python: >=3.12
7
- Requires-Dist: aikernel==0.1.41
7
+ Requires-Dist: aikernel==0.1.42
8
8
  Requires-Dist: jsonpath-ng>=1.7.0
9
9
  Requires-Dist: pydantic>=2.8.2
10
10
  Description-Content-Type: text/markdown
{goose_py-0.11.21.dist-info → goose_py-0.11.22.dist-info}/RECORD RENAMED
@@ -4,15 +4,15 @@ goose/flow.py,sha256=YsZLBa5I1W27_P6LYGWbtFX8ZYx9vJG3KtENYChHm5E,111
4
4
  goose/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
5
5
  goose/runs.py,sha256=ub-r_gzbUbaIzWXX-jc-dncNxEh6zTfzIkmnDfCSbRI,160
6
6
  goose/task.py,sha256=95rspdxETJoY12IHBl3KjnVIdqQnf1jDKlnGWNWOTvQ,53
7
- goose/_internal/agent.py,sha256=CpVvuyDkNrFLsVIHaf0Tbvp48vgTxyjQdlETsFdbvJ8,9088
7
+ goose/_internal/agent.py,sha256=Wn-bWCG14D60XBmlZoyp2OFX7khw8tdtYv5zsPC5mwA,9235
8
8
  goose/_internal/conversation.py,sha256=vhJwe1pHk2lV60DaB9Tz9KbpzQo7_thRYInPjbIoUTE,1437
9
9
  goose/_internal/flow.py,sha256=8MJxlhHYSAzUHZefpF_sRJc37o532OF0X7l3KRopDmc,4115
10
10
  goose/_internal/result.py,sha256=vtJMfBxb9skfl8st2tn4hBmEq6qmXiJTme_B5QTgu2M,538
11
11
  goose/_internal/state.py,sha256=kA116MpsetsQz6nYodsXOqE3uYz37OTgjC9Vcy_3Qvg,8065
12
12
  goose/_internal/store.py,sha256=tWmKfa1-yq1jU6lT3l6kSOmVt2m3H7I1xLMTrxnUDI8,889
13
- goose/_internal/task.py,sha256=Qj7Z5CowDAwjcdNJTgfIDEsu7bYr5wUchgJxF4mDp04,6547
13
+ goose/_internal/task.py,sha256=27xzbD_4E_j3Cn7JCIxXFiYMgLzVomCKF5I1QInRb2U,6627
14
14
  goose/_internal/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
15
15
  goose/_internal/types/telemetry.py,sha256=xpfkhx7zCdZjjKU8rPuUJ2UHgqmZ084ZupzbTU36gSM,3791
16
- goose_py-0.11.21.dist-info/METADATA,sha256=aGnScXNJXuSV-QRChM_UqGcP6WqcinEkEJ3D8IPGOZI,444
17
- goose_py-0.11.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
18
- goose_py-0.11.21.dist-info/RECORD,,
16
+ goose_py-0.11.22.dist-info/METADATA,sha256=FwtMXDWnfQrb4Z2zmBCAalMgGAnqtNc1wxwCGQlx6uM,444
17
+ goose_py-0.11.22.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
18
+ goose_py-0.11.22.dist-info/RECORD,,