repoburg 1.3.126 → 1.3.128
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/backend/.env +3 -1
- package/backend/dist/src/core-entities/system-prompt.entity.d.ts +1 -0
- package/backend/dist/src/core-entities/system-prompt.entity.js +8 -0
- package/backend/dist/src/core-entities/system-prompt.entity.js.map +1 -1
- package/backend/dist/src/llm-orchestration/action-handlers/invoke-subagent.handler.d.ts +3 -13
- package/backend/dist/src/llm-orchestration/action-handlers/invoke-subagent.handler.js +65 -187
- package/backend/dist/src/llm-orchestration/action-handlers/invoke-subagent.handler.js.map +1 -1
- package/backend/dist/src/llm-orchestration/action-handlers/mcp-tool-action.handler.js +2 -0
- package/backend/dist/src/llm-orchestration/action-handlers/mcp-tool-action.handler.js.map +1 -1
- package/backend/dist/src/llm-orchestration/llm-orchestration.module.js +3 -12
- package/backend/dist/src/llm-orchestration/llm-orchestration.module.js.map +1 -1
- package/backend/dist/src/llm-orchestration/llm-turn-processor.service.d.ts +1 -3
- package/backend/dist/src/llm-orchestration/llm-turn-processor.service.js +2 -33
- package/backend/dist/src/llm-orchestration/llm-turn-processor.service.js.map +1 -1
- package/backend/dist/src/llm-orchestration/tool-schema.service.js +8 -4
- package/backend/dist/src/llm-orchestration/tool-schema.service.js.map +1 -1
- package/backend/dist/src/mcp/mcp.service.d.ts +1 -0
- package/backend/dist/src/mcp/mcp.service.js +16 -0
- package/backend/dist/src/mcp/mcp.service.js.map +1 -1
- package/backend/dist/src/seeding/data/sub-agents/browser-use.d.ts +9 -0
- package/backend/dist/src/seeding/data/sub-agents/browser-use.js +13 -0
- package/backend/dist/src/seeding/data/sub-agents/browser-use.js.map +1 -0
- package/backend/dist/src/seeding/data/sub-agents/explore-codebase.d.ts +3 -3
- package/backend/dist/src/seeding/data/sub-agents/explore-codebase.js +3 -3
- package/backend/dist/src/seeding/data/sub-agents/explore-codebase.js.map +1 -1
- package/backend/dist/src/seeding/data/system-prompts/browser-use-agent.d.ts +3 -0
- package/backend/dist/src/seeding/data/system-prompts/browser-use-agent.js +39 -0
- package/backend/dist/src/seeding/data/system-prompts/browser-use-agent.js.map +1 -0
- package/backend/dist/src/seeding/data/system-prompts/default_native_tool_agent.d.ts +1 -1
- package/backend/dist/src/seeding/data/system-prompts/default_native_tool_agent.js +5 -5
- package/backend/dist/src/seeding/data/system-prompts/master-agent.d.ts +1 -1
- package/backend/dist/src/seeding/data/system-prompts/master-agent.js +14 -3
- package/backend/dist/src/seeding/data/system-prompts/master-agent.js.map +1 -1
- package/backend/dist/src/seeding/sub-agent-seeding.service.js +14 -1
- package/backend/dist/src/seeding/sub-agent-seeding.service.js.map +1 -1
- package/backend/dist/src/sub-agents/dto/create-sub-agent.dto.d.ts +1 -0
- package/backend/dist/src/sub-agents/dto/create-sub-agent.dto.js +8 -0
- package/backend/dist/src/sub-agents/dto/create-sub-agent.dto.js.map +1 -1
- package/backend/dist/src/sub-agents/dto/update-sub-agent.dto.d.ts +1 -0
- package/backend/dist/src/sub-agents/dto/update-sub-agent.dto.js +7 -0
- package/backend/dist/src/sub-agents/dto/update-sub-agent.dto.js.map +1 -1
- package/backend/dist/src/sub-agents/sub-agent-runner.service.d.ts +47 -0
- package/backend/dist/src/sub-agents/sub-agent-runner.service.js +523 -0
- package/backend/dist/src/sub-agents/sub-agent-runner.service.js.map +1 -0
- package/backend/dist/src/sub-agents/sub-agent-runs.controller.d.ts +4 -3
- package/backend/dist/src/sub-agents/sub-agent-runs.controller.js +10 -5
- package/backend/dist/src/sub-agents/sub-agent-runs.controller.js.map +1 -1
- package/backend/dist/src/sub-agents/sub-agent.entity.d.ts +1 -0
- package/backend/dist/src/sub-agents/sub-agent.entity.js +8 -0
- package/backend/dist/src/sub-agents/sub-agent.entity.js.map +1 -1
- package/backend/dist/src/sub-agents/sub-agents.module.js +16 -5
- package/backend/dist/src/sub-agents/sub-agents.module.js.map +1 -1
- package/backend/dist/src/sub-agents/sub-agents.service.d.ts +2 -0
- package/backend/dist/src/sub-agents/sub-agents.service.js +42 -2
- package/backend/dist/src/sub-agents/sub-agents.service.js.map +1 -1
- package/backend/dist/src/system-prompts/dto/set-enabled-mcp-tools.dto.d.ts +4 -0
- package/backend/dist/src/system-prompts/dto/set-enabled-mcp-tools.dto.js +38 -0
- package/backend/dist/src/system-prompts/dto/set-enabled-mcp-tools.dto.js.map +1 -0
- package/backend/dist/src/system-prompts/dto/system-prompt.dto.d.ts +2 -0
- package/backend/dist/src/system-prompts/dto/system-prompt.dto.js +16 -0
- package/backend/dist/src/system-prompts/dto/system-prompt.dto.js.map +1 -1
- package/backend/dist/src/system-prompts/dto/toggle-mcp-tool.dto.d.ts +3 -0
- package/backend/dist/src/system-prompts/dto/toggle-mcp-tool.dto.js +27 -0
- package/backend/dist/src/system-prompts/dto/toggle-mcp-tool.dto.js.map +1 -0
- package/backend/dist/src/system-prompts/system-prompts.controller.d.ts +17 -0
- package/backend/dist/src/system-prompts/system-prompts.controller.js +83 -0
- package/backend/dist/src/system-prompts/system-prompts.controller.js.map +1 -1
- package/backend/dist/src/system-prompts/system-prompts.service.d.ts +11 -0
- package/backend/dist/src/system-prompts/system-prompts.service.js +115 -1
- package/backend/dist/src/system-prompts/system-prompts.service.js.map +1 -1
- package/backend/dist/tsconfig.build.tsbuildinfo +1 -1
- package/package.json +1 -1
- package/backend/dist/src/llm-orchestration/hooks/sub-agent-final-response.hook.d.ts +0 -14
- package/backend/dist/src/llm-orchestration/hooks/sub-agent-final-response.hook.js +0 -80
- package/backend/dist/src/llm-orchestration/hooks/sub-agent-final-response.hook.js.map +0 -1
- package/backend/dist/src/seeding/data/sub-agents/handover-session.d.ts +0 -8
- package/backend/dist/src/seeding/data/sub-agents/handover-session.js +0 -18
- package/backend/dist/src/seeding/data/sub-agents/handover-session.js.map +0 -1
- package/backend/dist/src/seeding/data/system-prompts/handover-agent.d.ts +0 -3
- package/backend/dist/src/seeding/data/system-prompts/handover-agent.js +0 -69
- package/backend/dist/src/seeding/data/system-prompts/handover-agent.js.map +0 -1
|
@@ -0,0 +1,523 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
var __param = (this && this.__param) || function (paramIndex, decorator) {
|
|
12
|
+
return function (target, key) { decorator(target, key, paramIndex); }
|
|
13
|
+
};
|
|
14
|
+
var SubAgentRunner_1;
|
|
15
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
16
|
+
exports.SubAgentRunner = void 0;
|
|
17
|
+
const common_1 = require("@nestjs/common");
|
|
18
|
+
const core_1 = require("@nestjs/core");
|
|
19
|
+
const llm_provider_interface_1 = require("../llm-provider/llm-provider.interface");
|
|
20
|
+
const sub_agents_service_1 = require("./sub-agents.service");
|
|
21
|
+
const sub_agent_runs_service_1 = require("./sub-agent-runs.service");
|
|
22
|
+
const sessions_service_1 = require("../sessions/sessions.service");
|
|
23
|
+
const system_prompts_service_1 = require("../system-prompts/system-prompts.service");
|
|
24
|
+
const tool_schema_service_1 = require("../llm-orchestration/tool-schema.service");
|
|
25
|
+
const history_compression_service_1 = require("../llm-orchestration/history-compression.service");
|
|
26
|
+
const application_state_service_1 = require("../application-state/application-state.service");
|
|
27
|
+
const mcp_tool_action_handler_1 = require("../llm-orchestration/action-handlers/mcp-tool-action.handler");
|
|
28
|
+
const mcp_service_1 = require("../mcp/mcp.service");
|
|
29
|
+
const llm_orchestration_interfaces_1 = require("../llm-orchestration/llm-orchestration.interfaces");
|
|
30
|
+
const typeorm_1 = require("@nestjs/typeorm");
|
|
31
|
+
const typeorm_2 = require("typeorm");
|
|
32
|
+
const session_input_entity_1 = require("../core-entities/session-input.entity");
|
|
33
|
+
const ai_action_entity_1 = require("../core-entities/ai-action.entity");
|
|
34
|
+
const ai_actions_service_1 = require("../ai-actions/ai-actions.service");
|
|
35
|
+
const execution_logs_service_1 = require("../execution-logs/execution-logs.service");
|
|
36
|
+
const events_gateway_1 = require("../events/events.gateway");
|
|
37
|
+
const MAX_ITERATIONS = 1000;
|
|
38
|
+
let SubAgentRunner = SubAgentRunner_1 = class SubAgentRunner {
|
|
39
|
+
constructor(subAgentsService, subAgentRunsService, sessionsService, applicationStateService, executionLogsService, eventsGateway, llmProvider, moduleRef, sessionInputsRepository, aiActionsRepository) {
|
|
40
|
+
this.subAgentsService = subAgentsService;
|
|
41
|
+
this.subAgentRunsService = subAgentRunsService;
|
|
42
|
+
this.sessionsService = sessionsService;
|
|
43
|
+
this.applicationStateService = applicationStateService;
|
|
44
|
+
this.executionLogsService = executionLogsService;
|
|
45
|
+
this.eventsGateway = eventsGateway;
|
|
46
|
+
this.llmProvider = llmProvider;
|
|
47
|
+
this.moduleRef = moduleRef;
|
|
48
|
+
this.sessionInputsRepository = sessionInputsRepository;
|
|
49
|
+
this.aiActionsRepository = aiActionsRepository;
|
|
50
|
+
this.logger = new common_1.Logger(SubAgentRunner_1.name);
|
|
51
|
+
}
|
|
52
|
+
getHandlerRegistry() {
|
|
53
|
+
return this.moduleRef.get('ACTION_HANDLER_REGISTRY', { strict: false });
|
|
54
|
+
}
|
|
55
|
+
getToolSchemaService() {
|
|
56
|
+
return this.moduleRef.get(tool_schema_service_1.ToolSchemaService, { strict: false });
|
|
57
|
+
}
|
|
58
|
+
getHistoryCompressionService() {
|
|
59
|
+
return this.moduleRef.get(history_compression_service_1.HistoryCompressionService, { strict: false });
|
|
60
|
+
}
|
|
61
|
+
getSystemPromptsService() {
|
|
62
|
+
return this.moduleRef.get(system_prompts_service_1.SystemPromptsService, { strict: false });
|
|
63
|
+
}
|
|
64
|
+
getMcpService() {
|
|
65
|
+
return this.moduleRef.get(mcp_service_1.McpService, { strict: false });
|
|
66
|
+
}
|
|
67
|
+
async runAgent(agentName, prompt, parentSessionId, options) {
|
|
68
|
+
this.logger.log(`runAgent called for "${agentName}" from parent session ${parentSessionId}`);
|
|
69
|
+
const subAgent = await this.subAgentsService.getAgentByName(agentName);
|
|
70
|
+
if (!subAgent) {
|
|
71
|
+
const availableAgents = (await this.subAgentsService.findActive())
|
|
72
|
+
.map((a) => a.name)
|
|
73
|
+
.join(', ');
|
|
74
|
+
return {
|
|
75
|
+
content: `Sub-agent "${agentName}" not found or not active. Available: ${availableAgents}`,
|
|
76
|
+
success: false,
|
|
77
|
+
iterations: 0,
|
|
78
|
+
childSessionId: '',
|
|
79
|
+
};
|
|
80
|
+
}
|
|
81
|
+
const startTime = Date.now();
|
|
82
|
+
const handlerRegistry = this.getHandlerRegistry();
|
|
83
|
+
const toolSchemaService = this.getToolSchemaService();
|
|
84
|
+
const historyCompressionService = this.getHistoryCompressionService();
|
|
85
|
+
const systemPromptsService = this.getSystemPromptsService();
|
|
86
|
+
let childSession;
|
|
87
|
+
let existingHistory = [];
|
|
88
|
+
if (options?.sessionId) {
|
|
89
|
+
childSession = await this.sessionsService.findOne(options.sessionId);
|
|
90
|
+
if (!childSession) {
|
|
91
|
+
return {
|
|
92
|
+
content: `Session "${options.sessionId}" not found. Cannot reuse a non-existent session.`,
|
|
93
|
+
success: false,
|
|
94
|
+
iterations: 0,
|
|
95
|
+
childSessionId: options.sessionId,
|
|
96
|
+
};
|
|
97
|
+
}
|
|
98
|
+
this.logger.log(`Reusing existing session ${childSession.id} for sub-agent "${agentName}"`);
|
|
99
|
+
const existingInputs = await this.sessionInputsRepository.find({
|
|
100
|
+
where: { session: { id: childSession.id } },
|
|
101
|
+
order: { sequence_number: 'ASC' },
|
|
102
|
+
});
|
|
103
|
+
existingHistory = this.reconstructHistory(existingInputs);
|
|
104
|
+
}
|
|
105
|
+
else {
|
|
106
|
+
childSession = await this.sessionsService.create({
|
|
107
|
+
session_title: `Sub-Agent: ${agentName}`,
|
|
108
|
+
system_prompt_id: subAgent.system_prompt_id || undefined,
|
|
109
|
+
default_initial_context_template_id: subAgent.context_template_id || undefined,
|
|
110
|
+
default_followup_context_template_id: subAgent.followup_context_template_id || undefined,
|
|
111
|
+
model_id: subAgent.model_id || undefined,
|
|
112
|
+
sub_agent_id: subAgent.id,
|
|
113
|
+
parent_session_id: parentSessionId,
|
|
114
|
+
});
|
|
115
|
+
this.logger.log(`Created child session ${childSession.id} for sub-agent "${agentName}"`);
|
|
116
|
+
}
|
|
117
|
+
await this.subAgentRunsService.create({
|
|
118
|
+
parent_session_id: parentSessionId,
|
|
119
|
+
child_session_id: childSession.id,
|
|
120
|
+
sub_agent_id: subAgent.id,
|
|
121
|
+
agent_name: agentName,
|
|
122
|
+
initial_prompt: prompt,
|
|
123
|
+
});
|
|
124
|
+
let systemInstruction = '';
|
|
125
|
+
if (subAgent.system_prompt_id) {
|
|
126
|
+
const rendered = await systemPromptsService.findOneWithSession(subAgent.system_prompt_id, childSession.id, childSession.session_title);
|
|
127
|
+
systemInstruction = rendered?.prompt_content || '';
|
|
128
|
+
}
|
|
129
|
+
else {
|
|
130
|
+
const defaultPrompt = await systemPromptsService.findDefaultWithSession(childSession.id, childSession.session_title);
|
|
131
|
+
systemInstruction = defaultPrompt?.prompt_content || '';
|
|
132
|
+
}
|
|
133
|
+
const mcpService = this.getMcpService();
|
|
134
|
+
let tools;
|
|
135
|
+
if (subAgent.enabled_tools && subAgent.enabled_tools.length > 0) {
|
|
136
|
+
const filteredHandlers = new Map();
|
|
137
|
+
for (const toolName of subAgent.enabled_tools) {
|
|
138
|
+
const handler = handlerRegistry.get(toolName);
|
|
139
|
+
if (handler) {
|
|
140
|
+
filteredHandlers.set(toolName, handler);
|
|
141
|
+
}
|
|
142
|
+
}
|
|
143
|
+
tools = toolSchemaService.generateToolDefinitions(filteredHandlers);
|
|
144
|
+
}
|
|
145
|
+
if (subAgent.enabled_mcp_tools !== null &&
|
|
146
|
+
subAgent.enabled_mcp_tools !== undefined) {
|
|
147
|
+
try {
|
|
148
|
+
const allMcpTools = await mcpService.getActiveMcpToolDefinitions();
|
|
149
|
+
if (allMcpTools.length > 0) {
|
|
150
|
+
let enabledMcpToolNames;
|
|
151
|
+
if (subAgent.enabled_mcp_tools === 'all') {
|
|
152
|
+
enabledMcpToolNames = allMcpTools.map((t) => t.function.name);
|
|
153
|
+
}
|
|
154
|
+
else {
|
|
155
|
+
enabledMcpToolNames =
|
|
156
|
+
this.subAgentsService.parseEnabledMcpTools(subAgent.enabled_mcp_tools) || [];
|
|
157
|
+
}
|
|
158
|
+
const filteredMcpTools = allMcpTools.filter((tool) => enabledMcpToolNames.includes(tool.function.name));
|
|
159
|
+
if (filteredMcpTools.length > 0) {
|
|
160
|
+
tools = [...(tools || []), ...filteredMcpTools];
|
|
161
|
+
this.logger.log(`Added ${filteredMcpTools.length} MCP tool definitions for sub-agent "${agentName}".`);
|
|
162
|
+
}
|
|
163
|
+
}
|
|
164
|
+
}
|
|
165
|
+
catch (error) {
|
|
166
|
+
this.logger.warn(`Failed to load MCP tool definitions for sub-agent "${agentName}": ${error.message}. Continuing without MCP tools.`);
|
|
167
|
+
}
|
|
168
|
+
}
|
|
169
|
+
const history = [
|
|
170
|
+
...existingHistory,
|
|
171
|
+
{ role: 'user', parts: [{ text: prompt }] },
|
|
172
|
+
];
|
|
173
|
+
await this.sessionInputsRepository.save(this.sessionInputsRepository.create({
|
|
174
|
+
session: { id: childSession.id },
|
|
175
|
+
role: 'user',
|
|
176
|
+
user_prompt: prompt,
|
|
177
|
+
execution_strategy: 'auto_apply',
|
|
178
|
+
}));
|
|
179
|
+
this.eventsGateway.sendToAll('refresh-ui', {
|
|
180
|
+
sessionId: childSession.id,
|
|
181
|
+
});
|
|
182
|
+
let iterations = 0;
|
|
183
|
+
let finalContent = '';
|
|
184
|
+
let success = true;
|
|
185
|
+
try {
|
|
186
|
+
const abortController = new AbortController();
|
|
187
|
+
const executionStrategy = await this.applicationStateService.getExecutionStrategy();
|
|
188
|
+
while (iterations < MAX_ITERATIONS) {
|
|
189
|
+
iterations++;
|
|
190
|
+
this.logger.log(`Sub-agent "${agentName}" iteration ${iterations}/${MAX_ITERATIONS}`);
|
|
191
|
+
const compressedHistory = await historyCompressionService.compress(history);
|
|
192
|
+
const llmResponse = await this.callWithRetry({
|
|
193
|
+
prompt: '',
|
|
194
|
+
systemInstruction,
|
|
195
|
+
history: compressedHistory,
|
|
196
|
+
modelId: childSession.model_id || undefined,
|
|
197
|
+
tools,
|
|
198
|
+
abortController,
|
|
199
|
+
}, childSession.id);
|
|
200
|
+
if (!llmResponse.text &&
|
|
201
|
+
(!llmResponse.tool_calls || llmResponse.tool_calls.length === 0)) {
|
|
202
|
+
this.logger.log(`Sub-agent "${agentName}" returned empty response with no tool calls. Ending loop.`);
|
|
203
|
+
finalContent = llmResponse.text || '(No response from sub-agent)';
|
|
204
|
+
break;
|
|
205
|
+
}
|
|
206
|
+
const modelMessage = {
|
|
207
|
+
role: 'model',
|
|
208
|
+
parts: [{ text: llmResponse.text || '' }],
|
|
209
|
+
};
|
|
210
|
+
if (llmResponse.tool_calls && llmResponse.tool_calls.length > 0) {
|
|
211
|
+
modelMessage.tool_calls = llmResponse.tool_calls;
|
|
212
|
+
}
|
|
213
|
+
if (llmResponse.thoughts) {
|
|
214
|
+
modelMessage.thoughts = llmResponse.thoughts;
|
|
215
|
+
}
|
|
216
|
+
history.push(modelMessage);
|
|
217
|
+
const modelInput = this.sessionInputsRepository.create({
|
|
218
|
+
session: { id: childSession.id },
|
|
219
|
+
role: 'model',
|
|
220
|
+
raw_llm_response: llmResponse.text || '',
|
|
221
|
+
tool_calls: llmResponse.tool_calls
|
|
222
|
+
? JSON.stringify(llmResponse.tool_calls)
|
|
223
|
+
: null,
|
|
224
|
+
thoughts: llmResponse.thoughts || null,
|
|
225
|
+
input_token_count: llmResponse.usage?.inputTokens || null,
|
|
226
|
+
output_token_count: llmResponse.usage?.outputTokens || null,
|
|
227
|
+
cached_token_count: llmResponse.usage?.cachedTokens || null,
|
|
228
|
+
execution_strategy: 'auto_apply',
|
|
229
|
+
});
|
|
230
|
+
await this.sessionInputsRepository.save(modelInput);
|
|
231
|
+
this.eventsGateway.sendToAll('refresh-ui', {
|
|
232
|
+
sessionId: childSession.id,
|
|
233
|
+
});
|
|
234
|
+
if (!llmResponse.tool_calls || llmResponse.tool_calls.length === 0) {
|
|
235
|
+
this.logger.log(`Sub-agent "${agentName}" returned text with no tool calls. Ending loop.`);
|
|
236
|
+
finalContent = llmResponse.text || '';
|
|
237
|
+
break;
|
|
238
|
+
}
|
|
239
|
+
const executionContext = new llm_orchestration_interfaces_1.PlanExecutionContext();
|
|
240
|
+
executionContext.session_id = childSession.id;
|
|
241
|
+
executionContext.system_prompt_id = subAgent.system_prompt_id || null;
|
|
242
|
+
let shouldBreak = false;
|
|
243
|
+
const toolResults = [];
|
|
244
|
+
for (const toolCall of llmResponse.tool_calls) {
|
|
245
|
+
const toolName = toolCall.function.name;
|
|
246
|
+
const args = typeof toolCall.function.arguments === 'string'
|
|
247
|
+
? JSON.parse(toolCall.function.arguments)
|
|
248
|
+
: toolCall.function.arguments;
|
|
249
|
+
this.logger.log(`Sub-agent "${agentName}" calling tool: ${toolName}`);
|
|
250
|
+
if (toolName === 'final') {
|
|
251
|
+
const plainText = args.plain || args.content || '';
|
|
252
|
+
finalContent = plainText;
|
|
253
|
+
shouldBreak = true;
|
|
254
|
+
const actionEntity = this.aiActionsRepository.create({
|
|
255
|
+
input_id: modelInput.id,
|
|
256
|
+
sessionInput: { id: modelInput.id },
|
|
257
|
+
action_type: 'final',
|
|
258
|
+
status: ai_actions_service_1.AIActionStatus.CONFIRMED_KEPT,
|
|
259
|
+
order_of_execution: toolResults.length,
|
|
260
|
+
original_content_for_revert: null,
|
|
261
|
+
tool_call_id: toolCall.id || null,
|
|
262
|
+
plain: plainText,
|
|
263
|
+
selections: args.selections || null,
|
|
264
|
+
});
|
|
265
|
+
await this.aiActionsRepository.save(actionEntity);
|
|
266
|
+
toolResults.push({
|
|
267
|
+
toolCallId: toolCall.id || '',
|
|
268
|
+
toolName,
|
|
269
|
+
result: 'Task completed.',
|
|
270
|
+
});
|
|
271
|
+
continue;
|
|
272
|
+
}
|
|
273
|
+
let handler = handlerRegistry.get(toolName);
|
|
274
|
+
if (!handler) {
|
|
275
|
+
const mcpMatch = toolName.match(/^([^_]+)__(.+)$/);
|
|
276
|
+
if (mcpMatch) {
|
|
277
|
+
const [, serverName, toolNameRaw] = mcpMatch;
|
|
278
|
+
handler = new mcp_tool_action_handler_1.McpToolActionHandler(serverName, toolNameRaw, mcpService, '');
|
|
279
|
+
}
|
|
280
|
+
}
|
|
281
|
+
if (!handler) {
|
|
282
|
+
this.logger.warn(`No handler found for tool: ${toolName}. Skipping.`);
|
|
283
|
+
const errorMsg = `Tool '${toolName}' is not a valid tool. Please use a valid tool.`;
|
|
284
|
+
toolResults.push({
|
|
285
|
+
toolCallId: toolCall.id || '',
|
|
286
|
+
toolName,
|
|
287
|
+
result: errorMsg,
|
|
288
|
+
});
|
|
289
|
+
const actionEntity = this.aiActionsRepository.create({
|
|
290
|
+
input_id: modelInput.id,
|
|
291
|
+
sessionInput: { id: modelInput.id },
|
|
292
|
+
action_type: toolName,
|
|
293
|
+
status: ai_actions_service_1.AIActionStatus.EXECUTION_FAILED,
|
|
294
|
+
order_of_execution: toolResults.length,
|
|
295
|
+
original_content_for_revert: null,
|
|
296
|
+
tool_call_id: toolCall.id || null,
|
|
297
|
+
});
|
|
298
|
+
await this.aiActionsRepository.save(actionEntity);
|
|
299
|
+
continue;
|
|
300
|
+
}
|
|
301
|
+
let result;
|
|
302
|
+
try {
|
|
303
|
+
result = await handler.execute(args, executionContext);
|
|
304
|
+
}
|
|
305
|
+
catch (error) {
|
|
306
|
+
this.logger.error(`Handler for ${toolName} failed: ${error.message}`);
|
|
307
|
+
result = {
|
|
308
|
+
status: 'FAILURE',
|
|
309
|
+
summary: `An unexpected error occurred: ${error.message}`,
|
|
310
|
+
error_message: error.message,
|
|
311
|
+
persisted_args: args,
|
|
312
|
+
};
|
|
313
|
+
}
|
|
314
|
+
const successStatus = this.getSuccessStatusForStrategy(executionStrategy);
|
|
315
|
+
const actionEntity = this.aiActionsRepository.create({
|
|
316
|
+
input_id: modelInput.id,
|
|
317
|
+
sessionInput: { id: modelInput.id },
|
|
318
|
+
action_type: toolName,
|
|
319
|
+
status: result.status === 'SUCCESS'
|
|
320
|
+
? successStatus
|
|
321
|
+
: ai_actions_service_1.AIActionStatus.EXECUTION_FAILED,
|
|
322
|
+
order_of_execution: toolResults.length,
|
|
323
|
+
original_content_for_revert: result.original_content_for_revert || null,
|
|
324
|
+
tool_call_id: toolCall.id || null,
|
|
325
|
+
...result.persisted_args,
|
|
326
|
+
});
|
|
327
|
+
const savedAction = await this.aiActionsRepository.save(actionEntity);
|
|
328
|
+
if (result.execution_log) {
|
|
329
|
+
await this.executionLogsService.createLog({
|
|
330
|
+
action_id: savedAction.id,
|
|
331
|
+
output: result.execution_log.output,
|
|
332
|
+
error_message: result.execution_log.error_message,
|
|
333
|
+
});
|
|
334
|
+
}
|
|
335
|
+
const output = result.execution_log?.output || '';
|
|
336
|
+
const error = result.execution_log?.error_message || result.error_message || '';
|
|
337
|
+
const combinedResult = error
|
|
338
|
+
? `${output}\n\nError: ${error}`.trim()
|
|
339
|
+
: output;
|
|
340
|
+
toolResults.push({
|
|
341
|
+
toolCallId: toolCall.id || '',
|
|
342
|
+
toolName,
|
|
343
|
+
result: combinedResult,
|
|
344
|
+
});
|
|
345
|
+
if (executionContext.flags.should_halt) {
|
|
346
|
+
this.logger.log(`Execution context halted. Ending sub-agent loop.`);
|
|
347
|
+
shouldBreak = true;
|
|
348
|
+
break;
|
|
349
|
+
}
|
|
350
|
+
}
|
|
351
|
+
for (const tr of toolResults) {
|
|
352
|
+
history.push({
|
|
353
|
+
role: 'tool',
|
|
354
|
+
tool_call_id: tr.toolCallId,
|
|
355
|
+
tool_name: tr.toolName,
|
|
356
|
+
parts: [{ text: tr.result }],
|
|
357
|
+
});
|
|
358
|
+
}
|
|
359
|
+
await this.sessionInputsRepository.save(this.sessionInputsRepository.create({
|
|
360
|
+
session: { id: childSession.id },
|
|
361
|
+
role: 'user',
|
|
362
|
+
user_prompt: '',
|
|
363
|
+
execution_strategy: 'auto_apply',
|
|
364
|
+
raw_llm_response: null,
|
|
365
|
+
tool_calls: JSON.stringify(toolResults.map((tr) => ({
|
|
366
|
+
id: tr.toolCallId,
|
|
367
|
+
type: 'function',
|
|
368
|
+
function: { name: tr.toolName, arguments: '{}' },
|
|
369
|
+
}))),
|
|
370
|
+
}));
|
|
371
|
+
this.eventsGateway.sendToAll('refresh-ui', {
|
|
372
|
+
sessionId: childSession.id,
|
|
373
|
+
});
|
|
374
|
+
if (shouldBreak) {
|
|
375
|
+
break;
|
|
376
|
+
}
|
|
377
|
+
}
|
|
378
|
+
if (iterations >= MAX_ITERATIONS) {
|
|
379
|
+
this.logger.warn(`Sub-agent "${agentName}" reached max iterations (${MAX_ITERATIONS}).`);
|
|
380
|
+
if (!finalContent) {
|
|
381
|
+
finalContent =
|
|
382
|
+
'Sub-agent reached maximum iterations without completing.';
|
|
383
|
+
success = false;
|
|
384
|
+
}
|
|
385
|
+
}
|
|
386
|
+
await this.subAgentRunsService.markCompleted(childSession.id, finalContent.substring(0, 5000));
|
|
387
|
+
this.eventsGateway.sendToAll('refresh-ui', {
|
|
388
|
+
sessionId: childSession.id,
|
|
389
|
+
});
|
|
390
|
+
}
|
|
391
|
+
catch (error) {
|
|
392
|
+
this.logger.error(`Sub-agent "${agentName}" failed: ${error.message}`, error.stack);
|
|
393
|
+
success = false;
|
|
394
|
+
finalContent = `Sub-agent "${agentName}" encountered an error: ${error.message}`;
|
|
395
|
+
await this.subAgentRunsService.markFailed(childSession.id, error.message);
|
|
396
|
+
this.eventsGateway.sendToAll('refresh-ui', {
|
|
397
|
+
sessionId: childSession.id,
|
|
398
|
+
});
|
|
399
|
+
}
|
|
400
|
+
const durationMs = Date.now() - startTime;
|
|
401
|
+
this.logger.log(`Sub-agent "${agentName}" completed in ${durationMs}ms, ${iterations} iterations, success: ${success}`);
|
|
402
|
+
return {
|
|
403
|
+
content: finalContent,
|
|
404
|
+
success,
|
|
405
|
+
iterations,
|
|
406
|
+
childSessionId: childSession.id,
|
|
407
|
+
};
|
|
408
|
+
}
|
|
409
|
+
reconstructHistory(inputs) {
|
|
410
|
+
const history = [];
|
|
411
|
+
for (const input of inputs) {
|
|
412
|
+
if (input.role === 'user') {
|
|
413
|
+
const content = {
|
|
414
|
+
role: 'user',
|
|
415
|
+
parts: [{ text: input.user_prompt || input.raw_llm_response || '' }],
|
|
416
|
+
};
|
|
417
|
+
history.push(content);
|
|
418
|
+
}
|
|
419
|
+
else if (input.role === 'model') {
|
|
420
|
+
const content = {
|
|
421
|
+
role: 'model',
|
|
422
|
+
parts: [{ text: input.raw_llm_response || '' }],
|
|
423
|
+
};
|
|
424
|
+
if (input.tool_calls) {
|
|
425
|
+
try {
|
|
426
|
+
const parsed = JSON.parse(input.tool_calls);
|
|
427
|
+
if (Array.isArray(parsed)) {
|
|
428
|
+
content.tool_calls = parsed;
|
|
429
|
+
}
|
|
430
|
+
}
|
|
431
|
+
catch {
|
|
432
|
+
}
|
|
433
|
+
}
|
|
434
|
+
history.push(content);
|
|
435
|
+
}
|
|
436
|
+
}
|
|
437
|
+
return history;
|
|
438
|
+
}
|
|
439
|
+
async callWithRetry(request, sessionId) {
|
|
440
|
+
const retryEnabled = await this.applicationStateService.getLlmRetryEnabled();
|
|
441
|
+
const maxAttempts = await this.applicationStateService.getLlmRetryMaxAttempts();
|
|
442
|
+
let lastError;
|
|
443
|
+
let attempt = 1;
|
|
444
|
+
while (attempt <= maxAttempts) {
|
|
445
|
+
try {
|
|
446
|
+
if (attempt > 1) {
|
|
447
|
+
this.logger.log(`Retry attempt ${attempt}/${maxAttempts} for sub-agent session ${sessionId}`);
|
|
448
|
+
}
|
|
449
|
+
return await this.llmProvider.generateContent({
|
|
450
|
+
...request,
|
|
451
|
+
abortController: request.abortController,
|
|
452
|
+
});
|
|
453
|
+
}
|
|
454
|
+
catch (error) {
|
|
455
|
+
lastError = error;
|
|
456
|
+
if (error.name === 'AbortError' || error.code === 'ABORT_ERR') {
|
|
457
|
+
throw error;
|
|
458
|
+
}
|
|
459
|
+
const shouldRetry = retryEnabled && attempt < maxAttempts && this.isRetryableError(error);
|
|
460
|
+
if (!shouldRetry) {
|
|
461
|
+
throw error;
|
|
462
|
+
}
|
|
463
|
+
const backoffMs = Math.pow(2, attempt - 1) * 1000;
|
|
464
|
+
this.logger.log(`LLM call failed (attempt ${attempt}/${maxAttempts}): ${error.message}. Retrying in ${backoffMs}ms...`);
|
|
465
|
+
await new Promise((resolve) => setTimeout(resolve, backoffMs));
|
|
466
|
+
attempt++;
|
|
467
|
+
}
|
|
468
|
+
}
|
|
469
|
+
throw lastError;
|
|
470
|
+
}
|
|
471
|
+
isRetryableError(error) {
|
|
472
|
+
const errorMessage = error.message?.toLowerCase() || '';
|
|
473
|
+
const errorCode = error.code;
|
|
474
|
+
const status = error.status;
|
|
475
|
+
if (error.name === 'AbortError' || error.code === 'ABORT_ERR')
|
|
476
|
+
return false;
|
|
477
|
+
const networkErrors = [
|
|
478
|
+
'ECONNRESET',
|
|
479
|
+
'ETIMEDOUT',
|
|
480
|
+
'ENOTFOUND',
|
|
481
|
+
'ENETUNREACH',
|
|
482
|
+
'ECONNREFUSED',
|
|
483
|
+
];
|
|
484
|
+
if (errorCode && networkErrors.includes(errorCode))
|
|
485
|
+
return true;
|
|
486
|
+
if (status === 429 ||
|
|
487
|
+
errorMessage.includes('rate limit') ||
|
|
488
|
+
errorMessage.includes('too many requests'))
|
|
489
|
+
return true;
|
|
490
|
+
if (status >= 500 && status < 600)
|
|
491
|
+
return true;
|
|
492
|
+
if (errorMessage.includes('empty response') ||
|
|
493
|
+
errorMessage.includes('timeout'))
|
|
494
|
+
return true;
|
|
495
|
+
return false;
|
|
496
|
+
}
|
|
497
|
+
getSuccessStatusForStrategy(strategy) {
|
|
498
|
+
switch (strategy) {
|
|
499
|
+
case 'auto_apply':
|
|
500
|
+
return ai_actions_service_1.AIActionStatus.CONFIRMED_KEPT;
|
|
501
|
+
case 'review_first':
|
|
502
|
+
case 'apply_revert':
|
|
503
|
+
default:
|
|
504
|
+
return ai_actions_service_1.AIActionStatus.APPLIED_PENDING_REVIEW;
|
|
505
|
+
}
|
|
506
|
+
}
|
|
507
|
+
};
|
|
508
|
+
exports.SubAgentRunner = SubAgentRunner;
|
|
509
|
+
exports.SubAgentRunner = SubAgentRunner = SubAgentRunner_1 = __decorate([
|
|
510
|
+
(0, common_1.Injectable)(),
|
|
511
|
+
__param(6, (0, common_1.Inject)(llm_provider_interface_1.LLM_PROVIDER)),
|
|
512
|
+
__param(8, (0, typeorm_1.InjectRepository)(session_input_entity_1.SessionInput)),
|
|
513
|
+
__param(9, (0, typeorm_1.InjectRepository)(ai_action_entity_1.AIAction)),
|
|
514
|
+
__metadata("design:paramtypes", [sub_agents_service_1.SubAgentsService,
|
|
515
|
+
sub_agent_runs_service_1.SubAgentRunsService,
|
|
516
|
+
sessions_service_1.SessionsService,
|
|
517
|
+
application_state_service_1.ApplicationStateService,
|
|
518
|
+
execution_logs_service_1.ExecutionLogsService,
|
|
519
|
+
events_gateway_1.EventsGateway, Object, core_1.ModuleRef,
|
|
520
|
+
typeorm_2.Repository,
|
|
521
|
+
typeorm_2.Repository])
|
|
522
|
+
], SubAgentRunner);
|
|
523
|
+
//# sourceMappingURL=sub-agent-runner.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sub-agent-runner.service.js","sourceRoot":"","sources":["../../../src/sub-agents/sub-agent-runner.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAA4D;AAC5D,uCAAyC;AACzC,mFAKgD;AAChD,6DAAwD;AACxD,qEAA+D;AAC/D,mEAA+D;AAC/D,qFAAgF;AAChF,kFAA6E;AAC7E,kGAA6F;AAC7F,8FAAyF;AAEzF,0GAAoG;AACpG,oDAAgD;AAChD,oGAG2D;AAE3D,6CAAmD;AACnD,qCAAqC;AACrC,gFAAqE;AACrE,wEAA6D;AAC7D,yEAAkE;AAClE,qFAAgF;AAChF,6DAAyD;AAiBzD,MAAM,cAAc,GAAG,IAAI,CAAC;AAGrB,IAAM,cAAc,sBAApB,MAAM,cAAc;IAGzB,YACmB,gBAAkC,EAClC,mBAAwC,EACxC,eAAgC,EAChC,uBAAgD,EAChD,oBAA0C,EAC1C,aAA4B,EAE7C,WAAyC,EACxB,SAAoB,EAErC,uBAAkE,EAElE,mBAA0D;QAZzC,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,wBAAmB,GAAnB,mBAAmB,CAAqB;QACxC,oBAAe,GAAf,eAAe,CAAiB;QAChC,4BAAuB,GAAvB,uBAAuB,CAAyB;QAChD,yBAAoB,GAApB,oBAAoB,CAAsB;QAC1C,kBAAa,GAAb,aAAa,CAAe;QAE5B,gBAAW,GAAX,WAAW,CAAa;QACxB,cAAS,GAAT,SAAS,CAAW;QAEpB,4BAAuB,GAAvB,uBAAuB,CAA0B;QAEjD,wBAAmB,GAAnB,mBAAmB,CAAsB;QAf3C,WAAM,GAAG,IAAI,eAAM,CAAC,gBAAc,CAAC,IAAI,CAAC,CAAC;IAgBvD,CAAC;IAMI,kBAAkB;QACxB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,yBAAyB,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1E,CAAC;IAKO,oBAAoB;QAC1B,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,uCAAiB,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IAClE,CAAC;IAKO,4BAA4B;QAClC,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,uDAAyB,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1E,CAAC;IAMO,uBAAuB;QAC7B,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,6CAAoB,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IACrE,CAAC;IAKO,aAAa;QACnB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,wBAAU,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IAC3D,CAAC;IAMD,KAAK,CAAC,QAAQ,CACZ,SAAiB,EACjB,MAAc,EACd,eAAuB,EACvB,OAAyB;QAEzB,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,wBAAwB,SAAS,yBAAyB,eAAe,EAAE,CAC5E,CAAC;QAGF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QACvE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,eAAe,GAAG,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,CAAC;iBAC/D,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;iBAClB,IAAI,CAAC,IAAI,CAAC,CAAC;YACd,OAAO;gBACL,OAAO,EAAE,cAAc,SAAS,yCAAyC,eAAe,EAAE;gBAC1F,OAAO,EAAE,KAAK;gBACd,UAAU,EAAE,CAAC;gBACb,cAAc,EAAE,EAAE;aACnB,CAAC;QACJ,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,eAAe,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAClD,MAAM,iBAAiB,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAC;QACtD,MAAM,yBAAyB,GAAG,IAAI,CAAC,4BAA4B,EAAE,CAAC;QACtE,MAAM,oBAAoB,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAG5D,IAAI,YAAqB,CAAC;QAC1B,IAAI,eAAe,GAAiB,EAAE,CAAC;QAEvC,IAAI,OAAO,EAAE,SAAS,EAAE,CAAC;YAEvB,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACrE,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,OAAO;oBACL,OAAO,EAAE,YAAY,OAAO,CAAC,SAAS,mDAAmD;oBACzF,OAAO,EAAE,KAAK;oBACd,UAAU,EAAE,CAAC;oBACb,cAAc,EAAE,OAAO,CAAC,SAAS;iBAClC,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,4BAA4B,YAAY,CAAC,EAAE,mBAAmB,SAAS,GAAG,CAC3E,CAAC;YAGF,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC;gBAC7D,KAAK,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,YAAY,CAAC,EAAE,EAAE,EAAE;gBAC3C,KAAK,EAAE,EAAE,eAAe,EAAE,KAAc,EAAE;aAC3C,CAAC,CAAC;YACH,eAAe,GAAG,IAAI,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC5D,CAAC;aAAM,CAAC;YAEN,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC;gBAC/C,aAAa,EAAE,cAAc,SAAS,EAAE;gBACxC,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB,IAAI,SAAS;gBACxD,mCAAmC,EACjC,QAAQ,CAAC,mBAAmB,IAAI,SAAS;gBAC3C,oCAAoC,EAClC,QAAQ,CAAC,4BAA4B,IAAI,SAAS;gBACpD,QAAQ,EAAE,QAAQ,CAAC,QAAQ,IAAI,SAAS;gBACxC,YAAY,EAAE,QAAQ,CAAC,EAAE;gBACzB,iBAAiB,EAAE,eAAe;aACnC,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,yBAAyB,YAAY,CAAC,EAAE,mBAAmB,SAAS,GAAG,CACxE,CAAC;QACJ,CAAC;QAGD,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC;YACpC,iBAAiB,EAAE,eAAe;YAClC,gBAAgB,EAAE,YAAY,CAAC,EAAE;YACjC,YAAY,EAAE,QAAQ,CAAC,EAAE;YACzB,UAAU,EAAE,SAAS;YACrB,cAAc,EAAE,MAAM;SACvB,CAAC,CAAC;QAGH,IAAI,iBAAiB,GAAG,EAAE,CAAC;QAC3B,IAAI,QAAQ,CAAC,gBAAgB,EAAE,CAAC;YAC9B,MAAM,QAAQ,GAAG,MAAM,oBAAoB,CAAC,kBAAkB,CAC5D,QAAQ,CAAC,gBAAgB,EACzB,YAAY,CAAC,EAAE,EACf,YAAY,CAAC,aAAa,CAC3B,CAAC;YACF,iBAAiB,GAAG,QAAQ,EAAE,cAAc,IAAI,EAAE,CAAC;QACrD,CAAC;aAAM,CAAC;YACN,MAAM,aAAa,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CACrE,YAAY,CAAC,EAAE,EACf,YAAY,CAAC,aAAa,CAC3B,CAAC;YACF,iBAAiB,GAAG,aAAa,EAAE,cAAc,IAAI,EAAE,CAAC;QAC1D,CAAC;QAGD,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QACxC,IAAI,KAAwB,CAAC;QAC7B,IAAI,QAAQ,CAAC,aAAa,IAAI,QAAQ,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAEhE,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAyB,CAAC;YAC1D,KAAK,MAAM,QAAQ,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;gBAC9C,MAAM,OAAO,GAAG,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBAC9C,IAAI,OAAO,EAAE,CAAC;oBACZ,gBAAgB,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC;YACD,KAAK,GAAG,iBAAiB,CAAC,uBAAuB,CAAC,gBAAgB,CAAC,CAAC;QACtE,CAAC;QAGD,IACE,QAAQ,CAAC,iBAAiB,KAAK,IAAI;YACnC,QAAQ,CAAC,iBAAiB,KAAK,SAAS,EACxC,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,WAAW,GAAG,MAAM,UAAU,CAAC,2BAA2B,EAAE,CAAC;gBACnE,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC3B,IAAI,mBAA6B,CAAC;oBAClC,IAAI,QAAQ,CAAC,iBAAiB,KAAK,KAAK,EAAE,CAAC;wBACzC,mBAAmB,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;oBAChE,CAAC;yBAAM,CAAC;wBAEN,mBAAmB;4BAChB,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,CACzC,QAAQ,CAAC,iBAAiB,CACd,IAAI,EAAE,CAAC;oBACzB,CAAC;oBACD,MAAM,gBAAgB,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CACnD,mBAAmB,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CACjD,CAAC;oBACF,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAChC,KAAK,GAAG,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,GAAG,gBAAgB,CAAC,CAAC;wBAChD,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,SAAS,gBAAgB,CAAC,MAAM,wCAAwC,SAAS,IAAI,CACtF,CAAC;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,sDAAsD,SAAS,MAAM,KAAK,CAAC,OAAO,iCAAiC,CACpH,CAAC;YACJ,CAAC;QACH,CAAC;QAGD,MAAM,OAAO,GAAiB;YAC5B,GAAG,eAAe;YAClB,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE;SAC5C,CAAC;QAGF,MAAM,IAAI,CAAC,uBAAuB,CAAC,IAAI,CACrC,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC;YAClC,OAAO,EAAE,EAAE,EAAE,EAAE,YAAY,CAAC,EAAE,EAAE;YAChC,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,MAAM;YACnB,kBAAkB,EAAE,YAAY;SACjC,CAAC,CACH,CAAC;QAGF,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,YAAY,EAAE;YACzC,SAAS,EAAE,YAAY,CAAC,EAAE;SAC3B,CAAC,CAAC;QAGH,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,YAAY,GAAG,EAAE,CAAC;QACtB,IAAI,OAAO,GAAG,IAAI,CAAC;QAEnB,IAAI,CAAC;YACH,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;YAC9C,MAAM,iBAAiB,GACrB,MAAM,IAAI,CAAC,uBAAuB,CAAC,oBAAoB,EAAE,CAAC;YAE5D,OAAO,UAAU,GAAG,cAAc,EAAE,CAAC;gBACnC,UAAU,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,cAAc,SAAS,eAAe,UAAU,IAAI,cAAc,EAAE,CACrE,CAAC;gBAGF,MAAM,iBAAiB,GACrB,MAAM,yBAAyB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAGpD,MAAM,WAAW,GAAgB,MAAM,IAAI,CAAC,aAAa,CACvD;oBACE,MAAM,EAAE,EAAE;oBACV,iBAAiB;oBACjB,OAAO,EAAE,iBAAiB;oBAC1B,OAAO,EAAE,YAAY,CAAC,QAAQ,IAAI,SAAS;oBAC3C,KAAK;oBACL,eAAe;iBAChB,EACD,YAAY,CAAC,EAAE,CAChB,CAAC;gBAGF,IACE,CAAC,WAAW,CAAC,IAAI;oBACjB,CAAC,CAAC,WAAW,CAAC,UAAU,IAAI,WAAW,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,CAAC,EAChE,CAAC;oBACD,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,cAAc,SAAS,4DAA4D,CACpF,CAAC;oBACF,YAAY,GAAG,WAAW,CAAC,IAAI,IAAI,8BAA8B,CAAC;oBAClE,MAAM;gBACR,CAAC;gBAGD,MAAM,YAAY,GAAe;oBAC/B,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC;iBAC1C,CAAC;gBACF,IAAI,WAAW,CAAC,UAAU,IAAI,WAAW,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAChE,YAAY,CAAC,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC;gBACnD,CAAC;gBACD,IAAI,WAAW,CAAC,QAAQ,EAAE,CAAC;oBACzB,YAAY,CAAC,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC;gBAC/C,CAAC;gBACD,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAG3B,MAAM,UAAU,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC;oBACrD,OAAO,EAAE,EAAE,EAAE,EAAE,YAAY,CAAC,EAAE,EAAE;oBAChC,IAAI,EAAE,OAAO;oBACb,gBAAgB,EAAE,WAAW,CAAC,IAAI,IAAI,EAAE;oBACxC,UAAU,EAAE,WAAW,CAAC,UAAU;wBAChC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,UAAU,CAAC;wBACxC,CAAC,CAAC,IAAI;oBACR,QAAQ,EAAE,WAAW,CAAC,QAAQ,IAAI,IAAI;oBACtC,iBAAiB,EAAE,WAAW,CAAC,KAAK,EAAE,WAAW,IAAI,IAAI;oBACzD,kBAAkB,EAAE,WAAW,CAAC,KAAK,EAAE,YAAY,IAAI,IAAI;oBAC3D,kBAAkB,EAAE,WAAW,CAAC,KAAK,EAAE,YAAY,IAAI,IAAI;oBAC3D,kBAAkB,EAAE,YAAY;iBACjC,CAAC,CAAC;gBACH,MAAM,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBAGpD,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,YAAY,EAAE;oBACzC,SAAS,EAAE,YAAY,CAAC,EAAE;iBAC3B,CAAC,CAAC;gBAGH,IAAI,CAAC,WAAW,CAAC,UAAU,IAAI,WAAW,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACnE,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,cAAc,SAAS,kDAAkD,CAC1E,CAAC;oBACF,YAAY,GAAG,WAAW,CAAC,IAAI,IAAI,EAAE,CAAC;oBACtC,MAAM;gBACR,CAAC;gBAGD,MAAM,gBAAgB,GAAG,IAAI,mDAAoB,EAAE,CAAC;gBACpD,gBAAgB,CAAC,UAAU,GAAG,YAAY,CAAC,EAAE,CAAC;gBAC9C,gBAAgB,CAAC,gBAAgB,GAAG,QAAQ,CAAC,gBAAgB,IAAI,IAAI,CAAC;gBAEtE,IAAI,WAAW,GAAG,KAAK,CAAC;gBACxB,MAAM,WAAW,GAIX,EAAE,CAAC;gBAET,KAAK,MAAM,QAAQ,IAAI,WAAW,CAAC,UAAU,EAAE,CAAC;oBAC9C,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;oBACxC,MAAM,IAAI,GACR,OAAO,QAAQ,CAAC,QAAQ,CAAC,SAAS,KAAK,QAAQ;wBAC7C,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;wBACzC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;oBAElC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,cAAc,SAAS,mBAAmB,QAAQ,EAAE,CAAC,CAAC;oBAGtE,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;wBACzB,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC;wBACnD,YAAY,GAAG,SAAS,CAAC;wBACzB,WAAW,GAAG,IAAI,CAAC;wBAGnB,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC;4BACnD,QAAQ,EAAE,UAAU,CAAC,EAAE;4BACvB,YAAY,EAAE,EAAE,EAAE,EAAE,UAAU,CAAC,EAAE,EAAE;4BACnC,WAAW,EAAE,OAAO;4BACpB,MAAM,EAAE,mCAAc,CAAC,cAAc;4BACrC,kBAAkB,EAAE,WAAW,CAAC,MAAM;4BACtC,2BAA2B,EAAE,IAAI;4BACjC,YAAY,EAAE,QAAQ,CAAC,EAAE,IAAI,IAAI;4BACjC,KAAK,EAAE,SAAS;4BAChB,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI;yBACpC,CAAC,CAAC;wBACH,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;wBAElD,WAAW,CAAC,IAAI,CAAC;4BACf,UAAU,EAAE,QAAQ,CAAC,EAAE,IAAI,EAAE;4BAC7B,QAAQ;4BACR,MAAM,EAAE,iBAAiB;yBAC1B,CAAC,CAAC;wBACH,SAAS;oBACX,CAAC;oBAGD,IAAI,OAAO,GAAG,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;oBAC5C,IAAI,CAAC,OAAO,EAAE,CAAC;wBAGb,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;wBACnD,IAAI,QAAQ,EAAE,CAAC;4BACb,MAAM,CAAC,EAAE,UAAU,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAC;4BAC7C,OAAO,GAAG,IAAI,8CAAoB,CAChC,UAAU,EACV,WAAW,EACX,UAAU,EACV,EAAE,CACH,CAAC;wBACJ,CAAC;oBACH,CAAC;oBACD,IAAI,CAAC,OAAO,EAAE,CAAC;wBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,8BAA8B,QAAQ,aAAa,CACpD,CAAC;wBACF,MAAM,QAAQ,GAAG,SAAS,QAAQ,iDAAiD,CAAC;wBACpF,WAAW,CAAC,IAAI,CAAC;4BACf,UAAU,EAAE,QAAQ,CAAC,EAAE,IAAI,EAAE;4BAC7B,QAAQ;4BACR,MAAM,EAAE,QAAQ;yBACjB,CAAC,CAAC;wBAGH,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC;4BACnD,QAAQ,EAAE,UAAU,CAAC,EAAE;4BACvB,YAAY,EAAE,EAAE,EAAE,EAAE,UAAU,CAAC,EAAE,EAAE;4BACnC,WAAW,EAAE,QAAQ;4BACrB,MAAM,EAAE,mCAAc,CAAC,gBAAgB;4BACvC,kBAAkB,EAAE,WAAW,CAAC,MAAM;4BACtC,2BAA2B,EAAE,IAAI;4BACjC,YAAY,EAAE,QAAQ,CAAC,EAAE,IAAI,IAAI;yBAClC,CAAC,CAAC;wBACH,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;wBAClD,SAAS;oBACX,CAAC;oBAGD,IAAI,MAA6B,CAAC;oBAClC,IAAI,CAAC;wBACH,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;oBACzD,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,eAAe,QAAQ,YAAY,KAAK,CAAC,OAAO,EAAE,CACnD,CAAC;wBACF,MAAM,GAAG;4BACP,MAAM,EAAE,SAAS;4BACjB,OAAO,EAAE,iCAAiC,KAAK,CAAC,OAAO,EAAE;4BACzD,aAAa,EAAE,KAAK,CAAC,OAAO;4BAC5B,cAAc,EAAE,IAAI;yBACrB,CAAC;oBACJ,CAAC;oBAGD,MAAM,aAAa,GACjB,IAAI,CAAC,2BAA2B,CAAC,iBAAiB,CAAC,CAAC;oBACtD,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC;wBACnD,QAAQ,EAAE,UAAU,CAAC,EAAE;wBACvB,YAAY,EAAE,EAAE,EAAE,EAAE,UAAU,CAAC,EAAE,EAAE;wBACnC,WAAW,EAAE,QAAQ;wBACrB,MAAM,EACJ,MAAM,CAAC,MAAM,KAAK,SAAS;4BACzB,CAAC,CAAC,aAAa;4BACf,CAAC,CAAC,mCAAc,CAAC,gBAAgB;wBACrC,kBAAkB,EAAE,WAAW,CAAC,MAAM;wBACtC,2BAA2B,EACzB,MAAM,CAAC,2BAA2B,IAAI,IAAI;wBAC5C,YAAY,EAAE,QAAQ,CAAC,EAAE,IAAI,IAAI;wBACjC,GAAG,MAAM,CAAC,cAAc;qBACzB,CAAC,CAAC;oBACH,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;oBAGtE,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;wBACzB,MAAM,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC;4BACxC,SAAS,EAAE,WAAW,CAAC,EAAE;4BACzB,MAAM,EAAE,MAAM,CAAC,aAAa,CAAC,MAAM;4BACnC,aAAa,EAAE,MAAM,CAAC,aAAa,CAAC,aAAa;yBAClD,CAAC,CAAC;oBACL,CAAC;oBAGD,MAAM,MAAM,GAAG,MAAM,CAAC,aAAa,EAAE,MAAM,IAAI,EAAE,CAAC;oBAClD,MAAM,KAAK,GACT,MAAM,CAAC,aAAa,EAAE,aAAa,IAAI,MAAM,CAAC,aAAa,IAAI,EAAE,CAAC;oBACpE,MAAM,cAAc,GAAG,KAAK;wBAC1B,CAAC,CAAC,GAAG,MAAM,cAAc,KAAK,EAAE,CAAC,IAAI,EAAE;wBACvC,CAAC,CAAC,MAAM,CAAC;oBAEX,WAAW,CAAC,IAAI,CAAC;wBACf,UAAU,EAAE,QAAQ,CAAC,EAAE,IAAI,EAAE;wBAC7B,QAAQ;wBACR,MAAM,EAAE,cAAc;qBACvB,CAAC,CAAC;oBAGH,IAAI,gBAAgB,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;wBACvC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;wBACpE,WAAW,GAAG,IAAI,CAAC;wBACnB,MAAM;oBACR,CAAC;gBACH,CAAC;gBAGD,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;oBAC7B,OAAO,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,MAAM;wBACZ,YAAY,EAAE,EAAE,CAAC,UAAU;wBAC3B,SAAS,EAAE,EAAE,CAAC,QAAQ;wBACtB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC;qBAC7B,CAAC,CAAC;gBACL,CAAC;gBAGD,MAAM,IAAI,CAAC,uBAAuB,CAAC,IAAI,CACrC,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC;oBAClC,OAAO,EAAE,EAAE,EAAE,EAAE,YAAY,CAAC,EAAE,EAAE;oBAChC,IAAI,EAAE,MAAM;oBACZ,WAAW,EAAE,EAAE;oBACf,kBAAkB,EAAE,YAAY;oBAChC,gBAAgB,EAAE,IAAI;oBACtB,UAAU,EAAE,IAAI,CAAC,SAAS,CACxB,WAAW,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;wBACvB,EAAE,EAAE,EAAE,CAAC,UAAU;wBACjB,IAAI,EAAE,UAAU;wBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,IAAI,EAAE;qBACjD,CAAC,CAAC,CACJ;iBACF,CAAC,CACH,CAAC;gBAGF,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,YAAY,EAAE;oBACzC,SAAS,EAAE,YAAY,CAAC,EAAE;iBAC3B,CAAC,CAAC;gBAEH,IAAI,WAAW,EAAE,CAAC;oBAChB,MAAM;gBACR,CAAC;YAGH,CAAC;YAGD,IAAI,UAAU,IAAI,cAAc,EAAE,CAAC;gBACjC,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,cAAc,SAAS,6BAA6B,cAAc,IAAI,CACvE,CAAC;gBACF,IAAI,CAAC,YAAY,EAAE,CAAC;oBAClB,YAAY;wBACV,0DAA0D,CAAC;oBAC7D,OAAO,GAAG,KAAK,CAAC;gBAClB,CAAC;YACH,CAAC;YAGD,MAAM,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAC1C,YAAY,CAAC,EAAE,EACf,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,CAChC,CAAC;YAGF,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,YAAY,EAAE;gBACzC,SAAS,EAAE,YAAY,CAAC,EAAE;aAC3B,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,cAAc,SAAS,aAAa,KAAK,CAAC,OAAO,EAAE,EACnD,KAAK,CAAC,KAAK,CACZ,CAAC;YACF,OAAO,GAAG,KAAK,CAAC;YAChB,YAAY,GAAG,cAAc,SAAS,2BAA2B,KAAK,CAAC,OAAO,EAAE,CAAC;YAGjF,MAAM,IAAI,CAAC,mBAAmB,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YAG1E,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,YAAY,EAAE;gBACzC,SAAS,EAAE,YAAY,CAAC,EAAE;aAC3B,CAAC,CAAC;QACL,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QAC1C,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,cAAc,SAAS,kBAAkB,UAAU,OAAO,UAAU,yBAAyB,OAAO,EAAE,CACvG,CAAC;QAEF,OAAO;YACL,OAAO,EAAE,YAAY;YACrB,OAAO;YACP,UAAU;YACV,cAAc,EAAE,YAAY,CAAC,EAAE;SAChC,CAAC;IACJ,CAAC;IAOO,kBAAkB,CAAC,MAAsB;QAC/C,MAAM,OAAO,GAAiB,EAAE,CAAC;QACjC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,MAAM,OAAO,GAAe;oBAC1B,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,gBAAgB,IAAI,EAAE,EAAE,CAAC;iBACrE,CAAC;gBACF,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACxB,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBAClC,MAAM,OAAO,GAAe;oBAC1B,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,gBAAgB,IAAI,EAAE,EAAE,CAAC;iBAChD,CAAC;gBACF,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACrB,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;wBAC5C,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;4BAC1B,OAAO,CAAC,UAAU,GAAG,MAAM,CAAC;wBAC9B,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;oBAET,CAAC;gBACH,CAAC;gBACD,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACxB,CAAC;QAEH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAKO,KAAK,CAAC,aAAa,CACzB,OAOC,EACD,SAAiB;QAEjB,MAAM,YAAY,GAChB,MAAM,IAAI,CAAC,uBAAuB,CAAC,kBAAkB,EAAE,CAAC;QAC1D,MAAM,WAAW,GACf,MAAM,IAAI,CAAC,uBAAuB,CAAC,sBAAsB,EAAE,CAAC;QAE9D,IAAI,SAA4B,CAAC;QACjC,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,OAAO,OAAO,IAAI,WAAW,EAAE,CAAC;YAC9B,IAAI,CAAC;gBACH,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;oBAChB,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,iBAAiB,OAAO,IAAI,WAAW,0BAA0B,SAAS,EAAE,CAC7E,CAAC;gBACJ,CAAC;gBAED,OAAO,MAAM,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC;oBAC5C,GAAG,OAAO;oBACV,eAAe,EAAE,OAAO,CAAC,eAAe;iBACzC,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,SAAS,GAAG,KAAc,CAAC;gBAG3B,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC9D,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,MAAM,WAAW,GACf,YAAY,IAAI,OAAO,GAAG,WAAW,IAAI,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;gBACxE,IAAI,CAAC,WAAW,EAAE,CAAC;oBACjB,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC;gBAClD,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,4BAA4B,OAAO,IAAI,WAAW,MAAM,KAAK,CAAC,OAAO,iBAAiB,SAAS,OAAO,CACvG,CAAC;gBACF,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;gBAC/D,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QAED,MAAM,SAAS,CAAC;IAClB,CAAC;IAEO,gBAAgB,CAAC,KAAU;QACjC,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QACxD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC;QAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;QAE5B,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW;YAAE,OAAO,KAAK,CAAC;QAE5E,MAAM,aAAa,GAAG;YACpB,YAAY;YACZ,WAAW;YACX,WAAW;YACX,aAAa;YACb,cAAc;SACf,CAAC;QACF,IAAI,SAAS,IAAI,aAAa,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,OAAO,IAAI,CAAC;QAChE,IACE,MAAM,KAAK,GAAG;YACd,YAAY,CAAC,QAAQ,CAAC,YAAY,CAAC;YACnC,YAAY,CAAC,QAAQ,CAAC,mBAAmB,CAAC;YAE1C,OAAO,IAAI,CAAC;QACd,IAAI,MAAM,IAAI,GAAG,IAAI,MAAM,GAAG,GAAG;YAAE,OAAO,IAAI,CAAC;QAC/C,IACE,YAAY,CAAC,QAAQ,CAAC,gBAAgB,CAAC;YACvC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC;YAEhC,OAAO,IAAI,CAAC;QAEd,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,2BAA2B,CAAC,QAAgB;QAClD,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,YAAY;gBACf,OAAO,mCAAc,CAAC,cAAc,CAAC;YACvC,KAAK,cAAc,CAAC;YACpB,KAAK,cAAc,CAAC;YACpB;gBACE,OAAO,mCAAc,CAAC,sBAAsB,CAAC;QACjD,CAAC;IACH,CAAC;CACF,CAAA;AA/rBY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,mBAAU,GAAE;IAWR,WAAA,IAAA,eAAM,EAAC,qCAAY,CAAC,CAAA;IAGpB,WAAA,IAAA,0BAAgB,EAAC,mCAAY,CAAC,CAAA;IAE9B,WAAA,IAAA,0BAAgB,EAAC,2BAAQ,CAAC,CAAA;qCAXQ,qCAAgB;QACb,4CAAmB;QACvB,kCAAe;QACP,mDAAuB;QAC1B,6CAAoB;QAC3B,8BAAa,UAGjB,gBAAS;QAEK,oBAAU;QAEd,oBAAU;GAhBvC,cAAc,CA+rB1B"}
|
|
@@ -1,11 +1,12 @@
|
|
|
1
|
+
import { ModuleRef } from '@nestjs/core';
|
|
1
2
|
import { SubAgentRunsService } from './sub-agent-runs.service';
|
|
2
3
|
import { SubAgentRun } from './sub-agent-run.entity';
|
|
3
|
-
import { ChatService } from '../interactive-chat/chat.service';
|
|
4
4
|
export declare class SubAgentRunsController {
|
|
5
5
|
private readonly subAgentRunsService;
|
|
6
|
-
private readonly
|
|
6
|
+
private readonly moduleRef;
|
|
7
7
|
private readonly logger;
|
|
8
|
-
constructor(subAgentRunsService: SubAgentRunsService,
|
|
8
|
+
constructor(subAgentRunsService: SubAgentRunsService, moduleRef: ModuleRef);
|
|
9
|
+
private getChatService;
|
|
9
10
|
getAllActiveRuns(): Promise<SubAgentRun[]>;
|
|
10
11
|
getActiveRuns(sessionId: string): Promise<SubAgentRun[]>;
|
|
11
12
|
cancelRun(runId: string): Promise<{
|
|
@@ -16,15 +16,19 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
16
16
|
exports.SubAgentRunsController = void 0;
|
|
17
17
|
const common_1 = require("@nestjs/common");
|
|
18
18
|
const swagger_1 = require("@nestjs/swagger");
|
|
19
|
+
const core_1 = require("@nestjs/core");
|
|
19
20
|
const sub_agent_runs_service_1 = require("./sub-agent-runs.service");
|
|
20
21
|
const sub_agent_run_entity_1 = require("./sub-agent-run.entity");
|
|
21
22
|
const chat_service_1 = require("../interactive-chat/chat.service");
|
|
22
23
|
let SubAgentRunsController = SubAgentRunsController_1 = class SubAgentRunsController {
|
|
23
|
-
constructor(subAgentRunsService,
|
|
24
|
+
constructor(subAgentRunsService, moduleRef) {
|
|
24
25
|
this.subAgentRunsService = subAgentRunsService;
|
|
25
|
-
this.
|
|
26
|
+
this.moduleRef = moduleRef;
|
|
26
27
|
this.logger = new common_1.Logger(SubAgentRunsController_1.name);
|
|
27
28
|
}
|
|
29
|
+
getChatService() {
|
|
30
|
+
return this.moduleRef.get(chat_service_1.ChatService, { strict: false });
|
|
31
|
+
}
|
|
28
32
|
async getAllActiveRuns() {
|
|
29
33
|
return this.subAgentRunsService.getAllActiveRuns();
|
|
30
34
|
}
|
|
@@ -35,15 +39,16 @@ let SubAgentRunsController = SubAgentRunsController_1 = class SubAgentRunsContro
|
|
|
35
39
|
const run = await this.subAgentRunsService.cancel(runId);
|
|
36
40
|
if (run) {
|
|
37
41
|
this.logger.log(`Stopping LLM streaming for child session ${run.child_session_id}`);
|
|
38
|
-
this.
|
|
42
|
+
this.getChatService().cancelRequest(run.child_session_id);
|
|
39
43
|
}
|
|
40
44
|
return { success: run !== null, run };
|
|
41
45
|
}
|
|
42
46
|
async cancelAllRuns(sessionId) {
|
|
43
47
|
const activeRuns = await this.subAgentRunsService.getActiveRuns(sessionId);
|
|
48
|
+
const chatService = this.getChatService();
|
|
44
49
|
for (const run of activeRuns) {
|
|
45
50
|
this.logger.log(`Stopping LLM streaming for child session ${run.child_session_id}`);
|
|
46
|
-
|
|
51
|
+
chatService.cancelRequest(run.child_session_id);
|
|
47
52
|
}
|
|
48
53
|
const count = await this.subAgentRunsService.cancelAllActiveRuns(sessionId);
|
|
49
54
|
return { count };
|
|
@@ -143,6 +148,6 @@ exports.SubAgentRunsController = SubAgentRunsController = SubAgentRunsController
|
|
|
143
148
|
(0, swagger_1.ApiTags)('sub-agent-runs'),
|
|
144
149
|
(0, common_1.Controller)('sub-agent-runs'),
|
|
145
150
|
__metadata("design:paramtypes", [sub_agent_runs_service_1.SubAgentRunsService,
|
|
146
|
-
|
|
151
|
+
core_1.ModuleRef])
|
|
147
152
|
], SubAgentRunsController);
|
|
148
153
|
//# sourceMappingURL=sub-agent-runs.controller.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"sub-agent-runs.controller.js","sourceRoot":"","sources":["../../../src/sub-agents/sub-agent-runs.controller.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CASwB;AACxB,6CAA+E;AAC/E,qEAA+D;AAC/D,iEAAqD;AACrD,mEAA+D;AAkBxD,IAAM,sBAAsB,8BAA5B,MAAM,sBAAsB;IAGjC,YACmB,mBAAwC,EACxC,
|
|
1
|
+
{"version":3,"file":"sub-agent-runs.controller.js","sourceRoot":"","sources":["../../../src/sub-agents/sub-agent-runs.controller.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CASwB;AACxB,6CAA+E;AAC/E,uCAAyC;AACzC,qEAA+D;AAC/D,iEAAqD;AACrD,mEAA+D;AAkBxD,IAAM,sBAAsB,8BAA5B,MAAM,sBAAsB;IAGjC,YACmB,mBAAwC,EACxC,SAAoB;QADpB,wBAAmB,GAAnB,mBAAmB,CAAqB;QACxC,cAAS,GAAT,SAAS,CAAW;QAJtB,WAAM,GAAG,IAAI,eAAM,CAAC,wBAAsB,CAAC,IAAI,CAAC,CAAC;IAK/D,CAAC;IAMI,cAAc;QACpB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,0BAAW,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IAC5D,CAAC;IAaK,AAAN,KAAK,CAAC,gBAAgB;QACpB,OAAO,IAAI,CAAC,mBAAmB,CAAC,gBAAgB,EAAE,CAAC;IACrD,CAAC;IAkBK,AAAN,KAAK,CAAC,aAAa,CACkB,SAAiB;QAEpD,OAAO,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;IAC3D,CAAC;IAwBK,AAAN,KAAK,CAAC,SAAS,CACkB,KAAa;QAE5C,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAEzD,IAAI,GAAG,EAAE,CAAC;YAER,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,4CAA4C,GAAG,CAAC,gBAAgB,EAAE,CACnE,CAAC;YACF,IAAI,CAAC,cAAc,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC5D,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,GAAG,KAAK,IAAI,EAAE,GAAG,EAAE,CAAC;IACxC,CAAC;IAuBK,AAAN,KAAK,CAAC,aAAa,CACkB,SAAiB;QAGpD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QAE3E,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAG1C,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;YAC7B,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,4CAA4C,GAAG,CAAC,gBAAgB,EAAE,CACnE,CAAC;YACF,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAClD,CAAC;QAGD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAC5E,OAAO,EAAE,KAAK,EAAE,CAAC;IACnB,CAAC;CACF,CAAA;AApIY,wDAAsB;AA2B3B;IAXL,IAAA,YAAG,EAAC,QAAQ,CAAC;IACb,IAAA,sBAAY,EAAC;QACZ,OAAO,EAAE,+BAA+B;QACxC,WAAW,EACT,yEAAyE;KAC5E,CAAC;IACD,IAAA,qBAAW,EAAC;QACX,MAAM,EAAE,GAAG;QACX,WAAW,EAAE,qBAAqB;QAClC,IAAI,EAAE,CAAC,kCAAW,CAAC;KACpB,CAAC;;;;8DAGD;AAkBK;IAhBL,IAAA,YAAG,EAAC,mBAAmB,CAAC;IACxB,IAAA,sBAAY,EAAC;QACZ,OAAO,EAAE,+BAA+B;QACxC,WAAW,EACT,4EAA4E;KAC/E,CAAC;IACD,IAAA,kBAAQ,EAAC;QACR,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,qBAAqB;QAClC,MAAM,EAAE,MAAM;KACf,CAAC;IACD,IAAA,qBAAW,EAAC;QACX,MAAM,EAAE,GAAG;QACX,WAAW,EAAE,iCAAiC;QAC9C,IAAI,EAAE,CAAC,kCAAW,CAAC;KACpB,CAAC;IAEC,WAAA,IAAA,cAAK,EAAC,WAAW,EAAE,sBAAa,CAAC,CAAA;;;;2DAGnC;AAwBK;IAtBL,IAAA,aAAI,EAAC,eAAe,CAAC;IACrB,IAAA,iBAAQ,EAAC,mBAAU,CAAC,EAAE,CAAC;IACvB,IAAA,sBAAY,EAAC;QACZ,OAAO,EAAE,wBAAwB;QACjC,WAAW,EACT,+FAA+F;KAClG,CAAC;IACD,IAAA,kBAAQ,EAAC;QACR,IAAI,EAAE,OAAO;QACb,WAAW,EAAE,oBAAoB;QACjC,MAAM,EAAE,MAAM;KACf,CAAC;IACD,IAAA,qBAAW,EAAC;QACX,MAAM,EAAE,GAAG;QACX,WAAW,EAAE,eAAe;QAC5B,MAAM,EAAE;YACN,UAAU,EAAE;gBACV,OAAO,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;gBAC5B,GAAG,EAAE,EAAE,IAAI,EAAE,kCAAkC,EAAE,QAAQ,EAAE,IAAI,EAAE;aAClE;SACF;KACF,CAAC;IAEC,WAAA,IAAA,cAAK,EAAC,OAAO,EAAE,sBAAa,CAAC,CAAA;;;;uDAa/B;AAuBK;IArBL,IAAA,aAAI,EAAC,+BAA+B,CAAC;IACrC,IAAA,iBAAQ,EAAC,mBAAU,CAAC,EAAE,CAAC;IACvB,IAAA,sBAAY,EAAC;QACZ,OAAO,EAAE,+BAA+B;QACxC,WAAW,EACT,uGAAuG;KAC1G,CAAC;IACD,IAAA,kBAAQ,EAAC;QACR,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,qBAAqB;QAClC,MAAM,EAAE,MAAM;KACf,CAAC;IACD,IAAA,qBAAW,EAAC;QACX,MAAM,EAAE,GAAG;QACX,WAAW,EAAE,gBAAgB;QAC7B,MAAM,EAAE;YACN,UAAU,EAAE;gBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,0BAA0B,EAAE;aACnE;SACF;KACF,CAAC;IAEC,WAAA,IAAA,cAAK,EAAC,WAAW,EAAE,sBAAa,CAAC,CAAA;;;;2DAkBnC;iCAnIU,sBAAsB;IAFlC,IAAA,iBAAO,EAAC,gBAAgB,CAAC;IACzB,IAAA,mBAAU,EAAC,gBAAgB,CAAC;qCAKa,4CAAmB;QAC7B,gBAAS;GAL5B,sBAAsB,CAoIlC"}
|
|
@@ -11,6 +11,7 @@ export declare class SubAgent extends BaseEntity {
|
|
|
11
11
|
followup_context_template_id: string | null;
|
|
12
12
|
followupContextTemplate: ContextTemplate;
|
|
13
13
|
enabled_tools: string[] | null;
|
|
14
|
+
enabled_mcp_tools: string | null;
|
|
14
15
|
is_active: boolean;
|
|
15
16
|
model_id: string | null;
|
|
16
17
|
is_builtin: boolean;
|