triggerware 1.0.1__tar.gz → 1.0.3__tar.gz
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.
- {triggerware-1.0.1 → triggerware-1.0.3}/PKG-INFO +1 -1
- {triggerware-1.0.1 → triggerware-1.0.3}/pyproject.toml +1 -1
- triggerware-1.0.3/src/triggerware/_agents.py +13 -0
- {triggerware-1.0.1 → triggerware-1.0.3}/src/triggerware/_client.py +2 -2
- {triggerware-1.0.1 → triggerware-1.0.3}/src/triggerware/_configuration.py +8 -8
- {triggerware-1.0.1 → triggerware-1.0.3}/src/triggerware/_queries.py +0 -18
- {triggerware-1.0.1 → triggerware-1.0.3}/src/triggerware/_triggers.py +0 -6
- triggerware-1.0.1/src/triggerware/_prompts.py +0 -25
- {triggerware-1.0.1 → triggerware-1.0.3}/README.md +0 -0
- {triggerware-1.0.1 → triggerware-1.0.3}/src/triggerware/__init__.py +0 -0
- {triggerware-1.0.1 → triggerware-1.0.3}/src/triggerware/_connectors.py +0 -0
- {triggerware-1.0.1 → triggerware-1.0.3}/src/triggerware/_cursors.py +0 -0
- {triggerware-1.0.1 → triggerware-1.0.3}/src/triggerware/_extensions.py +0 -0
- {triggerware-1.0.1 → triggerware-1.0.3}/src/triggerware/_logging.py +0 -0
- {triggerware-1.0.1 → triggerware-1.0.3}/src/triggerware/_runtimes.py +0 -0
- {triggerware-1.0.1 → triggerware-1.0.3}/src/triggerware/_system.py +0 -0
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
|
|
3
|
+
from typing import Any
|
|
4
|
+
|
|
5
|
+
import tw2_jsonrpc as rpc
|
|
6
|
+
|
|
7
|
+
|
|
8
|
+
class AgentsService:
|
|
9
|
+
def __init__(self, peer: rpc.Peer):
|
|
10
|
+
self._peer = peer
|
|
11
|
+
|
|
12
|
+
async def run(self, name: str, data: dict[str, Any]) -> None:
|
|
13
|
+
await self._peer.call("agents.run", {"name": name, "data": data})
|
|
@@ -9,7 +9,7 @@ from ._connectors import ConnectorsService
|
|
|
9
9
|
from ._cursors import CursorsService
|
|
10
10
|
from ._extensions import ExtensionsService
|
|
11
11
|
from ._logging import LoggingService
|
|
12
|
-
from .
|
|
12
|
+
from ._agents import AgentsService
|
|
13
13
|
from ._queries import QueriesService
|
|
14
14
|
from ._runtimes import RuntimesService
|
|
15
15
|
from ._system import SystemService
|
|
@@ -25,7 +25,7 @@ class TriggerwareClient:
|
|
|
25
25
|
self.cursors = CursorsService(peer)
|
|
26
26
|
self.extensions = ExtensionsService(peer)
|
|
27
27
|
self.logging = LoggingService(peer)
|
|
28
|
-
self.
|
|
28
|
+
self.agents = AgentsService(peer)
|
|
29
29
|
self.queries = QueriesService(peer)
|
|
30
30
|
self.runtimes = RuntimesService(peer)
|
|
31
31
|
self.system = SystemService(peer)
|
|
@@ -9,8 +9,8 @@ class Configuration(TypedDict):
|
|
|
9
9
|
default_timeout_ms: NotRequired[int]
|
|
10
10
|
default_batch_size: NotRequired[int]
|
|
11
11
|
default_trigger_batch_size: NotRequired[int]
|
|
12
|
-
|
|
13
|
-
|
|
12
|
+
agent_provider: NotRequired[str]
|
|
13
|
+
agent_model: NotRequired[str]
|
|
14
14
|
max_log_file_size: NotRequired[str]
|
|
15
15
|
|
|
16
16
|
|
|
@@ -27,18 +27,18 @@ class ConfigurationService:
|
|
|
27
27
|
default_timeout_ms: int | None = None,
|
|
28
28
|
default_batch_size: int | None = None,
|
|
29
29
|
default_trigger_batch_size: int | None = None,
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
30
|
+
agent_provider: str | None = None,
|
|
31
|
+
agent_model: str | None = None,
|
|
32
|
+
agent_api_key: str | None = None,
|
|
33
33
|
max_log_file_size: str | None = None,
|
|
34
34
|
) -> Configuration:
|
|
35
35
|
params = {
|
|
36
36
|
"default_timeout_ms": default_timeout_ms,
|
|
37
37
|
"default_batch_size": default_batch_size,
|
|
38
38
|
"default_trigger_batch_size": default_trigger_batch_size,
|
|
39
|
-
"
|
|
40
|
-
"
|
|
41
|
-
"
|
|
39
|
+
"agent_provider": agent_provider,
|
|
40
|
+
"agent_model": agent_model,
|
|
41
|
+
"agent_api_key": agent_api_key,
|
|
42
42
|
"max_log_file_size": max_log_file_size,
|
|
43
43
|
}
|
|
44
44
|
return await self._peer.call(
|
|
@@ -37,24 +37,6 @@ class Query:
|
|
|
37
37
|
self._cache: list[list[Any]] = []
|
|
38
38
|
self._cache_position = 0
|
|
39
39
|
|
|
40
|
-
async def update(
|
|
41
|
-
self,
|
|
42
|
-
*,
|
|
43
|
-
sql: str | None = None,
|
|
44
|
-
prompt: str | None = None,
|
|
45
|
-
batch_size: int | None = None,
|
|
46
|
-
timeout_ms: int | None = None,
|
|
47
|
-
) -> None:
|
|
48
|
-
params = _query_params(
|
|
49
|
-
name=self.name,
|
|
50
|
-
sql=sql,
|
|
51
|
-
prompt=prompt,
|
|
52
|
-
batch_size=batch_size,
|
|
53
|
-
timeout_ms=timeout_ms,
|
|
54
|
-
)
|
|
55
|
-
definition: QueryDefinition = await self._peer.call("queries.update", params)
|
|
56
|
-
self._replace(definition)
|
|
57
|
-
|
|
58
40
|
async def reset(self) -> None:
|
|
59
41
|
definition: QueryDefinition = await self._peer.call(
|
|
60
42
|
"queries.reset",
|
|
@@ -16,7 +16,6 @@ class TriggerDefinition(TypedDict):
|
|
|
16
16
|
prompt: NotRequired[str]
|
|
17
17
|
sql: NotRequired[str]
|
|
18
18
|
schedule: NotRequired[str]
|
|
19
|
-
delivery: NotRequired[list[str]]
|
|
20
19
|
batch_size: NotRequired[int]
|
|
21
20
|
timeout_ms: NotRequired[int]
|
|
22
21
|
report_noops: bool
|
|
@@ -51,7 +50,6 @@ class Trigger:
|
|
|
51
50
|
sql: str | None = None,
|
|
52
51
|
prompt: str | None = None,
|
|
53
52
|
schedule: str | None = None,
|
|
54
|
-
delivery: list[str] | None = None,
|
|
55
53
|
batch_size: int | None = None,
|
|
56
54
|
timeout_ms: int | None = None,
|
|
57
55
|
report_noops: bool | None = None,
|
|
@@ -62,7 +60,6 @@ class Trigger:
|
|
|
62
60
|
sql=sql,
|
|
63
61
|
prompt=prompt,
|
|
64
62
|
schedule=schedule,
|
|
65
|
-
delivery=delivery,
|
|
66
63
|
batch_size=batch_size,
|
|
67
64
|
timeout_ms=timeout_ms,
|
|
68
65
|
report_noops=report_noops,
|
|
@@ -165,7 +162,6 @@ class Trigger:
|
|
|
165
162
|
self.sql = definition.get("sql")
|
|
166
163
|
self.prompt = definition.get("prompt")
|
|
167
164
|
self.schedule = definition.get("schedule")
|
|
168
|
-
self.delivery = definition.get("delivery", [])
|
|
169
165
|
self.batch_size = definition.get("batch_size")
|
|
170
166
|
self.timeout_ms = definition.get("timeout_ms")
|
|
171
167
|
self.report_noops = definition["report_noops"]
|
|
@@ -193,7 +189,6 @@ class TriggersService:
|
|
|
193
189
|
sql: str | None = None,
|
|
194
190
|
prompt: str | None = None,
|
|
195
191
|
schedule: str | None = None,
|
|
196
|
-
delivery: list[str] | None = None,
|
|
197
192
|
batch_size: int | None = None,
|
|
198
193
|
timeout_ms: int | None = None,
|
|
199
194
|
report_noops: bool | None = None,
|
|
@@ -206,7 +201,6 @@ class TriggersService:
|
|
|
206
201
|
sql=sql,
|
|
207
202
|
prompt=prompt,
|
|
208
203
|
schedule=schedule,
|
|
209
|
-
delivery=delivery,
|
|
210
204
|
batch_size=batch_size,
|
|
211
205
|
timeout_ms=timeout_ms,
|
|
212
206
|
report_noops=report_noops,
|
|
@@ -1,25 +0,0 @@
|
|
|
1
|
-
from __future__ import annotations
|
|
2
|
-
|
|
3
|
-
import tw2_jsonrpc as rpc
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
class PromptsService:
|
|
7
|
-
def __init__(self, peer: rpc.Peer):
|
|
8
|
-
self._peer = peer
|
|
9
|
-
|
|
10
|
-
async def get(self, name: str) -> str:
|
|
11
|
-
response = await self._peer.call("prompts.get", {"name": name})
|
|
12
|
-
return response["prompt"]
|
|
13
|
-
|
|
14
|
-
async def set(self, name: str, prompt: str) -> None:
|
|
15
|
-
await self._peer.call(
|
|
16
|
-
"prompts.set",
|
|
17
|
-
{
|
|
18
|
-
"name": name,
|
|
19
|
-
"prompt": prompt,
|
|
20
|
-
},
|
|
21
|
-
)
|
|
22
|
-
|
|
23
|
-
async def reset(self, name: str | None = None) -> None:
|
|
24
|
-
params = {} if name is None else {"name": name}
|
|
25
|
-
await self._peer.call("prompts.reset", params)
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|