@n8n/agents 0.1.0
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/LICENSE.md +88 -0
- package/dist/build.tsbuildinfo +1 -0
- package/dist/evals/categorization.d.ts +2 -0
- package/dist/evals/categorization.js +23 -0
- package/dist/evals/categorization.js.map +1 -0
- package/dist/evals/contains-keywords.d.ts +2 -0
- package/dist/evals/contains-keywords.js +29 -0
- package/dist/evals/contains-keywords.js.map +1 -0
- package/dist/evals/correctness.d.ts +2 -0
- package/dist/evals/correctness.js +28 -0
- package/dist/evals/correctness.js.map +1 -0
- package/dist/evals/helpfulness.d.ts +2 -0
- package/dist/evals/helpfulness.js +26 -0
- package/dist/evals/helpfulness.js.map +1 -0
- package/dist/evals/index.d.ts +7 -0
- package/dist/evals/index.js +18 -0
- package/dist/evals/index.js.map +1 -0
- package/dist/evals/json-validity.d.ts +2 -0
- package/dist/evals/json-validity.js +21 -0
- package/dist/evals/json-validity.js.map +1 -0
- package/dist/evals/parse-judge-response.d.ts +2 -0
- package/dist/evals/parse-judge-response.js +25 -0
- package/dist/evals/parse-judge-response.js.map +1 -0
- package/dist/evals/string-similarity.d.ts +2 -0
- package/dist/evals/string-similarity.js +42 -0
- package/dist/evals/string-similarity.js.map +1 -0
- package/dist/evals/tool-call-accuracy.d.ts +2 -0
- package/dist/evals/tool-call-accuracy.js +29 -0
- package/dist/evals/tool-call-accuracy.js.map +1 -0
- package/dist/index.d.ts +34 -0
- package/dist/index.js +85 -0
- package/dist/index.js.map +1 -0
- package/dist/integrations/langsmith.d.ts +13 -0
- package/dist/integrations/langsmith.js +77 -0
- package/dist/integrations/langsmith.js.map +1 -0
- package/dist/runtime/agent-runtime.d.ts +80 -0
- package/dist/runtime/agent-runtime.js +1246 -0
- package/dist/runtime/agent-runtime.js.map +1 -0
- package/dist/runtime/event-bus.d.ts +14 -0
- package/dist/runtime/event-bus.js +53 -0
- package/dist/runtime/event-bus.js.map +1 -0
- package/dist/runtime/logger.d.ts +22 -0
- package/dist/runtime/logger.js +40 -0
- package/dist/runtime/logger.js.map +1 -0
- package/dist/runtime/mcp-connection.d.ts +23 -0
- package/dist/runtime/mcp-connection.js +135 -0
- package/dist/runtime/mcp-connection.js.map +1 -0
- package/dist/runtime/mcp-tool-resolver.d.ts +7 -0
- package/dist/runtime/mcp-tool-resolver.js +64 -0
- package/dist/runtime/mcp-tool-resolver.js.map +1 -0
- package/dist/runtime/memory-store.d.ts +29 -0
- package/dist/runtime/memory-store.js +68 -0
- package/dist/runtime/memory-store.js.map +1 -0
- package/dist/runtime/message-list.d.ts +25 -0
- package/dist/runtime/message-list.js +79 -0
- package/dist/runtime/message-list.js.map +1 -0
- package/dist/runtime/messages.d.ts +8 -0
- package/dist/runtime/messages.js +239 -0
- package/dist/runtime/messages.js.map +1 -0
- package/dist/runtime/model-factory.d.ts +33 -0
- package/dist/runtime/model-factory.js +68 -0
- package/dist/runtime/model-factory.js.map +1 -0
- package/dist/runtime/run-state.d.ts +9 -0
- package/dist/runtime/run-state.js +49 -0
- package/dist/runtime/run-state.js.map +1 -0
- package/dist/runtime/runtime-helpers.d.ts +21 -0
- package/dist/runtime/runtime-helpers.js +96 -0
- package/dist/runtime/runtime-helpers.js.map +1 -0
- package/dist/runtime/stream.d.ts +15 -0
- package/dist/runtime/stream.js +94 -0
- package/dist/runtime/stream.js.map +1 -0
- package/dist/runtime/strip-orphaned-tool-messages.d.ts +2 -0
- package/dist/runtime/strip-orphaned-tool-messages.js +48 -0
- package/dist/runtime/strip-orphaned-tool-messages.js.map +1 -0
- package/dist/runtime/title-generation.d.ts +11 -0
- package/dist/runtime/title-generation.js +56 -0
- package/dist/runtime/title-generation.js.map +1 -0
- package/dist/runtime/tool-adapter.d.ts +23 -0
- package/dist/runtime/tool-adapter.js +104 -0
- package/dist/runtime/tool-adapter.js.map +1 -0
- package/dist/runtime/working-memory.d.ts +21 -0
- package/dist/runtime/working-memory.js +138 -0
- package/dist/runtime/working-memory.js.map +1 -0
- package/dist/sdk/agent.d.ts +88 -0
- package/dist/sdk/agent.js +327 -0
- package/dist/sdk/agent.js.map +1 -0
- package/dist/sdk/catalog.d.ts +31 -0
- package/dist/sdk/catalog.js +80 -0
- package/dist/sdk/catalog.js.map +1 -0
- package/dist/sdk/eval.d.ts +23 -0
- package/dist/sdk/eval.js +99 -0
- package/dist/sdk/eval.js.map +1 -0
- package/dist/sdk/evaluate.d.ts +13 -0
- package/dist/sdk/evaluate.js +84 -0
- package/dist/sdk/evaluate.js.map +1 -0
- package/dist/sdk/guardrail.d.ts +14 -0
- package/dist/sdk/guardrail.js +41 -0
- package/dist/sdk/guardrail.js.map +1 -0
- package/dist/sdk/mcp-client.d.ts +16 -0
- package/dist/sdk/mcp-client.js +128 -0
- package/dist/sdk/mcp-client.js.map +1 -0
- package/dist/sdk/memory.d.ts +22 -0
- package/dist/sdk/memory.js +101 -0
- package/dist/sdk/memory.js.map +1 -0
- package/dist/sdk/message.d.ts +4 -0
- package/dist/sdk/message.js +18 -0
- package/dist/sdk/message.js.map +1 -0
- package/dist/sdk/network.d.ts +20 -0
- package/dist/sdk/network.js +46 -0
- package/dist/sdk/network.js.map +1 -0
- package/dist/sdk/provider-tools.d.ts +18 -0
- package/dist/sdk/provider-tools.js +31 -0
- package/dist/sdk/provider-tools.js.map +1 -0
- package/dist/sdk/telemetry.d.ts +31 -0
- package/dist/sdk/telemetry.js +186 -0
- package/dist/sdk/telemetry.js.map +1 -0
- package/dist/sdk/tool.d.ts +38 -0
- package/dist/sdk/tool.js +143 -0
- package/dist/sdk/tool.js.map +1 -0
- package/dist/sdk/verify.d.ts +5 -0
- package/dist/sdk/verify.js +37 -0
- package/dist/sdk/verify.js.map +1 -0
- package/dist/storage/postgres-memory.d.ts +79 -0
- package/dist/storage/postgres-memory.js +411 -0
- package/dist/storage/postgres-memory.js.map +1 -0
- package/dist/storage/sqlite-memory.d.ts +60 -0
- package/dist/storage/sqlite-memory.js +349 -0
- package/dist/storage/sqlite-memory.js.map +1 -0
- package/dist/types/index.d.ts +13 -0
- package/dist/types/index.js +6 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/runtime/event.d.ts +41 -0
- package/dist/types/runtime/event.js +14 -0
- package/dist/types/runtime/event.js.map +1 -0
- package/dist/types/runtime/message-list.d.ts +7 -0
- package/dist/types/runtime/message-list.js +3 -0
- package/dist/types/runtime/message-list.js.map +1 -0
- package/dist/types/sdk/agent.d.ts +174 -0
- package/dist/types/sdk/agent.js +3 -0
- package/dist/types/sdk/agent.js.map +1 -0
- package/dist/types/sdk/eval.d.ts +41 -0
- package/dist/types/sdk/eval.js +3 -0
- package/dist/types/sdk/eval.js.map +1 -0
- package/dist/types/sdk/guardrail.d.ts +9 -0
- package/dist/types/sdk/guardrail.js +3 -0
- package/dist/types/sdk/guardrail.js.map +1 -0
- package/dist/types/sdk/mcp.d.ts +24 -0
- package/dist/types/sdk/mcp.js +3 -0
- package/dist/types/sdk/mcp.js.map +1 -0
- package/dist/types/sdk/memory.d.ts +99 -0
- package/dist/types/sdk/memory.js +3 -0
- package/dist/types/sdk/memory.js.map +1 -0
- package/dist/types/sdk/message.d.ts +79 -0
- package/dist/types/sdk/message.js +3 -0
- package/dist/types/sdk/message.js.map +1 -0
- package/dist/types/sdk/provider.d.ts +16 -0
- package/dist/types/sdk/provider.js +3 -0
- package/dist/types/sdk/provider.js.map +1 -0
- package/dist/types/sdk/tool.d.ts +32 -0
- package/dist/types/sdk/tool.js +3 -0
- package/dist/types/sdk/tool.js.map +1 -0
- package/dist/types/telemetry.d.ts +18 -0
- package/dist/types/telemetry.js +3 -0
- package/dist/types/telemetry.js.map +1 -0
- package/dist/types/utils/json.d.ts +5 -0
- package/dist/types/utils/json.js +3 -0
- package/dist/types/utils/json.js.map +1 -0
- package/dist/utils/zod.d.ts +3 -0
- package/dist/utils/zod.js +7 -0
- package/dist/utils/zod.js.map +1 -0
- package/dist/workspace/filesystem/base-filesystem.d.ts +40 -0
- package/dist/workspace/filesystem/base-filesystem.js +99 -0
- package/dist/workspace/filesystem/base-filesystem.js.map +1 -0
- package/dist/workspace/index.d.ts +8 -0
- package/dist/workspace/index.js +17 -0
- package/dist/workspace/index.js.map +1 -0
- package/dist/workspace/lifecycle.d.ts +13 -0
- package/dist/workspace/lifecycle.js +11 -0
- package/dist/workspace/lifecycle.js.map +1 -0
- package/dist/workspace/process.d.ts +36 -0
- package/dist/workspace/process.js +76 -0
- package/dist/workspace/process.js.map +1 -0
- package/dist/workspace/sandbox/base-sandbox.d.ts +28 -0
- package/dist/workspace/sandbox/base-sandbox.js +163 -0
- package/dist/workspace/sandbox/base-sandbox.js.map +1 -0
- package/dist/workspace/tools/append-file.d.ts +3 -0
- package/dist/workspace/tools/append-file.js +22 -0
- package/dist/workspace/tools/append-file.js.map +1 -0
- package/dist/workspace/tools/copy-file.d.ts +3 -0
- package/dist/workspace/tools/copy-file.js +26 -0
- package/dist/workspace/tools/copy-file.js.map +1 -0
- package/dist/workspace/tools/delete-file.d.ts +3 -0
- package/dist/workspace/tools/delete-file.js +26 -0
- package/dist/workspace/tools/delete-file.js.map +1 -0
- package/dist/workspace/tools/execute-command.d.ts +3 -0
- package/dist/workspace/tools/execute-command.js +39 -0
- package/dist/workspace/tools/execute-command.js.map +1 -0
- package/dist/workspace/tools/file-stat.d.ts +3 -0
- package/dist/workspace/tools/file-stat.js +33 -0
- package/dist/workspace/tools/file-stat.js.map +1 -0
- package/dist/workspace/tools/list-files.d.ts +3 -0
- package/dist/workspace/tools/list-files.js +28 -0
- package/dist/workspace/tools/list-files.js.map +1 -0
- package/dist/workspace/tools/mkdir.d.ts +3 -0
- package/dist/workspace/tools/mkdir.js +22 -0
- package/dist/workspace/tools/mkdir.js.map +1 -0
- package/dist/workspace/tools/move-file.d.ts +3 -0
- package/dist/workspace/tools/move-file.js +26 -0
- package/dist/workspace/tools/move-file.js.map +1 -0
- package/dist/workspace/tools/process-tools.d.ts +4 -0
- package/dist/workspace/tools/process-tools.js +37 -0
- package/dist/workspace/tools/process-tools.js.map +1 -0
- package/dist/workspace/tools/read-file.d.ts +3 -0
- package/dist/workspace/tools/read-file.js +24 -0
- package/dist/workspace/tools/read-file.js.map +1 -0
- package/dist/workspace/tools/rmdir.d.ts +3 -0
- package/dist/workspace/tools/rmdir.js +26 -0
- package/dist/workspace/tools/rmdir.js.map +1 -0
- package/dist/workspace/tools/workspace-tools.d.ts +8 -0
- package/dist/workspace/tools/workspace-tools.js +39 -0
- package/dist/workspace/tools/workspace-tools.js.map +1 -0
- package/dist/workspace/tools/write-file.d.ts +3 -0
- package/dist/workspace/tools/write-file.js +26 -0
- package/dist/workspace/tools/write-file.js.map +1 -0
- package/dist/workspace/types.d.ts +178 -0
- package/dist/workspace/types.js +7 -0
- package/dist/workspace/types.js.map +1 -0
- package/dist/workspace/workspace.d.ts +21 -0
- package/dist/workspace/workspace.js +131 -0
- package/dist/workspace/workspace.js.map +1 -0
- package/package.json +65 -0
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.AgentMessageList = void 0;
|
|
4
|
+
const messages_1 = require("./messages");
|
|
5
|
+
const strip_orphaned_tool_messages_1 = require("./strip-orphaned-tool-messages");
|
|
6
|
+
const working_memory_1 = require("./working-memory");
|
|
7
|
+
const message_1 = require("../sdk/message");
|
|
8
|
+
class AgentMessageList {
|
|
9
|
+
constructor() {
|
|
10
|
+
this.all = [];
|
|
11
|
+
this.historySet = new Set();
|
|
12
|
+
this.inputSet = new Set();
|
|
13
|
+
this.responseSet = new Set();
|
|
14
|
+
}
|
|
15
|
+
addHistory(messages) {
|
|
16
|
+
for (const m of messages) {
|
|
17
|
+
this.all.push(m);
|
|
18
|
+
this.historySet.add(m);
|
|
19
|
+
}
|
|
20
|
+
}
|
|
21
|
+
addInput(messages) {
|
|
22
|
+
for (const m of messages) {
|
|
23
|
+
this.all.push(m);
|
|
24
|
+
this.inputSet.add(m);
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
addResponse(messages) {
|
|
28
|
+
for (const m of messages) {
|
|
29
|
+
this.all.push(m);
|
|
30
|
+
this.responseSet.add(m);
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
forLlm(baseInstructions, instructionProviderOptions) {
|
|
34
|
+
let systemPrompt = baseInstructions;
|
|
35
|
+
if (this.workingMemory) {
|
|
36
|
+
const wmInstruction = (0, working_memory_1.buildWorkingMemoryInstruction)(this.workingMemory.template, this.workingMemory.structured);
|
|
37
|
+
const wmState = this.workingMemory.state ?? this.workingMemory.template;
|
|
38
|
+
systemPrompt +=
|
|
39
|
+
wmInstruction + '\n\nCurrent working memory state:\n```\n' + wmState + '\n```';
|
|
40
|
+
}
|
|
41
|
+
const systemMessage = instructionProviderOptions
|
|
42
|
+
? { role: 'system', content: systemPrompt, providerOptions: instructionProviderOptions }
|
|
43
|
+
: { role: 'system', content: systemPrompt };
|
|
44
|
+
return [systemMessage, ...(0, messages_1.toAiMessages)((0, message_1.filterLlmMessages)((0, strip_orphaned_tool_messages_1.stripOrphanedToolMessages)(this.all)))];
|
|
45
|
+
}
|
|
46
|
+
turnDelta() {
|
|
47
|
+
return this.all.filter((m) => this.inputSet.has(m) || this.responseSet.has(m));
|
|
48
|
+
}
|
|
49
|
+
responseDelta() {
|
|
50
|
+
return this.all.filter((m) => this.responseSet.has(m));
|
|
51
|
+
}
|
|
52
|
+
serialize() {
|
|
53
|
+
const toIds = (set) => Array.from(set).map((m) => m.id);
|
|
54
|
+
return {
|
|
55
|
+
messages: [...this.all],
|
|
56
|
+
historyIds: toIds(this.historySet),
|
|
57
|
+
inputIds: toIds(this.inputSet),
|
|
58
|
+
responseIds: toIds(this.responseSet),
|
|
59
|
+
};
|
|
60
|
+
}
|
|
61
|
+
static deserialize(data) {
|
|
62
|
+
const list = new AgentMessageList();
|
|
63
|
+
const historyIdSet = new Set(data.historyIds);
|
|
64
|
+
const inputIdSet = new Set(data.inputIds);
|
|
65
|
+
const responseIdSet = new Set(data.responseIds);
|
|
66
|
+
for (const m of data.messages) {
|
|
67
|
+
list.all.push(m);
|
|
68
|
+
if (historyIdSet.has(m.id))
|
|
69
|
+
list.historySet.add(m);
|
|
70
|
+
if (inputIdSet.has(m.id))
|
|
71
|
+
list.inputSet.add(m);
|
|
72
|
+
if (responseIdSet.has(m.id))
|
|
73
|
+
list.responseSet.add(m);
|
|
74
|
+
}
|
|
75
|
+
return list;
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
exports.AgentMessageList = AgentMessageList;
|
|
79
|
+
//# sourceMappingURL=message-list.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"message-list.js","sourceRoot":"","sources":["../../src/runtime/message-list.ts"],"names":[],"mappings":";;;AAGA,yCAA0C;AAC1C,iFAA2E;AAC3E,qDAAiE;AACjE,4CAAmD;AAyBnD,MAAa,gBAAgB;IAA7B;QACS,QAAG,GAAqB,EAAE,CAAC;QAE3B,eAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;QAEvC,aAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;QAErC,gBAAW,GAAG,IAAI,GAAG,EAAkB,CAAC;IAyFjD,CAAC;IApFA,UAAU,CAAC,QAA0B;QACpC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACjB,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACxB,CAAC;IACF,CAAC;IAED,QAAQ,CAAC,QAA0B;QAClC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACjB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACtB,CAAC;IACF,CAAC;IAED,WAAW,CAAC,QAA0B;QACrC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACjB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACzB,CAAC;IACF,CAAC;IAOD,MAAM,CAAC,gBAAwB,EAAE,0BAA4C;QAC5E,IAAI,YAAY,GAAG,gBAAgB,CAAC;QAEpC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACxB,MAAM,aAAa,GAAG,IAAA,8CAA6B,EAClD,IAAI,CAAC,aAAa,CAAC,QAAQ,EAC3B,IAAI,CAAC,aAAa,CAAC,UAAU,CAC7B,CAAC;YACF,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,IAAI,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC;YACxE,YAAY;gBACX,aAAa,GAAG,0CAA0C,GAAG,OAAO,GAAG,OAAO,CAAC;QACjF,CAAC;QAED,MAAM,aAAa,GAAiB,0BAA0B;YAC7D,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,0BAA0B,EAAE;YACxF,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC;QAC7C,OAAO,CAAC,aAAa,EAAE,GAAG,IAAA,uBAAY,EAAC,IAAA,2BAAiB,EAAC,IAAA,wDAAyB,EAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACjG,CAAC;IAMD,SAAS;QACR,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAChF,CAAC;IAMD,aAAa;QACZ,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACxD,CAAC;IAED,SAAS;QACR,MAAM,KAAK,GAAG,CAAC,GAAwB,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC7E,OAAO;YACN,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC;YACvB,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC;YAClC,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC;YAC9B,WAAW,EAAE,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC;SACpC,CAAC;IACH,CAAC;IAED,MAAM,CAAC,WAAW,CAAC,IAA2B;QAC7C,MAAM,IAAI,GAAG,IAAI,gBAAgB,EAAE,CAAC;QACpC,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC9C,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAChD,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC/B,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACjB,IAAI,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBAAE,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACnD,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAC/C,IAAI,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBAAE,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,IAAI,CAAC;IACb,CAAC;CACD;AAhGD,4CAgGC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import type { ModelMessage, FinishReason as AiFinishReason } from 'ai';
|
|
2
|
+
import type { FinishReason } from '../types';
|
|
3
|
+
import type { AgentDbMessage, Message } from '../types/sdk/message';
|
|
4
|
+
export declare function toAiMessage(msg: Message): ModelMessage;
|
|
5
|
+
export declare function toAiMessages(messages: Message[]): ModelMessage[];
|
|
6
|
+
export declare function fromAiMessage(msg: ModelMessage): AgentDbMessage;
|
|
7
|
+
export declare function fromAiMessages(messages: ModelMessage[]): AgentDbMessage[];
|
|
8
|
+
export declare function fromAiFinishReason(reason: AiFinishReason): FinishReason;
|
|
@@ -0,0 +1,239 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.toAiMessage = toAiMessage;
|
|
4
|
+
exports.toAiMessages = toAiMessages;
|
|
5
|
+
exports.fromAiMessage = fromAiMessage;
|
|
6
|
+
exports.fromAiMessages = fromAiMessages;
|
|
7
|
+
exports.fromAiFinishReason = fromAiFinishReason;
|
|
8
|
+
const message_1 = require("../sdk/message");
|
|
9
|
+
function isText(block) {
|
|
10
|
+
return block.type === 'text';
|
|
11
|
+
}
|
|
12
|
+
function isReasoning(block) {
|
|
13
|
+
return block.type === 'reasoning';
|
|
14
|
+
}
|
|
15
|
+
function isFile(block) {
|
|
16
|
+
return block.type === 'file';
|
|
17
|
+
}
|
|
18
|
+
function isToolCall(block) {
|
|
19
|
+
return block.type === 'tool-call';
|
|
20
|
+
}
|
|
21
|
+
function isToolResult(block) {
|
|
22
|
+
return block.type === 'tool-result';
|
|
23
|
+
}
|
|
24
|
+
function parseJsonValue(value) {
|
|
25
|
+
if (typeof value === 'string') {
|
|
26
|
+
try {
|
|
27
|
+
return JSON.parse(value);
|
|
28
|
+
}
|
|
29
|
+
catch {
|
|
30
|
+
return value;
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
return value;
|
|
34
|
+
}
|
|
35
|
+
function toAiContent(block) {
|
|
36
|
+
let base;
|
|
37
|
+
if (isText(block)) {
|
|
38
|
+
base = { type: 'text', text: block.text };
|
|
39
|
+
}
|
|
40
|
+
else if (isFile(block)) {
|
|
41
|
+
base = {
|
|
42
|
+
type: 'file',
|
|
43
|
+
data: block.data,
|
|
44
|
+
mediaType: block.mediaType ?? 'application/octet-stream',
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
else if (isToolCall(block)) {
|
|
48
|
+
base = {
|
|
49
|
+
type: 'tool-call',
|
|
50
|
+
toolCallId: block.toolCallId ?? '',
|
|
51
|
+
toolName: block.toolName,
|
|
52
|
+
input: parseJsonValue(block.input),
|
|
53
|
+
providerExecuted: block.providerExecuted,
|
|
54
|
+
};
|
|
55
|
+
}
|
|
56
|
+
if (isToolResult(block)) {
|
|
57
|
+
if (block.isError) {
|
|
58
|
+
if (typeof block.result === 'string') {
|
|
59
|
+
base = {
|
|
60
|
+
type: 'tool-result',
|
|
61
|
+
toolCallId: block.toolCallId,
|
|
62
|
+
toolName: block.toolName,
|
|
63
|
+
output: { type: 'error-text', value: block.result },
|
|
64
|
+
};
|
|
65
|
+
}
|
|
66
|
+
else {
|
|
67
|
+
base = {
|
|
68
|
+
type: 'tool-result',
|
|
69
|
+
toolCallId: block.toolCallId,
|
|
70
|
+
toolName: block.toolName,
|
|
71
|
+
output: { type: 'error-json', value: block.result },
|
|
72
|
+
};
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
else {
|
|
76
|
+
base = {
|
|
77
|
+
type: 'tool-result',
|
|
78
|
+
toolCallId: block.toolCallId,
|
|
79
|
+
toolName: block.toolName,
|
|
80
|
+
output: { type: 'json', value: block.result },
|
|
81
|
+
};
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
else if (isReasoning(block)) {
|
|
85
|
+
base = { type: 'reasoning', text: block.text };
|
|
86
|
+
}
|
|
87
|
+
if (base && block.providerOptions) {
|
|
88
|
+
return { ...base, providerOptions: block.providerOptions };
|
|
89
|
+
}
|
|
90
|
+
return base;
|
|
91
|
+
}
|
|
92
|
+
function fromAiContent(part) {
|
|
93
|
+
const providerOptions = 'providerOptions' in part ? part.providerOptions : undefined;
|
|
94
|
+
let base;
|
|
95
|
+
switch (part.type) {
|
|
96
|
+
case 'text':
|
|
97
|
+
base = { type: 'text', text: part.text };
|
|
98
|
+
break;
|
|
99
|
+
case 'file': {
|
|
100
|
+
const data = part.data instanceof URL ? part.data.toString() : part.data;
|
|
101
|
+
base = { type: 'file', data, mediaType: part.mediaType };
|
|
102
|
+
break;
|
|
103
|
+
}
|
|
104
|
+
case 'image': {
|
|
105
|
+
const data = part.image instanceof URL ? part.image.toString() : part.image;
|
|
106
|
+
base = { type: 'file', data, mediaType: part.mediaType };
|
|
107
|
+
break;
|
|
108
|
+
}
|
|
109
|
+
case 'reasoning':
|
|
110
|
+
base = { type: 'reasoning', text: part.text };
|
|
111
|
+
break;
|
|
112
|
+
case 'tool-call':
|
|
113
|
+
base = {
|
|
114
|
+
type: 'tool-call',
|
|
115
|
+
toolCallId: part.toolCallId,
|
|
116
|
+
toolName: part.toolName,
|
|
117
|
+
input: part.input,
|
|
118
|
+
providerExecuted: part.providerExecuted,
|
|
119
|
+
};
|
|
120
|
+
break;
|
|
121
|
+
case 'tool-result': {
|
|
122
|
+
const { output } = part;
|
|
123
|
+
let result;
|
|
124
|
+
let isError;
|
|
125
|
+
if (output.type === 'json') {
|
|
126
|
+
result = output.value;
|
|
127
|
+
}
|
|
128
|
+
else if (output.type === 'text') {
|
|
129
|
+
result = output.value;
|
|
130
|
+
}
|
|
131
|
+
else if (output.type === 'error-json') {
|
|
132
|
+
result = output.value;
|
|
133
|
+
isError = true;
|
|
134
|
+
}
|
|
135
|
+
else if (output.type === 'error-text') {
|
|
136
|
+
result = output.value;
|
|
137
|
+
isError = true;
|
|
138
|
+
}
|
|
139
|
+
else {
|
|
140
|
+
result = null;
|
|
141
|
+
isError = true;
|
|
142
|
+
}
|
|
143
|
+
base = {
|
|
144
|
+
type: 'tool-result',
|
|
145
|
+
toolCallId: part.toolCallId,
|
|
146
|
+
toolName: part.toolName,
|
|
147
|
+
result,
|
|
148
|
+
isError,
|
|
149
|
+
};
|
|
150
|
+
break;
|
|
151
|
+
}
|
|
152
|
+
case 'tool-approval-request':
|
|
153
|
+
case 'tool-approval-response':
|
|
154
|
+
default:
|
|
155
|
+
return undefined;
|
|
156
|
+
}
|
|
157
|
+
if (base && providerOptions) {
|
|
158
|
+
return { ...base, providerOptions };
|
|
159
|
+
}
|
|
160
|
+
return base;
|
|
161
|
+
}
|
|
162
|
+
function toAiMessage(msg) {
|
|
163
|
+
let base;
|
|
164
|
+
switch (msg.role) {
|
|
165
|
+
case 'system': {
|
|
166
|
+
const text = msg.content
|
|
167
|
+
.filter(isText)
|
|
168
|
+
.map((b) => b.text)
|
|
169
|
+
.join('');
|
|
170
|
+
base = { role: 'system', content: text };
|
|
171
|
+
break;
|
|
172
|
+
}
|
|
173
|
+
case 'user': {
|
|
174
|
+
const parts = msg.content
|
|
175
|
+
.map(toAiContent)
|
|
176
|
+
.filter((p) => p?.type === 'text' || p?.type === 'file');
|
|
177
|
+
base = { role: 'user', content: parts };
|
|
178
|
+
break;
|
|
179
|
+
}
|
|
180
|
+
case 'assistant': {
|
|
181
|
+
const parts = msg.content
|
|
182
|
+
.map(toAiContent)
|
|
183
|
+
.filter((p) => p?.type === 'text' ||
|
|
184
|
+
p?.type === 'reasoning' ||
|
|
185
|
+
p?.type === 'tool-call' ||
|
|
186
|
+
p?.type === 'tool-result' ||
|
|
187
|
+
p?.type === 'file');
|
|
188
|
+
base = { role: 'assistant', content: parts };
|
|
189
|
+
break;
|
|
190
|
+
}
|
|
191
|
+
case 'tool': {
|
|
192
|
+
const parts = msg.content
|
|
193
|
+
.map(toAiContent)
|
|
194
|
+
.filter((p) => p?.type === 'tool-result');
|
|
195
|
+
base = { role: 'tool', content: parts };
|
|
196
|
+
break;
|
|
197
|
+
}
|
|
198
|
+
default:
|
|
199
|
+
throw new Error(`Unknown role: ${msg.role}`);
|
|
200
|
+
}
|
|
201
|
+
if (msg.providerOptions) {
|
|
202
|
+
return { ...base, providerOptions: msg.providerOptions };
|
|
203
|
+
}
|
|
204
|
+
return base;
|
|
205
|
+
}
|
|
206
|
+
function toAiMessages(messages) {
|
|
207
|
+
return messages.map(toAiMessage);
|
|
208
|
+
}
|
|
209
|
+
function fromAiMessage(msg) {
|
|
210
|
+
const rawContent = msg.content;
|
|
211
|
+
const content = typeof rawContent === 'string'
|
|
212
|
+
? [{ type: 'text', text: rawContent }]
|
|
213
|
+
: rawContent.map(fromAiContent).filter((p) => p !== undefined);
|
|
214
|
+
const message = { role: msg.role, content };
|
|
215
|
+
if ('providerOptions' in msg && msg.providerOptions) {
|
|
216
|
+
message.providerOptions = msg.providerOptions;
|
|
217
|
+
}
|
|
218
|
+
return (0, message_1.toDbMessage)(message);
|
|
219
|
+
}
|
|
220
|
+
function fromAiMessages(messages) {
|
|
221
|
+
return messages.map(fromAiMessage);
|
|
222
|
+
}
|
|
223
|
+
function fromAiFinishReason(reason) {
|
|
224
|
+
switch (reason) {
|
|
225
|
+
case 'stop':
|
|
226
|
+
return 'stop';
|
|
227
|
+
case 'length':
|
|
228
|
+
return 'length';
|
|
229
|
+
case 'content-filter':
|
|
230
|
+
return 'content-filter';
|
|
231
|
+
case 'tool-calls':
|
|
232
|
+
return 'tool-calls';
|
|
233
|
+
case 'error':
|
|
234
|
+
return 'error';
|
|
235
|
+
case 'other':
|
|
236
|
+
return 'other';
|
|
237
|
+
}
|
|
238
|
+
}
|
|
239
|
+
//# sourceMappingURL=messages.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"messages.js","sourceRoot":"","sources":["../../src/runtime/messages.ts"],"names":[],"mappings":";;AA8MA,kCAmDC;AAGD,oCAEC;AAGD,sCAWC;AAGD,wCAEC;AAED,gDAeC;AA9RD,4CAA6C;AA8B7C,SAAS,MAAM,CAAC,KAAqB;IACpC,OAAO,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC;AAC9B,CAAC;AAED,SAAS,WAAW,CAAC,KAAqB;IACzC,OAAO,KAAK,CAAC,IAAI,KAAK,WAAW,CAAC;AACnC,CAAC;AAED,SAAS,MAAM,CAAC,KAAqB;IACpC,OAAO,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC;AAC9B,CAAC;AAED,SAAS,UAAU,CAAC,KAAqB;IACxC,OAAO,KAAK,CAAC,IAAI,KAAK,WAAW,CAAC;AACnC,CAAC;AAED,SAAS,YAAY,CAAC,KAAqB;IAC1C,OAAO,KAAK,CAAC,IAAI,KAAK,aAAa,CAAC;AACrC,CAAC;AAMD,SAAS,cAAc,CAAC,KAAgB;IACvC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC/B,IAAI,CAAC;YACJ,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;QAAC,MAAM,CAAC;YACR,OAAO,KAAK,CAAC;QACd,CAAC;IACF,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAGD,SAAS,WAAW,CAAC,KAAqB;IACzC,IAAI,IAA+B,CAAC;IACpC,IAAI,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACnB,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;IAC3C,CAAC;SAAM,IAAI,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,IAAI,GAAG;YACN,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,0BAA0B;SACxD,CAAC;IACH,CAAC;SAAM,IAAI,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;QAC9B,IAAI,GAAG;YACN,IAAI,EAAE,WAAW;YACjB,UAAU,EAAE,KAAK,CAAC,UAAU,IAAI,EAAE;YAClC,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,KAAK,EAAE,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC;YAClC,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;SACxC,CAAC;IACH,CAAC;IACD,IAAI,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YACnB,IAAI,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBACtC,IAAI,GAAG;oBACN,IAAI,EAAE,aAAa;oBACnB,UAAU,EAAE,KAAK,CAAC,UAAU;oBAC5B,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,MAAM,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE;iBACnD,CAAC;YACH,CAAC;iBAAM,CAAC;gBACP,IAAI,GAAG;oBACN,IAAI,EAAE,aAAa;oBACnB,UAAU,EAAE,KAAK,CAAC,UAAU;oBAC5B,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,MAAM,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE;iBACnD,CAAC;YACH,CAAC;QACF,CAAC;aAAM,CAAC;YACP,IAAI,GAAG;gBACN,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE;aAC7C,CAAC;QACH,CAAC;IACF,CAAC;SAAM,IAAI,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC;QAC/B,IAAI,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;IAChD,CAAC;IAED,IAAI,IAAI,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;QACnC,OAAO,EAAE,GAAG,IAAI,EAAE,eAAe,EAAE,KAAK,CAAC,eAAe,EAAmB,CAAC;IAC7E,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC;AAGD,SAAS,aAAa,CAAC,IAAmB;IACzC,MAAM,eAAe,GAAG,iBAAiB,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS,CAAC;IAErF,IAAI,IAAgC,CAAC;IACrC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,MAAM;YACV,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YACzC,MAAM;QACP,KAAK,MAAM,CAAC,CAAC,CAAC;YACb,MAAM,IAAI,GACT,IAAI,CAAC,IAAI,YAAY,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAE,IAAI,CAAC,IAA4B,CAAC;YACtF,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC;YACzD,MAAM;QACP,CAAC;QACD,KAAK,OAAO,CAAC,CAAC,CAAC;YACd,MAAM,IAAI,GACT,IAAI,CAAC,KAAK,YAAY,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAE,IAAI,CAAC,KAA6B,CAAC;YACzF,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC;YACzD,MAAM;QACP,CAAC;QACD,KAAK,WAAW;YACf,IAAI,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YAC9C,MAAM;QACP,KAAK,WAAW;YACf,IAAI,GAAG;gBACN,IAAI,EAAE,WAAW;gBACjB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,KAAK,EAAE,IAAI,CAAC,KAAkB;gBAC9B,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;aACvC,CAAC;YACF,MAAM;QACP,KAAK,aAAa,CAAC,CAAC,CAAC;YACpB,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;YACxB,IAAI,MAAiB,CAAC;YACtB,IAAI,OAA4B,CAAC;YACjC,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC5B,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC;YACvB,CAAC;iBAAM,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACnC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC;YACvB,CAAC;iBAAM,IAAI,MAAM,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACzC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC;gBACtB,OAAO,GAAG,IAAI,CAAC;YAChB,CAAC;iBAAM,IAAI,MAAM,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACzC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC;gBACtB,OAAO,GAAG,IAAI,CAAC;YAChB,CAAC;iBAAM,CAAC;gBACP,MAAM,GAAG,IAAI,CAAC;gBACd,OAAO,GAAG,IAAI,CAAC;YAChB,CAAC;YACD,IAAI,GAAG;gBACN,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,MAAM;gBACN,OAAO;aACP,CAAC;YACF,MAAM;QACP,CAAC;QAED,KAAK,uBAAuB,CAAC;QAC7B,KAAK,wBAAwB,CAAC;QAC9B;YACC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,IAAI,IAAI,eAAe,EAAE,CAAC;QAC7B,OAAO,EAAE,GAAG,IAAI,EAAE,eAAe,EAAE,CAAC;IACrC,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC;AAGD,SAAgB,WAAW,CAAC,GAAY;IACvC,IAAI,IAAkB,CAAC;IACvB,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,QAAQ,CAAC,CAAC,CAAC;YACf,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO;iBACtB,MAAM,CAAC,MAAM,CAAC;iBACd,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;iBAClB,IAAI,CAAC,EAAE,CAAC,CAAC;YACX,IAAI,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACzC,MAAM;QACP,CAAC;QAED,KAAK,MAAM,CAAC,CAAC,CAAC;YACb,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO;iBACvB,GAAG,CAAC,WAAW,CAAC;iBAChB,MAAM,CAAC,CAAC,CAAC,EAA4B,EAAE,CAAC,CAAC,EAAE,IAAI,KAAK,MAAM,IAAI,CAAC,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC;YACpF,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YACxC,MAAM;QACP,CAAC;QAED,KAAK,WAAW,CAAC,CAAC,CAAC;YAClB,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO;iBACvB,GAAG,CAAC,WAAW,CAAC;iBAChB,MAAM,CACN,CAAC,CAAC,EAA4E,EAAE,CAC/E,CAAC,EAAE,IAAI,KAAK,MAAM;gBAClB,CAAC,EAAE,IAAI,KAAK,WAAW;gBACvB,CAAC,EAAE,IAAI,KAAK,WAAW;gBACvB,CAAC,EAAE,IAAI,KAAK,aAAa;gBACzB,CAAC,EAAE,IAAI,KAAK,MAAM,CACnB,CAAC;YACH,IAAI,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC7C,MAAM;QACP,CAAC;QAED,KAAK,MAAM,CAAC,CAAC,CAAC;YACb,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO;iBACvB,GAAG,CAAC,WAAW,CAAC;iBAChB,MAAM,CAAC,CAAC,CAAC,EAAuB,EAAE,CAAC,CAAC,EAAE,IAAI,KAAK,aAAa,CAAC,CAAC;YAChE,IAAI,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YACxC,MAAM;QACP,CAAC;QAED;YACC,MAAM,IAAI,KAAK,CAAC,iBAAiB,GAAG,CAAC,IAAc,EAAE,CAAC,CAAC;IACzD,CAAC;IAED,IAAI,GAAG,CAAC,eAAe,EAAE,CAAC;QACzB,OAAO,EAAE,GAAG,IAAI,EAAE,eAAe,EAAE,GAAG,CAAC,eAAe,EAAE,CAAC;IAC1D,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC;AAGD,SAAgB,YAAY,CAAC,QAAmB;IAC/C,OAAO,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;AAClC,CAAC;AAGD,SAAgB,aAAa,CAAC,GAAiB;IAC9C,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC;IAC/B,MAAM,OAAO,GACZ,OAAO,UAAU,KAAK,QAAQ;QAC7B,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;QACtC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAuB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;IACtF,MAAM,OAAO,GAAiB,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC;IAC1D,IAAI,iBAAiB,IAAI,GAAG,IAAI,GAAG,CAAC,eAAe,EAAE,CAAC;QACrD,OAAO,CAAC,eAAe,GAAG,GAAG,CAAC,eAAe,CAAC;IAC/C,CAAC;IACD,OAAO,IAAA,qBAAW,EAAC,OAAO,CAAC,CAAC;AAC7B,CAAC;AAGD,SAAgB,cAAc,CAAC,QAAwB;IACtD,OAAO,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;AACpC,CAAC;AAED,SAAgB,kBAAkB,CAAC,MAAsB;IACxD,QAAQ,MAAM,EAAE,CAAC;QAChB,KAAK,MAAM;YACV,OAAO,MAAM,CAAC;QACf,KAAK,QAAQ;YACZ,OAAO,QAAQ,CAAC;QACjB,KAAK,gBAAgB;YACpB,OAAO,gBAAgB,CAAC;QACzB,KAAK,YAAY;YAChB,OAAO,YAAY,CAAC;QACrB,KAAK,OAAO;YACX,OAAO,OAAO,CAAC;QAChB,KAAK,OAAO;YACX,OAAO,OAAO,CAAC;IACjB,CAAC;AACF,CAAC"}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
import type { EmbeddingModel, LanguageModel } from 'ai';
|
|
2
|
+
import type { ModelConfig } from '../types/sdk/agent';
|
|
3
|
+
export declare function createModel(config: ModelConfig): LanguageModel;
|
|
4
|
+
declare const EMBEDDING_PROVIDERS: {
|
|
5
|
+
readonly openai: {
|
|
6
|
+
readonly pkg: "@ai-sdk/openai";
|
|
7
|
+
readonly factory: "createOpenAI";
|
|
8
|
+
};
|
|
9
|
+
readonly google: {
|
|
10
|
+
readonly pkg: "@ai-sdk/google";
|
|
11
|
+
readonly factory: "createGoogleGenerativeAI";
|
|
12
|
+
};
|
|
13
|
+
readonly mistral: {
|
|
14
|
+
readonly pkg: "@ai-sdk/mistral";
|
|
15
|
+
readonly factory: "createMistral";
|
|
16
|
+
};
|
|
17
|
+
readonly cohere: {
|
|
18
|
+
readonly pkg: "@ai-sdk/cohere";
|
|
19
|
+
readonly factory: "createCohere";
|
|
20
|
+
};
|
|
21
|
+
readonly amazon: {
|
|
22
|
+
readonly pkg: "@ai-sdk/amazon-bedrock";
|
|
23
|
+
readonly factory: "createAmazonBedrock";
|
|
24
|
+
};
|
|
25
|
+
readonly bedrock: {
|
|
26
|
+
readonly pkg: "@ai-sdk/amazon-bedrock";
|
|
27
|
+
readonly factory: "createAmazonBedrock";
|
|
28
|
+
};
|
|
29
|
+
};
|
|
30
|
+
type EmbeddingProvider = keyof typeof EMBEDDING_PROVIDERS;
|
|
31
|
+
type EmbeddingModelId = `${EmbeddingProvider}/${string}`;
|
|
32
|
+
export declare function createEmbeddingModel(embedderString: EmbeddingModelId | (string & {}), apiKey?: string): EmbeddingModel;
|
|
33
|
+
export {};
|
|
@@ -0,0 +1,68 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.createModel = createModel;
|
|
4
|
+
exports.createEmbeddingModel = createEmbeddingModel;
|
|
5
|
+
function isLanguageModel(config) {
|
|
6
|
+
return typeof config === 'object' && config !== null && 'doGenerate' in config;
|
|
7
|
+
}
|
|
8
|
+
function createModel(config) {
|
|
9
|
+
if (isLanguageModel(config)) {
|
|
10
|
+
return config;
|
|
11
|
+
}
|
|
12
|
+
const stripEmpty = (value) => {
|
|
13
|
+
if (!value)
|
|
14
|
+
return undefined;
|
|
15
|
+
if (typeof value === 'string' && value.trim() === '')
|
|
16
|
+
return undefined;
|
|
17
|
+
return value;
|
|
18
|
+
};
|
|
19
|
+
const modelId = stripEmpty(typeof config === 'string' ? config : config.id);
|
|
20
|
+
const apiKey = stripEmpty(typeof config === 'string' ? undefined : config.apiKey);
|
|
21
|
+
const baseURL = stripEmpty(typeof config === 'string' ? undefined : config.url);
|
|
22
|
+
if (!modelId) {
|
|
23
|
+
throw new Error('Model ID is required');
|
|
24
|
+
}
|
|
25
|
+
const [provider, ...rest] = modelId.split('/');
|
|
26
|
+
const modelName = rest.join('/');
|
|
27
|
+
switch (provider) {
|
|
28
|
+
case 'anthropic': {
|
|
29
|
+
const { createAnthropic } = require('@ai-sdk/anthropic');
|
|
30
|
+
return createAnthropic({ apiKey, baseURL })(modelName);
|
|
31
|
+
}
|
|
32
|
+
case 'openai': {
|
|
33
|
+
const { createOpenAI } = require('@ai-sdk/openai');
|
|
34
|
+
return createOpenAI({ apiKey, baseURL })(modelName);
|
|
35
|
+
}
|
|
36
|
+
case 'google': {
|
|
37
|
+
const { createGoogleGenerativeAI } = require('@ai-sdk/google');
|
|
38
|
+
return createGoogleGenerativeAI({ apiKey, baseURL })(modelName);
|
|
39
|
+
}
|
|
40
|
+
case 'xai': {
|
|
41
|
+
const { createXai } = require('@ai-sdk/xai');
|
|
42
|
+
return createXai({ apiKey, baseURL })(modelName);
|
|
43
|
+
}
|
|
44
|
+
default:
|
|
45
|
+
throw new Error(`Unsupported provider: "${provider}". Supported: anthropic, openai, google, xai`);
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
const EMBEDDING_PROVIDERS = {
|
|
49
|
+
openai: { pkg: '@ai-sdk/openai', factory: 'createOpenAI' },
|
|
50
|
+
google: { pkg: '@ai-sdk/google', factory: 'createGoogleGenerativeAI' },
|
|
51
|
+
mistral: { pkg: '@ai-sdk/mistral', factory: 'createMistral' },
|
|
52
|
+
cohere: { pkg: '@ai-sdk/cohere', factory: 'createCohere' },
|
|
53
|
+
amazon: { pkg: '@ai-sdk/amazon-bedrock', factory: 'createAmazonBedrock' },
|
|
54
|
+
bedrock: { pkg: '@ai-sdk/amazon-bedrock', factory: 'createAmazonBedrock' },
|
|
55
|
+
};
|
|
56
|
+
function createEmbeddingModel(embedderString, apiKey) {
|
|
57
|
+
const [provider, ...rest] = embedderString.split('/');
|
|
58
|
+
const modelName = rest.join('/');
|
|
59
|
+
const entry = EMBEDDING_PROVIDERS[provider];
|
|
60
|
+
if (!entry) {
|
|
61
|
+
const supported = Object.keys(EMBEDDING_PROVIDERS).join(', ');
|
|
62
|
+
throw new Error(`Unsupported embedding provider: "${provider}". Supported: ${supported}`);
|
|
63
|
+
}
|
|
64
|
+
const mod = require(entry.pkg);
|
|
65
|
+
const factory = mod[entry.factory];
|
|
66
|
+
return factory({ apiKey }).embeddingModel(modelName);
|
|
67
|
+
}
|
|
68
|
+
//# sourceMappingURL=model-factory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-factory.js","sourceRoot":"","sources":["../../src/runtime/model-factory.ts"],"names":[],"mappings":";;AAqBA,kCAoDC;AA0BD,oDAgBC;AAtGD,SAAS,eAAe,CAAC,MAAe;IACvC,OAAO,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,IAAI,YAAY,IAAI,MAAM,CAAC;AAChF,CAAC;AAMD,SAAgB,WAAW,CAAC,MAAmB;IAC9C,IAAI,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;QAC7B,OAAO,MAAM,CAAC;IACf,CAAC;IAED,MAAM,UAAU,GAAG,CAAI,KAAoB,EAAiB,EAAE;QAC7D,IAAI,CAAC,KAAK;YAAE,OAAO,SAAS,CAAC;QAC7B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE;YAAE,OAAO,SAAS,CAAC;QACvE,OAAO,KAAK,CAAC;IACd,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC5E,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAClF,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAEhF,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IACzC,CAAC;IAED,MAAM,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAEjC,QAAQ,QAAQ,EAAE,CAAC;QAClB,KAAK,WAAW,CAAC,CAAC,CAAC;YAClB,MAAM,EAAE,eAAe,EAAE,GAAG,OAAO,CAAC,mBAAmB,CAEtD,CAAC;YACF,OAAO,eAAe,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC;QACxD,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACf,MAAM,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC,gBAAgB,CAEhD,CAAC;YACF,OAAO,YAAY,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC;QACrD,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACf,MAAM,EAAE,wBAAwB,EAAE,GAAG,OAAO,CAAC,gBAAgB,CAE5D,CAAC;YACF,OAAO,wBAAwB,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC;QACjE,CAAC;QACD,KAAK,KAAK,CAAC,CAAC,CAAC;YACZ,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,aAAa,CAE1C,CAAC;YACF,OAAO,SAAS,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC;QAClD,CAAC;QACD;YACC,MAAM,IAAI,KAAK,CACd,0BAA0B,QAAQ,8CAA8C,CAChF,CAAC;IACJ,CAAC;AACF,CAAC;AASD,MAAM,mBAAmB,GAAG;IAC3B,MAAM,EAAE,EAAE,GAAG,EAAE,gBAAgB,EAAE,OAAO,EAAE,cAAc,EAAE;IAC1D,MAAM,EAAE,EAAE,GAAG,EAAE,gBAAgB,EAAE,OAAO,EAAE,0BAA0B,EAAE;IACtE,OAAO,EAAE,EAAE,GAAG,EAAE,iBAAiB,EAAE,OAAO,EAAE,eAAe,EAAE;IAC7D,MAAM,EAAE,EAAE,GAAG,EAAE,gBAAgB,EAAE,OAAO,EAAE,cAAc,EAAE;IAC1D,MAAM,EAAE,EAAE,GAAG,EAAE,wBAAwB,EAAE,OAAO,EAAE,qBAAqB,EAAE;IACzE,OAAO,EAAE,EAAE,GAAG,EAAE,wBAAwB,EAAE,OAAO,EAAE,qBAAqB,EAAE;CACjE,CAAC;AAUX,SAAgB,oBAAoB,CACnC,cAAgD,EAChD,MAAe;IAEf,MAAM,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,GAAG,cAAc,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACtD,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAEjC,MAAM,KAAK,GAAG,mBAAmB,CAAC,QAA6B,CAAC,CAAC;IACjE,IAAI,CAAC,KAAK,EAAE,CAAC;QACZ,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9D,MAAM,IAAI,KAAK,CAAC,oCAAoC,QAAQ,iBAAiB,SAAS,EAAE,CAAC,CAAC;IAC3F,CAAC;IAED,MAAM,GAAG,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAA8C,CAAC;IAC5E,MAAM,OAAO,GAAG,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACnC,OAAO,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;AACtD,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import type { CheckpointStore, SerializableAgentState } from '../types';
|
|
2
|
+
export declare class RunStateManager {
|
|
3
|
+
private store;
|
|
4
|
+
constructor(storage?: 'memory' | CheckpointStore);
|
|
5
|
+
suspend(runId: string, state: SerializableAgentState): Promise<void>;
|
|
6
|
+
resume(runId: string): Promise<SerializableAgentState | undefined>;
|
|
7
|
+
complete(runId: string): Promise<void>;
|
|
8
|
+
}
|
|
9
|
+
export declare function generateRunId(): string;
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.RunStateManager = void 0;
|
|
4
|
+
exports.generateRunId = generateRunId;
|
|
5
|
+
class MemoryCheckpointStore {
|
|
6
|
+
constructor() {
|
|
7
|
+
this.store = new Map();
|
|
8
|
+
}
|
|
9
|
+
async save(key, state) {
|
|
10
|
+
await Promise.resolve(this.store.set(key, state));
|
|
11
|
+
}
|
|
12
|
+
async load(key) {
|
|
13
|
+
return await Promise.resolve(this.store.get(key));
|
|
14
|
+
}
|
|
15
|
+
async delete(key) {
|
|
16
|
+
await Promise.resolve(this.store.delete(key));
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
class RunStateManager {
|
|
20
|
+
constructor(storage) {
|
|
21
|
+
this.store = storage && storage !== 'memory' ? storage : new MemoryCheckpointStore();
|
|
22
|
+
}
|
|
23
|
+
async suspend(runId, state) {
|
|
24
|
+
await this.store.save(runId, { ...state, status: 'suspended' });
|
|
25
|
+
}
|
|
26
|
+
async resume(runId) {
|
|
27
|
+
const state = await this.store.load(runId);
|
|
28
|
+
if (!state)
|
|
29
|
+
return undefined;
|
|
30
|
+
if (state.status !== 'suspended') {
|
|
31
|
+
throw new Error(`Run ${runId} is not suspended. Cannot resume.`);
|
|
32
|
+
}
|
|
33
|
+
const newState = { ...state, status: 'running' };
|
|
34
|
+
return newState;
|
|
35
|
+
}
|
|
36
|
+
async complete(runId) {
|
|
37
|
+
try {
|
|
38
|
+
await this.store.delete(runId);
|
|
39
|
+
}
|
|
40
|
+
catch (deleteError) {
|
|
41
|
+
console.error(`[RunStateManager] Failed to delete checkpoint ${runId}:`, deleteError);
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
exports.RunStateManager = RunStateManager;
|
|
46
|
+
function generateRunId() {
|
|
47
|
+
return `run_${crypto.randomUUID()}`;
|
|
48
|
+
}
|
|
49
|
+
//# sourceMappingURL=run-state.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"run-state.js","sourceRoot":"","sources":["../../src/runtime/run-state.ts"],"names":[],"mappings":";;;AAiEA,sCAEC;AAzDD,MAAM,qBAAqB;IAA3B;QACS,UAAK,GAAG,IAAI,GAAG,EAAkC,CAAC;IAa3D,CAAC;IAXA,KAAK,CAAC,IAAI,CAAC,GAAW,EAAE,KAA6B;QACpD,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,GAAW;QACrB,OAAO,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,GAAW;QACvB,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;IAC/C,CAAC;CACD;AAOD,MAAa,eAAe;IAG3B,YAAY,OAAoC;QAC/C,IAAI,CAAC,KAAK,GAAG,OAAO,IAAI,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,qBAAqB,EAAE,CAAC;IACtF,CAAC;IAGD,KAAK,CAAC,OAAO,CAAC,KAAa,EAAE,KAA6B;QACzD,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;IACjE,CAAC;IAGD,KAAK,CAAC,MAAM,CAAC,KAAa;QACzB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK;YAAE,OAAO,SAAS,CAAC;QAC7B,IAAI,KAAK,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,OAAO,KAAK,mCAAmC,CAAC,CAAC;QAClE,CAAC;QACD,MAAM,QAAQ,GAA2B,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QACzE,OAAO,QAAQ,CAAC;IACjB,CAAC;IAGD,KAAK,CAAC,QAAQ,CAAC,KAAa;QAC3B,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,WAAoB,EAAE,CAAC;YAC/B,OAAO,CAAC,KAAK,CAAC,iDAAiD,KAAK,GAAG,EAAE,WAAW,CAAC,CAAC;QACvF,CAAC;IACF,CAAC;CACD;AA/BD,0CA+BC;AAGD,SAAgB,aAAa;IAC5B,OAAO,OAAO,MAAM,CAAC,UAAU,EAAE,EAAE,CAAC;AACrC,CAAC"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import type { GenerateResult, StreamChunk, TokenUsage } from '../types';
|
|
2
|
+
import type { AgentDbMessage, AgentMessage, ContentToolResult } from '../types/sdk/message';
|
|
3
|
+
export declare function normalizeInput(input: AgentMessage[] | string): AgentDbMessage[];
|
|
4
|
+
export declare function makeToolResultMessage(toolCallId: string, toolName: string, result: unknown): AgentDbMessage;
|
|
5
|
+
export declare function makeErrorToolResultMessage(toolCallId: string, toolName: string, error: unknown): AgentDbMessage;
|
|
6
|
+
export declare function extractToolResults(messages: AgentDbMessage[]): ContentToolResult[];
|
|
7
|
+
export declare function makeErrorStream(error: unknown): ReadableStream<StreamChunk>;
|
|
8
|
+
export declare function mergeUsage(current: TokenUsage | undefined, next: TokenUsage | undefined): TokenUsage | undefined;
|
|
9
|
+
export declare function accumulateUsage(current: TokenUsage | undefined, raw: {
|
|
10
|
+
inputTokens?: number | undefined;
|
|
11
|
+
outputTokens?: number | undefined;
|
|
12
|
+
totalTokens?: number | undefined;
|
|
13
|
+
inputTokenDetails?: {
|
|
14
|
+
cacheReadTokens?: number;
|
|
15
|
+
cacheWriteTokens?: number;
|
|
16
|
+
};
|
|
17
|
+
outputTokenDetails?: {
|
|
18
|
+
reasoningTokens?: number;
|
|
19
|
+
};
|
|
20
|
+
} | undefined): TokenUsage | undefined;
|
|
21
|
+
export declare function applySubAgentUsage(result: GenerateResult): GenerateResult;
|
|
@@ -0,0 +1,96 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.normalizeInput = normalizeInput;
|
|
4
|
+
exports.makeToolResultMessage = makeToolResultMessage;
|
|
5
|
+
exports.makeErrorToolResultMessage = makeErrorToolResultMessage;
|
|
6
|
+
exports.extractToolResults = extractToolResults;
|
|
7
|
+
exports.makeErrorStream = makeErrorStream;
|
|
8
|
+
exports.mergeUsage = mergeUsage;
|
|
9
|
+
exports.accumulateUsage = accumulateUsage;
|
|
10
|
+
exports.applySubAgentUsage = applySubAgentUsage;
|
|
11
|
+
const message_1 = require("../sdk/message");
|
|
12
|
+
const stream_1 = require("./stream");
|
|
13
|
+
function normalizeInput(input) {
|
|
14
|
+
if (typeof input === 'string') {
|
|
15
|
+
return [(0, message_1.toDbMessage)({ role: 'user', content: [{ type: 'text', text: input }] })];
|
|
16
|
+
}
|
|
17
|
+
return input.map(message_1.toDbMessage);
|
|
18
|
+
}
|
|
19
|
+
function makeToolResultMessage(toolCallId, toolName, result) {
|
|
20
|
+
return (0, message_1.toDbMessage)({
|
|
21
|
+
role: 'tool',
|
|
22
|
+
content: [
|
|
23
|
+
{
|
|
24
|
+
type: 'tool-result',
|
|
25
|
+
toolCallId,
|
|
26
|
+
toolName,
|
|
27
|
+
result: result,
|
|
28
|
+
},
|
|
29
|
+
],
|
|
30
|
+
});
|
|
31
|
+
}
|
|
32
|
+
function makeErrorToolResultMessage(toolCallId, toolName, error) {
|
|
33
|
+
const message = error instanceof Error ? `${error.name}: ${error.message}` : String(error);
|
|
34
|
+
return (0, message_1.toDbMessage)({
|
|
35
|
+
role: 'tool',
|
|
36
|
+
content: [
|
|
37
|
+
{
|
|
38
|
+
type: 'tool-result',
|
|
39
|
+
toolCallId,
|
|
40
|
+
toolName,
|
|
41
|
+
result: { error: message },
|
|
42
|
+
isError: true,
|
|
43
|
+
},
|
|
44
|
+
],
|
|
45
|
+
});
|
|
46
|
+
}
|
|
47
|
+
function extractToolResults(messages) {
|
|
48
|
+
return messages
|
|
49
|
+
.flatMap((m) => ('content' in m ? m.content : []))
|
|
50
|
+
.filter((c) => c.type === 'tool-result');
|
|
51
|
+
}
|
|
52
|
+
function makeErrorStream(error) {
|
|
53
|
+
const { readable, writable } = new TransformStream();
|
|
54
|
+
const writer = writable.getWriter();
|
|
55
|
+
writer.write({ type: 'error', error }).catch(() => { });
|
|
56
|
+
writer.write({ type: 'finish', finishReason: 'error' }).catch(() => { });
|
|
57
|
+
writer.close().catch(() => { });
|
|
58
|
+
return readable;
|
|
59
|
+
}
|
|
60
|
+
function mergeUsage(current, next) {
|
|
61
|
+
if (!next)
|
|
62
|
+
return current;
|
|
63
|
+
if (!current)
|
|
64
|
+
return next;
|
|
65
|
+
const merged = {
|
|
66
|
+
promptTokens: current.promptTokens + next.promptTokens,
|
|
67
|
+
completionTokens: current.completionTokens + next.completionTokens,
|
|
68
|
+
totalTokens: current.totalTokens + next.totalTokens,
|
|
69
|
+
};
|
|
70
|
+
const cacheRead = (current.inputTokenDetails?.cacheRead ?? 0) + (next.inputTokenDetails?.cacheRead ?? 0);
|
|
71
|
+
const cacheWrite = (current.inputTokenDetails?.cacheWrite ?? 0) + (next.inputTokenDetails?.cacheWrite ?? 0);
|
|
72
|
+
if (cacheRead > 0 || cacheWrite > 0) {
|
|
73
|
+
merged.inputTokenDetails = {
|
|
74
|
+
...(cacheRead > 0 && { cacheRead }),
|
|
75
|
+
...(cacheWrite > 0 && { cacheWrite }),
|
|
76
|
+
};
|
|
77
|
+
}
|
|
78
|
+
const reasoning = (current.outputTokenDetails?.reasoning ?? 0) + (next.outputTokenDetails?.reasoning ?? 0);
|
|
79
|
+
if (reasoning > 0) {
|
|
80
|
+
merged.outputTokenDetails = { reasoning };
|
|
81
|
+
}
|
|
82
|
+
return merged;
|
|
83
|
+
}
|
|
84
|
+
function accumulateUsage(current, raw) {
|
|
85
|
+
if (!raw)
|
|
86
|
+
return current;
|
|
87
|
+
return mergeUsage(current, (0, stream_1.toTokenUsage)(raw));
|
|
88
|
+
}
|
|
89
|
+
function applySubAgentUsage(result) {
|
|
90
|
+
if (!result.subAgentUsage || result.subAgentUsage.length === 0)
|
|
91
|
+
return result;
|
|
92
|
+
const parentCost = result.usage?.cost ?? 0;
|
|
93
|
+
const subCost = result.subAgentUsage.reduce((sum, s) => sum + (s.usage.cost ?? 0), 0);
|
|
94
|
+
return { ...result, totalCost: parentCost + subCost };
|
|
95
|
+
}
|
|
96
|
+
//# sourceMappingURL=runtime-helpers.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runtime-helpers.js","sourceRoot":"","sources":["../../src/runtime/runtime-helpers.ts"],"names":[],"mappings":";;AAWA,wCAKC;AAGD,sDAgBC;AAOD,gEAkBC;AAGD,gDAIC;AAOD,0CAOC;AAGD,gCA8BC;AAMD,0CAcC;AAGD,gDAOC;AA5ID,4CAA6C;AAE7C,qCAAwC;AAKxC,SAAgB,cAAc,CAAC,KAA8B;IAC5D,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,CAAC,IAAA,qBAAW,EAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAClF,CAAC;IACD,OAAO,KAAK,CAAC,GAAG,CAAC,qBAAW,CAAC,CAAC;AAC/B,CAAC;AAGD,SAAgB,qBAAqB,CACpC,UAAkB,EAClB,QAAgB,EAChB,MAAe;IAEf,OAAO,IAAA,qBAAW,EAAC;QAClB,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE;YACR;gBACC,IAAI,EAAE,aAAa;gBACnB,UAAU;gBACV,QAAQ;gBACR,MAAM,EAAE,MAAmB;aAC3B;SACD;KACD,CAAC,CAAC;AACJ,CAAC;AAOD,SAAgB,0BAA0B,CACzC,UAAkB,EAClB,QAAgB,EAChB,KAAc;IAEd,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC3F,OAAO,IAAA,qBAAW,EAAC;QAClB,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE;YACR;gBACC,IAAI,EAAE,aAAa;gBACnB,UAAU;gBACV,QAAQ;gBACR,MAAM,EAAE,EAAE,KAAK,EAAE,OAAO,EAAe;gBACvC,OAAO,EAAE,IAAI;aACb;SACD;KACD,CAAC,CAAC;AACJ,CAAC;AAGD,SAAgB,kBAAkB,CAAC,QAA0B;IAC5D,OAAO,QAAQ;SACb,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;SACjD,MAAM,CAAC,CAAC,CAAC,EAA0B,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC;AACnE,CAAC;AAOD,SAAgB,eAAe,CAAC,KAAc;IAC7C,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,IAAI,eAAe,EAA4B,CAAC;IAC/E,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,EAAE,CAAC;IACpC,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACvD,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACxE,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAC/B,OAAO,QAAQ,CAAC;AACjB,CAAC;AAGD,SAAgB,UAAU,CACzB,OAA+B,EAC/B,IAA4B;IAE5B,IAAI,CAAC,IAAI;QAAE,OAAO,OAAO,CAAC;IAC1B,IAAI,CAAC,OAAO;QAAE,OAAO,IAAI,CAAC;IAC1B,MAAM,MAAM,GAAe;QAC1B,YAAY,EAAE,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY;QACtD,gBAAgB,EAAE,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB;QAClE,WAAW,EAAE,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW;KACnD,CAAC;IAEF,MAAM,SAAS,GACd,CAAC,OAAO,CAAC,iBAAiB,EAAE,SAAS,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,iBAAiB,EAAE,SAAS,IAAI,CAAC,CAAC,CAAC;IACxF,MAAM,UAAU,GACf,CAAC,OAAO,CAAC,iBAAiB,EAAE,UAAU,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,iBAAiB,EAAE,UAAU,IAAI,CAAC,CAAC,CAAC;IAC1F,IAAI,SAAS,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;QACrC,MAAM,CAAC,iBAAiB,GAAG;YAC1B,GAAG,CAAC,SAAS,GAAG,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YACnC,GAAG,CAAC,UAAU,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,CAAC;SACrC,CAAC;IACH,CAAC;IAED,MAAM,SAAS,GACd,CAAC,OAAO,CAAC,kBAAkB,EAAE,SAAS,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,kBAAkB,EAAE,SAAS,IAAI,CAAC,CAAC,CAAC;IAC1F,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;QACnB,MAAM,CAAC,kBAAkB,GAAG,EAAE,SAAS,EAAE,CAAC;IAC3C,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC;AAMD,SAAgB,eAAe,CAC9B,OAA+B,EAC/B,GAQY;IAEZ,IAAI,CAAC,GAAG;QAAE,OAAO,OAAO,CAAC;IACzB,OAAO,UAAU,CAAC,OAAO,EAAE,IAAA,qBAAY,EAAC,GAAG,CAAC,CAAC,CAAC;AAC/C,CAAC;AAGD,SAAgB,kBAAkB,CAAC,MAAsB;IACxD,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,aAAa,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,MAAM,CAAC;IAE9E,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,EAAE,IAAI,IAAI,CAAC,CAAC;IAC3C,MAAM,OAAO,GAAG,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAEtF,OAAO,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,UAAU,GAAG,OAAO,EAAE,CAAC;AACvD,CAAC"}
|