@downcity/agent 1.1.79 → 1.1.81
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 +1 -1
- package/bin/agent/local/Agent.d.ts +8 -121
- package/bin/agent/local/Agent.d.ts.map +1 -1
- package/bin/agent/local/Agent.js +63 -381
- package/bin/agent/local/Agent.js.map +1 -1
- package/bin/agent/local/AgentRuntimeFactory.d.ts +2 -2
- package/bin/agent/local/AgentRuntimeFactory.d.ts.map +1 -1
- package/bin/agent/local/AgentRuntimeFactory.js +2 -23
- package/bin/agent/local/AgentRuntimeFactory.js.map +1 -1
- package/bin/agent/local/services/AgentAssemblyService.d.ts +112 -0
- package/bin/agent/local/services/AgentAssemblyService.d.ts.map +1 -0
- package/bin/agent/local/services/AgentAssemblyService.js +135 -0
- package/bin/agent/local/services/AgentAssemblyService.js.map +1 -0
- package/bin/agent/local/services/AgentLifecycleService.d.ts +59 -0
- package/bin/agent/local/services/AgentLifecycleService.d.ts.map +1 -0
- package/bin/agent/local/services/AgentLifecycleService.js +136 -0
- package/bin/agent/local/services/AgentLifecycleService.js.map +1 -0
- package/bin/agent/local/services/AgentSessionManager.d.ts +106 -0
- package/bin/agent/local/services/AgentSessionManager.d.ts.map +1 -0
- package/bin/agent/local/services/AgentSessionManager.js +182 -0
- package/bin/agent/local/services/AgentSessionManager.js.map +1 -0
- package/bin/executor/Executor.d.ts +7 -24
- package/bin/executor/Executor.d.ts.map +1 -1
- package/bin/executor/Executor.js +73 -361
- package/bin/executor/Executor.js.map +1 -1
- package/bin/executor/SessionRunScope.d.ts +18 -34
- package/bin/executor/SessionRunScope.d.ts.map +1 -1
- package/bin/executor/SessionRunScope.js +42 -28
- package/bin/executor/SessionRunScope.js.map +1 -1
- package/bin/executor/composer/context/LocalSessionContextComposer.d.ts +5 -3
- package/bin/executor/composer/context/LocalSessionContextComposer.d.ts.map +1 -1
- package/bin/executor/composer/context/LocalSessionContextComposer.js +11 -18
- package/bin/executor/composer/context/LocalSessionContextComposer.js.map +1 -1
- package/bin/executor/composer/context/SessionContextComposer.d.ts +8 -3
- package/bin/executor/composer/context/SessionContextComposer.d.ts.map +1 -1
- package/bin/executor/composer/system/SessionSystemComposer.d.ts +2 -1
- package/bin/executor/composer/system/SessionSystemComposer.d.ts.map +1 -1
- package/bin/executor/composer/system/default/DefaultSessionSystemComposer.d.ts +2 -1
- package/bin/executor/composer/system/default/DefaultSessionSystemComposer.d.ts.map +1 -1
- package/bin/executor/composer/system/default/DefaultSessionSystemComposer.js +2 -4
- package/bin/executor/composer/system/default/DefaultSessionSystemComposer.js.map +1 -1
- package/bin/executor/core-engine/CoreEngineRunner.d.ts +62 -0
- package/bin/executor/core-engine/CoreEngineRunner.d.ts.map +1 -0
- package/bin/executor/core-engine/CoreEngineRunner.js +309 -0
- package/bin/executor/core-engine/CoreEngineRunner.js.map +1 -0
- package/bin/executor/core-engine/CoreEngineUiStreamCollector.d.ts +5 -0
- package/bin/executor/core-engine/CoreEngineUiStreamCollector.d.ts.map +1 -1
- package/bin/executor/core-engine/CoreEngineUiStreamCollector.js +2 -4
- package/bin/executor/core-engine/CoreEngineUiStreamCollector.js.map +1 -1
- package/bin/executor/services/ExecutorInflightService.d.ts +39 -0
- package/bin/executor/services/ExecutorInflightService.d.ts.map +1 -0
- package/bin/executor/services/ExecutorInflightService.js +75 -0
- package/bin/executor/services/ExecutorInflightService.js.map +1 -0
- package/bin/executor/services/ExecutorRecoveryPolicy.d.ts +103 -0
- package/bin/executor/services/ExecutorRecoveryPolicy.d.ts.map +1 -0
- package/bin/executor/services/ExecutorRecoveryPolicy.js +87 -0
- package/bin/executor/services/ExecutorRecoveryPolicy.js.map +1 -0
- package/bin/executor/tools/plugin/PluginToolBridge.d.ts +19 -0
- package/bin/executor/tools/plugin/PluginToolBridge.d.ts.map +1 -0
- package/bin/executor/tools/plugin/PluginToolBridge.js +143 -0
- package/bin/executor/tools/plugin/PluginToolBridge.js.map +1 -0
- package/bin/executor/tools/plugin/PluginToolDefinition.d.ts +32 -0
- package/bin/executor/tools/plugin/PluginToolDefinition.d.ts.map +1 -0
- package/bin/executor/tools/plugin/PluginToolDefinition.js +27 -0
- package/bin/executor/tools/plugin/PluginToolDefinition.js.map +1 -0
- package/bin/executor/tools/plugin/PluginToolSchemas.d.ts +14 -0
- package/bin/executor/tools/plugin/PluginToolSchemas.d.ts.map +1 -0
- package/bin/executor/tools/plugin/PluginToolSchemas.js +19 -0
- package/bin/executor/tools/plugin/PluginToolSchemas.js.map +1 -0
- package/bin/executor/tools/plugin/types/PluginTool.d.ts +39 -0
- package/bin/executor/tools/plugin/types/PluginTool.d.ts.map +1 -0
- package/bin/executor/tools/plugin/types/PluginTool.js +9 -0
- package/bin/executor/tools/plugin/types/PluginTool.js.map +1 -0
- package/bin/executor/tools/shell/ShellToolBridge.js +3 -3
- package/bin/executor/tools/shell/ShellToolBridge.js.map +1 -1
- package/bin/executor/types/SessionRun.d.ts +18 -0
- package/bin/executor/types/SessionRun.d.ts.map +1 -1
- package/bin/index.d.ts +10 -1
- package/bin/index.d.ts.map +1 -1
- package/bin/index.js +3 -0
- package/bin/index.js.map +1 -1
- package/bin/plugin/core/BasePlugin.d.ts +2 -2
- package/bin/plugin/core/BasePlugin.d.ts.map +1 -1
- package/bin/plugin/core/BasePlugin.js.map +1 -1
- package/bin/plugin/core/ImagePlugin.d.ts +56 -0
- package/bin/plugin/core/ImagePlugin.d.ts.map +1 -0
- package/bin/plugin/core/ImagePlugin.js +109 -0
- package/bin/plugin/core/ImagePlugin.js.map +1 -0
- package/bin/session/Session.d.ts +14 -83
- package/bin/session/Session.d.ts.map +1 -1
- package/bin/session/Session.js +139 -362
- package/bin/session/Session.js.map +1 -1
- package/bin/session/SessionSystemBuilder.d.ts +2 -1
- package/bin/session/SessionSystemBuilder.d.ts.map +1 -1
- package/bin/session/SessionSystemBuilder.js +2 -3
- package/bin/session/SessionSystemBuilder.js.map +1 -1
- package/bin/session/services/SessionStateService.d.ts +132 -0
- package/bin/session/services/SessionStateService.d.ts.map +1 -0
- package/bin/session/services/SessionStateService.js +242 -0
- package/bin/session/services/SessionStateService.js.map +1 -0
- package/bin/session/services/SessionTurnService.d.ts +66 -0
- package/bin/session/services/SessionTurnService.d.ts.map +1 -0
- package/bin/session/services/SessionTurnService.js +137 -0
- package/bin/session/services/SessionTurnService.js.map +1 -0
- package/bin/session/services/SessionViewService.d.ts +105 -0
- package/bin/session/services/SessionViewService.d.ts.map +1 -0
- package/bin/session/services/SessionViewService.js +184 -0
- package/bin/session/services/SessionViewService.js.map +1 -0
- package/bin/types/agent/AgentOptions.d.ts +18 -0
- package/bin/types/agent/AgentOptions.d.ts.map +1 -1
- package/bin/types/agent/AgentTypes.d.ts +3 -1
- package/bin/types/agent/AgentTypes.d.ts.map +1 -1
- package/bin/types/agent/SessionTypes.d.ts.map +1 -1
- package/bin/types/executor/SessionRunContext.d.ts +66 -0
- package/bin/types/executor/SessionRunContext.d.ts.map +1 -0
- package/bin/types/executor/SessionRunContext.js +10 -0
- package/bin/types/executor/SessionRunContext.js.map +1 -0
- package/bin/types/plugin/ImagePlugin.d.ts +94 -0
- package/bin/types/plugin/ImagePlugin.d.ts.map +1 -0
- package/bin/types/plugin/ImagePlugin.js +10 -0
- package/bin/types/plugin/ImagePlugin.js.map +1 -0
- package/bin/types/session/SessionComposerOptions.d.ts +90 -0
- package/bin/types/session/SessionComposerOptions.d.ts.map +1 -0
- package/bin/types/session/SessionComposerOptions.js +10 -0
- package/bin/types/session/SessionComposerOptions.js.map +1 -0
- package/bin/types/session/SessionLocalState.d.ts +35 -0
- package/bin/types/session/SessionLocalState.d.ts.map +1 -0
- package/bin/types/session/SessionLocalState.js +10 -0
- package/bin/types/session/SessionLocalState.js.map +1 -0
- package/bin/types/session/SessionOptions.d.ts +85 -0
- package/bin/types/session/SessionOptions.d.ts.map +1 -0
- package/bin/types/session/SessionOptions.js +10 -0
- package/bin/types/session/SessionOptions.js.map +1 -0
- package/package.json +1 -1
- package/src/agent/local/Agent.ts +74 -433
- package/src/agent/local/AgentRuntimeFactory.ts +4 -25
- package/src/agent/local/services/AgentAssemblyService.ts +268 -0
- package/src/agent/local/services/AgentLifecycleService.ts +187 -0
- package/src/agent/local/services/AgentSessionManager.ts +291 -0
- package/src/executor/Executor.ts +103 -441
- package/src/executor/README.md +4 -4
- package/src/executor/SessionRunScope.ts +47 -71
- package/src/executor/composer/context/LocalSessionContextComposer.ts +24 -20
- package/src/executor/composer/context/SessionContextComposer.ts +13 -3
- package/src/executor/composer/system/SessionSystemComposer.ts +2 -1
- package/src/executor/composer/system/default/DefaultSessionSystemComposer.ts +3 -4
- package/src/executor/core-engine/CoreEngineRunner.ts +433 -0
- package/src/executor/core-engine/CoreEngineUiStreamCollector.ts +7 -5
- package/src/executor/services/ExecutorInflightService.ts +101 -0
- package/src/executor/services/ExecutorRecoveryPolicy.ts +213 -0
- package/src/executor/tools/plugin/PluginToolBridge.ts +161 -0
- package/src/executor/tools/plugin/PluginToolDefinition.ts +32 -0
- package/src/executor/tools/plugin/PluginToolSchemas.ts +20 -0
- package/src/executor/tools/plugin/types/PluginTool.ts +41 -0
- package/src/executor/tools/shell/ShellToolBridge.ts +3 -3
- package/src/executor/types/SessionRun.ts +20 -0
- package/src/index.ts +33 -0
- package/src/plugin/core/BasePlugin.ts +2 -2
- package/src/plugin/core/ImagePlugin.ts +128 -0
- package/src/session/Session.ts +178 -485
- package/src/session/SessionSystemBuilder.ts +3 -3
- package/src/session/services/SessionStateService.ts +341 -0
- package/src/session/services/SessionTurnService.ts +202 -0
- package/src/session/services/SessionViewService.ts +301 -0
- package/src/types/agent/AgentOptions.ts +25 -0
- package/src/types/agent/AgentTypes.ts +10 -0
- package/src/types/agent/SessionTypes.ts +1 -0
- package/src/types/executor/SessionRunContext.ts +76 -0
- package/src/types/plugin/ImagePlugin.ts +103 -0
- package/src/types/session/SessionComposerOptions.ts +107 -0
- package/src/types/session/SessionLocalState.ts +40 -0
- package/src/types/session/SessionOptions.ts +99 -0
- package/tsconfig.tsbuildinfo +1 -1
|
@@ -0,0 +1,309 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* CoreEngineRunner:模型与 tool-loop 主循环执行器。
|
|
3
|
+
*
|
|
4
|
+
* 关键点(中文)
|
|
5
|
+
* - 只负责单次已装配输入的执行,不负责 run scope、外层重试与历史准备。
|
|
6
|
+
* - 把 step 循环、续写恢复、最终 assistant 汇总等细节从 Executor 中剥离。
|
|
7
|
+
* - 保持失败返回结构稳定,避免对外 Session 行为变化。
|
|
8
|
+
*/
|
|
9
|
+
import { streamText } from "ai";
|
|
10
|
+
import { buildOpenAIResponsesProviderOptions } from "../messages/SessionMessageCodec.js";
|
|
11
|
+
import { logAssistantMessageNow } from "../messages/SessionMessageLog.js";
|
|
12
|
+
import { MAX_INCOMPLETE_RESPONSE_RECOVERIES, MAX_TEXT_ONLY_CONTINUATIONS, MAX_TOOL_LOOP_STEPS, buildIncompleteResponseRecoveryNudge, buildTextOnlyContinuationNudge, detectIncompleteResponse, detectTextOnlyContinuationReason, mergeAssistantUiMessages, summarizeStepForDebug, summarizeUiMessageForDebug, toInlinePreview, } from "../core-engine/CoreEngineSignals.js";
|
|
13
|
+
import { evaluateCoreEngineLoopDecision, shouldContinueForTailMergedUserMessages, } from "../core-engine/CoreEngineLoopDecision.js";
|
|
14
|
+
import { resolveEffectiveCoreEngineError, summarizeStreamError, } from "../core-engine/CoreEngineError.js";
|
|
15
|
+
import { collectFinalAssistantMessageFromUiStream } from "../core-engine/CoreEngineUiStreamCollector.js";
|
|
16
|
+
import { CoreEngineMessageState } from "../core-engine/CoreEngineMessageState.js";
|
|
17
|
+
/**
|
|
18
|
+
* 生成 file part 去重 key。
|
|
19
|
+
*/
|
|
20
|
+
function build_file_part_key(part) {
|
|
21
|
+
return [
|
|
22
|
+
String(part.type || ""),
|
|
23
|
+
String(part.mediaType || ""),
|
|
24
|
+
String(part.filename || ""),
|
|
25
|
+
String(part.url || ""),
|
|
26
|
+
].join("\n");
|
|
27
|
+
}
|
|
28
|
+
/**
|
|
29
|
+
* 把 tool/plugin 运行期产生的 file parts 并入最终 assistant UIMessage。
|
|
30
|
+
*/
|
|
31
|
+
function mergePendingAssistantFileParts(message, parts) {
|
|
32
|
+
if (!Array.isArray(parts) || parts.length === 0)
|
|
33
|
+
return message;
|
|
34
|
+
const current_parts = Array.isArray(message.parts) ? message.parts : [];
|
|
35
|
+
const seen = new Set();
|
|
36
|
+
for (const part of current_parts) {
|
|
37
|
+
const candidate = part;
|
|
38
|
+
if (candidate?.type !== "file")
|
|
39
|
+
continue;
|
|
40
|
+
seen.add(build_file_part_key(candidate));
|
|
41
|
+
}
|
|
42
|
+
const next_file_parts = parts.filter((part) => {
|
|
43
|
+
const key = build_file_part_key(part);
|
|
44
|
+
if (seen.has(key))
|
|
45
|
+
return false;
|
|
46
|
+
seen.add(key);
|
|
47
|
+
return true;
|
|
48
|
+
});
|
|
49
|
+
if (next_file_parts.length === 0)
|
|
50
|
+
return message;
|
|
51
|
+
return {
|
|
52
|
+
...message,
|
|
53
|
+
parts: [...current_parts, ...next_file_parts],
|
|
54
|
+
};
|
|
55
|
+
}
|
|
56
|
+
/**
|
|
57
|
+
* 模型与 tool-loop 主循环执行器。
|
|
58
|
+
*/
|
|
59
|
+
export class CoreEngineRunner {
|
|
60
|
+
history_store;
|
|
61
|
+
context_composer;
|
|
62
|
+
logger;
|
|
63
|
+
should_compact_on_error;
|
|
64
|
+
constructor(options) {
|
|
65
|
+
this.history_store = options.history_store;
|
|
66
|
+
this.context_composer = options.context_composer;
|
|
67
|
+
this.logger = options.logger;
|
|
68
|
+
this.should_compact_on_error = options.should_compact_on_error;
|
|
69
|
+
}
|
|
70
|
+
/**
|
|
71
|
+
* 执行一次已装配完成的模型/tool-loop 运行。
|
|
72
|
+
*/
|
|
73
|
+
async run(input) {
|
|
74
|
+
const start_time = Date.now();
|
|
75
|
+
const session_id = String(this.history_store.sessionId || "").trim();
|
|
76
|
+
const system = Array.isArray(input.execute_input.system)
|
|
77
|
+
? input.execute_input.system
|
|
78
|
+
: [];
|
|
79
|
+
const tools = input.execute_input.tools;
|
|
80
|
+
let last_observed_stream_error = undefined;
|
|
81
|
+
try {
|
|
82
|
+
const message_state = await CoreEngineMessageState.create({
|
|
83
|
+
messages: input.execute_input.messages,
|
|
84
|
+
tools,
|
|
85
|
+
});
|
|
86
|
+
const append_merged_user_messages = (messages) => message_state.appendMergedUserMessages(messages);
|
|
87
|
+
const context_composer_on_step_finish = this.context_composer.createOnStepFinishHandler(input.run_context);
|
|
88
|
+
let step_count = 0;
|
|
89
|
+
let total_tool_call_count = 0;
|
|
90
|
+
let total_tool_result_count = 0;
|
|
91
|
+
const on_step_finish = async (step_result) => {
|
|
92
|
+
step_count += 1;
|
|
93
|
+
const summary = summarizeStepForDebug(step_result);
|
|
94
|
+
total_tool_call_count +=
|
|
95
|
+
typeof summary.toolCallCount === "number" ? summary.toolCallCount : 0;
|
|
96
|
+
total_tool_result_count +=
|
|
97
|
+
typeof summary.toolResultCount === "number"
|
|
98
|
+
? summary.toolResultCount
|
|
99
|
+
: 0;
|
|
100
|
+
await this.logger.log("info", "[agent] step.finish", {
|
|
101
|
+
sessionId: session_id,
|
|
102
|
+
stepIndex: step_count,
|
|
103
|
+
...summary,
|
|
104
|
+
});
|
|
105
|
+
await context_composer_on_step_finish(step_result);
|
|
106
|
+
};
|
|
107
|
+
const prepare_step = this.context_composer.createPrepareStepHandler({
|
|
108
|
+
system,
|
|
109
|
+
appendMergedUserMessages: append_merged_user_messages,
|
|
110
|
+
runContext: input.run_context,
|
|
111
|
+
});
|
|
112
|
+
let final_assistant_ui_message = null;
|
|
113
|
+
let text_only_continuation_count = 0;
|
|
114
|
+
let incomplete_response_recovery_count = 0;
|
|
115
|
+
while (step_count < MAX_TOOL_LOOP_STEPS) {
|
|
116
|
+
const result = streamText({
|
|
117
|
+
model: input.model,
|
|
118
|
+
system,
|
|
119
|
+
onStepFinish: on_step_finish,
|
|
120
|
+
prepareStep: prepare_step,
|
|
121
|
+
messages: message_state.modelMessages,
|
|
122
|
+
tools,
|
|
123
|
+
providerOptions: buildOpenAIResponsesProviderOptions(),
|
|
124
|
+
onError: async ({ error }) => {
|
|
125
|
+
last_observed_stream_error = error;
|
|
126
|
+
await this.logger.log("error", "[agent] stream.error", {
|
|
127
|
+
sessionId: session_id,
|
|
128
|
+
...summarizeStreamError(error),
|
|
129
|
+
});
|
|
130
|
+
},
|
|
131
|
+
});
|
|
132
|
+
const step_assistant_ui_message = await collectFinalAssistantMessageFromUiStream({
|
|
133
|
+
result,
|
|
134
|
+
sessionId: session_id,
|
|
135
|
+
logger: this.logger,
|
|
136
|
+
buildFallbackAssistantMessage: (text) => this.context_composer.buildFallbackAssistantMessage(text, input.run_context),
|
|
137
|
+
onUiMessageChunkCallback: input.run_context.onUiMessageChunkCallback,
|
|
138
|
+
});
|
|
139
|
+
const executed_steps = await result.steps;
|
|
140
|
+
const last_step = executed_steps[executed_steps.length - 1];
|
|
141
|
+
if (!last_step)
|
|
142
|
+
break;
|
|
143
|
+
const incomplete_response = detectIncompleteResponse({
|
|
144
|
+
stepResult: last_step,
|
|
145
|
+
assistantMessage: step_assistant_ui_message,
|
|
146
|
+
});
|
|
147
|
+
const text_only_continuation_reason = detectTextOnlyContinuationReason(last_step);
|
|
148
|
+
const loop_decision = evaluateCoreEngineLoopDecision({
|
|
149
|
+
hasIncompleteResponse: incomplete_response !== null,
|
|
150
|
+
incompleteRecoveryCount: incomplete_response_recovery_count,
|
|
151
|
+
maxIncompleteRecoveries: MAX_INCOMPLETE_RESPONSE_RECOVERIES,
|
|
152
|
+
textOnlyContinuationReason: text_only_continuation_reason,
|
|
153
|
+
textOnlyContinuationCount: text_only_continuation_count,
|
|
154
|
+
maxTextOnlyContinuations: MAX_TEXT_ONLY_CONTINUATIONS,
|
|
155
|
+
hasTools: Object.keys(tools).length > 0,
|
|
156
|
+
toolCallCount: last_step.toolCalls.length,
|
|
157
|
+
});
|
|
158
|
+
await this.logger.log("info", "[agent] loop.decision", {
|
|
159
|
+
sessionId: session_id,
|
|
160
|
+
stepIndex: step_count,
|
|
161
|
+
continueForToolCalls: loop_decision.continueForToolCalls,
|
|
162
|
+
continueForTextOnly: loop_decision.continueForTextOnly,
|
|
163
|
+
continueForIncompleteRecovery: loop_decision.continueForIncompleteRecovery,
|
|
164
|
+
decisionKind: loop_decision.kind,
|
|
165
|
+
textOnlyContinuationReason: text_only_continuation_reason,
|
|
166
|
+
textOnlyContinuationCount: text_only_continuation_count,
|
|
167
|
+
incompleteResponseReason: incomplete_response?.reason ?? null,
|
|
168
|
+
incompleteResponseRecoveryCount: incomplete_response_recovery_count,
|
|
169
|
+
toolCallCount: last_step.toolCalls.length,
|
|
170
|
+
toolResultCount: last_step.toolResults.length,
|
|
171
|
+
finishReason: last_step.finishReason,
|
|
172
|
+
textPreview: toInlinePreview(last_step.text),
|
|
173
|
+
});
|
|
174
|
+
if (loop_decision.continueForIncompleteRecovery &&
|
|
175
|
+
incomplete_response) {
|
|
176
|
+
incomplete_response_recovery_count += 1;
|
|
177
|
+
await this.logger.log("warn", "[agent] incomplete_response.recover", {
|
|
178
|
+
sessionId: session_id,
|
|
179
|
+
stepIndex: step_count,
|
|
180
|
+
recoveryCount: incomplete_response_recovery_count,
|
|
181
|
+
reason: incomplete_response.reason,
|
|
182
|
+
...incomplete_response.details,
|
|
183
|
+
});
|
|
184
|
+
const recovery_message = this.history_store.userText({
|
|
185
|
+
text: buildIncompleteResponseRecoveryNudge(incomplete_response_recovery_count),
|
|
186
|
+
metadata: {
|
|
187
|
+
sessionId: session_id,
|
|
188
|
+
extra: {
|
|
189
|
+
internal: "agent_incomplete_response_recover",
|
|
190
|
+
reason: incomplete_response.reason,
|
|
191
|
+
stepIndex: step_count,
|
|
192
|
+
},
|
|
193
|
+
},
|
|
194
|
+
});
|
|
195
|
+
await message_state.appendUserTextMessage(recovery_message);
|
|
196
|
+
continue;
|
|
197
|
+
}
|
|
198
|
+
if (incomplete_response) {
|
|
199
|
+
await this.logger.log("error", "[agent] incomplete_response", {
|
|
200
|
+
sessionId: session_id,
|
|
201
|
+
stepIndex: step_count,
|
|
202
|
+
reason: incomplete_response.reason,
|
|
203
|
+
recoveryCount: incomplete_response_recovery_count,
|
|
204
|
+
...incomplete_response.details,
|
|
205
|
+
});
|
|
206
|
+
throw new Error(`Agent received incomplete response (${incomplete_response.reason})`);
|
|
207
|
+
}
|
|
208
|
+
const response_messages = Array.isArray(last_step.response?.messages)
|
|
209
|
+
? last_step.response.messages
|
|
210
|
+
: [];
|
|
211
|
+
message_state.appendModelMessages(response_messages);
|
|
212
|
+
final_assistant_ui_message = mergeAssistantUiMessages(final_assistant_ui_message, step_assistant_ui_message);
|
|
213
|
+
// 关键点(中文):把本 step 的 assistant UI 消息并入运行时上下文,保证后续全量重算不丢历史。
|
|
214
|
+
message_state.appendRuntimeSessionMessage(step_assistant_ui_message);
|
|
215
|
+
if (loop_decision.continueForToolCalls) {
|
|
216
|
+
text_only_continuation_count = 0;
|
|
217
|
+
incomplete_response_recovery_count = 0;
|
|
218
|
+
continue;
|
|
219
|
+
}
|
|
220
|
+
if (loop_decision.continueForTextOnly) {
|
|
221
|
+
text_only_continuation_count += 1;
|
|
222
|
+
incomplete_response_recovery_count = 0;
|
|
223
|
+
const continuation_message = this.history_store.userText({
|
|
224
|
+
text: buildTextOnlyContinuationNudge(text_only_continuation_count),
|
|
225
|
+
metadata: {
|
|
226
|
+
sessionId: session_id,
|
|
227
|
+
extra: {
|
|
228
|
+
internal: "agent_loop_auto_continue",
|
|
229
|
+
reason: text_only_continuation_reason,
|
|
230
|
+
stepIndex: step_count,
|
|
231
|
+
},
|
|
232
|
+
},
|
|
233
|
+
});
|
|
234
|
+
await message_state.appendUserTextMessage(continuation_message);
|
|
235
|
+
continue;
|
|
236
|
+
}
|
|
237
|
+
// 关键点(中文):stop 前做 tail merge,覆盖最后一个 step 后才入队的新 user 消息。
|
|
238
|
+
const tail_prepared = await prepare_step({ messages: [] });
|
|
239
|
+
const tail_merged_message_count = Array.isArray(tail_prepared.messages)
|
|
240
|
+
? tail_prepared.messages.length
|
|
241
|
+
: 0;
|
|
242
|
+
if (shouldContinueForTailMergedUserMessages({
|
|
243
|
+
mergedUserMessageCount: tail_merged_message_count,
|
|
244
|
+
})) {
|
|
245
|
+
text_only_continuation_count = 0;
|
|
246
|
+
incomplete_response_recovery_count = 0;
|
|
247
|
+
await this.logger.log("info", "[agent] loop.tail_merge_continue", {
|
|
248
|
+
sessionId: session_id,
|
|
249
|
+
stepIndex: step_count,
|
|
250
|
+
mergedUserMessageCount: tail_merged_message_count,
|
|
251
|
+
});
|
|
252
|
+
continue;
|
|
253
|
+
}
|
|
254
|
+
break;
|
|
255
|
+
}
|
|
256
|
+
if (step_count >= MAX_TOOL_LOOP_STEPS) {
|
|
257
|
+
await this.logger.log("warn", "[agent] loop.max_steps_reached", {
|
|
258
|
+
sessionId: session_id,
|
|
259
|
+
stepCount: step_count,
|
|
260
|
+
totalToolCallCount: total_tool_call_count,
|
|
261
|
+
totalToolResultCount: total_tool_result_count,
|
|
262
|
+
});
|
|
263
|
+
}
|
|
264
|
+
const final_message = mergePendingAssistantFileParts(final_assistant_ui_message ||
|
|
265
|
+
this.context_composer.buildFallbackAssistantMessage("Execution completed", input.run_context), input.run_context.pendingAssistantFileParts);
|
|
266
|
+
await this.logger.log("info", "[agent] final.message", {
|
|
267
|
+
sessionId: session_id,
|
|
268
|
+
...summarizeUiMessageForDebug(final_message),
|
|
269
|
+
});
|
|
270
|
+
await logAssistantMessageNow(this.logger, final_message);
|
|
271
|
+
const duration = Date.now() - start_time;
|
|
272
|
+
await this.logger.log("info", "[agent] finish", {
|
|
273
|
+
sessionId: session_id,
|
|
274
|
+
duration,
|
|
275
|
+
stepCount: step_count,
|
|
276
|
+
totalToolCallCount: total_tool_call_count,
|
|
277
|
+
totalToolResultCount: total_tool_result_count,
|
|
278
|
+
});
|
|
279
|
+
return {
|
|
280
|
+
success: true,
|
|
281
|
+
assistantMessage: final_message,
|
|
282
|
+
deferredPersistedUserMessages: [
|
|
283
|
+
...input.run_context.deferredPersistedUserMessages,
|
|
284
|
+
],
|
|
285
|
+
};
|
|
286
|
+
}
|
|
287
|
+
catch (error) {
|
|
288
|
+
if (this.should_compact_on_error(error)) {
|
|
289
|
+
throw error;
|
|
290
|
+
}
|
|
291
|
+
const error_text = resolveEffectiveCoreEngineError({
|
|
292
|
+
error,
|
|
293
|
+
streamError: last_observed_stream_error,
|
|
294
|
+
});
|
|
295
|
+
await this.logger.log("error", "CoreEngine execution failed", {
|
|
296
|
+
error: error_text,
|
|
297
|
+
});
|
|
298
|
+
return {
|
|
299
|
+
success: false,
|
|
300
|
+
error: error_text,
|
|
301
|
+
assistantMessage: this.context_composer.buildFallbackAssistantMessage(`Execution failed: ${error_text}`, input.run_context),
|
|
302
|
+
deferredPersistedUserMessages: [
|
|
303
|
+
...input.run_context.deferredPersistedUserMessages,
|
|
304
|
+
],
|
|
305
|
+
};
|
|
306
|
+
}
|
|
307
|
+
}
|
|
308
|
+
}
|
|
309
|
+
//# sourceMappingURL=CoreEngineRunner.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"CoreEngineRunner.js","sourceRoot":"","sources":["../../../src/executor/core-engine/CoreEngineRunner.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,UAAU,EAAuC,MAAM,IAAI,CAAC;AACrE,OAAO,EAAE,mCAAmC,EAAE,MAAM,2CAA2C,CAAC;AAChG,OAAO,EAAE,sBAAsB,EAAE,MAAM,yCAAyC,CAAC;AACjF,OAAO,EACL,kCAAkC,EAClC,2BAA2B,EAC3B,mBAAmB,EACnB,oCAAoC,EACpC,8BAA8B,EAC9B,wBAAwB,EACxB,gCAAgC,EAChC,wBAAwB,EACxB,qBAAqB,EACrB,0BAA0B,EAC1B,eAAe,GAChB,MAAM,4CAA4C,CAAC;AACpD,OAAO,EACL,8BAA8B,EAC9B,uCAAuC,GACxC,MAAM,iDAAiD,CAAC;AACzD,OAAO,EACL,+BAA+B,EAC/B,oBAAoB,GACrB,MAAM,0CAA0C,CAAC;AAClD,OAAO,EAAE,wCAAwC,EAAE,MAAM,sDAAsD,CAAC;AAChH,OAAO,EAAE,sBAAsB,EAAE,MAAM,iDAAiD,CAAC;AAWzF;;GAEG;AACH,SAAS,mBAAmB,CAAC,IAAgB;IAC3C,OAAO;QACL,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;QAC3B,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,EAAE,CAAC;KACvB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED;;GAEG;AACH,SAAS,8BAA8B,CACrC,OAAyB,EACzB,KAAmB;IAEnB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC;IAChE,MAAM,aAAa,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;IACxE,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;QACjC,MAAM,SAAS,GAAG,IAAkB,CAAC;QACrC,IAAI,SAAS,EAAE,IAAI,KAAK,MAAM;YAAE,SAAS;QACzC,IAAI,CAAC,GAAG,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3C,CAAC;IACD,MAAM,eAAe,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;QAC5C,MAAM,GAAG,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,OAAO,KAAK,CAAC;QAChC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACd,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IACH,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC;IACjD,OAAO;QACL,GAAG,OAAO;QACV,KAAK,EAAE,CAAC,GAAG,aAAa,EAAE,GAAG,eAAe,CAAC;KAC9C,CAAC;AACJ,CAAC;AAyCD;;GAEG;AACH,MAAM,OAAO,gBAAgB;IACV,aAAa,CAAsB;IACnC,gBAAgB,CAAyB;IACzC,MAAM,CAAS;IACf,uBAAuB,CAAqD;IAE7F,YAAY,OAAgC;QAC1C,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;QAC3C,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;QACjD,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,uBAAuB,GAAG,OAAO,CAAC,uBAAuB,CAAC;IACjE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,GAAG,CAAC,KAAyB;QACjC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC9B,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACrE,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC;YACtD,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM;YAC5B,CAAC,CAAC,EAAE,CAAC;QACP,MAAM,KAAK,GAAG,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC;QACxC,IAAI,0BAA0B,GAAY,SAAS,CAAC;QAEpD,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,MAAM,sBAAsB,CAAC,MAAM,CAAC;gBACxD,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,QAAQ;gBACtC,KAAK;aACN,CAAC,CAAC;YAEH,MAAM,2BAA2B,GAAG,CAAC,QAA4B,EAAE,EAAE,CACnE,aAAa,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;YAEnD,MAAM,+BAA+B,GACnC,IAAI,CAAC,gBAAgB,CAAC,yBAAyB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YACrE,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,IAAI,qBAAqB,GAAG,CAAC,CAAC;YAC9B,IAAI,uBAAuB,GAAG,CAAC,CAAC;YAChC,MAAM,cAAc,GAAG,KAAK,EAAE,WAAoB,EAAiB,EAAE;gBACnE,UAAU,IAAI,CAAC,CAAC;gBAChB,MAAM,OAAO,GAAG,qBAAqB,CAAC,WAAW,CAAC,CAAC;gBACnD,qBAAqB;oBACnB,OAAO,OAAO,CAAC,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;gBACxE,uBAAuB;oBACrB,OAAO,OAAO,CAAC,eAAe,KAAK,QAAQ;wBACzC,CAAC,CAAC,OAAO,CAAC,eAAe;wBACzB,CAAC,CAAC,CAAC,CAAC;gBACR,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,qBAAqB,EAAE;oBACnD,SAAS,EAAE,UAAU;oBACrB,SAAS,EAAE,UAAU;oBACrB,GAAG,OAAO;iBACX,CAAC,CAAC;gBACH,MAAM,+BAA+B,CAAC,WAAW,CAAC,CAAC;YACrD,CAAC,CAAC;YAEF,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,wBAAwB,CAAC;gBAClE,MAAM;gBACN,wBAAwB,EAAE,2BAA2B;gBACrD,UAAU,EAAE,KAAK,CAAC,WAAW;aAC9B,CAAC,CAAC;YAEH,IAAI,0BAA0B,GAA4B,IAAI,CAAC;YAC/D,IAAI,4BAA4B,GAAG,CAAC,CAAC;YACrC,IAAI,kCAAkC,GAAG,CAAC,CAAC;YAE3C,OAAO,UAAU,GAAG,mBAAmB,EAAE,CAAC;gBACxC,MAAM,MAAM,GAAG,UAAU,CAAC;oBACxB,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,MAAM;oBACN,YAAY,EAAE,cAAc;oBAC5B,WAAW,EAAE,YAAY;oBACzB,QAAQ,EAAE,aAAa,CAAC,aAAa;oBACrC,KAAK;oBACL,eAAe,EAAE,mCAAmC,EAAE;oBACtD,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;wBAC3B,0BAA0B,GAAG,KAAK,CAAC;wBACnC,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,sBAAsB,EAAE;4BACrD,SAAS,EAAE,UAAU;4BACrB,GAAG,oBAAoB,CAAC,KAAK,CAAC;yBAC/B,CAAC,CAAC;oBACL,CAAC;iBACF,CAAC,CAAC;gBAEH,MAAM,yBAAyB,GAC7B,MAAM,wCAAwC,CAAC;oBAC7C,MAAM;oBACN,SAAS,EAAE,UAAU;oBACrB,MAAM,EAAE,IAAI,CAAC,MAAM;oBACnB,6BAA6B,EAAE,CAAC,IAAI,EAAE,EAAE,CACtC,IAAI,CAAC,gBAAgB,CAAC,6BAA6B,CACjD,IAAI,EACJ,KAAK,CAAC,WAAW,CAClB;oBACH,wBAAwB,EAAE,KAAK,CAAC,WAAW,CAAC,wBAAwB;iBACrE,CAAC,CAAC;gBAEL,MAAM,cAAc,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;gBAC1C,MAAM,SAAS,GAAG,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAC5D,IAAI,CAAC,SAAS;oBAAE,MAAM;gBAEtB,MAAM,mBAAmB,GAAG,wBAAwB,CAAC;oBACnD,UAAU,EAAE,SAAS;oBACrB,gBAAgB,EAAE,yBAAyB;iBAC5C,CAAC,CAAC;gBACH,MAAM,6BAA6B,GACjC,gCAAgC,CAAC,SAAS,CAAC,CAAC;gBAC9C,MAAM,aAAa,GAAG,8BAA8B,CAAC;oBACnD,qBAAqB,EAAE,mBAAmB,KAAK,IAAI;oBACnD,uBAAuB,EAAE,kCAAkC;oBAC3D,uBAAuB,EAAE,kCAAkC;oBAC3D,0BAA0B,EAAE,6BAA6B;oBACzD,yBAAyB,EAAE,4BAA4B;oBACvD,wBAAwB,EAAE,2BAA2B;oBACrD,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC;oBACvC,aAAa,EAAE,SAAS,CAAC,SAAS,CAAC,MAAM;iBAC1C,CAAC,CAAC;gBAEH,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,uBAAuB,EAAE;oBACrD,SAAS,EAAE,UAAU;oBACrB,SAAS,EAAE,UAAU;oBACrB,oBAAoB,EAAE,aAAa,CAAC,oBAAoB;oBACxD,mBAAmB,EAAE,aAAa,CAAC,mBAAmB;oBACtD,6BAA6B,EAC3B,aAAa,CAAC,6BAA6B;oBAC7C,YAAY,EAAE,aAAa,CAAC,IAAI;oBAChC,0BAA0B,EAAE,6BAA6B;oBACzD,yBAAyB,EAAE,4BAA4B;oBACvD,wBAAwB,EAAE,mBAAmB,EAAE,MAAM,IAAI,IAAI;oBAC7D,+BAA+B,EAAE,kCAAkC;oBACnE,aAAa,EAAE,SAAS,CAAC,SAAS,CAAC,MAAM;oBACzC,eAAe,EAAE,SAAS,CAAC,WAAW,CAAC,MAAM;oBAC7C,YAAY,EAAE,SAAS,CAAC,YAAY;oBACpC,WAAW,EAAE,eAAe,CAAC,SAAS,CAAC,IAAI,CAAC;iBAC7C,CAAC,CAAC;gBAEH,IACE,aAAa,CAAC,6BAA6B;oBAC3C,mBAAmB,EACnB,CAAC;oBACD,kCAAkC,IAAI,CAAC,CAAC;oBACxC,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,qCAAqC,EAAE;wBACnE,SAAS,EAAE,UAAU;wBACrB,SAAS,EAAE,UAAU;wBACrB,aAAa,EAAE,kCAAkC;wBACjD,MAAM,EAAE,mBAAmB,CAAC,MAAM;wBAClC,GAAG,mBAAmB,CAAC,OAAO;qBAC/B,CAAC,CAAC;oBACH,MAAM,gBAAgB,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC;wBACnD,IAAI,EAAE,oCAAoC,CACxC,kCAAkC,CACnC;wBACD,QAAQ,EAAE;4BACR,SAAS,EAAE,UAAU;4BACrB,KAAK,EAAE;gCACL,QAAQ,EAAE,mCAAmC;gCAC7C,MAAM,EAAE,mBAAmB,CAAC,MAAM;gCAClC,SAAS,EAAE,UAAU;6BACtB;yBACF;qBACF,CAAC,CAAC;oBACH,MAAM,aAAa,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;oBAC5D,SAAS;gBACX,CAAC;gBAED,IAAI,mBAAmB,EAAE,CAAC;oBACxB,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,6BAA6B,EAAE;wBAC5D,SAAS,EAAE,UAAU;wBACrB,SAAS,EAAE,UAAU;wBACrB,MAAM,EAAE,mBAAmB,CAAC,MAAM;wBAClC,aAAa,EAAE,kCAAkC;wBACjD,GAAG,mBAAmB,CAAC,OAAO;qBAC/B,CAAC,CAAC;oBACH,MAAM,IAAI,KAAK,CACb,uCAAuC,mBAAmB,CAAC,MAAM,GAAG,CACrE,CAAC;gBACJ,CAAC;gBAED,MAAM,iBAAiB,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,EAAE,QAAQ,CAAC;oBACnE,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ;oBAC7B,CAAC,CAAC,EAAE,CAAC;gBACP,aAAa,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;gBAErD,0BAA0B,GAAG,wBAAwB,CACnD,0BAA0B,EAC1B,yBAAyB,CAC1B,CAAC;gBAEF,0DAA0D;gBAC1D,aAAa,CAAC,2BAA2B,CAAC,yBAAyB,CAAC,CAAC;gBAErE,IAAI,aAAa,CAAC,oBAAoB,EAAE,CAAC;oBACvC,4BAA4B,GAAG,CAAC,CAAC;oBACjC,kCAAkC,GAAG,CAAC,CAAC;oBACvC,SAAS;gBACX,CAAC;gBAED,IAAI,aAAa,CAAC,mBAAmB,EAAE,CAAC;oBACtC,4BAA4B,IAAI,CAAC,CAAC;oBAClC,kCAAkC,GAAG,CAAC,CAAC;oBACvC,MAAM,oBAAoB,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC;wBACvD,IAAI,EAAE,8BAA8B,CAAC,4BAA4B,CAAC;wBAClE,QAAQ,EAAE;4BACR,SAAS,EAAE,UAAU;4BACrB,KAAK,EAAE;gCACL,QAAQ,EAAE,0BAA0B;gCACpC,MAAM,EAAE,6BAA6B;gCACrC,SAAS,EAAE,UAAU;6BACtB;yBACF;qBACF,CAAC,CAAC;oBACH,MAAM,aAAa,CAAC,qBAAqB,CAAC,oBAAoB,CAAC,CAAC;oBAChE,SAAS;gBACX,CAAC;gBAED,yDAAyD;gBACzD,MAAM,aAAa,GAAG,MAAM,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC3D,MAAM,yBAAyB,GAAG,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC;oBACrE,CAAC,CAAC,aAAa,CAAC,QAAQ,CAAC,MAAM;oBAC/B,CAAC,CAAC,CAAC,CAAC;gBACN,IACE,uCAAuC,CAAC;oBACtC,sBAAsB,EAAE,yBAAyB;iBAClD,CAAC,EACF,CAAC;oBACD,4BAA4B,GAAG,CAAC,CAAC;oBACjC,kCAAkC,GAAG,CAAC,CAAC;oBACvC,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,kCAAkC,EAAE;wBAChE,SAAS,EAAE,UAAU;wBACrB,SAAS,EAAE,UAAU;wBACrB,sBAAsB,EAAE,yBAAyB;qBAClD,CAAC,CAAC;oBACH,SAAS;gBACX,CAAC;gBAED,MAAM;YACR,CAAC;YAED,IAAI,UAAU,IAAI,mBAAmB,EAAE,CAAC;gBACtC,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,gCAAgC,EAAE;oBAC9D,SAAS,EAAE,UAAU;oBACrB,SAAS,EAAE,UAAU;oBACrB,kBAAkB,EAAE,qBAAqB;oBACzC,oBAAoB,EAAE,uBAAuB;iBAC9C,CAAC,CAAC;YACL,CAAC;YAED,MAAM,aAAa,GAAG,8BAA8B,CAClD,0BAA0B;gBACxB,IAAI,CAAC,gBAAgB,CAAC,6BAA6B,CACjD,qBAAqB,EACrB,KAAK,CAAC,WAAW,CAClB,EACH,KAAK,CAAC,WAAW,CAAC,yBAAyB,CAC5C,CAAC;YAEF,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,uBAAuB,EAAE;gBACrD,SAAS,EAAE,UAAU;gBACrB,GAAG,0BAA0B,CAAC,aAAa,CAAC;aAC7C,CAAC,CAAC;YACH,MAAM,sBAAsB,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;YAEzD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,UAAU,CAAC;YACzC,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,gBAAgB,EAAE;gBAC9C,SAAS,EAAE,UAAU;gBACrB,QAAQ;gBACR,SAAS,EAAE,UAAU;gBACrB,kBAAkB,EAAE,qBAAqB;gBACzC,oBAAoB,EAAE,uBAAuB;aAC9C,CAAC,CAAC;YAEH,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,gBAAgB,EAAE,aAAa;gBAC/B,6BAA6B,EAAE;oBAC7B,GAAG,KAAK,CAAC,WAAW,CAAC,6BAA6B;iBACnD;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC,EAAE,CAAC;gBACxC,MAAM,KAAK,CAAC;YACd,CAAC;YAED,MAAM,UAAU,GAAG,+BAA+B,CAAC;gBACjD,KAAK;gBACL,WAAW,EAAE,0BAA0B;aACxC,CAAC,CAAC;YAEH,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,6BAA6B,EAAE;gBAC5D,KAAK,EAAE,UAAU;aAClB,CAAC,CAAC;YAEH,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,UAAU;gBACjB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,CAAC,6BAA6B,CACnE,qBAAqB,UAAU,EAAE,EACjC,KAAK,CAAC,WAAW,CAClB;gBACD,6BAA6B,EAAE;oBAC7B,GAAG,KAAK,CAAC,WAAW,CAAC,6BAA6B;iBACnD;aACF,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}
|
|
@@ -9,6 +9,7 @@
|
|
|
9
9
|
import type { streamText } from "ai";
|
|
10
10
|
import type { Logger } from "../../utils/logger/Logger.js";
|
|
11
11
|
import type { SessionMessageV1 } from "../../executor/types/SessionMessages.js";
|
|
12
|
+
import type { SessionUiMessageChunkCallback } from "../../executor/types/SessionRun.js";
|
|
12
13
|
/**
|
|
13
14
|
* 收敛 UI stream 中的最终 assistant 消息。
|
|
14
15
|
*/
|
|
@@ -29,5 +30,9 @@ export declare function collectFinalAssistantMessageFromUiStream(params: {
|
|
|
29
30
|
* 构造 fallback assistant 消息的工厂函数。
|
|
30
31
|
*/
|
|
31
32
|
buildFallbackAssistantMessage: (text: string) => SessionMessageV1;
|
|
33
|
+
/**
|
|
34
|
+
* UI stream chunk 回调。
|
|
35
|
+
*/
|
|
36
|
+
onUiMessageChunkCallback?: SessionUiMessageChunkCallback;
|
|
32
37
|
}): Promise<SessionMessageV1>;
|
|
33
38
|
//# sourceMappingURL=CoreEngineUiStreamCollector.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"CoreEngineUiStreamCollector.d.ts","sourceRoot":"","sources":["../../../src/executor/core-engine/CoreEngineUiStreamCollector.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AACrC,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,0BAA0B,CAAC;AAEvD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qCAAqC,CAAC;
|
|
1
|
+
{"version":3,"file":"CoreEngineUiStreamCollector.d.ts","sourceRoot":"","sources":["../../../src/executor/core-engine/CoreEngineUiStreamCollector.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AACrC,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,0BAA0B,CAAC;AAEvD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qCAAqC,CAAC;AAC5E,OAAO,KAAK,EAAE,6BAA6B,EAAE,MAAM,gCAAgC,CAAC;AAMpF;;GAEG;AACH,wBAAsB,wCAAwC,CAAC,MAAM,EAAE;IACrE;;OAEG;IACH,MAAM,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,CAAC;IACtC;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IACf;;OAEG;IACH,6BAA6B,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,gBAAgB,CAAC;IAClE;;OAEG;IACH,wBAAwB,CAAC,EAAE,6BAA6B,CAAC;CAC1D,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAsD5B"}
|
|
@@ -6,7 +6,6 @@
|
|
|
6
6
|
* - 优先使用结构化 `responseMessage`;缺失时才回退到纯文本。
|
|
7
7
|
* - UI chunk 回调是展示层副作用,失败不应阻断 session 执行。
|
|
8
8
|
*/
|
|
9
|
-
import { getSessionRunScope } from "../SessionRunScope.js";
|
|
10
9
|
import { summarizeUiMessageForDebug, toInlinePreview, } from "../core-engine/CoreEngineSignals.js";
|
|
11
10
|
/**
|
|
12
11
|
* 收敛 UI stream 中的最终 assistant 消息。
|
|
@@ -28,12 +27,11 @@ export async function collectFinalAssistantMessageFromUiStream(params) {
|
|
|
28
27
|
};
|
|
29
28
|
},
|
|
30
29
|
});
|
|
31
|
-
const onUiMessageChunkCallback = getSessionRunScope()?.onUiMessageChunkCallback;
|
|
32
30
|
for await (const chunk of uiStream) {
|
|
33
|
-
if (typeof onUiMessageChunkCallback !== "function")
|
|
31
|
+
if (typeof params.onUiMessageChunkCallback !== "function")
|
|
34
32
|
continue;
|
|
35
33
|
try {
|
|
36
|
-
await onUiMessageChunkCallback(chunk);
|
|
34
|
+
await params.onUiMessageChunkCallback(chunk);
|
|
37
35
|
}
|
|
38
36
|
catch {
|
|
39
37
|
// ignore UI stream callback failures
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"CoreEngineUiStreamCollector.js","sourceRoot":"","sources":["../../../src/executor/core-engine/CoreEngineUiStreamCollector.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;
|
|
1
|
+
{"version":3,"file":"CoreEngineUiStreamCollector.js","sourceRoot":"","sources":["../../../src/executor/core-engine/CoreEngineUiStreamCollector.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAOH,OAAO,EACL,0BAA0B,EAC1B,eAAe,GAChB,MAAM,4CAA4C,CAAC;AAEpD;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,wCAAwC,CAAC,MAqB9D;IACC,IAAI,wBAAwB,GAA4B,IAAI,CAAC;IAC7D,IAAI,eAAe,GAAsB,IAAI,CAAC;IAE9C,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAmB;QACjE,wEAAwE;QACxE,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,KAAK;QAClB,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;YAClB,wBAAwB,GAAG,KAAK,CAAC,eAAe,IAAI,IAAI,CAAC;YACzD,eAAe,GAAG;gBAChB,cAAc,EAAE,KAAK,CAAC,cAAc;gBACpC,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,YAAY,EACV,OAAO,KAAK,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI;gBACpE,GAAG,0BAA0B,CAAC,KAAK,CAAC,eAAe,CAAC;aACrD,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;QACnC,IAAI,OAAO,MAAM,CAAC,wBAAwB,KAAK,UAAU;YAAE,SAAS;QACpE,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,qCAAqC;QACvC,CAAC;IACH,CAAC;IAED,MAAM,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,mBAAmB,EAAE;QACnD,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,GAAG,CAAC,eAAe,IAAI;YACrB,sBAAsB,EAAE,IAAI;SAC7B,CAAC;KACH,CAAC,CAAC;IAEH,IAAI,wBAAwB;QAAE,OAAO,wBAAwB,CAAC;IAE9D,IAAI,aAAa,GAAG,EAAE,CAAC;IACvB,IAAI,CAAC;QACH,aAAa,GAAG,MAAM,CAAC,CAAC,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAClE,CAAC;IAAC,MAAM,CAAC;QACP,aAAa,GAAG,EAAE,CAAC;IACrB,CAAC;IAED,MAAM,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,gCAAgC,EAAE;QAChE,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,mBAAmB,EAAE,aAAa,CAAC,MAAM;QACzC,oBAAoB,EAAE,eAAe,CAAC,aAAa,CAAC;KACrD,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC,6BAA6B,CACzC,aAAa,IAAI,qBAAqB,CACvC,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ExecutorInflightService:运行中 assistant 快照写入服务。
|
|
3
|
+
*
|
|
4
|
+
* 关键点(中文)
|
|
5
|
+
* - 只负责把 step/tool 增量写入当前 inflight assistant 快照。
|
|
6
|
+
* - 不负责长期 history 组装,不负责主执行循环。
|
|
7
|
+
* - 保持与 SessionHistoryStore 的交互边界清晰,避免 Executor 自身同时承担编排与写入细节。
|
|
8
|
+
*/
|
|
9
|
+
import type { SessionHistoryStore } from "../../executor/store/history/SessionHistoryStore.js";
|
|
10
|
+
import type { SessionMessageV1 } from "../../executor/types/SessionMessages.js";
|
|
11
|
+
interface ExecutorInflightServiceOptions {
|
|
12
|
+
/**
|
|
13
|
+
* 当前 session 标识。
|
|
14
|
+
*/
|
|
15
|
+
session_id: string;
|
|
16
|
+
/**
|
|
17
|
+
* 当前 session 对应的 history 事实源。
|
|
18
|
+
*/
|
|
19
|
+
history_store: SessionHistoryStore;
|
|
20
|
+
/**
|
|
21
|
+
* inflight 更新完成后的异步通知。
|
|
22
|
+
*/
|
|
23
|
+
run_after_session_updated_async?: () => Promise<void>;
|
|
24
|
+
}
|
|
25
|
+
/**
|
|
26
|
+
* 运行中 assistant 快照写入服务。
|
|
27
|
+
*/
|
|
28
|
+
export declare class ExecutorInflightService {
|
|
29
|
+
private readonly session_id;
|
|
30
|
+
private readonly history_store;
|
|
31
|
+
private readonly run_after_session_updated_async?;
|
|
32
|
+
constructor(options: ExecutorInflightServiceOptions);
|
|
33
|
+
/**
|
|
34
|
+
* 把 step/tool 过程增量写入当前运行中的 assistant 快照。
|
|
35
|
+
*/
|
|
36
|
+
append_assistant_step_parts(parts: SessionMessageV1["parts"]): Promise<void>;
|
|
37
|
+
}
|
|
38
|
+
export {};
|
|
39
|
+
//# sourceMappingURL=ExecutorInflightService.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ExecutorInflightService.d.ts","sourceRoot":"","sources":["../../../src/executor/services/ExecutorInflightService.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAGH,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,iDAAiD,CAAC;AAC3F,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qCAAqC,CAAC;AAE5E,UAAU,8BAA8B;IACtC;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IAEnB;;OAEG;IACH,aAAa,EAAE,mBAAmB,CAAC;IAEnC;;OAEG;IACH,+BAA+B,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CACvD;AAED;;GAEG;AACH,qBAAa,uBAAuB;IAClC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAsB;IACpD,OAAO,CAAC,QAAQ,CAAC,+BAA+B,CAAC,CAAoE;gBAEzG,OAAO,EAAE,8BAA8B;IAWnD;;OAEG;IACG,2BAA2B,CAC/B,KAAK,EAAE,gBAAgB,CAAC,OAAO,CAAC,GAC/B,OAAO,CAAC,IAAI,CAAC;CA8CjB"}
|
|
@@ -0,0 +1,75 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ExecutorInflightService:运行中 assistant 快照写入服务。
|
|
3
|
+
*
|
|
4
|
+
* 关键点(中文)
|
|
5
|
+
* - 只负责把 step/tool 增量写入当前 inflight assistant 快照。
|
|
6
|
+
* - 不负责长期 history 组装,不负责主执行循环。
|
|
7
|
+
* - 保持与 SessionHistoryStore 的交互边界清晰,避免 Executor 自身同时承担编排与写入细节。
|
|
8
|
+
*/
|
|
9
|
+
import { generateId } from "../../utils/Id.js";
|
|
10
|
+
/**
|
|
11
|
+
* 运行中 assistant 快照写入服务。
|
|
12
|
+
*/
|
|
13
|
+
export class ExecutorInflightService {
|
|
14
|
+
session_id;
|
|
15
|
+
history_store;
|
|
16
|
+
run_after_session_updated_async;
|
|
17
|
+
constructor(options) {
|
|
18
|
+
this.session_id = String(options.session_id || "").trim();
|
|
19
|
+
this.history_store = options.history_store;
|
|
20
|
+
this.run_after_session_updated_async =
|
|
21
|
+
options.run_after_session_updated_async;
|
|
22
|
+
if (!this.session_id) {
|
|
23
|
+
throw new Error("ExecutorInflightService requires a non-empty session_id");
|
|
24
|
+
}
|
|
25
|
+
}
|
|
26
|
+
/**
|
|
27
|
+
* 把 step/tool 过程增量写入当前运行中的 assistant 快照。
|
|
28
|
+
*/
|
|
29
|
+
async append_assistant_step_parts(parts) {
|
|
30
|
+
const normalized_parts = Array.isArray(parts)
|
|
31
|
+
? parts.filter((part) => part && typeof part === "object")
|
|
32
|
+
: [];
|
|
33
|
+
if (normalized_parts.length === 0)
|
|
34
|
+
return;
|
|
35
|
+
const current_inflight = await this.history_store.readInflight();
|
|
36
|
+
const next_message = current_inflight
|
|
37
|
+
? {
|
|
38
|
+
...current_inflight,
|
|
39
|
+
metadata: {
|
|
40
|
+
...(current_inflight.metadata || {
|
|
41
|
+
v: 1,
|
|
42
|
+
ts: Date.now(),
|
|
43
|
+
sessionId: this.session_id,
|
|
44
|
+
}),
|
|
45
|
+
ts: Date.now(),
|
|
46
|
+
sessionId: this.session_id,
|
|
47
|
+
source: "egress",
|
|
48
|
+
kind: "normal",
|
|
49
|
+
},
|
|
50
|
+
parts: [
|
|
51
|
+
...(Array.isArray(current_inflight.parts)
|
|
52
|
+
? current_inflight.parts
|
|
53
|
+
: []),
|
|
54
|
+
...normalized_parts,
|
|
55
|
+
],
|
|
56
|
+
}
|
|
57
|
+
: {
|
|
58
|
+
id: `a:${this.session_id}:${generateId()}`,
|
|
59
|
+
role: "assistant",
|
|
60
|
+
metadata: {
|
|
61
|
+
v: 1,
|
|
62
|
+
ts: Date.now(),
|
|
63
|
+
sessionId: this.session_id,
|
|
64
|
+
source: "egress",
|
|
65
|
+
kind: "normal",
|
|
66
|
+
},
|
|
67
|
+
parts: normalized_parts,
|
|
68
|
+
};
|
|
69
|
+
await this.history_store.writeInflight(next_message);
|
|
70
|
+
if (this.run_after_session_updated_async) {
|
|
71
|
+
await this.run_after_session_updated_async();
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
//# sourceMappingURL=ExecutorInflightService.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ExecutorInflightService.js","sourceRoot":"","sources":["../../../src/executor/services/ExecutorInflightService.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAqB3C;;GAEG;AACH,MAAM,OAAO,uBAAuB;IACjB,UAAU,CAAS;IACnB,aAAa,CAAsB;IACnC,+BAA+B,CAAqE;IAErH,YAAY,OAAuC;QACjD,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC1D,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;QAC3C,IAAI,CAAC,+BAA+B;YAClC,OAAO,CAAC,+BAA+B,CAAC;QAE1C,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;QAC7E,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,2BAA2B,CAC/B,KAAgC;QAEhC,MAAM,gBAAgB,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YAC3C,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,CAAC;YAC1D,CAAC,CAAC,EAAE,CAAC;QACP,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE1C,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE,CAAC;QACjE,MAAM,YAAY,GAAqB,gBAAgB;YACrD,CAAC,CAAC;gBACE,GAAG,gBAAgB;gBACnB,QAAQ,EAAE;oBACR,GAAG,CAAC,gBAAgB,CAAC,QAAQ,IAAI;wBAC/B,CAAC,EAAE,CAAU;wBACb,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;wBACd,SAAS,EAAE,IAAI,CAAC,UAAU;qBAC3B,CAAC;oBACF,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;oBACd,SAAS,EAAE,IAAI,CAAC,UAAU;oBAC1B,MAAM,EAAE,QAAQ;oBAChB,IAAI,EAAE,QAAQ;iBACf;gBACD,KAAK,EAAE;oBACL,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAC,KAAK,CAAC;wBACvC,CAAC,CAAC,gBAAgB,CAAC,KAAK;wBACxB,CAAC,CAAC,EAAE,CAAC;oBACP,GAAG,gBAAgB;iBACpB;aACF;YACH,CAAC,CAAC;gBACE,EAAE,EAAE,KAAK,IAAI,CAAC,UAAU,IAAI,UAAU,EAAE,EAAE;gBAC1C,IAAI,EAAE,WAAW;gBACjB,QAAQ,EAAE;oBACR,CAAC,EAAE,CAAC;oBACJ,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE;oBACd,SAAS,EAAE,IAAI,CAAC,UAAU;oBAC1B,MAAM,EAAE,QAAQ;oBAChB,IAAI,EAAE,QAAQ;iBACf;gBACD,KAAK,EAAE,gBAAgB;aACxB,CAAC;QAEN,MAAM,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QACrD,IAAI,IAAI,CAAC,+BAA+B,EAAE,CAAC;YACzC,MAAM,IAAI,CAAC,+BAA+B,EAAE,CAAC;QAC/C,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,103 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ExecutorRecoveryPolicy:执行恢复与重试策略。
|
|
3
|
+
*
|
|
4
|
+
* 关键点(中文)
|
|
5
|
+
* - 统一封装“压缩后重试”和“普通失败兜底”逻辑。
|
|
6
|
+
* - Executor 只负责准备输入与调用策略,不再直接承载重试状态机。
|
|
7
|
+
* - 不改变外部行为,只把异常分流规则集中到一个地方。
|
|
8
|
+
*/
|
|
9
|
+
import type { LanguageModel } from "ai";
|
|
10
|
+
import type { Logger } from "../../utils/logger/Logger.js";
|
|
11
|
+
import type { SessionCompactionComposer } from "../composer/compaction/SessionCompactionComposer.js";
|
|
12
|
+
import type { SessionContextComposer } from "../composer/context/SessionContextComposer.js";
|
|
13
|
+
import type { SessionRunContext } from "../../types/executor/SessionRunContext.js";
|
|
14
|
+
import type { SessionExecuteInput, SessionRunResult } from "../../executor/types/SessionRun.js";
|
|
15
|
+
interface ExecutorRecoveryPolicyOptions {
|
|
16
|
+
/**
|
|
17
|
+
* 当前 session 对应的 compaction Composer。
|
|
18
|
+
*/
|
|
19
|
+
compaction_composer: SessionCompactionComposer;
|
|
20
|
+
/**
|
|
21
|
+
* 当前 session 对应的 context Composer。
|
|
22
|
+
*/
|
|
23
|
+
context_composer: SessionContextComposer;
|
|
24
|
+
/**
|
|
25
|
+
* 当前 session 统一日志器。
|
|
26
|
+
*/
|
|
27
|
+
logger: Logger;
|
|
28
|
+
}
|
|
29
|
+
interface ExecutorPrepareRunInput {
|
|
30
|
+
/**
|
|
31
|
+
* 当前轮用户 query。
|
|
32
|
+
*/
|
|
33
|
+
query: string;
|
|
34
|
+
/**
|
|
35
|
+
* 当前轮模型实例。
|
|
36
|
+
*/
|
|
37
|
+
model: LanguageModel;
|
|
38
|
+
/**
|
|
39
|
+
* 当前显式运行上下文。
|
|
40
|
+
*/
|
|
41
|
+
run_context: SessionRunContext;
|
|
42
|
+
/**
|
|
43
|
+
* 当前压缩重试次数。
|
|
44
|
+
*/
|
|
45
|
+
retry_count: number;
|
|
46
|
+
}
|
|
47
|
+
interface ExecutorExecutePreparedRunInput {
|
|
48
|
+
/**
|
|
49
|
+
* 已装配好的执行输入。
|
|
50
|
+
*/
|
|
51
|
+
execute_input: SessionExecuteInput;
|
|
52
|
+
/**
|
|
53
|
+
* 当前轮模型实例。
|
|
54
|
+
*/
|
|
55
|
+
model: LanguageModel;
|
|
56
|
+
/**
|
|
57
|
+
* 当前显式运行上下文。
|
|
58
|
+
*/
|
|
59
|
+
run_context: SessionRunContext;
|
|
60
|
+
}
|
|
61
|
+
interface ExecutorRecoveryRunInput {
|
|
62
|
+
/**
|
|
63
|
+
* 当前轮用户 query。
|
|
64
|
+
*/
|
|
65
|
+
query: string;
|
|
66
|
+
/**
|
|
67
|
+
* 当前轮模型实例。
|
|
68
|
+
*/
|
|
69
|
+
model: LanguageModel;
|
|
70
|
+
/**
|
|
71
|
+
* 当前显式运行上下文。
|
|
72
|
+
*/
|
|
73
|
+
run_context: SessionRunContext;
|
|
74
|
+
/**
|
|
75
|
+
* 运行前装配执行输入。
|
|
76
|
+
*/
|
|
77
|
+
prepare_execute_input: (input: ExecutorPrepareRunInput) => Promise<SessionExecuteInput>;
|
|
78
|
+
/**
|
|
79
|
+
* 执行已装配好的运行输入。
|
|
80
|
+
*/
|
|
81
|
+
execute_prepared_run: (input: ExecutorExecutePreparedRunInput) => Promise<SessionRunResult>;
|
|
82
|
+
}
|
|
83
|
+
/**
|
|
84
|
+
* 执行恢复与重试策略服务。
|
|
85
|
+
*/
|
|
86
|
+
export declare class ExecutorRecoveryPolicy {
|
|
87
|
+
private readonly compaction_composer;
|
|
88
|
+
private readonly context_composer;
|
|
89
|
+
private readonly logger;
|
|
90
|
+
private retry_count;
|
|
91
|
+
constructor(options: ExecutorRecoveryPolicyOptions);
|
|
92
|
+
/**
|
|
93
|
+
* 重置当前 run 级状态。
|
|
94
|
+
*/
|
|
95
|
+
reset_run_state(): void;
|
|
96
|
+
/**
|
|
97
|
+
* 执行一次带恢复策略的 session run。
|
|
98
|
+
*/
|
|
99
|
+
run_with_retry(input: ExecutorRecoveryRunInput): Promise<SessionRunResult>;
|
|
100
|
+
private build_failure_result;
|
|
101
|
+
}
|
|
102
|
+
export {};
|
|
103
|
+
//# sourceMappingURL=ExecutorRecoveryPolicy.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ExecutorRecoveryPolicy.d.ts","sourceRoot":"","sources":["../../../src/executor/services/ExecutorRecoveryPolicy.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,0BAA0B,CAAC;AACvD,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,4DAA4D,CAAC;AAC5G,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,sDAAsD,CAAC;AACnG,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAC/E,OAAO,KAAK,EACV,mBAAmB,EACnB,gBAAgB,EACjB,MAAM,gCAAgC,CAAC;AAOxC,UAAU,6BAA6B;IACrC;;OAEG;IACH,mBAAmB,EAAE,yBAAyB,CAAC;IAE/C;;OAEG;IACH,gBAAgB,EAAE,sBAAsB,CAAC;IAEzC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,UAAU,uBAAuB;IAC/B;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,KAAK,EAAE,aAAa,CAAC;IAErB;;OAEG;IACH,WAAW,EAAE,iBAAiB,CAAC;IAE/B;;OAEG;IACH,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,UAAU,+BAA+B;IACvC;;OAEG;IACH,aAAa,EAAE,mBAAmB,CAAC;IAEnC;;OAEG;IACH,KAAK,EAAE,aAAa,CAAC;IAErB;;OAEG;IACH,WAAW,EAAE,iBAAiB,CAAC;CAChC;AAED,UAAU,wBAAwB;IAChC;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,KAAK,EAAE,aAAa,CAAC;IAErB;;OAEG;IACH,WAAW,EAAE,iBAAiB,CAAC;IAE/B;;OAEG;IACH,qBAAqB,EAAE,CACrB,KAAK,EAAE,uBAAuB,KAC3B,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAElC;;OAEG;IACH,oBAAoB,EAAE,CACpB,KAAK,EAAE,+BAA+B,KACnC,OAAO,CAAC,gBAAgB,CAAC,CAAC;CAChC;AAED;;GAEG;AACH,qBAAa,sBAAsB;IACjC,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAA4B;IAChE,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAyB;IAC1D,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,WAAW,CAAK;gBAEZ,OAAO,EAAE,6BAA6B;IAMlD;;OAEG;IACH,eAAe,IAAI,IAAI;IAIvB;;OAEG;IACG,cAAc,CAClB,KAAK,EAAE,wBAAwB,GAC9B,OAAO,CAAC,gBAAgB,CAAC;IA8C5B,OAAO,CAAC,oBAAoB;CA4B7B"}
|