@mcp-abap-adt/llm-agent 2.9.2 → 2.10.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/dist/index.d.ts +11 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +7 -0
- package/dist/index.js.map +1 -1
- package/dist/smart-agent/agent.d.ts +18 -1
- package/dist/smart-agent/agent.d.ts.map +1 -1
- package/dist/smart-agent/agent.js +176 -21
- package/dist/smart-agent/agent.js.map +1 -1
- package/dist/smart-agent/builder.d.ts +114 -51
- package/dist/smart-agent/builder.d.ts.map +1 -1
- package/dist/smart-agent/builder.js +234 -110
- package/dist/smart-agent/builder.js.map +1 -1
- package/dist/smart-agent/cli.d.ts +1 -0
- package/dist/smart-agent/cli.d.ts.map +1 -1
- package/dist/smart-agent/cli.js +2 -0
- package/dist/smart-agent/cli.js.map +1 -1
- package/dist/smart-agent/config/config-watcher.d.ts +4 -0
- package/dist/smart-agent/config/config-watcher.d.ts.map +1 -1
- package/dist/smart-agent/config/config-watcher.js +8 -0
- package/dist/smart-agent/config/config-watcher.js.map +1 -1
- package/dist/smart-agent/config.d.ts +2 -1
- package/dist/smart-agent/config.d.ts.map +1 -1
- package/dist/smart-agent/config.js +46 -4
- package/dist/smart-agent/config.js.map +1 -1
- package/dist/smart-agent/interfaces/index.d.ts +1 -1
- package/dist/smart-agent/interfaces/index.d.ts.map +1 -1
- package/dist/smart-agent/interfaces/index.js.map +1 -1
- package/dist/smart-agent/interfaces/rag.d.ts +16 -0
- package/dist/smart-agent/interfaces/rag.d.ts.map +1 -1
- package/dist/smart-agent/pipeline/condition-evaluator.d.ts +35 -0
- package/dist/smart-agent/pipeline/condition-evaluator.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/condition-evaluator.js +143 -0
- package/dist/smart-agent/pipeline/condition-evaluator.js.map +1 -0
- package/dist/smart-agent/pipeline/context.d.ts +104 -0
- package/dist/smart-agent/pipeline/context.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/context.js +19 -0
- package/dist/smart-agent/pipeline/context.js.map +1 -0
- package/dist/smart-agent/pipeline/default-pipeline.d.ts +45 -0
- package/dist/smart-agent/pipeline/default-pipeline.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/default-pipeline.js +118 -0
- package/dist/smart-agent/pipeline/default-pipeline.js.map +1 -0
- package/dist/smart-agent/pipeline/executor.d.ts +54 -0
- package/dist/smart-agent/pipeline/executor.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/executor.js +133 -0
- package/dist/smart-agent/pipeline/executor.js.map +1 -0
- package/dist/smart-agent/pipeline/handlers/assemble.d.ts +16 -0
- package/dist/smart-agent/pipeline/handlers/assemble.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/handlers/assemble.js +57 -0
- package/dist/smart-agent/pipeline/handlers/assemble.js.map +1 -0
- package/dist/smart-agent/pipeline/handlers/classify.d.ts +21 -0
- package/dist/smart-agent/pipeline/handlers/classify.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/handlers/classify.js +53 -0
- package/dist/smart-agent/pipeline/handlers/classify.js.map +1 -0
- package/dist/smart-agent/pipeline/handlers/expand.d.ts +16 -0
- package/dist/smart-agent/pipeline/handlers/expand.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/handlers/expand.js +24 -0
- package/dist/smart-agent/pipeline/handlers/expand.js.map +1 -0
- package/dist/smart-agent/pipeline/handlers/index.d.ts +28 -0
- package/dist/smart-agent/pipeline/handlers/index.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/handlers/index.js +39 -0
- package/dist/smart-agent/pipeline/handlers/index.js.map +1 -0
- package/dist/smart-agent/pipeline/handlers/rag-query.d.ts +25 -0
- package/dist/smart-agent/pipeline/handlers/rag-query.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/handlers/rag-query.js +53 -0
- package/dist/smart-agent/pipeline/handlers/rag-query.js.map +1 -0
- package/dist/smart-agent/pipeline/handlers/rag-upsert.d.ts +17 -0
- package/dist/smart-agent/pipeline/handlers/rag-upsert.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/handlers/rag-upsert.js +47 -0
- package/dist/smart-agent/pipeline/handlers/rag-upsert.js.map +1 -0
- package/dist/smart-agent/pipeline/handlers/rerank.d.ts +16 -0
- package/dist/smart-agent/pipeline/handlers/rerank.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/handlers/rerank.js +38 -0
- package/dist/smart-agent/pipeline/handlers/rerank.js.map +1 -0
- package/dist/smart-agent/pipeline/handlers/summarize.d.ts +17 -0
- package/dist/smart-agent/pipeline/handlers/summarize.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/handlers/summarize.js +47 -0
- package/dist/smart-agent/pipeline/handlers/summarize.js.map +1 -0
- package/dist/smart-agent/pipeline/handlers/tool-loop.d.ts +32 -0
- package/dist/smart-agent/pipeline/handlers/tool-loop.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/handlers/tool-loop.js +416 -0
- package/dist/smart-agent/pipeline/handlers/tool-loop.js.map +1 -0
- package/dist/smart-agent/pipeline/handlers/tool-select.d.ts +21 -0
- package/dist/smart-agent/pipeline/handlers/tool-select.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/handlers/tool-select.js +95 -0
- package/dist/smart-agent/pipeline/handlers/tool-select.js.map +1 -0
- package/dist/smart-agent/pipeline/handlers/translate.d.ts +18 -0
- package/dist/smart-agent/pipeline/handlers/translate.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/handlers/translate.js +39 -0
- package/dist/smart-agent/pipeline/handlers/translate.js.map +1 -0
- package/dist/smart-agent/pipeline/index.d.ts +14 -0
- package/dist/smart-agent/pipeline/index.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/index.js +16 -0
- package/dist/smart-agent/pipeline/index.js.map +1 -0
- package/dist/smart-agent/pipeline/stage-handler.d.ts +36 -0
- package/dist/smart-agent/pipeline/stage-handler.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/stage-handler.js +24 -0
- package/dist/smart-agent/pipeline/stage-handler.js.map +1 -0
- package/dist/smart-agent/pipeline/types.d.ts +129 -0
- package/dist/smart-agent/pipeline/types.d.ts.map +1 -0
- package/dist/smart-agent/pipeline/types.js +54 -0
- package/dist/smart-agent/pipeline/types.js.map +1 -0
- package/dist/smart-agent/pipeline.d.ts +22 -6
- package/dist/smart-agent/pipeline.d.ts.map +1 -1
- package/dist/smart-agent/pipeline.js +5 -142
- package/dist/smart-agent/pipeline.js.map +1 -1
- package/dist/smart-agent/plugins/index.d.ts +4 -0
- package/dist/smart-agent/plugins/index.d.ts.map +1 -0
- package/dist/smart-agent/plugins/index.js +3 -0
- package/dist/smart-agent/plugins/index.js.map +1 -0
- package/dist/smart-agent/plugins/loader.d.ts +67 -0
- package/dist/smart-agent/plugins/loader.d.ts.map +1 -0
- package/dist/smart-agent/plugins/loader.js +114 -0
- package/dist/smart-agent/plugins/loader.js.map +1 -0
- package/dist/smart-agent/plugins/types.d.ts +149 -0
- package/dist/smart-agent/plugins/types.d.ts.map +1 -0
- package/dist/smart-agent/plugins/types.js +111 -0
- package/dist/smart-agent/plugins/types.js.map +1 -0
- package/dist/smart-agent/providers.d.ts +76 -0
- package/dist/smart-agent/providers.d.ts.map +1 -0
- package/dist/smart-agent/providers.js +173 -0
- package/dist/smart-agent/providers.js.map +1 -0
- package/dist/smart-agent/rag/embedder-factories.d.ts +9 -0
- package/dist/smart-agent/rag/embedder-factories.d.ts.map +1 -0
- package/dist/smart-agent/rag/embedder-factories.js +27 -0
- package/dist/smart-agent/rag/embedder-factories.js.map +1 -0
- package/dist/smart-agent/rag/index.d.ts +7 -0
- package/dist/smart-agent/rag/index.d.ts.map +1 -1
- package/dist/smart-agent/rag/index.js +4 -0
- package/dist/smart-agent/rag/index.js.map +1 -1
- package/dist/smart-agent/smart-server.d.ts +25 -1
- package/dist/smart-agent/smart-server.d.ts.map +1 -1
- package/dist/smart-agent/smart-server.js +122 -27
- package/dist/smart-agent/smart-server.js.map +1 -1
- package/package.json +2 -1
|
@@ -0,0 +1,416 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ToolLoopHandler — streaming LLM call + MCP tool execution loop.
|
|
3
|
+
*
|
|
4
|
+
* This is the "terminal" pipeline stage. It takes over the streaming yield
|
|
5
|
+
* mechanism to push SSE chunks back to the consumer.
|
|
6
|
+
*
|
|
7
|
+
* Reads: `ctx.assembledMessages`, `ctx.activeTools`, `ctx.toolClientMap`,
|
|
8
|
+
* `ctx.externalTools`, `ctx.mainLlm`
|
|
9
|
+
* Writes: yields chunks via `ctx.yield()`, updates `ctx.timing`
|
|
10
|
+
*
|
|
11
|
+
* ## Config
|
|
12
|
+
*
|
|
13
|
+
* | Field | Type | Default | Description |
|
|
14
|
+
* |--------------------|--------|-------------|---------------------------------|
|
|
15
|
+
* | `maxIterations` | number | from ctx | Max tool-loop iterations |
|
|
16
|
+
* | `maxToolCalls` | number | from ctx | Max total tool calls per request|
|
|
17
|
+
* | `heartbeatIntervalMs` | number | 5000 | SSE heartbeat interval (ms) |
|
|
18
|
+
*
|
|
19
|
+
* ## Includes
|
|
20
|
+
*
|
|
21
|
+
* - Output validation (re-prompts on invalid LLM output)
|
|
22
|
+
* - Tool call classification (internal / external / hallucinated / blocked)
|
|
23
|
+
* - Concurrent tool execution with heartbeat
|
|
24
|
+
* - Tool availability tracking (temporary blacklist)
|
|
25
|
+
*/
|
|
26
|
+
import { OrchestratorError } from '../../agent.js';
|
|
27
|
+
import { isToolContextUnavailableError } from '../../policy/tool-availability-registry.js';
|
|
28
|
+
import { getStreamToolCallName, toToolCallDelta, } from '../../utils/tool-call-deltas.js';
|
|
29
|
+
export class ToolLoopHandler {
|
|
30
|
+
async execute(ctx, config, parentSpan) {
|
|
31
|
+
const maxIterations = config.maxIterations ?? ctx.config.maxIterations;
|
|
32
|
+
const maxToolCalls = config.maxToolCalls ?? ctx.config.maxToolCalls;
|
|
33
|
+
const heartbeatMs = config.heartbeatIntervalMs ??
|
|
34
|
+
ctx.config.heartbeatIntervalMs ??
|
|
35
|
+
5000;
|
|
36
|
+
const mode = ctx.config.mode || 'smart';
|
|
37
|
+
const externalTools = mode === 'hard' ? [] : ctx.externalTools;
|
|
38
|
+
const externalToolNames = new Set(externalTools.map((t) => t.name));
|
|
39
|
+
let toolCallCount = 0;
|
|
40
|
+
let messages = ctx.assembledMessages;
|
|
41
|
+
const usage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 };
|
|
42
|
+
const timingLog = [];
|
|
43
|
+
const loopStart = Date.now();
|
|
44
|
+
let currentTools = ctx.activeTools;
|
|
45
|
+
for (let iteration = 0;; iteration++) {
|
|
46
|
+
if (ctx.options?.signal?.aborted) {
|
|
47
|
+
ctx.yield({
|
|
48
|
+
ok: false,
|
|
49
|
+
error: new OrchestratorError('Aborted', 'ABORTED'),
|
|
50
|
+
});
|
|
51
|
+
return false;
|
|
52
|
+
}
|
|
53
|
+
if (iteration >= maxIterations) {
|
|
54
|
+
timingLog.push({ phase: 'total', duration: Date.now() - loopStart });
|
|
55
|
+
ctx.timing.push(...timingLog);
|
|
56
|
+
ctx.yield({
|
|
57
|
+
ok: true,
|
|
58
|
+
value: {
|
|
59
|
+
content: '',
|
|
60
|
+
finishReason: 'length',
|
|
61
|
+
usage,
|
|
62
|
+
timing: timingLog,
|
|
63
|
+
},
|
|
64
|
+
});
|
|
65
|
+
return true;
|
|
66
|
+
}
|
|
67
|
+
// Filter tools per iteration
|
|
68
|
+
const filteredForIteration = ctx.toolAvailabilityRegistry.filterTools(ctx.sessionId, currentTools);
|
|
69
|
+
currentTools = filteredForIteration.allowed;
|
|
70
|
+
if (filteredForIteration.blocked.length > 0) {
|
|
71
|
+
ctx.options?.sessionLogger?.logStep('active_tools_filtered_in_iteration', { iteration: iteration + 1, blocked: filteredForIteration.blocked });
|
|
72
|
+
}
|
|
73
|
+
ctx.options?.sessionLogger?.logStep(`llm_request_iter_${iteration + 1}`, {
|
|
74
|
+
messages,
|
|
75
|
+
tools: currentTools,
|
|
76
|
+
});
|
|
77
|
+
const llmSpan = ctx.tracer.startSpan('smart_agent.llm_call', {
|
|
78
|
+
parent: parentSpan,
|
|
79
|
+
attributes: { 'llm.iteration': iteration + 1 },
|
|
80
|
+
});
|
|
81
|
+
ctx.metrics.llmCallCount.add();
|
|
82
|
+
const llmCallStart = Date.now();
|
|
83
|
+
const stream = ctx.mainLlm.streamChat(messages, currentTools, ctx.options);
|
|
84
|
+
let content = '';
|
|
85
|
+
let finishReason;
|
|
86
|
+
const toolCallsMap = new Map();
|
|
87
|
+
for await (const chunkResult of stream) {
|
|
88
|
+
if (!chunkResult.ok) {
|
|
89
|
+
llmSpan.setStatus('error', chunkResult.error.message);
|
|
90
|
+
llmSpan.end();
|
|
91
|
+
ctx.yield({
|
|
92
|
+
ok: false,
|
|
93
|
+
error: new OrchestratorError(chunkResult.error.message, 'LLM_ERROR'),
|
|
94
|
+
});
|
|
95
|
+
return false;
|
|
96
|
+
}
|
|
97
|
+
const chunk = chunkResult.value;
|
|
98
|
+
if (chunk.content) {
|
|
99
|
+
content += chunk.content;
|
|
100
|
+
ctx.yield({ ok: true, value: { content: chunk.content } });
|
|
101
|
+
}
|
|
102
|
+
if (chunk.toolCalls) {
|
|
103
|
+
const externalDeltas = chunk.toolCalls.filter((tc) => externalToolNames.has(getStreamToolCallName(tc) ?? ''));
|
|
104
|
+
if (externalDeltas.length > 0) {
|
|
105
|
+
ctx.yield({
|
|
106
|
+
ok: true,
|
|
107
|
+
value: { content: '', toolCalls: externalDeltas },
|
|
108
|
+
});
|
|
109
|
+
}
|
|
110
|
+
for (const [fallbackIndex, rawToolCall,] of chunk.toolCalls.entries()) {
|
|
111
|
+
const tc = toToolCallDelta(rawToolCall, fallbackIndex);
|
|
112
|
+
if (!toolCallsMap.has(tc.index)) {
|
|
113
|
+
toolCallsMap.set(tc.index, {
|
|
114
|
+
id: tc.id || '',
|
|
115
|
+
name: tc.name || '',
|
|
116
|
+
arguments: tc.arguments || '',
|
|
117
|
+
});
|
|
118
|
+
}
|
|
119
|
+
else {
|
|
120
|
+
const ex = toolCallsMap.get(tc.index);
|
|
121
|
+
if (ex) {
|
|
122
|
+
if (tc.id)
|
|
123
|
+
ex.id = tc.id;
|
|
124
|
+
if (tc.name)
|
|
125
|
+
ex.name = tc.name;
|
|
126
|
+
if (tc.arguments)
|
|
127
|
+
ex.arguments += tc.arguments;
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
}
|
|
131
|
+
}
|
|
132
|
+
if (chunk.finishReason)
|
|
133
|
+
finishReason = chunk.finishReason;
|
|
134
|
+
if (chunk.usage) {
|
|
135
|
+
usage.promptTokens += chunk.usage.promptTokens;
|
|
136
|
+
usage.completionTokens += chunk.usage.completionTokens;
|
|
137
|
+
usage.totalTokens += chunk.usage.totalTokens;
|
|
138
|
+
ctx.sessionManager.addTokens(chunk.usage.totalTokens);
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
llmSpan.setStatus('ok');
|
|
142
|
+
llmSpan.end();
|
|
143
|
+
const llmCallDuration = Date.now() - llmCallStart;
|
|
144
|
+
ctx.metrics.llmCallLatency.record(llmCallDuration);
|
|
145
|
+
timingLog.push({
|
|
146
|
+
phase: `llm_call_${iteration + 1}`,
|
|
147
|
+
duration: llmCallDuration,
|
|
148
|
+
});
|
|
149
|
+
const toolCalls = Array.from(toolCallsMap.values()).map((tc) => {
|
|
150
|
+
let args = {};
|
|
151
|
+
try {
|
|
152
|
+
args = JSON.parse(tc.arguments);
|
|
153
|
+
}
|
|
154
|
+
catch {
|
|
155
|
+
args = {};
|
|
156
|
+
}
|
|
157
|
+
return { id: tc.id, name: tc.name, arguments: args };
|
|
158
|
+
});
|
|
159
|
+
ctx.options?.sessionLogger?.logStep(`llm_response_iter_${iteration + 1}`, { content, toolCalls, finishReason });
|
|
160
|
+
// -- No tool calls: validate and finish --------------------------------
|
|
161
|
+
if (finishReason !== 'tool_calls' || toolCalls.length === 0) {
|
|
162
|
+
const valResult = await ctx.outputValidator.validate(content, { messages, tools: currentTools }, ctx.options);
|
|
163
|
+
if (valResult.ok && !valResult.value.valid) {
|
|
164
|
+
const correction = valResult.value.correctedContent ?? valResult.value.reason;
|
|
165
|
+
messages = [
|
|
166
|
+
...messages,
|
|
167
|
+
{ role: 'assistant', content },
|
|
168
|
+
{
|
|
169
|
+
role: 'user',
|
|
170
|
+
content: `Your previous response was rejected by validation: ${correction}. Please try again.`,
|
|
171
|
+
},
|
|
172
|
+
];
|
|
173
|
+
continue;
|
|
174
|
+
}
|
|
175
|
+
ctx.options?.sessionLogger?.logStep('final_response', {
|
|
176
|
+
content,
|
|
177
|
+
usage,
|
|
178
|
+
});
|
|
179
|
+
timingLog.push({ phase: 'total', duration: Date.now() - loopStart });
|
|
180
|
+
ctx.timing.push(...timingLog);
|
|
181
|
+
ctx.yield({
|
|
182
|
+
ok: true,
|
|
183
|
+
value: {
|
|
184
|
+
content: '',
|
|
185
|
+
finishReason: finishReason || 'stop',
|
|
186
|
+
usage,
|
|
187
|
+
timing: timingLog,
|
|
188
|
+
},
|
|
189
|
+
});
|
|
190
|
+
return true;
|
|
191
|
+
}
|
|
192
|
+
// -- Classify tool calls -----------------------------------------------
|
|
193
|
+
const internalCalls = toolCalls.filter((tc) => ctx.toolClientMap.has(tc.name));
|
|
194
|
+
const validExternalCalls = toolCalls.filter((tc) => externalToolNames.has(tc.name));
|
|
195
|
+
const blockedToolNames = ctx.toolAvailabilityRegistry.getBlockedToolNames(ctx.sessionId);
|
|
196
|
+
const blockedCalls = toolCalls.filter((tc) => blockedToolNames.has(tc.name));
|
|
197
|
+
const hallucinations = toolCalls.filter((tc) => !blockedToolNames.has(tc.name) &&
|
|
198
|
+
!ctx.toolClientMap.has(tc.name) &&
|
|
199
|
+
!externalToolNames.has(tc.name));
|
|
200
|
+
// -- Handle blocked tools ----------------------------------------------
|
|
201
|
+
if (blockedCalls.length > 0) {
|
|
202
|
+
messages = [
|
|
203
|
+
...messages,
|
|
204
|
+
{
|
|
205
|
+
role: 'assistant',
|
|
206
|
+
content: content || null,
|
|
207
|
+
tool_calls: blockedCalls.map((tc) => ({
|
|
208
|
+
id: tc.id,
|
|
209
|
+
type: 'function',
|
|
210
|
+
function: {
|
|
211
|
+
name: tc.name,
|
|
212
|
+
arguments: JSON.stringify(tc.arguments),
|
|
213
|
+
},
|
|
214
|
+
})),
|
|
215
|
+
},
|
|
216
|
+
];
|
|
217
|
+
for (const blocked of blockedCalls) {
|
|
218
|
+
messages = [
|
|
219
|
+
...messages,
|
|
220
|
+
{
|
|
221
|
+
role: 'tool',
|
|
222
|
+
content: `Error: Tool "${blocked.name}" is temporarily unavailable in this session.`,
|
|
223
|
+
tool_call_id: blocked.id,
|
|
224
|
+
},
|
|
225
|
+
];
|
|
226
|
+
}
|
|
227
|
+
ctx.options?.sessionLogger?.logStep('blocked_tool_calls_intercepted', {
|
|
228
|
+
toolNames: blockedCalls.map((tc) => tc.name),
|
|
229
|
+
});
|
|
230
|
+
continue;
|
|
231
|
+
}
|
|
232
|
+
// -- Handle hallucinated tools -----------------------------------------
|
|
233
|
+
if (hallucinations.length > 0) {
|
|
234
|
+
messages = [
|
|
235
|
+
...messages,
|
|
236
|
+
{
|
|
237
|
+
role: 'assistant',
|
|
238
|
+
content: content || null,
|
|
239
|
+
tool_calls: toolCalls.map((tc) => ({
|
|
240
|
+
id: tc.id,
|
|
241
|
+
type: 'function',
|
|
242
|
+
function: {
|
|
243
|
+
name: tc.name,
|
|
244
|
+
arguments: JSON.stringify(tc.arguments),
|
|
245
|
+
},
|
|
246
|
+
})),
|
|
247
|
+
},
|
|
248
|
+
];
|
|
249
|
+
for (const h of hallucinations) {
|
|
250
|
+
messages = [
|
|
251
|
+
...messages,
|
|
252
|
+
{
|
|
253
|
+
role: 'tool',
|
|
254
|
+
content: `Error: Tool "${h.name}" not found.`,
|
|
255
|
+
tool_call_id: h.id,
|
|
256
|
+
},
|
|
257
|
+
];
|
|
258
|
+
}
|
|
259
|
+
continue;
|
|
260
|
+
}
|
|
261
|
+
// -- Handle external tool calls (delegate to consumer) -----------------
|
|
262
|
+
if (validExternalCalls.length > 0) {
|
|
263
|
+
timingLog.push({ phase: 'total', duration: Date.now() - loopStart });
|
|
264
|
+
ctx.timing.push(...timingLog);
|
|
265
|
+
ctx.yield({
|
|
266
|
+
ok: true,
|
|
267
|
+
value: {
|
|
268
|
+
content: '',
|
|
269
|
+
finishReason: 'tool_calls',
|
|
270
|
+
usage,
|
|
271
|
+
timing: timingLog,
|
|
272
|
+
},
|
|
273
|
+
});
|
|
274
|
+
return true;
|
|
275
|
+
}
|
|
276
|
+
// -- Execute internal MCP tool calls -----------------------------------
|
|
277
|
+
if (content || internalCalls.length > 0) {
|
|
278
|
+
messages = [
|
|
279
|
+
...messages,
|
|
280
|
+
{
|
|
281
|
+
role: 'assistant',
|
|
282
|
+
content: content || null,
|
|
283
|
+
tool_calls: internalCalls.map((tc) => ({
|
|
284
|
+
id: tc.id,
|
|
285
|
+
type: 'function',
|
|
286
|
+
function: {
|
|
287
|
+
name: tc.name,
|
|
288
|
+
arguments: JSON.stringify(tc.arguments),
|
|
289
|
+
},
|
|
290
|
+
})),
|
|
291
|
+
},
|
|
292
|
+
];
|
|
293
|
+
}
|
|
294
|
+
// Check tool call budget
|
|
295
|
+
const remaining = maxToolCalls !== undefined
|
|
296
|
+
? maxToolCalls - toolCallCount
|
|
297
|
+
: internalCalls.length;
|
|
298
|
+
if (remaining <= 0) {
|
|
299
|
+
timingLog.push({ phase: 'total', duration: Date.now() - loopStart });
|
|
300
|
+
ctx.timing.push(...timingLog);
|
|
301
|
+
ctx.yield({
|
|
302
|
+
ok: true,
|
|
303
|
+
value: {
|
|
304
|
+
content: '',
|
|
305
|
+
finishReason: 'length',
|
|
306
|
+
usage,
|
|
307
|
+
timing: timingLog,
|
|
308
|
+
},
|
|
309
|
+
});
|
|
310
|
+
return true;
|
|
311
|
+
}
|
|
312
|
+
const batch = internalCalls.slice(0, remaining);
|
|
313
|
+
// Yield progress messages
|
|
314
|
+
for (const tc of batch) {
|
|
315
|
+
ctx.yield({
|
|
316
|
+
ok: true,
|
|
317
|
+
value: { content: `\n\n[SmartAgent: Executing ${tc.name}...]\n` },
|
|
318
|
+
});
|
|
319
|
+
}
|
|
320
|
+
const toolExecPromises = batch.map(async (tc) => {
|
|
321
|
+
const toolStart = Date.now();
|
|
322
|
+
ctx.options?.sessionLogger?.logStep(`mcp_call_${tc.name}`, {
|
|
323
|
+
arguments: tc.arguments,
|
|
324
|
+
});
|
|
325
|
+
const client = ctx.toolClientMap.get(tc.name);
|
|
326
|
+
if (!client)
|
|
327
|
+
return { tc, text: '', res: null, duration: 0 };
|
|
328
|
+
const toolSpan = ctx.tracer.startSpan('smart_agent.tool_call', {
|
|
329
|
+
parent: parentSpan,
|
|
330
|
+
attributes: { 'tool.name': tc.name },
|
|
331
|
+
});
|
|
332
|
+
const cached = ctx.toolCache.get(tc.name, tc.arguments);
|
|
333
|
+
const res = cached
|
|
334
|
+
? (() => {
|
|
335
|
+
ctx.metrics.toolCacheHitCount.add();
|
|
336
|
+
toolSpan.setAttribute('cache', 'hit');
|
|
337
|
+
return { ok: true, value: cached };
|
|
338
|
+
})()
|
|
339
|
+
: await (async () => {
|
|
340
|
+
const r = await client.callTool(tc.name, tc.arguments, ctx.options);
|
|
341
|
+
if (r.ok)
|
|
342
|
+
ctx.toolCache.set(tc.name, tc.arguments, r.value);
|
|
343
|
+
return r;
|
|
344
|
+
})();
|
|
345
|
+
const text = !res.ok
|
|
346
|
+
? res.error.message
|
|
347
|
+
: typeof res.value.content === 'string'
|
|
348
|
+
? res.value.content
|
|
349
|
+
: JSON.stringify(res.value.content);
|
|
350
|
+
toolSpan.setStatus(res.ok ? 'ok' : 'error', res.ok ? undefined : text);
|
|
351
|
+
toolSpan.end();
|
|
352
|
+
return { tc, text, res, duration: Date.now() - toolStart };
|
|
353
|
+
});
|
|
354
|
+
// Race: tool execution vs periodic heartbeat
|
|
355
|
+
const allDone = Promise.all(toolExecPromises);
|
|
356
|
+
const pendingTools = new Set(batch.map((tc) => tc.name));
|
|
357
|
+
const toolStartTime = Date.now();
|
|
358
|
+
let results = [];
|
|
359
|
+
let settled = false;
|
|
360
|
+
for (const [i, p] of toolExecPromises.entries()) {
|
|
361
|
+
p.then(() => pendingTools.delete(batch[i].name));
|
|
362
|
+
}
|
|
363
|
+
while (!settled) {
|
|
364
|
+
const winner = await Promise.race([
|
|
365
|
+
allDone.then((r) => ({ tag: 'done', results: r })),
|
|
366
|
+
new Promise((resolve) => setTimeout(() => resolve({ tag: 'tick' }), heartbeatMs)),
|
|
367
|
+
]);
|
|
368
|
+
if (winner.tag === 'done') {
|
|
369
|
+
results = winner.results;
|
|
370
|
+
settled = true;
|
|
371
|
+
}
|
|
372
|
+
else {
|
|
373
|
+
for (const tool of pendingTools) {
|
|
374
|
+
ctx.yield({
|
|
375
|
+
ok: true,
|
|
376
|
+
value: {
|
|
377
|
+
content: '',
|
|
378
|
+
heartbeat: { tool, elapsed: Date.now() - toolStartTime },
|
|
379
|
+
},
|
|
380
|
+
});
|
|
381
|
+
}
|
|
382
|
+
}
|
|
383
|
+
}
|
|
384
|
+
// Collect timing
|
|
385
|
+
for (const r of results) {
|
|
386
|
+
timingLog.push({ phase: `tool_${r.tc.name}`, duration: r.duration });
|
|
387
|
+
}
|
|
388
|
+
// Process results
|
|
389
|
+
const toolMessages = [];
|
|
390
|
+
for (const { tc, text, res } of results) {
|
|
391
|
+
if (!res)
|
|
392
|
+
continue;
|
|
393
|
+
if (!res.ok && isToolContextUnavailableError(text)) {
|
|
394
|
+
const entry = ctx.toolAvailabilityRegistry.block(ctx.sessionId, tc.name, text);
|
|
395
|
+
currentTools = currentTools.filter((t) => t.name !== tc.name);
|
|
396
|
+
ctx.options?.sessionLogger?.logStep(`tool_blacklisted_${tc.name}`, {
|
|
397
|
+
reason: text,
|
|
398
|
+
blockedUntil: entry.blockedUntil,
|
|
399
|
+
});
|
|
400
|
+
}
|
|
401
|
+
ctx.options?.sessionLogger?.logStep(`mcp_result_${tc.name}`, {
|
|
402
|
+
result: text,
|
|
403
|
+
});
|
|
404
|
+
toolCallCount++;
|
|
405
|
+
ctx.metrics.toolCallCount.add();
|
|
406
|
+
toolMessages.push({
|
|
407
|
+
role: 'tool',
|
|
408
|
+
content: text,
|
|
409
|
+
tool_call_id: tc.id,
|
|
410
|
+
});
|
|
411
|
+
}
|
|
412
|
+
messages = [...messages, ...toolMessages];
|
|
413
|
+
}
|
|
414
|
+
}
|
|
415
|
+
}
|
|
416
|
+
//# sourceMappingURL=tool-loop.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-loop.js","sourceRoot":"","sources":["../../../../src/smart-agent/pipeline/handlers/tool-loop.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AAGH,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAOnD,OAAO,EAAE,6BAA6B,EAAE,MAAM,4CAA4C,CAAC;AAE3F,OAAO,EACL,qBAAqB,EACrB,eAAe,GAChB,MAAM,iCAAiC,CAAC;AAIzC,MAAM,OAAO,eAAe;IAC1B,KAAK,CAAC,OAAO,CACX,GAAoB,EACpB,MAA+B,EAC/B,UAAiB;QAEjB,MAAM,aAAa,GAChB,MAAM,CAAC,aAAwB,IAAI,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC;QAC/D,MAAM,YAAY,GACf,MAAM,CAAC,YAAuB,IAAI,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC;QAC7D,MAAM,WAAW,GACd,MAAM,CAAC,mBAA8B;YACtC,GAAG,CAAC,MAAM,CAAC,mBAAmB;YAC9B,IAAI,CAAC;QAEP,MAAM,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,OAAO,CAAC;QACxC,MAAM,aAAa,GAAG,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC;QAC/D,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAEpE,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,IAAI,QAAQ,GAAG,GAAG,CAAC,iBAAiB,CAAC;QACrC,MAAM,KAAK,GAAG,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;QACvE,MAAM,SAAS,GAAkB,EAAE,CAAC;QACpC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,YAAY,GAAc,GAAG,CAAC,WAAW,CAAC;QAE9C,KAAK,IAAI,SAAS,GAAG,CAAC,GAAI,SAAS,EAAE,EAAE,CAAC;YACtC,IAAI,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;gBACjC,GAAG,CAAC,KAAK,CAAC;oBACR,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE,IAAI,iBAAiB,CAAC,SAAS,EAAE,SAAS,CAAC;iBACnD,CAAC,CAAC;gBACH,OAAO,KAAK,CAAC;YACf,CAAC;YACD,IAAI,SAAS,IAAI,aAAa,EAAE,CAAC;gBAC/B,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC;gBACrE,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;gBAC9B,GAAG,CAAC,KAAK,CAAC;oBACR,EAAE,EAAE,IAAI;oBACR,KAAK,EAAE;wBACL,OAAO,EAAE,EAAE;wBACX,YAAY,EAAE,QAAQ;wBACtB,KAAK;wBACL,MAAM,EAAE,SAAS;qBAClB;iBACF,CAAC,CAAC;gBACH,OAAO,IAAI,CAAC;YACd,CAAC;YAED,6BAA6B;YAC7B,MAAM,oBAAoB,GAAG,GAAG,CAAC,wBAAwB,CAAC,WAAW,CACnE,GAAG,CAAC,SAAS,EACb,YAAY,CACb,CAAC;YACF,YAAY,GAAG,oBAAoB,CAAC,OAAO,CAAC;YAC5C,IAAI,oBAAoB,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5C,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CACjC,oCAAoC,EACpC,EAAE,SAAS,EAAE,SAAS,GAAG,CAAC,EAAE,OAAO,EAAE,oBAAoB,CAAC,OAAO,EAAE,CACpE,CAAC;YACJ,CAAC;YAED,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,oBAAoB,SAAS,GAAG,CAAC,EAAE,EAAE;gBACvE,QAAQ;gBACR,KAAK,EAAE,YAAY;aACpB,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,sBAAsB,EAAE;gBAC3D,MAAM,EAAE,UAAU;gBAClB,UAAU,EAAE,EAAE,eAAe,EAAE,SAAS,GAAG,CAAC,EAAE;aAC/C,CAAC,CAAC;YACH,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC;YAC/B,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,UAAU,CACnC,QAAQ,EACR,YAAY,EACZ,GAAG,CAAC,OAAO,CACZ,CAAC;YAEF,IAAI,OAAO,GAAG,EAAE,CAAC;YACjB,IAAI,YAAyC,CAAC;YAC9C,MAAM,YAAY,GAAG,IAAI,GAAG,EAGzB,CAAC;YAEJ,IAAI,KAAK,EAAE,MAAM,WAAW,IAAI,MAAM,EAAE,CAAC;gBACvC,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC;oBACpB,OAAO,CAAC,SAAS,CAAC,OAAO,EAAE,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACtD,OAAO,CAAC,GAAG,EAAE,CAAC;oBACd,GAAG,CAAC,KAAK,CAAC;wBACR,EAAE,EAAE,KAAK;wBACT,KAAK,EAAE,IAAI,iBAAiB,CAC1B,WAAW,CAAC,KAAK,CAAC,OAAO,EACzB,WAAW,CACZ;qBACF,CAAC,CAAC;oBACH,OAAO,KAAK,CAAC;gBACf,CAAC;gBACD,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC;gBAChC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oBAClB,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC;oBACzB,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC7D,CAAC;gBACD,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;oBACpB,MAAM,cAAc,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CACnD,iBAAiB,CAAC,GAAG,CAAC,qBAAqB,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CACvD,CAAC;oBACF,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC9B,GAAG,CAAC,KAAK,CAAC;4BACR,EAAE,EAAE,IAAI;4BACR,KAAK,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,cAAc,EAAE;yBAClD,CAAC,CAAC;oBACL,CAAC;oBACD,KAAK,MAAM,CACT,aAAa,EACb,WAAW,EACZ,IAAI,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,CAAC;wBAC/B,MAAM,EAAE,GAAG,eAAe,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;wBACvD,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;4BAChC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE;gCACzB,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE;gCACf,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,EAAE;gCACnB,SAAS,EAAE,EAAE,CAAC,SAAS,IAAI,EAAE;6BAC9B,CAAC,CAAC;wBACL,CAAC;6BAAM,CAAC;4BACN,MAAM,EAAE,GAAG,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;4BACtC,IAAI,EAAE,EAAE,CAAC;gCACP,IAAI,EAAE,CAAC,EAAE;oCAAE,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;gCACzB,IAAI,EAAE,CAAC,IAAI;oCAAE,EAAE,CAAC,IAAI,GAAG,EAAE,CAAC,IAAI,CAAC;gCAC/B,IAAI,EAAE,CAAC,SAAS;oCAAE,EAAE,CAAC,SAAS,IAAI,EAAE,CAAC,SAAS,CAAC;4BACjD,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,IAAI,KAAK,CAAC,YAAY;oBAAE,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;gBAC1D,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;oBAChB,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC;oBAC/C,KAAK,CAAC,gBAAgB,IAAI,KAAK,CAAC,KAAK,CAAC,gBAAgB,CAAC;oBACvD,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;oBAC7C,GAAG,CAAC,cAAc,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;gBACxD,CAAC;YACH,CAAC;YAED,OAAO,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACxB,OAAO,CAAC,GAAG,EAAE,CAAC;YACd,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY,CAAC;YAClD,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;YACnD,SAAS,CAAC,IAAI,CAAC;gBACb,KAAK,EAAE,YAAY,SAAS,GAAG,CAAC,EAAE;gBAClC,QAAQ,EAAE,eAAe;aAC1B,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE;gBAC7D,IAAI,IAAI,GAA4B,EAAE,CAAC;gBACvC,IAAI,CAAC;oBACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;gBAClC,CAAC;gBAAC,MAAM,CAAC;oBACP,IAAI,GAAG,EAAE,CAAC;gBACZ,CAAC;gBACD,OAAO,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;YACvD,CAAC,CAAC,CAAC;YAEH,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CACjC,qBAAqB,SAAS,GAAG,CAAC,EAAE,EACpC,EAAE,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,CACrC,CAAC;YAEF,yEAAyE;YACzE,IAAI,YAAY,KAAK,YAAY,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC5D,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,eAAe,CAAC,QAAQ,CAClD,OAAO,EACP,EAAE,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,EACjC,GAAG,CAAC,OAAO,CACZ,CAAC;gBACF,IAAI,SAAS,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;oBAC3C,MAAM,UAAU,GACd,SAAS,CAAC,KAAK,CAAC,gBAAgB,IAAI,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC;oBAC7D,QAAQ,GAAG;wBACT,GAAG,QAAQ;wBACX,EAAE,IAAI,EAAE,WAAoB,EAAE,OAAO,EAAE;wBACvC;4BACE,IAAI,EAAE,MAAe;4BACrB,OAAO,EAAE,sDAAsD,UAAU,qBAAqB;yBAC/F;qBACF,CAAC;oBACF,SAAS;gBACX,CAAC;gBACD,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,gBAAgB,EAAE;oBACpD,OAAO;oBACP,KAAK;iBACN,CAAC,CAAC;gBACH,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC;gBACrE,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;gBAC9B,GAAG,CAAC,KAAK,CAAC;oBACR,EAAE,EAAE,IAAI;oBACR,KAAK,EAAE;wBACL,OAAO,EAAE,EAAE;wBACX,YAAY,EAAE,YAAY,IAAI,MAAM;wBACpC,KAAK;wBACL,MAAM,EAAE,SAAS;qBAClB;iBACF,CAAC,CAAC;gBACH,OAAO,IAAI,CAAC;YACd,CAAC;YAED,yEAAyE;YACzE,MAAM,aAAa,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAC5C,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAC/B,CAAC;YACF,MAAM,kBAAkB,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CACjD,iBAAiB,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAC/B,CAAC;YACF,MAAM,gBAAgB,GAAG,GAAG,CAAC,wBAAwB,CAAC,mBAAmB,CACvE,GAAG,CAAC,SAAS,CACd,CAAC;YACF,MAAM,YAAY,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAC3C,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAC9B,CAAC;YACF,MAAM,cAAc,GAAG,SAAS,CAAC,MAAM,CACrC,CAAC,EAAE,EAAE,EAAE,CACL,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC;gBAC9B,CAAC,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC;gBAC/B,CAAC,iBAAiB,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAClC,CAAC;YAEF,yEAAyE;YACzE,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5B,QAAQ,GAAG;oBACT,GAAG,QAAQ;oBACX;wBACE,IAAI,EAAE,WAAoB;wBAC1B,OAAO,EAAE,OAAO,IAAI,IAAI;wBACxB,UAAU,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;4BACpC,EAAE,EAAE,EAAE,CAAC,EAAE;4BACT,IAAI,EAAE,UAAmB;4BACzB,QAAQ,EAAE;gCACR,IAAI,EAAE,EAAE,CAAC,IAAI;gCACb,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC;6BACxC;yBACF,CAAC,CAAC;qBACJ;iBACF,CAAC;gBACF,KAAK,MAAM,OAAO,IAAI,YAAY,EAAE,CAAC;oBACnC,QAAQ,GAAG;wBACT,GAAG,QAAQ;wBACX;4BACE,IAAI,EAAE,MAAe;4BACrB,OAAO,EAAE,gBAAgB,OAAO,CAAC,IAAI,+CAA+C;4BACpF,YAAY,EAAE,OAAO,CAAC,EAAE;yBACzB;qBACF,CAAC;gBACJ,CAAC;gBACD,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,gCAAgC,EAAE;oBACpE,SAAS,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;iBAC7C,CAAC,CAAC;gBACH,SAAS;YACX,CAAC;YAED,yEAAyE;YACzE,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9B,QAAQ,GAAG;oBACT,GAAG,QAAQ;oBACX;wBACE,IAAI,EAAE,WAAoB;wBAC1B,OAAO,EAAE,OAAO,IAAI,IAAI;wBACxB,UAAU,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;4BACjC,EAAE,EAAE,EAAE,CAAC,EAAE;4BACT,IAAI,EAAE,UAAmB;4BACzB,QAAQ,EAAE;gCACR,IAAI,EAAE,EAAE,CAAC,IAAI;gCACb,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC;6BACxC;yBACF,CAAC,CAAC;qBACJ;iBACF,CAAC;gBACF,KAAK,MAAM,CAAC,IAAI,cAAc,EAAE,CAAC;oBAC/B,QAAQ,GAAG;wBACT,GAAG,QAAQ;wBACX;4BACE,IAAI,EAAE,MAAe;4BACrB,OAAO,EAAE,gBAAgB,CAAC,CAAC,IAAI,cAAc;4BAC7C,YAAY,EAAE,CAAC,CAAC,EAAE;yBACnB;qBACF,CAAC;gBACJ,CAAC;gBACD,SAAS;YACX,CAAC;YAED,yEAAyE;YACzE,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClC,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC;gBACrE,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;gBAC9B,GAAG,CAAC,KAAK,CAAC;oBACR,EAAE,EAAE,IAAI;oBACR,KAAK,EAAE;wBACL,OAAO,EAAE,EAAE;wBACX,YAAY,EAAE,YAAY;wBAC1B,KAAK;wBACL,MAAM,EAAE,SAAS;qBAClB;iBACF,CAAC,CAAC;gBACH,OAAO,IAAI,CAAC;YACd,CAAC;YAED,yEAAyE;YACzE,IAAI,OAAO,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxC,QAAQ,GAAG;oBACT,GAAG,QAAQ;oBACX;wBACE,IAAI,EAAE,WAAoB;wBAC1B,OAAO,EAAE,OAAO,IAAI,IAAI;wBACxB,UAAU,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;4BACrC,EAAE,EAAE,EAAE,CAAC,EAAE;4BACT,IAAI,EAAE,UAAmB;4BACzB,QAAQ,EAAE;gCACR,IAAI,EAAE,EAAE,CAAC,IAAI;gCACb,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC;6BACxC;yBACF,CAAC,CAAC;qBACJ;iBACF,CAAC;YACJ,CAAC;YAED,yBAAyB;YACzB,MAAM,SAAS,GACb,YAAY,KAAK,SAAS;gBACxB,CAAC,CAAC,YAAY,GAAG,aAAa;gBAC9B,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC;YAC3B,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;gBACnB,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC;gBACrE,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;gBAC9B,GAAG,CAAC,KAAK,CAAC;oBACR,EAAE,EAAE,IAAI;oBACR,KAAK,EAAE;wBACL,OAAO,EAAE,EAAE;wBACX,YAAY,EAAE,QAAQ;wBACtB,KAAK;wBACL,MAAM,EAAE,SAAS;qBAClB;iBACF,CAAC,CAAC;gBACH,OAAO,IAAI,CAAC;YACd,CAAC;YAED,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YAEhD,0BAA0B;YAC1B,KAAK,MAAM,EAAE,IAAI,KAAK,EAAE,CAAC;gBACvB,GAAG,CAAC,KAAK,CAAC;oBACR,EAAE,EAAE,IAAI;oBACR,KAAK,EAAE,EAAE,OAAO,EAAE,8BAA8B,EAAE,CAAC,IAAI,QAAQ,EAAE;iBAClE,CAAC,CAAC;YACL,CAAC;YAaD,MAAM,gBAAgB,GAAG,KAAK,CAAC,GAAG,CAChC,KAAK,EAAE,EAAE,EAA2B,EAAE;gBACpC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC7B,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC,IAAI,EAAE,EAAE;oBACzD,SAAS,EAAE,EAAE,CAAC,SAAS;iBACxB,CAAC,CAAC;gBACH,MAAM,MAAM,GAAG,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;gBAC9C,IAAI,CAAC,MAAM;oBAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;gBAC7D,MAAM,QAAQ,GAAG,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,uBAAuB,EAAE;oBAC7D,MAAM,EAAE,UAAU;oBAClB,UAAU,EAAE,EAAE,WAAW,EAAE,EAAE,CAAC,IAAI,EAAE;iBACrC,CAAC,CAAC;gBACH,MAAM,MAAM,GAAG,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC;gBACxD,MAAM,GAAG,GAAG,MAAM;oBAChB,CAAC,CAAC,CAAC,GAAG,EAAE;wBACJ,GAAG,CAAC,OAAO,CAAC,iBAAiB,CAAC,GAAG,EAAE,CAAC;wBACpC,QAAQ,CAAC,YAAY,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;wBACtC,OAAO,EAAE,EAAE,EAAE,IAAa,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;oBAC9C,CAAC,CAAC,EAAE;oBACN,CAAC,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE;wBAChB,MAAM,CAAC,GAAG,MAAM,MAAM,CAAC,QAAQ,CAC7B,EAAE,CAAC,IAAI,EACP,EAAE,CAAC,SAAS,EACZ,GAAG,CAAC,OAAO,CACZ,CAAC;wBACF,IAAI,CAAC,CAAC,EAAE;4BAAE,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;wBAC5D,OAAO,CAAC,CAAC;oBACX,CAAC,CAAC,EAAE,CAAC;gBACT,MAAM,IAAI,GAAG,CAAC,GAAG,CAAC,EAAE;oBAClB,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO;oBACnB,CAAC,CAAC,OAAO,GAAG,CAAC,KAAK,CAAC,OAAO,KAAK,QAAQ;wBACrC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO;wBACnB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBACxC,QAAQ,CAAC,SAAS,CAChB,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,EACvB,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAC1B,CAAC;gBACF,QAAQ,CAAC,GAAG,EAAE,CAAC;gBACf,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,CAAC;YAC7D,CAAC,CACF,CAAC;YAEF,6CAA6C;YAC7C,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YAC9C,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;YACzD,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACjC,IAAI,OAAO,GAAqB,EAAE,CAAC;YACnC,IAAI,OAAO,GAAG,KAAK,CAAC;YAEpB,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,gBAAgB,CAAC,OAAO,EAAE,EAAE,CAAC;gBAChD,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YACnD,CAAC;YAED,OAAO,CAAC,OAAO,EAAE,CAAC;gBAChB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;oBAChC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,MAAe,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;oBAC3D,IAAI,OAAO,CAAkB,CAAC,OAAO,EAAE,EAAE,CACvC,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE,WAAW,CAAC,CACxD;iBACF,CAAC,CAAC;gBACH,IAAI,MAAM,CAAC,GAAG,KAAK,MAAM,EAAE,CAAC;oBAC1B,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;oBACzB,OAAO,GAAG,IAAI,CAAC;gBACjB,CAAC;qBAAM,CAAC;oBACN,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;wBAChC,GAAG,CAAC,KAAK,CAAC;4BACR,EAAE,EAAE,IAAI;4BACR,KAAK,EAAE;gCACL,OAAO,EAAE,EAAE;gCACX,SAAS,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,aAAa,EAAE;6BACzD;yBACF,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;YAED,iBAAiB;YACjB,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBACxB,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;YACvE,CAAC;YAED,kBAAkB;YAClB,MAAM,YAAY,GAAc,EAAE,CAAC;YACnC,KAAK,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,OAAO,EAAE,CAAC;gBACxC,IAAI,CAAC,GAAG;oBAAE,SAAS;gBACnB,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,6BAA6B,CAAC,IAAI,CAAC,EAAE,CAAC;oBACnD,MAAM,KAAK,GAAG,GAAG,CAAC,wBAAwB,CAAC,KAAK,CAC9C,GAAG,CAAC,SAAS,EACb,EAAE,CAAC,IAAI,EACP,IAAI,CACL,CAAC;oBACF,YAAY,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,EAAE,CAAC,IAAI,CAAC,CAAC;oBAC9D,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,oBAAoB,EAAE,CAAC,IAAI,EAAE,EAAE;wBACjE,MAAM,EAAE,IAAI;wBACZ,YAAY,EAAE,KAAK,CAAC,YAAY;qBACjC,CAAC,CAAC;gBACL,CAAC;gBACD,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,cAAc,EAAE,CAAC,IAAI,EAAE,EAAE;oBAC3D,MAAM,EAAE,IAAI;iBACb,CAAC,CAAC;gBACH,aAAa,EAAE,CAAC;gBAChB,GAAG,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,EAAE,CAAC;gBAChC,YAAY,CAAC,IAAI,CAAC;oBAChB,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,IAAI;oBACb,YAAY,EAAE,EAAE,CAAC,EAAE;iBACpB,CAAC,CAAC;YACL,CAAC;YACD,QAAQ,GAAG,CAAC,GAAG,QAAQ,EAAE,GAAG,YAAY,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ToolSelectHandler — selects MCP tools based on RAG results.
|
|
3
|
+
*
|
|
4
|
+
* Reads: `ctx.ragResults.facts`, `ctx.mcpTools`, `ctx.externalTools`, `ctx.toolClientMap`
|
|
5
|
+
* Writes: `ctx.selectedTools`, `ctx.activeTools`
|
|
6
|
+
*
|
|
7
|
+
* Uses RAG fact IDs with the `tool:` prefix to identify relevant MCP tools.
|
|
8
|
+
*
|
|
9
|
+
* If RAG retrieval was skipped (e.g. `shouldRetrieve` was false), the handler
|
|
10
|
+
* performs its own facts RAG query to discover tools. This ensures tools are
|
|
11
|
+
* always discoverable regardless of domain context detection.
|
|
12
|
+
*
|
|
13
|
+
* Falls back to all MCP tools in `hard` mode or external-only in `smart` mode.
|
|
14
|
+
*/
|
|
15
|
+
import type { ISpan } from '../../tracer/types.js';
|
|
16
|
+
import type { PipelineContext } from '../context.js';
|
|
17
|
+
import type { IStageHandler } from '../stage-handler.js';
|
|
18
|
+
export declare class ToolSelectHandler implements IStageHandler {
|
|
19
|
+
execute(ctx: PipelineContext, config: Record<string, unknown>, span: ISpan): Promise<boolean>;
|
|
20
|
+
}
|
|
21
|
+
//# sourceMappingURL=tool-select.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-select.d.ts","sourceRoot":"","sources":["../../../../src/smart-agent/pipeline/handlers/tool-select.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAGH,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AACnD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AACrD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEzD,qBAAa,iBAAkB,YAAW,aAAa;IAC/C,OAAO,CACX,GAAG,EAAE,eAAe,EACpB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC/B,IAAI,EAAE,KAAK,GACV,OAAO,CAAC,OAAO,CAAC;CAmGpB"}
|
|
@@ -0,0 +1,95 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ToolSelectHandler — selects MCP tools based on RAG results.
|
|
3
|
+
*
|
|
4
|
+
* Reads: `ctx.ragResults.facts`, `ctx.mcpTools`, `ctx.externalTools`, `ctx.toolClientMap`
|
|
5
|
+
* Writes: `ctx.selectedTools`, `ctx.activeTools`
|
|
6
|
+
*
|
|
7
|
+
* Uses RAG fact IDs with the `tool:` prefix to identify relevant MCP tools.
|
|
8
|
+
*
|
|
9
|
+
* If RAG retrieval was skipped (e.g. `shouldRetrieve` was false), the handler
|
|
10
|
+
* performs its own facts RAG query to discover tools. This ensures tools are
|
|
11
|
+
* always discoverable regardless of domain context detection.
|
|
12
|
+
*
|
|
13
|
+
* Falls back to all MCP tools in `hard` mode or external-only in `smart` mode.
|
|
14
|
+
*/
|
|
15
|
+
export class ToolSelectHandler {
|
|
16
|
+
async execute(ctx, config, span) {
|
|
17
|
+
const mode = ctx.config.mode || 'smart';
|
|
18
|
+
// List all MCP tools if not already done
|
|
19
|
+
if (ctx.mcpTools.length === 0 && ctx.mcpClients.length > 0) {
|
|
20
|
+
const settled = await Promise.allSettled(ctx.mcpClients.map(async (client) => ({
|
|
21
|
+
client,
|
|
22
|
+
result: await client.listTools(ctx.options),
|
|
23
|
+
})));
|
|
24
|
+
for (const entry of settled) {
|
|
25
|
+
if (entry.status === 'fulfilled' && entry.value.result.ok) {
|
|
26
|
+
for (const t of entry.value.result.value) {
|
|
27
|
+
if (!ctx.toolClientMap.has(t.name)) {
|
|
28
|
+
ctx.mcpTools.push(t);
|
|
29
|
+
ctx.toolClientMap.set(t.name, entry.value.client);
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
// If RAG retrieval was skipped, query facts store directly for tool discovery.
|
|
36
|
+
// Tools should always be discoverable regardless of shouldRetrieve flag.
|
|
37
|
+
let factsResults = ctx.ragResults.facts;
|
|
38
|
+
if (factsResults.length === 0 && ctx.mcpTools.length > 0) {
|
|
39
|
+
const k = config.k ?? ctx.config.ragQueryK ?? 20;
|
|
40
|
+
const queryText = ctx.ragText || ctx.inputText;
|
|
41
|
+
const result = await ctx.ragStores.facts.query(queryText, k, ctx.options);
|
|
42
|
+
if (result.ok) {
|
|
43
|
+
factsResults = result.value;
|
|
44
|
+
// Also populate ctx.ragResults.facts so assemble can use them
|
|
45
|
+
ctx.ragResults.facts = result.value;
|
|
46
|
+
ctx.options?.sessionLogger?.logStep('tool_select_rag_fallback', {
|
|
47
|
+
reason: 'RAG retrieval was skipped, querying facts for tool discovery',
|
|
48
|
+
query: queryText.slice(0, 200),
|
|
49
|
+
k,
|
|
50
|
+
resultCount: result.value.length,
|
|
51
|
+
results: result.value.map((r) => ({
|
|
52
|
+
id: r.metadata.id,
|
|
53
|
+
score: r.score,
|
|
54
|
+
text: r.text.slice(0, 120),
|
|
55
|
+
})),
|
|
56
|
+
});
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
// Select tools based on RAG results
|
|
60
|
+
const ragToolNames = new Set(factsResults
|
|
61
|
+
.map((r) => r.metadata.id)
|
|
62
|
+
.filter((id) => id?.startsWith('tool:'))
|
|
63
|
+
.map((id) => id.slice(5)));
|
|
64
|
+
const selectedMcpTools = ragToolNames.size > 0
|
|
65
|
+
? ctx.mcpTools.filter((t) => ragToolNames.has(t.name))
|
|
66
|
+
: mode === 'hard'
|
|
67
|
+
? ctx.mcpTools
|
|
68
|
+
: [];
|
|
69
|
+
ctx.selectedTools =
|
|
70
|
+
mode === 'hard'
|
|
71
|
+
? selectedMcpTools
|
|
72
|
+
: [...selectedMcpTools, ...ctx.externalTools];
|
|
73
|
+
// Apply availability filtering
|
|
74
|
+
const filtered = ctx.toolAvailabilityRegistry.filterTools(ctx.sessionId, ctx.selectedTools);
|
|
75
|
+
ctx.activeTools = filtered.allowed;
|
|
76
|
+
if (filtered.blocked.length > 0) {
|
|
77
|
+
ctx.options?.sessionLogger?.logStep('active_tools_filtered_by_registry', {
|
|
78
|
+
blocked: filtered.blocked,
|
|
79
|
+
});
|
|
80
|
+
}
|
|
81
|
+
span.setAttribute('mcp_tools', ctx.mcpTools.length);
|
|
82
|
+
span.setAttribute('selected', ctx.selectedTools.length);
|
|
83
|
+
span.setAttribute('active', ctx.activeTools.length);
|
|
84
|
+
// Log tool selection diagnostics
|
|
85
|
+
ctx.options?.sessionLogger?.logStep('tools_selected', {
|
|
86
|
+
totalMcp: ctx.mcpTools.length,
|
|
87
|
+
ragMatchedTools: [...ragToolNames],
|
|
88
|
+
selectedCount: ctx.selectedTools.length,
|
|
89
|
+
selectedNames: ctx.selectedTools.map((t) => t.name),
|
|
90
|
+
activeCount: ctx.activeTools.length,
|
|
91
|
+
});
|
|
92
|
+
return true;
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
//# sourceMappingURL=tool-select.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-select.js","sourceRoot":"","sources":["../../../../src/smart-agent/pipeline/handlers/tool-select.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAOH,MAAM,OAAO,iBAAiB;IAC5B,KAAK,CAAC,OAAO,CACX,GAAoB,EACpB,MAA+B,EAC/B,IAAW;QAEX,MAAM,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,OAAO,CAAC;QAExC,yCAAyC;QACzC,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3D,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CACtC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;gBACpC,MAAM;gBACN,MAAM,EAAE,MAAM,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC;aAC5C,CAAC,CAAC,CACJ,CAAC;YACF,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,KAAK,CAAC,MAAM,KAAK,WAAW,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;oBAC1D,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;wBACzC,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;4BACnC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;4BACrB,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;wBACpD,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,+EAA+E;QAC/E,yEAAyE;QACzE,IAAI,YAAY,GAAG,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC;QACxC,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzD,MAAM,CAAC,GAAI,MAAM,CAAC,CAAY,IAAI,GAAG,CAAC,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC;YAC7D,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC,SAAS,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;YAE1E,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;gBACd,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC;gBAC5B,8DAA8D;gBAC9D,GAAG,CAAC,UAAU,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;gBAEpC,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,0BAA0B,EAAE;oBAC9D,MAAM,EACJ,8DAA8D;oBAChE,KAAK,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;oBAC9B,CAAC;oBACD,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM;oBAChC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAChC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE;wBACjB,KAAK,EAAE,CAAC,CAAC,KAAK;wBACd,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;qBAC3B,CAAC,CAAC;iBACJ,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,oCAAoC;QACpC,MAAM,YAAY,GAAG,IAAI,GAAG,CAC1B,YAAY;aACT,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAY,CAAC;aACnC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;aACvC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAC5B,CAAC;QAEF,MAAM,gBAAgB,GACpB,YAAY,CAAC,IAAI,GAAG,CAAC;YACnB,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACtD,CAAC,CAAC,IAAI,KAAK,MAAM;gBACf,CAAC,CAAC,GAAG,CAAC,QAAQ;gBACd,CAAC,CAAC,EAAE,CAAC;QAEX,GAAG,CAAC,aAAa;YACf,IAAI,KAAK,MAAM;gBACb,CAAC,CAAE,gBAA8B;gBACjC,CAAC,CAAC,CAAC,GAAI,gBAA8B,EAAE,GAAG,GAAG,CAAC,aAAa,CAAC,CAAC;QAEjE,+BAA+B;QAC/B,MAAM,QAAQ,GAAG,GAAG,CAAC,wBAAwB,CAAC,WAAW,CACvD,GAAG,CAAC,SAAS,EACb,GAAG,CAAC,aAAa,CAClB,CAAC;QACF,GAAG,CAAC,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC;QAEnC,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,mCAAmC,EAAE;gBACvE,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,WAAW,EAAE,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACpD,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QACxD,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,GAAG,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAEpD,iCAAiC;QACjC,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,gBAAgB,EAAE;YACpD,QAAQ,EAAE,GAAG,CAAC,QAAQ,CAAC,MAAM;YAC7B,eAAe,EAAE,CAAC,GAAG,YAAY,CAAC;YAClC,aAAa,EAAE,GAAG,CAAC,aAAa,CAAC,MAAM;YACvC,aAAa,EAAE,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;YACnD,WAAW,EAAE,GAAG,CAAC,WAAW,CAAC,MAAM;SACpC,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* TranslateHandler — translates non-ASCII RAG query text to English.
|
|
3
|
+
*
|
|
4
|
+
* Reads: `ctx.ragText`, `ctx.helperLlm` (or `ctx.mainLlm` as fallback)
|
|
5
|
+
* Writes: `ctx.ragText`
|
|
6
|
+
*
|
|
7
|
+
* Skips translation when:
|
|
8
|
+
* - `ragTranslationEnabled` is false
|
|
9
|
+
* - Text is ASCII-only
|
|
10
|
+
* - Text is shorter than 15 characters
|
|
11
|
+
*/
|
|
12
|
+
import type { ISpan } from '../../tracer/types.js';
|
|
13
|
+
import type { PipelineContext } from '../context.js';
|
|
14
|
+
import type { IStageHandler } from '../stage-handler.js';
|
|
15
|
+
export declare class TranslateHandler implements IStageHandler {
|
|
16
|
+
execute(ctx: PipelineContext, _config: Record<string, unknown>, span: ISpan): Promise<boolean>;
|
|
17
|
+
}
|
|
18
|
+
//# sourceMappingURL=translate.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"translate.d.ts","sourceRoot":"","sources":["../../../../src/smart-agent/pipeline/handlers/translate.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AACnD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AACrD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEzD,qBAAa,gBAAiB,YAAW,aAAa;IAC9C,OAAO,CACX,GAAG,EAAE,eAAe,EACpB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAChC,IAAI,EAAE,KAAK,GACV,OAAO,CAAC,OAAO,CAAC;CAoCpB"}
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* TranslateHandler — translates non-ASCII RAG query text to English.
|
|
3
|
+
*
|
|
4
|
+
* Reads: `ctx.ragText`, `ctx.helperLlm` (or `ctx.mainLlm` as fallback)
|
|
5
|
+
* Writes: `ctx.ragText`
|
|
6
|
+
*
|
|
7
|
+
* Skips translation when:
|
|
8
|
+
* - `ragTranslationEnabled` is false
|
|
9
|
+
* - Text is ASCII-only
|
|
10
|
+
* - Text is shorter than 15 characters
|
|
11
|
+
*/
|
|
12
|
+
export class TranslateHandler {
|
|
13
|
+
async execute(ctx, _config, span) {
|
|
14
|
+
if (ctx.config.ragTranslationEnabled === false) {
|
|
15
|
+
span.setAttribute('skipped', true);
|
|
16
|
+
span.setAttribute('reason', 'disabled');
|
|
17
|
+
return true;
|
|
18
|
+
}
|
|
19
|
+
ctx.isAscii = /^[\p{ASCII}]+$/u.test(ctx.ragText);
|
|
20
|
+
if (ctx.isAscii || ctx.ragText.length < 15) {
|
|
21
|
+
span.setAttribute('skipped', true);
|
|
22
|
+
span.setAttribute('reason', ctx.isAscii ? 'ascii' : 'too_short');
|
|
23
|
+
return true;
|
|
24
|
+
}
|
|
25
|
+
const prompt = ctx.config.ragTranslatePrompt ??
|
|
26
|
+
'Translate the user request to English for search purposes. Preserve technical terms if present. Reply with only the expanded English terms, no explanation.';
|
|
27
|
+
const llm = ctx.helperLlm || ctx.mainLlm;
|
|
28
|
+
const res = await llm.chat([
|
|
29
|
+
{ role: 'system', content: prompt },
|
|
30
|
+
{ role: 'user', content: ctx.ragText },
|
|
31
|
+
], [], ctx.options);
|
|
32
|
+
if (res.ok && res.value.content.trim()) {
|
|
33
|
+
ctx.ragText = res.value.content.trim();
|
|
34
|
+
span.setAttribute('translated', true);
|
|
35
|
+
}
|
|
36
|
+
return true;
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
//# sourceMappingURL=translate.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"translate.js","sourceRoot":"","sources":["../../../../src/smart-agent/pipeline/handlers/translate.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAMH,MAAM,OAAO,gBAAgB;IAC3B,KAAK,CAAC,OAAO,CACX,GAAoB,EACpB,OAAgC,EAChC,IAAW;QAEX,IAAI,GAAG,CAAC,MAAM,CAAC,qBAAqB,KAAK,KAAK,EAAE,CAAC;YAC/C,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YACnC,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACxC,OAAO,IAAI,CAAC;QACd,CAAC;QAED,GAAG,CAAC,OAAO,GAAG,iBAAiB,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAElD,IAAI,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC,OAAO,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YACnC,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;YACjE,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,MAAM,GACV,GAAG,CAAC,MAAM,CAAC,kBAAkB;YAC7B,6JAA6J,CAAC;QAEhK,MAAM,GAAG,GAAG,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,OAAO,CAAC;QACzC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,IAAI,CACxB;YACE,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,MAAM,EAAE;YAC5C,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE;SAChD,EACD,EAAE,EACF,GAAG,CAAC,OAAO,CACZ,CAAC;QAEF,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;YACvC,GAAG,CAAC,OAAO,GAAG,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YACvC,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QACxC,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Structured pipeline — public API.
|
|
3
|
+
*
|
|
4
|
+
* Exports the pipeline DSL types, executor, context, condition evaluator,
|
|
5
|
+
* default pipeline definition, and all built-in stage handlers.
|
|
6
|
+
*/
|
|
7
|
+
export { evaluateCondition } from './condition-evaluator.js';
|
|
8
|
+
export type { PipelineContext } from './context.js';
|
|
9
|
+
export { getDefaultPipelineDefinition, getDefaultStages, } from './default-pipeline.js';
|
|
10
|
+
export { PipelineExecutor } from './executor.js';
|
|
11
|
+
export { AssembleHandler, buildDefaultHandlerRegistry, ClassifyHandler, ExpandHandler, RagQueryHandler, RagUpsertHandler, RerankHandler, type StageHandlerRegistry, SummarizeHandler, ToolLoopHandler, ToolSelectHandler, TranslateHandler, } from './handlers/index.js';
|
|
12
|
+
export type { IStageHandler } from './stage-handler.js';
|
|
13
|
+
export type { BuiltInStageType, ControlFlowType, StageDefinition, StageType, StructuredPipelineDefinition, } from './types.js';
|
|
14
|
+
//# sourceMappingURL=index.d.ts.map
|