@inkeep/agents-cli 0.59.3 → 0.60.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/agents-cli/package.js +1 -1
- package/dist/commands/pull-v4/collector-common.js +128 -0
- package/dist/commands/pull-v4/collector-common.js.map +1 -0
- package/dist/commands/pull-v4/collector-reference-helpers.js +323 -0
- package/dist/commands/pull-v4/collector-reference-helpers.js.map +1 -0
- package/dist/commands/pull-v4/component-parser.js +2 -1
- package/dist/commands/pull-v4/component-parser.js.map +1 -1
- package/dist/commands/pull-v4/file-scope.js +43 -0
- package/dist/commands/pull-v4/file-scope.js.map +1 -0
- package/dist/commands/pull-v4/generation-resolver.js +305 -0
- package/dist/commands/pull-v4/generation-resolver.js.map +1 -0
- package/dist/commands/pull-v4/generation-types.js +56 -0
- package/dist/commands/pull-v4/generation-types.js.map +1 -0
- package/dist/commands/pull-v4/generators/agent-generator.helpers.js +4 -10
- package/dist/commands/pull-v4/generators/agent-generator.helpers.js.map +1 -1
- package/dist/commands/pull-v4/generators/agent-generator.js +154 -81
- package/dist/commands/pull-v4/generators/agent-generator.js.map +1 -1
- package/dist/commands/pull-v4/generators/artifact-component-generator.js +48 -27
- package/dist/commands/pull-v4/generators/artifact-component-generator.js.map +1 -1
- package/dist/commands/pull-v4/generators/context-config-generator.js +147 -129
- package/dist/commands/pull-v4/generators/context-config-generator.js.map +1 -1
- package/dist/commands/pull-v4/generators/credential-generator.js +36 -14
- package/dist/commands/pull-v4/generators/credential-generator.js.map +1 -1
- package/dist/commands/pull-v4/generators/data-component-generator.js +55 -19
- package/dist/commands/pull-v4/generators/data-component-generator.js.map +1 -1
- package/dist/commands/pull-v4/generators/environment-generator.js +29 -33
- package/dist/commands/pull-v4/generators/environment-generator.js.map +1 -1
- package/dist/commands/pull-v4/generators/environment-settings-generator.js +57 -0
- package/dist/commands/pull-v4/generators/environment-settings-generator.js.map +1 -0
- package/dist/commands/pull-v4/generators/external-agent-generator.js +51 -23
- package/dist/commands/pull-v4/generators/external-agent-generator.js.map +1 -1
- package/dist/commands/pull-v4/generators/function-tool-generator.js +50 -14
- package/dist/commands/pull-v4/generators/function-tool-generator.js.map +1 -1
- package/dist/commands/pull-v4/generators/helpers/agent.js +129 -0
- package/dist/commands/pull-v4/generators/helpers/agent.js.map +1 -0
- package/dist/commands/pull-v4/generators/helpers/sub-agent.js +65 -0
- package/dist/commands/pull-v4/generators/helpers/sub-agent.js.map +1 -0
- package/dist/commands/pull-v4/generators/index.js +38 -0
- package/dist/commands/pull-v4/generators/index.js.map +1 -0
- package/dist/commands/pull-v4/generators/mcp-tool-generator.js +48 -27
- package/dist/commands/pull-v4/generators/mcp-tool-generator.js.map +1 -1
- package/dist/commands/pull-v4/generators/project-generator.js +144 -110
- package/dist/commands/pull-v4/generators/project-generator.js.map +1 -1
- package/dist/commands/pull-v4/generators/scheduled-trigger-generator.js +66 -0
- package/dist/commands/pull-v4/generators/scheduled-trigger-generator.js.map +1 -0
- package/dist/commands/pull-v4/generators/status-component-generator.js +55 -20
- package/dist/commands/pull-v4/generators/status-component-generator.js.map +1 -1
- package/dist/commands/pull-v4/generators/sub-agent-generator.helpers.js +4 -1
- package/dist/commands/pull-v4/generators/sub-agent-generator.helpers.js.map +1 -1
- package/dist/commands/pull-v4/generators/sub-agent-generator.js +282 -166
- package/dist/commands/pull-v4/generators/sub-agent-generator.js.map +1 -1
- package/dist/commands/pull-v4/generators/trigger-generator.js +62 -27
- package/dist/commands/pull-v4/generators/trigger-generator.js.map +1 -1
- package/dist/commands/pull-v4/import-plan.js +40 -0
- package/dist/commands/pull-v4/import-plan.js.map +1 -0
- package/dist/commands/pull-v4/introspect/index.js +11 -7
- package/dist/commands/pull-v4/introspect/index.js.map +1 -1
- package/dist/commands/pull-v4/introspect-generator.js +15 -1218
- package/dist/commands/pull-v4/introspect-generator.js.map +1 -1
- package/dist/commands/pull-v4/module-merge.js +14 -5
- package/dist/commands/pull-v4/module-merge.js.map +1 -1
- package/dist/commands/pull-v4/reference-resolution.js +111 -0
- package/dist/commands/pull-v4/reference-resolution.js.map +1 -0
- package/dist/commands/pull-v4/simple-factory-generator.js +54 -0
- package/dist/commands/pull-v4/simple-factory-generator.js.map +1 -0
- package/dist/commands/pull-v4/{generators/skill-generator.js → skill.js} +18 -6
- package/dist/commands/pull-v4/skill.js.map +1 -0
- package/dist/commands/pull-v4/typescript-file-writer.js +78 -0
- package/dist/commands/pull-v4/typescript-file-writer.js.map +1 -0
- package/dist/commands/pull-v4/utils/code-values.js +64 -0
- package/dist/commands/pull-v4/utils/code-values.js.map +1 -0
- package/dist/commands/pull-v4/utils/factory-writer.js +128 -0
- package/dist/commands/pull-v4/utils/factory-writer.js.map +1 -0
- package/dist/commands/pull-v4/utils/index.js +8 -0
- package/dist/commands/pull-v4/utils/naming.js +74 -0
- package/dist/commands/pull-v4/utils/naming.js.map +1 -0
- package/dist/commands/pull-v4/utils/schema-rendering.js +20 -0
- package/dist/commands/pull-v4/utils/schema-rendering.js.map +1 -0
- package/dist/commands/pull-v4/utils/shared.js +18 -0
- package/dist/commands/pull-v4/utils/shared.js.map +1 -0
- package/dist/commands/pull-v4/utils/templates.js +58 -0
- package/dist/commands/pull-v4/utils/templates.js.map +1 -0
- package/package.json +4 -4
- package/dist/commands/pull-v4/generators/skill-generator.js.map +0 -1
- package/dist/commands/pull-v4/scheduled-trigger-generator.js +0 -38
- package/dist/commands/pull-v4/scheduled-trigger-generator.js.map +0 -1
- package/dist/commands/pull-v4/utils.js +0 -320
- package/dist/commands/pull-v4/utils.js.map +0 -1
|
@@ -1,8 +1,17 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
1
|
+
import { hasReferences, isPlainObject } from "../utils/shared.js";
|
|
2
|
+
import { collectTemplateVariableNames, formatTemplate } from "../utils/templates.js";
|
|
3
|
+
import { codeReference, createArrayGetterValue, createReferenceGetterValue } from "../utils/code-values.js";
|
|
4
|
+
import { buildComponentFileName, resolveReferenceName, toCamelCase } from "../utils/naming.js";
|
|
5
|
+
import { addValueToObject } from "../utils/factory-writer.js";
|
|
6
|
+
import "../utils/index.js";
|
|
7
|
+
import { asRecord } from "../collector-common.js";
|
|
8
|
+
import { collectCanDelegateToReferences, collectCanUseReferences, collectSkills, resolveSubAgentName, resolveSubAgentVariableName } from "./helpers/sub-agent.js";
|
|
9
|
+
import { collectContextTemplateReferences, collectSubAgentDependencyReferenceOverrides } from "../collector-reference-helpers.js";
|
|
10
|
+
import { addResolvedReferenceImports, resolveReferenceBinding, resolveReferenceBindingsFromIds, toReferenceNameRecord } from "../reference-resolution.js";
|
|
11
|
+
import { generateFactorySourceFile } from "../simple-factory-generator.js";
|
|
3
12
|
import { FullAgentAgentInsertSchema } from "@inkeep/agents-core";
|
|
4
13
|
import { z } from "zod";
|
|
5
|
-
import {
|
|
14
|
+
import { join } from "node:path";
|
|
6
15
|
|
|
7
16
|
//#region src/commands/pull-v4/generators/sub-agent-generator.ts
|
|
8
17
|
const ReferenceNameByIdSchema = z.record(z.string(), z.string().nonempty());
|
|
@@ -48,145 +57,226 @@ const SubAgentSchema = z.strictObject({
|
|
|
48
57
|
contextConfigId: z.string().nonempty().optional(),
|
|
49
58
|
contextConfigReference: ContextTemplateReferenceSchema.optional(),
|
|
50
59
|
contextConfigHeadersReference: ContextTemplateReferenceSchema.optional()
|
|
51
|
-
})
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
const hasHeadersTemplateVariables = promptTemplateVariables.some((variableName) => variableName.startsWith("headers."));
|
|
65
|
-
const contextImportResolution = resolveContextTemplateImports({
|
|
66
|
-
reservedNames: reservedReferenceNames,
|
|
67
|
-
shouldResolveContextReference: hasContextTemplateVariables && Boolean(parsed.contextConfigId),
|
|
68
|
-
shouldResolveHeadersReference: hasHeadersTemplateVariables && Boolean(parsed.contextConfigId),
|
|
69
|
-
contextConfigReference: parsed.contextConfigReference,
|
|
70
|
-
contextConfigHeadersReference: parsed.contextConfigHeadersReference
|
|
71
|
-
});
|
|
72
|
-
if (contextImportResolution.namedImports.length > 0 && parsed.contextConfigId) sourceFile.addImportDeclaration({
|
|
73
|
-
namedImports: contextImportResolution.namedImports,
|
|
74
|
-
moduleSpecifier: `../../context-configs/${parsed.contextConfigId}`
|
|
75
|
-
});
|
|
76
|
-
const canUseToolReferenceOverrides = addCanUseToolImports(sourceFile, parsed.canUse, parsed.referenceOverrides?.tools, parsed.referencePathOverrides?.tools, reservedReferenceNames);
|
|
77
|
-
addDataComponentImports(sourceFile, parsed.dataComponents, parsed.referenceOverrides?.dataComponents);
|
|
78
|
-
addArtifactComponentImports(sourceFile, parsed.artifactComponents, parsed.referenceOverrides?.artifactComponents);
|
|
79
|
-
addDelegateTargetImports(sourceFile, {
|
|
80
|
-
currentSubAgentId: parsed.id,
|
|
81
|
-
canDelegateTo: parsed.canDelegateTo,
|
|
82
|
-
canTransferTo: parsed.canTransferTo,
|
|
83
|
-
referenceOverrides: {
|
|
84
|
-
subAgents: parsed.referenceOverrides?.subAgents,
|
|
85
|
-
agents: parsed.referenceOverrides?.agents,
|
|
86
|
-
externalAgents: parsed.referenceOverrides?.externalAgents
|
|
87
|
-
},
|
|
88
|
-
referencePathOverrides: {
|
|
89
|
-
subAgents: parsed.referencePathOverrides?.subAgents,
|
|
90
|
-
agents: parsed.referencePathOverrides?.agents,
|
|
91
|
-
externalAgents: parsed.referencePathOverrides?.externalAgents
|
|
60
|
+
}).transform((data) => ({
|
|
61
|
+
...data,
|
|
62
|
+
normalizedCanUse: normalizeCanUseEntries(data.canUse),
|
|
63
|
+
normalizedCanDelegateTo: normalizeCanDelegateToEntries(data.canDelegateTo),
|
|
64
|
+
normalizedSkills: normalizeSkills(data.skills)
|
|
65
|
+
}));
|
|
66
|
+
function normalizeCanUseEntries(canUse) {
|
|
67
|
+
if (!Array.isArray(canUse)) return [];
|
|
68
|
+
const entries = [];
|
|
69
|
+
for (const item of canUse) {
|
|
70
|
+
if (typeof item === "string") {
|
|
71
|
+
entries.push({ toolId: item });
|
|
72
|
+
continue;
|
|
92
73
|
}
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
}
|
|
100
|
-
|
|
101
|
-
|
|
102
|
-
for (const item of canUse ?? []) {
|
|
103
|
-
const toolId = typeof item === "string" ? item : isPlainObject(item) && typeof item.toolId === "string" ? item.toolId : void 0;
|
|
104
|
-
if (!toolId || toolImportsById.has(toolId)) continue;
|
|
105
|
-
toolImportsById.set(toolId, {
|
|
106
|
-
importName: resolveReferenceName(toolId, [toolReferenceOverrides]),
|
|
107
|
-
modulePath: resolveReferenceName(toolId, [toolReferencePathOverrides])
|
|
74
|
+
if (!isPlainObject(item) || typeof item.toolId !== "string") continue;
|
|
75
|
+
const selectedTools = Array.isArray(item.toolSelection) && item.toolSelection.length ? item.toolSelection : Array.isArray(item.selectedTools) && item.selectedTools.length ? item.selectedTools : void 0;
|
|
76
|
+
const headers = isPlainObject(item.headers) && Object.keys(item.headers).length > 0 ? item.headers : void 0;
|
|
77
|
+
const toolPolicies = isPlainObject(item.toolPolicies) && Object.keys(item.toolPolicies).length > 0 ? item.toolPolicies : void 0;
|
|
78
|
+
entries.push({
|
|
79
|
+
toolId: item.toolId,
|
|
80
|
+
...selectedTools && { selectedTools },
|
|
81
|
+
...headers && { headers },
|
|
82
|
+
...toolPolicies && { toolPolicies }
|
|
108
83
|
});
|
|
109
84
|
}
|
|
110
|
-
|
|
111
|
-
const localNames = reservedReferenceNames ?? /* @__PURE__ */ new Set();
|
|
112
|
-
for (const [toolId, { importName, modulePath }] of toolImportsById) {
|
|
113
|
-
const localName = resolveNonCollidingName(importName, localNames);
|
|
114
|
-
localReferenceNamesByToolId[toolId] = localName;
|
|
115
|
-
sourceFile.addImportDeclaration({
|
|
116
|
-
namedImports: [importName === localName ? importName : {
|
|
117
|
-
name: importName,
|
|
118
|
-
alias: localName
|
|
119
|
-
}],
|
|
120
|
-
moduleSpecifier: `../../tools/${modulePath}`
|
|
121
|
-
});
|
|
122
|
-
}
|
|
123
|
-
return localReferenceNamesByToolId;
|
|
124
|
-
}
|
|
125
|
-
function addDataComponentImports(sourceFile, dataComponents, dataComponentReferenceOverrides) {
|
|
126
|
-
addReferenceImports(sourceFile, dataComponents, "../../data-components", dataComponentReferenceOverrides);
|
|
85
|
+
return entries;
|
|
127
86
|
}
|
|
128
|
-
function
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
87
|
+
function normalizeCanDelegateToEntries(canDelegateTo) {
|
|
88
|
+
if (!Array.isArray(canDelegateTo)) return [];
|
|
89
|
+
const entries = [];
|
|
90
|
+
for (const item of canDelegateTo) {
|
|
91
|
+
if (typeof item === "string") {
|
|
92
|
+
entries.push({ id: item });
|
|
93
|
+
continue;
|
|
94
|
+
}
|
|
95
|
+
if (!isPlainObject(item)) continue;
|
|
96
|
+
const headers = isPlainObject(item.headers) && Object.keys(item.headers).length > 0 ? item.headers : void 0;
|
|
97
|
+
const addEntry = (type, id) => {
|
|
98
|
+
if (typeof id !== "string") return;
|
|
99
|
+
entries.push({
|
|
100
|
+
id,
|
|
101
|
+
type,
|
|
102
|
+
...headers && { headers }
|
|
103
|
+
});
|
|
104
|
+
};
|
|
105
|
+
addEntry("subAgents", item.subAgentId);
|
|
106
|
+
addEntry("agents", item.agentId);
|
|
107
|
+
addEntry("externalAgents", item.externalAgentId);
|
|
136
108
|
}
|
|
137
|
-
|
|
138
|
-
namedImports: [referenceName],
|
|
139
|
-
moduleSpecifier: `${basePath}/${referenceId}`
|
|
140
|
-
});
|
|
109
|
+
return entries;
|
|
141
110
|
}
|
|
142
|
-
function
|
|
143
|
-
|
|
144
|
-
|
|
145
|
-
|
|
146
|
-
if (
|
|
147
|
-
|
|
148
|
-
|
|
149
|
-
|
|
150
|
-
|
|
151
|
-
const
|
|
152
|
-
if (!
|
|
153
|
-
|
|
154
|
-
|
|
111
|
+
function normalizeSkills(skills) {
|
|
112
|
+
if (!Array.isArray(skills)) return [];
|
|
113
|
+
const normalizedSkills = [];
|
|
114
|
+
for (const skill of skills) {
|
|
115
|
+
if (typeof skill === "string") {
|
|
116
|
+
normalizedSkills.push(skill);
|
|
117
|
+
continue;
|
|
118
|
+
}
|
|
119
|
+
if (!isPlainObject(skill)) continue;
|
|
120
|
+
const skillId = typeof skill.id === "string" ? skill.id : typeof skill.skillId === "string" ? skill.skillId : void 0;
|
|
121
|
+
if (!skillId) continue;
|
|
122
|
+
normalizedSkills.push({
|
|
123
|
+
id: skillId,
|
|
124
|
+
...typeof skill.index === "number" && Number.isInteger(skill.index) ? { index: skill.index } : {},
|
|
125
|
+
...typeof skill.alwaysLoaded === "boolean" ? { alwaysLoaded: skill.alwaysLoaded } : {}
|
|
126
|
+
});
|
|
155
127
|
}
|
|
156
|
-
|
|
157
|
-
|
|
158
|
-
|
|
128
|
+
return normalizedSkills;
|
|
129
|
+
}
|
|
130
|
+
function generateSubAgentDefinition({ subAgentId, ...data }) {
|
|
131
|
+
return generateFactorySourceFile(data, {
|
|
132
|
+
schema: SubAgentSchema,
|
|
133
|
+
factory: {
|
|
134
|
+
importName: "subAgent",
|
|
135
|
+
variableName: (parsed) => resolveSubAgentVariableName(parsed.id, parsed.name)
|
|
136
|
+
},
|
|
137
|
+
render({ parsed, sourceFile, configObject }) {
|
|
138
|
+
const subAgentVariableName = resolveSubAgentVariableName(parsed.id, parsed.name);
|
|
139
|
+
const reservedReferenceNames = new Set([subAgentVariableName]);
|
|
140
|
+
const promptTemplateVariables = typeof parsed.prompt === "string" ? collectTemplateVariableNames(parsed.prompt) : [];
|
|
141
|
+
const hasContextTemplateVariables = promptTemplateVariables.some((variableName) => !variableName.startsWith("headers."));
|
|
142
|
+
const hasHeadersTemplateVariables = promptTemplateVariables.some((variableName) => variableName.startsWith("headers."));
|
|
143
|
+
let contextReferenceName;
|
|
144
|
+
let headersReferenceName;
|
|
145
|
+
if (parsed.contextConfigId) {
|
|
146
|
+
const contextReferences = [];
|
|
147
|
+
if (hasContextTemplateVariables) {
|
|
148
|
+
const contextReference = resolveReferenceBinding({
|
|
149
|
+
id: `${parsed.contextConfigId}:context`,
|
|
150
|
+
importName: parsed.contextConfigReference?.name ?? toCamelCase(parsed.contextConfigId),
|
|
151
|
+
modulePath: parsed.contextConfigId,
|
|
152
|
+
local: parsed.contextConfigReference?.local === true,
|
|
153
|
+
conflictSuffix: "ContextConfig"
|
|
154
|
+
}, { reservedNames: reservedReferenceNames });
|
|
155
|
+
contextReferenceName = contextReference.localName;
|
|
156
|
+
contextReferences.push(contextReference);
|
|
157
|
+
}
|
|
158
|
+
if (hasHeadersTemplateVariables) {
|
|
159
|
+
const headersReference = resolveReferenceBinding({
|
|
160
|
+
id: `${parsed.contextConfigId}:headers`,
|
|
161
|
+
importName: parsed.contextConfigHeadersReference?.name ?? `${toCamelCase(parsed.contextConfigId)}Headers`,
|
|
162
|
+
modulePath: parsed.contextConfigId,
|
|
163
|
+
local: parsed.contextConfigHeadersReference?.local === true,
|
|
164
|
+
conflictSuffix: "Headers"
|
|
165
|
+
}, { reservedNames: reservedReferenceNames });
|
|
166
|
+
headersReferenceName = headersReference.localName;
|
|
167
|
+
contextReferences.push(headersReference);
|
|
168
|
+
}
|
|
169
|
+
addResolvedReferenceImports(sourceFile, contextReferences, () => {
|
|
170
|
+
return `../../context-configs/${parsed.contextConfigId}`;
|
|
171
|
+
});
|
|
172
|
+
}
|
|
173
|
+
const canUseToolReferences = resolveReferenceBindingsFromIds({
|
|
174
|
+
ids: collectCanUseToolIds(parsed.normalizedCanUse),
|
|
175
|
+
reservedNames: reservedReferenceNames,
|
|
176
|
+
conflictSuffix: "Tool",
|
|
177
|
+
collisionStrategy: "numeric",
|
|
178
|
+
referenceOverrides: parsed.referenceOverrides?.tools,
|
|
179
|
+
referencePathOverrides: parsed.referencePathOverrides?.tools,
|
|
180
|
+
defaultModulePath: toCamelCase
|
|
181
|
+
});
|
|
182
|
+
addResolvedReferenceImports(sourceFile, canUseToolReferences, (reference) => {
|
|
183
|
+
return `../../tools/${reference.modulePath}`;
|
|
184
|
+
});
|
|
185
|
+
const dataComponentReferences = resolveReferenceBindingsFromIds({
|
|
186
|
+
ids: parsed.dataComponents ?? [],
|
|
187
|
+
reservedNames: reservedReferenceNames,
|
|
188
|
+
conflictSuffix: "DataComponent",
|
|
189
|
+
referenceOverrides: parsed.referenceOverrides?.dataComponents,
|
|
190
|
+
defaultModulePath: (id) => id
|
|
191
|
+
});
|
|
192
|
+
addResolvedReferenceImports(sourceFile, dataComponentReferences, (reference) => {
|
|
193
|
+
return `../../data-components/${reference.modulePath}`;
|
|
194
|
+
});
|
|
195
|
+
const artifactComponentReferences = resolveReferenceBindingsFromIds({
|
|
196
|
+
ids: parsed.artifactComponents ?? [],
|
|
197
|
+
reservedNames: reservedReferenceNames,
|
|
198
|
+
conflictSuffix: "ArtifactComponent",
|
|
199
|
+
referenceOverrides: parsed.referenceOverrides?.artifactComponents,
|
|
200
|
+
defaultModulePath: (id) => id
|
|
201
|
+
});
|
|
202
|
+
addResolvedReferenceImports(sourceFile, artifactComponentReferences, (reference) => {
|
|
203
|
+
return `../../artifact-components/${reference.modulePath}`;
|
|
204
|
+
});
|
|
205
|
+
const delegateTargetIds = collectDelegateTargetIds(parsed.normalizedCanDelegateTo, parsed.canTransferTo, parsed.id, {
|
|
206
|
+
subAgents: parsed.referenceOverrides?.subAgents,
|
|
207
|
+
agents: parsed.referenceOverrides?.agents,
|
|
208
|
+
externalAgents: parsed.referenceOverrides?.externalAgents
|
|
209
|
+
});
|
|
210
|
+
const subAgentDelegateReferences = resolveReferenceBindingsFromIds({
|
|
211
|
+
ids: delegateTargetIds.subAgents,
|
|
212
|
+
reservedNames: reservedReferenceNames,
|
|
213
|
+
conflictSuffix: "SubAgent",
|
|
214
|
+
referenceOverrides: parsed.referenceOverrides?.subAgents,
|
|
215
|
+
referencePathOverrides: parsed.referencePathOverrides?.subAgents,
|
|
216
|
+
defaultModulePath: (id) => id
|
|
217
|
+
});
|
|
218
|
+
addResolvedReferenceImports(sourceFile, subAgentDelegateReferences, (reference) => {
|
|
219
|
+
return `./${reference.modulePath}`;
|
|
220
|
+
});
|
|
221
|
+
const agentDelegateReferences = resolveReferenceBindingsFromIds({
|
|
222
|
+
ids: delegateTargetIds.agents,
|
|
223
|
+
reservedNames: reservedReferenceNames,
|
|
224
|
+
conflictSuffix: "Agent",
|
|
225
|
+
referenceOverrides: parsed.referenceOverrides?.agents,
|
|
226
|
+
referencePathOverrides: parsed.referencePathOverrides?.agents,
|
|
227
|
+
defaultModulePath: (id) => id
|
|
228
|
+
});
|
|
229
|
+
addResolvedReferenceImports(sourceFile, agentDelegateReferences, (reference) => {
|
|
230
|
+
return `../${reference.modulePath}`;
|
|
231
|
+
});
|
|
232
|
+
const externalAgentDelegateReferences = resolveReferenceBindingsFromIds({
|
|
233
|
+
ids: delegateTargetIds.externalAgents,
|
|
234
|
+
reservedNames: reservedReferenceNames,
|
|
235
|
+
conflictSuffix: "ExternalAgent",
|
|
236
|
+
referenceOverrides: parsed.referenceOverrides?.externalAgents,
|
|
237
|
+
referencePathOverrides: parsed.referencePathOverrides?.externalAgents,
|
|
238
|
+
defaultModulePath: (id) => id
|
|
239
|
+
});
|
|
240
|
+
addResolvedReferenceImports(sourceFile, externalAgentDelegateReferences, (reference) => {
|
|
241
|
+
return `../../external-agents/${reference.modulePath}`;
|
|
242
|
+
});
|
|
243
|
+
writeSubAgentConfig(configObject, {
|
|
244
|
+
contextReference: contextReferenceName,
|
|
245
|
+
headersReference: headersReferenceName
|
|
246
|
+
}, {
|
|
247
|
+
tools: toReferenceNameRecord(canUseToolReferences),
|
|
248
|
+
subAgents: toReferenceNameRecord(subAgentDelegateReferences),
|
|
249
|
+
agents: toReferenceNameRecord(agentDelegateReferences),
|
|
250
|
+
externalAgents: toReferenceNameRecord(externalAgentDelegateReferences),
|
|
251
|
+
dataComponents: toReferenceNameRecord(dataComponentReferences),
|
|
252
|
+
artifactComponents: toReferenceNameRecord(artifactComponentReferences)
|
|
253
|
+
}, parsed);
|
|
254
|
+
}
|
|
159
255
|
});
|
|
160
256
|
}
|
|
161
|
-
function
|
|
162
|
-
if (
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
|
|
168
|
-
|
|
169
|
-
|
|
170
|
-
};
|
|
171
|
-
}
|
|
172
|
-
if (!isPlainObject(canDelegateToEntry)) return;
|
|
173
|
-
if (typeof canDelegateToEntry.subAgentId === "string") return {
|
|
174
|
-
type: "subAgents",
|
|
175
|
-
id: canDelegateToEntry.subAgentId,
|
|
176
|
-
name: resolveReferenceName(canDelegateToEntry.subAgentId, [referenceOverrides.subAgents]),
|
|
177
|
-
modulePath: referencePathOverrides?.subAgents?.[canDelegateToEntry.subAgentId]
|
|
178
|
-
};
|
|
179
|
-
if (typeof canDelegateToEntry.agentId === "string") return {
|
|
180
|
-
type: "agents",
|
|
181
|
-
id: canDelegateToEntry.agentId,
|
|
182
|
-
name: resolveReferenceName(canDelegateToEntry.agentId, [referenceOverrides.agents]),
|
|
183
|
-
modulePath: referencePathOverrides?.agents?.[canDelegateToEntry.agentId]
|
|
257
|
+
function collectCanUseToolIds(canUse) {
|
|
258
|
+
if (!canUse?.length) return [];
|
|
259
|
+
return [...new Set(canUse.map((item) => item.toolId))];
|
|
260
|
+
}
|
|
261
|
+
function collectDelegateTargetIds(canDelegateTo, canTransferTo, currentSubAgentId, referenceOverrides) {
|
|
262
|
+
const idsByType = {
|
|
263
|
+
subAgents: /* @__PURE__ */ new Set(),
|
|
264
|
+
agents: /* @__PURE__ */ new Set(),
|
|
265
|
+
externalAgents: /* @__PURE__ */ new Set()
|
|
184
266
|
};
|
|
185
|
-
|
|
186
|
-
type
|
|
187
|
-
id
|
|
188
|
-
|
|
189
|
-
|
|
267
|
+
for (const item of canDelegateTo ?? []) {
|
|
268
|
+
const type = item.type ?? resolveDelegateTargetType(item.id, referenceOverrides);
|
|
269
|
+
if (!(type === "subAgents" && item.id === currentSubAgentId)) idsByType[type].add(item.id);
|
|
270
|
+
}
|
|
271
|
+
for (const targetId of canTransferTo ?? []) {
|
|
272
|
+
const type = resolveDelegateTargetType(targetId, referenceOverrides);
|
|
273
|
+
if (type === "subAgents" && targetId === currentSubAgentId) continue;
|
|
274
|
+
idsByType[type].add(targetId);
|
|
275
|
+
}
|
|
276
|
+
return {
|
|
277
|
+
subAgents: [...idsByType.subAgents],
|
|
278
|
+
agents: [...idsByType.agents],
|
|
279
|
+
externalAgents: [...idsByType.externalAgents]
|
|
190
280
|
};
|
|
191
281
|
}
|
|
192
282
|
function resolveDelegateTargetType(targetId, referenceOverrides) {
|
|
@@ -195,44 +285,70 @@ function resolveDelegateTargetType(targetId, referenceOverrides) {
|
|
|
195
285
|
if (referenceOverrides.externalAgents?.[targetId]) return "externalAgents";
|
|
196
286
|
return "subAgents";
|
|
197
287
|
}
|
|
198
|
-
function
|
|
199
|
-
const path = modulePath ?? id;
|
|
200
|
-
switch (type) {
|
|
201
|
-
case "subAgents": return `./${path}`;
|
|
202
|
-
case "agents": return `../${path}`;
|
|
203
|
-
case "externalAgents": return `../../external-agents/${path}`;
|
|
204
|
-
}
|
|
205
|
-
}
|
|
206
|
-
function writeSubAgentConfig(configObject, templateReferences, canUseToolReferenceOverrides, { dataComponents, name, canDelegateTo, canTransferTo, skills, artifactComponents, canUse, referenceOverrides, referencePathOverrides: _referencePathOverrides, contextConfigId: _contextConfigId, contextConfigReference: _contextConfigReference, contextConfigHeadersReference: _contextConfigHeadersReference, ...rest }) {
|
|
288
|
+
function writeSubAgentConfig(configObject, templateReferences, referenceNames, { dataComponents, name, canDelegateTo: _canDelegateTo, canTransferTo, skills: _skills, artifactComponents, canUse: _canUse, normalizedCanDelegateTo, normalizedCanUse, normalizedSkills, referenceOverrides: _referenceOverrides, referencePathOverrides: _referencePathOverrides, contextConfigId: _contextConfigId, contextConfigReference: _contextConfigReference, contextConfigHeadersReference: _contextConfigHeadersReference, ...rest }) {
|
|
207
289
|
rest = { ...rest };
|
|
208
290
|
rest.prompt &&= formatTemplate(rest.prompt, templateReferences);
|
|
209
|
-
|
|
210
|
-
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
const
|
|
214
|
-
|
|
215
|
-
|
|
216
|
-
|
|
291
|
+
const subAgentConfig = {
|
|
292
|
+
...rest,
|
|
293
|
+
name: resolveSubAgentName(rest.id, name)
|
|
294
|
+
};
|
|
295
|
+
const canUseReferences = collectCanUseReferences(normalizedCanUse, Object.keys(referenceNames.tools ?? {}).length ? referenceNames.tools : void 0);
|
|
296
|
+
if (canUseReferences.length) subAgentConfig.canUse = createReferenceGetterValue(canUseReferences);
|
|
297
|
+
const canDelegateToReferences = collectCanDelegateToReferences(normalizedCanDelegateTo, {
|
|
298
|
+
subAgents: referenceNames.subAgents,
|
|
299
|
+
agents: referenceNames.agents,
|
|
300
|
+
externalAgents: referenceNames.externalAgents
|
|
217
301
|
});
|
|
218
|
-
if (canDelegateToReferences.length)
|
|
219
|
-
if (hasReferences(canTransferTo))
|
|
220
|
-
|
|
221
|
-
|
|
222
|
-
|
|
223
|
-
])));
|
|
224
|
-
if (hasReferences(dataComponents))
|
|
225
|
-
if (hasReferences(artifactComponents))
|
|
226
|
-
const collectedSkills = collectSkills(
|
|
227
|
-
if (collectedSkills.length > 0)
|
|
228
|
-
|
|
229
|
-
name: "skills",
|
|
230
|
-
initializer: "() => []"
|
|
231
|
-
}).getInitializerIfKindOrThrow(SyntaxKind.ArrowFunction).getBody().asKindOrThrow(SyntaxKind.ArrayLiteralExpression);
|
|
232
|
-
for (const skill of collectedSkills) skillsArray.addElement(skill);
|
|
233
|
-
}
|
|
302
|
+
if (canDelegateToReferences.length) subAgentConfig.canDelegateTo = createReferenceGetterValue(canDelegateToReferences);
|
|
303
|
+
if (hasReferences(canTransferTo)) subAgentConfig.canTransferTo = createReferenceGetterValue(canTransferTo.map((id) => codeReference(resolveReferenceName(id, [
|
|
304
|
+
referenceNames.subAgents,
|
|
305
|
+
referenceNames.agents,
|
|
306
|
+
referenceNames.externalAgents
|
|
307
|
+
]))));
|
|
308
|
+
if (hasReferences(dataComponents)) subAgentConfig.dataComponents = createReferenceGetterValue(dataComponents.map((id) => codeReference(resolveReferenceName(id, [referenceNames.dataComponents]))));
|
|
309
|
+
if (hasReferences(artifactComponents)) subAgentConfig.artifactComponents = createReferenceGetterValue(artifactComponents.map((id) => codeReference(resolveReferenceName(id, [referenceNames.artifactComponents]))));
|
|
310
|
+
const collectedSkills = collectSkills(normalizedSkills);
|
|
311
|
+
if (collectedSkills.length > 0) subAgentConfig.skills = createArrayGetterValue(collectedSkills);
|
|
312
|
+
for (const [key, value] of Object.entries(subAgentConfig)) addValueToObject(configObject, key, value);
|
|
234
313
|
}
|
|
314
|
+
const task = {
|
|
315
|
+
type: "sub-agent",
|
|
316
|
+
collect(context) {
|
|
317
|
+
if (!context.project.agents) return [];
|
|
318
|
+
const recordsBySubAgentId = /* @__PURE__ */ new Map();
|
|
319
|
+
for (const agentId of context.completeAgentIds) {
|
|
320
|
+
const agentData = context.project.agents[agentId];
|
|
321
|
+
const subAgents = asRecord(agentData?.subAgents);
|
|
322
|
+
if (!subAgents) continue;
|
|
323
|
+
for (const [subAgentId, subAgentData] of Object.entries(subAgents)) {
|
|
324
|
+
const payload = asRecord(subAgentData);
|
|
325
|
+
if (!payload) continue;
|
|
326
|
+
const dependencyReferences = collectSubAgentDependencyReferenceOverrides(context, payload);
|
|
327
|
+
const subAgentName = typeof payload.name === "string" ? payload.name : void 0;
|
|
328
|
+
const subAgentFilePath = context.resolver.resolveOutputFilePath("subAgents", subAgentId, join(context.paths.agentsDir, "sub-agents", buildComponentFileName(subAgentId, subAgentName)));
|
|
329
|
+
const contextTemplateReferences = collectContextTemplateReferences(context, agentData, subAgentFilePath, [typeof payload.prompt === "string" ? payload.prompt : void 0]);
|
|
330
|
+
recordsBySubAgentId.set(subAgentId, {
|
|
331
|
+
id: subAgentId,
|
|
332
|
+
filePath: subAgentFilePath,
|
|
333
|
+
payload: {
|
|
334
|
+
subAgentId,
|
|
335
|
+
...payload,
|
|
336
|
+
...dependencyReferences?.referenceOverrides && { referenceOverrides: dependencyReferences.referenceOverrides },
|
|
337
|
+
...dependencyReferences?.referencePathOverrides && { referencePathOverrides: dependencyReferences.referencePathOverrides },
|
|
338
|
+
...contextTemplateReferences && {
|
|
339
|
+
contextConfigId: contextTemplateReferences.contextConfigId,
|
|
340
|
+
contextConfigReference: contextTemplateReferences.contextConfigReference
|
|
341
|
+
},
|
|
342
|
+
...contextTemplateReferences?.contextConfigHeadersReference && { contextConfigHeadersReference: contextTemplateReferences.contextConfigHeadersReference }
|
|
343
|
+
}
|
|
344
|
+
});
|
|
345
|
+
}
|
|
346
|
+
}
|
|
347
|
+
return [...recordsBySubAgentId.values()];
|
|
348
|
+
},
|
|
349
|
+
generate: generateSubAgentDefinition
|
|
350
|
+
};
|
|
235
351
|
|
|
236
352
|
//#endregion
|
|
237
|
-
export { generateSubAgentDefinition };
|
|
353
|
+
export { generateSubAgentDefinition, task };
|
|
238
354
|
//# sourceMappingURL=sub-agent-generator.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"sub-agent-generator.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/sub-agent-generator.ts"],"sourcesContent":["import { FullAgentAgentInsertSchema } from '@inkeep/agents-core';\nimport { type ObjectLiteralExpression, type SourceFile, SyntaxKind } from 'ts-morph';\nimport { z } from 'zod';\nimport {\n addReferenceGetterProperty,\n addStringProperty,\n addValueToObject,\n collectTemplateVariableNames,\n createFactoryDefinition,\n formatTemplate,\n hasReferences,\n isPlainObject,\n resolveContextTemplateImports,\n resolveNonCollidingName,\n resolveReferenceName,\n} from '../utils';\nimport {\n collectCanDelegateToReferences,\n collectCanUseReferences,\n collectSkills,\n resolveSubAgentName,\n resolveSubAgentVariableName,\n} from './sub-agent-generator.helpers';\n\nconst ReferenceNameByIdSchema = z.record(z.string(), z.string().nonempty());\n\nconst ReferenceOverridesSchema = z.object({\n tools: ReferenceNameByIdSchema.optional(),\n subAgents: ReferenceNameByIdSchema.optional(),\n agents: ReferenceNameByIdSchema.optional(),\n externalAgents: ReferenceNameByIdSchema.optional(),\n dataComponents: ReferenceNameByIdSchema.optional(),\n artifactComponents: ReferenceNameByIdSchema.optional(),\n});\n\nconst ReferencePathOverridesSchema = z.object({\n tools: ReferenceNameByIdSchema.optional(),\n subAgents: ReferenceNameByIdSchema.optional(),\n agents: ReferenceNameByIdSchema.optional(),\n externalAgents: ReferenceNameByIdSchema.optional(),\n});\n\nconst ContextTemplateReferenceSchema = z.object({\n name: z.string().nonempty(),\n local: z.boolean().optional(),\n});\n\nconst MySchema = FullAgentAgentInsertSchema.pick({\n id: true,\n prompt: true,\n name: true,\n description: true,\n stopWhen: true,\n});\n\nconst SubAgentSchema = z.strictObject({\n ...MySchema.shape,\n prompt: z.preprocess((v) => v || undefined, MySchema.shape.prompt),\n description: z.preprocess((v) => v || undefined, MySchema.shape.description),\n stopWhen: z.preprocess((v) => v ?? undefined, MySchema.shape.stopWhen),\n models: z.preprocess((v) => v ?? undefined, z.looseObject({}).optional()),\n skills: z.array(z.unknown()).optional(),\n canUse: z.array(z.unknown()).optional(),\n canDelegateTo: z.array(z.unknown()).optional(),\n canTransferTo: z.array(z.string()).optional(),\n dataComponents: z.array(z.string()).optional(),\n artifactComponents: z.array(z.string()).optional(),\n referenceOverrides: ReferenceOverridesSchema.optional(),\n referencePathOverrides: ReferencePathOverridesSchema.optional(),\n contextConfigId: z.string().nonempty().optional(),\n contextConfigReference: ContextTemplateReferenceSchema.optional(),\n contextConfigHeadersReference: ContextTemplateReferenceSchema.optional(),\n});\n\ntype SubAgentInput = z.input<typeof SubAgentSchema>;\ntype SubAgentOutput = z.output<typeof SubAgentSchema>;\n\nexport function generateSubAgentDefinition({\n subAgentId,\n ...data\n}: SubAgentInput & Record<string, unknown>): SourceFile {\n const result = SubAgentSchema.safeParse(data);\n if (!result.success) {\n throw new Error(`Validation failed for sub-agent:\\n${z.prettifyError(result.error)}`);\n }\n\n const parsed = result.data;\n const subAgentVariableName = resolveSubAgentVariableName(parsed.id, parsed.name);\n const { sourceFile, configObject } = createFactoryDefinition({\n importName: 'subAgent',\n variableName: subAgentVariableName,\n });\n const reservedReferenceNames = new Set([subAgentVariableName]);\n\n const promptTemplateVariables =\n typeof parsed.prompt === 'string' ? collectTemplateVariableNames(parsed.prompt) : [];\n const hasContextTemplateVariables = promptTemplateVariables.some(\n (variableName) => !variableName.startsWith('headers.')\n );\n const hasHeadersTemplateVariables = promptTemplateVariables.some((variableName) =>\n variableName.startsWith('headers.')\n );\n const contextImportResolution = resolveContextTemplateImports({\n reservedNames: reservedReferenceNames,\n shouldResolveContextReference: hasContextTemplateVariables && Boolean(parsed.contextConfigId),\n shouldResolveHeadersReference: hasHeadersTemplateVariables && Boolean(parsed.contextConfigId),\n contextConfigReference: parsed.contextConfigReference,\n contextConfigHeadersReference: parsed.contextConfigHeadersReference,\n });\n if (contextImportResolution.namedImports.length > 0 && parsed.contextConfigId) {\n sourceFile.addImportDeclaration({\n namedImports: contextImportResolution.namedImports,\n moduleSpecifier: `../../context-configs/${parsed.contextConfigId}`,\n });\n }\n\n const canUseToolReferenceOverrides = addCanUseToolImports(\n sourceFile,\n parsed.canUse,\n parsed.referenceOverrides?.tools,\n parsed.referencePathOverrides?.tools,\n reservedReferenceNames\n );\n addDataComponentImports(\n sourceFile,\n parsed.dataComponents,\n parsed.referenceOverrides?.dataComponents\n );\n addArtifactComponentImports(\n sourceFile,\n parsed.artifactComponents,\n parsed.referenceOverrides?.artifactComponents\n );\n addDelegateTargetImports(sourceFile, {\n currentSubAgentId: parsed.id,\n canDelegateTo: parsed.canDelegateTo,\n canTransferTo: parsed.canTransferTo,\n referenceOverrides: {\n subAgents: parsed.referenceOverrides?.subAgents,\n agents: parsed.referenceOverrides?.agents,\n externalAgents: parsed.referenceOverrides?.externalAgents,\n },\n referencePathOverrides: {\n subAgents: parsed.referencePathOverrides?.subAgents,\n agents: parsed.referencePathOverrides?.agents,\n externalAgents: parsed.referencePathOverrides?.externalAgents,\n },\n });\n\n writeSubAgentConfig(\n configObject,\n {\n contextReference: contextImportResolution.contextReferenceName,\n headersReference: contextImportResolution.headersReferenceName,\n },\n canUseToolReferenceOverrides,\n parsed\n );\n\n return sourceFile;\n}\n\nfunction addCanUseToolImports(\n sourceFile: SourceFile,\n canUse?: unknown[],\n toolReferenceOverrides?: Record<string, string>,\n toolReferencePathOverrides?: Record<string, string>,\n reservedReferenceNames?: Set<string>\n): Record<string, string> {\n const toolImportsById = new Map<string, { importName: string; modulePath: string }>();\n for (const item of canUse ?? []) {\n const toolId =\n typeof item === 'string'\n ? item\n : isPlainObject(item) && typeof item.toolId === 'string'\n ? item.toolId\n : undefined;\n if (!toolId || toolImportsById.has(toolId)) {\n continue;\n }\n toolImportsById.set(toolId, {\n importName: resolveReferenceName(toolId, [toolReferenceOverrides]),\n modulePath: resolveReferenceName(toolId, [toolReferencePathOverrides]),\n });\n }\n\n const localReferenceNamesByToolId: Record<string, string> = {};\n const localNames = reservedReferenceNames ?? new Set<string>();\n for (const [toolId, { importName, modulePath }] of toolImportsById) {\n const localName = resolveNonCollidingName(importName, localNames);\n localReferenceNamesByToolId[toolId] = localName;\n sourceFile.addImportDeclaration({\n namedImports: [\n importName === localName ? importName : { name: importName, alias: localName },\n ],\n moduleSpecifier: `../../tools/${modulePath}`,\n });\n }\n\n return localReferenceNamesByToolId;\n}\n\nfunction addDataComponentImports(\n sourceFile: SourceFile,\n dataComponents?: string[],\n dataComponentReferenceOverrides?: Record<string, string>\n): void {\n addReferenceImports(\n sourceFile,\n dataComponents,\n '../../data-components',\n dataComponentReferenceOverrides\n );\n}\n\nfunction addArtifactComponentImports(\n sourceFile: SourceFile,\n artifactComponents?: string[],\n artifactComponentReferenceOverrides?: Record<string, string>\n): void {\n addReferenceImports(\n sourceFile,\n artifactComponents,\n '../../artifact-components',\n artifactComponentReferenceOverrides\n );\n}\n\nfunction addReferenceImports(\n sourceFile: SourceFile,\n references: string[] | undefined,\n basePath: string,\n referenceOverrides?: Record<string, string>\n): void {\n const importByReferenceId = new Map<string, string>();\n for (const referenceId of references ?? []) {\n if (!referenceId || importByReferenceId.has(referenceId)) {\n continue;\n }\n\n importByReferenceId.set(referenceId, resolveReferenceName(referenceId, [referenceOverrides]));\n }\n\n for (const [referenceId, referenceName] of importByReferenceId) {\n sourceFile.addImportDeclaration({\n namedImports: [referenceName],\n moduleSpecifier: `${basePath}/${referenceId}`,\n });\n }\n}\n\ntype DelegateTargetType = 'subAgents' | 'agents' | 'externalAgents';\n\nfunction addDelegateTargetImports(\n sourceFile: SourceFile,\n options: {\n currentSubAgentId: string;\n canDelegateTo?: unknown[];\n canTransferTo?: string[];\n referenceOverrides: {\n subAgents?: Record<string, string>;\n agents?: Record<string, string>;\n externalAgents?: Record<string, string>;\n };\n referencePathOverrides?: {\n subAgents?: Record<string, string>;\n agents?: Record<string, string>;\n externalAgents?: Record<string, string>;\n };\n }\n): void {\n const importsByTarget = new Map<\n string,\n { type: DelegateTargetType; id: string; name: string; modulePath?: string }\n >();\n\n for (const item of options.canDelegateTo ?? []) {\n const resolvedTarget = resolveDelegateTargetImport(\n item,\n options.referenceOverrides,\n options.referencePathOverrides\n );\n if (!resolvedTarget) {\n continue;\n }\n if (resolvedTarget.type === 'subAgents' && resolvedTarget.id === options.currentSubAgentId) {\n continue;\n }\n importsByTarget.set(`${resolvedTarget.type}:${resolvedTarget.id}`, resolvedTarget);\n }\n\n for (const targetId of options.canTransferTo ?? []) {\n const resolvedTarget = resolveDelegateTargetImport(\n targetId,\n options.referenceOverrides,\n options.referencePathOverrides\n );\n if (!resolvedTarget) {\n continue;\n }\n if (resolvedTarget.type === 'subAgents' && resolvedTarget.id === options.currentSubAgentId) {\n continue;\n }\n importsByTarget.set(`${resolvedTarget.type}:${resolvedTarget.id}`, resolvedTarget);\n }\n\n for (const target of importsByTarget.values()) {\n sourceFile.addImportDeclaration({\n namedImports: [target.name],\n moduleSpecifier: resolveDelegateImportModuleSpecifier(\n target.type,\n target.id,\n target.modulePath\n ),\n });\n }\n}\n\nfunction resolveDelegateTargetImport(\n canDelegateToEntry: unknown,\n referenceOverrides: {\n subAgents?: Record<string, string>;\n agents?: Record<string, string>;\n externalAgents?: Record<string, string>;\n },\n referencePathOverrides?: {\n subAgents?: Record<string, string>;\n agents?: Record<string, string>;\n externalAgents?: Record<string, string>;\n }\n): { type: DelegateTargetType; id: string; name: string; modulePath?: string } | undefined {\n if (typeof canDelegateToEntry === 'string') {\n const resolvedType = resolveDelegateTargetType(canDelegateToEntry, referenceOverrides);\n if (!resolvedType) {\n return;\n }\n\n return {\n type: resolvedType,\n id: canDelegateToEntry,\n name: resolveReferenceName(canDelegateToEntry, [referenceOverrides[resolvedType]]),\n ...((resolvedType === 'subAgents' && {\n modulePath: referencePathOverrides?.subAgents?.[canDelegateToEntry],\n }) ||\n (resolvedType === 'agents' && {\n modulePath: referencePathOverrides?.agents?.[canDelegateToEntry],\n }) ||\n (resolvedType === 'externalAgents' && {\n modulePath: referencePathOverrides?.externalAgents?.[canDelegateToEntry],\n })),\n };\n }\n\n if (!isPlainObject(canDelegateToEntry)) {\n return;\n }\n\n if (typeof canDelegateToEntry.subAgentId === 'string') {\n return {\n type: 'subAgents',\n id: canDelegateToEntry.subAgentId,\n name: resolveReferenceName(canDelegateToEntry.subAgentId, [referenceOverrides.subAgents]),\n modulePath: referencePathOverrides?.subAgents?.[canDelegateToEntry.subAgentId],\n };\n }\n\n if (typeof canDelegateToEntry.agentId === 'string') {\n return {\n type: 'agents',\n id: canDelegateToEntry.agentId,\n name: resolveReferenceName(canDelegateToEntry.agentId, [referenceOverrides.agents]),\n modulePath: referencePathOverrides?.agents?.[canDelegateToEntry.agentId],\n };\n }\n\n if (typeof canDelegateToEntry.externalAgentId === 'string') {\n return {\n type: 'externalAgents',\n id: canDelegateToEntry.externalAgentId,\n name: resolveReferenceName(canDelegateToEntry.externalAgentId, [\n referenceOverrides.externalAgents,\n ]),\n modulePath: referencePathOverrides?.externalAgents?.[canDelegateToEntry.externalAgentId],\n };\n }\n}\n\nfunction resolveDelegateTargetType(\n targetId: string,\n referenceOverrides: {\n subAgents?: Record<string, string>;\n agents?: Record<string, string>;\n externalAgents?: Record<string, string>;\n }\n): DelegateTargetType | undefined {\n if (referenceOverrides.subAgents?.[targetId]) {\n return 'subAgents';\n }\n if (referenceOverrides.agents?.[targetId]) {\n return 'agents';\n }\n if (referenceOverrides.externalAgents?.[targetId]) {\n return 'externalAgents';\n }\n\n return 'subAgents';\n}\n\nfunction resolveDelegateImportModuleSpecifier(\n type: DelegateTargetType,\n id: string,\n modulePath?: string\n): string {\n const path = modulePath ?? id;\n switch (type) {\n case 'subAgents':\n return `./${path}`;\n case 'agents':\n return `../${path}`;\n case 'externalAgents':\n return `../../external-agents/${path}`;\n }\n}\n\nfunction writeSubAgentConfig(\n configObject: ObjectLiteralExpression,\n templateReferences: {\n contextReference?: string;\n headersReference?: string;\n },\n canUseToolReferenceOverrides: Record<string, string>,\n {\n dataComponents,\n name,\n canDelegateTo,\n canTransferTo,\n skills,\n artifactComponents,\n canUse,\n referenceOverrides,\n referencePathOverrides: _referencePathOverrides,\n contextConfigId: _contextConfigId,\n contextConfigReference: _contextConfigReference,\n contextConfigHeadersReference: _contextConfigHeadersReference,\n ...rest\n }: SubAgentOutput\n) {\n rest = { ...rest };\n rest.prompt &&= formatTemplate(rest.prompt, templateReferences);\n for (const [k, v] of Object.entries(rest)) {\n addValueToObject(configObject, k, v);\n }\n addStringProperty(configObject, 'name', resolveSubAgentName(rest.id, name));\n\n const canUseReferences = collectCanUseReferences(\n canUse,\n Object.keys(canUseToolReferenceOverrides).length\n ? canUseToolReferenceOverrides\n : referenceOverrides?.tools\n );\n if (canUseReferences.length) {\n addReferenceGetterProperty(configObject, 'canUse', canUseReferences);\n }\n\n const canDelegateToReferences = collectCanDelegateToReferences(canDelegateTo, {\n subAgents: referenceOverrides?.subAgents,\n agents: referenceOverrides?.agents,\n externalAgents: referenceOverrides?.externalAgents,\n });\n if (canDelegateToReferences.length) {\n addReferenceGetterProperty(configObject, 'canDelegateTo', canDelegateToReferences);\n }\n\n if (hasReferences(canTransferTo)) {\n addReferenceGetterProperty(\n configObject,\n 'canTransferTo',\n canTransferTo.map((id) =>\n resolveReferenceName(id, [\n referenceOverrides?.subAgents,\n referenceOverrides?.agents,\n referenceOverrides?.externalAgents,\n ])\n )\n );\n }\n\n if (hasReferences(dataComponents)) {\n addReferenceGetterProperty(\n configObject,\n 'dataComponents',\n dataComponents.map((id) => resolveReferenceName(id, [referenceOverrides?.dataComponents]))\n );\n }\n\n if (hasReferences(artifactComponents)) {\n addReferenceGetterProperty(\n configObject,\n 'artifactComponents',\n artifactComponents.map((id) =>\n resolveReferenceName(id, [referenceOverrides?.artifactComponents])\n )\n );\n }\n\n const collectedSkills = collectSkills(skills);\n if (collectedSkills.length > 0) {\n const skillsProperty = configObject.addPropertyAssignment({\n name: 'skills',\n initializer: '() => []',\n });\n const skillsGetter = skillsProperty.getInitializerIfKindOrThrow(SyntaxKind.ArrowFunction);\n const skillsArray = skillsGetter.getBody().asKindOrThrow(SyntaxKind.ArrayLiteralExpression);\n for (const skill of collectedSkills) {\n skillsArray.addElement(skill);\n }\n }\n}\n"],"mappings":";;;;;;;AAwBA,MAAM,0BAA0B,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC;AAE3E,MAAM,2BAA2B,EAAE,OAAO;CACxC,OAAO,wBAAwB,UAAU;CACzC,WAAW,wBAAwB,UAAU;CAC7C,QAAQ,wBAAwB,UAAU;CAC1C,gBAAgB,wBAAwB,UAAU;CAClD,gBAAgB,wBAAwB,UAAU;CAClD,oBAAoB,wBAAwB,UAAU;CACvD,CAAC;AAEF,MAAM,+BAA+B,EAAE,OAAO;CAC5C,OAAO,wBAAwB,UAAU;CACzC,WAAW,wBAAwB,UAAU;CAC7C,QAAQ,wBAAwB,UAAU;CAC1C,gBAAgB,wBAAwB,UAAU;CACnD,CAAC;AAEF,MAAM,iCAAiC,EAAE,OAAO;CAC9C,MAAM,EAAE,QAAQ,CAAC,UAAU;CAC3B,OAAO,EAAE,SAAS,CAAC,UAAU;CAC9B,CAAC;AAEF,MAAM,WAAW,2BAA2B,KAAK;CAC/C,IAAI;CACJ,QAAQ;CACR,MAAM;CACN,aAAa;CACb,UAAU;CACX,CAAC;AAEF,MAAM,iBAAiB,EAAE,aAAa;CACpC,GAAG,SAAS;CACZ,QAAQ,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,OAAO;CAClE,aAAa,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,YAAY;CAC5E,UAAU,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,SAAS;CACtE,QAAQ,EAAE,YAAY,MAAM,KAAK,QAAW,EAAE,YAAY,EAAE,CAAC,CAAC,UAAU,CAAC;CACzE,QAAQ,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,UAAU;CACvC,QAAQ,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,UAAU;CACvC,eAAe,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,UAAU;CAC9C,eAAe,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,UAAU;CAC7C,gBAAgB,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,UAAU;CAC9C,oBAAoB,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,UAAU;CAClD,oBAAoB,yBAAyB,UAAU;CACvD,wBAAwB,6BAA6B,UAAU;CAC/D,iBAAiB,EAAE,QAAQ,CAAC,UAAU,CAAC,UAAU;CACjD,wBAAwB,+BAA+B,UAAU;CACjE,+BAA+B,+BAA+B,UAAU;CACzE,CAAC;AAKF,SAAgB,2BAA2B,EACzC,YACA,GAAG,QACmD;CACtD,MAAM,SAAS,eAAe,UAAU,KAAK;AAC7C,KAAI,CAAC,OAAO,QACV,OAAM,IAAI,MAAM,qCAAqC,EAAE,cAAc,OAAO,MAAM,GAAG;CAGvF,MAAM,SAAS,OAAO;CACtB,MAAM,uBAAuB,4BAA4B,OAAO,IAAI,OAAO,KAAK;CAChF,MAAM,EAAE,YAAY,iBAAiB,wBAAwB;EAC3D,YAAY;EACZ,cAAc;EACf,CAAC;CACF,MAAM,yBAAyB,IAAI,IAAI,CAAC,qBAAqB,CAAC;CAE9D,MAAM,0BACJ,OAAO,OAAO,WAAW,WAAW,6BAA6B,OAAO,OAAO,GAAG,EAAE;CACtF,MAAM,8BAA8B,wBAAwB,MACzD,iBAAiB,CAAC,aAAa,WAAW,WAAW,CACvD;CACD,MAAM,8BAA8B,wBAAwB,MAAM,iBAChE,aAAa,WAAW,WAAW,CACpC;CACD,MAAM,0BAA0B,8BAA8B;EAC5D,eAAe;EACf,+BAA+B,+BAA+B,QAAQ,OAAO,gBAAgB;EAC7F,+BAA+B,+BAA+B,QAAQ,OAAO,gBAAgB;EAC7F,wBAAwB,OAAO;EAC/B,+BAA+B,OAAO;EACvC,CAAC;AACF,KAAI,wBAAwB,aAAa,SAAS,KAAK,OAAO,gBAC5D,YAAW,qBAAqB;EAC9B,cAAc,wBAAwB;EACtC,iBAAiB,yBAAyB,OAAO;EAClD,CAAC;CAGJ,MAAM,+BAA+B,qBACnC,YACA,OAAO,QACP,OAAO,oBAAoB,OAC3B,OAAO,wBAAwB,OAC/B,uBACD;AACD,yBACE,YACA,OAAO,gBACP,OAAO,oBAAoB,eAC5B;AACD,6BACE,YACA,OAAO,oBACP,OAAO,oBAAoB,mBAC5B;AACD,0BAAyB,YAAY;EACnC,mBAAmB,OAAO;EAC1B,eAAe,OAAO;EACtB,eAAe,OAAO;EACtB,oBAAoB;GAClB,WAAW,OAAO,oBAAoB;GACtC,QAAQ,OAAO,oBAAoB;GACnC,gBAAgB,OAAO,oBAAoB;GAC5C;EACD,wBAAwB;GACtB,WAAW,OAAO,wBAAwB;GAC1C,QAAQ,OAAO,wBAAwB;GACvC,gBAAgB,OAAO,wBAAwB;GAChD;EACF,CAAC;AAEF,qBACE,cACA;EACE,kBAAkB,wBAAwB;EAC1C,kBAAkB,wBAAwB;EAC3C,EACD,8BACA,OACD;AAED,QAAO;;AAGT,SAAS,qBACP,YACA,QACA,wBACA,4BACA,wBACwB;CACxB,MAAM,kCAAkB,IAAI,KAAyD;AACrF,MAAK,MAAM,QAAQ,UAAU,EAAE,EAAE;EAC/B,MAAM,SACJ,OAAO,SAAS,WACZ,OACA,cAAc,KAAK,IAAI,OAAO,KAAK,WAAW,WAC5C,KAAK,SACL;AACR,MAAI,CAAC,UAAU,gBAAgB,IAAI,OAAO,CACxC;AAEF,kBAAgB,IAAI,QAAQ;GAC1B,YAAY,qBAAqB,QAAQ,CAAC,uBAAuB,CAAC;GAClE,YAAY,qBAAqB,QAAQ,CAAC,2BAA2B,CAAC;GACvE,CAAC;;CAGJ,MAAM,8BAAsD,EAAE;CAC9D,MAAM,aAAa,0CAA0B,IAAI,KAAa;AAC9D,MAAK,MAAM,CAAC,QAAQ,EAAE,YAAY,iBAAiB,iBAAiB;EAClE,MAAM,YAAY,wBAAwB,YAAY,WAAW;AACjE,8BAA4B,UAAU;AACtC,aAAW,qBAAqB;GAC9B,cAAc,CACZ,eAAe,YAAY,aAAa;IAAE,MAAM;IAAY,OAAO;IAAW,CAC/E;GACD,iBAAiB,eAAe;GACjC,CAAC;;AAGJ,QAAO;;AAGT,SAAS,wBACP,YACA,gBACA,iCACM;AACN,qBACE,YACA,gBACA,yBACA,gCACD;;AAGH,SAAS,4BACP,YACA,oBACA,qCACM;AACN,qBACE,YACA,oBACA,6BACA,oCACD;;AAGH,SAAS,oBACP,YACA,YACA,UACA,oBACM;CACN,MAAM,sCAAsB,IAAI,KAAqB;AACrD,MAAK,MAAM,eAAe,cAAc,EAAE,EAAE;AAC1C,MAAI,CAAC,eAAe,oBAAoB,IAAI,YAAY,CACtD;AAGF,sBAAoB,IAAI,aAAa,qBAAqB,aAAa,CAAC,mBAAmB,CAAC,CAAC;;AAG/F,MAAK,MAAM,CAAC,aAAa,kBAAkB,oBACzC,YAAW,qBAAqB;EAC9B,cAAc,CAAC,cAAc;EAC7B,iBAAiB,GAAG,SAAS,GAAG;EACjC,CAAC;;AAMN,SAAS,yBACP,YACA,SAeM;CACN,MAAM,kCAAkB,IAAI,KAGzB;AAEH,MAAK,MAAM,QAAQ,QAAQ,iBAAiB,EAAE,EAAE;EAC9C,MAAM,iBAAiB,4BACrB,MACA,QAAQ,oBACR,QAAQ,uBACT;AACD,MAAI,CAAC,eACH;AAEF,MAAI,eAAe,SAAS,eAAe,eAAe,OAAO,QAAQ,kBACvE;AAEF,kBAAgB,IAAI,GAAG,eAAe,KAAK,GAAG,eAAe,MAAM,eAAe;;AAGpF,MAAK,MAAM,YAAY,QAAQ,iBAAiB,EAAE,EAAE;EAClD,MAAM,iBAAiB,4BACrB,UACA,QAAQ,oBACR,QAAQ,uBACT;AACD,MAAI,CAAC,eACH;AAEF,MAAI,eAAe,SAAS,eAAe,eAAe,OAAO,QAAQ,kBACvE;AAEF,kBAAgB,IAAI,GAAG,eAAe,KAAK,GAAG,eAAe,MAAM,eAAe;;AAGpF,MAAK,MAAM,UAAU,gBAAgB,QAAQ,CAC3C,YAAW,qBAAqB;EAC9B,cAAc,CAAC,OAAO,KAAK;EAC3B,iBAAiB,qCACf,OAAO,MACP,OAAO,IACP,OAAO,WACR;EACF,CAAC;;AAIN,SAAS,4BACP,oBACA,oBAKA,wBAKyF;AACzF,KAAI,OAAO,uBAAuB,UAAU;EAC1C,MAAM,eAAe,0BAA0B,oBAAoB,mBAAmB;AACtF,MAAI,CAAC,aACH;AAGF,SAAO;GACL,MAAM;GACN,IAAI;GACJ,MAAM,qBAAqB,oBAAoB,CAAC,mBAAmB,cAAc,CAAC;GAClF,GAAK,iBAAiB,eAAe,EACnC,YAAY,wBAAwB,YAAY,qBACjD,IACE,iBAAiB,YAAY,EAC5B,YAAY,wBAAwB,SAAS,qBAC9C,IACA,iBAAiB,oBAAoB,EACpC,YAAY,wBAAwB,iBAAiB,qBACtD;GACJ;;AAGH,KAAI,CAAC,cAAc,mBAAmB,CACpC;AAGF,KAAI,OAAO,mBAAmB,eAAe,SAC3C,QAAO;EACL,MAAM;EACN,IAAI,mBAAmB;EACvB,MAAM,qBAAqB,mBAAmB,YAAY,CAAC,mBAAmB,UAAU,CAAC;EACzF,YAAY,wBAAwB,YAAY,mBAAmB;EACpE;AAGH,KAAI,OAAO,mBAAmB,YAAY,SACxC,QAAO;EACL,MAAM;EACN,IAAI,mBAAmB;EACvB,MAAM,qBAAqB,mBAAmB,SAAS,CAAC,mBAAmB,OAAO,CAAC;EACnF,YAAY,wBAAwB,SAAS,mBAAmB;EACjE;AAGH,KAAI,OAAO,mBAAmB,oBAAoB,SAChD,QAAO;EACL,MAAM;EACN,IAAI,mBAAmB;EACvB,MAAM,qBAAqB,mBAAmB,iBAAiB,CAC7D,mBAAmB,eACpB,CAAC;EACF,YAAY,wBAAwB,iBAAiB,mBAAmB;EACzE;;AAIL,SAAS,0BACP,UACA,oBAKgC;AAChC,KAAI,mBAAmB,YAAY,UACjC,QAAO;AAET,KAAI,mBAAmB,SAAS,UAC9B,QAAO;AAET,KAAI,mBAAmB,iBAAiB,UACtC,QAAO;AAGT,QAAO;;AAGT,SAAS,qCACP,MACA,IACA,YACQ;CACR,MAAM,OAAO,cAAc;AAC3B,SAAQ,MAAR;EACE,KAAK,YACH,QAAO,KAAK;EACd,KAAK,SACH,QAAO,MAAM;EACf,KAAK,iBACH,QAAO,yBAAyB;;;AAItC,SAAS,oBACP,cACA,oBAIA,8BACA,EACE,gBACA,MACA,eACA,eACA,QACA,oBACA,QACA,oBACA,wBAAwB,yBACxB,iBAAiB,kBACjB,wBAAwB,yBACxB,+BAA+B,gCAC/B,GAAG,QAEL;AACA,QAAO,EAAE,GAAG,MAAM;AAClB,MAAK,WAAW,eAAe,KAAK,QAAQ,mBAAmB;AAC/D,MAAK,MAAM,CAAC,GAAG,MAAM,OAAO,QAAQ,KAAK,CACvC,kBAAiB,cAAc,GAAG,EAAE;AAEtC,mBAAkB,cAAc,QAAQ,oBAAoB,KAAK,IAAI,KAAK,CAAC;CAE3E,MAAM,mBAAmB,wBACvB,QACA,OAAO,KAAK,6BAA6B,CAAC,SACtC,+BACA,oBAAoB,MACzB;AACD,KAAI,iBAAiB,OACnB,4BAA2B,cAAc,UAAU,iBAAiB;CAGtE,MAAM,0BAA0B,+BAA+B,eAAe;EAC5E,WAAW,oBAAoB;EAC/B,QAAQ,oBAAoB;EAC5B,gBAAgB,oBAAoB;EACrC,CAAC;AACF,KAAI,wBAAwB,OAC1B,4BAA2B,cAAc,iBAAiB,wBAAwB;AAGpF,KAAI,cAAc,cAAc,CAC9B,4BACE,cACA,iBACA,cAAc,KAAK,OACjB,qBAAqB,IAAI;EACvB,oBAAoB;EACpB,oBAAoB;EACpB,oBAAoB;EACrB,CAAC,CACH,CACF;AAGH,KAAI,cAAc,eAAe,CAC/B,4BACE,cACA,kBACA,eAAe,KAAK,OAAO,qBAAqB,IAAI,CAAC,oBAAoB,eAAe,CAAC,CAAC,CAC3F;AAGH,KAAI,cAAc,mBAAmB,CACnC,4BACE,cACA,sBACA,mBAAmB,KAAK,OACtB,qBAAqB,IAAI,CAAC,oBAAoB,mBAAmB,CAAC,CACnE,CACF;CAGH,MAAM,kBAAkB,cAAc,OAAO;AAC7C,KAAI,gBAAgB,SAAS,GAAG;EAM9B,MAAM,cALiB,aAAa,sBAAsB;GACxD,MAAM;GACN,aAAa;GACd,CAAC,CACkC,4BAA4B,WAAW,cAAc,CACxD,SAAS,CAAC,cAAc,WAAW,uBAAuB;AAC3F,OAAK,MAAM,SAAS,gBAClB,aAAY,WAAW,MAAM"}
|
|
1
|
+
{"version":3,"file":"sub-agent-generator.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/sub-agent-generator.ts"],"sourcesContent":["import { join } from 'node:path';\nimport { FullAgentAgentInsertSchema } from '@inkeep/agents-core';\nimport type { ObjectLiteralExpression, SourceFile } from 'ts-morph';\nimport { z } from 'zod';\nimport { asRecord } from '../collector-common';\nimport {\n collectContextTemplateReferences,\n collectSubAgentDependencyReferenceOverrides,\n} from '../collector-reference-helpers';\nimport type { GenerationTask } from '../generation-types';\nimport {\n addResolvedReferenceImports,\n resolveReferenceBinding,\n resolveReferenceBindingsFromIds,\n toReferenceNameRecord,\n} from '../reference-resolution';\nimport { generateFactorySourceFile } from '../simple-factory-generator';\nimport {\n addValueToObject,\n buildComponentFileName,\n codeReference,\n collectTemplateVariableNames,\n createArrayGetterValue,\n createReferenceGetterValue,\n formatTemplate,\n hasReferences,\n isPlainObject,\n resolveReferenceName,\n toCamelCase,\n} from '../utils';\nimport {\n collectCanDelegateToReferences,\n collectCanUseReferences,\n collectSkills,\n type NormalizedCanDelegateToEntry,\n type NormalizedCanUseEntry,\n type NormalizedDelegateTargetType,\n type NormalizedSkillEntry,\n resolveSubAgentName,\n resolveSubAgentVariableName,\n} from './helpers/sub-agent';\n\nconst ReferenceNameByIdSchema = z.record(z.string(), z.string().nonempty());\n\nconst ReferenceOverridesSchema = z.object({\n tools: ReferenceNameByIdSchema.optional(),\n subAgents: ReferenceNameByIdSchema.optional(),\n agents: ReferenceNameByIdSchema.optional(),\n externalAgents: ReferenceNameByIdSchema.optional(),\n dataComponents: ReferenceNameByIdSchema.optional(),\n artifactComponents: ReferenceNameByIdSchema.optional(),\n});\n\nconst ReferencePathOverridesSchema = z.object({\n tools: ReferenceNameByIdSchema.optional(),\n subAgents: ReferenceNameByIdSchema.optional(),\n agents: ReferenceNameByIdSchema.optional(),\n externalAgents: ReferenceNameByIdSchema.optional(),\n});\n\nconst ContextTemplateReferenceSchema = z.object({\n name: z.string().nonempty(),\n local: z.boolean().optional(),\n});\n\nconst MySchema = FullAgentAgentInsertSchema.pick({\n id: true,\n prompt: true,\n name: true,\n description: true,\n stopWhen: true,\n});\n\nconst BaseSubAgentSchema = z.strictObject({\n ...MySchema.shape,\n prompt: z.preprocess((v) => v || undefined, MySchema.shape.prompt),\n description: z.preprocess((v) => v || undefined, MySchema.shape.description),\n stopWhen: z.preprocess((v) => v ?? undefined, MySchema.shape.stopWhen),\n models: z.preprocess((v) => v ?? undefined, z.looseObject({}).optional()),\n skills: z.array(z.unknown()).optional(),\n canUse: z.array(z.unknown()).optional(),\n canDelegateTo: z.array(z.unknown()).optional(),\n canTransferTo: z.array(z.string()).optional(),\n dataComponents: z.array(z.string()).optional(),\n artifactComponents: z.array(z.string()).optional(),\n referenceOverrides: ReferenceOverridesSchema.optional(),\n referencePathOverrides: ReferencePathOverridesSchema.optional(),\n contextConfigId: z.string().nonempty().optional(),\n contextConfigReference: ContextTemplateReferenceSchema.optional(),\n contextConfigHeadersReference: ContextTemplateReferenceSchema.optional(),\n});\n\nconst SubAgentSchema = BaseSubAgentSchema.transform((data) => ({\n ...data,\n normalizedCanUse: normalizeCanUseEntries(data.canUse),\n normalizedCanDelegateTo: normalizeCanDelegateToEntries(data.canDelegateTo),\n normalizedSkills: normalizeSkills(data.skills),\n}));\n\ntype SubAgentInput = z.input<typeof SubAgentSchema>;\ntype SubAgentOutput = z.output<typeof SubAgentSchema>;\n\nfunction normalizeCanUseEntries(canUse?: unknown[]): NormalizedCanUseEntry[] {\n if (!Array.isArray(canUse)) {\n return [];\n }\n\n const entries: NormalizedCanUseEntry[] = [];\n for (const item of canUse) {\n if (typeof item === 'string') {\n entries.push({ toolId: item });\n continue;\n }\n\n if (!isPlainObject(item) || typeof item.toolId !== 'string') {\n continue;\n }\n\n const selectedTools =\n Array.isArray(item.toolSelection) && item.toolSelection.length\n ? item.toolSelection\n : Array.isArray(item.selectedTools) && item.selectedTools.length\n ? item.selectedTools\n : undefined;\n const headers =\n isPlainObject(item.headers) && Object.keys(item.headers).length > 0\n ? item.headers\n : undefined;\n const toolPolicies =\n isPlainObject(item.toolPolicies) && Object.keys(item.toolPolicies).length > 0\n ? item.toolPolicies\n : undefined;\n\n entries.push({\n toolId: item.toolId,\n ...(selectedTools && { selectedTools }),\n ...(headers && { headers }),\n ...(toolPolicies && { toolPolicies }),\n });\n }\n\n return entries;\n}\n\nfunction normalizeCanDelegateToEntries(canDelegateTo?: unknown[]): NormalizedCanDelegateToEntry[] {\n if (!Array.isArray(canDelegateTo)) {\n return [];\n }\n\n const entries: NormalizedCanDelegateToEntry[] = [];\n for (const item of canDelegateTo) {\n if (typeof item === 'string') {\n entries.push({ id: item });\n continue;\n }\n\n if (!isPlainObject(item)) {\n continue;\n }\n\n const headers =\n isPlainObject(item.headers) && Object.keys(item.headers).length > 0\n ? item.headers\n : undefined;\n const addEntry = (type: NormalizedDelegateTargetType, id: unknown) => {\n if (typeof id !== 'string') {\n return;\n }\n\n entries.push({\n id,\n type,\n ...(headers && { headers }),\n });\n };\n\n addEntry('subAgents', item.subAgentId);\n addEntry('agents', item.agentId);\n addEntry('externalAgents', item.externalAgentId);\n }\n\n return entries;\n}\n\nfunction normalizeSkills(skills?: unknown[]): NormalizedSkillEntry[] {\n if (!Array.isArray(skills)) {\n return [];\n }\n\n const normalizedSkills: NormalizedSkillEntry[] = [];\n for (const skill of skills) {\n if (typeof skill === 'string') {\n normalizedSkills.push(skill);\n continue;\n }\n\n if (!isPlainObject(skill)) {\n continue;\n }\n\n const skillId =\n typeof skill.id === 'string'\n ? skill.id\n : typeof skill.skillId === 'string'\n ? skill.skillId\n : undefined;\n if (!skillId) {\n continue;\n }\n\n normalizedSkills.push({\n id: skillId,\n ...(typeof skill.index === 'number' && Number.isInteger(skill.index)\n ? { index: skill.index }\n : {}),\n ...(typeof skill.alwaysLoaded === 'boolean' ? { alwaysLoaded: skill.alwaysLoaded } : {}),\n });\n }\n\n return normalizedSkills;\n}\n\nexport function generateSubAgentDefinition({\n subAgentId,\n ...data\n}: SubAgentInput & Record<string, unknown>): SourceFile {\n return generateFactorySourceFile(data, {\n schema: SubAgentSchema,\n factory: {\n importName: 'subAgent',\n variableName: (parsed) => resolveSubAgentVariableName(parsed.id, parsed.name),\n },\n render({ parsed, sourceFile, configObject }) {\n const subAgentVariableName = resolveSubAgentVariableName(parsed.id, parsed.name);\n const reservedReferenceNames = new Set([subAgentVariableName]);\n\n const promptTemplateVariables =\n typeof parsed.prompt === 'string' ? collectTemplateVariableNames(parsed.prompt) : [];\n const hasContextTemplateVariables = promptTemplateVariables.some(\n (variableName) => !variableName.startsWith('headers.')\n );\n const hasHeadersTemplateVariables = promptTemplateVariables.some((variableName) =>\n variableName.startsWith('headers.')\n );\n let contextReferenceName: string | undefined;\n let headersReferenceName: string | undefined;\n if (parsed.contextConfigId) {\n const contextReferences = [];\n if (hasContextTemplateVariables) {\n const contextReference = resolveReferenceBinding(\n {\n id: `${parsed.contextConfigId}:context`,\n importName:\n parsed.contextConfigReference?.name ?? toCamelCase(parsed.contextConfigId),\n modulePath: parsed.contextConfigId,\n local: parsed.contextConfigReference?.local === true,\n conflictSuffix: 'ContextConfig',\n },\n {\n reservedNames: reservedReferenceNames,\n }\n );\n contextReferenceName = contextReference.localName;\n contextReferences.push(contextReference);\n }\n\n if (hasHeadersTemplateVariables) {\n const headersReference = resolveReferenceBinding(\n {\n id: `${parsed.contextConfigId}:headers`,\n importName:\n parsed.contextConfigHeadersReference?.name ??\n `${toCamelCase(parsed.contextConfigId)}Headers`,\n modulePath: parsed.contextConfigId,\n local: parsed.contextConfigHeadersReference?.local === true,\n conflictSuffix: 'Headers',\n },\n {\n reservedNames: reservedReferenceNames,\n }\n );\n headersReferenceName = headersReference.localName;\n contextReferences.push(headersReference);\n }\n\n addResolvedReferenceImports(sourceFile, contextReferences, () => {\n return `../../context-configs/${parsed.contextConfigId}`;\n });\n }\n\n const canUseToolReferences = resolveReferenceBindingsFromIds({\n ids: collectCanUseToolIds(parsed.normalizedCanUse),\n reservedNames: reservedReferenceNames,\n conflictSuffix: 'Tool',\n collisionStrategy: 'numeric',\n referenceOverrides: parsed.referenceOverrides?.tools,\n referencePathOverrides: parsed.referencePathOverrides?.tools,\n defaultModulePath: toCamelCase,\n });\n addResolvedReferenceImports(sourceFile, canUseToolReferences, (reference) => {\n return `../../tools/${reference.modulePath}`;\n });\n\n const dataComponentReferences = resolveReferenceBindingsFromIds({\n ids: parsed.dataComponents ?? [],\n reservedNames: reservedReferenceNames,\n conflictSuffix: 'DataComponent',\n referenceOverrides: parsed.referenceOverrides?.dataComponents,\n defaultModulePath: (id) => id,\n });\n addResolvedReferenceImports(sourceFile, dataComponentReferences, (reference) => {\n return `../../data-components/${reference.modulePath}`;\n });\n\n const artifactComponentReferences = resolveReferenceBindingsFromIds({\n ids: parsed.artifactComponents ?? [],\n reservedNames: reservedReferenceNames,\n conflictSuffix: 'ArtifactComponent',\n referenceOverrides: parsed.referenceOverrides?.artifactComponents,\n defaultModulePath: (id) => id,\n });\n addResolvedReferenceImports(sourceFile, artifactComponentReferences, (reference) => {\n return `../../artifact-components/${reference.modulePath}`;\n });\n\n const delegateTargetIds = collectDelegateTargetIds(\n parsed.normalizedCanDelegateTo,\n parsed.canTransferTo,\n parsed.id,\n {\n subAgents: parsed.referenceOverrides?.subAgents,\n agents: parsed.referenceOverrides?.agents,\n externalAgents: parsed.referenceOverrides?.externalAgents,\n }\n );\n const subAgentDelegateReferences = resolveReferenceBindingsFromIds({\n ids: delegateTargetIds.subAgents,\n reservedNames: reservedReferenceNames,\n conflictSuffix: 'SubAgent',\n referenceOverrides: parsed.referenceOverrides?.subAgents,\n referencePathOverrides: parsed.referencePathOverrides?.subAgents,\n defaultModulePath: (id) => id,\n });\n addResolvedReferenceImports(sourceFile, subAgentDelegateReferences, (reference) => {\n return `./${reference.modulePath}`;\n });\n const agentDelegateReferences = resolveReferenceBindingsFromIds({\n ids: delegateTargetIds.agents,\n reservedNames: reservedReferenceNames,\n conflictSuffix: 'Agent',\n referenceOverrides: parsed.referenceOverrides?.agents,\n referencePathOverrides: parsed.referencePathOverrides?.agents,\n defaultModulePath: (id) => id,\n });\n addResolvedReferenceImports(sourceFile, agentDelegateReferences, (reference) => {\n return `../${reference.modulePath}`;\n });\n const externalAgentDelegateReferences = resolveReferenceBindingsFromIds({\n ids: delegateTargetIds.externalAgents,\n reservedNames: reservedReferenceNames,\n conflictSuffix: 'ExternalAgent',\n referenceOverrides: parsed.referenceOverrides?.externalAgents,\n referencePathOverrides: parsed.referencePathOverrides?.externalAgents,\n defaultModulePath: (id) => id,\n });\n addResolvedReferenceImports(sourceFile, externalAgentDelegateReferences, (reference) => {\n return `../../external-agents/${reference.modulePath}`;\n });\n\n writeSubAgentConfig(\n configObject,\n {\n contextReference: contextReferenceName,\n headersReference: headersReferenceName,\n },\n {\n tools: toReferenceNameRecord(canUseToolReferences),\n subAgents: toReferenceNameRecord(subAgentDelegateReferences),\n agents: toReferenceNameRecord(agentDelegateReferences),\n externalAgents: toReferenceNameRecord(externalAgentDelegateReferences),\n dataComponents: toReferenceNameRecord(dataComponentReferences),\n artifactComponents: toReferenceNameRecord(artifactComponentReferences),\n },\n parsed\n );\n },\n });\n}\n\ntype DelegateTargetType = 'subAgents' | 'agents' | 'externalAgents';\n\nfunction collectCanUseToolIds(canUse?: NormalizedCanUseEntry[]): string[] {\n if (!canUse?.length) {\n return [];\n }\n\n return [...new Set(canUse.map((item) => item.toolId))];\n}\n\nfunction collectDelegateTargetIds(\n canDelegateTo: NormalizedCanDelegateToEntry[] | undefined,\n canTransferTo: string[] | undefined,\n currentSubAgentId: string,\n referenceOverrides: {\n subAgents?: Record<string, string>;\n agents?: Record<string, string>;\n externalAgents?: Record<string, string>;\n }\n): Record<DelegateTargetType, string[]> {\n const idsByType: Record<DelegateTargetType, Set<string>> = {\n subAgents: new Set<string>(),\n agents: new Set<string>(),\n externalAgents: new Set<string>(),\n };\n\n for (const item of canDelegateTo ?? []) {\n const type = item.type ?? resolveDelegateTargetType(item.id, referenceOverrides);\n if (!(type === 'subAgents' && item.id === currentSubAgentId)) {\n idsByType[type].add(item.id);\n }\n }\n\n for (const targetId of canTransferTo ?? []) {\n const type = resolveDelegateTargetType(targetId, referenceOverrides);\n if (type === 'subAgents' && targetId === currentSubAgentId) {\n continue;\n }\n idsByType[type].add(targetId);\n }\n\n return {\n subAgents: [...idsByType.subAgents],\n agents: [...idsByType.agents],\n externalAgents: [...idsByType.externalAgents],\n };\n}\n\nfunction resolveDelegateTargetType(\n targetId: string,\n referenceOverrides: {\n subAgents?: Record<string, string>;\n agents?: Record<string, string>;\n externalAgents?: Record<string, string>;\n }\n): DelegateTargetType {\n if (referenceOverrides.subAgents?.[targetId]) {\n return 'subAgents';\n }\n if (referenceOverrides.agents?.[targetId]) {\n return 'agents';\n }\n if (referenceOverrides.externalAgents?.[targetId]) {\n return 'externalAgents';\n }\n\n return 'subAgents';\n}\n\nfunction writeSubAgentConfig(\n configObject: ObjectLiteralExpression,\n templateReferences: {\n contextReference?: string;\n headersReference?: string;\n },\n referenceNames: {\n tools?: Record<string, string>;\n subAgents?: Record<string, string>;\n agents?: Record<string, string>;\n externalAgents?: Record<string, string>;\n dataComponents?: Record<string, string>;\n artifactComponents?: Record<string, string>;\n },\n {\n dataComponents,\n name,\n canDelegateTo: _canDelegateTo,\n canTransferTo,\n skills: _skills,\n artifactComponents,\n canUse: _canUse,\n normalizedCanDelegateTo,\n normalizedCanUse,\n normalizedSkills,\n referenceOverrides: _referenceOverrides,\n referencePathOverrides: _referencePathOverrides,\n contextConfigId: _contextConfigId,\n contextConfigReference: _contextConfigReference,\n contextConfigHeadersReference: _contextConfigHeadersReference,\n ...rest\n }: SubAgentOutput\n) {\n rest = { ...rest };\n rest.prompt &&= formatTemplate(rest.prompt, templateReferences);\n const subAgentConfig: Record<string, unknown> = {\n ...rest,\n name: resolveSubAgentName(rest.id, name),\n };\n\n const canUseReferences = collectCanUseReferences(\n normalizedCanUse,\n Object.keys(referenceNames.tools ?? {}).length ? referenceNames.tools : undefined\n );\n if (canUseReferences.length) {\n subAgentConfig.canUse = createReferenceGetterValue(canUseReferences);\n }\n\n const canDelegateToReferences = collectCanDelegateToReferences(normalizedCanDelegateTo, {\n subAgents: referenceNames.subAgents,\n agents: referenceNames.agents,\n externalAgents: referenceNames.externalAgents,\n });\n if (canDelegateToReferences.length) {\n subAgentConfig.canDelegateTo = createReferenceGetterValue(canDelegateToReferences);\n }\n\n if (hasReferences(canTransferTo)) {\n subAgentConfig.canTransferTo = createReferenceGetterValue(\n canTransferTo.map((id) =>\n codeReference(\n resolveReferenceName(id, [\n referenceNames.subAgents,\n referenceNames.agents,\n referenceNames.externalAgents,\n ])\n )\n )\n );\n }\n\n if (hasReferences(dataComponents)) {\n subAgentConfig.dataComponents = createReferenceGetterValue(\n dataComponents.map((id) =>\n codeReference(resolveReferenceName(id, [referenceNames.dataComponents]))\n )\n );\n }\n\n if (hasReferences(artifactComponents)) {\n subAgentConfig.artifactComponents = createReferenceGetterValue(\n artifactComponents.map((id) =>\n codeReference(resolveReferenceName(id, [referenceNames.artifactComponents]))\n )\n );\n }\n\n const collectedSkills = collectSkills(normalizedSkills);\n if (collectedSkills.length > 0) {\n subAgentConfig.skills = createArrayGetterValue(collectedSkills);\n }\n\n for (const [key, value] of Object.entries(subAgentConfig)) {\n addValueToObject(configObject, key, value);\n }\n}\n\nexport const task = {\n type: 'sub-agent',\n collect(context) {\n if (!context.project.agents) {\n return [];\n }\n\n const recordsBySubAgentId = new Map<\n string,\n ReturnType<\n GenerationTask<Parameters<typeof generateSubAgentDefinition>[0]>['collect']\n >[number]\n >();\n\n for (const agentId of context.completeAgentIds) {\n const agentData = context.project.agents[agentId];\n const subAgents = asRecord(agentData?.subAgents);\n if (!subAgents) {\n continue;\n }\n\n for (const [subAgentId, subAgentData] of Object.entries(subAgents)) {\n const payload = asRecord(subAgentData);\n if (!payload) {\n continue;\n }\n\n const dependencyReferences = collectSubAgentDependencyReferenceOverrides(context, payload);\n const subAgentName = typeof payload.name === 'string' ? payload.name : undefined;\n const subAgentFilePath = context.resolver.resolveOutputFilePath(\n 'subAgents',\n subAgentId,\n join(\n context.paths.agentsDir,\n 'sub-agents',\n buildComponentFileName(subAgentId, subAgentName)\n )\n );\n const contextTemplateReferences = collectContextTemplateReferences(\n context,\n agentData,\n subAgentFilePath,\n [typeof payload.prompt === 'string' ? payload.prompt : undefined]\n );\n\n recordsBySubAgentId.set(subAgentId, {\n id: subAgentId,\n filePath: subAgentFilePath,\n payload: {\n subAgentId,\n ...payload,\n ...(dependencyReferences?.referenceOverrides && {\n referenceOverrides: dependencyReferences.referenceOverrides,\n }),\n ...(dependencyReferences?.referencePathOverrides && {\n referencePathOverrides: dependencyReferences.referencePathOverrides,\n }),\n ...(contextTemplateReferences && {\n contextConfigId: contextTemplateReferences.contextConfigId,\n contextConfigReference: contextTemplateReferences.contextConfigReference,\n }),\n ...(contextTemplateReferences?.contextConfigHeadersReference && {\n contextConfigHeadersReference:\n contextTemplateReferences.contextConfigHeadersReference,\n }),\n } as unknown as Parameters<typeof generateSubAgentDefinition>[0],\n });\n }\n }\n\n return [...recordsBySubAgentId.values()];\n },\n generate: generateSubAgentDefinition,\n} satisfies GenerationTask<Parameters<typeof generateSubAgentDefinition>[0]>;\n"],"mappings":";;;;;;;;;;;;;;;;AA0CA,MAAM,0BAA0B,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC;AAE3E,MAAM,2BAA2B,EAAE,OAAO;CACxC,OAAO,wBAAwB,UAAU;CACzC,WAAW,wBAAwB,UAAU;CAC7C,QAAQ,wBAAwB,UAAU;CAC1C,gBAAgB,wBAAwB,UAAU;CAClD,gBAAgB,wBAAwB,UAAU;CAClD,oBAAoB,wBAAwB,UAAU;CACvD,CAAC;AAEF,MAAM,+BAA+B,EAAE,OAAO;CAC5C,OAAO,wBAAwB,UAAU;CACzC,WAAW,wBAAwB,UAAU;CAC7C,QAAQ,wBAAwB,UAAU;CAC1C,gBAAgB,wBAAwB,UAAU;CACnD,CAAC;AAEF,MAAM,iCAAiC,EAAE,OAAO;CAC9C,MAAM,EAAE,QAAQ,CAAC,UAAU;CAC3B,OAAO,EAAE,SAAS,CAAC,UAAU;CAC9B,CAAC;AAEF,MAAM,WAAW,2BAA2B,KAAK;CAC/C,IAAI;CACJ,QAAQ;CACR,MAAM;CACN,aAAa;CACb,UAAU;CACX,CAAC;AAqBF,MAAM,iBAnBqB,EAAE,aAAa;CACxC,GAAG,SAAS;CACZ,QAAQ,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,OAAO;CAClE,aAAa,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,YAAY;CAC5E,UAAU,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,SAAS;CACtE,QAAQ,EAAE,YAAY,MAAM,KAAK,QAAW,EAAE,YAAY,EAAE,CAAC,CAAC,UAAU,CAAC;CACzE,QAAQ,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,UAAU;CACvC,QAAQ,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,UAAU;CACvC,eAAe,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,UAAU;CAC9C,eAAe,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,UAAU;CAC7C,gBAAgB,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,UAAU;CAC9C,oBAAoB,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,UAAU;CAClD,oBAAoB,yBAAyB,UAAU;CACvD,wBAAwB,6BAA6B,UAAU;CAC/D,iBAAiB,EAAE,QAAQ,CAAC,UAAU,CAAC,UAAU;CACjD,wBAAwB,+BAA+B,UAAU;CACjE,+BAA+B,+BAA+B,UAAU;CACzE,CAAC,CAEwC,WAAW,UAAU;CAC7D,GAAG;CACH,kBAAkB,uBAAuB,KAAK,OAAO;CACrD,yBAAyB,8BAA8B,KAAK,cAAc;CAC1E,kBAAkB,gBAAgB,KAAK,OAAO;CAC/C,EAAE;AAKH,SAAS,uBAAuB,QAA6C;AAC3E,KAAI,CAAC,MAAM,QAAQ,OAAO,CACxB,QAAO,EAAE;CAGX,MAAM,UAAmC,EAAE;AAC3C,MAAK,MAAM,QAAQ,QAAQ;AACzB,MAAI,OAAO,SAAS,UAAU;AAC5B,WAAQ,KAAK,EAAE,QAAQ,MAAM,CAAC;AAC9B;;AAGF,MAAI,CAAC,cAAc,KAAK,IAAI,OAAO,KAAK,WAAW,SACjD;EAGF,MAAM,gBACJ,MAAM,QAAQ,KAAK,cAAc,IAAI,KAAK,cAAc,SACpD,KAAK,gBACL,MAAM,QAAQ,KAAK,cAAc,IAAI,KAAK,cAAc,SACtD,KAAK,gBACL;EACR,MAAM,UACJ,cAAc,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,QAAQ,CAAC,SAAS,IAC9D,KAAK,UACL;EACN,MAAM,eACJ,cAAc,KAAK,aAAa,IAAI,OAAO,KAAK,KAAK,aAAa,CAAC,SAAS,IACxE,KAAK,eACL;AAEN,UAAQ,KAAK;GACX,QAAQ,KAAK;GACb,GAAI,iBAAiB,EAAE,eAAe;GACtC,GAAI,WAAW,EAAE,SAAS;GAC1B,GAAI,gBAAgB,EAAE,cAAc;GACrC,CAAC;;AAGJ,QAAO;;AAGT,SAAS,8BAA8B,eAA2D;AAChG,KAAI,CAAC,MAAM,QAAQ,cAAc,CAC/B,QAAO,EAAE;CAGX,MAAM,UAA0C,EAAE;AAClD,MAAK,MAAM,QAAQ,eAAe;AAChC,MAAI,OAAO,SAAS,UAAU;AAC5B,WAAQ,KAAK,EAAE,IAAI,MAAM,CAAC;AAC1B;;AAGF,MAAI,CAAC,cAAc,KAAK,CACtB;EAGF,MAAM,UACJ,cAAc,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,QAAQ,CAAC,SAAS,IAC9D,KAAK,UACL;EACN,MAAM,YAAY,MAAoC,OAAgB;AACpE,OAAI,OAAO,OAAO,SAChB;AAGF,WAAQ,KAAK;IACX;IACA;IACA,GAAI,WAAW,EAAE,SAAS;IAC3B,CAAC;;AAGJ,WAAS,aAAa,KAAK,WAAW;AACtC,WAAS,UAAU,KAAK,QAAQ;AAChC,WAAS,kBAAkB,KAAK,gBAAgB;;AAGlD,QAAO;;AAGT,SAAS,gBAAgB,QAA4C;AACnE,KAAI,CAAC,MAAM,QAAQ,OAAO,CACxB,QAAO,EAAE;CAGX,MAAM,mBAA2C,EAAE;AACnD,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,OAAO,UAAU,UAAU;AAC7B,oBAAiB,KAAK,MAAM;AAC5B;;AAGF,MAAI,CAAC,cAAc,MAAM,CACvB;EAGF,MAAM,UACJ,OAAO,MAAM,OAAO,WAChB,MAAM,KACN,OAAO,MAAM,YAAY,WACvB,MAAM,UACN;AACR,MAAI,CAAC,QACH;AAGF,mBAAiB,KAAK;GACpB,IAAI;GACJ,GAAI,OAAO,MAAM,UAAU,YAAY,OAAO,UAAU,MAAM,MAAM,GAChE,EAAE,OAAO,MAAM,OAAO,GACtB,EAAE;GACN,GAAI,OAAO,MAAM,iBAAiB,YAAY,EAAE,cAAc,MAAM,cAAc,GAAG,EAAE;GACxF,CAAC;;AAGJ,QAAO;;AAGT,SAAgB,2BAA2B,EACzC,YACA,GAAG,QACmD;AACtD,QAAO,0BAA0B,MAAM;EACrC,QAAQ;EACR,SAAS;GACP,YAAY;GACZ,eAAe,WAAW,4BAA4B,OAAO,IAAI,OAAO,KAAK;GAC9E;EACD,OAAO,EAAE,QAAQ,YAAY,gBAAgB;GAC3C,MAAM,uBAAuB,4BAA4B,OAAO,IAAI,OAAO,KAAK;GAChF,MAAM,yBAAyB,IAAI,IAAI,CAAC,qBAAqB,CAAC;GAE9D,MAAM,0BACJ,OAAO,OAAO,WAAW,WAAW,6BAA6B,OAAO,OAAO,GAAG,EAAE;GACtF,MAAM,8BAA8B,wBAAwB,MACzD,iBAAiB,CAAC,aAAa,WAAW,WAAW,CACvD;GACD,MAAM,8BAA8B,wBAAwB,MAAM,iBAChE,aAAa,WAAW,WAAW,CACpC;GACD,IAAI;GACJ,IAAI;AACJ,OAAI,OAAO,iBAAiB;IAC1B,MAAM,oBAAoB,EAAE;AAC5B,QAAI,6BAA6B;KAC/B,MAAM,mBAAmB,wBACvB;MACE,IAAI,GAAG,OAAO,gBAAgB;MAC9B,YACE,OAAO,wBAAwB,QAAQ,YAAY,OAAO,gBAAgB;MAC5E,YAAY,OAAO;MACnB,OAAO,OAAO,wBAAwB,UAAU;MAChD,gBAAgB;MACjB,EACD,EACE,eAAe,wBAChB,CACF;AACD,4BAAuB,iBAAiB;AACxC,uBAAkB,KAAK,iBAAiB;;AAG1C,QAAI,6BAA6B;KAC/B,MAAM,mBAAmB,wBACvB;MACE,IAAI,GAAG,OAAO,gBAAgB;MAC9B,YACE,OAAO,+BAA+B,QACtC,GAAG,YAAY,OAAO,gBAAgB,CAAC;MACzC,YAAY,OAAO;MACnB,OAAO,OAAO,+BAA+B,UAAU;MACvD,gBAAgB;MACjB,EACD,EACE,eAAe,wBAChB,CACF;AACD,4BAAuB,iBAAiB;AACxC,uBAAkB,KAAK,iBAAiB;;AAG1C,gCAA4B,YAAY,yBAAyB;AAC/D,YAAO,yBAAyB,OAAO;MACvC;;GAGJ,MAAM,uBAAuB,gCAAgC;IAC3D,KAAK,qBAAqB,OAAO,iBAAiB;IAClD,eAAe;IACf,gBAAgB;IAChB,mBAAmB;IACnB,oBAAoB,OAAO,oBAAoB;IAC/C,wBAAwB,OAAO,wBAAwB;IACvD,mBAAmB;IACpB,CAAC;AACF,+BAA4B,YAAY,uBAAuB,cAAc;AAC3E,WAAO,eAAe,UAAU;KAChC;GAEF,MAAM,0BAA0B,gCAAgC;IAC9D,KAAK,OAAO,kBAAkB,EAAE;IAChC,eAAe;IACf,gBAAgB;IAChB,oBAAoB,OAAO,oBAAoB;IAC/C,oBAAoB,OAAO;IAC5B,CAAC;AACF,+BAA4B,YAAY,0BAA0B,cAAc;AAC9E,WAAO,yBAAyB,UAAU;KAC1C;GAEF,MAAM,8BAA8B,gCAAgC;IAClE,KAAK,OAAO,sBAAsB,EAAE;IACpC,eAAe;IACf,gBAAgB;IAChB,oBAAoB,OAAO,oBAAoB;IAC/C,oBAAoB,OAAO;IAC5B,CAAC;AACF,+BAA4B,YAAY,8BAA8B,cAAc;AAClF,WAAO,6BAA6B,UAAU;KAC9C;GAEF,MAAM,oBAAoB,yBACxB,OAAO,yBACP,OAAO,eACP,OAAO,IACP;IACE,WAAW,OAAO,oBAAoB;IACtC,QAAQ,OAAO,oBAAoB;IACnC,gBAAgB,OAAO,oBAAoB;IAC5C,CACF;GACD,MAAM,6BAA6B,gCAAgC;IACjE,KAAK,kBAAkB;IACvB,eAAe;IACf,gBAAgB;IAChB,oBAAoB,OAAO,oBAAoB;IAC/C,wBAAwB,OAAO,wBAAwB;IACvD,oBAAoB,OAAO;IAC5B,CAAC;AACF,+BAA4B,YAAY,6BAA6B,cAAc;AACjF,WAAO,KAAK,UAAU;KACtB;GACF,MAAM,0BAA0B,gCAAgC;IAC9D,KAAK,kBAAkB;IACvB,eAAe;IACf,gBAAgB;IAChB,oBAAoB,OAAO,oBAAoB;IAC/C,wBAAwB,OAAO,wBAAwB;IACvD,oBAAoB,OAAO;IAC5B,CAAC;AACF,+BAA4B,YAAY,0BAA0B,cAAc;AAC9E,WAAO,MAAM,UAAU;KACvB;GACF,MAAM,kCAAkC,gCAAgC;IACtE,KAAK,kBAAkB;IACvB,eAAe;IACf,gBAAgB;IAChB,oBAAoB,OAAO,oBAAoB;IAC/C,wBAAwB,OAAO,wBAAwB;IACvD,oBAAoB,OAAO;IAC5B,CAAC;AACF,+BAA4B,YAAY,kCAAkC,cAAc;AACtF,WAAO,yBAAyB,UAAU;KAC1C;AAEF,uBACE,cACA;IACE,kBAAkB;IAClB,kBAAkB;IACnB,EACD;IACE,OAAO,sBAAsB,qBAAqB;IAClD,WAAW,sBAAsB,2BAA2B;IAC5D,QAAQ,sBAAsB,wBAAwB;IACtD,gBAAgB,sBAAsB,gCAAgC;IACtE,gBAAgB,sBAAsB,wBAAwB;IAC9D,oBAAoB,sBAAsB,4BAA4B;IACvE,EACD,OACD;;EAEJ,CAAC;;AAKJ,SAAS,qBAAqB,QAA4C;AACxE,KAAI,CAAC,QAAQ,OACX,QAAO,EAAE;AAGX,QAAO,CAAC,GAAG,IAAI,IAAI,OAAO,KAAK,SAAS,KAAK,OAAO,CAAC,CAAC;;AAGxD,SAAS,yBACP,eACA,eACA,mBACA,oBAKsC;CACtC,MAAM,YAAqD;EACzD,2BAAW,IAAI,KAAa;EAC5B,wBAAQ,IAAI,KAAa;EACzB,gCAAgB,IAAI,KAAa;EAClC;AAED,MAAK,MAAM,QAAQ,iBAAiB,EAAE,EAAE;EACtC,MAAM,OAAO,KAAK,QAAQ,0BAA0B,KAAK,IAAI,mBAAmB;AAChF,MAAI,EAAE,SAAS,eAAe,KAAK,OAAO,mBACxC,WAAU,MAAM,IAAI,KAAK,GAAG;;AAIhC,MAAK,MAAM,YAAY,iBAAiB,EAAE,EAAE;EAC1C,MAAM,OAAO,0BAA0B,UAAU,mBAAmB;AACpE,MAAI,SAAS,eAAe,aAAa,kBACvC;AAEF,YAAU,MAAM,IAAI,SAAS;;AAG/B,QAAO;EACL,WAAW,CAAC,GAAG,UAAU,UAAU;EACnC,QAAQ,CAAC,GAAG,UAAU,OAAO;EAC7B,gBAAgB,CAAC,GAAG,UAAU,eAAe;EAC9C;;AAGH,SAAS,0BACP,UACA,oBAKoB;AACpB,KAAI,mBAAmB,YAAY,UACjC,QAAO;AAET,KAAI,mBAAmB,SAAS,UAC9B,QAAO;AAET,KAAI,mBAAmB,iBAAiB,UACtC,QAAO;AAGT,QAAO;;AAGT,SAAS,oBACP,cACA,oBAIA,gBAQA,EACE,gBACA,MACA,eAAe,gBACf,eACA,QAAQ,SACR,oBACA,QAAQ,SACR,yBACA,kBACA,kBACA,oBAAoB,qBACpB,wBAAwB,yBACxB,iBAAiB,kBACjB,wBAAwB,yBACxB,+BAA+B,gCAC/B,GAAG,QAEL;AACA,QAAO,EAAE,GAAG,MAAM;AAClB,MAAK,WAAW,eAAe,KAAK,QAAQ,mBAAmB;CAC/D,MAAM,iBAA0C;EAC9C,GAAG;EACH,MAAM,oBAAoB,KAAK,IAAI,KAAK;EACzC;CAED,MAAM,mBAAmB,wBACvB,kBACA,OAAO,KAAK,eAAe,SAAS,EAAE,CAAC,CAAC,SAAS,eAAe,QAAQ,OACzE;AACD,KAAI,iBAAiB,OACnB,gBAAe,SAAS,2BAA2B,iBAAiB;CAGtE,MAAM,0BAA0B,+BAA+B,yBAAyB;EACtF,WAAW,eAAe;EAC1B,QAAQ,eAAe;EACvB,gBAAgB,eAAe;EAChC,CAAC;AACF,KAAI,wBAAwB,OAC1B,gBAAe,gBAAgB,2BAA2B,wBAAwB;AAGpF,KAAI,cAAc,cAAc,CAC9B,gBAAe,gBAAgB,2BAC7B,cAAc,KAAK,OACjB,cACE,qBAAqB,IAAI;EACvB,eAAe;EACf,eAAe;EACf,eAAe;EAChB,CAAC,CACH,CACF,CACF;AAGH,KAAI,cAAc,eAAe,CAC/B,gBAAe,iBAAiB,2BAC9B,eAAe,KAAK,OAClB,cAAc,qBAAqB,IAAI,CAAC,eAAe,eAAe,CAAC,CAAC,CACzE,CACF;AAGH,KAAI,cAAc,mBAAmB,CACnC,gBAAe,qBAAqB,2BAClC,mBAAmB,KAAK,OACtB,cAAc,qBAAqB,IAAI,CAAC,eAAe,mBAAmB,CAAC,CAAC,CAC7E,CACF;CAGH,MAAM,kBAAkB,cAAc,iBAAiB;AACvD,KAAI,gBAAgB,SAAS,EAC3B,gBAAe,SAAS,uBAAuB,gBAAgB;AAGjE,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,eAAe,CACvD,kBAAiB,cAAc,KAAK,MAAM;;AAI9C,MAAa,OAAO;CAClB,MAAM;CACN,QAAQ,SAAS;AACf,MAAI,CAAC,QAAQ,QAAQ,OACnB,QAAO,EAAE;EAGX,MAAM,sCAAsB,IAAI,KAK7B;AAEH,OAAK,MAAM,WAAW,QAAQ,kBAAkB;GAC9C,MAAM,YAAY,QAAQ,QAAQ,OAAO;GACzC,MAAM,YAAY,SAAS,WAAW,UAAU;AAChD,OAAI,CAAC,UACH;AAGF,QAAK,MAAM,CAAC,YAAY,iBAAiB,OAAO,QAAQ,UAAU,EAAE;IAClE,MAAM,UAAU,SAAS,aAAa;AACtC,QAAI,CAAC,QACH;IAGF,MAAM,uBAAuB,4CAA4C,SAAS,QAAQ;IAC1F,MAAM,eAAe,OAAO,QAAQ,SAAS,WAAW,QAAQ,OAAO;IACvE,MAAM,mBAAmB,QAAQ,SAAS,sBACxC,aACA,YACA,KACE,QAAQ,MAAM,WACd,cACA,uBAAuB,YAAY,aAAa,CACjD,CACF;IACD,MAAM,4BAA4B,iCAChC,SACA,WACA,kBACA,CAAC,OAAO,QAAQ,WAAW,WAAW,QAAQ,SAAS,OAAU,CAClE;AAED,wBAAoB,IAAI,YAAY;KAClC,IAAI;KACJ,UAAU;KACV,SAAS;MACP;MACA,GAAG;MACH,GAAI,sBAAsB,sBAAsB,EAC9C,oBAAoB,qBAAqB,oBAC1C;MACD,GAAI,sBAAsB,0BAA0B,EAClD,wBAAwB,qBAAqB,wBAC9C;MACD,GAAI,6BAA6B;OAC/B,iBAAiB,0BAA0B;OAC3C,wBAAwB,0BAA0B;OACnD;MACD,GAAI,2BAA2B,iCAAiC,EAC9D,+BACE,0BAA0B,+BAC7B;MACF;KACF,CAAC;;;AAIN,SAAO,CAAC,GAAG,oBAAoB,QAAQ,CAAC;;CAE1C,UAAU;CACX"}
|