@inkeep/agents-cli 0.59.2 → 0.59.4

package/dist/commands/pull-v4/generators/agent-generator.helpers.js

@@ -1,14 +1,8 @@
-import { collectTemplateVariableNames, createUniqueReferenceName, isPlainObject, toCamelCase, toKebabCase, toTriggerReferenceName } from "../utils.js";
+import { isPlainObject } from "../utils/shared.js";
+import { createUniqueReferenceName, toCamelCase, toKebabCase, toTriggerReferenceName } from "../utils/naming.js";
+import "../utils/index.js";
 
 //#region src/commands/pull-v4/generators/agent-generator.helpers.ts
-function collectTemplateVariableNamesFromFields(values) {
-	const variables = [];
-	for (const value of values) {
-		if (typeof value !== "string") continue;
-		variables.push(...collectTemplateVariableNames(value));
-	}
-	return variables;
-}
 function extractIds(value) {
 	if (Array.isArray(value)) return value.map((item) => {
 		if (typeof item === "string") return item;
@@ -182,5 +176,5 @@ function createNumericReferenceName(baseName, reservedNames) {
 }
 
 //#endregion
-export { addScheduledTriggerImports, addStatusComponentImports, addSubAgentImports, addTriggerImports, collectTemplateVariableNamesFromFields, createReferenceNameMap, createScheduledTriggerReferenceMaps, createSubAgentReferenceMaps, createTriggerReferenceMaps, extractContextConfigId, extractIds, extractStatusComponentIds, resolveStatusComponentId };
+export { addScheduledTriggerImports, addStatusComponentImports, addSubAgentImports, addTriggerImports, createReferenceNameMap, createScheduledTriggerReferenceMaps, createSubAgentReferenceMaps, createTriggerReferenceMaps, extractContextConfigId, extractIds, extractStatusComponentIds, resolveStatusComponentId };
 //# sourceMappingURL=agent-generator.helpers.js.map

package/dist/commands/pull-v4/generators/agent-generator.helpers.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-generator.helpers.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/agent-generator.helpers.ts"],"sourcesContent":["import type { SourceFile } from 'ts-morph';\nimport {\n  collectTemplateVariableNames,\n  createUniqueReferenceName,\n  isPlainObject,\n  toCamelCase,\n  toKebabCase,\n  toTriggerReferenceName,\n} from '../utils';\n\nexport type ReferenceNameMap = Map<string, string>;\nexport type TriggerImportMap = Map<string, { importName: string; modulePath: string }>;\n\ninterface SubAgentReferenceOverride {\n  name: string;\n  local?: boolean;\n}\n\ntype StatusComponentLike = string | { id?: string; type?: string; name?: string };\ntype StatusUpdatesLike = { statusComponents?: StatusComponentLike[] } | undefined;\n\nexport function collectTemplateVariableNamesFromFields(\n  values: Array<string | undefined>\n): string[] {\n  const variables: string[] = [];\n  for (const value of values) {\n    if (typeof value !== 'string') {\n      continue;\n    }\n    variables.push(...collectTemplateVariableNames(value));\n  }\n  return variables;\n}\n\nexport function extractIds(value: unknown[] | Record<string, unknown>): string[] {\n  if (Array.isArray(value)) {\n    return value\n      .map((item) => {\n        if (typeof item === 'string') {\n          return item;\n        }\n        if (isPlainObject(item) && typeof item.id === 'string') {\n          return item.id;\n        }\n        return null;\n      })\n      .filter((id): id is string => Boolean(id));\n  }\n  return Object.keys(value);\n}\n\nexport function extractContextConfigId(\n  contextConfig?: string | { id?: string }\n): string | undefined {\n  if (!contextConfig) {\n    return;\n  }\n  if (typeof contextConfig === 'string') {\n    return contextConfig;\n  }\n  return contextConfig.id;\n}\n\nexport function addSubAgentImports(\n  sourceFile: SourceFile,\n  referenceNames: ReferenceNameMap,\n  importNames: ReferenceNameMap,\n  modulePathOverrides?: Record<string, string>\n): void {\n  for (const [subAgentId, referenceName] of referenceNames) {\n    const importName = importNames.get(subAgentId);\n    if (!importName) {\n      continue;\n    }\n\n    sourceFile.addImportDeclaration({\n      namedImports: [\n        importName === referenceName ? importName : { name: importName, alias: referenceName },\n      ],\n      moduleSpecifier: `./sub-agents/${modulePathOverrides?.[subAgentId] ?? subAgentId}`,\n    });\n  }\n}\n\nexport function addTriggerImports(\n  sourceFile: SourceFile,\n  referenceNames: ReferenceNameMap,\n  importRefs: TriggerImportMap\n): void {\n  for (const [triggerId, referenceName] of referenceNames) {\n    const importRef = importRefs.get(triggerId);\n    if (!importRef) {\n      continue;\n    }\n\n    const { importName, modulePath } = importRef;\n    sourceFile.addImportDeclaration({\n      namedImports: [\n        importName === referenceName ? importName : { name: importName, alias: referenceName },\n      ],\n      moduleSpecifier: `./triggers/${modulePath}`,\n    });\n  }\n}\n\nexport function addScheduledTriggerImports(\n  sourceFile: SourceFile,\n  referenceNames: ReferenceNameMap,\n  importRefs: TriggerImportMap\n): void {\n  for (const [scheduledTriggerId, referenceName] of referenceNames) {\n    const importRef = importRefs.get(scheduledTriggerId);\n    if (!importRef) {\n      continue;\n    }\n\n    const { importName, modulePath } = importRef;\n    sourceFile.addImportDeclaration({\n      namedImports: [\n        importName === referenceName ? importName : { name: importName, alias: referenceName },\n      ],\n      moduleSpecifier: `./scheduled-triggers/${modulePath}`,\n    });\n  }\n}\n\nexport function extractStatusComponentIds(statusUpdates: StatusUpdatesLike): string[] {\n  if (!statusUpdates?.statusComponents?.length) {\n    return [];\n  }\n\n  const statusComponentIds = statusUpdates.statusComponents.map(resolveStatusComponentId);\n  return [...new Set(statusComponentIds)];\n}\n\nexport function resolveStatusComponentId(statusComponent: StatusComponentLike): string {\n  const id =\n    typeof statusComponent === 'string'\n      ? statusComponent\n      : statusComponent.id || statusComponent.type;\n  if (!id) {\n    throw new Error(\n      `Unable to resolve status component with id ${JSON.stringify(statusComponent)}`\n    );\n  }\n  return id;\n}\n\nexport function addStatusComponentImports(\n  sourceFile: SourceFile,\n  referenceNames: ReferenceNameMap\n): void {\n  for (const [statusComponentId, referenceName] of referenceNames) {\n    const importName = toCamelCase(statusComponentId);\n    sourceFile.addImportDeclaration({\n      namedImports: [\n        importName === referenceName ? importName : { name: importName, alias: referenceName },\n      ],\n      moduleSpecifier: `../status-components/${statusComponentId}`,\n    });\n  }\n}\n\nexport function createSubAgentReferenceMaps(\n  ids: Iterable<string>,\n  reservedNames: Set<string>,\n  conflictSuffix: string,\n  overrides?: Record<string, SubAgentReferenceOverride>\n): {\n  referenceNames: ReferenceNameMap;\n  importNames: ReferenceNameMap;\n} {\n  const referenceNames: ReferenceNameMap = new Map();\n  const importNames: ReferenceNameMap = new Map();\n\n  for (const id of ids) {\n    if (referenceNames.has(id)) {\n      continue;\n    }\n\n    const override = overrides?.[id];\n    const importName = override?.name ?? toCamelCase(id);\n    const isLocal = override?.local === true;\n    const referenceName = isLocal\n      ? importName\n      : createUniqueReferenceName(importName, reservedNames, conflictSuffix);\n\n    if (isLocal) {\n      reservedNames.add(referenceName);\n    } else {\n      importNames.set(id, importName);\n    }\n\n    referenceNames.set(id, referenceName);\n  }\n\n  return { referenceNames, importNames };\n}\n\nexport function createReferenceNameMap(\n  ids: Iterable<string>,\n  reservedNames: Set<string>,\n  conflictSuffix: string\n): ReferenceNameMap {\n  const map: ReferenceNameMap = new Map();\n  for (const id of ids) {\n    if (map.has(id)) {\n      continue;\n    }\n    map.set(id, createUniqueReferenceName(toCamelCase(id), reservedNames, conflictSuffix));\n  }\n  return map;\n}\n\nexport function createScheduledTriggerReferenceMaps(\n  scheduledTriggers: unknown,\n  reservedNames: Set<string>\n): {\n  referenceNames: ReferenceNameMap;\n  importRefs: TriggerImportMap;\n} {\n  const referenceNames: ReferenceNameMap = new Map();\n  const importRefs: TriggerImportMap = new Map();\n\n  if (!scheduledTriggers || !isPlainObject(scheduledTriggers)) {\n    return { referenceNames, importRefs };\n  }\n\n  const moduleNameCounts = new Map<string, number>();\n\n  for (const [scheduledTriggerId, scheduledTriggerData] of Object.entries(scheduledTriggers)) {\n    if (referenceNames.has(scheduledTriggerId)) {\n      continue;\n    }\n\n    const scheduledTriggerRecord = isPlainObject(scheduledTriggerData)\n      ? scheduledTriggerData\n      : undefined;\n    const scheduledTriggerName =\n      typeof scheduledTriggerRecord?.name === 'string' && scheduledTriggerRecord.name.length > 0\n        ? scheduledTriggerRecord.name\n        : scheduledTriggerId;\n\n    const importName = toTriggerReferenceName(scheduledTriggerName);\n    const referenceName = createNumericReferenceName(importName, reservedNames);\n\n    const baseModuleName =\n      toKebabCase(scheduledTriggerName) || toKebabCase(scheduledTriggerId) || scheduledTriggerId;\n    const moduleCount = moduleNameCounts.get(baseModuleName) ?? 0;\n    moduleNameCounts.set(baseModuleName, moduleCount + 1);\n    const modulePath = moduleCount === 0 ? baseModuleName : `${baseModuleName}-${moduleCount}`;\n\n    referenceNames.set(scheduledTriggerId, referenceName);\n    importRefs.set(scheduledTriggerId, { importName, modulePath });\n  }\n\n  return { referenceNames, importRefs };\n}\n\nexport function createTriggerReferenceMaps(\n  triggers: unknown,\n  reservedNames: Set<string>\n): {\n  referenceNames: ReferenceNameMap;\n  importRefs: TriggerImportMap;\n} {\n  const referenceNames: ReferenceNameMap = new Map();\n  const importRefs: TriggerImportMap = new Map();\n\n  if (!triggers || !isPlainObject(triggers)) {\n    return { referenceNames, importRefs };\n  }\n\n  const moduleNameCounts = new Map<string, number>();\n\n  for (const [triggerId, triggerData] of Object.entries(triggers)) {\n    if (referenceNames.has(triggerId)) {\n      continue;\n    }\n\n    const triggerRecord = isPlainObject(triggerData) ? triggerData : undefined;\n    const triggerName =\n      typeof triggerRecord?.name === 'string' && triggerRecord.name.length > 0\n        ? triggerRecord.name\n        : triggerId;\n\n    const importName = toTriggerReferenceName(triggerName);\n    const referenceName = createNumericReferenceName(importName, reservedNames);\n\n    const baseModuleName = toKebabCase(triggerName) || toKebabCase(triggerId) || triggerId;\n    const moduleCount = moduleNameCounts.get(baseModuleName) ?? 0;\n    moduleNameCounts.set(baseModuleName, moduleCount + 1);\n    const modulePath = moduleCount === 0 ? baseModuleName : `${baseModuleName}-${moduleCount}`;\n\n    referenceNames.set(triggerId, referenceName);\n    importRefs.set(triggerId, { importName, modulePath });\n  }\n\n  return { referenceNames, importRefs };\n}\n\nfunction createNumericReferenceName(baseName: string, reservedNames: Set<string>): string {\n  if (!reservedNames.has(baseName)) {\n    reservedNames.add(baseName);\n    return baseName;\n  }\n\n  let index = 1;\n  while (reservedNames.has(`${baseName}${index}`)) {\n    index += 1;\n  }\n\n  const uniqueName = `${baseName}${index}`;\n  reservedNames.add(uniqueName);\n  return uniqueName;\n}\n"],"mappings":";;;AAqBA,SAAgB,uCACd,QACU;CACV,MAAM,YAAsB,EAAE;AAC9B,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,OAAO,UAAU,SACnB;AAEF,YAAU,KAAK,GAAG,6BAA6B,MAAM,CAAC;;AAExD,QAAO;;AAGT,SAAgB,WAAW,OAAsD;AAC/E,KAAI,MAAM,QAAQ,MAAM,CACtB,QAAO,MACJ,KAAK,SAAS;AACb,MAAI,OAAO,SAAS,SAClB,QAAO;AAET,MAAI,cAAc,KAAK,IAAI,OAAO,KAAK,OAAO,SAC5C,QAAO,KAAK;AAEd,SAAO;GACP,CACD,QAAQ,OAAqB,QAAQ,GAAG,CAAC;AAE9C,QAAO,OAAO,KAAK,MAAM;;AAG3B,SAAgB,uBACd,eACoB;AACpB,KAAI,CAAC,cACH;AAEF,KAAI,OAAO,kBAAkB,SAC3B,QAAO;AAET,QAAO,cAAc;;AAGvB,SAAgB,mBACd,YACA,gBACA,aACA,qBACM;AACN,MAAK,MAAM,CAAC,YAAY,kBAAkB,gBAAgB;EACxD,MAAM,aAAa,YAAY,IAAI,WAAW;AAC9C,MAAI,CAAC,WACH;AAGF,aAAW,qBAAqB;GAC9B,cAAc,CACZ,eAAe,gBAAgB,aAAa;IAAE,MAAM;IAAY,OAAO;IAAe,CACvF;GACD,iBAAiB,gBAAgB,sBAAsB,eAAe;GACvE,CAAC;;;AAIN,SAAgB,kBACd,YACA,gBACA,YACM;AACN,MAAK,MAAM,CAAC,WAAW,kBAAkB,gBAAgB;EACvD,MAAM,YAAY,WAAW,IAAI,UAAU;AAC3C,MAAI,CAAC,UACH;EAGF,MAAM,EAAE,YAAY,eAAe;AACnC,aAAW,qBAAqB;GAC9B,cAAc,CACZ,eAAe,gBAAgB,aAAa;IAAE,MAAM;IAAY,OAAO;IAAe,CACvF;GACD,iBAAiB,cAAc;GAChC,CAAC;;;AAIN,SAAgB,2BACd,YACA,gBACA,YACM;AACN,MAAK,MAAM,CAAC,oBAAoB,kBAAkB,gBAAgB;EAChE,MAAM,YAAY,WAAW,IAAI,mBAAmB;AACpD,MAAI,CAAC,UACH;EAGF,MAAM,EAAE,YAAY,eAAe;AACnC,aAAW,qBAAqB;GAC9B,cAAc,CACZ,eAAe,gBAAgB,aAAa;IAAE,MAAM;IAAY,OAAO;IAAe,CACvF;GACD,iBAAiB,wBAAwB;GAC1C,CAAC;;;AAIN,SAAgB,0BAA0B,eAA4C;AACpF,KAAI,CAAC,eAAe,kBAAkB,OACpC,QAAO,EAAE;CAGX,MAAM,qBAAqB,cAAc,iBAAiB,IAAI,yBAAyB;AACvF,QAAO,CAAC,GAAG,IAAI,IAAI,mBAAmB,CAAC;;AAGzC,SAAgB,yBAAyB,iBAA8C;CACrF,MAAM,KACJ,OAAO,oBAAoB,WACvB,kBACA,gBAAgB,MAAM,gBAAgB;AAC5C,KAAI,CAAC,GACH,OAAM,IAAI,MACR,8CAA8C,KAAK,UAAU,gBAAgB,GAC9E;AAEH,QAAO;;AAGT,SAAgB,0BACd,YACA,gBACM;AACN,MAAK,MAAM,CAAC,mBAAmB,kBAAkB,gBAAgB;EAC/D,MAAM,aAAa,YAAY,kBAAkB;AACjD,aAAW,qBAAqB;GAC9B,cAAc,CACZ,eAAe,gBAAgB,aAAa;IAAE,MAAM;IAAY,OAAO;IAAe,CACvF;GACD,iBAAiB,wBAAwB;GAC1C,CAAC;;;AAIN,SAAgB,4BACd,KACA,eACA,gBACA,WAIA;CACA,MAAM,iCAAmC,IAAI,KAAK;CAClD,MAAM,8BAAgC,IAAI,KAAK;AAE/C,MAAK,MAAM,MAAM,KAAK;AACpB,MAAI,eAAe,IAAI,GAAG,CACxB;EAGF,MAAM,WAAW,YAAY;EAC7B,MAAM,aAAa,UAAU,QAAQ,YAAY,GAAG;EACpD,MAAM,UAAU,UAAU,UAAU;EACpC,MAAM,gBAAgB,UAClB,aACA,0BAA0B,YAAY,eAAe,eAAe;AAExE,MAAI,QACF,eAAc,IAAI,cAAc;MAEhC,aAAY,IAAI,IAAI,WAAW;AAGjC,iBAAe,IAAI,IAAI,cAAc;;AAGvC,QAAO;EAAE;EAAgB;EAAa;;AAGxC,SAAgB,uBACd,KACA,eACA,gBACkB;CAClB,MAAM,sBAAwB,IAAI,KAAK;AACvC,MAAK,MAAM,MAAM,KAAK;AACpB,MAAI,IAAI,IAAI,GAAG,CACb;AAEF,MAAI,IAAI,IAAI,0BAA0B,YAAY,GAAG,EAAE,eAAe,eAAe,CAAC;;AAExF,QAAO;;AAGT,SAAgB,oCACd,mBACA,eAIA;CACA,MAAM,iCAAmC,IAAI,KAAK;CAClD,MAAM,6BAA+B,IAAI,KAAK;AAE9C,KAAI,CAAC,qBAAqB,CAAC,cAAc,kBAAkB,CACzD,QAAO;EAAE;EAAgB;EAAY;CAGvC,MAAM,mCAAmB,IAAI,KAAqB;AAElD,MAAK,MAAM,CAAC,oBAAoB,yBAAyB,OAAO,QAAQ,kBAAkB,EAAE;AAC1F,MAAI,eAAe,IAAI,mBAAmB,CACxC;EAGF,MAAM,yBAAyB,cAAc,qBAAqB,GAC9D,uBACA;EACJ,MAAM,uBACJ,OAAO,wBAAwB,SAAS,YAAY,uBAAuB,KAAK,SAAS,IACrF,uBAAuB,OACvB;EAEN,MAAM,aAAa,uBAAuB,qBAAqB;EAC/D,MAAM,gBAAgB,2BAA2B,YAAY,cAAc;EAE3E,MAAM,iBACJ,YAAY,qBAAqB,IAAI,YAAY,mBAAmB,IAAI;EAC1E,MAAM,cAAc,iBAAiB,IAAI,eAAe,IAAI;AAC5D,mBAAiB,IAAI,gBAAgB,cAAc,EAAE;EACrD,MAAM,aAAa,gBAAgB,IAAI,iBAAiB,GAAG,eAAe,GAAG;AAE7E,iBAAe,IAAI,oBAAoB,cAAc;AACrD,aAAW,IAAI,oBAAoB;GAAE;GAAY;GAAY,CAAC;;AAGhE,QAAO;EAAE;EAAgB;EAAY;;AAGvC,SAAgB,2BACd,UACA,eAIA;CACA,MAAM,iCAAmC,IAAI,KAAK;CAClD,MAAM,6BAA+B,IAAI,KAAK;AAE9C,KAAI,CAAC,YAAY,CAAC,cAAc,SAAS,CACvC,QAAO;EAAE;EAAgB;EAAY;CAGvC,MAAM,mCAAmB,IAAI,KAAqB;AAElD,MAAK,MAAM,CAAC,WAAW,gBAAgB,OAAO,QAAQ,SAAS,EAAE;AAC/D,MAAI,eAAe,IAAI,UAAU,CAC/B;EAGF,MAAM,gBAAgB,cAAc,YAAY,GAAG,cAAc;EACjE,MAAM,cACJ,OAAO,eAAe,SAAS,YAAY,cAAc,KAAK,SAAS,IACnE,cAAc,OACd;EAEN,MAAM,aAAa,uBAAuB,YAAY;EACtD,MAAM,gBAAgB,2BAA2B,YAAY,cAAc;EAE3E,MAAM,iBAAiB,YAAY,YAAY,IAAI,YAAY,UAAU,IAAI;EAC7E,MAAM,cAAc,iBAAiB,IAAI,eAAe,IAAI;AAC5D,mBAAiB,IAAI,gBAAgB,cAAc,EAAE;EACrD,MAAM,aAAa,gBAAgB,IAAI,iBAAiB,GAAG,eAAe,GAAG;AAE7E,iBAAe,IAAI,WAAW,cAAc;AAC5C,aAAW,IAAI,WAAW;GAAE;GAAY;GAAY,CAAC;;AAGvD,QAAO;EAAE;EAAgB;EAAY;;AAGvC,SAAS,2BAA2B,UAAkB,eAAoC;AACxF,KAAI,CAAC,cAAc,IAAI,SAAS,EAAE;AAChC,gBAAc,IAAI,SAAS;AAC3B,SAAO;;CAGT,IAAI,QAAQ;AACZ,QAAO,cAAc,IAAI,GAAG,WAAW,QAAQ,CAC7C,UAAS;CAGX,MAAM,aAAa,GAAG,WAAW;AACjC,eAAc,IAAI,WAAW;AAC7B,QAAO"}
+{"version":3,"file":"agent-generator.helpers.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/agent-generator.helpers.ts"],"sourcesContent":["import type { SourceFile } from 'ts-morph';\nimport {\n  createUniqueReferenceName,\n  isPlainObject,\n  toCamelCase,\n  toKebabCase,\n  toTriggerReferenceName,\n} from '../utils';\n\nexport type ReferenceNameMap = Map<string, string>;\nexport type TriggerImportMap = Map<string, { importName: string; modulePath: string }>;\n\ninterface SubAgentReferenceOverride {\n  name: string;\n  local?: boolean;\n}\n\ntype StatusComponentLike = string | { id?: string; type?: string; name?: string };\ntype StatusUpdatesLike = { statusComponents?: StatusComponentLike[] } | undefined;\n\nexport function extractIds(value: unknown[] | Record<string, unknown>): string[] {\n  if (Array.isArray(value)) {\n    return value\n      .map((item) => {\n        if (typeof item === 'string') {\n          return item;\n        }\n        if (isPlainObject(item) && typeof item.id === 'string') {\n          return item.id;\n        }\n        return null;\n      })\n      .filter((id): id is string => Boolean(id));\n  }\n  return Object.keys(value);\n}\n\nexport function extractContextConfigId(\n  contextConfig?: string | { id?: string }\n): string | undefined {\n  if (!contextConfig) {\n    return;\n  }\n  if (typeof contextConfig === 'string') {\n    return contextConfig;\n  }\n  return contextConfig.id;\n}\n\nexport function addSubAgentImports(\n  sourceFile: SourceFile,\n  referenceNames: ReferenceNameMap,\n  importNames: ReferenceNameMap,\n  modulePathOverrides?: Record<string, string>\n): void {\n  for (const [subAgentId, referenceName] of referenceNames) {\n    const importName = importNames.get(subAgentId);\n    if (!importName) {\n      continue;\n    }\n\n    sourceFile.addImportDeclaration({\n      namedImports: [\n        importName === referenceName ? importName : { name: importName, alias: referenceName },\n      ],\n      moduleSpecifier: `./sub-agents/${modulePathOverrides?.[subAgentId] ?? subAgentId}`,\n    });\n  }\n}\n\nexport function addTriggerImports(\n  sourceFile: SourceFile,\n  referenceNames: ReferenceNameMap,\n  importRefs: TriggerImportMap\n): void {\n  for (const [triggerId, referenceName] of referenceNames) {\n    const importRef = importRefs.get(triggerId);\n    if (!importRef) {\n      continue;\n    }\n\n    const { importName, modulePath } = importRef;\n    sourceFile.addImportDeclaration({\n      namedImports: [\n        importName === referenceName ? importName : { name: importName, alias: referenceName },\n      ],\n      moduleSpecifier: `./triggers/${modulePath}`,\n    });\n  }\n}\n\nexport function addScheduledTriggerImports(\n  sourceFile: SourceFile,\n  referenceNames: ReferenceNameMap,\n  importRefs: TriggerImportMap\n): void {\n  for (const [scheduledTriggerId, referenceName] of referenceNames) {\n    const importRef = importRefs.get(scheduledTriggerId);\n    if (!importRef) {\n      continue;\n    }\n\n    const { importName, modulePath } = importRef;\n    sourceFile.addImportDeclaration({\n      namedImports: [\n        importName === referenceName ? importName : { name: importName, alias: referenceName },\n      ],\n      moduleSpecifier: `./scheduled-triggers/${modulePath}`,\n    });\n  }\n}\n\nexport function extractStatusComponentIds(statusUpdates: StatusUpdatesLike): string[] {\n  if (!statusUpdates?.statusComponents?.length) {\n    return [];\n  }\n\n  const statusComponentIds = statusUpdates.statusComponents.map(resolveStatusComponentId);\n  return [...new Set(statusComponentIds)];\n}\n\nexport function resolveStatusComponentId(statusComponent: StatusComponentLike): string {\n  const id =\n    typeof statusComponent === 'string'\n      ? statusComponent\n      : statusComponent.id || statusComponent.type;\n  if (!id) {\n    throw new Error(\n      `Unable to resolve status component with id ${JSON.stringify(statusComponent)}`\n    );\n  }\n  return id;\n}\n\nexport function addStatusComponentImports(\n  sourceFile: SourceFile,\n  referenceNames: ReferenceNameMap\n): void {\n  for (const [statusComponentId, referenceName] of referenceNames) {\n    const importName = toCamelCase(statusComponentId);\n    sourceFile.addImportDeclaration({\n      namedImports: [\n        importName === referenceName ? importName : { name: importName, alias: referenceName },\n      ],\n      moduleSpecifier: `../status-components/${statusComponentId}`,\n    });\n  }\n}\n\nexport function createSubAgentReferenceMaps(\n  ids: Iterable<string>,\n  reservedNames: Set<string>,\n  conflictSuffix: string,\n  overrides?: Record<string, SubAgentReferenceOverride>\n): {\n  referenceNames: ReferenceNameMap;\n  importNames: ReferenceNameMap;\n} {\n  const referenceNames: ReferenceNameMap = new Map();\n  const importNames: ReferenceNameMap = new Map();\n\n  for (const id of ids) {\n    if (referenceNames.has(id)) {\n      continue;\n    }\n\n    const override = overrides?.[id];\n    const importName = override?.name ?? toCamelCase(id);\n    const isLocal = override?.local === true;\n    const referenceName = isLocal\n      ? importName\n      : createUniqueReferenceName(importName, reservedNames, conflictSuffix);\n\n    if (isLocal) {\n      reservedNames.add(referenceName);\n    } else {\n      importNames.set(id, importName);\n    }\n\n    referenceNames.set(id, referenceName);\n  }\n\n  return { referenceNames, importNames };\n}\n\nexport function createReferenceNameMap(\n  ids: Iterable<string>,\n  reservedNames: Set<string>,\n  conflictSuffix: string\n): ReferenceNameMap {\n  const map: ReferenceNameMap = new Map();\n  for (const id of ids) {\n    if (map.has(id)) {\n      continue;\n    }\n    map.set(id, createUniqueReferenceName(toCamelCase(id), reservedNames, conflictSuffix));\n  }\n  return map;\n}\n\nexport function createScheduledTriggerReferenceMaps(\n  scheduledTriggers: unknown,\n  reservedNames: Set<string>\n): {\n  referenceNames: ReferenceNameMap;\n  importRefs: TriggerImportMap;\n} {\n  const referenceNames: ReferenceNameMap = new Map();\n  const importRefs: TriggerImportMap = new Map();\n\n  if (!scheduledTriggers || !isPlainObject(scheduledTriggers)) {\n    return { referenceNames, importRefs };\n  }\n\n  const moduleNameCounts = new Map<string, number>();\n\n  for (const [scheduledTriggerId, scheduledTriggerData] of Object.entries(scheduledTriggers)) {\n    if (referenceNames.has(scheduledTriggerId)) {\n      continue;\n    }\n\n    const scheduledTriggerRecord = isPlainObject(scheduledTriggerData)\n      ? scheduledTriggerData\n      : undefined;\n    const scheduledTriggerName =\n      typeof scheduledTriggerRecord?.name === 'string' && scheduledTriggerRecord.name.length > 0\n        ? scheduledTriggerRecord.name\n        : scheduledTriggerId;\n\n    const importName = toTriggerReferenceName(scheduledTriggerName);\n    const referenceName = createNumericReferenceName(importName, reservedNames);\n\n    const baseModuleName =\n      toKebabCase(scheduledTriggerName) || toKebabCase(scheduledTriggerId) || scheduledTriggerId;\n    const moduleCount = moduleNameCounts.get(baseModuleName) ?? 0;\n    moduleNameCounts.set(baseModuleName, moduleCount + 1);\n    const modulePath = moduleCount === 0 ? baseModuleName : `${baseModuleName}-${moduleCount}`;\n\n    referenceNames.set(scheduledTriggerId, referenceName);\n    importRefs.set(scheduledTriggerId, { importName, modulePath });\n  }\n\n  return { referenceNames, importRefs };\n}\n\nexport function createTriggerReferenceMaps(\n  triggers: unknown,\n  reservedNames: Set<string>\n): {\n  referenceNames: ReferenceNameMap;\n  importRefs: TriggerImportMap;\n} {\n  const referenceNames: ReferenceNameMap = new Map();\n  const importRefs: TriggerImportMap = new Map();\n\n  if (!triggers || !isPlainObject(triggers)) {\n    return { referenceNames, importRefs };\n  }\n\n  const moduleNameCounts = new Map<string, number>();\n\n  for (const [triggerId, triggerData] of Object.entries(triggers)) {\n    if (referenceNames.has(triggerId)) {\n      continue;\n    }\n\n    const triggerRecord = isPlainObject(triggerData) ? triggerData : undefined;\n    const triggerName =\n      typeof triggerRecord?.name === 'string' && triggerRecord.name.length > 0\n        ? triggerRecord.name\n        : triggerId;\n\n    const importName = toTriggerReferenceName(triggerName);\n    const referenceName = createNumericReferenceName(importName, reservedNames);\n\n    const baseModuleName = toKebabCase(triggerName) || toKebabCase(triggerId) || triggerId;\n    const moduleCount = moduleNameCounts.get(baseModuleName) ?? 0;\n    moduleNameCounts.set(baseModuleName, moduleCount + 1);\n    const modulePath = moduleCount === 0 ? baseModuleName : `${baseModuleName}-${moduleCount}`;\n\n    referenceNames.set(triggerId, referenceName);\n    importRefs.set(triggerId, { importName, modulePath });\n  }\n\n  return { referenceNames, importRefs };\n}\n\nfunction createNumericReferenceName(baseName: string, reservedNames: Set<string>): string {\n  if (!reservedNames.has(baseName)) {\n    reservedNames.add(baseName);\n    return baseName;\n  }\n\n  let index = 1;\n  while (reservedNames.has(`${baseName}${index}`)) {\n    index += 1;\n  }\n\n  const uniqueName = `${baseName}${index}`;\n  reservedNames.add(uniqueName);\n  return uniqueName;\n}\n"],"mappings":";;;;;AAoBA,SAAgB,WAAW,OAAsD;AAC/E,KAAI,MAAM,QAAQ,MAAM,CACtB,QAAO,MACJ,KAAK,SAAS;AACb,MAAI,OAAO,SAAS,SAClB,QAAO;AAET,MAAI,cAAc,KAAK,IAAI,OAAO,KAAK,OAAO,SAC5C,QAAO,KAAK;AAEd,SAAO;GACP,CACD,QAAQ,OAAqB,QAAQ,GAAG,CAAC;AAE9C,QAAO,OAAO,KAAK,MAAM;;AAG3B,SAAgB,uBACd,eACoB;AACpB,KAAI,CAAC,cACH;AAEF,KAAI,OAAO,kBAAkB,SAC3B,QAAO;AAET,QAAO,cAAc;;AAGvB,SAAgB,mBACd,YACA,gBACA,aACA,qBACM;AACN,MAAK,MAAM,CAAC,YAAY,kBAAkB,gBAAgB;EACxD,MAAM,aAAa,YAAY,IAAI,WAAW;AAC9C,MAAI,CAAC,WACH;AAGF,aAAW,qBAAqB;GAC9B,cAAc,CACZ,eAAe,gBAAgB,aAAa;IAAE,MAAM;IAAY,OAAO;IAAe,CACvF;GACD,iBAAiB,gBAAgB,sBAAsB,eAAe;GACvE,CAAC;;;AAIN,SAAgB,kBACd,YACA,gBACA,YACM;AACN,MAAK,MAAM,CAAC,WAAW,kBAAkB,gBAAgB;EACvD,MAAM,YAAY,WAAW,IAAI,UAAU;AAC3C,MAAI,CAAC,UACH;EAGF,MAAM,EAAE,YAAY,eAAe;AACnC,aAAW,qBAAqB;GAC9B,cAAc,CACZ,eAAe,gBAAgB,aAAa;IAAE,MAAM;IAAY,OAAO;IAAe,CACvF;GACD,iBAAiB,cAAc;GAChC,CAAC;;;AAIN,SAAgB,2BACd,YACA,gBACA,YACM;AACN,MAAK,MAAM,CAAC,oBAAoB,kBAAkB,gBAAgB;EAChE,MAAM,YAAY,WAAW,IAAI,mBAAmB;AACpD,MAAI,CAAC,UACH;EAGF,MAAM,EAAE,YAAY,eAAe;AACnC,aAAW,qBAAqB;GAC9B,cAAc,CACZ,eAAe,gBAAgB,aAAa;IAAE,MAAM;IAAY,OAAO;IAAe,CACvF;GACD,iBAAiB,wBAAwB;GAC1C,CAAC;;;AAIN,SAAgB,0BAA0B,eAA4C;AACpF,KAAI,CAAC,eAAe,kBAAkB,OACpC,QAAO,EAAE;CAGX,MAAM,qBAAqB,cAAc,iBAAiB,IAAI,yBAAyB;AACvF,QAAO,CAAC,GAAG,IAAI,IAAI,mBAAmB,CAAC;;AAGzC,SAAgB,yBAAyB,iBAA8C;CACrF,MAAM,KACJ,OAAO,oBAAoB,WACvB,kBACA,gBAAgB,MAAM,gBAAgB;AAC5C,KAAI,CAAC,GACH,OAAM,IAAI,MACR,8CAA8C,KAAK,UAAU,gBAAgB,GAC9E;AAEH,QAAO;;AAGT,SAAgB,0BACd,YACA,gBACM;AACN,MAAK,MAAM,CAAC,mBAAmB,kBAAkB,gBAAgB;EAC/D,MAAM,aAAa,YAAY,kBAAkB;AACjD,aAAW,qBAAqB;GAC9B,cAAc,CACZ,eAAe,gBAAgB,aAAa;IAAE,MAAM;IAAY,OAAO;IAAe,CACvF;GACD,iBAAiB,wBAAwB;GAC1C,CAAC;;;AAIN,SAAgB,4BACd,KACA,eACA,gBACA,WAIA;CACA,MAAM,iCAAmC,IAAI,KAAK;CAClD,MAAM,8BAAgC,IAAI,KAAK;AAE/C,MAAK,MAAM,MAAM,KAAK;AACpB,MAAI,eAAe,IAAI,GAAG,CACxB;EAGF,MAAM,WAAW,YAAY;EAC7B,MAAM,aAAa,UAAU,QAAQ,YAAY,GAAG;EACpD,MAAM,UAAU,UAAU,UAAU;EACpC,MAAM,gBAAgB,UAClB,aACA,0BAA0B,YAAY,eAAe,eAAe;AAExE,MAAI,QACF,eAAc,IAAI,cAAc;MAEhC,aAAY,IAAI,IAAI,WAAW;AAGjC,iBAAe,IAAI,IAAI,cAAc;;AAGvC,QAAO;EAAE;EAAgB;EAAa;;AAGxC,SAAgB,uBACd,KACA,eACA,gBACkB;CAClB,MAAM,sBAAwB,IAAI,KAAK;AACvC,MAAK,MAAM,MAAM,KAAK;AACpB,MAAI,IAAI,IAAI,GAAG,CACb;AAEF,MAAI,IAAI,IAAI,0BAA0B,YAAY,GAAG,EAAE,eAAe,eAAe,CAAC;;AAExF,QAAO;;AAGT,SAAgB,oCACd,mBACA,eAIA;CACA,MAAM,iCAAmC,IAAI,KAAK;CAClD,MAAM,6BAA+B,IAAI,KAAK;AAE9C,KAAI,CAAC,qBAAqB,CAAC,cAAc,kBAAkB,CACzD,QAAO;EAAE;EAAgB;EAAY;CAGvC,MAAM,mCAAmB,IAAI,KAAqB;AAElD,MAAK,MAAM,CAAC,oBAAoB,yBAAyB,OAAO,QAAQ,kBAAkB,EAAE;AAC1F,MAAI,eAAe,IAAI,mBAAmB,CACxC;EAGF,MAAM,yBAAyB,cAAc,qBAAqB,GAC9D,uBACA;EACJ,MAAM,uBACJ,OAAO,wBAAwB,SAAS,YAAY,uBAAuB,KAAK,SAAS,IACrF,uBAAuB,OACvB;EAEN,MAAM,aAAa,uBAAuB,qBAAqB;EAC/D,MAAM,gBAAgB,2BAA2B,YAAY,cAAc;EAE3E,MAAM,iBACJ,YAAY,qBAAqB,IAAI,YAAY,mBAAmB,IAAI;EAC1E,MAAM,cAAc,iBAAiB,IAAI,eAAe,IAAI;AAC5D,mBAAiB,IAAI,gBAAgB,cAAc,EAAE;EACrD,MAAM,aAAa,gBAAgB,IAAI,iBAAiB,GAAG,eAAe,GAAG;AAE7E,iBAAe,IAAI,oBAAoB,cAAc;AACrD,aAAW,IAAI,oBAAoB;GAAE;GAAY;GAAY,CAAC;;AAGhE,QAAO;EAAE;EAAgB;EAAY;;AAGvC,SAAgB,2BACd,UACA,eAIA;CACA,MAAM,iCAAmC,IAAI,KAAK;CAClD,MAAM,6BAA+B,IAAI,KAAK;AAE9C,KAAI,CAAC,YAAY,CAAC,cAAc,SAAS,CACvC,QAAO;EAAE;EAAgB;EAAY;CAGvC,MAAM,mCAAmB,IAAI,KAAqB;AAElD,MAAK,MAAM,CAAC,WAAW,gBAAgB,OAAO,QAAQ,SAAS,EAAE;AAC/D,MAAI,eAAe,IAAI,UAAU,CAC/B;EAGF,MAAM,gBAAgB,cAAc,YAAY,GAAG,cAAc;EACjE,MAAM,cACJ,OAAO,eAAe,SAAS,YAAY,cAAc,KAAK,SAAS,IACnE,cAAc,OACd;EAEN,MAAM,aAAa,uBAAuB,YAAY;EACtD,MAAM,gBAAgB,2BAA2B,YAAY,cAAc;EAE3E,MAAM,iBAAiB,YAAY,YAAY,IAAI,YAAY,UAAU,IAAI;EAC7E,MAAM,cAAc,iBAAiB,IAAI,eAAe,IAAI;AAC5D,mBAAiB,IAAI,gBAAgB,cAAc,EAAE;EACrD,MAAM,aAAa,gBAAgB,IAAI,iBAAiB,GAAG,eAAe,GAAG;AAE7E,iBAAe,IAAI,WAAW,cAAc;AAC5C,aAAW,IAAI,WAAW;GAAE;GAAY;GAAY,CAAC;;AAGvD,QAAO;EAAE;EAAgB;EAAY;;AAGvC,SAAS,2BAA2B,UAAkB,eAAoC;AACxF,KAAI,CAAC,cAAc,IAAI,SAAS,EAAE;AAChC,gBAAc,IAAI,SAAS;AAC3B,SAAO;;CAGT,IAAI,QAAQ;AACZ,QAAO,cAAc,IAAI,GAAG,WAAW,QAAQ,CAC7C,UAAS;CAGX,MAAM,aAAa,GAAG,WAAW;AACjC,eAAc,IAAI,WAAW;AAC7B,QAAO"}

package/dist/commands/pull-v4/generators/agent-generator.js

@@ -1,8 +1,17 @@
-import { addReferenceGetterProperty, addValueToObject, createFactoryDefinition, formatTemplate, resolveContextTemplateImports, toCamelCase, toTriggerReferenceName } from "../utils.js";
-import { addScheduledTriggerImports, addStatusComponentImports, addSubAgentImports, addTriggerImports, collectTemplateVariableNamesFromFields, createReferenceNameMap, createScheduledTriggerReferenceMaps, createSubAgentReferenceMaps, createTriggerReferenceMaps, extractContextConfigId, extractIds, extractStatusComponentIds, resolveStatusComponentId } from "./agent-generator.helpers.js";
+import { isPlainObject } from "../utils/shared.js";
+import { formatTemplate } from "../utils/templates.js";
+import { codePropertyAccess, codeReference, createReferenceGetterValue } from "../utils/code-values.js";
+import { buildComponentFileName, toCamelCase, toTriggerReferenceName } from "../utils/naming.js";
+import { addValueToObject } from "../utils/factory-writer.js";
+import "../utils/index.js";
+import { asRecord, collectTemplateVariablesFromValues } from "../collector-common.js";
+import { collectContextTemplateReferences, collectSubAgentReferenceOverrides, collectSubAgentReferencePathOverrides } from "../collector-reference-helpers.js";
+import { addResolvedReferenceImports, resolveReferenceBinding, resolveReferenceBindingsFromIds, toReferenceNameMap } from "../reference-resolution.js";
+import { generateFactorySourceFile } from "../simple-factory-generator.js";
+import { addScheduledTriggerImports, addStatusComponentImports, addTriggerImports, createReferenceNameMap, createScheduledTriggerReferenceMaps, createTriggerReferenceMaps, extractIds } from "./helpers/agent.js";
 import { FullProjectDefinitionSchema } from "@inkeep/agents-core";
 import { z } from "zod";
-import { SyntaxKind } from "ts-morph";
+import { join } from "node:path";
 
 //#region src/commands/pull-v4/generators/agent-generator.ts
 const SubAgentReferenceSchema = z.object({
@@ -35,60 +44,101 @@ const AgentSchema = z.strictObject({
 	subAgentReferencePathOverrides: z.record(z.string(), z.string().nonempty()).optional(),
 	contextConfigReference: SubAgentReferenceSchema.optional(),
 	contextConfigHeadersReference: SubAgentReferenceSchema.optional()
-});
+}).transform((data) => ({
+	...data,
+	normalizedContextConfigId: normalizeContextConfigId(data.contextConfig),
+	normalizedStatusComponentIds: normalizeStatusComponentIds(data.statusUpdates?.statusComponents),
+	normalizedStatusComponentSequence: normalizeStatusComponentSequence(data.statusUpdates?.statusComponents)
+}));
+function normalizeContextConfigId(contextConfig) {
+	if (typeof contextConfig === "string") return contextConfig;
+	if (isPlainObject(contextConfig) && typeof contextConfig.id === "string") return contextConfig.id;
+}
+function normalizeStatusComponentId(statusComponent) {
+	if (typeof statusComponent === "string") return statusComponent;
+	if (!isPlainObject(statusComponent)) return;
+	if (typeof statusComponent.id === "string") return statusComponent.id;
+	if (typeof statusComponent.type === "string") return statusComponent.type;
+}
+function normalizeStatusComponentSequence(statusComponents) {
+	if (!Array.isArray(statusComponents)) return [];
+	return statusComponents.map((statusComponent) => normalizeStatusComponentId(statusComponent)).filter((statusComponentId) => Boolean(statusComponentId));
+}
+function normalizeStatusComponentIds(statusComponents) {
+	return [...new Set(normalizeStatusComponentSequence(statusComponents))];
+}
 function generateAgentDefinition({ id, createdAt, updatedAt, ...data }) {
-	const result = AgentSchema.safeParse(data);
-	if (!result.success) throw new Error(`Validation failed for agent:\n${z.prettifyError(result.error)}`);
-	const parsed = result.data;
-	const subAgentIds = new Set(extractIds(parsed.subAgents));
-	if (parsed.defaultSubAgentId) subAgentIds.add(parsed.defaultSubAgentId);
-	const agentVarName = parsed.agentVariableName || toCamelCase(parsed.agentId);
-	const { sourceFile, configObject } = createFactoryDefinition({
-		importName: "agent",
-		variableName: agentVarName
+	return generateFactorySourceFile(data, {
+		schema: AgentSchema,
+		factory: {
+			importName: "agent",
+			variableName: (parsed) => parsed.agentVariableName || toCamelCase(parsed.agentId)
+		},
+		render({ parsed, sourceFile, configObject }) {
+			const subAgentIds = new Set(extractIds(parsed.subAgents));
+			if (parsed.defaultSubAgentId) subAgentIds.add(parsed.defaultSubAgentId);
+			const agentVarName = parsed.agentVariableName || toCamelCase(parsed.agentId);
+			const reservedReferenceNames = new Set([agentVarName]);
+			const subAgentReferences = resolveReferenceBindingsFromIds({
+				ids: subAgentIds,
+				reservedNames: reservedReferenceNames,
+				conflictSuffix: "SubAgent",
+				referenceOverrides: parsed.subAgentReferences,
+				referencePathOverrides: parsed.subAgentReferencePathOverrides,
+				defaultModulePath: (id$1) => id$1
+			});
+			const subAgentReferenceNames = toReferenceNameMap(subAgentReferences);
+			addResolvedReferenceImports(sourceFile, subAgentReferences, (reference) => {
+				return `./sub-agents/${reference.modulePath}`;
+			});
+			const contextConfigId = parsed.normalizedContextConfigId;
+			let contextConfigReferenceName;
+			let contextHeadersReferenceName;
+			const hasHeadersTemplateVariables = collectTemplateVariablesFromValues([parsed.prompt, parsed.statusUpdates?.prompt]).some((variableName) => variableName.startsWith("headers."));
+			if (contextConfigId) {
+				const contextConfigReference = resolveReferenceBinding({
+					id: `${contextConfigId}:context`,
+					importName: parsed.contextConfigReference?.name ?? toCamelCase(contextConfigId),
+					modulePath: contextConfigId,
+					local: parsed.contextConfigReference?.local === true,
+					conflictSuffix: "ContextConfig"
+				}, { reservedNames: reservedReferenceNames });
+				contextConfigReferenceName = contextConfigReference.localName;
+				const contextReferences = [contextConfigReference];
+				if (hasHeadersTemplateVariables) {
+					const headersReference = resolveReferenceBinding({
+						id: `${contextConfigId}:headers`,
+						importName: parsed.contextConfigHeadersReference?.name ?? `${toCamelCase(contextConfigId)}Headers`,
+						modulePath: contextConfigId,
+						local: parsed.contextConfigHeadersReference?.local === true,
+						conflictSuffix: "Headers"
+					}, { reservedNames: reservedReferenceNames });
+					contextHeadersReferenceName = headersReference.localName;
+					contextReferences.push(headersReference);
+				}
+				addResolvedReferenceImports(sourceFile, contextReferences, () => {
+					return `../context-configs/${contextConfigId}`;
+				});
+			}
+			const { referenceNames: triggerReferenceNames, importRefs: triggerImportRefs } = createTriggerReferenceMaps(parsed.triggers, reservedReferenceNames);
+			addTriggerImports(sourceFile, triggerReferenceNames, triggerImportRefs);
+			const { referenceNames: scheduledTriggerReferenceNames, importRefs: scheduledTriggerImportRefs } = createScheduledTriggerReferenceMaps(parsed.scheduledTriggers, reservedReferenceNames);
+			addScheduledTriggerImports(sourceFile, scheduledTriggerReferenceNames, scheduledTriggerImportRefs);
+			const statusComponentReferenceNames = createReferenceNameMap(parsed.normalizedStatusComponentIds, reservedReferenceNames, "StatusComponent");
+			addStatusComponentImports(sourceFile, statusComponentReferenceNames);
+			writeAgentConfig(configObject, parsed, {
+				subAgents: subAgentReferenceNames,
+				contextConfig: contextConfigReferenceName,
+				contextHeaders: contextHeadersReferenceName,
+				triggers: triggerReferenceNames,
+				scheduledTriggers: scheduledTriggerReferenceNames,
+				statusComponents: statusComponentReferenceNames
+			});
+		}
 	});
-	const reservedReferenceNames = new Set([agentVarName]);
-	const { referenceNames: subAgentReferenceNames, importNames: subAgentImportNames } = createSubAgentReferenceMaps(subAgentIds, reservedReferenceNames, "SubAgent", parsed.subAgentReferences);
-	addSubAgentImports(sourceFile, subAgentReferenceNames, subAgentImportNames, parsed.subAgentReferencePathOverrides);
-	const contextConfigId = extractContextConfigId(parsed.contextConfig);
-	let contextConfigReferenceName;
-	let contextHeadersReferenceName;
-	const hasHeadersTemplateVariables = collectTemplateVariableNamesFromFields([parsed.prompt, parsed.statusUpdates?.prompt]).some((variableName) => variableName.startsWith("headers."));
-	if (contextConfigId) {
-		const contextImportResolution = resolveContextTemplateImports({
-			reservedNames: reservedReferenceNames,
-			shouldResolveContextReference: true,
-			shouldResolveHeadersReference: hasHeadersTemplateVariables,
-			contextConfigReference: parsed.contextConfigReference,
-			contextConfigHeadersReference: parsed.contextConfigHeadersReference,
-			defaultContextImportName: toCamelCase(contextConfigId),
-			defaultHeadersImportName: `${toCamelCase(contextConfigId)}Headers`
-		});
-		contextConfigReferenceName = contextImportResolution.contextReferenceName;
-		contextHeadersReferenceName = contextImportResolution.headersReferenceName;
-		if (contextImportResolution.namedImports.length > 0) sourceFile.addImportDeclaration({
-			namedImports: contextImportResolution.namedImports,
-			moduleSpecifier: `../context-configs/${contextConfigId}`
-		});
-	}
-	const { referenceNames: triggerReferenceNames, importRefs: triggerImportRefs } = createTriggerReferenceMaps(parsed.triggers, reservedReferenceNames);
-	addTriggerImports(sourceFile, triggerReferenceNames, triggerImportRefs);
-	const { referenceNames: scheduledTriggerReferenceNames, importRefs: scheduledTriggerImportRefs } = createScheduledTriggerReferenceMaps(parsed.scheduledTriggers, reservedReferenceNames);
-	addScheduledTriggerImports(sourceFile, scheduledTriggerReferenceNames, scheduledTriggerImportRefs);
-	const statusComponentReferenceNames = createReferenceNameMap(extractStatusComponentIds(parsed.statusUpdates), reservedReferenceNames, "StatusComponent");
-	addStatusComponentImports(sourceFile, statusComponentReferenceNames);
-	writeAgentConfig(configObject, parsed, {
-		subAgents: subAgentReferenceNames,
-		contextConfig: contextConfigReferenceName,
-		contextHeaders: contextHeadersReferenceName,
-		triggers: triggerReferenceNames,
-		scheduledTriggers: scheduledTriggerReferenceNames,
-		statusComponents: statusComponentReferenceNames
-	});
-	return sourceFile;
 }
 function writeAgentConfig(configObject, data, referenceNames) {
-	for (const [key, value] of Object.entries({
+	const agentConfig = {
 		id: data.agentId,
 		name: data.name,
 		description: data.description,
@@ -98,40 +148,63 @@ function writeAgentConfig(configObject, data, referenceNames) {
 		}),
 		models: data.models,
 		stopWhen: data.stopWhen
-	})) addValueToObject(configObject, key, value);
+	};
 	const { defaultSubAgentId } = data;
-	if (defaultSubAgentId) configObject.addPropertyAssignment({
-		name: "defaultSubAgent",
-		initializer: referenceNames.subAgents.get(defaultSubAgentId) ?? toCamelCase(defaultSubAgentId)
-	});
-	addReferenceGetterProperty(configObject, "subAgents", extractIds(data.subAgents).map((id) => referenceNames.subAgents.get(id) ?? toCamelCase(id)));
-	if (extractContextConfigId(data.contextConfig) && referenceNames.contextConfig) configObject.addPropertyAssignment({
-		name: "contextConfig",
-		initializer: referenceNames.contextConfig
-	});
+	if (defaultSubAgentId) agentConfig.defaultSubAgent = codeReference(referenceNames.subAgents.get(defaultSubAgentId) ?? toCamelCase(defaultSubAgentId));
+	agentConfig.subAgents = createReferenceGetterValue(extractIds(data.subAgents).map((id) => referenceNames.subAgents.get(id) ?? toCamelCase(id)));
+	if (data.normalizedContextConfigId && referenceNames.contextConfig) agentConfig.contextConfig = codeReference(referenceNames.contextConfig);
 	const triggerIds = data.triggers ? extractIds(data.triggers) : [];
-	if (triggerIds.length) addReferenceGetterProperty(configObject, "triggers", triggerIds.map((id) => referenceNames.triggers.get(id) ?? toTriggerReferenceName(id)));
+	if (triggerIds.length) agentConfig.triggers = createReferenceGetterValue(triggerIds.map((id) => referenceNames.triggers.get(id) ?? toTriggerReferenceName(id)));
 	const scheduledTriggerIds = data.scheduledTriggers ? extractIds(data.scheduledTriggers) : [];
-	if (scheduledTriggerIds.length) addReferenceGetterProperty(configObject, "scheduledTriggers", scheduledTriggerIds.map((id) => referenceNames.scheduledTriggers.get(id) ?? toTriggerReferenceName(id)));
+	if (scheduledTriggerIds.length) agentConfig.scheduledTriggers = createReferenceGetterValue(scheduledTriggerIds.map((id) => referenceNames.scheduledTriggers.get(id) ?? toTriggerReferenceName(id)));
 	if (data.statusUpdates) {
-		const statusUpdatesObject = configObject.addPropertyAssignment({
-			name: "statusUpdates",
-			initializer: "{}"
-		}).getInitializerIfKindOrThrow(SyntaxKind.ObjectLiteralExpression);
-		addValueToObject(statusUpdatesObject, "numEvents", data.statusUpdates.numEvents);
-		addValueToObject(statusUpdatesObject, "timeInSeconds", data.statusUpdates.timeInSeconds);
-		addValueToObject(statusUpdatesObject, "prompt", data.statusUpdates.prompt && formatTemplate(data.statusUpdates.prompt, {
-			contextReference: referenceNames.contextConfig,
-			headersReference: referenceNames.contextHeaders
-		}));
-		const statusComponentRefs = data.statusUpdates.statusComponents?.map((statusComponent) => `${referenceNames.statusComponents.get(resolveStatusComponentId(statusComponent)) ?? toCamelCase(resolveStatusComponentId(statusComponent))}.config`);
-		if (statusComponentRefs?.length) statusUpdatesObject.addPropertyAssignment({
-			name: "statusComponents",
-			initializer: "[]"
-		}).getInitializerIfKindOrThrow(SyntaxKind.ArrayLiteralExpression).addElements(statusComponentRefs);
+		const statusComponentRefs = data.normalizedStatusComponentSequence.map((statusComponentId) => codePropertyAccess(referenceNames.statusComponents.get(statusComponentId) ?? toCamelCase(statusComponentId), "config"));
+		agentConfig.statusUpdates = {
+			numEvents: data.statusUpdates.numEvents,
+			timeInSeconds: data.statusUpdates.timeInSeconds,
+			prompt: data.statusUpdates.prompt && formatTemplate(data.statusUpdates.prompt, {
+				contextReference: referenceNames.contextConfig,
+				headersReference: referenceNames.contextHeaders
+			}),
+			...statusComponentRefs?.length && { statusComponents: statusComponentRefs }
+		};
 	}
+	for (const [key, value] of Object.entries(agentConfig)) addValueToObject(configObject, key, value);
 }
+const task = {
+	type: "agent",
+	collect(context) {
+		if (!context.project.agents) return [];
+		const records = [];
+		for (const agentId of context.completeAgentIds) {
+			const agentData = context.project.agents[agentId];
+			if (!agentData) continue;
+			const agentName = typeof agentData.name === "string" ? agentData.name : void 0;
+			const agentFilePath = context.resolver.resolveOutputFilePath("agents", agentId, join(context.paths.agentsDir, buildComponentFileName(agentId, agentName)));
+			const existingAgent = context.resolver.getExistingComponent(agentId, "agents");
+			const subAgentReferences = collectSubAgentReferenceOverrides(context, agentData, agentFilePath);
+			const subAgentReferencePathOverrides = collectSubAgentReferencePathOverrides(context, agentData);
+			const statusUpdates = asRecord(agentData.statusUpdates);
+			const contextTemplateReferences = collectContextTemplateReferences(context, agentData, agentFilePath, [typeof agentData.prompt === "string" ? agentData.prompt : void 0, typeof statusUpdates?.prompt === "string" ? statusUpdates.prompt : void 0]);
+			records.push({
+				id: agentId,
+				filePath: agentFilePath,
+				payload: {
+					agentId,
+					...agentData,
+					...existingAgent?.name?.length && { agentVariableName: existingAgent.name },
+					...Object.keys(subAgentReferences).length && { subAgentReferences },
+					...Object.keys(subAgentReferencePathOverrides).length && { subAgentReferencePathOverrides },
+					...contextTemplateReferences && { contextConfigReference: contextTemplateReferences.contextConfigReference },
+					...contextTemplateReferences?.contextConfigHeadersReference && { contextConfigHeadersReference: contextTemplateReferences.contextConfigHeadersReference }
+				}
+			});
+		}
+		return records;
+	},
+	generate: generateAgentDefinition
+};
 
 //#endregion
-export { generateAgentDefinition };
+export { generateAgentDefinition, task };
 //# sourceMappingURL=agent-generator.js.map

package/dist/commands/pull-v4/generators/agent-generator.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-generator.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/agent-generator.ts"],"sourcesContent":["import { FullProjectDefinitionSchema } from '@inkeep/agents-core';\nimport { type ObjectLiteralExpression, type SourceFile, SyntaxKind } from 'ts-morph';\nimport { z } from 'zod';\nimport {\n  addReferenceGetterProperty,\n  addValueToObject,\n  createFactoryDefinition,\n  formatTemplate,\n  resolveContextTemplateImports,\n  toCamelCase,\n  toTriggerReferenceName,\n} from '../utils';\nimport {\n  addScheduledTriggerImports,\n  addStatusComponentImports,\n  addSubAgentImports,\n  addTriggerImports,\n  collectTemplateVariableNamesFromFields,\n  createReferenceNameMap,\n  createScheduledTriggerReferenceMaps,\n  createSubAgentReferenceMaps,\n  createTriggerReferenceMaps,\n  extractContextConfigId,\n  extractIds,\n  extractStatusComponentIds,\n  type ReferenceNameMap,\n  resolveStatusComponentId,\n} from './agent-generator.helpers';\n\nconst SubAgentReferenceSchema = z.object({\n  name: z.string().nonempty(),\n  local: z.boolean().optional(),\n});\n\nconst MySchema = FullProjectDefinitionSchema.shape.agents.valueType.omit({\n  id: true,\n});\n\nconst SubAgentSchema = MySchema.shape.subAgents.valueType.omit({\n  // Invalid input: expected \"internal\"\n  type: true,\n});\n\nconst ToolSchema = MySchema.shape.tools.unwrap().valueType;\n\nconst AgentSchema = z.strictObject({\n  agentId: z.string().nonempty(),\n  ...MySchema.shape,\n  description: z.preprocess((v) => v || undefined, MySchema.shape.description),\n  models: z.preprocess((v) => v ?? undefined, MySchema.shape.models),\n  stopWhen: z.preprocess(\n    (v) => (v && Object.keys(v).length && v) || undefined,\n    MySchema.shape.stopWhen\n  ),\n  subAgents: z.record(\n    z.string(),\n    z.strictObject({\n      ...SubAgentSchema.shape,\n      models: z.preprocess((v) => v ?? undefined, SubAgentSchema.shape.models),\n      stopWhen: z.preprocess((v) => v ?? undefined, SubAgentSchema.shape.stopWhen),\n      // Unrecognized keys: \"name\", \"description\", \"content\", \"metadata\", \"subAgentSkillId\", \"subAgentId\", \"createdAt\", \"updatedAt\"\n      skills: z.unknown(),\n      // Invalid input\n      canDelegateTo: z.unknown(),\n    })\n  ),\n  tools: z\n    .record(\n      z.string(),\n      z.strictObject({\n        ...ToolSchema.shape,\n        // Invalid input: expected string, received null\n        imageUrl: z.preprocess((v) => v ?? undefined, ToolSchema.shape.imageUrl),\n      })\n    )\n    .optional(),\n  // ✖ Invalid input: expected string, received undefined\n  // → at triggers.t546ck7yueh52jils88rm.authentication.headers[0].value\n  triggers: z.record(z.string(), z.unknown()).optional(),\n  agentVariableName: z.string().nonempty().optional(),\n  subAgentReferences: z.record(z.string(), SubAgentReferenceSchema).optional(),\n  subAgentReferencePathOverrides: z.record(z.string(), z.string().nonempty()).optional(),\n  contextConfigReference: SubAgentReferenceSchema.optional(),\n  contextConfigHeadersReference: SubAgentReferenceSchema.optional(),\n});\n\ntype AgentInput = z.input<typeof AgentSchema>;\ntype AgentOutput = z.output<typeof AgentSchema>;\n\ninterface AgentReferenceNames {\n  subAgents: ReferenceNameMap;\n  contextConfig?: string;\n  contextHeaders?: string;\n  triggers: ReferenceNameMap;\n  scheduledTriggers: ReferenceNameMap;\n  statusComponents: ReferenceNameMap;\n}\n\nexport function generateAgentDefinition({\n  id,\n  createdAt,\n  updatedAt,\n  ...data\n}: AgentInput & Record<string, unknown>): SourceFile {\n  const result = AgentSchema.safeParse(data);\n  if (!result.success) {\n    throw new Error(`Validation failed for agent:\\n${z.prettifyError(result.error)}`);\n  }\n\n  const parsed = result.data;\n\n  const subAgentIds = new Set(extractIds(parsed.subAgents));\n  if (parsed.defaultSubAgentId) {\n    subAgentIds.add(parsed.defaultSubAgentId);\n  }\n  const agentVarName = parsed.agentVariableName || toCamelCase(parsed.agentId);\n  const { sourceFile, configObject } = createFactoryDefinition({\n    importName: 'agent',\n    variableName: agentVarName,\n  });\n  const reservedReferenceNames = new Set([agentVarName]);\n  const { referenceNames: subAgentReferenceNames, importNames: subAgentImportNames } =\n    createSubAgentReferenceMaps(\n      subAgentIds,\n      reservedReferenceNames,\n      'SubAgent',\n      parsed.subAgentReferences\n    );\n  addSubAgentImports(\n    sourceFile,\n    subAgentReferenceNames,\n    subAgentImportNames,\n    parsed.subAgentReferencePathOverrides\n  );\n\n  const contextConfigId = extractContextConfigId(parsed.contextConfig);\n  let contextConfigReferenceName: string | undefined;\n  let contextHeadersReferenceName: string | undefined;\n  const promptTemplateVariables = collectTemplateVariableNamesFromFields([\n    parsed.prompt,\n    parsed.statusUpdates?.prompt,\n  ]);\n  const hasHeadersTemplateVariables = promptTemplateVariables.some((variableName) =>\n    variableName.startsWith('headers.')\n  );\n  if (contextConfigId) {\n    const contextImportResolution = resolveContextTemplateImports({\n      reservedNames: reservedReferenceNames,\n      shouldResolveContextReference: true,\n      shouldResolveHeadersReference: hasHeadersTemplateVariables,\n      contextConfigReference: parsed.contextConfigReference,\n      contextConfigHeadersReference: parsed.contextConfigHeadersReference,\n      defaultContextImportName: toCamelCase(contextConfigId),\n      defaultHeadersImportName: `${toCamelCase(contextConfigId)}Headers`,\n    });\n    contextConfigReferenceName = contextImportResolution.contextReferenceName;\n    contextHeadersReferenceName = contextImportResolution.headersReferenceName;\n\n    if (contextImportResolution.namedImports.length > 0) {\n      sourceFile.addImportDeclaration({\n        namedImports: contextImportResolution.namedImports,\n        moduleSpecifier: `../context-configs/${contextConfigId}`,\n      });\n    }\n  }\n\n  const { referenceNames: triggerReferenceNames, importRefs: triggerImportRefs } =\n    createTriggerReferenceMaps(parsed.triggers, reservedReferenceNames);\n  addTriggerImports(sourceFile, triggerReferenceNames, triggerImportRefs);\n\n  const { referenceNames: scheduledTriggerReferenceNames, importRefs: scheduledTriggerImportRefs } =\n    createScheduledTriggerReferenceMaps(parsed.scheduledTriggers, reservedReferenceNames);\n  addScheduledTriggerImports(\n    sourceFile,\n    scheduledTriggerReferenceNames,\n    scheduledTriggerImportRefs\n  );\n\n  const statusComponentReferenceNames = createReferenceNameMap(\n    extractStatusComponentIds(parsed.statusUpdates),\n    reservedReferenceNames,\n    'StatusComponent'\n  );\n  addStatusComponentImports(sourceFile, statusComponentReferenceNames);\n\n  writeAgentConfig(configObject, parsed, {\n    subAgents: subAgentReferenceNames,\n    contextConfig: contextConfigReferenceName,\n    contextHeaders: contextHeadersReferenceName,\n    triggers: triggerReferenceNames,\n    scheduledTriggers: scheduledTriggerReferenceNames,\n    statusComponents: statusComponentReferenceNames,\n  });\n  return sourceFile;\n}\n\nfunction writeAgentConfig(\n  configObject: ObjectLiteralExpression,\n  data: AgentOutput,\n  referenceNames: AgentReferenceNames\n) {\n  for (const [key, value] of Object.entries({\n    id: data.agentId,\n    name: data.name,\n    description: data.description,\n    prompt:\n      data.prompt &&\n      formatTemplate(data.prompt, {\n        contextReference: referenceNames.contextConfig,\n        headersReference: referenceNames.contextHeaders,\n      }),\n    models: data.models,\n    stopWhen: data.stopWhen,\n  })) {\n    addValueToObject(configObject, key, value);\n  }\n\n  const { defaultSubAgentId } = data;\n  if (defaultSubAgentId) {\n    configObject.addPropertyAssignment({\n      name: 'defaultSubAgent',\n      initializer:\n        referenceNames.subAgents.get(defaultSubAgentId) ?? toCamelCase(defaultSubAgentId),\n    });\n  }\n\n  const subAgentIds = extractIds(data.subAgents);\n  addReferenceGetterProperty(\n    configObject,\n    'subAgents',\n    subAgentIds.map((id) => referenceNames.subAgents.get(id) ?? toCamelCase(id))\n  );\n\n  const contextConfigId = extractContextConfigId(data.contextConfig);\n  if (contextConfigId && referenceNames.contextConfig) {\n    configObject.addPropertyAssignment({\n      name: 'contextConfig',\n      initializer: referenceNames.contextConfig,\n    });\n  }\n\n  const triggerIds = data.triggers ? extractIds(data.triggers) : [];\n  if (triggerIds.length) {\n    addReferenceGetterProperty(\n      configObject,\n      'triggers',\n      triggerIds.map((id) => referenceNames.triggers.get(id) ?? toTriggerReferenceName(id))\n    );\n  }\n\n  const scheduledTriggerIds = data.scheduledTriggers ? extractIds(data.scheduledTriggers) : [];\n  if (scheduledTriggerIds.length) {\n    addReferenceGetterProperty(\n      configObject,\n      'scheduledTriggers',\n      scheduledTriggerIds.map(\n        (id) => referenceNames.scheduledTriggers.get(id) ?? toTriggerReferenceName(id)\n      )\n    );\n  }\n\n  if (data.statusUpdates) {\n    const statusUpdatesProperty = configObject.addPropertyAssignment({\n      name: 'statusUpdates',\n      initializer: '{}',\n    });\n    const statusUpdatesObject = statusUpdatesProperty.getInitializerIfKindOrThrow(\n      SyntaxKind.ObjectLiteralExpression\n    );\n    addValueToObject(statusUpdatesObject, 'numEvents', data.statusUpdates.numEvents);\n    addValueToObject(statusUpdatesObject, 'timeInSeconds', data.statusUpdates.timeInSeconds);\n    addValueToObject(\n      statusUpdatesObject,\n      'prompt',\n      data.statusUpdates.prompt &&\n        formatTemplate(data.statusUpdates.prompt, {\n          contextReference: referenceNames.contextConfig,\n          headersReference: referenceNames.contextHeaders,\n        })\n    );\n    const statusComponentRefs = data.statusUpdates.statusComponents?.map(\n      (statusComponent) =>\n        `${referenceNames.statusComponents.get(resolveStatusComponentId(statusComponent)) ?? toCamelCase(resolveStatusComponentId(statusComponent))}.config`\n    );\n    if (statusComponentRefs?.length) {\n      const statusComponentsProperty = statusUpdatesObject.addPropertyAssignment({\n        name: 'statusComponents',\n        initializer: '[]',\n      });\n      const statusComponentsArray = statusComponentsProperty.getInitializerIfKindOrThrow(\n        SyntaxKind.ArrayLiteralExpression\n      );\n      statusComponentsArray.addElements(statusComponentRefs);\n    }\n  }\n}\n"],"mappings":";;;;;;;AA6BA,MAAM,0BAA0B,EAAE,OAAO;CACvC,MAAM,EAAE,QAAQ,CAAC,UAAU;CAC3B,OAAO,EAAE,SAAS,CAAC,UAAU;CAC9B,CAAC;AAEF,MAAM,WAAW,4BAA4B,MAAM,OAAO,UAAU,KAAK,EACvE,IAAI,MACL,CAAC;AAEF,MAAM,iBAAiB,SAAS,MAAM,UAAU,UAAU,KAAK,EAE7D,MAAM,MACP,CAAC;AAEF,MAAM,aAAa,SAAS,MAAM,MAAM,QAAQ,CAAC;AAEjD,MAAM,cAAc,EAAE,aAAa;CACjC,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,GAAG,SAAS;CACZ,aAAa,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,YAAY;CAC5E,QAAQ,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,OAAO;CAClE,UAAU,EAAE,YACT,MAAO,KAAK,OAAO,KAAK,EAAE,CAAC,UAAU,KAAM,QAC5C,SAAS,MAAM,SAChB;CACD,WAAW,EAAE,OACX,EAAE,QAAQ,EACV,EAAE,aAAa;EACb,GAAG,eAAe;EAClB,QAAQ,EAAE,YAAY,MAAM,KAAK,QAAW,eAAe,MAAM,OAAO;EACxE,UAAU,EAAE,YAAY,MAAM,KAAK,QAAW,eAAe,MAAM,SAAS;EAE5E,QAAQ,EAAE,SAAS;EAEnB,eAAe,EAAE,SAAS;EAC3B,CAAC,CACH;CACD,OAAO,EACJ,OACC,EAAE,QAAQ,EACV,EAAE,aAAa;EACb,GAAG,WAAW;EAEd,UAAU,EAAE,YAAY,MAAM,KAAK,QAAW,WAAW,MAAM,SAAS;EACzE,CAAC,CACH,CACA,UAAU;CAGb,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,SAAS,CAAC,CAAC,UAAU;CACtD,mBAAmB,EAAE,QAAQ,CAAC,UAAU,CAAC,UAAU;CACnD,oBAAoB,EAAE,OAAO,EAAE,QAAQ,EAAE,wBAAwB,CAAC,UAAU;CAC5E,gCAAgC,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC,UAAU;CACtF,wBAAwB,wBAAwB,UAAU;CAC1D,+BAA+B,wBAAwB,UAAU;CAClE,CAAC;AAcF,SAAgB,wBAAwB,EACtC,IACA,WACA,WACA,GAAG,QACgD;CACnD,MAAM,SAAS,YAAY,UAAU,KAAK;AAC1C,KAAI,CAAC,OAAO,QACV,OAAM,IAAI,MAAM,iCAAiC,EAAE,cAAc,OAAO,MAAM,GAAG;CAGnF,MAAM,SAAS,OAAO;CAEtB,MAAM,cAAc,IAAI,IAAI,WAAW,OAAO,UAAU,CAAC;AACzD,KAAI,OAAO,kBACT,aAAY,IAAI,OAAO,kBAAkB;CAE3C,MAAM,eAAe,OAAO,qBAAqB,YAAY,OAAO,QAAQ;CAC5E,MAAM,EAAE,YAAY,iBAAiB,wBAAwB;EAC3D,YAAY;EACZ,cAAc;EACf,CAAC;CACF,MAAM,yBAAyB,IAAI,IAAI,CAAC,aAAa,CAAC;CACtD,MAAM,EAAE,gBAAgB,wBAAwB,aAAa,wBAC3D,4BACE,aACA,wBACA,YACA,OAAO,mBACR;AACH,oBACE,YACA,wBACA,qBACA,OAAO,+BACR;CAED,MAAM,kBAAkB,uBAAuB,OAAO,cAAc;CACpE,IAAI;CACJ,IAAI;CAKJ,MAAM,8BAJ0B,uCAAuC,CACrE,OAAO,QACP,OAAO,eAAe,OACvB,CAAC,CAC0D,MAAM,iBAChE,aAAa,WAAW,WAAW,CACpC;AACD,KAAI,iBAAiB;EACnB,MAAM,0BAA0B,8BAA8B;GAC5D,eAAe;GACf,+BAA+B;GAC/B,+BAA+B;GAC/B,wBAAwB,OAAO;GAC/B,+BAA+B,OAAO;GACtC,0BAA0B,YAAY,gBAAgB;GACtD,0BAA0B,GAAG,YAAY,gBAAgB,CAAC;GAC3D,CAAC;AACF,+BAA6B,wBAAwB;AACrD,gCAA8B,wBAAwB;AAEtD,MAAI,wBAAwB,aAAa,SAAS,EAChD,YAAW,qBAAqB;GAC9B,cAAc,wBAAwB;GACtC,iBAAiB,sBAAsB;GACxC,CAAC;;CAIN,MAAM,EAAE,gBAAgB,uBAAuB,YAAY,sBACzD,2BAA2B,OAAO,UAAU,uBAAuB;AACrE,mBAAkB,YAAY,uBAAuB,kBAAkB;CAEvE,MAAM,EAAE,gBAAgB,gCAAgC,YAAY,+BAClE,oCAAoC,OAAO,mBAAmB,uBAAuB;AACvF,4BACE,YACA,gCACA,2BACD;CAED,MAAM,gCAAgC,uBACpC,0BAA0B,OAAO,cAAc,EAC/C,wBACA,kBACD;AACD,2BAA0B,YAAY,8BAA8B;AAEpE,kBAAiB,cAAc,QAAQ;EACrC,WAAW;EACX,eAAe;EACf,gBAAgB;EAChB,UAAU;EACV,mBAAmB;EACnB,kBAAkB;EACnB,CAAC;AACF,QAAO;;AAGT,SAAS,iBACP,cACA,MACA,gBACA;AACA,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ;EACxC,IAAI,KAAK;EACT,MAAM,KAAK;EACX,aAAa,KAAK;EAClB,QACE,KAAK,UACL,eAAe,KAAK,QAAQ;GAC1B,kBAAkB,eAAe;GACjC,kBAAkB,eAAe;GAClC,CAAC;EACJ,QAAQ,KAAK;EACb,UAAU,KAAK;EAChB,CAAC,CACA,kBAAiB,cAAc,KAAK,MAAM;CAG5C,MAAM,EAAE,sBAAsB;AAC9B,KAAI,kBACF,cAAa,sBAAsB;EACjC,MAAM;EACN,aACE,eAAe,UAAU,IAAI,kBAAkB,IAAI,YAAY,kBAAkB;EACpF,CAAC;AAIJ,4BACE,cACA,aAHkB,WAAW,KAAK,UAAU,CAIhC,KAAK,OAAO,eAAe,UAAU,IAAI,GAAG,IAAI,YAAY,GAAG,CAAC,CAC7E;AAGD,KADwB,uBAAuB,KAAK,cAAc,IAC3C,eAAe,cACpC,cAAa,sBAAsB;EACjC,MAAM;EACN,aAAa,eAAe;EAC7B,CAAC;CAGJ,MAAM,aAAa,KAAK,WAAW,WAAW,KAAK,SAAS,GAAG,EAAE;AACjE,KAAI,WAAW,OACb,4BACE,cACA,YACA,WAAW,KAAK,OAAO,eAAe,SAAS,IAAI,GAAG,IAAI,uBAAuB,GAAG,CAAC,CACtF;CAGH,MAAM,sBAAsB,KAAK,oBAAoB,WAAW,KAAK,kBAAkB,GAAG,EAAE;AAC5F,KAAI,oBAAoB,OACtB,4BACE,cACA,qBACA,oBAAoB,KACjB,OAAO,eAAe,kBAAkB,IAAI,GAAG,IAAI,uBAAuB,GAAG,CAC/E,CACF;AAGH,KAAI,KAAK,eAAe;EAKtB,MAAM,sBAJwB,aAAa,sBAAsB;GAC/D,MAAM;GACN,aAAa;GACd,CAAC,CACgD,4BAChD,WAAW,wBACZ;AACD,mBAAiB,qBAAqB,aAAa,KAAK,cAAc,UAAU;AAChF,mBAAiB,qBAAqB,iBAAiB,KAAK,cAAc,cAAc;AACxF,mBACE,qBACA,UACA,KAAK,cAAc,UACjB,eAAe,KAAK,cAAc,QAAQ;GACxC,kBAAkB,eAAe;GACjC,kBAAkB,eAAe;GAClC,CAAC,CACL;EACD,MAAM,sBAAsB,KAAK,cAAc,kBAAkB,KAC9D,oBACC,GAAG,eAAe,iBAAiB,IAAI,yBAAyB,gBAAgB,CAAC,IAAI,YAAY,yBAAyB,gBAAgB,CAAC,CAAC,SAC/I;AACD,MAAI,qBAAqB,OAQvB,CAPiC,oBAAoB,sBAAsB;GACzE,MAAM;GACN,aAAa;GACd,CAAC,CACqD,4BACrD,WAAW,uBACZ,CACqB,YAAY,oBAAoB"}
+{"version":3,"file":"agent-generator.js","names":["id"],"sources":["../../../../src/commands/pull-v4/generators/agent-generator.ts"],"sourcesContent":["import { join } from 'node:path';\nimport { FullProjectDefinitionSchema } from '@inkeep/agents-core';\nimport type { ObjectLiteralExpression, SourceFile } from 'ts-morph';\nimport { z } from 'zod';\nimport { asRecord, collectTemplateVariablesFromValues } from '../collector-common';\nimport {\n  collectContextTemplateReferences,\n  collectSubAgentReferenceOverrides,\n  collectSubAgentReferencePathOverrides,\n} from '../collector-reference-helpers';\nimport type { GenerationTask } from '../generation-types';\nimport {\n  addResolvedReferenceImports,\n  resolveReferenceBinding,\n  resolveReferenceBindingsFromIds,\n  toReferenceNameMap,\n} from '../reference-resolution';\nimport { generateFactorySourceFile } from '../simple-factory-generator';\nimport {\n  addValueToObject,\n  buildComponentFileName,\n  codePropertyAccess,\n  codeReference,\n  createReferenceGetterValue,\n  formatTemplate,\n  isPlainObject,\n  toCamelCase,\n  toTriggerReferenceName,\n} from '../utils';\nimport {\n  addScheduledTriggerImports,\n  addStatusComponentImports,\n  addTriggerImports,\n  createReferenceNameMap,\n  createScheduledTriggerReferenceMaps,\n  createTriggerReferenceMaps,\n  extractIds,\n  type ReferenceNameMap,\n} from './helpers/agent';\n\nconst SubAgentReferenceSchema = z.object({\n  name: z.string().nonempty(),\n  local: z.boolean().optional(),\n});\n\nconst MySchema = FullProjectDefinitionSchema.shape.agents.valueType.omit({\n  id: true,\n});\n\nconst SubAgentSchema = MySchema.shape.subAgents.valueType.omit({\n  // Invalid input: expected \"internal\"\n  type: true,\n});\n\nconst ToolSchema = MySchema.shape.tools.unwrap().valueType;\n\nconst BaseAgentSchema = z.strictObject({\n  agentId: z.string().nonempty(),\n  ...MySchema.shape,\n  description: z.preprocess((v) => v || undefined, MySchema.shape.description),\n  models: z.preprocess((v) => v ?? undefined, MySchema.shape.models),\n  stopWhen: z.preprocess(\n    (v) => (v && Object.keys(v).length && v) || undefined,\n    MySchema.shape.stopWhen\n  ),\n  subAgents: z.record(\n    z.string(),\n    z.strictObject({\n      ...SubAgentSchema.shape,\n      models: z.preprocess((v) => v ?? undefined, SubAgentSchema.shape.models),\n      stopWhen: z.preprocess((v) => v ?? undefined, SubAgentSchema.shape.stopWhen),\n      // Unrecognized keys: \"name\", \"description\", \"content\", \"metadata\", \"subAgentSkillId\", \"subAgentId\", \"createdAt\", \"updatedAt\"\n      skills: z.unknown(),\n      // Invalid input\n      canDelegateTo: z.unknown(),\n    })\n  ),\n  tools: z\n    .record(\n      z.string(),\n      z.strictObject({\n        ...ToolSchema.shape,\n        // Invalid input: expected string, received null\n        imageUrl: z.preprocess((v) => v ?? undefined, ToolSchema.shape.imageUrl),\n      })\n    )\n    .optional(),\n  // ✖ Invalid input: expected string, received undefined\n  // → at triggers.t546ck7yueh52jils88rm.authentication.headers[0].value\n  triggers: z.record(z.string(), z.unknown()).optional(),\n  agentVariableName: z.string().nonempty().optional(),\n  subAgentReferences: z.record(z.string(), SubAgentReferenceSchema).optional(),\n  subAgentReferencePathOverrides: z.record(z.string(), z.string().nonempty()).optional(),\n  contextConfigReference: SubAgentReferenceSchema.optional(),\n  contextConfigHeadersReference: SubAgentReferenceSchema.optional(),\n});\n\nconst AgentSchema = BaseAgentSchema.transform((data) => ({\n  ...data,\n  normalizedContextConfigId: normalizeContextConfigId(data.contextConfig),\n  normalizedStatusComponentIds: normalizeStatusComponentIds(data.statusUpdates?.statusComponents),\n  normalizedStatusComponentSequence: normalizeStatusComponentSequence(\n    data.statusUpdates?.statusComponents\n  ),\n}));\n\ntype AgentInput = z.input<typeof AgentSchema>;\ntype AgentOutput = z.output<typeof AgentSchema>;\n\nfunction normalizeContextConfigId(contextConfig: unknown): string | undefined {\n  if (typeof contextConfig === 'string') {\n    return contextConfig;\n  }\n\n  if (isPlainObject(contextConfig) && typeof contextConfig.id === 'string') {\n    return contextConfig.id;\n  }\n\n  return undefined;\n}\n\nfunction normalizeStatusComponentId(statusComponent: unknown): string | undefined {\n  if (typeof statusComponent === 'string') {\n    return statusComponent;\n  }\n\n  if (!isPlainObject(statusComponent)) {\n    return undefined;\n  }\n\n  if (typeof statusComponent.id === 'string') {\n    return statusComponent.id;\n  }\n\n  if (typeof statusComponent.type === 'string') {\n    return statusComponent.type;\n  }\n\n  return undefined;\n}\n\nfunction normalizeStatusComponentSequence(statusComponents: unknown[] | undefined): string[] {\n  if (!Array.isArray(statusComponents)) {\n    return [];\n  }\n\n  return statusComponents\n    .map((statusComponent) => normalizeStatusComponentId(statusComponent))\n    .filter((statusComponentId): statusComponentId is string => Boolean(statusComponentId));\n}\n\nfunction normalizeStatusComponentIds(statusComponents: unknown[] | undefined): string[] {\n  return [...new Set(normalizeStatusComponentSequence(statusComponents))];\n}\n\ninterface AgentReferenceNames {\n  subAgents: ReferenceNameMap;\n  contextConfig?: string;\n  contextHeaders?: string;\n  triggers: ReferenceNameMap;\n  scheduledTriggers: ReferenceNameMap;\n  statusComponents: ReferenceNameMap;\n}\n\nexport function generateAgentDefinition({\n  id,\n  createdAt,\n  updatedAt,\n  ...data\n}: AgentInput & Record<string, unknown>): SourceFile {\n  return generateFactorySourceFile(data, {\n    schema: AgentSchema,\n    factory: {\n      importName: 'agent',\n      variableName: (parsed) => parsed.agentVariableName || toCamelCase(parsed.agentId),\n    },\n    render({ parsed, sourceFile, configObject }) {\n      const subAgentIds = new Set(extractIds(parsed.subAgents));\n      if (parsed.defaultSubAgentId) {\n        subAgentIds.add(parsed.defaultSubAgentId);\n      }\n      const agentVarName = parsed.agentVariableName || toCamelCase(parsed.agentId);\n      const reservedReferenceNames = new Set([agentVarName]);\n      const subAgentReferences = resolveReferenceBindingsFromIds({\n        ids: subAgentIds,\n        reservedNames: reservedReferenceNames,\n        conflictSuffix: 'SubAgent',\n        referenceOverrides: parsed.subAgentReferences,\n        referencePathOverrides: parsed.subAgentReferencePathOverrides,\n        defaultModulePath: (id) => id,\n      });\n      const subAgentReferenceNames = toReferenceNameMap(subAgentReferences);\n      addResolvedReferenceImports(sourceFile, subAgentReferences, (reference) => {\n        return `./sub-agents/${reference.modulePath}`;\n      });\n\n      const contextConfigId = parsed.normalizedContextConfigId;\n      let contextConfigReferenceName: string | undefined;\n      let contextHeadersReferenceName: string | undefined;\n      const promptTemplateVariables = collectTemplateVariablesFromValues([\n        parsed.prompt,\n        parsed.statusUpdates?.prompt,\n      ]);\n      const hasHeadersTemplateVariables = promptTemplateVariables.some((variableName) =>\n        variableName.startsWith('headers.')\n      );\n      if (contextConfigId) {\n        const contextConfigReference = resolveReferenceBinding(\n          {\n            id: `${contextConfigId}:context`,\n            importName: parsed.contextConfigReference?.name ?? toCamelCase(contextConfigId),\n            modulePath: contextConfigId,\n            local: parsed.contextConfigReference?.local === true,\n            conflictSuffix: 'ContextConfig',\n          },\n          {\n            reservedNames: reservedReferenceNames,\n          }\n        );\n        contextConfigReferenceName = contextConfigReference.localName;\n\n        const contextReferences = [contextConfigReference];\n        if (hasHeadersTemplateVariables) {\n          const headersReference = resolveReferenceBinding(\n            {\n              id: `${contextConfigId}:headers`,\n              importName:\n                parsed.contextConfigHeadersReference?.name ??\n                `${toCamelCase(contextConfigId)}Headers`,\n              modulePath: contextConfigId,\n              local: parsed.contextConfigHeadersReference?.local === true,\n              conflictSuffix: 'Headers',\n            },\n            {\n              reservedNames: reservedReferenceNames,\n            }\n          );\n          contextHeadersReferenceName = headersReference.localName;\n          contextReferences.push(headersReference);\n        }\n\n        addResolvedReferenceImports(sourceFile, contextReferences, () => {\n          return `../context-configs/${contextConfigId}`;\n        });\n      }\n\n      const { referenceNames: triggerReferenceNames, importRefs: triggerImportRefs } =\n        createTriggerReferenceMaps(parsed.triggers, reservedReferenceNames);\n      addTriggerImports(sourceFile, triggerReferenceNames, triggerImportRefs);\n\n      const {\n        referenceNames: scheduledTriggerReferenceNames,\n        importRefs: scheduledTriggerImportRefs,\n      } = createScheduledTriggerReferenceMaps(parsed.scheduledTriggers, reservedReferenceNames);\n      addScheduledTriggerImports(\n        sourceFile,\n        scheduledTriggerReferenceNames,\n        scheduledTriggerImportRefs\n      );\n\n      const statusComponentReferenceNames = createReferenceNameMap(\n        parsed.normalizedStatusComponentIds,\n        reservedReferenceNames,\n        'StatusComponent'\n      );\n      addStatusComponentImports(sourceFile, statusComponentReferenceNames);\n\n      writeAgentConfig(configObject, parsed, {\n        subAgents: subAgentReferenceNames,\n        contextConfig: contextConfigReferenceName,\n        contextHeaders: contextHeadersReferenceName,\n        triggers: triggerReferenceNames,\n        scheduledTriggers: scheduledTriggerReferenceNames,\n        statusComponents: statusComponentReferenceNames,\n      });\n    },\n  });\n}\n\nfunction writeAgentConfig(\n  configObject: ObjectLiteralExpression,\n  data: AgentOutput,\n  referenceNames: AgentReferenceNames\n) {\n  const agentConfig: Record<string, unknown> = {\n    id: data.agentId,\n    name: data.name,\n    description: data.description,\n    prompt:\n      data.prompt &&\n      formatTemplate(data.prompt, {\n        contextReference: referenceNames.contextConfig,\n        headersReference: referenceNames.contextHeaders,\n      }),\n    models: data.models,\n    stopWhen: data.stopWhen,\n  };\n\n  const { defaultSubAgentId } = data;\n  if (defaultSubAgentId) {\n    agentConfig.defaultSubAgent = codeReference(\n      referenceNames.subAgents.get(defaultSubAgentId) ?? toCamelCase(defaultSubAgentId)\n    );\n  }\n\n  const subAgentIds = extractIds(data.subAgents);\n  agentConfig.subAgents = createReferenceGetterValue(\n    subAgentIds.map((id) => referenceNames.subAgents.get(id) ?? toCamelCase(id))\n  );\n\n  const contextConfigId = data.normalizedContextConfigId;\n  if (contextConfigId && referenceNames.contextConfig) {\n    agentConfig.contextConfig = codeReference(referenceNames.contextConfig);\n  }\n\n  const triggerIds = data.triggers ? extractIds(data.triggers) : [];\n  if (triggerIds.length) {\n    agentConfig.triggers = createReferenceGetterValue(\n      triggerIds.map((id) => referenceNames.triggers.get(id) ?? toTriggerReferenceName(id))\n    );\n  }\n\n  const scheduledTriggerIds = data.scheduledTriggers ? extractIds(data.scheduledTriggers) : [];\n  if (scheduledTriggerIds.length) {\n    agentConfig.scheduledTriggers = createReferenceGetterValue(\n      scheduledTriggerIds.map(\n        (id) => referenceNames.scheduledTriggers.get(id) ?? toTriggerReferenceName(id)\n      )\n    );\n  }\n\n  if (data.statusUpdates) {\n    const statusComponentRefs = data.normalizedStatusComponentSequence.map((statusComponentId) =>\n      codePropertyAccess(\n        referenceNames.statusComponents.get(statusComponentId) ?? toCamelCase(statusComponentId),\n        'config'\n      )\n    );\n    agentConfig.statusUpdates = {\n      numEvents: data.statusUpdates.numEvents,\n      timeInSeconds: data.statusUpdates.timeInSeconds,\n      prompt:\n        data.statusUpdates.prompt &&\n        formatTemplate(data.statusUpdates.prompt, {\n          contextReference: referenceNames.contextConfig,\n          headersReference: referenceNames.contextHeaders,\n        }),\n      ...(statusComponentRefs?.length && { statusComponents: statusComponentRefs }),\n    };\n  }\n\n  for (const [key, value] of Object.entries(agentConfig)) {\n    addValueToObject(configObject, key, value);\n  }\n}\n\nexport const task = {\n  type: 'agent',\n  collect(context) {\n    if (!context.project.agents) {\n      return [];\n    }\n\n    const records = [];\n    for (const agentId of context.completeAgentIds) {\n      const agentData = context.project.agents[agentId];\n      if (!agentData) {\n        continue;\n      }\n\n      const agentName = typeof agentData.name === 'string' ? agentData.name : undefined;\n      const agentFilePath = context.resolver.resolveOutputFilePath(\n        'agents',\n        agentId,\n        join(context.paths.agentsDir, buildComponentFileName(agentId, agentName))\n      );\n      const existingAgent = context.resolver.getExistingComponent(agentId, 'agents');\n      const subAgentReferences = collectSubAgentReferenceOverrides(\n        context,\n        agentData,\n        agentFilePath\n      );\n      const subAgentReferencePathOverrides = collectSubAgentReferencePathOverrides(\n        context,\n        agentData\n      );\n      const statusUpdates = asRecord(agentData.statusUpdates);\n      const contextTemplateReferences = collectContextTemplateReferences(\n        context,\n        agentData,\n        agentFilePath,\n        [\n          typeof agentData.prompt === 'string' ? agentData.prompt : undefined,\n          typeof statusUpdates?.prompt === 'string' ? statusUpdates.prompt : undefined,\n        ]\n      );\n\n      records.push({\n        id: agentId,\n        filePath: agentFilePath,\n        payload: {\n          agentId,\n          ...agentData,\n          ...(existingAgent?.name?.length && { agentVariableName: existingAgent.name }),\n          ...(Object.keys(subAgentReferences).length && { subAgentReferences }),\n          ...(Object.keys(subAgentReferencePathOverrides).length && {\n            subAgentReferencePathOverrides,\n          }),\n          ...(contextTemplateReferences && {\n            contextConfigReference: contextTemplateReferences.contextConfigReference,\n          }),\n          ...(contextTemplateReferences?.contextConfigHeadersReference && {\n            contextConfigHeadersReference: contextTemplateReferences.contextConfigHeadersReference,\n          }),\n        } as Parameters<typeof generateAgentDefinition>[0],\n      });\n    }\n\n    return records;\n  },\n  generate: generateAgentDefinition,\n} satisfies GenerationTask<Parameters<typeof generateAgentDefinition>[0]>;\n"],"mappings":";;;;;;;;;;;;;;;;AAwCA,MAAM,0BAA0B,EAAE,OAAO;CACvC,MAAM,EAAE,QAAQ,CAAC,UAAU;CAC3B,OAAO,EAAE,SAAS,CAAC,UAAU;CAC9B,CAAC;AAEF,MAAM,WAAW,4BAA4B,MAAM,OAAO,UAAU,KAAK,EACvE,IAAI,MACL,CAAC;AAEF,MAAM,iBAAiB,SAAS,MAAM,UAAU,UAAU,KAAK,EAE7D,MAAM,MACP,CAAC;AAEF,MAAM,aAAa,SAAS,MAAM,MAAM,QAAQ,CAAC;AA2CjD,MAAM,cAzCkB,EAAE,aAAa;CACrC,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,GAAG,SAAS;CACZ,aAAa,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,YAAY;CAC5E,QAAQ,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,OAAO;CAClE,UAAU,EAAE,YACT,MAAO,KAAK,OAAO,KAAK,EAAE,CAAC,UAAU,KAAM,QAC5C,SAAS,MAAM,SAChB;CACD,WAAW,EAAE,OACX,EAAE,QAAQ,EACV,EAAE,aAAa;EACb,GAAG,eAAe;EAClB,QAAQ,EAAE,YAAY,MAAM,KAAK,QAAW,eAAe,MAAM,OAAO;EACxE,UAAU,EAAE,YAAY,MAAM,KAAK,QAAW,eAAe,MAAM,SAAS;EAE5E,QAAQ,EAAE,SAAS;EAEnB,eAAe,EAAE,SAAS;EAC3B,CAAC,CACH;CACD,OAAO,EACJ,OACC,EAAE,QAAQ,EACV,EAAE,aAAa;EACb,GAAG,WAAW;EAEd,UAAU,EAAE,YAAY,MAAM,KAAK,QAAW,WAAW,MAAM,SAAS;EACzE,CAAC,CACH,CACA,UAAU;CAGb,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,SAAS,CAAC,CAAC,UAAU;CACtD,mBAAmB,EAAE,QAAQ,CAAC,UAAU,CAAC,UAAU;CACnD,oBAAoB,EAAE,OAAO,EAAE,QAAQ,EAAE,wBAAwB,CAAC,UAAU;CAC5E,gCAAgC,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC,UAAU;CACtF,wBAAwB,wBAAwB,UAAU;CAC1D,+BAA+B,wBAAwB,UAAU;CAClE,CAAC,CAEkC,WAAW,UAAU;CACvD,GAAG;CACH,2BAA2B,yBAAyB,KAAK,cAAc;CACvE,8BAA8B,4BAA4B,KAAK,eAAe,iBAAiB;CAC/F,mCAAmC,iCACjC,KAAK,eAAe,iBACrB;CACF,EAAE;AAKH,SAAS,yBAAyB,eAA4C;AAC5E,KAAI,OAAO,kBAAkB,SAC3B,QAAO;AAGT,KAAI,cAAc,cAAc,IAAI,OAAO,cAAc,OAAO,SAC9D,QAAO,cAAc;;AAMzB,SAAS,2BAA2B,iBAA8C;AAChF,KAAI,OAAO,oBAAoB,SAC7B,QAAO;AAGT,KAAI,CAAC,cAAc,gBAAgB,CACjC;AAGF,KAAI,OAAO,gBAAgB,OAAO,SAChC,QAAO,gBAAgB;AAGzB,KAAI,OAAO,gBAAgB,SAAS,SAClC,QAAO,gBAAgB;;AAM3B,SAAS,iCAAiC,kBAAmD;AAC3F,KAAI,CAAC,MAAM,QAAQ,iBAAiB,CAClC,QAAO,EAAE;AAGX,QAAO,iBACJ,KAAK,oBAAoB,2BAA2B,gBAAgB,CAAC,CACrE,QAAQ,sBAAmD,QAAQ,kBAAkB,CAAC;;AAG3F,SAAS,4BAA4B,kBAAmD;AACtF,QAAO,CAAC,GAAG,IAAI,IAAI,iCAAiC,iBAAiB,CAAC,CAAC;;AAYzE,SAAgB,wBAAwB,EACtC,IACA,WACA,WACA,GAAG,QACgD;AACnD,QAAO,0BAA0B,MAAM;EACrC,QAAQ;EACR,SAAS;GACP,YAAY;GACZ,eAAe,WAAW,OAAO,qBAAqB,YAAY,OAAO,QAAQ;GAClF;EACD,OAAO,EAAE,QAAQ,YAAY,gBAAgB;GAC3C,MAAM,cAAc,IAAI,IAAI,WAAW,OAAO,UAAU,CAAC;AACzD,OAAI,OAAO,kBACT,aAAY,IAAI,OAAO,kBAAkB;GAE3C,MAAM,eAAe,OAAO,qBAAqB,YAAY,OAAO,QAAQ;GAC5E,MAAM,yBAAyB,IAAI,IAAI,CAAC,aAAa,CAAC;GACtD,MAAM,qBAAqB,gCAAgC;IACzD,KAAK;IACL,eAAe;IACf,gBAAgB;IAChB,oBAAoB,OAAO;IAC3B,wBAAwB,OAAO;IAC/B,oBAAoB,SAAOA;IAC5B,CAAC;GACF,MAAM,yBAAyB,mBAAmB,mBAAmB;AACrE,+BAA4B,YAAY,qBAAqB,cAAc;AACzE,WAAO,gBAAgB,UAAU;KACjC;GAEF,MAAM,kBAAkB,OAAO;GAC/B,IAAI;GACJ,IAAI;GAKJ,MAAM,8BAJ0B,mCAAmC,CACjE,OAAO,QACP,OAAO,eAAe,OACvB,CAAC,CAC0D,MAAM,iBAChE,aAAa,WAAW,WAAW,CACpC;AACD,OAAI,iBAAiB;IACnB,MAAM,yBAAyB,wBAC7B;KACE,IAAI,GAAG,gBAAgB;KACvB,YAAY,OAAO,wBAAwB,QAAQ,YAAY,gBAAgB;KAC/E,YAAY;KACZ,OAAO,OAAO,wBAAwB,UAAU;KAChD,gBAAgB;KACjB,EACD,EACE,eAAe,wBAChB,CACF;AACD,iCAA6B,uBAAuB;IAEpD,MAAM,oBAAoB,CAAC,uBAAuB;AAClD,QAAI,6BAA6B;KAC/B,MAAM,mBAAmB,wBACvB;MACE,IAAI,GAAG,gBAAgB;MACvB,YACE,OAAO,+BAA+B,QACtC,GAAG,YAAY,gBAAgB,CAAC;MAClC,YAAY;MACZ,OAAO,OAAO,+BAA+B,UAAU;MACvD,gBAAgB;MACjB,EACD,EACE,eAAe,wBAChB,CACF;AACD,mCAA8B,iBAAiB;AAC/C,uBAAkB,KAAK,iBAAiB;;AAG1C,gCAA4B,YAAY,yBAAyB;AAC/D,YAAO,sBAAsB;MAC7B;;GAGJ,MAAM,EAAE,gBAAgB,uBAAuB,YAAY,sBACzD,2BAA2B,OAAO,UAAU,uBAAuB;AACrE,qBAAkB,YAAY,uBAAuB,kBAAkB;GAEvE,MAAM,EACJ,gBAAgB,gCAChB,YAAY,+BACV,oCAAoC,OAAO,mBAAmB,uBAAuB;AACzF,8BACE,YACA,gCACA,2BACD;GAED,MAAM,gCAAgC,uBACpC,OAAO,8BACP,wBACA,kBACD;AACD,6BAA0B,YAAY,8BAA8B;AAEpE,oBAAiB,cAAc,QAAQ;IACrC,WAAW;IACX,eAAe;IACf,gBAAgB;IAChB,UAAU;IACV,mBAAmB;IACnB,kBAAkB;IACnB,CAAC;;EAEL,CAAC;;AAGJ,SAAS,iBACP,cACA,MACA,gBACA;CACA,MAAM,cAAuC;EAC3C,IAAI,KAAK;EACT,MAAM,KAAK;EACX,aAAa,KAAK;EAClB,QACE,KAAK,UACL,eAAe,KAAK,QAAQ;GAC1B,kBAAkB,eAAe;GACjC,kBAAkB,eAAe;GAClC,CAAC;EACJ,QAAQ,KAAK;EACb,UAAU,KAAK;EAChB;CAED,MAAM,EAAE,sBAAsB;AAC9B,KAAI,kBACF,aAAY,kBAAkB,cAC5B,eAAe,UAAU,IAAI,kBAAkB,IAAI,YAAY,kBAAkB,CAClF;AAIH,aAAY,YAAY,2BADJ,WAAW,KAAK,UAAU,CAEhC,KAAK,OAAO,eAAe,UAAU,IAAI,GAAG,IAAI,YAAY,GAAG,CAAC,CAC7E;AAGD,KADwB,KAAK,6BACN,eAAe,cACpC,aAAY,gBAAgB,cAAc,eAAe,cAAc;CAGzE,MAAM,aAAa,KAAK,WAAW,WAAW,KAAK,SAAS,GAAG,EAAE;AACjE,KAAI,WAAW,OACb,aAAY,WAAW,2BACrB,WAAW,KAAK,OAAO,eAAe,SAAS,IAAI,GAAG,IAAI,uBAAuB,GAAG,CAAC,CACtF;CAGH,MAAM,sBAAsB,KAAK,oBAAoB,WAAW,KAAK,kBAAkB,GAAG,EAAE;AAC5F,KAAI,oBAAoB,OACtB,aAAY,oBAAoB,2BAC9B,oBAAoB,KACjB,OAAO,eAAe,kBAAkB,IAAI,GAAG,IAAI,uBAAuB,GAAG,CAC/E,CACF;AAGH,KAAI,KAAK,eAAe;EACtB,MAAM,sBAAsB,KAAK,kCAAkC,KAAK,sBACtE,mBACE,eAAe,iBAAiB,IAAI,kBAAkB,IAAI,YAAY,kBAAkB,EACxF,SACD,CACF;AACD,cAAY,gBAAgB;GAC1B,WAAW,KAAK,cAAc;GAC9B,eAAe,KAAK,cAAc;GAClC,QACE,KAAK,cAAc,UACnB,eAAe,KAAK,cAAc,QAAQ;IACxC,kBAAkB,eAAe;IACjC,kBAAkB,eAAe;IAClC,CAAC;GACJ,GAAI,qBAAqB,UAAU,EAAE,kBAAkB,qBAAqB;GAC7E;;AAGH,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,YAAY,CACpD,kBAAiB,cAAc,KAAK,MAAM;;AAI9C,MAAa,OAAO;CAClB,MAAM;CACN,QAAQ,SAAS;AACf,MAAI,CAAC,QAAQ,QAAQ,OACnB,QAAO,EAAE;EAGX,MAAM,UAAU,EAAE;AAClB,OAAK,MAAM,WAAW,QAAQ,kBAAkB;GAC9C,MAAM,YAAY,QAAQ,QAAQ,OAAO;AACzC,OAAI,CAAC,UACH;GAGF,MAAM,YAAY,OAAO,UAAU,SAAS,WAAW,UAAU,OAAO;GACxE,MAAM,gBAAgB,QAAQ,SAAS,sBACrC,UACA,SACA,KAAK,QAAQ,MAAM,WAAW,uBAAuB,SAAS,UAAU,CAAC,CAC1E;GACD,MAAM,gBAAgB,QAAQ,SAAS,qBAAqB,SAAS,SAAS;GAC9E,MAAM,qBAAqB,kCACzB,SACA,WACA,cACD;GACD,MAAM,iCAAiC,sCACrC,SACA,UACD;GACD,MAAM,gBAAgB,SAAS,UAAU,cAAc;GACvD,MAAM,4BAA4B,iCAChC,SACA,WACA,eACA,CACE,OAAO,UAAU,WAAW,WAAW,UAAU,SAAS,QAC1D,OAAO,eAAe,WAAW,WAAW,cAAc,SAAS,OACpE,CACF;AAED,WAAQ,KAAK;IACX,IAAI;IACJ,UAAU;IACV,SAAS;KACP;KACA,GAAG;KACH,GAAI,eAAe,MAAM,UAAU,EAAE,mBAAmB,cAAc,MAAM;KAC5E,GAAI,OAAO,KAAK,mBAAmB,CAAC,UAAU,EAAE,oBAAoB;KACpE,GAAI,OAAO,KAAK,+BAA+B,CAAC,UAAU,EACxD,gCACD;KACD,GAAI,6BAA6B,EAC/B,wBAAwB,0BAA0B,wBACnD;KACD,GAAI,2BAA2B,iCAAiC,EAC9D,+BAA+B,0BAA0B,+BAC1D;KACF;IACF,CAAC;;AAGJ,SAAO;;CAET,UAAU;CACX"}

package/dist/commands/pull-v4/generators/artifact-component-generator.js

@@ -1,6 +1,14 @@
-import { addValueToObject, convertJsonSchemaToZodSafe, createFactoryDefinition, formatPropertyName, formatStringLiteral, isPlainObject, toCamelCase } from "../utils.js";
+import { isPlainObject } from "../utils/shared.js";
+import { formatPropertyName, formatStringLiteral } from "../utils/templates.js";
+import { codeExpression } from "../utils/code-values.js";
+import { buildComponentFileName, toCamelCase } from "../utils/naming.js";
+import { addValueToObject } from "../utils/factory-writer.js";
+import { convertJsonSchemaToZodSafe } from "../utils/schema-rendering.js";
+import "../utils/index.js";
+import { generateFactorySourceFile } from "../simple-factory-generator.js";
 import { FullProjectDefinitionSchema } from "@inkeep/agents-core";
 import { z } from "zod";
+import { join } from "node:path";
 
 //#region src/commands/pull-v4/generators/artifact-component-generator.ts
 const MySchema = FullProjectDefinitionSchema.shape.artifactComponents.unwrap().valueType.omit({ id: true });
@@ -12,32 +20,30 @@ const ArtifactComponentSchema = z.strictObject({
 	props: z.unknown()
 });
 function generateArtifactComponentDefinition({ tenantId, id, projectId, createdAt, updatedAt, ...data }) {
-	const result = ArtifactComponentSchema.safeParse(data);
-	if (!result.success) throw new Error(`Validation failed for artifact component:\n${z.prettifyError(result.error)}`);
-	const parsed = result.data;
-	const schema = parsed.props;
-	const { sourceFile, configObject } = createFactoryDefinition({
-		importName: "artifactComponent",
-		variableName: toCamelCase(parsed.artifactComponentId)
-	});
-	if (hasInPreviewFields(schema)) sourceFile.addImportDeclaration({
-		namedImports: ["preview"],
-		moduleSpecifier: "@inkeep/agents-core"
-	});
-	if (schema) sourceFile.addImportDeclaration({
-		namedImports: ["z"],
-		moduleSpecifier: "zod"
-	});
-	const { artifactComponentId, props: _, ...rest } = parsed;
-	for (const [key, value] of Object.entries({
-		id: artifactComponentId,
-		...rest
-	})) addValueToObject(configObject, key, value);
-	if (schema) configObject.addPropertyAssignment({
-		name: "props",
-		initializer: formatArtifactSchema(schema)
+	return generateFactorySourceFile(data, {
+		schema: ArtifactComponentSchema,
+		factory: {
+			importName: "artifactComponent",
+			variableName: (parsed) => toCamelCase(parsed.artifactComponentId)
+		},
+		render({ parsed, sourceFile, configObject }) {
+			const schema = parsed.props;
+			if (hasInPreviewFields(schema)) sourceFile.addImportDeclaration({
+				namedImports: ["preview"],
+				moduleSpecifier: "@inkeep/agents-core"
+			});
+			if (schema) sourceFile.addImportDeclaration({
+				namedImports: ["z"],
+				moduleSpecifier: "zod"
+			});
+			const { artifactComponentId, props: _, ...rest } = parsed;
+			for (const [key, value] of Object.entries({
+				id: artifactComponentId,
+				...rest
+			})) addValueToObject(configObject, key, value);
+			if (schema) addValueToObject(configObject, "props", codeExpression(formatArtifactSchema(schema)));
+		}
 	});
-	return sourceFile;
 }
 function hasInPreviewFields(schema) {
 	if (!isPlainObject(schema) || schema.type !== "object" || !isPlainObject(schema.properties)) return false;
@@ -60,7 +66,22 @@ function formatArtifactSchema(schema) {
 	}
 	return convertJsonSchemaToZodSafe(schema);
 }
+const task = {
+	type: "artifact-component",
+	collect(context) {
+		if (!context.project.artifactComponents) return [];
+		return Object.entries(context.project.artifactComponents).map(([artifactComponentId, artifactComponentData]) => ({
+			id: artifactComponentId,
+			filePath: context.resolver.resolveOutputFilePath("artifactComponents", artifactComponentId, join(context.paths.artifactComponentsDir, buildComponentFileName(artifactComponentId, artifactComponentData.name ?? void 0))),
+			payload: {
+				artifactComponentId,
+				...artifactComponentData
+			}
+		}));
+	},
+	generate: generateArtifactComponentDefinition
+};
 
 //#endregion
-export { generateArtifactComponentDefinition };
+export { generateArtifactComponentDefinition, task };
 //# sourceMappingURL=artifact-component-generator.js.map

package/dist/commands/pull-v4/generators/artifact-component-generator.js.map

@@ -1 +1 @@
-{"version":3,"file":"artifact-component-generator.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/artifact-component-generator.ts"],"sourcesContent":["import { FullProjectDefinitionSchema } from '@inkeep/agents-core';\nimport type { SourceFile } from 'ts-morph';\nimport { z } from 'zod';\nimport {\n  addValueToObject,\n  convertJsonSchemaToZodSafe,\n  createFactoryDefinition,\n  formatPropertyName,\n  formatStringLiteral,\n  isPlainObject,\n  toCamelCase,\n} from '../utils';\n\nconst MySchema = FullProjectDefinitionSchema.shape.artifactComponents.unwrap().valueType.omit({\n  id: true,\n});\n\nconst ArtifactComponentSchema = z.strictObject({\n  artifactComponentId: z.string().nonempty(),\n  ...MySchema.shape,\n  description: z.preprocess((v) => v || undefined, MySchema.shape.description),\n  render: z.preprocess((v) => v ?? undefined, MySchema.shape.render),\n  // Invalid input\n  props: z.unknown(),\n});\n\ntype ArtifactComponentInput = z.input<typeof ArtifactComponentSchema>;\n\nexport function generateArtifactComponentDefinition({\n  tenantId,\n  id,\n  projectId,\n  createdAt,\n  updatedAt,\n  ...data\n}: ArtifactComponentInput & Record<string, unknown>): SourceFile {\n  const result = ArtifactComponentSchema.safeParse(data);\n  if (!result.success) {\n    throw new Error(`Validation failed for artifact component:\\n${z.prettifyError(result.error)}`);\n  }\n\n  const parsed = result.data;\n  const schema = parsed.props;\n  const { sourceFile, configObject } = createFactoryDefinition({\n    importName: 'artifactComponent',\n    variableName: toCamelCase(parsed.artifactComponentId),\n  });\n\n  if (hasInPreviewFields(schema)) {\n    sourceFile.addImportDeclaration({\n      namedImports: ['preview'],\n      moduleSpecifier: '@inkeep/agents-core',\n    });\n  }\n  if (schema) {\n    sourceFile.addImportDeclaration({ namedImports: ['z'], moduleSpecifier: 'zod' });\n  }\n\n  const { artifactComponentId, props: _, ...rest } = parsed;\n\n  for (const [key, value] of Object.entries({\n    id: artifactComponentId,\n    ...rest,\n  })) {\n    addValueToObject(configObject, key, value);\n  }\n  if (schema) {\n    configObject.addPropertyAssignment({\n      name: 'props',\n      initializer: formatArtifactSchema(schema),\n    });\n  }\n  return sourceFile;\n}\n\nfunction hasInPreviewFields(schema: unknown): boolean {\n  if (!isPlainObject(schema) || schema.type !== 'object' || !isPlainObject(schema.properties)) {\n    return false;\n  }\n\n  return Object.values(schema.properties).some(\n    (property) => isPlainObject(property) && property.inPreview === true\n  );\n}\n\nfunction formatArtifactSchema(schema: unknown): string {\n  if (!isPlainObject(schema)) {\n    return 'z.any()';\n  }\n\n  if (schema.type === 'object' && isPlainObject(schema.properties)) {\n    const lines = ['z.object({'];\n\n    for (const [key, propertySchema] of Object.entries(schema.properties)) {\n      const propertyWithoutPreview = isPlainObject(propertySchema)\n        ? { ...propertySchema }\n        : propertySchema;\n\n      if (isPlainObject(propertyWithoutPreview)) {\n        delete propertyWithoutPreview.inPreview;\n      }\n\n      const zodType = convertJsonSchemaToZodSafe(propertyWithoutPreview);\n      const propertyValue =\n        isPlainObject(propertySchema) && propertySchema.inPreview === true\n          ? `preview(${zodType})`\n          : zodType;\n\n      lines.push(`  ${formatPropertyName(key)}: ${propertyValue},`);\n    }\n\n    lines.push('})');\n\n    if (typeof schema.description === 'string') {\n      return `${lines.join('\\n')}.describe(${formatStringLiteral(schema.description)})`;\n    }\n\n    return lines.join('\\n');\n  }\n\n  return convertJsonSchemaToZodSafe(schema);\n}\n"],"mappings":";;;;;AAaA,MAAM,WAAW,4BAA4B,MAAM,mBAAmB,QAAQ,CAAC,UAAU,KAAK,EAC5F,IAAI,MACL,CAAC;AAEF,MAAM,0BAA0B,EAAE,aAAa;CAC7C,qBAAqB,EAAE,QAAQ,CAAC,UAAU;CAC1C,GAAG,SAAS;CACZ,aAAa,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,YAAY;CAC5E,QAAQ,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,OAAO;CAElE,OAAO,EAAE,SAAS;CACnB,CAAC;AAIF,SAAgB,oCAAoC,EAClD,UACA,IACA,WACA,WACA,WACA,GAAG,QAC4D;CAC/D,MAAM,SAAS,wBAAwB,UAAU,KAAK;AACtD,KAAI,CAAC,OAAO,QACV,OAAM,IAAI,MAAM,8CAA8C,EAAE,cAAc,OAAO,MAAM,GAAG;CAGhG,MAAM,SAAS,OAAO;CACtB,MAAM,SAAS,OAAO;CACtB,MAAM,EAAE,YAAY,iBAAiB,wBAAwB;EAC3D,YAAY;EACZ,cAAc,YAAY,OAAO,oBAAoB;EACtD,CAAC;AAEF,KAAI,mBAAmB,OAAO,CAC5B,YAAW,qBAAqB;EAC9B,cAAc,CAAC,UAAU;EACzB,iBAAiB;EAClB,CAAC;AAEJ,KAAI,OACF,YAAW,qBAAqB;EAAE,cAAc,CAAC,IAAI;EAAE,iBAAiB;EAAO,CAAC;CAGlF,MAAM,EAAE,qBAAqB,OAAO,GAAG,GAAG,SAAS;AAEnD,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ;EACxC,IAAI;EACJ,GAAG;EACJ,CAAC,CACA,kBAAiB,cAAc,KAAK,MAAM;AAE5C,KAAI,OACF,cAAa,sBAAsB;EACjC,MAAM;EACN,aAAa,qBAAqB,OAAO;EAC1C,CAAC;AAEJ,QAAO;;AAGT,SAAS,mBAAmB,QAA0B;AACpD,KAAI,CAAC,cAAc,OAAO,IAAI,OAAO,SAAS,YAAY,CAAC,cAAc,OAAO,WAAW,CACzF,QAAO;AAGT,QAAO,OAAO,OAAO,OAAO,WAAW,CAAC,MACrC,aAAa,cAAc,SAAS,IAAI,SAAS,cAAc,KACjE;;AAGH,SAAS,qBAAqB,QAAyB;AACrD,KAAI,CAAC,cAAc,OAAO,CACxB,QAAO;AAGT,KAAI,OAAO,SAAS,YAAY,cAAc,OAAO,WAAW,EAAE;EAChE,MAAM,QAAQ,CAAC,aAAa;AAE5B,OAAK,MAAM,CAAC,KAAK,mBAAmB,OAAO,QAAQ,OAAO,WAAW,EAAE;GACrE,MAAM,yBAAyB,cAAc,eAAe,GACxD,EAAE,GAAG,gBAAgB,GACrB;AAEJ,OAAI,cAAc,uBAAuB,CACvC,QAAO,uBAAuB;GAGhC,MAAM,UAAU,2BAA2B,uBAAuB;GAClE,MAAM,gBACJ,cAAc,eAAe,IAAI,eAAe,cAAc,OAC1D,WAAW,QAAQ,KACnB;AAEN,SAAM,KAAK,KAAK,mBAAmB,IAAI,CAAC,IAAI,cAAc,GAAG;;AAG/D,QAAM,KAAK,KAAK;AAEhB,MAAI,OAAO,OAAO,gBAAgB,SAChC,QAAO,GAAG,MAAM,KAAK,KAAK,CAAC,YAAY,oBAAoB,OAAO,YAAY,CAAC;AAGjF,SAAO,MAAM,KAAK,KAAK;;AAGzB,QAAO,2BAA2B,OAAO"}
+{"version":3,"file":"artifact-component-generator.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/artifact-component-generator.ts"],"sourcesContent":["import { join } from 'node:path';\nimport { FullProjectDefinitionSchema } from '@inkeep/agents-core';\nimport type { SourceFile } from 'ts-morph';\nimport { z } from 'zod';\nimport type { GenerationTask } from '../generation-types';\nimport { generateFactorySourceFile } from '../simple-factory-generator';\nimport {\n  addValueToObject,\n  buildComponentFileName,\n  codeExpression,\n  convertJsonSchemaToZodSafe,\n  formatPropertyName,\n  formatStringLiteral,\n  isPlainObject,\n  toCamelCase,\n} from '../utils';\n\nconst MySchema = FullProjectDefinitionSchema.shape.artifactComponents.unwrap().valueType.omit({\n  id: true,\n});\n\nconst ArtifactComponentSchema = z.strictObject({\n  artifactComponentId: z.string().nonempty(),\n  ...MySchema.shape,\n  description: z.preprocess((v) => v || undefined, MySchema.shape.description),\n  render: z.preprocess((v) => v ?? undefined, MySchema.shape.render),\n  // Invalid input\n  props: z.unknown(),\n});\n\ntype ArtifactComponentInput = z.input<typeof ArtifactComponentSchema>;\n\nexport function generateArtifactComponentDefinition({\n  tenantId,\n  id,\n  projectId,\n  createdAt,\n  updatedAt,\n  ...data\n}: ArtifactComponentInput & Record<string, unknown>): SourceFile {\n  return generateFactorySourceFile(data, {\n    schema: ArtifactComponentSchema,\n    factory: {\n      importName: 'artifactComponent',\n      variableName: (parsed) => toCamelCase(parsed.artifactComponentId),\n    },\n    render({ parsed, sourceFile, configObject }) {\n      const schema = parsed.props;\n\n      if (hasInPreviewFields(schema)) {\n        sourceFile.addImportDeclaration({\n          namedImports: ['preview'],\n          moduleSpecifier: '@inkeep/agents-core',\n        });\n      }\n      if (schema) {\n        sourceFile.addImportDeclaration({ namedImports: ['z'], moduleSpecifier: 'zod' });\n      }\n\n      const { artifactComponentId, props: _, ...rest } = parsed;\n\n      for (const [key, value] of Object.entries({\n        id: artifactComponentId,\n        ...rest,\n      })) {\n        addValueToObject(configObject, key, value);\n      }\n      if (schema) {\n        addValueToObject(configObject, 'props', codeExpression(formatArtifactSchema(schema)));\n      }\n    },\n  });\n}\n\nfunction hasInPreviewFields(schema: unknown): boolean {\n  if (!isPlainObject(schema) || schema.type !== 'object' || !isPlainObject(schema.properties)) {\n    return false;\n  }\n\n  return Object.values(schema.properties).some(\n    (property) => isPlainObject(property) && property.inPreview === true\n  );\n}\n\nfunction formatArtifactSchema(schema: unknown): string {\n  if (!isPlainObject(schema)) {\n    return 'z.any()';\n  }\n\n  if (schema.type === 'object' && isPlainObject(schema.properties)) {\n    const lines = ['z.object({'];\n\n    for (const [key, propertySchema] of Object.entries(schema.properties)) {\n      const propertyWithoutPreview = isPlainObject(propertySchema)\n        ? { ...propertySchema }\n        : propertySchema;\n\n      if (isPlainObject(propertyWithoutPreview)) {\n        delete propertyWithoutPreview.inPreview;\n      }\n\n      const zodType = convertJsonSchemaToZodSafe(propertyWithoutPreview);\n      const propertyValue =\n        isPlainObject(propertySchema) && propertySchema.inPreview === true\n          ? `preview(${zodType})`\n          : zodType;\n\n      lines.push(`  ${formatPropertyName(key)}: ${propertyValue},`);\n    }\n\n    lines.push('})');\n\n    if (typeof schema.description === 'string') {\n      return `${lines.join('\\n')}.describe(${formatStringLiteral(schema.description)})`;\n    }\n\n    return lines.join('\\n');\n  }\n\n  return convertJsonSchemaToZodSafe(schema);\n}\n\nexport const task = {\n  type: 'artifact-component',\n  collect(context) {\n    if (!context.project.artifactComponents) {\n      return [];\n    }\n\n    return Object.entries(context.project.artifactComponents).map(\n      ([artifactComponentId, artifactComponentData]) => ({\n        id: artifactComponentId,\n        filePath: context.resolver.resolveOutputFilePath(\n          'artifactComponents',\n          artifactComponentId,\n          join(\n            context.paths.artifactComponentsDir,\n            buildComponentFileName(artifactComponentId, artifactComponentData.name ?? undefined)\n          )\n        ),\n        payload: {\n          artifactComponentId,\n          ...artifactComponentData,\n        } as Parameters<typeof generateArtifactComponentDefinition>[0],\n      })\n    );\n  },\n  generate: generateArtifactComponentDefinition,\n} satisfies GenerationTask<Parameters<typeof generateArtifactComponentDefinition>[0]>;\n"],"mappings":";;;;;;;;;;;;;AAiBA,MAAM,WAAW,4BAA4B,MAAM,mBAAmB,QAAQ,CAAC,UAAU,KAAK,EAC5F,IAAI,MACL,CAAC;AAEF,MAAM,0BAA0B,EAAE,aAAa;CAC7C,qBAAqB,EAAE,QAAQ,CAAC,UAAU;CAC1C,GAAG,SAAS;CACZ,aAAa,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,YAAY;CAC5E,QAAQ,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,OAAO;CAElE,OAAO,EAAE,SAAS;CACnB,CAAC;AAIF,SAAgB,oCAAoC,EAClD,UACA,IACA,WACA,WACA,WACA,GAAG,QAC4D;AAC/D,QAAO,0BAA0B,MAAM;EACrC,QAAQ;EACR,SAAS;GACP,YAAY;GACZ,eAAe,WAAW,YAAY,OAAO,oBAAoB;GAClE;EACD,OAAO,EAAE,QAAQ,YAAY,gBAAgB;GAC3C,MAAM,SAAS,OAAO;AAEtB,OAAI,mBAAmB,OAAO,CAC5B,YAAW,qBAAqB;IAC9B,cAAc,CAAC,UAAU;IACzB,iBAAiB;IAClB,CAAC;AAEJ,OAAI,OACF,YAAW,qBAAqB;IAAE,cAAc,CAAC,IAAI;IAAE,iBAAiB;IAAO,CAAC;GAGlF,MAAM,EAAE,qBAAqB,OAAO,GAAG,GAAG,SAAS;AAEnD,QAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ;IACxC,IAAI;IACJ,GAAG;IACJ,CAAC,CACA,kBAAiB,cAAc,KAAK,MAAM;AAE5C,OAAI,OACF,kBAAiB,cAAc,SAAS,eAAe,qBAAqB,OAAO,CAAC,CAAC;;EAG1F,CAAC;;AAGJ,SAAS,mBAAmB,QAA0B;AACpD,KAAI,CAAC,cAAc,OAAO,IAAI,OAAO,SAAS,YAAY,CAAC,cAAc,OAAO,WAAW,CACzF,QAAO;AAGT,QAAO,OAAO,OAAO,OAAO,WAAW,CAAC,MACrC,aAAa,cAAc,SAAS,IAAI,SAAS,cAAc,KACjE;;AAGH,SAAS,qBAAqB,QAAyB;AACrD,KAAI,CAAC,cAAc,OAAO,CACxB,QAAO;AAGT,KAAI,OAAO,SAAS,YAAY,cAAc,OAAO,WAAW,EAAE;EAChE,MAAM,QAAQ,CAAC,aAAa;AAE5B,OAAK,MAAM,CAAC,KAAK,mBAAmB,OAAO,QAAQ,OAAO,WAAW,EAAE;GACrE,MAAM,yBAAyB,cAAc,eAAe,GACxD,EAAE,GAAG,gBAAgB,GACrB;AAEJ,OAAI,cAAc,uBAAuB,CACvC,QAAO,uBAAuB;GAGhC,MAAM,UAAU,2BAA2B,uBAAuB;GAClE,MAAM,gBACJ,cAAc,eAAe,IAAI,eAAe,cAAc,OAC1D,WAAW,QAAQ,KACnB;AAEN,SAAM,KAAK,KAAK,mBAAmB,IAAI,CAAC,IAAI,cAAc,GAAG;;AAG/D,QAAM,KAAK,KAAK;AAEhB,MAAI,OAAO,OAAO,gBAAgB,SAChC,QAAO,GAAG,MAAM,KAAK,KAAK,CAAC,YAAY,oBAAoB,OAAO,YAAY,CAAC;AAGjF,SAAO,MAAM,KAAK,KAAK;;AAGzB,QAAO,2BAA2B,OAAO;;AAG3C,MAAa,OAAO;CAClB,MAAM;CACN,QAAQ,SAAS;AACf,MAAI,CAAC,QAAQ,QAAQ,mBACnB,QAAO,EAAE;AAGX,SAAO,OAAO,QAAQ,QAAQ,QAAQ,mBAAmB,CAAC,KACvD,CAAC,qBAAqB,4BAA4B;GACjD,IAAI;GACJ,UAAU,QAAQ,SAAS,sBACzB,sBACA,qBACA,KACE,QAAQ,MAAM,uBACd,uBAAuB,qBAAqB,sBAAsB,QAAQ,OAAU,CACrF,CACF;GACD,SAAS;IACP;IACA,GAAG;IACJ;GACF,EACF;;CAEH,UAAU;CACX"}