@oni.bot/core 0.6.1
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.
- package/README.md +308 -0
- package/dist/agents/context.d.ts +24 -0
- package/dist/agents/context.d.ts.map +1 -0
- package/dist/agents/context.js +93 -0
- package/dist/agents/context.js.map +1 -0
- package/dist/agents/define-agent.d.ts +10 -0
- package/dist/agents/define-agent.d.ts.map +1 -0
- package/dist/agents/define-agent.js +121 -0
- package/dist/agents/define-agent.js.map +1 -0
- package/dist/agents/functional-agent.d.ts +12 -0
- package/dist/agents/functional-agent.d.ts.map +1 -0
- package/dist/agents/functional-agent.js +115 -0
- package/dist/agents/functional-agent.js.map +1 -0
- package/dist/agents/index.d.ts +6 -0
- package/dist/agents/index.d.ts.map +1 -0
- package/dist/agents/index.js +7 -0
- package/dist/agents/index.js.map +1 -0
- package/dist/agents/types.d.ts +62 -0
- package/dist/agents/types.d.ts.map +1 -0
- package/dist/agents/types.js +8 -0
- package/dist/agents/types.js.map +1 -0
- package/dist/checkpoint.d.ts +27 -0
- package/dist/checkpoint.d.ts.map +1 -0
- package/dist/checkpoint.js +91 -0
- package/dist/checkpoint.js.map +1 -0
- package/dist/checkpointers/index.d.ts +4 -0
- package/dist/checkpointers/index.d.ts.map +1 -0
- package/dist/checkpointers/index.js +5 -0
- package/dist/checkpointers/index.js.map +1 -0
- package/dist/checkpointers/namespaced.d.ts +12 -0
- package/dist/checkpointers/namespaced.d.ts.map +1 -0
- package/dist/checkpointers/namespaced.js +37 -0
- package/dist/checkpointers/namespaced.js.map +1 -0
- package/dist/checkpointers/postgres.d.ts +13 -0
- package/dist/checkpointers/postgres.d.ts.map +1 -0
- package/dist/checkpointers/postgres.js +100 -0
- package/dist/checkpointers/postgres.js.map +1 -0
- package/dist/checkpointers/sqlite.d.ts +14 -0
- package/dist/checkpointers/sqlite.d.ts.map +1 -0
- package/dist/checkpointers/sqlite.js +98 -0
- package/dist/checkpointers/sqlite.js.map +1 -0
- package/dist/context.d.ts +24 -0
- package/dist/context.d.ts.map +1 -0
- package/dist/context.js +47 -0
- package/dist/context.js.map +1 -0
- package/dist/coordination/index.d.ts +3 -0
- package/dist/coordination/index.d.ts.map +1 -0
- package/dist/coordination/index.js +3 -0
- package/dist/coordination/index.js.map +1 -0
- package/dist/coordination/pubsub.d.ts +38 -0
- package/dist/coordination/pubsub.d.ts.map +1 -0
- package/dist/coordination/pubsub.js +73 -0
- package/dist/coordination/pubsub.js.map +1 -0
- package/dist/coordination/request-reply.d.ts +40 -0
- package/dist/coordination/request-reply.d.ts.map +1 -0
- package/dist/coordination/request-reply.js +77 -0
- package/dist/coordination/request-reply.js.map +1 -0
- package/dist/errors.d.ts +27 -0
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +41 -0
- package/dist/errors.js.map +1 -0
- package/dist/events/bus.d.ts +13 -0
- package/dist/events/bus.d.ts.map +1 -0
- package/dist/events/bus.js +52 -0
- package/dist/events/bus.js.map +1 -0
- package/dist/events/index.d.ts +3 -0
- package/dist/events/index.d.ts.map +1 -0
- package/dist/events/index.js +3 -0
- package/dist/events/index.js.map +1 -0
- package/dist/events/types.d.ts +87 -0
- package/dist/events/types.d.ts.map +1 -0
- package/dist/events/types.js +2 -0
- package/dist/events/types.js.map +1 -0
- package/dist/functional.d.ts +36 -0
- package/dist/functional.d.ts.map +1 -0
- package/dist/functional.js +103 -0
- package/dist/functional.js.map +1 -0
- package/dist/graph.d.ts +74 -0
- package/dist/graph.d.ts.map +1 -0
- package/dist/graph.js +119 -0
- package/dist/graph.js.map +1 -0
- package/dist/guardrails/audit.d.ts +13 -0
- package/dist/guardrails/audit.d.ts.map +1 -0
- package/dist/guardrails/audit.js +31 -0
- package/dist/guardrails/audit.js.map +1 -0
- package/dist/guardrails/budget.d.ts +30 -0
- package/dist/guardrails/budget.d.ts.map +1 -0
- package/dist/guardrails/budget.js +154 -0
- package/dist/guardrails/budget.js.map +1 -0
- package/dist/guardrails/filters.d.ts +24 -0
- package/dist/guardrails/filters.d.ts.map +1 -0
- package/dist/guardrails/filters.js +87 -0
- package/dist/guardrails/filters.js.map +1 -0
- package/dist/guardrails/index.d.ts +8 -0
- package/dist/guardrails/index.d.ts.map +1 -0
- package/dist/guardrails/index.js +9 -0
- package/dist/guardrails/index.js.map +1 -0
- package/dist/guardrails/permissions.d.ts +9 -0
- package/dist/guardrails/permissions.d.ts.map +1 -0
- package/dist/guardrails/permissions.js +29 -0
- package/dist/guardrails/permissions.js.map +1 -0
- package/dist/guardrails/types.d.ts +30 -0
- package/dist/guardrails/types.d.ts.map +1 -0
- package/dist/guardrails/types.js +2 -0
- package/dist/guardrails/types.js.map +1 -0
- package/dist/hitl/index.d.ts +5 -0
- package/dist/hitl/index.d.ts.map +1 -0
- package/dist/hitl/index.js +3 -0
- package/dist/hitl/index.js.map +1 -0
- package/dist/hitl/interrupt.d.ts +81 -0
- package/dist/hitl/interrupt.d.ts.map +1 -0
- package/dist/hitl/interrupt.js +116 -0
- package/dist/hitl/interrupt.js.map +1 -0
- package/dist/hitl/resume.d.ts +33 -0
- package/dist/hitl/resume.d.ts.map +1 -0
- package/dist/hitl/resume.js +72 -0
- package/dist/hitl/resume.js.map +1 -0
- package/dist/index.d.ts +57 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +58 -0
- package/dist/index.js.map +1 -0
- package/dist/injected.d.ts +26 -0
- package/dist/injected.d.ts.map +1 -0
- package/dist/injected.js +25 -0
- package/dist/injected.js.map +1 -0
- package/dist/inspect.d.ts +31 -0
- package/dist/inspect.d.ts.map +1 -0
- package/dist/inspect.js +139 -0
- package/dist/inspect.js.map +1 -0
- package/dist/messages/index.d.ts +66 -0
- package/dist/messages/index.d.ts.map +1 -0
- package/dist/messages/index.js +123 -0
- package/dist/messages/index.js.map +1 -0
- package/dist/models/anthropic.d.ts +6 -0
- package/dist/models/anthropic.d.ts.map +1 -0
- package/dist/models/anthropic.js +317 -0
- package/dist/models/anthropic.js.map +1 -0
- package/dist/models/google.d.ts +3 -0
- package/dist/models/google.d.ts.map +1 -0
- package/dist/models/google.js +310 -0
- package/dist/models/google.js.map +1 -0
- package/dist/models/index.d.ts +6 -0
- package/dist/models/index.d.ts.map +1 -0
- package/dist/models/index.js +5 -0
- package/dist/models/index.js.map +1 -0
- package/dist/models/ollama.d.ts +7 -0
- package/dist/models/ollama.d.ts.map +1 -0
- package/dist/models/ollama.js +205 -0
- package/dist/models/ollama.js.map +1 -0
- package/dist/models/openai.d.ts +3 -0
- package/dist/models/openai.d.ts.map +1 -0
- package/dist/models/openai.js +331 -0
- package/dist/models/openai.js.map +1 -0
- package/dist/models/types.d.ts +77 -0
- package/dist/models/types.d.ts.map +1 -0
- package/dist/models/types.js +2 -0
- package/dist/models/types.js.map +1 -0
- package/dist/prebuilt/index.d.ts +5 -0
- package/dist/prebuilt/index.d.ts.map +1 -0
- package/dist/prebuilt/index.js +4 -0
- package/dist/prebuilt/index.js.map +1 -0
- package/dist/prebuilt/react-agent.d.ts +35 -0
- package/dist/prebuilt/react-agent.d.ts.map +1 -0
- package/dist/prebuilt/react-agent.js +55 -0
- package/dist/prebuilt/react-agent.js.map +1 -0
- package/dist/prebuilt/tool-node.d.ts +15 -0
- package/dist/prebuilt/tool-node.d.ts.map +1 -0
- package/dist/prebuilt/tool-node.js +61 -0
- package/dist/prebuilt/tool-node.js.map +1 -0
- package/dist/pregel.d.ts +48 -0
- package/dist/pregel.d.ts.map +1 -0
- package/dist/pregel.js +583 -0
- package/dist/pregel.js.map +1 -0
- package/dist/retry.d.ts +3 -0
- package/dist/retry.d.ts.map +1 -0
- package/dist/retry.js +40 -0
- package/dist/retry.js.map +1 -0
- package/dist/store/index.d.ts +90 -0
- package/dist/store/index.d.ts.map +1 -0
- package/dist/store/index.js +214 -0
- package/dist/store/index.js.map +1 -0
- package/dist/stream-events.d.ts +15 -0
- package/dist/stream-events.d.ts.map +1 -0
- package/dist/stream-events.js +53 -0
- package/dist/stream-events.js.map +1 -0
- package/dist/streaming.d.ts +56 -0
- package/dist/streaming.d.ts.map +1 -0
- package/dist/streaming.js +143 -0
- package/dist/streaming.js.map +1 -0
- package/dist/swarm/graph.d.ts +184 -0
- package/dist/swarm/graph.d.ts.map +1 -0
- package/dist/swarm/graph.js +534 -0
- package/dist/swarm/graph.js.map +1 -0
- package/dist/swarm/index.d.ts +11 -0
- package/dist/swarm/index.d.ts.map +1 -0
- package/dist/swarm/index.js +10 -0
- package/dist/swarm/index.js.map +1 -0
- package/dist/swarm/mailbox.d.ts +12 -0
- package/dist/swarm/mailbox.d.ts.map +1 -0
- package/dist/swarm/mailbox.js +36 -0
- package/dist/swarm/mailbox.js.map +1 -0
- package/dist/swarm/pool.d.ts +27 -0
- package/dist/swarm/pool.d.ts.map +1 -0
- package/dist/swarm/pool.js +87 -0
- package/dist/swarm/pool.js.map +1 -0
- package/dist/swarm/registry.d.ts +37 -0
- package/dist/swarm/registry.d.ts.map +1 -0
- package/dist/swarm/registry.js +115 -0
- package/dist/swarm/registry.js.map +1 -0
- package/dist/swarm/supervisor.d.ts +18 -0
- package/dist/swarm/supervisor.d.ts.map +1 -0
- package/dist/swarm/supervisor.js +117 -0
- package/dist/swarm/supervisor.js.map +1 -0
- package/dist/swarm/types.d.ts +111 -0
- package/dist/swarm/types.d.ts.map +1 -0
- package/dist/swarm/types.js +16 -0
- package/dist/swarm/types.js.map +1 -0
- package/dist/tools/define.d.ts +14 -0
- package/dist/tools/define.d.ts.map +1 -0
- package/dist/tools/define.js +39 -0
- package/dist/tools/define.js.map +1 -0
- package/dist/tools/index.d.ts +3 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +2 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/types.d.ts +23 -0
- package/dist/tools/types.d.ts.map +1 -0
- package/dist/tools/types.js +2 -0
- package/dist/tools/types.js.map +1 -0
- package/dist/types.d.ts +204 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +45 -0
- package/dist/types.js.map +1 -0
- package/package.json +108 -0
|
@@ -0,0 +1,115 @@
|
|
|
1
|
+
// ============================================================
|
|
2
|
+
// @oni.bot/core — agent() — Functional Agent Factory
|
|
3
|
+
// ============================================================
|
|
4
|
+
// Creates an AgentNode from a user-supplied handler function.
|
|
5
|
+
//
|
|
6
|
+
// Usage:
|
|
7
|
+
// agent("myAgent", async (ctx) => { ... })
|
|
8
|
+
// agent("myAgent", { model, tools }, async (ctx) => { ... })
|
|
9
|
+
//
|
|
10
|
+
// The handler receives an AgentContext with chat(), stream(),
|
|
11
|
+
// executeTools(), coordination methods, and runtime context.
|
|
12
|
+
// ============================================================
|
|
13
|
+
import { buildAgentContext } from "./context.js";
|
|
14
|
+
// Try to import runtime context getters — these work inside Pregel
|
|
15
|
+
import { _getRunContext, } from "../context.js";
|
|
16
|
+
/**
|
|
17
|
+
* agent() — create a functional agent node
|
|
18
|
+
*
|
|
19
|
+
* Overloads:
|
|
20
|
+
* agent(name, handler)
|
|
21
|
+
* agent(name, opts, handler)
|
|
22
|
+
*/
|
|
23
|
+
export function agent(name, fnOrOpts, fn) {
|
|
24
|
+
let handler;
|
|
25
|
+
let opts = {};
|
|
26
|
+
if (typeof fnOrOpts === "function") {
|
|
27
|
+
handler = fnOrOpts;
|
|
28
|
+
}
|
|
29
|
+
else {
|
|
30
|
+
opts = fnOrOpts;
|
|
31
|
+
if (!fn) {
|
|
32
|
+
throw new Error(`agent("${name}", opts, fn): handler function is required when passing options.`);
|
|
33
|
+
}
|
|
34
|
+
handler = fn;
|
|
35
|
+
}
|
|
36
|
+
const { model, tools = [], systemPrompt, maxSteps = 10, maxTokens, } = opts;
|
|
37
|
+
const _nodeFn = async (state, config) => {
|
|
38
|
+
// Accumulator for send() calls — merged into state update as swarmMessages
|
|
39
|
+
const pendingMessages = [];
|
|
40
|
+
// Try to get runtime context from AsyncLocalStorage (available inside Pregel)
|
|
41
|
+
const runCtx = _getRunContext();
|
|
42
|
+
const effectiveConfig = config ?? runCtx?.config ?? {};
|
|
43
|
+
const effectiveStore = runCtx?.store ?? null;
|
|
44
|
+
const effectiveWriter = runCtx?.writer ?? null;
|
|
45
|
+
const effectiveRemaining = runCtx
|
|
46
|
+
? runCtx.recursionLimit - runCtx.step
|
|
47
|
+
: maxSteps;
|
|
48
|
+
const ctx = buildAgentContext({
|
|
49
|
+
model,
|
|
50
|
+
tools,
|
|
51
|
+
agentName: name,
|
|
52
|
+
systemPrompt,
|
|
53
|
+
config: effectiveConfig,
|
|
54
|
+
store: effectiveStore,
|
|
55
|
+
state,
|
|
56
|
+
streamWriter: effectiveWriter,
|
|
57
|
+
remainingSteps: effectiveRemaining,
|
|
58
|
+
onSend: (agent, payload) => {
|
|
59
|
+
pendingMessages.push({
|
|
60
|
+
to: agent,
|
|
61
|
+
content: payload,
|
|
62
|
+
timestamp: Date.now(),
|
|
63
|
+
});
|
|
64
|
+
},
|
|
65
|
+
getInbox: () => {
|
|
66
|
+
// Read swarmMessages from state if available
|
|
67
|
+
const msgs = state.swarmMessages;
|
|
68
|
+
if (!Array.isArray(msgs))
|
|
69
|
+
return [];
|
|
70
|
+
return msgs.filter((m) => m !== null && typeof m === "object");
|
|
71
|
+
},
|
|
72
|
+
onRequest: async (_agent, _payload) => {
|
|
73
|
+
// Request/response coordination will be fully wired in Task 12
|
|
74
|
+
throw new Error("request() is not yet wired — see Task 12 (enhanced coordination).");
|
|
75
|
+
},
|
|
76
|
+
onReply: (msg, payload) => {
|
|
77
|
+
if (msg._replyChannel) {
|
|
78
|
+
msg._replyChannel(payload);
|
|
79
|
+
}
|
|
80
|
+
},
|
|
81
|
+
onPublish: (_topic, _data) => {
|
|
82
|
+
// Pub/sub coordination will be fully wired in Task 12
|
|
83
|
+
},
|
|
84
|
+
});
|
|
85
|
+
// Execute the user handler
|
|
86
|
+
const result = await handler(ctx);
|
|
87
|
+
// Merge pending swarm messages into the result
|
|
88
|
+
if (pendingMessages.length > 0) {
|
|
89
|
+
const existing = Array.isArray(result.swarmMessages)
|
|
90
|
+
? result.swarmMessages
|
|
91
|
+
: [];
|
|
92
|
+
result.swarmMessages = [
|
|
93
|
+
...existing,
|
|
94
|
+
...pendingMessages.map((m) => ({
|
|
95
|
+
id: `${name}-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`,
|
|
96
|
+
from: name,
|
|
97
|
+
content: m.content,
|
|
98
|
+
timestamp: m.timestamp,
|
|
99
|
+
})),
|
|
100
|
+
];
|
|
101
|
+
}
|
|
102
|
+
return result;
|
|
103
|
+
};
|
|
104
|
+
return {
|
|
105
|
+
name,
|
|
106
|
+
model,
|
|
107
|
+
tools,
|
|
108
|
+
systemPrompt,
|
|
109
|
+
maxSteps,
|
|
110
|
+
maxTokens,
|
|
111
|
+
_nodeFn,
|
|
112
|
+
_isAgent: true,
|
|
113
|
+
};
|
|
114
|
+
}
|
|
115
|
+
//# sourceMappingURL=functional-agent.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"functional-agent.js","sourceRoot":"","sources":["../../src/agents/functional-agent.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,qDAAqD;AACrD,+DAA+D;AAC/D,8DAA8D;AAC9D,EAAE;AACF,SAAS;AACT,6CAA6C;AAC7C,+DAA+D;AAC/D,EAAE;AACF,8DAA8D;AAC9D,6DAA6D;AAC7D,+DAA+D;AAI/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAEjD,mEAAmE;AACnE,OAAO,EACL,cAAc,GACf,MAAM,eAAe,CAAC;AAIvB;;;;;;GAMG;AACH,MAAM,UAAU,KAAK,CACnB,IAAY,EACZ,QAAkD,EAClD,EAAoB;IAEpB,IAAI,OAAwB,CAAC;IAC7B,IAAI,IAAI,GAA2B,EAAE,CAAC;IAEtC,IAAI,OAAO,QAAQ,KAAK,UAAU,EAAE,CAAC;QACnC,OAAO,GAAG,QAAQ,CAAC;IACrB,CAAC;SAAM,CAAC;QACN,IAAI,GAAG,QAAQ,CAAC;QAChB,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,MAAM,IAAI,KAAK,CACb,UAAU,IAAI,kEAAkE,CACjF,CAAC;QACJ,CAAC;QACD,OAAO,GAAG,EAAE,CAAC;IACf,CAAC;IAED,MAAM,EACJ,KAAK,EACL,KAAK,GAAG,EAAE,EACV,YAAY,EACZ,QAAQ,GAAG,EAAE,EACb,SAAS,GACV,GAAG,IAAI,CAAC;IAET,MAAM,OAAO,GAAG,KAAK,EAAE,KAAQ,EAAE,MAAkB,EAAuB,EAAE;QAC1E,2EAA2E;QAC3E,MAAM,eAAe,GAA+D,EAAE,CAAC;QAEvF,8EAA8E;QAC9E,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;QAEhC,MAAM,eAAe,GAAG,MAAM,IAAI,MAAM,EAAE,MAAM,IAAI,EAAE,CAAC;QACvD,MAAM,cAAc,GAAG,MAAM,EAAE,KAAK,IAAI,IAAI,CAAC;QAC7C,MAAM,eAAe,GAAG,MAAM,EAAE,MAAM,IAAI,IAAI,CAAC;QAC/C,MAAM,kBAAkB,GAAG,MAAM;YAC/B,CAAC,CAAC,MAAM,CAAC,cAAc,GAAG,MAAM,CAAC,IAAI;YACrC,CAAC,CAAC,QAAQ,CAAC;QAEb,MAAM,GAAG,GAAG,iBAAiB,CAAI;YAC/B,KAAK;YACL,KAAK;YACL,SAAS,EAAE,IAAI;YACf,YAAY;YACZ,MAAM,EAAE,eAAe;YACvB,KAAK,EAAE,cAAc;YACrB,KAAK;YACL,YAAY,EAAE,eAAe;YAC7B,cAAc,EAAE,kBAAkB;YAElC,MAAM,EAAE,CAAC,KAAa,EAAE,OAAgB,EAAE,EAAE;gBAC1C,eAAe,CAAC,IAAI,CAAC;oBACnB,EAAE,EAAE,KAAK;oBACT,OAAO,EAAE,OAAO;oBAChB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iBACtB,CAAC,CAAC;YACL,CAAC;YAED,QAAQ,EAAE,GAAG,EAAE;gBACb,6CAA6C;gBAC7C,MAAM,IAAI,GAAI,KAAiC,CAAC,aAAa,CAAC;gBAC9D,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;oBAAE,OAAO,EAAE,CAAC;gBACpC,OAAO,IAAI,CAAC,MAAM,CAChB,CAAC,CAAmB,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,IAAI,OAAO,CAAC,KAAK,QAAQ,CACvC,CAAC;YAC1B,CAAC;YAED,SAAS,EAAE,KAAK,EAAE,MAAc,EAAE,QAAiB,EAAE,EAAE;gBACrD,+DAA+D;gBAC/D,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;YACvF,CAAC;YAED,OAAO,EAAE,CAAC,GAAqB,EAAE,OAAgB,EAAE,EAAE;gBACnD,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;oBACtB,GAAG,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;gBAC7B,CAAC;YACH,CAAC;YAED,SAAS,EAAE,CAAC,MAAc,EAAE,KAAc,EAAE,EAAE;gBAC5C,sDAAsD;YACxD,CAAC;SACF,CAAC,CAAC;QAEH,2BAA2B;QAC3B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC;QAElC,+CAA+C;QAC/C,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAE,MAAkC,CAAC,aAAa,CAAC;gBAC/E,CAAC,CAAE,MAAkC,CAAC,aAA0B;gBAChE,CAAC,CAAC,EAAE,CAAC;YACN,MAAkC,CAAC,aAAa,GAAG;gBAClD,GAAG,QAAQ;gBACX,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC7B,EAAE,EAAE,GAAG,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;oBACrE,IAAI,EAAE,IAAI;oBACV,OAAO,EAAE,CAAC,CAAC,OAAO;oBAClB,SAAS,EAAE,CAAC,CAAC,SAAS;iBACvB,CAAC,CAAC;aACJ,CAAC;QACJ,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;IAEF,OAAO;QACL,IAAI;QACJ,KAAK;QACL,KAAK;QACL,YAAY;QACZ,QAAQ;QACR,SAAS;QACT,OAAO;QACP,QAAQ,EAAE,IAAI;KACf,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
export type { AgentContext, AgentNode, DefineAgentOptions, FunctionalAgentOptions, SwarmMessageView, } from "./types.js";
|
|
2
|
+
export { buildAgentContext } from "./context.js";
|
|
3
|
+
export type { BuildAgentContextOptions } from "./context.js";
|
|
4
|
+
export { defineAgent } from "./define-agent.js";
|
|
5
|
+
export { agent } from "./functional-agent.js";
|
|
6
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/agents/index.ts"],"names":[],"mappings":"AAIA,YAAY,EACV,YAAY,EACZ,SAAS,EACT,kBAAkB,EAClB,sBAAsB,EACtB,gBAAgB,GACjB,MAAM,YAAY,CAAC;AAEpB,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACjD,YAAY,EAAE,wBAAwB,EAAE,MAAM,cAAc,CAAC;AAE7D,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
// ============================================================
|
|
2
|
+
// @oni.bot/core — Agents barrel exports
|
|
3
|
+
// ============================================================
|
|
4
|
+
export { buildAgentContext } from "./context.js";
|
|
5
|
+
export { defineAgent } from "./define-agent.js";
|
|
6
|
+
export { agent } from "./functional-agent.js";
|
|
7
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/agents/index.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,wCAAwC;AACxC,+DAA+D;AAU/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAGjD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC"}
|
|
@@ -0,0 +1,62 @@
|
|
|
1
|
+
import type { ONIModel, ONIModelMessage, ChatResponse, ChatChunk, ToolResult } from "../models/types.js";
|
|
2
|
+
import type { ToolDefinition } from "../tools/types.js";
|
|
3
|
+
import type { ONIConfig } from "../types.js";
|
|
4
|
+
import type { BaseStore } from "../store/index.js";
|
|
5
|
+
import type { StreamWriter } from "../context.js";
|
|
6
|
+
export interface SwarmMessageView {
|
|
7
|
+
id: string;
|
|
8
|
+
from: string;
|
|
9
|
+
content: unknown;
|
|
10
|
+
timestamp: number;
|
|
11
|
+
replyTo?: string;
|
|
12
|
+
/** Internal: channel used by reply() — not part of public API contract */
|
|
13
|
+
_replyChannel?: (payload: unknown) => void;
|
|
14
|
+
}
|
|
15
|
+
export interface AgentContext<S = Record<string, unknown>> {
|
|
16
|
+
chat(messages: ONIModelMessage[]): Promise<ChatResponse>;
|
|
17
|
+
stream(messages: ONIModelMessage[]): AsyncGenerator<ChatChunk>;
|
|
18
|
+
executeTools(calls: Array<{
|
|
19
|
+
id: string;
|
|
20
|
+
name: string;
|
|
21
|
+
args: Record<string, unknown>;
|
|
22
|
+
}>): Promise<ToolResult[]>;
|
|
23
|
+
send(agent: string, payload: unknown): void;
|
|
24
|
+
inbox(): SwarmMessageView[];
|
|
25
|
+
request(agent: string, payload: unknown): Promise<unknown>;
|
|
26
|
+
reply(msg: SwarmMessageView, payload: unknown): void;
|
|
27
|
+
publish(topic: string, data: unknown): void;
|
|
28
|
+
config: ONIConfig;
|
|
29
|
+
store: BaseStore | null;
|
|
30
|
+
state: S;
|
|
31
|
+
streamWriter: StreamWriter | null;
|
|
32
|
+
remainingSteps: number;
|
|
33
|
+
agentName: string;
|
|
34
|
+
}
|
|
35
|
+
export interface DefineAgentOptions {
|
|
36
|
+
name: string;
|
|
37
|
+
description?: string;
|
|
38
|
+
model: ONIModel;
|
|
39
|
+
tools?: ToolDefinition[];
|
|
40
|
+
systemPrompt?: string;
|
|
41
|
+
maxSteps?: number;
|
|
42
|
+
maxTokens?: number;
|
|
43
|
+
}
|
|
44
|
+
export interface FunctionalAgentOptions {
|
|
45
|
+
model?: ONIModel;
|
|
46
|
+
tools?: ToolDefinition[];
|
|
47
|
+
systemPrompt?: string;
|
|
48
|
+
maxSteps?: number;
|
|
49
|
+
maxTokens?: number;
|
|
50
|
+
}
|
|
51
|
+
export interface AgentNode<S = Record<string, unknown>> {
|
|
52
|
+
name: string;
|
|
53
|
+
description?: string;
|
|
54
|
+
model?: ONIModel;
|
|
55
|
+
tools: ToolDefinition[];
|
|
56
|
+
systemPrompt?: string;
|
|
57
|
+
maxSteps: number;
|
|
58
|
+
maxTokens?: number;
|
|
59
|
+
_nodeFn: (state: S, config?: ONIConfig) => Promise<Partial<S>>;
|
|
60
|
+
_isAgent: true;
|
|
61
|
+
}
|
|
62
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/agents/types.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EACV,QAAQ,EACR,eAAe,EACf,YAAY,EACZ,SAAS,EACT,UAAU,EACX,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAMlD,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,0EAA0E;IAC1E,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;CAC5C;AAMD,MAAM,WAAW,YAAY,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAEvD,IAAI,CAAC,QAAQ,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IACzD,MAAM,CAAC,QAAQ,EAAE,eAAe,EAAE,GAAG,cAAc,CAAC,SAAS,CAAC,CAAC;IAC/D,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;IAG/G,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI,CAAC;IAC5C,KAAK,IAAI,gBAAgB,EAAE,CAAC;IAC5B,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3D,KAAK,CAAC,GAAG,EAAE,gBAAgB,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI,CAAC;IACrD,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,IAAI,CAAC;IAG5C,MAAM,EAAE,SAAS,CAAC;IAClB,KAAK,EAAE,SAAS,GAAG,IAAI,CAAC;IACxB,KAAK,EAAE,CAAC,CAAC;IACT,YAAY,EAAE,YAAY,GAAG,IAAI,CAAC;IAClC,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;CACnB;AAMD,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,QAAQ,CAAC;IAChB,KAAK,CAAC,EAAE,cAAc,EAAE,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAMD,MAAM,WAAW,sBAAsB;IACrC,KAAK,CAAC,EAAE,QAAQ,CAAC;IACjB,KAAK,CAAC,EAAE,cAAc,EAAE,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAMD,MAAM,WAAW,SAAS,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IACpD,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,QAAQ,CAAC;IACjB,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,MAAM,CAAC,EAAE,SAAS,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/D,QAAQ,EAAE,IAAI,CAAC;CAChB"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
// ============================================================
|
|
2
|
+
// @oni.bot/core — Agent Types
|
|
3
|
+
// ============================================================
|
|
4
|
+
// Defines the type contracts for both declarative (defineAgent)
|
|
5
|
+
// and functional (agent) agent factories.
|
|
6
|
+
// ============================================================
|
|
7
|
+
export {};
|
|
8
|
+
//# sourceMappingURL=types.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/agents/types.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,8BAA8B;AAC9B,+DAA+D;AAC/D,gEAAgE;AAChE,0CAA0C;AAC1C,+DAA+D"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
import type { ONICheckpoint, ONICheckpointer, CheckpointListOptions } from "./types.js";
|
|
2
|
+
export declare class MemoryCheckpointer<S> implements ONICheckpointer<S> {
|
|
3
|
+
private store;
|
|
4
|
+
get(threadId: string): Promise<ONICheckpoint<S> | null>;
|
|
5
|
+
put(checkpoint: ONICheckpoint<S>): Promise<void>;
|
|
6
|
+
list(threadId: string, opts?: CheckpointListOptions): Promise<ONICheckpoint<S>[]>;
|
|
7
|
+
delete(threadId: string): Promise<void>;
|
|
8
|
+
getAt(threadId: string, step: number): Promise<ONICheckpoint<S> | null>;
|
|
9
|
+
/** Clone a thread's history up to `step` under a new threadId */
|
|
10
|
+
fork(sourceThreadId: string, step: number, newThreadId: string): Promise<boolean>;
|
|
11
|
+
clear(): void;
|
|
12
|
+
}
|
|
13
|
+
export declare class NoopCheckpointer<S> implements ONICheckpointer<S> {
|
|
14
|
+
get(_: string): Promise<null>;
|
|
15
|
+
put(_: ONICheckpoint<S>): Promise<void>;
|
|
16
|
+
list(_: string): Promise<ONICheckpoint<S>[]>;
|
|
17
|
+
delete(_: string): Promise<void>;
|
|
18
|
+
}
|
|
19
|
+
export declare abstract class PersistentCheckpointer<S> implements ONICheckpointer<S> {
|
|
20
|
+
abstract get(threadId: string): Promise<ONICheckpoint<S> | null>;
|
|
21
|
+
abstract put(checkpoint: ONICheckpoint<S>): Promise<void>;
|
|
22
|
+
abstract list(threadId: string, opts?: CheckpointListOptions): Promise<ONICheckpoint<S>[]>;
|
|
23
|
+
abstract delete(threadId: string): Promise<void>;
|
|
24
|
+
getAt(threadId: string, step: number): Promise<ONICheckpoint<S> | null>;
|
|
25
|
+
fork(source: string, step: number, dest: string): Promise<boolean>;
|
|
26
|
+
}
|
|
27
|
+
//# sourceMappingURL=checkpoint.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"checkpoint.d.ts","sourceRoot":"","sources":["../src/checkpoint.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,aAAa,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AAMxF,qBAAa,kBAAkB,CAAC,CAAC,CAAE,YAAW,eAAe,CAAC,CAAC,CAAC;IAC9D,OAAO,CAAC,KAAK,CAAyC;IAEhD,GAAG,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;IAMvD,GAAG,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAShD,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,qBAAqB,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;IAiBjF,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMvC,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;IAK7E,iEAAiE;IAC3D,IAAI,CACR,cAAc,EAAE,MAAM,EACtB,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,MAAM,GAClB,OAAO,CAAC,OAAO,CAAC;IAWnB,KAAK,IAAI,IAAI;CACd;AAMD,qBAAa,gBAAgB,CAAC,CAAC,CAAE,YAAW,eAAe,CAAC,CAAC,CAAC;IACtD,GAAG,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAC7B,GAAG,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IACvC,IAAI,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;IAC5C,MAAM,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CACvC;AAOD,8BAAsB,sBAAsB,CAAC,CAAC,CAAE,YAAW,eAAe,CAAC,CAAC,CAAC;IAC3E,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,GAAyC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;IACtG,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC,CAAC,GAA6B,OAAO,CAAC,IAAI,CAAC;IACnF,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,qBAAqB,GAAU,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;IACjG,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAsC,OAAO,CAAC,IAAI,CAAC;IAE7E,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;IAKvE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;CASzE"}
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
// ============================================================
|
|
2
|
+
// @oni.bot/core — Checkpointers v2
|
|
3
|
+
// Adds: time-travel, fork, SQLite-ready interface stubs
|
|
4
|
+
// ============================================================
|
|
5
|
+
// ----------------------------------------------------------------
|
|
6
|
+
// MemoryCheckpointer — dev/test, full history retained
|
|
7
|
+
// ----------------------------------------------------------------
|
|
8
|
+
export class MemoryCheckpointer {
|
|
9
|
+
store = new Map();
|
|
10
|
+
async get(threadId) {
|
|
11
|
+
const history = this.store.get(threadId);
|
|
12
|
+
if (!history?.length)
|
|
13
|
+
return null;
|
|
14
|
+
return history[history.length - 1] ?? null;
|
|
15
|
+
}
|
|
16
|
+
async put(checkpoint) {
|
|
17
|
+
const existing = this.store.get(checkpoint.threadId) ?? [];
|
|
18
|
+
// Overwrite if same step exists (idempotent)
|
|
19
|
+
const idx = existing.findIndex((c) => c.step === checkpoint.step);
|
|
20
|
+
if (idx >= 0)
|
|
21
|
+
existing[idx] = { ...checkpoint };
|
|
22
|
+
else
|
|
23
|
+
existing.push({ ...checkpoint });
|
|
24
|
+
this.store.set(checkpoint.threadId, existing);
|
|
25
|
+
}
|
|
26
|
+
async list(threadId, opts) {
|
|
27
|
+
let items = [...(this.store.get(threadId) ?? [])];
|
|
28
|
+
if (opts?.before !== undefined) {
|
|
29
|
+
items = items.filter(c => c.step < opts.before);
|
|
30
|
+
}
|
|
31
|
+
if (opts?.filter) {
|
|
32
|
+
items = items.filter(c => {
|
|
33
|
+
if (!c.metadata)
|
|
34
|
+
return false;
|
|
35
|
+
return Object.entries(opts.filter).every(([k, v]) => c.metadata[k] === v);
|
|
36
|
+
});
|
|
37
|
+
}
|
|
38
|
+
if (opts?.limit !== undefined) {
|
|
39
|
+
items = items.slice(0, opts.limit);
|
|
40
|
+
}
|
|
41
|
+
return items;
|
|
42
|
+
}
|
|
43
|
+
async delete(threadId) {
|
|
44
|
+
this.store.delete(threadId);
|
|
45
|
+
}
|
|
46
|
+
// ---- Time-travel extensions ----
|
|
47
|
+
async getAt(threadId, step) {
|
|
48
|
+
const history = this.store.get(threadId) ?? [];
|
|
49
|
+
return history.find((c) => c.step === step) ?? null;
|
|
50
|
+
}
|
|
51
|
+
/** Clone a thread's history up to `step` under a new threadId */
|
|
52
|
+
async fork(sourceThreadId, step, newThreadId) {
|
|
53
|
+
const history = this.store.get(sourceThreadId) ?? [];
|
|
54
|
+
const upTo = history.filter((c) => c.step <= step);
|
|
55
|
+
if (!upTo.length)
|
|
56
|
+
return false;
|
|
57
|
+
this.store.set(newThreadId, upTo.map((c) => ({ ...c, threadId: newThreadId })));
|
|
58
|
+
return true;
|
|
59
|
+
}
|
|
60
|
+
clear() { this.store.clear(); }
|
|
61
|
+
}
|
|
62
|
+
// ----------------------------------------------------------------
|
|
63
|
+
// NoopCheckpointer — stateless default
|
|
64
|
+
// ----------------------------------------------------------------
|
|
65
|
+
export class NoopCheckpointer {
|
|
66
|
+
async get(_) { return null; }
|
|
67
|
+
async put(_) { }
|
|
68
|
+
async list(_) { return []; }
|
|
69
|
+
async delete(_) { }
|
|
70
|
+
}
|
|
71
|
+
// ----------------------------------------------------------------
|
|
72
|
+
// Base class for persistent checkpointers (SQLite, Redis, etc.)
|
|
73
|
+
// Subclass this and implement the abstract methods.
|
|
74
|
+
// ----------------------------------------------------------------
|
|
75
|
+
export class PersistentCheckpointer {
|
|
76
|
+
async getAt(threadId, step) {
|
|
77
|
+
const all = await this.list(threadId);
|
|
78
|
+
return all.find((c) => c.step === step) ?? null;
|
|
79
|
+
}
|
|
80
|
+
async fork(source, step, dest) {
|
|
81
|
+
const all = await this.list(source);
|
|
82
|
+
const upTo = all.filter((c) => c.step <= step);
|
|
83
|
+
if (!upTo.length)
|
|
84
|
+
return false;
|
|
85
|
+
for (const c of upTo) {
|
|
86
|
+
await this.put({ ...c, threadId: dest });
|
|
87
|
+
}
|
|
88
|
+
return true;
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
//# sourceMappingURL=checkpoint.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"checkpoint.js","sourceRoot":"","sources":["../src/checkpoint.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,mCAAmC;AACnC,wDAAwD;AACxD,+DAA+D;AAI/D,mEAAmE;AACnE,uDAAuD;AACvD,mEAAmE;AAEnE,MAAM,OAAO,kBAAkB;IACrB,KAAK,GAAG,IAAI,GAAG,EAA8B,CAAC;IAEtD,KAAK,CAAC,GAAG,CAAC,QAAgB;QACxB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACzC,IAAI,CAAC,OAAO,EAAE,MAAM;YAAE,OAAO,IAAI,CAAC;QAClC,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,UAA4B;QACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC3D,6CAA6C;QAC7C,MAAM,GAAG,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,CAAC,CAAC;QAClE,IAAI,GAAG,IAAI,CAAC;YAAE,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,UAAU,EAAE,CAAC;;YAC3C,QAAQ,CAAC,IAAI,CAAC,EAAE,GAAG,UAAU,EAAE,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE,IAA4B;QACvD,IAAI,KAAK,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAClD,IAAI,IAAI,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;YAC/B,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,MAAO,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,IAAI,EAAE,MAAM,EAAE,CAAC;YACjB,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE;gBACvB,IAAI,CAAC,CAAC,CAAC,QAAQ;oBAAE,OAAO,KAAK,CAAC;gBAC9B,OAAO,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;YAC9E,CAAC,CAAC,CAAC;QACL,CAAC;QACD,IAAI,IAAI,EAAE,KAAK,KAAK,SAAS,EAAE,CAAC;YAC9B,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACrC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAgB;QAC3B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC9B,CAAC;IAED,mCAAmC;IAEnC,KAAK,CAAC,KAAK,CAAC,QAAgB,EAAE,IAAY;QACxC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC/C,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,IAAI,CAAC;IACtD,CAAC;IAED,iEAAiE;IACjE,KAAK,CAAC,IAAI,CACR,cAAsB,EACtB,IAAY,EACZ,WAAmB;QAEnB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;QACrD,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC;QACnD,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,KAAK,CAAC;QAC/B,IAAI,CAAC,KAAK,CAAC,GAAG,CACZ,WAAW,EACX,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC,CACnD,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,KAAW,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;CACtC;AAED,mEAAmE;AACnE,uCAAuC;AACvC,mEAAmE;AAEnE,MAAM,OAAO,gBAAgB;IAC3B,KAAK,CAAC,GAAG,CAAC,CAAS,IAAmB,OAAO,IAAI,CAAC,CAAC,CAAC;IACpD,KAAK,CAAC,GAAG,CAAC,CAAmB,IAAkB,CAAC;IAChD,KAAK,CAAC,IAAI,CAAC,CAAS,IAAiC,OAAO,EAAE,CAAC,CAAC,CAAC;IACjE,KAAK,CAAC,MAAM,CAAC,CAAS,IAAkB,CAAC;CAC1C;AAED,mEAAmE;AACnE,gEAAgE;AAChE,oDAAoD;AACpD,mEAAmE;AAEnE,MAAM,OAAgB,sBAAsB;IAM1C,KAAK,CAAC,KAAK,CAAC,QAAgB,EAAE,IAAY;QACxC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,OAAO,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,IAAI,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAc,EAAE,IAAY,EAAE,IAAY;QACnD,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,KAAK,CAAC;QAC/B,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/checkpointers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AACzD,OAAO,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
export { SqliteCheckpointer } from "./sqlite.js";
|
|
2
|
+
export { NamespacedCheckpointer } from "./namespaced.js";
|
|
3
|
+
export { PostgresCheckpointer } from "./postgres.js";
|
|
4
|
+
// Redis coming next: export { RedisCheckpointer } from "./redis.js";
|
|
5
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/checkpointers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AACzD,OAAO,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AACrD,qEAAqE"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { ONICheckpoint, ONICheckpointer, CheckpointListOptions } from "../types.js";
|
|
2
|
+
export declare class NamespacedCheckpointer<S> implements ONICheckpointer<S> {
|
|
3
|
+
private readonly inner;
|
|
4
|
+
private readonly ns;
|
|
5
|
+
constructor(inner: ONICheckpointer<S>, ns: string);
|
|
6
|
+
private prefix;
|
|
7
|
+
get(threadId: string): Promise<ONICheckpoint<S> | null>;
|
|
8
|
+
put(checkpoint: ONICheckpoint<S>): Promise<void>;
|
|
9
|
+
list(threadId: string, opts?: CheckpointListOptions): Promise<ONICheckpoint<S>[]>;
|
|
10
|
+
delete(threadId: string): Promise<void>;
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=namespaced.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"namespaced.d.ts","sourceRoot":"","sources":["../../src/checkpointers/namespaced.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EAAE,aAAa,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAEzF,qBAAa,sBAAsB,CAAC,CAAC,CAAE,YAAW,eAAe,CAAC,CAAC,CAAC;IAEhE,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,EAAE;gBADF,KAAK,EAAE,eAAe,CAAC,CAAC,CAAC,EACzB,EAAE,EAAE,MAAM;IAG7B,OAAO,CAAC,MAAM;IAIR,GAAG,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;IAMvD,GAAG,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAOhD,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,qBAAqB,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;IAKjF,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAG9C"}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
// ============================================================
|
|
2
|
+
// @oni.bot/core — Namespaced Checkpointer Wrapper
|
|
3
|
+
// ============================================================
|
|
4
|
+
// Prefixes threadId with a namespace for checkpoint isolation.
|
|
5
|
+
// Used by subgraphs to avoid checkpoint collisions with parent.
|
|
6
|
+
// ============================================================
|
|
7
|
+
export class NamespacedCheckpointer {
|
|
8
|
+
inner;
|
|
9
|
+
ns;
|
|
10
|
+
constructor(inner, ns) {
|
|
11
|
+
this.inner = inner;
|
|
12
|
+
this.ns = ns;
|
|
13
|
+
}
|
|
14
|
+
prefix(threadId) {
|
|
15
|
+
return `${threadId}:${this.ns}`;
|
|
16
|
+
}
|
|
17
|
+
async get(threadId) {
|
|
18
|
+
const cp = await this.inner.get(this.prefix(threadId));
|
|
19
|
+
if (!cp)
|
|
20
|
+
return null;
|
|
21
|
+
return { ...cp, threadId };
|
|
22
|
+
}
|
|
23
|
+
async put(checkpoint) {
|
|
24
|
+
await this.inner.put({
|
|
25
|
+
...checkpoint,
|
|
26
|
+
threadId: this.prefix(checkpoint.threadId),
|
|
27
|
+
});
|
|
28
|
+
}
|
|
29
|
+
async list(threadId, opts) {
|
|
30
|
+
const items = await this.inner.list(this.prefix(threadId), opts);
|
|
31
|
+
return items.map(c => ({ ...c, threadId }));
|
|
32
|
+
}
|
|
33
|
+
async delete(threadId) {
|
|
34
|
+
await this.inner.delete(this.prefix(threadId));
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
//# sourceMappingURL=namespaced.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"namespaced.js","sourceRoot":"","sources":["../../src/checkpointers/namespaced.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,kDAAkD;AAClD,+DAA+D;AAC/D,+DAA+D;AAC/D,gEAAgE;AAChE,+DAA+D;AAI/D,MAAM,OAAO,sBAAsB;IAEd;IACA;IAFnB,YACmB,KAAyB,EACzB,EAAU;QADV,UAAK,GAAL,KAAK,CAAoB;QACzB,OAAE,GAAF,EAAE,CAAQ;IAC1B,CAAC;IAEI,MAAM,CAAC,QAAgB;QAC7B,OAAO,GAAG,QAAQ,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,QAAgB;QACxB,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;QACvD,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC;QACrB,OAAO,EAAE,GAAG,EAAE,EAAE,QAAQ,EAAE,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,UAA4B;QACpC,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC;YACnB,GAAG,UAAU;YACb,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC;SAC3C,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE,IAA4B;QACvD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,CAAC;QACjE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAgB;QAC3B,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;IACjD,CAAC;CACF"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { ONICheckpoint, ONICheckpointer, CheckpointListOptions } from "../types.js";
|
|
2
|
+
export declare class PostgresCheckpointer<S> implements ONICheckpointer<S> {
|
|
3
|
+
private readonly pool;
|
|
4
|
+
private constructor();
|
|
5
|
+
static create<S>(connectionString: string): Promise<PostgresCheckpointer<S>>;
|
|
6
|
+
get(threadId: string): Promise<ONICheckpoint<S> | null>;
|
|
7
|
+
put(cp: ONICheckpoint<S>): Promise<void>;
|
|
8
|
+
list(threadId: string, opts?: CheckpointListOptions): Promise<ONICheckpoint<S>[]>;
|
|
9
|
+
delete(threadId: string): Promise<void>;
|
|
10
|
+
close(): Promise<void>;
|
|
11
|
+
private deserialize;
|
|
12
|
+
}
|
|
13
|
+
//# sourceMappingURL=postgres.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../../src/checkpointers/postgres.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EAAE,aAAa,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAOzF,qBAAa,oBAAoB,CAAC,CAAC,CAAE,YAAW,eAAe,CAAC,CAAC,CAAC;IAC5C,OAAO,CAAC,QAAQ,CAAC,IAAI;IAAzC,OAAO;WAEM,MAAM,CAAC,CAAC,EAAE,gBAAgB,EAAE,MAAM,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;IA2B5E,GAAG,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;IAQvD,GAAG,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBxC,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,qBAAqB,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;IA8BjF,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAOvC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAI5B,OAAO,CAAC,WAAW;CAgBpB"}
|
|
@@ -0,0 +1,100 @@
|
|
|
1
|
+
// ============================================================
|
|
2
|
+
// @oni.bot/core — PostgreSQL Checkpointer
|
|
3
|
+
// ============================================================
|
|
4
|
+
// Production-grade checkpointer using PostgreSQL.
|
|
5
|
+
// Requires `pg` as an optional peer dependency.
|
|
6
|
+
// ============================================================
|
|
7
|
+
export class PostgresCheckpointer {
|
|
8
|
+
pool;
|
|
9
|
+
constructor(pool) {
|
|
10
|
+
this.pool = pool;
|
|
11
|
+
}
|
|
12
|
+
static async create(connectionString) {
|
|
13
|
+
// @ts-ignore — pg is an optional peer dependency
|
|
14
|
+
const pg = await import("pg");
|
|
15
|
+
const Pool = pg.Pool ?? pg.default?.Pool ?? pg;
|
|
16
|
+
const pool = new Pool({ connectionString });
|
|
17
|
+
await pool.query(`
|
|
18
|
+
CREATE TABLE IF NOT EXISTS oni_checkpoints (
|
|
19
|
+
thread_id TEXT NOT NULL,
|
|
20
|
+
step INTEGER NOT NULL,
|
|
21
|
+
agent_id TEXT,
|
|
22
|
+
state JSONB NOT NULL,
|
|
23
|
+
next_nodes JSONB NOT NULL,
|
|
24
|
+
pending_sends JSONB NOT NULL DEFAULT '[]',
|
|
25
|
+
metadata JSONB,
|
|
26
|
+
pending_writes JSONB,
|
|
27
|
+
timestamp BIGINT NOT NULL,
|
|
28
|
+
PRIMARY KEY (thread_id, step)
|
|
29
|
+
)
|
|
30
|
+
`);
|
|
31
|
+
await pool.query(`CREATE INDEX IF NOT EXISTS idx_oni_cp_thread ON oni_checkpoints (thread_id)`);
|
|
32
|
+
return new PostgresCheckpointer(pool);
|
|
33
|
+
}
|
|
34
|
+
async get(threadId) {
|
|
35
|
+
const { rows } = await this.pool.query("SELECT * FROM oni_checkpoints WHERE thread_id=$1 ORDER BY step DESC LIMIT 1", [threadId]);
|
|
36
|
+
return rows.length > 0 ? this.deserialize(rows[0]) : null;
|
|
37
|
+
}
|
|
38
|
+
async put(cp) {
|
|
39
|
+
await this.pool.query(`INSERT INTO oni_checkpoints
|
|
40
|
+
(thread_id, step, agent_id, state, next_nodes, pending_sends, metadata, pending_writes, timestamp)
|
|
41
|
+
VALUES ($1,$2,$3,$4,$5,$6,$7,$8,$9)
|
|
42
|
+
ON CONFLICT (thread_id, step)
|
|
43
|
+
DO UPDATE SET state=$4, next_nodes=$5, pending_sends=$6, metadata=$7, pending_writes=$8, timestamp=$9`, [
|
|
44
|
+
cp.threadId,
|
|
45
|
+
cp.step,
|
|
46
|
+
cp.agentId ?? null,
|
|
47
|
+
JSON.stringify(cp.state),
|
|
48
|
+
JSON.stringify(cp.nextNodes),
|
|
49
|
+
JSON.stringify(cp.pendingSends ?? []),
|
|
50
|
+
cp.metadata ? JSON.stringify(cp.metadata) : null,
|
|
51
|
+
cp.pendingWrites ? JSON.stringify(cp.pendingWrites) : null,
|
|
52
|
+
cp.timestamp,
|
|
53
|
+
]);
|
|
54
|
+
}
|
|
55
|
+
async list(threadId, opts) {
|
|
56
|
+
let sql = "SELECT * FROM oni_checkpoints WHERE thread_id=$1";
|
|
57
|
+
const params = [threadId];
|
|
58
|
+
let paramIdx = 2;
|
|
59
|
+
if (opts?.before !== undefined) {
|
|
60
|
+
sql += ` AND step < $${paramIdx++}`;
|
|
61
|
+
params.push(opts.before);
|
|
62
|
+
}
|
|
63
|
+
sql += " ORDER BY step ASC";
|
|
64
|
+
if (opts?.limit !== undefined) {
|
|
65
|
+
sql += ` LIMIT $${paramIdx++}`;
|
|
66
|
+
params.push(opts.limit);
|
|
67
|
+
}
|
|
68
|
+
const { rows } = await this.pool.query(sql, params);
|
|
69
|
+
let items = rows.map(r => this.deserialize(r));
|
|
70
|
+
if (opts?.filter) {
|
|
71
|
+
items = items.filter(c => {
|
|
72
|
+
if (!c.metadata)
|
|
73
|
+
return false;
|
|
74
|
+
return Object.entries(opts.filter).every(([k, v]) => c.metadata[k] === v);
|
|
75
|
+
});
|
|
76
|
+
}
|
|
77
|
+
return items;
|
|
78
|
+
}
|
|
79
|
+
async delete(threadId) {
|
|
80
|
+
await this.pool.query("DELETE FROM oni_checkpoints WHERE thread_id=$1", [threadId]);
|
|
81
|
+
}
|
|
82
|
+
async close() {
|
|
83
|
+
await this.pool.end();
|
|
84
|
+
}
|
|
85
|
+
deserialize(row) {
|
|
86
|
+
const parse = (val) => typeof val === "string" ? JSON.parse(val) : val;
|
|
87
|
+
return {
|
|
88
|
+
threadId: row.thread_id,
|
|
89
|
+
step: row.step,
|
|
90
|
+
agentId: row.agent_id ?? undefined,
|
|
91
|
+
state: parse(row.state),
|
|
92
|
+
nextNodes: parse(row.next_nodes),
|
|
93
|
+
pendingSends: parse(row.pending_sends),
|
|
94
|
+
metadata: row.metadata ? parse(row.metadata) : undefined,
|
|
95
|
+
pendingWrites: row.pending_writes ? parse(row.pending_writes) : undefined,
|
|
96
|
+
timestamp: Number(row.timestamp),
|
|
97
|
+
};
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
//# sourceMappingURL=postgres.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"postgres.js","sourceRoot":"","sources":["../../src/checkpointers/postgres.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,0CAA0C;AAC1C,+DAA+D;AAC/D,kDAAkD;AAClD,gDAAgD;AAChD,+DAA+D;AAS/D,MAAM,OAAO,oBAAoB;IACM;IAArC,YAAqC,IAAY;QAAZ,SAAI,GAAJ,IAAI,CAAQ;IAAG,CAAC;IAErD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAI,gBAAwB;QAC7C,iDAAiD;QACjD,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,IAAI,GAAG,EAAE,CAAC,IAAI,IAAK,EAAE,CAAC,OAAe,EAAE,IAAI,IAAI,EAAE,CAAC;QACxD,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,EAAE,gBAAgB,EAAE,CAAW,CAAC;QAEtD,MAAM,IAAI,CAAC,KAAK,CAAC;;;;;;;;;;;;;KAahB,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,KAAK,CACd,6EAA6E,CAC9E,CAAC;QAEF,OAAO,IAAI,oBAAoB,CAAI,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,QAAgB;QACxB,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACpC,6EAA6E,EAC7E,CAAC,QAAQ,CAAC,CACX,CAAC;QACF,OAAO,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,EAAoB;QAC5B,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACnB;;;;6GAIuG,EACvG;YACE,EAAE,CAAC,QAAQ;YACX,EAAE,CAAC,IAAI;YACP,EAAE,CAAC,OAAO,IAAI,IAAI;YAClB,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,KAAK,CAAC;YACxB,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC;YAC5B,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,YAAY,IAAI,EAAE,CAAC;YACrC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI;YAChD,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI;YAC1D,EAAE,CAAC,SAAS;SACb,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE,IAA4B;QACvD,IAAI,GAAG,GAAG,kDAAkD,CAAC;QAC7D,MAAM,MAAM,GAAc,CAAC,QAAQ,CAAC,CAAC;QACrC,IAAI,QAAQ,GAAG,CAAC,CAAC;QAEjB,IAAI,IAAI,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;YAC/B,GAAG,IAAI,gBAAgB,QAAQ,EAAE,EAAE,CAAC;YACpC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;QAED,GAAG,IAAI,oBAAoB,CAAC;QAE5B,IAAI,IAAI,EAAE,KAAK,KAAK,SAAS,EAAE,CAAC;YAC9B,GAAG,IAAI,WAAW,QAAQ,EAAE,EAAE,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;QAED,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACpD,IAAI,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;QAE/C,IAAI,IAAI,EAAE,MAAM,EAAE,CAAC;YACjB,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE;gBACvB,IAAI,CAAC,CAAC,CAAC,QAAQ;oBAAE,OAAO,KAAK,CAAC;gBAC9B,OAAO,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;YAC9E,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAgB;QAC3B,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACnB,gDAAgD,EAChD,CAAC,QAAQ,CAAC,CACX,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;IACxB,CAAC;IAEO,WAAW,CAAC,GAA4B;QAC9C,MAAM,KAAK,GAAG,CAAC,GAAY,EAAE,EAAE,CAC7B,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;QAElD,OAAO;YACL,QAAQ,EAAO,GAAG,CAAC,SAAmB;YACtC,IAAI,EAAW,GAAG,CAAC,IAAc;YACjC,OAAO,EAAS,GAAG,CAAC,QAA0B,IAAI,SAAS;YAC3D,KAAK,EAAU,KAAK,CAAC,GAAG,CAAC,KAAK,CAAM;YACpC,SAAS,EAAM,KAAK,CAAC,GAAG,CAAC,UAAU,CAAa;YAChD,YAAY,EAAG,KAAK,CAAC,GAAG,CAAC,aAAa,CAA2D;YACjG,QAAQ,EAAO,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAA4B,CAAC,CAAC,CAAC,SAAS;YACxF,aAAa,EAAE,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,CAA+D,CAAC,CAAC,CAAC,SAAS;YACvI,SAAS,EAAM,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC;SACrC,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { ONICheckpoint, ONICheckpointer, CheckpointListOptions } from "../types.js";
|
|
2
|
+
export declare class SqliteCheckpointer<S> implements ONICheckpointer<S> {
|
|
3
|
+
private readonly db;
|
|
4
|
+
private constructor();
|
|
5
|
+
static create<S>(dbPath: string): Promise<SqliteCheckpointer<S>>;
|
|
6
|
+
get(threadId: string): Promise<ONICheckpoint<S> | null>;
|
|
7
|
+
put(cp: ONICheckpoint<S>): Promise<void>;
|
|
8
|
+
list(threadId: string, opts?: CheckpointListOptions): Promise<ONICheckpoint<S>[]>;
|
|
9
|
+
delete(threadId: string): Promise<void>;
|
|
10
|
+
getAt(threadId: string, step: number): Promise<ONICheckpoint<S> | null>;
|
|
11
|
+
close(): void;
|
|
12
|
+
private deserialize;
|
|
13
|
+
}
|
|
14
|
+
//# sourceMappingURL=sqlite.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sqlite.d.ts","sourceRoot":"","sources":["../../src/checkpointers/sqlite.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EAAE,aAAa,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAQzF,qBAAa,kBAAkB,CAAC,CAAC,CAAE,YAAW,eAAe,CAAC,CAAC,CAAC;IAC1C,OAAO,CAAC,QAAQ,CAAC,EAAE;IAAvC,OAAO;WAEM,MAAM,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC;IA2BhE,GAAG,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;IAOvD,GAAG,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAYxC,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,qBAAqB,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;IA6BjF,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIvC,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;IAK7E,KAAK,IAAI,IAAI;IAEb,OAAO,CAAC,WAAW;CAYpB"}
|