neoctl 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/README.md +146 -0
- package/dist/agents/agent-definition.d.ts +39 -0
- package/dist/agents/agent-definition.js +57 -0
- package/dist/agents/agent-definition.js.map +1 -0
- package/dist/agents/agent-tool.d.ts +38 -0
- package/dist/agents/agent-tool.js +336 -0
- package/dist/agents/agent-tool.js.map +1 -0
- package/dist/agents/local-agent-task.d.ts +52 -0
- package/dist/agents/local-agent-task.js +63 -0
- package/dist/agents/local-agent-task.js.map +1 -0
- package/dist/agents/smoke-agents.d.ts +1 -0
- package/dist/agents/smoke-agents.js +142 -0
- package/dist/agents/smoke-agents.js.map +1 -0
- package/dist/agents/team.d.ts +10 -0
- package/dist/agents/team.js +2 -0
- package/dist/agents/team.js.map +1 -0
- package/dist/app/app-state.d.ts +21 -0
- package/dist/app/app-state.js +24 -0
- package/dist/app/app-state.js.map +1 -0
- package/dist/context/compaction.d.ts +49 -0
- package/dist/context/compaction.js +334 -0
- package/dist/context/compaction.js.map +1 -0
- package/dist/context/context-manager.d.ts +53 -0
- package/dist/context/context-manager.js +98 -0
- package/dist/context/context-manager.js.map +1 -0
- package/dist/context/prompts.d.ts +22 -0
- package/dist/context/prompts.js +84 -0
- package/dist/context/prompts.js.map +1 -0
- package/dist/context/smoke-context.d.ts +1 -0
- package/dist/context/smoke-context.js +151 -0
- package/dist/context/smoke-context.js.map +1 -0
- package/dist/core/assistant-output-filter.d.ts +9 -0
- package/dist/core/assistant-output-filter.js +78 -0
- package/dist/core/assistant-output-filter.js.map +1 -0
- package/dist/core/context-metrics.d.ts +10 -0
- package/dist/core/context-metrics.js +77 -0
- package/dist/core/context-metrics.js.map +1 -0
- package/dist/core/message-pipeline.d.ts +10 -0
- package/dist/core/message-pipeline.js +138 -0
- package/dist/core/message-pipeline.js.map +1 -0
- package/dist/core/query-engine.d.ts +86 -0
- package/dist/core/query-engine.js +337 -0
- package/dist/core/query-engine.js.map +1 -0
- package/dist/core/query.d.ts +47 -0
- package/dist/core/query.js +408 -0
- package/dist/core/query.js.map +1 -0
- package/dist/core/run-agent.d.ts +48 -0
- package/dist/core/run-agent.js +150 -0
- package/dist/core/run-agent.js.map +1 -0
- package/dist/core/smoke-core-loop.d.ts +1 -0
- package/dist/core/smoke-core-loop.js +42 -0
- package/dist/core/smoke-core-loop.js.map +1 -0
- package/dist/core/state.d.ts +37 -0
- package/dist/core/state.js +31 -0
- package/dist/core/state.js.map +1 -0
- package/dist/index.d.ts +42 -0
- package/dist/index.js +43 -0
- package/dist/index.js.map +1 -0
- package/dist/model/communication-logger.d.ts +30 -0
- package/dist/model/communication-logger.js +218 -0
- package/dist/model/communication-logger.js.map +1 -0
- package/dist/model/config.d.ts +25 -0
- package/dist/model/config.js +65 -0
- package/dist/model/config.js.map +1 -0
- package/dist/model/context-window.d.ts +32 -0
- package/dist/model/context-window.js +76 -0
- package/dist/model/context-window.js.map +1 -0
- package/dist/model/credentials.d.ts +13 -0
- package/dist/model/credentials.js +19 -0
- package/dist/model/credentials.js.map +1 -0
- package/dist/model/env.d.ts +8 -0
- package/dist/model/env.js +36 -0
- package/dist/model/env.js.map +1 -0
- package/dist/model/errors.d.ts +42 -0
- package/dist/model/errors.js +74 -0
- package/dist/model/errors.js.map +1 -0
- package/dist/model/http-transport.d.ts +26 -0
- package/dist/model/http-transport.js +118 -0
- package/dist/model/http-transport.js.map +1 -0
- package/dist/model/model-gateway.d.ts +103 -0
- package/dist/model/model-gateway.js +15 -0
- package/dist/model/model-gateway.js.map +1 -0
- package/dist/model/model-metadata.json +677 -0
- package/dist/model/openai-adapter.d.ts +34 -0
- package/dist/model/openai-adapter.js +152 -0
- package/dist/model/openai-adapter.js.map +1 -0
- package/dist/model/openai-chat-mapper.d.ts +11 -0
- package/dist/model/openai-chat-mapper.js +114 -0
- package/dist/model/openai-chat-mapper.js.map +1 -0
- package/dist/model/openai-mappers.d.ts +21 -0
- package/dist/model/openai-mappers.js +261 -0
- package/dist/model/openai-mappers.js.map +1 -0
- package/dist/model/openai-responses-adapter.d.ts +2 -0
- package/dist/model/openai-responses-adapter.js +2 -0
- package/dist/model/openai-responses-adapter.js.map +1 -0
- package/dist/model/openai-responses-mapper.d.ts +11 -0
- package/dist/model/openai-responses-mapper.js +218 -0
- package/dist/model/openai-responses-mapper.js.map +1 -0
- package/dist/model/provider-adapter.d.ts +17 -0
- package/dist/model/provider-adapter.js +2 -0
- package/dist/model/provider-adapter.js.map +1 -0
- package/dist/model/provider-factory.d.ts +4 -0
- package/dist/model/provider-factory.js +29 -0
- package/dist/model/provider-factory.js.map +1 -0
- package/dist/model/retry-runner.d.ts +8 -0
- package/dist/model/retry-runner.js +28 -0
- package/dist/model/retry-runner.js.map +1 -0
- package/dist/model/smoke-openai.d.ts +1 -0
- package/dist/model/smoke-openai.js +44 -0
- package/dist/model/smoke-openai.js.map +1 -0
- package/dist/model/smoke-responses-mapper.d.ts +1 -0
- package/dist/model/smoke-responses-mapper.js +72 -0
- package/dist/model/smoke-responses-mapper.js.map +1 -0
- package/dist/model/sse-decoder.d.ts +5 -0
- package/dist/model/sse-decoder.js +79 -0
- package/dist/model/sse-decoder.js.map +1 -0
- package/dist/repl/clipboard.d.ts +14 -0
- package/dist/repl/clipboard.js +92 -0
- package/dist/repl/clipboard.js.map +1 -0
- package/dist/repl/commands.d.ts +38 -0
- package/dist/repl/commands.js +86 -0
- package/dist/repl/commands.js.map +1 -0
- package/dist/repl/index.d.ts +2 -0
- package/dist/repl/index.js +2836 -0
- package/dist/repl/index.js.map +1 -0
- package/dist/repl/markdown-renderer.d.ts +81 -0
- package/dist/repl/markdown-renderer.js +546 -0
- package/dist/repl/markdown-renderer.js.map +1 -0
- package/dist/repl/render.d.ts +2 -0
- package/dist/repl/render.js +45 -0
- package/dist/repl/render.js.map +1 -0
- package/dist/repl/status-line.d.ts +19 -0
- package/dist/repl/status-line.js +160 -0
- package/dist/repl/status-line.js.map +1 -0
- package/dist/safety/audit.d.ts +10 -0
- package/dist/safety/audit.js +2 -0
- package/dist/safety/audit.js.map +1 -0
- package/dist/safety/permissions.d.ts +10 -0
- package/dist/safety/permissions.js +2 -0
- package/dist/safety/permissions.js.map +1 -0
- package/dist/safety/sandbox.d.ts +6 -0
- package/dist/safety/sandbox.js +2 -0
- package/dist/safety/sandbox.js.map +1 -0
- package/dist/session/session-store.d.ts +98 -0
- package/dist/session/session-store.js +249 -0
- package/dist/session/session-store.js.map +1 -0
- package/dist/session/smoke-session.d.ts +1 -0
- package/dist/session/smoke-session.js +85 -0
- package/dist/session/smoke-session.js.map +1 -0
- package/dist/session/tool-result-memory.d.ts +64 -0
- package/dist/session/tool-result-memory.js +303 -0
- package/dist/session/tool-result-memory.js.map +1 -0
- package/dist/skills/skill-tool.d.ts +28 -0
- package/dist/skills/skill-tool.js +107 -0
- package/dist/skills/skill-tool.js.map +1 -0
- package/dist/skills/smoke-skills.d.ts +1 -0
- package/dist/skills/smoke-skills.js +60 -0
- package/dist/skills/smoke-skills.js.map +1 -0
- package/dist/tasks/task-store.d.ts +43 -0
- package/dist/tasks/task-store.js +150 -0
- package/dist/tasks/task-store.js.map +1 -0
- package/dist/tasks/task-tools.d.ts +31 -0
- package/dist/tasks/task-tools.js +232 -0
- package/dist/tasks/task-tools.js.map +1 -0
- package/dist/tools/builtins/echo-tool.d.ts +4 -0
- package/dist/tools/builtins/echo-tool.js +26 -0
- package/dist/tools/builtins/echo-tool.js.map +1 -0
- package/dist/tools/builtins/edit-tool.d.ts +30 -0
- package/dist/tools/builtins/edit-tool.js +423 -0
- package/dist/tools/builtins/edit-tool.js.map +1 -0
- package/dist/tools/builtins/exec-tool.d.ts +18 -0
- package/dist/tools/builtins/exec-tool.js +290 -0
- package/dist/tools/builtins/exec-tool.js.map +1 -0
- package/dist/tools/builtins/filesystem-tools.d.ts +16 -0
- package/dist/tools/builtins/filesystem-tools.js +306 -0
- package/dist/tools/builtins/filesystem-tools.js.map +1 -0
- package/dist/tools/builtins/grep-tool.d.ts +56 -0
- package/dist/tools/builtins/grep-tool.js +320 -0
- package/dist/tools/builtins/grep-tool.js.map +1 -0
- package/dist/tools/builtins/ripgrep-binary.d.ts +6 -0
- package/dist/tools/builtins/ripgrep-binary.js +51 -0
- package/dist/tools/builtins/ripgrep-binary.js.map +1 -0
- package/dist/tools/builtins/search-providers.d.ts +49 -0
- package/dist/tools/builtins/search-providers.js +240 -0
- package/dist/tools/builtins/search-providers.js.map +1 -0
- package/dist/tools/builtins/search-tool.d.ts +18 -0
- package/dist/tools/builtins/search-tool.js +171 -0
- package/dist/tools/builtins/search-tool.js.map +1 -0
- package/dist/tools/registry.d.ts +18 -0
- package/dist/tools/registry.js +69 -0
- package/dist/tools/registry.js.map +1 -0
- package/dist/tools/run-tool-use.d.ts +13 -0
- package/dist/tools/run-tool-use.js +113 -0
- package/dist/tools/run-tool-use.js.map +1 -0
- package/dist/tools/schema.d.ts +2 -0
- package/dist/tools/schema.js +61 -0
- package/dist/tools/schema.js.map +1 -0
- package/dist/tools/smoke-tool-system.d.ts +1 -0
- package/dist/tools/smoke-tool-system.js +204 -0
- package/dist/tools/smoke-tool-system.js.map +1 -0
- package/dist/tools/streaming-tool-executor.d.ts +19 -0
- package/dist/tools/streaming-tool-executor.js +89 -0
- package/dist/tools/streaming-tool-executor.js.map +1 -0
- package/dist/tools/tool-orchestration.d.ts +15 -0
- package/dist/tools/tool-orchestration.js +89 -0
- package/dist/tools/tool-orchestration.js.map +1 -0
- package/dist/tools/tool.d.ts +121 -0
- package/dist/tools/tool.js +4 -0
- package/dist/tools/tool.js.map +1 -0
- package/dist/types/events.d.ts +66 -0
- package/dist/types/events.js +2 -0
- package/dist/types/events.js.map +1 -0
- package/dist/types/messages.d.ts +57 -0
- package/dist/types/messages.js +42 -0
- package/dist/types/messages.js.map +1 -0
- package/package.json +49 -0
- package/scripts/copy-model-metadata.mjs +4 -0
- package/scripts/install-ripgrep.cjs +152 -0
|
@@ -0,0 +1,113 @@
|
|
|
1
|
+
import { createTextMessage, createToolResultMessage } from "../types/messages.js";
|
|
2
|
+
import { validateJsonSchema } from "./schema.js";
|
|
3
|
+
export async function runToolUse(request, context, options = {}) {
|
|
4
|
+
const tool = context.tools.get(request.name) ?? context.tools.getByAlias?.(request.name);
|
|
5
|
+
if (!tool) {
|
|
6
|
+
return [{ message: createToolResultMessage(request, false, { error: `Unknown tool: ${request.name}` }) }];
|
|
7
|
+
}
|
|
8
|
+
if (context.abortSignal?.aborted) {
|
|
9
|
+
return [{ message: createToolResultMessage(request, false, { error: "Tool use aborted before execution" }) }];
|
|
10
|
+
}
|
|
11
|
+
const updates = [];
|
|
12
|
+
const contextWithToolUseId = { ...context, toolUseId: request.id };
|
|
13
|
+
const onProgress = (event) => {
|
|
14
|
+
const progress = { ...event, toolUseId: event.toolUseId ?? request.id };
|
|
15
|
+
context.emit(progress);
|
|
16
|
+
const rendered = tool.renderToolProgressMessage?.(progress) ?? createTextMessage("progress", progress.message);
|
|
17
|
+
updates.push({ message: rendered, progress });
|
|
18
|
+
};
|
|
19
|
+
try {
|
|
20
|
+
const parsed = parseAndValidateInput(tool, request.input, contextWithToolUseId);
|
|
21
|
+
if (!parsed.ok)
|
|
22
|
+
return [{ message: createToolResultMessage(request, false, { error: parsed.message }) }];
|
|
23
|
+
const customValidation = tool.validateInput
|
|
24
|
+
? await tool.validateInput(parsed.value, contextWithToolUseId)
|
|
25
|
+
: { ok: true, value: parsed.value };
|
|
26
|
+
if (!customValidation.ok) {
|
|
27
|
+
return [{ message: createToolResultMessage(request, false, { error: customValidation.message }) }];
|
|
28
|
+
}
|
|
29
|
+
const decision = await resolvePermissionDecision(request, contextWithToolUseId, options.canUseTool);
|
|
30
|
+
if (!decision.allowed) {
|
|
31
|
+
return [{ message: createToolResultMessage(request, false, { error: decision.reason ?? "Tool use denied" }) }];
|
|
32
|
+
}
|
|
33
|
+
onProgress({ toolName: tool.name, message: `Running ${tool.name}` });
|
|
34
|
+
const result = await callTool(tool, customValidation.value, contextWithToolUseId, {
|
|
35
|
+
canUseTool: options.canUseTool,
|
|
36
|
+
parentMessage: options.parentMessage,
|
|
37
|
+
onProgress,
|
|
38
|
+
});
|
|
39
|
+
const output = await processToolOutput(tool, request, result, contextWithToolUseId);
|
|
40
|
+
const resultMessage = tool.renderToolResultMessage?.(result) ?? createToolResultMessage(request, result.ok, output);
|
|
41
|
+
updates.push({ message: resultMessage, context: result.contextModifier?.(contextWithToolUseId) });
|
|
42
|
+
for (const message of result.newMessages ?? []) {
|
|
43
|
+
updates.push({ message });
|
|
44
|
+
}
|
|
45
|
+
return updates;
|
|
46
|
+
}
|
|
47
|
+
catch (error) {
|
|
48
|
+
return [
|
|
49
|
+
...updates,
|
|
50
|
+
{
|
|
51
|
+
message: createToolResultMessage(request, false, {
|
|
52
|
+
error: error instanceof Error ? error.message : String(error),
|
|
53
|
+
}),
|
|
54
|
+
},
|
|
55
|
+
];
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
export async function runToolUseToMessages(request, context, options = {}) {
|
|
59
|
+
return (await runToolUse(request, context, options)).map((update) => update.message);
|
|
60
|
+
}
|
|
61
|
+
function parseAndValidateInput(tool, input, context) {
|
|
62
|
+
const schemaResult = validateJsonSchema(input, tool.inputSchema);
|
|
63
|
+
if (!schemaResult.ok) {
|
|
64
|
+
const hint = tool.metadata.shouldDefer
|
|
65
|
+
? ` ${buildSchemaNotSentHint(tool.name)}`
|
|
66
|
+
: "";
|
|
67
|
+
return { ok: false, message: `${schemaResult.message}.${hint}`.trim() };
|
|
68
|
+
}
|
|
69
|
+
try {
|
|
70
|
+
const value = tool.validate ? tool.validate(schemaResult.value, context) : schemaResult.value;
|
|
71
|
+
return { ok: true, value };
|
|
72
|
+
}
|
|
73
|
+
catch (error) {
|
|
74
|
+
return { ok: false, message: error instanceof Error ? error.message : String(error) };
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
async function resolvePermissionDecision(request, context, canUseTool) {
|
|
78
|
+
if (!canUseTool)
|
|
79
|
+
return { allowed: true };
|
|
80
|
+
const decision = await canUseTool(request, context);
|
|
81
|
+
if (typeof decision === "boolean")
|
|
82
|
+
return { allowed: decision };
|
|
83
|
+
return decision;
|
|
84
|
+
}
|
|
85
|
+
async function callTool(tool, input, context, options) {
|
|
86
|
+
if (tool.call) {
|
|
87
|
+
return tool.call(input, context, options);
|
|
88
|
+
}
|
|
89
|
+
if (tool.execute) {
|
|
90
|
+
return tool.execute(input, context);
|
|
91
|
+
}
|
|
92
|
+
return { ok: false, output: { error: `Tool ${tool.name} has no call implementation` } };
|
|
93
|
+
}
|
|
94
|
+
async function processToolOutput(tool, request, result, context) {
|
|
95
|
+
const mapped = tool.mapResult ? tool.mapResult(result, request) : result.output;
|
|
96
|
+
if (context.toolResultMemory) {
|
|
97
|
+
const processed = await context.toolResultMemory.processToolResult(request.id, mapped, tool.metadata.maxResultSizeChars);
|
|
98
|
+
if (processed.record)
|
|
99
|
+
context.recordContentReplacements?.([processed.record]);
|
|
100
|
+
return processed.output;
|
|
101
|
+
}
|
|
102
|
+
const maxSize = tool.metadata.maxResultSizeChars;
|
|
103
|
+
if (!maxSize)
|
|
104
|
+
return mapped;
|
|
105
|
+
const serialized = typeof mapped === "string" ? mapped : JSON.stringify(mapped);
|
|
106
|
+
if (serialized.length <= maxSize)
|
|
107
|
+
return mapped;
|
|
108
|
+
return `[Tool result truncated by transport: original ${serialized.length} chars, showing first ${maxSize} chars${result.summary ? `, summary: ${result.summary}` : ""}]\n${serialized.slice(0, maxSize)}`;
|
|
109
|
+
}
|
|
110
|
+
function buildSchemaNotSentHint(toolName) {
|
|
111
|
+
return `If ${toolName} is deferred, select it with ToolSearch before retrying.`;
|
|
112
|
+
}
|
|
113
|
+
//# sourceMappingURL=run-tool-use.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"run-tool-use.js","sourceRoot":"","sources":["../../src/tools/run-tool-use.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,uBAAuB,EAAqC,MAAM,sBAAsB,CAAC;AACrH,OAAO,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAqBjD,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAAuB,EACvB,OAAuB,EACvB,UAA6B,EAAE;IAE/B,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACzF,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,CAAC,EAAE,OAAO,EAAE,uBAAuB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,iBAAiB,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;IAC5G,CAAC;IAED,IAAI,OAAO,CAAC,WAAW,EAAE,OAAO,EAAE,CAAC;QACjC,OAAO,CAAC,EAAE,OAAO,EAAE,uBAAuB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,mCAAmC,EAAE,CAAC,EAAE,CAAC,CAAC;IAChH,CAAC;IAED,MAAM,OAAO,GAAwB,EAAE,CAAC;IACxC,MAAM,oBAAoB,GAAG,EAAE,GAAG,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC;IACnE,MAAM,UAAU,GAAG,CAAC,KAAwB,EAAE,EAAE;QAC9C,MAAM,QAAQ,GAAG,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,OAAO,CAAC,EAAE,EAAE,CAAC;QACxE,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvB,MAAM,QAAQ,GAAG,IAAI,CAAC,yBAAyB,EAAE,CAAC,QAAQ,CAAC,IAAI,iBAAiB,CAAC,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC/G,OAAO,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;IAChD,CAAC,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,qBAAqB,CAAC,IAAI,EAAE,OAAO,CAAC,KAAK,EAAE,oBAAoB,CAAC,CAAC;QAChF,IAAI,CAAC,MAAM,CAAC,EAAE;YAAE,OAAO,CAAC,EAAE,OAAO,EAAE,uBAAuB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;QAEzG,MAAM,gBAAgB,GAAG,IAAI,CAAC,aAAa;YACzC,CAAC,CAAC,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,EAAE,oBAAoB,CAAC;YAC9D,CAAC,CAAC,EAAE,EAAE,EAAE,IAAa,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;QAC/C,IAAI,CAAC,gBAAgB,CAAC,EAAE,EAAE,CAAC;YACzB,OAAO,CAAC,EAAE,OAAO,EAAE,uBAAuB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,gBAAgB,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;QACrG,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,yBAAyB,CAAC,OAAO,EAAE,oBAAoB,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;QACpG,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACtB,OAAO,CAAC,EAAE,OAAO,EAAE,uBAAuB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,MAAM,IAAI,iBAAiB,EAAE,CAAC,EAAE,CAAC,CAAC;QACjH,CAAC;QAED,UAAU,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,WAAW,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACrE,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,gBAAgB,CAAC,KAAK,EAAE,oBAAoB,EAAE;YAChF,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,aAAa,EAAE,OAAO,CAAC,aAAa;YACpC,UAAU;SACX,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,oBAAoB,CAAC,CAAC;QACpF,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC,MAAM,CAAC,IAAI,uBAAuB,CAAC,OAAO,EAAE,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QACpH,OAAO,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,CAAC,eAAe,EAAE,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC;QAElG,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,WAAW,IAAI,EAAE,EAAE,CAAC;YAC/C,OAAO,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QAC5B,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,GAAG,OAAO;YACV;gBACE,OAAO,EAAE,uBAAuB,CAAC,OAAO,EAAE,KAAK,EAAE;oBAC/C,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC;aACH;SACF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,OAAuB,EACvB,OAAuB,EACvB,UAA6B,EAAE;IAE/B,OAAO,CAAC,MAAM,UAAU,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AACvF,CAAC;AAED,SAAS,qBAAqB,CAC5B,IAAkB,EAClB,KAAc,EACd,OAAuB;IAEvB,MAAM,YAAY,GAAG,kBAAkB,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IACjE,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,CAAC;QACrB,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,WAAW;YACpC,CAAC,CAAC,IAAI,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;YACzC,CAAC,CAAC,EAAE,CAAC;QACP,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,YAAY,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;IAC1E,CAAC;IAED,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,CAAE,YAAY,CAAC,KAAgB,CAAC;QAC1G,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;IAC7B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;IACxF,CAAC;AACH,CAAC;AAED,KAAK,UAAU,yBAAyB,CACtC,OAAuB,EACvB,OAAuB,EACvB,UAAuB;IAEvB,IAAI,CAAC,UAAU;QAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC1C,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACpD,IAAI,OAAO,QAAQ,KAAK,SAAS;QAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;IAChE,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,KAAK,UAAU,QAAQ,CACrB,IAAkB,EAClB,KAAa,EACb,OAAuB,EACvB,OAIC;IAED,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAC5C,CAAC;IACD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACtC,CAAC;IACD,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,QAAQ,IAAI,CAAC,IAAI,6BAA6B,EAAE,EAAE,CAAC;AAC1F,CAAC;AAED,KAAK,UAAU,iBAAiB,CAAC,IAAU,EAAE,OAAuB,EAAE,MAAkB,EAAE,OAAuB;IAC/G,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;IAChF,IAAI,OAAO,CAAC,gBAAgB,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,gBAAgB,CAAC,iBAAiB,CAChE,OAAO,CAAC,EAAE,EACV,MAAM,EACN,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CACjC,CAAC;QACF,IAAI,SAAS,CAAC,MAAM;YAAE,OAAO,CAAC,yBAAyB,EAAE,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;QAC9E,OAAO,SAAS,CAAC,MAAM,CAAC;IAC1B,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC;IACjD,IAAI,CAAC,OAAO;QAAE,OAAO,MAAM,CAAC;IAE5B,MAAM,UAAU,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChF,IAAI,UAAU,CAAC,MAAM,IAAI,OAAO;QAAE,OAAO,MAAM,CAAC;IAEhD,OAAO,iDAAiD,UAAU,CAAC,MAAM,yBAAyB,OAAO,SAAS,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,CAAC;AAC7M,CAAC;AAED,SAAS,sBAAsB,CAAC,QAAgB;IAC9C,OAAO,MAAM,QAAQ,0DAA0D,CAAC;AAClF,CAAC"}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
export function validateJsonSchema(input, schema, path = "input") {
|
|
2
|
+
if (!schema || Object.keys(schema).length === 0)
|
|
3
|
+
return { ok: true, value: input };
|
|
4
|
+
if (schema.enum && !schema.enum.includes(input)) {
|
|
5
|
+
return { ok: false, message: `${path} must be one of ${schema.enum.map(String).join(", ")}` };
|
|
6
|
+
}
|
|
7
|
+
if (schema.type) {
|
|
8
|
+
const typeResult = validateType(input, schema, path);
|
|
9
|
+
if (!typeResult.ok)
|
|
10
|
+
return typeResult;
|
|
11
|
+
}
|
|
12
|
+
if (schema.type === "object" && schema.properties) {
|
|
13
|
+
const record = input;
|
|
14
|
+
for (const required of schema.required ?? []) {
|
|
15
|
+
if (!(required in record))
|
|
16
|
+
return { ok: false, message: `${path}.${required} is required` };
|
|
17
|
+
}
|
|
18
|
+
for (const [key, value] of Object.entries(record)) {
|
|
19
|
+
const childSchema = schema.properties[key];
|
|
20
|
+
if (!childSchema) {
|
|
21
|
+
if (schema.additionalProperties === false)
|
|
22
|
+
return { ok: false, message: `${path}.${key} is not allowed` };
|
|
23
|
+
continue;
|
|
24
|
+
}
|
|
25
|
+
const childResult = validateJsonSchema(value, childSchema, `${path}.${key}`);
|
|
26
|
+
if (!childResult.ok)
|
|
27
|
+
return childResult;
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
if (schema.type === "array" && schema.items && Array.isArray(input)) {
|
|
31
|
+
for (let index = 0; index < input.length; index += 1) {
|
|
32
|
+
const childResult = validateJsonSchema(input[index], schema.items, `${path}[${index}]`);
|
|
33
|
+
if (!childResult.ok)
|
|
34
|
+
return childResult;
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
return { ok: true, value: input };
|
|
38
|
+
}
|
|
39
|
+
function validateType(input, schema, path) {
|
|
40
|
+
switch (schema.type) {
|
|
41
|
+
case "object":
|
|
42
|
+
return input !== null && typeof input === "object" && !Array.isArray(input)
|
|
43
|
+
? { ok: true, value: input }
|
|
44
|
+
: { ok: false, message: `${path} must be an object` };
|
|
45
|
+
case "array":
|
|
46
|
+
return Array.isArray(input) ? { ok: true, value: input } : { ok: false, message: `${path} must be an array` };
|
|
47
|
+
case "string":
|
|
48
|
+
return typeof input === "string" ? { ok: true, value: input } : { ok: false, message: `${path} must be a string` };
|
|
49
|
+
case "number":
|
|
50
|
+
return typeof input === "number" && Number.isFinite(input) ? { ok: true, value: input } : { ok: false, message: `${path} must be a number` };
|
|
51
|
+
case "integer":
|
|
52
|
+
return Number.isInteger(input) ? { ok: true, value: input } : { ok: false, message: `${path} must be an integer` };
|
|
53
|
+
case "boolean":
|
|
54
|
+
return typeof input === "boolean" ? { ok: true, value: input } : { ok: false, message: `${path} must be a boolean` };
|
|
55
|
+
case "null":
|
|
56
|
+
return input === null ? { ok: true, value: input } : { ok: false, message: `${path} must be null` };
|
|
57
|
+
default:
|
|
58
|
+
return { ok: true, value: input };
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
//# sourceMappingURL=schema.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../src/tools/schema.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,kBAAkB,CAAC,KAAc,EAAE,MAAkB,EAAE,IAAI,GAAG,OAAO;IACnF,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IAEnF,IAAI,MAAM,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QAChD,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,mBAAmB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;IAChG,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QAChB,MAAM,UAAU,GAAG,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QACrD,IAAI,CAAC,UAAU,CAAC,EAAE;YAAE,OAAO,UAAU,CAAC;IACxC,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QAClD,MAAM,MAAM,GAAG,KAAgC,CAAC;QAChD,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,QAAQ,IAAI,EAAE,EAAE,CAAC;YAC7C,IAAI,CAAC,CAAC,QAAQ,IAAI,MAAM,CAAC;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,IAAI,QAAQ,cAAc,EAAE,CAAC;QAC9F,CAAC;QACD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAClD,MAAM,WAAW,GAAG,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YAC3C,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,IAAI,MAAM,CAAC,oBAAoB,KAAK,KAAK;oBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,IAAI,GAAG,iBAAiB,EAAE,CAAC;gBAC1G,SAAS;YACX,CAAC;YACD,MAAM,WAAW,GAAG,kBAAkB,CAAC,KAAK,EAAE,WAAW,EAAE,GAAG,IAAI,IAAI,GAAG,EAAE,CAAC,CAAC;YAC7E,IAAI,CAAC,WAAW,CAAC,EAAE;gBAAE,OAAO,WAAW,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,IAAI,MAAM,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACpE,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YACrD,MAAM,WAAW,GAAG,kBAAkB,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,KAAK,EAAE,GAAG,IAAI,IAAI,KAAK,GAAG,CAAC,CAAC;YACxF,IAAI,CAAC,WAAW,CAAC,EAAE;gBAAE,OAAO,WAAW,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;AACpC,CAAC;AAED,SAAS,YAAY,CAAC,KAAc,EAAE,MAAkB,EAAE,IAAY;IACpE,QAAQ,MAAM,CAAC,IAAI,EAAE,CAAC;QACpB,KAAK,QAAQ;YACX,OAAO,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;gBACzE,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE;gBAC5B,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC1D,KAAK,OAAO;YACV,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;QAChH,KAAK,QAAQ;YACX,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;QACrH,KAAK,QAAQ;YACX,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,mBAAmB,EAAE,CAAC;QAC/I,KAAK,SAAS;YACZ,OAAO,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,qBAAqB,EAAE,CAAC;QACrH,KAAK,SAAS;YACZ,OAAO,OAAO,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,oBAAoB,EAAE,CAAC;QACvH,KAAK,MAAM;YACT,OAAO,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,eAAe,EAAE,CAAC;QACtG;YACE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IACtC,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,204 @@
|
|
|
1
|
+
import fs from "node:fs/promises";
|
|
2
|
+
import os from "node:os";
|
|
3
|
+
import path from "node:path";
|
|
4
|
+
import { InMemoryAppState } from "../app/app-state.js";
|
|
5
|
+
import { echoTool } from "./builtins/echo-tool.js";
|
|
6
|
+
import { editTool, writeTool } from "./builtins/edit-tool.js";
|
|
7
|
+
import { execTool } from "./builtins/exec-tool.js";
|
|
8
|
+
import { listDirectoryTool, readFileTool } from "./builtins/filesystem-tools.js";
|
|
9
|
+
import { grepTool } from "./builtins/grep-tool.js";
|
|
10
|
+
import { createSearchTool } from "./builtins/search-tool.js";
|
|
11
|
+
import { ToolRegistry } from "./registry.js";
|
|
12
|
+
import { runToolUseToMessages } from "./run-tool-use.js";
|
|
13
|
+
import { runTools } from "./tool-orchestration.js";
|
|
14
|
+
const delayTool = {
|
|
15
|
+
name: "delay",
|
|
16
|
+
description: "Wait briefly and return an id.",
|
|
17
|
+
inputSchema: {
|
|
18
|
+
type: "object",
|
|
19
|
+
properties: { id: { type: "string" }, delayMs: { type: "integer" } },
|
|
20
|
+
required: ["id", "delayMs"],
|
|
21
|
+
additionalProperties: false,
|
|
22
|
+
},
|
|
23
|
+
metadata: { readOnly: true, concurrent: true, visible: true },
|
|
24
|
+
validate(input) {
|
|
25
|
+
return input;
|
|
26
|
+
},
|
|
27
|
+
async call(input) {
|
|
28
|
+
await new Promise((resolve) => setTimeout(resolve, input.delayMs));
|
|
29
|
+
return { ok: true, output: input.id };
|
|
30
|
+
},
|
|
31
|
+
};
|
|
32
|
+
const mockSearchProvider = {
|
|
33
|
+
name: "mock",
|
|
34
|
+
async search(input) {
|
|
35
|
+
return {
|
|
36
|
+
provider: "mock",
|
|
37
|
+
query: input.query,
|
|
38
|
+
results: [
|
|
39
|
+
{
|
|
40
|
+
title: `Mock result for ${input.query}`,
|
|
41
|
+
url: "https://example.com/mock",
|
|
42
|
+
published: "2026-05-07",
|
|
43
|
+
highlights: ["Mock provider result used by smoke tests."],
|
|
44
|
+
},
|
|
45
|
+
],
|
|
46
|
+
};
|
|
47
|
+
},
|
|
48
|
+
};
|
|
49
|
+
const largeTool = {
|
|
50
|
+
name: "large",
|
|
51
|
+
description: "Return a large string for truncation tests.",
|
|
52
|
+
inputSchema: {
|
|
53
|
+
type: "object",
|
|
54
|
+
properties: { size: { type: "integer" } },
|
|
55
|
+
required: ["size"],
|
|
56
|
+
additionalProperties: false,
|
|
57
|
+
},
|
|
58
|
+
metadata: { readOnly: true, concurrent: true, visible: true, maxResultSizeChars: 8 },
|
|
59
|
+
validate(input) {
|
|
60
|
+
return input;
|
|
61
|
+
},
|
|
62
|
+
async call(input) {
|
|
63
|
+
return { ok: true, output: "x".repeat(input.size) };
|
|
64
|
+
},
|
|
65
|
+
};
|
|
66
|
+
async function main() {
|
|
67
|
+
const registry = new ToolRegistry();
|
|
68
|
+
registry.register(echoTool);
|
|
69
|
+
registry.register(editTool);
|
|
70
|
+
registry.register(writeTool);
|
|
71
|
+
registry.register(execTool);
|
|
72
|
+
registry.register(listDirectoryTool);
|
|
73
|
+
registry.register(readFileTool);
|
|
74
|
+
registry.register(grepTool);
|
|
75
|
+
registry.register(createSearchTool({ provider: mockSearchProvider }));
|
|
76
|
+
registry.register(delayTool);
|
|
77
|
+
registry.register(largeTool);
|
|
78
|
+
const context = {
|
|
79
|
+
agentId: "tool-smoke",
|
|
80
|
+
tools: registry,
|
|
81
|
+
appState: new InMemoryAppState("tool-smoke", process.cwd()),
|
|
82
|
+
emit: () => undefined,
|
|
83
|
+
};
|
|
84
|
+
const tempDir = await fs.mkdtemp(path.join(os.tmpdir(), "agent-tools-smoke-"));
|
|
85
|
+
const valid = await runToolUseToMessages({ id: "echo1", name: "say", input: { text: "ok" } }, context);
|
|
86
|
+
const invalid = await runToolUseToMessages({ id: "echo2", name: "echo", input: { text: "" } }, context);
|
|
87
|
+
const unknown = await runToolUseToMessages({ id: "missing", name: "missing", input: {} }, context);
|
|
88
|
+
const large = await runToolUseToMessages({ id: "large", name: "large", input: { size: 20 } }, context);
|
|
89
|
+
const grep = await runToolUseToMessages({ id: "grep", name: "grep", input: { query: "echoTool", path: "src/tools/builtins/echo-tool.ts", maxResults: 5 } }, context);
|
|
90
|
+
const truncatedGrep = await runToolUseToMessages({ id: "grep-truncated", name: "grep", input: { query: "import", path: "src", maxResults: 1 } }, context);
|
|
91
|
+
const webSearch = await runToolUseToMessages({ id: "web-search", name: "search", input: { query: "agent scaffold", numResults: 1 } }, context);
|
|
92
|
+
const contextGrep = await runToolUseToMessages({ id: "grep-context", name: "grep", input: { query: "description", path: "src/tools/builtins/echo-tool.ts", contextLines: 1, maxResults: 2 } }, context);
|
|
93
|
+
const read = await runToolUseToMessages({ id: "read", name: "read", input: { path: "src/tools/builtins/echo-tool.ts", offset: 1, limit: 5 } }, context);
|
|
94
|
+
const list = await runToolUseToMessages({ id: "list", name: "list", input: { path: "src/tools/builtins", recursive: false, maxEntries: 20 } }, context);
|
|
95
|
+
const recursiveListRoot = path.join(tempDir, "recursive-list");
|
|
96
|
+
await fs.mkdir(path.join(recursiveListRoot, ".idea"), { recursive: true });
|
|
97
|
+
await fs.mkdir(path.join(recursiveListRoot, ".agent-tasks"), { recursive: true });
|
|
98
|
+
await fs.mkdir(path.join(recursiveListRoot, "visible"), { recursive: true });
|
|
99
|
+
await fs.writeFile(path.join(recursiveListRoot, ".idea", "workspace.xml"), "ignored", "utf8");
|
|
100
|
+
await fs.writeFile(path.join(recursiveListRoot, ".agent-tasks", "task.json"), "ignored", "utf8");
|
|
101
|
+
await fs.writeFile(path.join(recursiveListRoot, "visible", "keep.ts"), "export {};", "utf8");
|
|
102
|
+
const recursiveList = await runToolUseToMessages({ id: "list-recursive", name: "list", input: { path: recursiveListRoot, recursive: true, maxEntries: 20, includeHidden: true } }, context);
|
|
103
|
+
const exec = await runToolUseToMessages({ id: "exec", name: "exec", input: { command: "node -e \"console.log(process.cwd()); console.error('warn')\"", timeoutMs: 10000, maxOutputChars: 4000 } }, context);
|
|
104
|
+
const execFailure = await runToolUseToMessages({ id: "exec-fail", name: "exec", input: { command: "node -e \"process.exit(7)\"", timeoutMs: 10000 } }, context);
|
|
105
|
+
const editCreate = await runToolUseToMessages({ id: "edit-create", name: "edit", input: { path: path.join(tempDir, "sample.txt"), oldString: "", newString: "alpha\nbeta\nalpha\n" } }, context);
|
|
106
|
+
const editAmbiguous = await runToolUseToMessages({ id: "edit-ambiguous", name: "edit", input: { path: path.join(tempDir, "sample.txt"), oldString: "alpha", newString: "gamma" } }, context);
|
|
107
|
+
const editReplaceAll = await runToolUseToMessages({ id: "edit-all", name: "edit", input: { path: path.join(tempDir, "sample.txt"), oldString: "alpha", newString: "gamma", replaceAll: true } }, context);
|
|
108
|
+
const crlfPath = path.join(tempDir, "crlf.txt");
|
|
109
|
+
await fs.writeFile(crlfPath, "one\r\ntwo\r\nthree\r\n", "utf8");
|
|
110
|
+
const editCrlfWithLfOldString = await runToolUseToMessages({ id: "edit-crlf-lf-old", name: "edit", input: { path: crlfPath, oldString: "one\ntwo\n", newString: "uno\ndos\n" } }, context);
|
|
111
|
+
const write = await runToolUseToMessages({ id: "write", name: "write", input: { path: path.join(tempDir, "nested", "write.txt"), content: "full\ncontent\n" } }, context);
|
|
112
|
+
const grepOutput = toolOutput(grep[grep.length - 1]);
|
|
113
|
+
const truncatedGrepOutput = toolOutput(truncatedGrep[truncatedGrep.length - 1]);
|
|
114
|
+
const contextGrepOutput = toolOutput(contextGrep[contextGrep.length - 1]);
|
|
115
|
+
const recursiveListOutput = toolOutput(recursiveList[recursiveList.length - 1]);
|
|
116
|
+
const execOutput = toolOutput(exec[exec.length - 1]);
|
|
117
|
+
const execFailureOutput = toolOutput(execFailure[execFailure.length - 1]);
|
|
118
|
+
const editCreateOutput = toolOutput(editCreate[editCreate.length - 1]);
|
|
119
|
+
const editReplaceAllOutput = toolOutput(editReplaceAll[editReplaceAll.length - 1]);
|
|
120
|
+
const editCrlfWithLfOldStringOutput = toolOutput(editCrlfWithLfOldString[editCrlfWithLfOldString.length - 1]);
|
|
121
|
+
const webSearchOutput = toolOutput(webSearch[webSearch.length - 1]);
|
|
122
|
+
const writeOutput = toolOutput(write[write.length - 1]);
|
|
123
|
+
const started = Date.now();
|
|
124
|
+
const batch = await runTools([
|
|
125
|
+
{ id: "d1", name: "delay", input: { id: "a", delayMs: 60 } },
|
|
126
|
+
{ id: "d2", name: "delay", input: { id: "b", delayMs: 60 } },
|
|
127
|
+
], context);
|
|
128
|
+
const elapsedMs = Date.now() - started;
|
|
129
|
+
const checks = {
|
|
130
|
+
validTool: toolOk(valid[valid.length - 1]),
|
|
131
|
+
invalidRejected: !toolOk(invalid[invalid.length - 1]),
|
|
132
|
+
unknownRejected: !toolOk(unknown[0]),
|
|
133
|
+
transportTruncationLabel: JSON.stringify(large[large.length - 1]).includes("truncated"),
|
|
134
|
+
grepOk: toolOk(grep[grep.length - 1]),
|
|
135
|
+
grepFindsFile: JSON.stringify(grep[grep.length - 1]).includes("echo-tool.ts"),
|
|
136
|
+
grepFields: grepOutput.cwd !== undefined && grepOutput.grepPath !== undefined,
|
|
137
|
+
grepKnownTotal: grepOutput.returnedMatches === grepOutput.totalMatchesKnown,
|
|
138
|
+
grepNoLegacyRoot: !JSON.stringify(grep[grep.length - 1]).includes('"root"'),
|
|
139
|
+
truncatedGrepOk: toolOk(truncatedGrep[truncatedGrep.length - 1]),
|
|
140
|
+
webSearchOk: toolOk(webSearch[webSearch.length - 1]),
|
|
141
|
+
webSearchUsesProvider: webSearchOutput.provider === "mock" && webSearchOutput.returnedResults === 1,
|
|
142
|
+
webSearchFindsUrl: webSearchOutput.results?.[0]?.url === "https://example.com/mock",
|
|
143
|
+
truncatedGrepCounts: truncatedGrepOutput.returnedMatches === 1 &&
|
|
144
|
+
truncatedGrepOutput.totalMatchesKnown === null &&
|
|
145
|
+
truncatedGrepOutput.truncated === true,
|
|
146
|
+
contextGrepOk: toolOk(contextGrep[contextGrep.length - 1]),
|
|
147
|
+
contextGrepLines: (contextGrepOutput.matches?.some((match) => (match.contextBefore?.length ?? 0) > 0) ?? false) ||
|
|
148
|
+
(contextGrepOutput.matches?.some((match) => (match.contextAfter?.length ?? 0) > 0) ?? false),
|
|
149
|
+
readOk: toolOk(read[read.length - 1]),
|
|
150
|
+
readLineMetadata: JSON.stringify(read[read.length - 1]).includes('"startLine":1'),
|
|
151
|
+
readContent: JSON.stringify(read[read.length - 1]).includes("echoTool"),
|
|
152
|
+
listOk: toolOk(list[list.length - 1]),
|
|
153
|
+
listFindsFile: JSON.stringify(list[list.length - 1]).includes("grep-tool.ts"),
|
|
154
|
+
recursiveListOk: toolOk(recursiveList[recursiveList.length - 1]),
|
|
155
|
+
recursiveListDefaultExcludes: recursiveListOutput.exclude?.includes(".git") === true &&
|
|
156
|
+
recursiveListOutput.exclude?.includes(".idea") === true &&
|
|
157
|
+
recursiveListOutput.exclude?.includes(".agent-tasks") === true,
|
|
158
|
+
recursiveListTracksExcluded: recursiveListOutput.excludedCounts?.[".idea"] !== undefined &&
|
|
159
|
+
recursiveListOutput.excludedCounts?.[".agent-tasks"] !== undefined,
|
|
160
|
+
recursiveListEntriesClean: recursiveListOutput.entries?.every((entry) => ![".git", ".idea", ".agent-tasks", "node_modules", "dist"].includes(entry.name)) === true,
|
|
161
|
+
execOk: toolOk(exec[exec.length - 1]) &&
|
|
162
|
+
execOutput.exitCode === 0 &&
|
|
163
|
+
execOutput.stdout?.includes(process.cwd()) === true &&
|
|
164
|
+
execOutput.stderr?.includes("warn") === true,
|
|
165
|
+
execFailureRejected: !toolOk(execFailure[execFailure.length - 1]) &&
|
|
166
|
+
typeof execFailureOutput.exitCode === "number" &&
|
|
167
|
+
execFailureOutput.exitCode !== 0,
|
|
168
|
+
editCreateOk: toolOk(editCreate[editCreate.length - 1]) &&
|
|
169
|
+
editCreateOutput.operation === "create" &&
|
|
170
|
+
editCreateOutput.replacements === 1 &&
|
|
171
|
+
(editCreateOutput.patch?.length ?? 0) > 0,
|
|
172
|
+
editAmbiguousRejected: !toolOk(editAmbiguous[editAmbiguous.length - 1]),
|
|
173
|
+
editReplaceAllOk: toolOk(editReplaceAll[editReplaceAll.length - 1]) &&
|
|
174
|
+
editReplaceAllOutput.operation === "edit" &&
|
|
175
|
+
editReplaceAllOutput.replacements === 2 &&
|
|
176
|
+
(await fs.readFile(path.join(tempDir, "sample.txt"), "utf8")).includes("gamma\nbeta\ngamma"),
|
|
177
|
+
editCrlfWithLfOldStringOk: toolOk(editCrlfWithLfOldString[editCrlfWithLfOldString.length - 1]) &&
|
|
178
|
+
editCrlfWithLfOldStringOutput.operation === "edit" &&
|
|
179
|
+
editCrlfWithLfOldStringOutput.replacements === 1 &&
|
|
180
|
+
(await fs.readFile(crlfPath, "utf8")) === "uno\r\ndos\r\nthree\r\n",
|
|
181
|
+
writeOk: toolOk(write[write.length - 1]) &&
|
|
182
|
+
writeOutput.operation === "create" &&
|
|
183
|
+
writeOutput.bytesAfter === Buffer.byteLength("full\ncontent\n") &&
|
|
184
|
+
(await fs.readFile(path.join(tempDir, "nested", "write.txt"), "utf8")) === "full\ncontent\n",
|
|
185
|
+
batchMessages: batch.messages.length === 4,
|
|
186
|
+
batchConcurrent: elapsedMs < 110,
|
|
187
|
+
};
|
|
188
|
+
const ok = Object.values(checks).every(Boolean);
|
|
189
|
+
console.log(JSON.stringify({ ok, elapsedMs, checks, counts: { valid: valid.length, invalid: invalid.length, unknown: unknown.length, large: large.length, grep: grep.length, truncatedGrep: truncatedGrep.length, webSearch: webSearch.length, contextGrep: contextGrep.length, read: read.length, list: list.length, recursiveList: recursiveList.length, exec: exec.length, execFailure: execFailure.length, editCreate: editCreate.length, editAmbiguous: editAmbiguous.length, editReplaceAll: editReplaceAll.length, editCrlfWithLfOldString: editCrlfWithLfOldString.length, write: write.length, batch: batch.messages.length } }, null, 2));
|
|
190
|
+
if (!ok)
|
|
191
|
+
process.exitCode = 1;
|
|
192
|
+
}
|
|
193
|
+
function toolOk(message) {
|
|
194
|
+
return message.blocks.every((block) => block.type !== "tool_result" || block.ok);
|
|
195
|
+
}
|
|
196
|
+
function toolOutput(message) {
|
|
197
|
+
const block = message.blocks.find((candidate) => candidate.type === "tool_result");
|
|
198
|
+
return block?.type === "tool_result" ? block.output : undefined;
|
|
199
|
+
}
|
|
200
|
+
main().catch((error) => {
|
|
201
|
+
console.error(error instanceof Error ? error.stack ?? error.message : String(error));
|
|
202
|
+
process.exitCode = 1;
|
|
203
|
+
});
|
|
204
|
+
//# sourceMappingURL=smoke-tool-system.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"smoke-tool-system.js","sourceRoot":"","sources":["../../src/tools/smoke-tool-system.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEvD,OAAO,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AACnD,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AACnD,OAAO,EAAE,iBAAiB,EAAE,YAAY,EAAE,MAAM,gCAAgC,CAAC;AACjF,OAAO,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAE7D,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAGnD,MAAM,SAAS,GAA0C;IACvD,IAAI,EAAE,OAAO;IACb,WAAW,EAAE,gCAAgC;IAC7C,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;QACpE,QAAQ,EAAE,CAAC,IAAI,EAAE,SAAS,CAAC;QAC3B,oBAAoB,EAAE,KAAK;KAC5B;IACD,QAAQ,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE;IAC7D,QAAQ,CAAC,KAAK;QACZ,OAAO,KAAwC,CAAC;IAClD,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,KAAK;QACd,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QACnE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;IACxC,CAAC;CACF,CAAC;AAGF,MAAM,kBAAkB,GAAmB;IACzC,IAAI,EAAE,MAAM;IACZ,KAAK,CAAC,MAAM,CAAC,KAAK;QAChB,OAAO;YACL,QAAQ,EAAE,MAAM;YAChB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,OAAO,EAAE;gBACP;oBACE,KAAK,EAAE,mBAAmB,KAAK,CAAC,KAAK,EAAE;oBACvC,GAAG,EAAE,0BAA0B;oBAC/B,SAAS,EAAE,YAAY;oBACvB,UAAU,EAAE,CAAC,2CAA2C,CAAC;iBAC1D;aACF;SACF,CAAC;IACJ,CAAC;CACF,CAAC;AAEF,MAAM,SAAS,GAA2B;IACxC,IAAI,EAAE,OAAO;IACb,WAAW,EAAE,6CAA6C;IAC1D,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;QACzC,QAAQ,EAAE,CAAC,MAAM,CAAC;QAClB,oBAAoB,EAAE,KAAK;KAC5B;IACD,QAAQ,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,EAAE;IACpF,QAAQ,CAAC,KAAK;QACZ,OAAO,KAAyB,CAAC;IACnC,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,KAAK;QACd,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;IACtD,CAAC;CACF,CAAC;AAEF,KAAK,UAAU,IAAI;IACjB,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;IACpC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC5B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC5B,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC7B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC5B,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;IACrC,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IAChC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC5B,QAAQ,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,QAAQ,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;IACtE,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC7B,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAE7B,MAAM,OAAO,GAAmB;QAC9B,OAAO,EAAE,YAAY;QACrB,KAAK,EAAE,QAAQ;QACf,QAAQ,EAAE,IAAI,gBAAgB,CAAC,YAAY,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC;QAC3D,IAAI,EAAE,GAAG,EAAE,CAAC,SAAS;KACtB,CAAC;IACF,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,oBAAoB,CAAC,CAAC,CAAC;IAE/E,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;IACvG,MAAM,OAAO,GAAG,MAAM,oBAAoB,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;IACxG,MAAM,OAAO,GAAG,MAAM,oBAAoB,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;IACnG,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;IACvG,MAAM,IAAI,GAAG,MAAM,oBAAoB,CACrC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,iCAAiC,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,EAClH,OAAO,CACR,CAAC;IACF,MAAM,aAAa,GAAG,MAAM,oBAAoB,CAC9C,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,EAC9F,OAAO,CACR,CAAC;IACF,MAAM,SAAS,GAAG,MAAM,oBAAoB,CAC1C,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,gBAAgB,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,EACvF,OAAO,CACR,CAAC;IACF,MAAM,WAAW,GAAG,MAAM,oBAAoB,CAC5C,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,iCAAiC,EAAE,YAAY,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,EAC9I,OAAO,CACR,CAAC;IACF,MAAM,IAAI,GAAG,MAAM,oBAAoB,CACrC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,iCAAiC,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,EACrG,OAAO,CACR,CAAC;IACF,MAAM,IAAI,GAAG,MAAM,oBAAoB,CACrC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,oBAAoB,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE,EACrG,OAAO,CACR,CAAC;IACF,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;IAC/D,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAO,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3E,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,cAAc,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAClF,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,SAAS,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7E,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAO,EAAE,eAAe,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;IAC9F,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,cAAc,EAAE,WAAW,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;IACjG,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;IAC7F,MAAM,aAAa,GAAG,MAAM,oBAAoB,CAC9C,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,EAChI,OAAO,CACR,CAAC;IACF,MAAM,IAAI,GAAG,MAAM,oBAAoB,CACrC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,+DAA+D,EAAE,SAAS,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,EACzJ,OAAO,CACR,CAAC;IACF,MAAM,WAAW,GAAG,MAAM,oBAAoB,CAC5C,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,6BAA6B,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EACtG,OAAO,CACR,CAAC;IACF,MAAM,UAAU,GAAG,MAAM,oBAAoB,CAC3C,EAAE,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,sBAAsB,EAAE,EAAE,EACxI,OAAO,CACR,CAAC;IACF,MAAM,aAAa,GAAG,MAAM,oBAAoB,CAC9C,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,EACjI,OAAO,CACR,CAAC;IACF,MAAM,cAAc,GAAG,MAAM,oBAAoB,CAC/C,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,EAC7I,OAAO,CACR,CAAC;IACF,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IAChD,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,yBAAyB,EAAE,MAAM,CAAC,CAAC;IAChE,MAAM,uBAAuB,GAAG,MAAM,oBAAoB,CACxD,EAAE,EAAE,EAAE,kBAAkB,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,YAAY,EAAE,SAAS,EAAE,YAAY,EAAE,EAAE,EACrH,OAAO,CACR,CAAC;IACF,MAAM,KAAK,GAAG,MAAM,oBAAoB,CACtC,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,WAAW,CAAC,EAAE,OAAO,EAAE,iBAAiB,EAAE,EAAE,EACtH,OAAO,CACR,CAAC;IACF,MAAM,UAAU,GAAG,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAKlD,CAAC;IACF,MAAM,mBAAmB,GAAG,UAAU,CAAC,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAI7E,CAAC;IACF,MAAM,iBAAiB,GAAG,UAAU,CAAC,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAEvE,CAAC;IACF,MAAM,mBAAmB,GAAG,UAAU,CAAC,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAI7E,CAAC;IACF,MAAM,UAAU,GAAG,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAKlD,CAAC;IACF,MAAM,iBAAiB,GAAG,UAAU,CAAC,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAEvE,CAAC;IACF,MAAM,gBAAgB,GAAG,UAAU,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAIpE,CAAC;IACF,MAAM,oBAAoB,GAAG,UAAU,CAAC,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAIhF,CAAC;IACF,MAAM,6BAA6B,GAAG,UAAU,CAAC,uBAAuB,CAAC,uBAAuB,CAAC,MAAM,GAAG,CAAC,CAAC,CAG3G,CAAC;IACF,MAAM,eAAe,GAAG,UAAU,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAIjE,CAAC;IACF,MAAM,WAAW,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAGrD,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC3B,MAAM,KAAK,GAAG,MAAM,QAAQ,CAC1B;QACE,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE;QAC5D,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE;KAC7D,EACD,OAAO,CACR,CAAC;IACF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC;IAEvC,MAAM,MAAM,GAAG;QACb,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC1C,eAAe,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACrD,eAAe,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACpC,wBAAwB,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC;QACvF,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACrC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC;QAC7E,UAAU,EAAE,UAAU,CAAC,GAAG,KAAK,SAAS,IAAI,UAAU,CAAC,QAAQ,KAAK,SAAS;QAC7E,cAAc,EAAE,UAAU,CAAC,eAAe,KAAK,UAAU,CAAC,iBAAiB;QAC3E,gBAAgB,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAC3E,eAAe,EAAE,MAAM,CAAC,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAChE,WAAW,EAAE,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACpD,qBAAqB,EAAE,eAAe,CAAC,QAAQ,KAAK,MAAM,IAAI,eAAe,CAAC,eAAe,KAAK,CAAC;QACnG,iBAAiB,EAAE,eAAe,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,KAAK,0BAA0B;QACnF,mBAAmB,EACjB,mBAAmB,CAAC,eAAe,KAAK,CAAC;YACzC,mBAAmB,CAAC,iBAAiB,KAAK,IAAI;YAC9C,mBAAmB,CAAC,SAAS,KAAK,IAAI;QACxC,aAAa,EAAE,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC1D,gBAAgB,EACd,CAAC,iBAAiB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,aAAa,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC;YAC7F,CAAC,iBAAiB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,YAAY,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC;QAC9F,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACrC,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,eAAe,CAAC;QACjF,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC;QACvE,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACrC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC;QAC7E,eAAe,EAAE,MAAM,CAAC,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAChE,4BAA4B,EAC1B,mBAAmB,CAAC,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,KAAK,IAAI;YACtD,mBAAmB,CAAC,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC,KAAK,IAAI;YACvD,mBAAmB,CAAC,OAAO,EAAE,QAAQ,CAAC,cAAc,CAAC,KAAK,IAAI;QAChE,2BAA2B,EACzB,mBAAmB,CAAC,cAAc,EAAE,CAAC,OAAO,CAAC,KAAK,SAAS;YAC3D,mBAAmB,CAAC,cAAc,EAAE,CAAC,cAAc,CAAC,KAAK,SAAS;QACpE,yBAAyB,EACvB,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,IAAI;QACzI,MAAM,EACJ,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC7B,UAAU,CAAC,QAAQ,KAAK,CAAC;YACzB,UAAU,CAAC,MAAM,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,KAAK,IAAI;YACnD,UAAU,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,KAAK,IAAI;QAC9C,mBAAmB,EACjB,CAAC,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC5C,OAAO,iBAAiB,CAAC,QAAQ,KAAK,QAAQ;YAC9C,iBAAiB,CAAC,QAAQ,KAAK,CAAC;QAClC,YAAY,EACV,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACzC,gBAAgB,CAAC,SAAS,KAAK,QAAQ;YACvC,gBAAgB,CAAC,YAAY,KAAK,CAAC;YACnC,CAAC,gBAAgB,CAAC,KAAK,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC;QAC3C,qBAAqB,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACvE,gBAAgB,EACd,MAAM,CAAC,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACjD,oBAAoB,CAAC,SAAS,KAAK,MAAM;YACzC,oBAAoB,CAAC,YAAY,KAAK,CAAC;YACvC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,oBAAoB,CAAC;QAC9F,yBAAyB,EACvB,MAAM,CAAC,uBAAuB,CAAC,uBAAuB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACnE,6BAA6B,CAAC,SAAS,KAAK,MAAM;YAClD,6BAA6B,CAAC,YAAY,KAAK,CAAC;YAChD,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,KAAK,yBAAyB;QACrE,OAAO,EACL,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC/B,WAAW,CAAC,SAAS,KAAK,QAAQ;YAClC,WAAW,CAAC,UAAU,KAAK,MAAM,CAAC,UAAU,CAAC,iBAAiB,CAAC;YAC/D,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,WAAW,CAAC,EAAE,MAAM,CAAC,CAAC,KAAK,iBAAiB;QAC9F,aAAa,EAAE,KAAK,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC;QAC1C,eAAe,EAAE,SAAS,GAAG,GAAG;KACjC,CAAC;IACF,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAEhD,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,aAAa,EAAE,aAAa,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,WAAW,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,aAAa,EAAE,aAAa,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,WAAW,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,MAAM,EAAE,aAAa,EAAE,aAAa,CAAC,MAAM,EAAE,cAAc,EAAE,cAAc,CAAC,MAAM,EAAE,uBAAuB,EAAE,uBAAuB,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACpnB,IAAI,CAAC,EAAE;QAAE,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;AAChC,CAAC;AAED,SAAS,MAAM,CAAC,OAAgB;IAC9B,OAAO,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,KAAK,CAAC,EAAE,CAAC,CAAC;AACnF,CAAC;AAED,SAAS,UAAU,CAAC,OAAgB;IAClC,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC;IACnF,OAAO,KAAK,EAAE,IAAI,KAAK,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;AAClE,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;IACrB,OAAO,CAAC,KAAK,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACrF,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import { type Message, type ToolUseRequest } from "../types/messages.js";
|
|
2
|
+
import { type RunToolUseOptions } from "./run-tool-use.js";
|
|
3
|
+
import type { ToolUseContext } from "./tool.js";
|
|
4
|
+
export interface StreamingToolExecutorOptions {
|
|
5
|
+
context: ToolUseContext;
|
|
6
|
+
runOptions?: RunToolUseOptions;
|
|
7
|
+
}
|
|
8
|
+
export declare class StreamingToolExecutor {
|
|
9
|
+
private readonly options;
|
|
10
|
+
private readonly tracked;
|
|
11
|
+
private discarded;
|
|
12
|
+
constructor(options: StreamingToolExecutorOptions);
|
|
13
|
+
addTool(request: ToolUseRequest): void;
|
|
14
|
+
getCompletedResults(): Message[];
|
|
15
|
+
getRemainingResults(): Promise<Message[]>;
|
|
16
|
+
discard(): Message[];
|
|
17
|
+
private processQueue;
|
|
18
|
+
private canExecute;
|
|
19
|
+
}
|
|
@@ -0,0 +1,89 @@
|
|
|
1
|
+
import { createToolResultMessage } from "../types/messages.js";
|
|
2
|
+
import { runToolUse } from "./run-tool-use.js";
|
|
3
|
+
export class StreamingToolExecutor {
|
|
4
|
+
options;
|
|
5
|
+
tracked = [];
|
|
6
|
+
discarded = false;
|
|
7
|
+
constructor(options) {
|
|
8
|
+
this.options = options;
|
|
9
|
+
}
|
|
10
|
+
addTool(request) {
|
|
11
|
+
if (this.discarded)
|
|
12
|
+
return;
|
|
13
|
+
const tool = this.options.context.tools.get(request.name) ?? this.options.context.tools.getByAlias?.(request.name);
|
|
14
|
+
const concurrencySafe = tool?.isConcurrencySafe?.(request.input, this.options.context) ?? tool?.metadata.concurrent ?? true;
|
|
15
|
+
this.tracked.push({ request, status: "queued", concurrencySafe });
|
|
16
|
+
this.processQueue();
|
|
17
|
+
}
|
|
18
|
+
getCompletedResults() {
|
|
19
|
+
const messages = [];
|
|
20
|
+
for (const tracked of this.tracked) {
|
|
21
|
+
if (tracked.status !== "completed" || !tracked.updates)
|
|
22
|
+
continue;
|
|
23
|
+
tracked.status = "yielded";
|
|
24
|
+
messages.push(...tracked.updates.map((update) => update.message));
|
|
25
|
+
}
|
|
26
|
+
this.processQueue();
|
|
27
|
+
return messages;
|
|
28
|
+
}
|
|
29
|
+
async getRemainingResults() {
|
|
30
|
+
while (this.tracked.some((tracked) => tracked.status === "queued" || tracked.status === "executing" || tracked.status === "completed")) {
|
|
31
|
+
const executing = this.tracked.filter((tracked) => tracked.status === "executing" && tracked.promise);
|
|
32
|
+
if (executing.length)
|
|
33
|
+
await Promise.race(executing.map((tracked) => tracked.promise));
|
|
34
|
+
const completed = this.getCompletedResults();
|
|
35
|
+
if (completed.length)
|
|
36
|
+
return [...completed, ...(await this.getRemainingResults())];
|
|
37
|
+
this.processQueue();
|
|
38
|
+
}
|
|
39
|
+
return [];
|
|
40
|
+
}
|
|
41
|
+
discard() {
|
|
42
|
+
this.discarded = true;
|
|
43
|
+
const synthetic = [];
|
|
44
|
+
for (const tracked of this.tracked) {
|
|
45
|
+
if (tracked.status === "yielded" || tracked.status === "discarded")
|
|
46
|
+
continue;
|
|
47
|
+
tracked.status = "discarded";
|
|
48
|
+
synthetic.push(createToolResultMessage(tracked.request, false, { error: "Tool result discarded after stream fallback or abort" }));
|
|
49
|
+
}
|
|
50
|
+
return synthetic;
|
|
51
|
+
}
|
|
52
|
+
processQueue() {
|
|
53
|
+
if (this.discarded)
|
|
54
|
+
return;
|
|
55
|
+
for (const tracked of this.tracked) {
|
|
56
|
+
if (tracked.status !== "queued")
|
|
57
|
+
continue;
|
|
58
|
+
if (!this.canExecute(tracked))
|
|
59
|
+
return;
|
|
60
|
+
tracked.status = "executing";
|
|
61
|
+
tracked.promise = runToolUse(tracked.request, this.options.context, this.options.runOptions)
|
|
62
|
+
.then((updates) => {
|
|
63
|
+
tracked.updates = updates;
|
|
64
|
+
tracked.status = "completed";
|
|
65
|
+
this.processQueue();
|
|
66
|
+
return updates;
|
|
67
|
+
})
|
|
68
|
+
.catch((error) => {
|
|
69
|
+
tracked.updates = [
|
|
70
|
+
{
|
|
71
|
+
message: createToolResultMessage(tracked.request, false, {
|
|
72
|
+
error: error instanceof Error ? error.message : String(error),
|
|
73
|
+
}),
|
|
74
|
+
},
|
|
75
|
+
];
|
|
76
|
+
tracked.status = "completed";
|
|
77
|
+
this.processQueue();
|
|
78
|
+
return tracked.updates;
|
|
79
|
+
});
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
canExecute(next) {
|
|
83
|
+
const executing = this.tracked.filter((tracked) => tracked.status === "executing");
|
|
84
|
+
if (executing.length === 0)
|
|
85
|
+
return true;
|
|
86
|
+
return next.concurrencySafe && executing.every((tracked) => tracked.concurrencySafe);
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
//# sourceMappingURL=streaming-tool-executor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"streaming-tool-executor.js","sourceRoot":"","sources":["../../src/tools/streaming-tool-executor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAqC,MAAM,sBAAsB,CAAC;AAClG,OAAO,EAAE,UAAU,EAAkD,MAAM,mBAAmB,CAAC;AAkB/F,MAAM,OAAO,qBAAqB;IAIH;IAHZ,OAAO,GAAkB,EAAE,CAAC;IACrC,SAAS,GAAG,KAAK,CAAC;IAE1B,YAA6B,OAAqC;QAArC,YAAO,GAAP,OAAO,CAA8B;IAAG,CAAC;IAEtE,OAAO,CAAC,OAAuB;QAC7B,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACnH,MAAM,eAAe,GAAG,IAAI,EAAE,iBAAiB,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,QAAQ,CAAC,UAAU,IAAI,IAAI,CAAC;QAC5H,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC,CAAC;QAClE,IAAI,CAAC,YAAY,EAAE,CAAC;IACtB,CAAC;IAED,mBAAmB;QACjB,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACnC,IAAI,OAAO,CAAC,MAAM,KAAK,WAAW,IAAI,CAAC,OAAO,CAAC,OAAO;gBAAE,SAAS;YACjE,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;YAC3B,QAAQ,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;QACpE,CAAC;QACD,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,mBAAmB;QACvB,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,KAAK,QAAQ,IAAI,OAAO,CAAC,MAAM,KAAK,WAAW,IAAI,OAAO,CAAC,MAAM,KAAK,WAAW,CAAC,EAAE,CAAC;YACvI,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,KAAK,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;YACtG,IAAI,SAAS,CAAC,MAAM;gBAAE,MAAM,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;YACtF,MAAM,SAAS,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC7C,IAAI,SAAS,CAAC,MAAM;gBAAE,OAAO,CAAC,GAAG,SAAS,EAAE,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC;YACnF,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO;QACL,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,MAAM,SAAS,GAAc,EAAE,CAAC;QAChC,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACnC,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,IAAI,OAAO,CAAC,MAAM,KAAK,WAAW;gBAAE,SAAS;YAC7E,OAAO,CAAC,MAAM,GAAG,WAAW,CAAC;YAC7B,SAAS,CAAC,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,sDAAsD,EAAE,CAAC,CAAC,CAAC;QACrI,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,YAAY;QAClB,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACnC,IAAI,OAAO,CAAC,MAAM,KAAK,QAAQ;gBAAE,SAAS;YAC1C,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;gBAAE,OAAO;YACtC,OAAO,CAAC,MAAM,GAAG,WAAW,CAAC;YAC7B,OAAO,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;iBACzF,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE;gBAChB,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC;gBAC1B,OAAO,CAAC,MAAM,GAAG,WAAW,CAAC;gBAC7B,IAAI,CAAC,YAAY,EAAE,CAAC;gBACpB,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACf,OAAO,CAAC,OAAO,GAAG;oBAChB;wBACE,OAAO,EAAE,uBAAuB,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE;4BACvD,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;yBAC9D,CAAC;qBACH;iBACF,CAAC;gBACF,OAAO,CAAC,MAAM,GAAG,WAAW,CAAC;gBAC7B,IAAI,CAAC,YAAY,EAAE,CAAC;gBACpB,OAAO,OAAO,CAAC,OAAO,CAAC;YACzB,CAAC,CAAC,CAAC;QACP,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,IAAiB;QAClC,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC;QACnF,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACxC,OAAO,IAAI,CAAC,eAAe,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IACvF,CAAC;CACF"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import type { Message, ToolUseRequest } from "../types/messages.js";
|
|
2
|
+
import { type RunToolUseOptions } from "./run-tool-use.js";
|
|
3
|
+
import type { Tool, ToolUseContext } from "./tool.js";
|
|
4
|
+
export interface RunToolsResult {
|
|
5
|
+
messages: Message[];
|
|
6
|
+
context: ToolUseContext;
|
|
7
|
+
}
|
|
8
|
+
interface ParsedToolCall {
|
|
9
|
+
request: ToolUseRequest;
|
|
10
|
+
tool?: Tool;
|
|
11
|
+
concurrencySafe: boolean;
|
|
12
|
+
}
|
|
13
|
+
export declare function runTools(requests: readonly ToolUseRequest[], context: ToolUseContext, options?: RunToolUseOptions): Promise<RunToolsResult>;
|
|
14
|
+
export declare function partitionToolCalls(requests: readonly ToolUseRequest[], context: ToolUseContext): ParsedToolCall[][];
|
|
15
|
+
export {};
|