@inkeep/agents-cli 0.58.9 → 0.58.12

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

@@ -0,0 +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"}

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

@@ -1,25 +1,21 @@
-import { addValueToObject, convertJsonSchemaToZodSafe, convertNullToUndefined, createFactoryDefinition, formatPropertyName, formatStringLiteral, isPlainObject, toCamelCase } from "../utils.js";
+import { addValueToObject, convertJsonSchemaToZodSafe, createFactoryDefinition, formatPropertyName, formatStringLiteral, isPlainObject, toCamelCase } from "../utils.js";
+import { FullProjectDefinitionSchema } from "@inkeep/agents-core";
 import { z } from "zod";
 
 //#region src/commands/pull-v4/generators/artifact-component-generator.ts
-const ArtifactComponentSchema = z.object({
+const MySchema = FullProjectDefinitionSchema.shape.artifactComponents.unwrap().valueType.omit({ id: true });
+const ArtifactComponentSchema = z.strictObject({
 	artifactComponentId: z.string().nonempty(),
-	name: z.string().nonempty(),
-	description: z.string().optional(),
-	props: z.looseObject({}),
-	schema: z.looseObject({}).optional(),
-	template: z.string().optional(),
-	contentType: z.string().optional(),
-	render: z.preprocess(convertNullToUndefined, z.looseObject({
-		component: z.string().optional(),
-		mockData: z.looseObject({}).optional()
-	}).optional())
+	...MySchema.shape,
+	description: z.preprocess((v) => v || void 0, MySchema.shape.description),
+	render: z.preprocess((v) => v ?? void 0, MySchema.shape.render),
+	props: z.unknown()
 });
-function generateArtifactComponentDefinition(data) {
+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 ?? parsed.schema;
+	const schema = parsed.props;
 	const { sourceFile, configObject } = createFactoryDefinition({
 		importName: "artifactComponent",
 		variableName: toCamelCase(parsed.artifactComponentId)
@@ -32,7 +28,7 @@ function generateArtifactComponentDefinition(data) {
 		namedImports: ["z"],
 		moduleSpecifier: "zod"
 	});
-	const { artifactComponentId, schema: _, props: _2, ...rest } = parsed;
+	const { artifactComponentId, props: _, ...rest } = parsed;
 	for (const [key, value] of Object.entries({
 		id: artifactComponentId,
 		...rest
@@ -66,4 +62,5 @@ function formatArtifactSchema(schema) {
 }
 
 //#endregion
-export { generateArtifactComponentDefinition };
+export { generateArtifactComponentDefinition };
+//# sourceMappingURL=artifact-component-generator.js.map

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

@@ -0,0 +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"}

package/dist/commands/pull-v4/generators/context-config-generator.js

@@ -1,18 +1,26 @@
-import { addFactoryConfigVariable, addStringProperty, addValueToObject, convertJsonSchemaToZodSafe, convertNullToUndefined, createInMemoryProject, formatPropertyName, isPlainObject, toCamelCase } from "../utils.js";
+import { addFactoryConfigVariable, addStringProperty, addValueToObject, convertJsonSchemaToZodSafe, createInMemoryProject, formatPropertyName, isPlainObject, toCamelCase } from "../utils.js";
+import { FullProjectDefinitionSchema } from "@inkeep/agents-core";
 import { z } from "zod";
 import { SyntaxKind } from "ts-morph";
 
 //#region src/commands/pull-v4/generators/context-config-generator.ts
+const MySchema = FullProjectDefinitionSchema.shape.agents.valueType.shape.contextConfig.unwrap();
 const ReferenceNameByIdSchema = z.record(z.string(), z.string().nonempty());
 const ReferenceOverridesSchema = z.object({ credentialReferences: ReferenceNameByIdSchema.optional() });
-const ContextConfigSchema = z.looseObject({
+const ReferencePathOverridesSchema = z.object({ credentialReferences: ReferenceNameByIdSchema.optional() });
+const ContextConfigSchema = z.strictObject({
 	contextConfigId: z.string().nonempty(),
-	id: z.string().optional(),
-	schema: z.looseObject({}).optional(),
-	headers: z.union([z.string(), z.looseObject({ id: z.string().optional() })]).optional(),
-	headersSchema: z.preprocess(convertNullToUndefined, z.looseObject({}).optional()),
-	contextVariables: z.record(z.string(), z.unknown()).optional(),
-	referenceOverrides: ReferenceOverridesSchema.optional()
+	...MySchema.shape,
+	referenceOverrides: ReferenceOverridesSchema.optional(),
+	referencePathOverrides: ReferencePathOverridesSchema.optional(),
+	headers: z.string().optional(),
+	headersReference: z.string().optional(),
+	schema: z.unknown(),
+	name: z.string().nullish(),
+	trigger: z.unknown(),
+	fetchConfig: z.record(z.string(), z.unknown()).optional(),
+	defaultValue: z.unknown(),
+	responseSchema: z.unknown()
 });
 function generateContextConfigDefinition(data) {
 	const result = ContextConfigSchema.safeParse(data);
@@ -22,15 +30,14 @@ function generateContextConfigDefinition(data) {
 	const sourceFile = project.createSourceFile("context-config-definition.ts", "", { overwrite: true });
 	if (isHeadersDefinitionData(parsed)) return generateStandaloneHeadersDefinition(sourceFile, parsed);
 	if (isFetchDefinitionData(parsed)) return generateStandaloneFetchDefinition(sourceFile, parsed);
-	const parsedContextConfig = result.data;
-	const explicitHeadersReference = extractHeadersReference(parsedContextConfig.headers);
-	const templateHeaderVariables = collectTemplateHeaderVariables(parsedContextConfig.contextVariables);
-	const inferredHeadersSchema = !isPlainObject(parsedContextConfig.headersSchema) && !explicitHeadersReference ? inferHeadersSchemaFromTemplateHeaderVariables(templateHeaderVariables) : void 0;
-	const headersSchema = isPlainObject(parsedContextConfig.headersSchema) ? parsedContextConfig.headersSchema : inferredHeadersSchema;
-	const headersReference = resolveHeadersReference(parsedContextConfig, Boolean(headersSchema));
+	const explicitHeadersReference = (typeof parsed.headersReference === "string" && parsed.headersReference.length > 0 ? parsed.headersReference : void 0) ?? extractHeadersReference(parsed.headers);
+	const templateHeaderVariables = collectTemplateHeaderVariables(parsed.contextVariables);
+	const inferredHeadersSchema = !isPlainObject(parsed.headersSchema) && !explicitHeadersReference ? inferHeadersSchemaFromTemplateHeaderVariables(templateHeaderVariables) : void 0;
+	const headersSchema = isPlainObject(parsed.headersSchema) ? parsed.headersSchema : inferredHeadersSchema;
+	const headersReference = resolveHeadersReference(parsed, Boolean(headersSchema));
 	const shouldDefineHeadersInFile = Boolean(headersReference) && isPlainObject(headersSchema);
-	const fetchDefinitions = collectFetchDefinitionEntries(parsedContextConfig.contextVariables);
-	const credentialReferenceNames = collectCredentialReferenceNames(fetchDefinitions, parsedContextConfig.referenceOverrides?.credentialReferences);
+	const fetchDefinitions = collectFetchDefinitionEntries(parsed.contextVariables);
+	const credentialReferenceNames = collectCredentialReferenceNames(fetchDefinitions, parsed.referenceOverrides?.credentialReferences);
 	const coreImports = ["contextConfig"];
 	if (shouldDefineHeadersInFile) coreImports.unshift("headers");
 	if (fetchDefinitions.length > 0) coreImports.splice(coreImports.length - 1, 0, "fetchDefinition");
@@ -43,13 +50,17 @@ function generateContextConfigDefinition(data) {
 		namedImports: ["z"],
 		moduleSpecifier: "zod"
 	});
-	for (const [credentialId, credentialReferenceName] of credentialReferenceNames) sourceFile.addImportDeclaration({
-		namedImports: [credentialReferenceName],
-		moduleSpecifier: `../credentials/${credentialId}`
-	});
+	for (const [credentialId, credentialReferenceName] of credentialReferenceNames) {
+		const credentialReferencePath = parsed.referencePathOverrides?.credentialReferences?.[credentialId] ?? credentialId;
+		sourceFile.addImportDeclaration({
+			namedImports: [credentialReferenceName],
+			moduleSpecifier: `../credentials/${credentialReferencePath}`
+		});
+	}
 	if (shouldDefineHeadersInFile && headersReference && headersSchema) {
 		const { configObject: headersObject } = addFactoryConfigVariable({
 			sourceFile,
+			isExported: true,
 			importName: "headers",
 			variableName: headersReference
 		});
@@ -69,10 +80,10 @@ function generateContextConfigDefinition(data) {
 	const { configObject } = addFactoryConfigVariable({
 		sourceFile,
 		importName: "contextConfig",
-		variableName: toContextConfigVariableName(parsedContextConfig.contextConfigId),
+		variableName: toContextConfigVariableName(parsed.contextConfigId),
 		isExported: true
 	});
-	writeContextConfig(configObject, parsedContextConfig, headersReference);
+	writeContextConfig(configObject, parsed, headersReference);
 	return sourceFile;
 }
 function writeContextConfig(configObject, data, headersReference) {
@@ -103,6 +114,7 @@ function extractHeadersReference(headers) {
 	if (typeof headers.name === "string") return headers.name;
 }
 function resolveHeadersReference(data, hasHeadersSchema) {
+	if (typeof data.headersReference === "string" && data.headersReference) return data.headersReference;
 	const headersRef = extractHeadersReference(data.headers);
 	if (headersRef) return toReferenceIdentifier(headersRef);
 	if (hasHeadersSchema) return `${toContextConfigVariableName(data.contextConfigId)}Headers`;
@@ -192,10 +204,11 @@ function generateStandaloneFetchDefinition(sourceFile, data) {
 	if (typeof data.credentialReferenceId === "string") {
 		const credentialReferenceId = data.credentialReferenceId;
 		const credentialReferenceName = data.referenceOverrides?.credentialReferences?.[credentialReferenceId] ?? toReferenceIdentifier(credentialReferenceId);
+		const credentialReferencePath = data.referencePathOverrides?.credentialReferences?.[credentialReferenceId] ?? credentialReferenceId;
 		credentialReferenceNames.set(credentialReferenceId, credentialReferenceName);
 		sourceFile.addImportDeclaration({
 			namedImports: [credentialReferenceName],
-			moduleSpecifier: `../credentials/${credentialReferenceId}`
+			moduleSpecifier: `../credentials/${credentialReferencePath}`
 		});
 	}
 	const { configObject } = addFactoryConfigVariable({
@@ -261,4 +274,5 @@ function inferHeadersSchemaFromTemplateHeaderVariables(variables) {
 }
 
 //#endregion
-export { generateContextConfigDefinition };
+export { generateContextConfigDefinition };
+//# sourceMappingURL=context-config-generator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"context-config-generator.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/context-config-generator.ts"],"sourcesContent":["import { FullProjectDefinitionSchema } from '@inkeep/agents-core';\nimport { type ObjectLiteralExpression, type SourceFile, SyntaxKind } from 'ts-morph';\nimport { z } from 'zod';\nimport {\n  addFactoryConfigVariable,\n  addStringProperty,\n  addValueToObject,\n  convertJsonSchemaToZodSafe,\n  createInMemoryProject,\n  formatPropertyName,\n  isPlainObject,\n  toCamelCase,\n} from '../utils';\n\nconst MySchema = FullProjectDefinitionSchema.shape.agents.valueType.shape.contextConfig.unwrap();\n\nconst ReferenceNameByIdSchema = z.record(z.string(), z.string().nonempty());\n\nconst ReferenceOverridesSchema = z.object({\n  credentialReferences: ReferenceNameByIdSchema.optional(),\n});\n\nconst ReferencePathOverridesSchema = z.object({\n  credentialReferences: ReferenceNameByIdSchema.optional(),\n});\n\nconst ContextConfigSchema = z.strictObject({\n  contextConfigId: z.string().nonempty(),\n  ...MySchema.shape,\n  referenceOverrides: ReferenceOverridesSchema.optional(),\n  referencePathOverrides: ReferencePathOverridesSchema.optional(),\n\n  // TODO check these fields\n  headers: z.string().optional(),\n  headersReference: z.string().optional(),\n  schema: z.unknown(),\n  name: z.string().nullish(),\n  trigger: z.unknown(),\n  fetchConfig: z.record(z.string(), z.unknown()).optional(),\n  defaultValue: z.unknown(),\n  responseSchema: z.unknown(),\n});\n\ntype ContextConfigInput = z.input<typeof ContextConfigSchema>;\ntype ContextConfigOutput = z.output<typeof ContextConfigSchema>;\n\nexport function generateContextConfigDefinition(data: ContextConfigInput): SourceFile {\n  const result = ContextConfigSchema.safeParse(data);\n  if (!result.success) {\n    throw new Error(`Validation failed for context config:\\n${z.prettifyError(result.error)}`);\n  }\n\n  const project = createInMemoryProject();\n\n  const parsed = result.data;\n  const sourceFile = project.createSourceFile('context-config-definition.ts', '', {\n    overwrite: true,\n  });\n\n  if (isHeadersDefinitionData(parsed)) {\n    return generateStandaloneHeadersDefinition(sourceFile, parsed);\n  }\n\n  if (isFetchDefinitionData(parsed)) {\n    return generateStandaloneFetchDefinition(sourceFile, parsed);\n  }\n\n  const explicitHeadersReference =\n    (typeof parsed.headersReference === 'string' && parsed.headersReference.length > 0\n      ? parsed.headersReference\n      : undefined) ?? extractHeadersReference(parsed.headers);\n  const templateHeaderVariables = collectTemplateHeaderVariables(parsed.contextVariables);\n  const inferredHeadersSchema =\n    !isPlainObject(parsed.headersSchema) && !explicitHeadersReference\n      ? inferHeadersSchemaFromTemplateHeaderVariables(templateHeaderVariables)\n      : undefined;\n  const headersSchema = isPlainObject(parsed.headersSchema)\n    ? parsed.headersSchema\n    : inferredHeadersSchema;\n  const headersReference = resolveHeadersReference(parsed, Boolean(headersSchema));\n  const shouldDefineHeadersInFile = Boolean(headersReference) && isPlainObject(headersSchema);\n  const fetchDefinitions = collectFetchDefinitionEntries(parsed.contextVariables);\n  const credentialReferenceNames = collectCredentialReferenceNames(\n    fetchDefinitions,\n    parsed.referenceOverrides?.credentialReferences\n  );\n  const coreImports = ['contextConfig'];\n  if (shouldDefineHeadersInFile) {\n    coreImports.unshift('headers');\n  }\n  if (fetchDefinitions.length > 0) {\n    coreImports.splice(coreImports.length - 1, 0, 'fetchDefinition');\n  }\n\n  sourceFile.addImportDeclaration({\n    namedImports: coreImports,\n    moduleSpecifier: '@inkeep/agents-core',\n  });\n\n  const hasResponseSchemas = fetchDefinitions.some((definition) =>\n    // @ts-expect-error -- fixme\n    isPlainObject(definition.data.responseSchema)\n  );\n  if (shouldDefineHeadersInFile || hasResponseSchemas) {\n    sourceFile.addImportDeclaration({\n      namedImports: ['z'],\n      moduleSpecifier: 'zod',\n    });\n  }\n\n  for (const [credentialId, credentialReferenceName] of credentialReferenceNames) {\n    const credentialReferencePath =\n      parsed.referencePathOverrides?.credentialReferences?.[credentialId] ?? credentialId;\n    sourceFile.addImportDeclaration({\n      namedImports: [credentialReferenceName],\n      moduleSpecifier: `../credentials/${credentialReferencePath}`,\n    });\n  }\n  if (shouldDefineHeadersInFile && headersReference && headersSchema) {\n    const { configObject: headersObject } = addFactoryConfigVariable({\n      sourceFile,\n      isExported: true,\n      importName: 'headers',\n      variableName: headersReference,\n    });\n\n    headersObject.addPropertyAssignment({\n      name: 'schema',\n      initializer: convertJsonSchemaToZod(headersSchema),\n    });\n  }\n\n  for (const fetchDefinition of fetchDefinitions) {\n    const { configObject: fetchConfigObject } = addFactoryConfigVariable({\n      sourceFile,\n      importName: 'fetchDefinition',\n      variableName: fetchDefinition.variableName,\n    });\n    writeFetchDefinition(\n      fetchConfigObject,\n      fetchDefinition.data,\n      credentialReferenceNames,\n      headersReference\n    );\n  }\n  const contextConfigVarName = toContextConfigVariableName(parsed.contextConfigId);\n  const { configObject } = addFactoryConfigVariable({\n    sourceFile,\n    importName: 'contextConfig',\n    variableName: contextConfigVarName,\n    isExported: true,\n  });\n\n  writeContextConfig(configObject, parsed, headersReference);\n\n  return sourceFile;\n}\n\nfunction writeContextConfig(\n  configObject: ObjectLiteralExpression,\n  data: ContextConfigOutput,\n  headersReference?: string\n) {\n  if (data.id !== undefined) {\n    addStringProperty(configObject, 'id', data.id);\n  }\n\n  if (headersReference) {\n    configObject.addPropertyAssignment({\n      name: 'headers',\n      initializer: headersReference,\n    });\n  }\n\n  if (data.contextVariables && Object.keys(data.contextVariables).length > 0) {\n    const contextVariablesProperty = configObject.addPropertyAssignment({\n      name: 'contextVariables',\n      initializer: '{}',\n    });\n    const contextVariablesObject = contextVariablesProperty.getInitializerIfKindOrThrow(\n      SyntaxKind.ObjectLiteralExpression\n    );\n\n    for (const [key, value] of Object.entries(data.contextVariables)) {\n      const reference = extractContextVariableReference(key, value);\n      if (!reference) {\n        continue;\n      }\n\n      contextVariablesObject.addPropertyAssignment({\n        name: formatPropertyName(key),\n        initializer: reference,\n      });\n    }\n  }\n}\n\nfunction extractHeadersReference(headers?: string | { id?: string; name?: string }) {\n  if (!headers) {\n    return undefined;\n  }\n  if (typeof headers === 'string') {\n    return headers;\n  }\n  if (typeof headers.id === 'string') {\n    return headers.id;\n  }\n  if (typeof headers.name === 'string') {\n    return headers.name;\n  }\n  return undefined;\n}\n\nfunction resolveHeadersReference(\n  data: ContextConfigOutput,\n  hasHeadersSchema: boolean\n): string | undefined {\n  if (typeof data.headersReference === 'string' && data.headersReference) {\n    return data.headersReference;\n  }\n\n  const headersRef = extractHeadersReference(data.headers);\n  if (headersRef) {\n    return toReferenceIdentifier(headersRef);\n  }\n\n  if (hasHeadersSchema) {\n    return `${toContextConfigVariableName(data.contextConfigId)}Headers`;\n  }\n\n  return undefined;\n}\n\nfunction isFetchDefinitionData(value: unknown): boolean {\n  if (!isPlainObject(value)) {\n    return false;\n  }\n\n  return value.fetchConfig !== undefined || value.responseSchema !== undefined;\n}\n\nfunction collectFetchDefinitionEntries(contextVariables?: Record<string, unknown>) {\n  if (!contextVariables) {\n    return [];\n  }\n\n  return Object.entries(contextVariables)\n    .filter(([, value]) => isFetchDefinitionData(value))\n    .map(([key, value]) => {\n      const variableName =\n        extractContextVariableReference(key, value) ?? toReferenceIdentifier(key);\n      return {\n        key,\n        variableName,\n        data: value,\n      };\n    });\n}\n\nfunction writeFetchDefinition(\n  configObject: ObjectLiteralExpression,\n  fetchDefinitionData: unknown,\n  credentialReferenceNames?: Map<string, string>,\n  headersReference?: string\n) {\n  const { contextConfigId, responseSchema, credentialReferenceId, ...rest } = isPlainObject(\n    fetchDefinitionData\n  )\n    ? fetchDefinitionData\n    : {};\n  const normalizedRest = rewriteHeaderTemplates(rest, headersReference);\n  for (const [k, v] of Object.entries({\n    id: contextConfigId,\n    ...normalizedRest,\n  })) {\n    if (v !== null) {\n      addValueToObject(configObject, k, v);\n    }\n  }\n  if (responseSchema) {\n    configObject.addPropertyAssignment({\n      name: 'responseSchema',\n      // @ts-expect-error -- fixme\n      initializer: convertJsonSchemaToZod(responseSchema),\n    });\n  }\n\n  if (\n    typeof credentialReferenceId === 'string' &&\n    credentialReferenceNames?.has(credentialReferenceId)\n  ) {\n    configObject.addPropertyAssignment({\n      name: 'credentialReference',\n      initializer: credentialReferenceNames.get(credentialReferenceId) as string,\n    });\n    return;\n  }\n\n  if (typeof credentialReferenceId === 'string') {\n    addStringProperty(configObject, 'credentialReferenceId', credentialReferenceId);\n  }\n}\n\nconst HEADER_TEMPLATE_REGEX = /\\{\\{headers\\.([^}]+)\\}\\}/g;\nconst HEADER_TO_TEMPLATE_CALL_REGEX =\n  /\\$\\{\\s*(headersSchema|headers)\\.toTemplate\\((['\"])([^'\"`]+)\\2\\)\\s*\\}/g;\n\nfunction rewriteHeaderTemplates<T>(value: T, headersReference?: string): T {\n  if (!headersReference) {\n    return value;\n  }\n\n  if (typeof value === 'string') {\n    const withHeaderTokensReplaced = value.replace(\n      HEADER_TEMPLATE_REGEX,\n      (_, variableName: string) => {\n        return `\\${${headersReference}.toTemplate(${JSON.stringify(variableName)})}`;\n      }\n    );\n    return withHeaderTokensReplaced.replace(\n      HEADER_TO_TEMPLATE_CALL_REGEX,\n      (_, __: string, ___: string, variableName: string) => {\n        return `\\${${headersReference}.toTemplate(${JSON.stringify(variableName)})}`;\n      }\n    ) as T;\n  }\n\n  if (Array.isArray(value)) {\n    return value.map((entry) => rewriteHeaderTemplates(entry, headersReference)) as T;\n  }\n\n  if (isPlainObject(value)) {\n    return Object.fromEntries(\n      Object.entries(value).map(([key, entryValue]) => [\n        key,\n        rewriteHeaderTemplates(entryValue, headersReference),\n      ])\n    ) as T;\n  }\n\n  return value;\n}\n\nfunction generateStandaloneHeadersDefinition(\n  sourceFile: SourceFile,\n  data: ContextConfigOutput & { schema: Record<string, unknown> }\n): SourceFile {\n  const importName = 'headers';\n  sourceFile.addImportDeclaration({\n    namedImports: [importName],\n    moduleSpecifier: '@inkeep/agents-core',\n  });\n  sourceFile.addImportDeclaration({\n    namedImports: ['z'],\n    moduleSpecifier: 'zod',\n  });\n\n  const headersVarName = toContextConfigVariableName(data.contextConfigId);\n  const { configObject } = addFactoryConfigVariable({\n    sourceFile,\n    importName,\n    variableName: headersVarName,\n  });\n\n  configObject.addPropertyAssignment({\n    name: 'schema',\n    initializer: convertJsonSchemaToZod(data.schema),\n  });\n  return sourceFile;\n}\n\nfunction isHeadersDefinitionData(\n  value: ContextConfigOutput\n): value is ContextConfigOutput & { schema: Record<string, unknown> } {\n  return isPlainObject(value.schema);\n}\n\nfunction generateStandaloneFetchDefinition(\n  sourceFile: SourceFile,\n  data: ContextConfigOutput\n): SourceFile {\n  const importName = 'fetchDefinition';\n  sourceFile.addImportDeclaration({\n    namedImports: [importName],\n    moduleSpecifier: '@inkeep/agents-core',\n  });\n\n  if (isPlainObject(data.responseSchema)) {\n    sourceFile.addImportDeclaration({\n      namedImports: ['z'],\n      moduleSpecifier: 'zod',\n    });\n  }\n\n  const credentialReferenceNames = new Map<string, string>();\n  if (typeof (data as Record<string, unknown>).credentialReferenceId === 'string') {\n    const credentialReferenceId = (data as Record<string, unknown>).credentialReferenceId as string;\n    const credentialReferenceName =\n      data.referenceOverrides?.credentialReferences?.[credentialReferenceId] ??\n      toReferenceIdentifier(credentialReferenceId);\n    const credentialReferencePath =\n      data.referencePathOverrides?.credentialReferences?.[credentialReferenceId] ??\n      credentialReferenceId;\n    credentialReferenceNames.set(credentialReferenceId, credentialReferenceName);\n    sourceFile.addImportDeclaration({\n      namedImports: [credentialReferenceName],\n      moduleSpecifier: `../credentials/${credentialReferencePath}`,\n    });\n  }\n\n  const fetchVarName = toContextConfigVariableName(data.contextConfigId);\n  const { configObject } = addFactoryConfigVariable({\n    sourceFile,\n    importName,\n    variableName: fetchVarName,\n  });\n\n  writeFetchDefinition(configObject, data, credentialReferenceNames);\n  return sourceFile;\n}\n\nfunction convertJsonSchemaToZod(schema: Record<string, unknown>): string {\n  return convertJsonSchemaToZodSafe(schema, {\n    conversionOptions: { module: 'none' },\n  });\n}\n\nfunction extractContextVariableReference(key: string, value: unknown): string | undefined {\n  if (typeof value === 'string') {\n    return toReferenceIdentifier(value);\n  }\n\n  if (!isPlainObject(value)) {\n    return;\n  }\n\n  if (typeof value.id === 'string') {\n    return toReferenceIdentifier(value.id);\n  }\n  if (typeof value.name === 'string') {\n    return toReferenceIdentifier(value.name);\n  }\n  if (typeof value.ref === 'string') {\n    return toReferenceIdentifier(value.ref);\n  }\n  if (typeof value.variable === 'string') {\n    return toReferenceIdentifier(value.variable);\n  }\n\n  if (value.fetchConfig || value.responseSchema) {\n    return toReferenceIdentifier(key);\n  }\n\n  return toReferenceIdentifier(key);\n}\n\nconst toContextConfigVariableName = toCamelCase;\n\nconst toReferenceIdentifier = toCamelCase;\n\nfunction collectCredentialReferenceNames(\n  fetchDefinitions: Array<{ data: unknown }>,\n  overrideNamesById?: Record<string, string>\n): Map<string, string> {\n  const credentialReferenceNames = new Map<string, string>();\n\n  for (const fetchDefinition of fetchDefinitions) {\n    const fetchDefinitionData = isPlainObject(fetchDefinition.data)\n      ? fetchDefinition.data\n      : undefined;\n    const credentialReferenceId =\n      fetchDefinitionData && typeof fetchDefinitionData.credentialReferenceId === 'string'\n        ? fetchDefinitionData.credentialReferenceId\n        : undefined;\n    if (!credentialReferenceId || credentialReferenceNames.has(credentialReferenceId)) {\n      continue;\n    }\n\n    credentialReferenceNames.set(\n      credentialReferenceId,\n      overrideNamesById?.[credentialReferenceId] ?? toReferenceIdentifier(credentialReferenceId)\n    );\n  }\n\n  return credentialReferenceNames;\n}\n\nfunction collectTemplateHeaderVariables(contextVariables?: Record<string, unknown>): Set<string> {\n  const variables = new Set<string>();\n  collectTemplateHeaderVariablesFromValue(contextVariables, variables);\n  return variables;\n}\n\nfunction collectTemplateHeaderVariablesFromValue(value: unknown, variables: Set<string>): void {\n  if (typeof value === 'string') {\n    for (const match of value.matchAll(HEADER_TEMPLATE_REGEX)) {\n      if (match[1]) {\n        variables.add(match[1]);\n      }\n    }\n    for (const match of value.matchAll(HEADER_TO_TEMPLATE_CALL_REGEX)) {\n      if (match[3]) {\n        variables.add(match[3]);\n      }\n    }\n    return;\n  }\n\n  if (Array.isArray(value)) {\n    for (const entry of value) {\n      collectTemplateHeaderVariablesFromValue(entry, variables);\n    }\n    return;\n  }\n\n  if (isPlainObject(value)) {\n    for (const entryValue of Object.values(value)) {\n      collectTemplateHeaderVariablesFromValue(entryValue, variables);\n    }\n  }\n}\n\nfunction inferHeadersSchemaFromTemplateHeaderVariables(\n  variables: Set<string>\n): Record<string, unknown> | undefined {\n  if (!variables.size) {\n    return;\n  }\n\n  const properties: Record<string, unknown> = {};\n  for (const variable of [...variables].sort()) {\n    properties[variable] = { type: 'string' };\n  }\n\n  return {\n    type: 'object',\n    properties,\n    required: [...variables].sort(),\n    additionalProperties: false,\n  };\n}\n"],"mappings":";;;;;;AAcA,MAAM,WAAW,4BAA4B,MAAM,OAAO,UAAU,MAAM,cAAc,QAAQ;AAEhG,MAAM,0BAA0B,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC;AAE3E,MAAM,2BAA2B,EAAE,OAAO,EACxC,sBAAsB,wBAAwB,UAAU,EACzD,CAAC;AAEF,MAAM,+BAA+B,EAAE,OAAO,EAC5C,sBAAsB,wBAAwB,UAAU,EACzD,CAAC;AAEF,MAAM,sBAAsB,EAAE,aAAa;CACzC,iBAAiB,EAAE,QAAQ,CAAC,UAAU;CACtC,GAAG,SAAS;CACZ,oBAAoB,yBAAyB,UAAU;CACvD,wBAAwB,6BAA6B,UAAU;CAG/D,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,kBAAkB,EAAE,QAAQ,CAAC,UAAU;CACvC,QAAQ,EAAE,SAAS;CACnB,MAAM,EAAE,QAAQ,CAAC,SAAS;CAC1B,SAAS,EAAE,SAAS;CACpB,aAAa,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,SAAS,CAAC,CAAC,UAAU;CACzD,cAAc,EAAE,SAAS;CACzB,gBAAgB,EAAE,SAAS;CAC5B,CAAC;AAKF,SAAgB,gCAAgC,MAAsC;CACpF,MAAM,SAAS,oBAAoB,UAAU,KAAK;AAClD,KAAI,CAAC,OAAO,QACV,OAAM,IAAI,MAAM,0CAA0C,EAAE,cAAc,OAAO,MAAM,GAAG;CAG5F,MAAM,UAAU,uBAAuB;CAEvC,MAAM,SAAS,OAAO;CACtB,MAAM,aAAa,QAAQ,iBAAiB,gCAAgC,IAAI,EAC9E,WAAW,MACZ,CAAC;AAEF,KAAI,wBAAwB,OAAO,CACjC,QAAO,oCAAoC,YAAY,OAAO;AAGhE,KAAI,sBAAsB,OAAO,CAC/B,QAAO,kCAAkC,YAAY,OAAO;CAG9D,MAAM,4BACH,OAAO,OAAO,qBAAqB,YAAY,OAAO,iBAAiB,SAAS,IAC7E,OAAO,mBACP,WAAc,wBAAwB,OAAO,QAAQ;CAC3D,MAAM,0BAA0B,+BAA+B,OAAO,iBAAiB;CACvF,MAAM,wBACJ,CAAC,cAAc,OAAO,cAAc,IAAI,CAAC,2BACrC,8CAA8C,wBAAwB,GACtE;CACN,MAAM,gBAAgB,cAAc,OAAO,cAAc,GACrD,OAAO,gBACP;CACJ,MAAM,mBAAmB,wBAAwB,QAAQ,QAAQ,cAAc,CAAC;CAChF,MAAM,4BAA4B,QAAQ,iBAAiB,IAAI,cAAc,cAAc;CAC3F,MAAM,mBAAmB,8BAA8B,OAAO,iBAAiB;CAC/E,MAAM,2BAA2B,gCAC/B,kBACA,OAAO,oBAAoB,qBAC5B;CACD,MAAM,cAAc,CAAC,gBAAgB;AACrC,KAAI,0BACF,aAAY,QAAQ,UAAU;AAEhC,KAAI,iBAAiB,SAAS,EAC5B,aAAY,OAAO,YAAY,SAAS,GAAG,GAAG,kBAAkB;AAGlE,YAAW,qBAAqB;EAC9B,cAAc;EACd,iBAAiB;EAClB,CAAC;CAEF,MAAM,qBAAqB,iBAAiB,MAAM,eAEhD,cAAc,WAAW,KAAK,eAAe,CAC9C;AACD,KAAI,6BAA6B,mBAC/B,YAAW,qBAAqB;EAC9B,cAAc,CAAC,IAAI;EACnB,iBAAiB;EAClB,CAAC;AAGJ,MAAK,MAAM,CAAC,cAAc,4BAA4B,0BAA0B;EAC9E,MAAM,0BACJ,OAAO,wBAAwB,uBAAuB,iBAAiB;AACzE,aAAW,qBAAqB;GAC9B,cAAc,CAAC,wBAAwB;GACvC,iBAAiB,kBAAkB;GACpC,CAAC;;AAEJ,KAAI,6BAA6B,oBAAoB,eAAe;EAClE,MAAM,EAAE,cAAc,kBAAkB,yBAAyB;GAC/D;GACA,YAAY;GACZ,YAAY;GACZ,cAAc;GACf,CAAC;AAEF,gBAAc,sBAAsB;GAClC,MAAM;GACN,aAAa,uBAAuB,cAAc;GACnD,CAAC;;AAGJ,MAAK,MAAM,mBAAmB,kBAAkB;EAC9C,MAAM,EAAE,cAAc,sBAAsB,yBAAyB;GACnE;GACA,YAAY;GACZ,cAAc,gBAAgB;GAC/B,CAAC;AACF,uBACE,mBACA,gBAAgB,MAChB,0BACA,iBACD;;CAGH,MAAM,EAAE,iBAAiB,yBAAyB;EAChD;EACA,YAAY;EACZ,cAJ2B,4BAA4B,OAAO,gBAAgB;EAK9E,YAAY;EACb,CAAC;AAEF,oBAAmB,cAAc,QAAQ,iBAAiB;AAE1D,QAAO;;AAGT,SAAS,mBACP,cACA,MACA,kBACA;AACA,KAAI,KAAK,OAAO,OACd,mBAAkB,cAAc,MAAM,KAAK,GAAG;AAGhD,KAAI,iBACF,cAAa,sBAAsB;EACjC,MAAM;EACN,aAAa;EACd,CAAC;AAGJ,KAAI,KAAK,oBAAoB,OAAO,KAAK,KAAK,iBAAiB,CAAC,SAAS,GAAG;EAK1E,MAAM,yBAJ2B,aAAa,sBAAsB;GAClE,MAAM;GACN,aAAa;GACd,CAAC,CACsD,4BACtD,WAAW,wBACZ;AAED,OAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,KAAK,iBAAiB,EAAE;GAChE,MAAM,YAAY,gCAAgC,KAAK,MAAM;AAC7D,OAAI,CAAC,UACH;AAGF,0BAAuB,sBAAsB;IAC3C,MAAM,mBAAmB,IAAI;IAC7B,aAAa;IACd,CAAC;;;;AAKR,SAAS,wBAAwB,SAAmD;AAClF,KAAI,CAAC,QACH;AAEF,KAAI,OAAO,YAAY,SACrB,QAAO;AAET,KAAI,OAAO,QAAQ,OAAO,SACxB,QAAO,QAAQ;AAEjB,KAAI,OAAO,QAAQ,SAAS,SAC1B,QAAO,QAAQ;;AAKnB,SAAS,wBACP,MACA,kBACoB;AACpB,KAAI,OAAO,KAAK,qBAAqB,YAAY,KAAK,iBACpD,QAAO,KAAK;CAGd,MAAM,aAAa,wBAAwB,KAAK,QAAQ;AACxD,KAAI,WACF,QAAO,sBAAsB,WAAW;AAG1C,KAAI,iBACF,QAAO,GAAG,4BAA4B,KAAK,gBAAgB,CAAC;;AAMhE,SAAS,sBAAsB,OAAyB;AACtD,KAAI,CAAC,cAAc,MAAM,CACvB,QAAO;AAGT,QAAO,MAAM,gBAAgB,UAAa,MAAM,mBAAmB;;AAGrE,SAAS,8BAA8B,kBAA4C;AACjF,KAAI,CAAC,iBACH,QAAO,EAAE;AAGX,QAAO,OAAO,QAAQ,iBAAiB,CACpC,QAAQ,GAAG,WAAW,sBAAsB,MAAM,CAAC,CACnD,KAAK,CAAC,KAAK,WAAW;AAGrB,SAAO;GACL;GACA,cAHA,gCAAgC,KAAK,MAAM,IAAI,sBAAsB,IAAI;GAIzE,MAAM;GACP;GACD;;AAGN,SAAS,qBACP,cACA,qBACA,0BACA,kBACA;CACA,MAAM,EAAE,iBAAiB,gBAAgB,uBAAuB,GAAG,SAAS,cAC1E,oBACD,GACG,sBACA,EAAE;CACN,MAAM,iBAAiB,uBAAuB,MAAM,iBAAiB;AACrE,MAAK,MAAM,CAAC,GAAG,MAAM,OAAO,QAAQ;EAClC,IAAI;EACJ,GAAG;EACJ,CAAC,CACA,KAAI,MAAM,KACR,kBAAiB,cAAc,GAAG,EAAE;AAGxC,KAAI,eACF,cAAa,sBAAsB;EACjC,MAAM;EAEN,aAAa,uBAAuB,eAAe;EACpD,CAAC;AAGJ,KACE,OAAO,0BAA0B,YACjC,0BAA0B,IAAI,sBAAsB,EACpD;AACA,eAAa,sBAAsB;GACjC,MAAM;GACN,aAAa,yBAAyB,IAAI,sBAAsB;GACjE,CAAC;AACF;;AAGF,KAAI,OAAO,0BAA0B,SACnC,mBAAkB,cAAc,yBAAyB,sBAAsB;;AAInF,MAAM,wBAAwB;AAC9B,MAAM,gCACJ;AAEF,SAAS,uBAA0B,OAAU,kBAA8B;AACzE,KAAI,CAAC,iBACH,QAAO;AAGT,KAAI,OAAO,UAAU,SAOnB,QANiC,MAAM,QACrC,wBACC,GAAG,iBAAyB;AAC3B,SAAO,MAAM,iBAAiB,cAAc,KAAK,UAAU,aAAa,CAAC;GAE5E,CAC+B,QAC9B,gCACC,GAAG,IAAY,KAAa,iBAAyB;AACpD,SAAO,MAAM,iBAAiB,cAAc,KAAK,UAAU,aAAa,CAAC;GAE5E;AAGH,KAAI,MAAM,QAAQ,MAAM,CACtB,QAAO,MAAM,KAAK,UAAU,uBAAuB,OAAO,iBAAiB,CAAC;AAG9E,KAAI,cAAc,MAAM,CACtB,QAAO,OAAO,YACZ,OAAO,QAAQ,MAAM,CAAC,KAAK,CAAC,KAAK,gBAAgB,CAC/C,KACA,uBAAuB,YAAY,iBAAiB,CACrD,CAAC,CACH;AAGH,QAAO;;AAGT,SAAS,oCACP,YACA,MACY;CACZ,MAAM,aAAa;AACnB,YAAW,qBAAqB;EAC9B,cAAc,CAAC,WAAW;EAC1B,iBAAiB;EAClB,CAAC;AACF,YAAW,qBAAqB;EAC9B,cAAc,CAAC,IAAI;EACnB,iBAAiB;EAClB,CAAC;CAGF,MAAM,EAAE,iBAAiB,yBAAyB;EAChD;EACA;EACA,cAJqB,4BAA4B,KAAK,gBAAgB;EAKvE,CAAC;AAEF,cAAa,sBAAsB;EACjC,MAAM;EACN,aAAa,uBAAuB,KAAK,OAAO;EACjD,CAAC;AACF,QAAO;;AAGT,SAAS,wBACP,OACoE;AACpE,QAAO,cAAc,MAAM,OAAO;;AAGpC,SAAS,kCACP,YACA,MACY;CACZ,MAAM,aAAa;AACnB,YAAW,qBAAqB;EAC9B,cAAc,CAAC,WAAW;EAC1B,iBAAiB;EAClB,CAAC;AAEF,KAAI,cAAc,KAAK,eAAe,CACpC,YAAW,qBAAqB;EAC9B,cAAc,CAAC,IAAI;EACnB,iBAAiB;EAClB,CAAC;CAGJ,MAAM,2CAA2B,IAAI,KAAqB;AAC1D,KAAI,OAAQ,KAAiC,0BAA0B,UAAU;EAC/E,MAAM,wBAAyB,KAAiC;EAChE,MAAM,0BACJ,KAAK,oBAAoB,uBAAuB,0BAChD,sBAAsB,sBAAsB;EAC9C,MAAM,0BACJ,KAAK,wBAAwB,uBAAuB,0BACpD;AACF,2BAAyB,IAAI,uBAAuB,wBAAwB;AAC5E,aAAW,qBAAqB;GAC9B,cAAc,CAAC,wBAAwB;GACvC,iBAAiB,kBAAkB;GACpC,CAAC;;CAIJ,MAAM,EAAE,iBAAiB,yBAAyB;EAChD;EACA;EACA,cAJmB,4BAA4B,KAAK,gBAAgB;EAKrE,CAAC;AAEF,sBAAqB,cAAc,MAAM,yBAAyB;AAClE,QAAO;;AAGT,SAAS,uBAAuB,QAAyC;AACvE,QAAO,2BAA2B,QAAQ,EACxC,mBAAmB,EAAE,QAAQ,QAAQ,EACtC,CAAC;;AAGJ,SAAS,gCAAgC,KAAa,OAAoC;AACxF,KAAI,OAAO,UAAU,SACnB,QAAO,sBAAsB,MAAM;AAGrC,KAAI,CAAC,cAAc,MAAM,CACvB;AAGF,KAAI,OAAO,MAAM,OAAO,SACtB,QAAO,sBAAsB,MAAM,GAAG;AAExC,KAAI,OAAO,MAAM,SAAS,SACxB,QAAO,sBAAsB,MAAM,KAAK;AAE1C,KAAI,OAAO,MAAM,QAAQ,SACvB,QAAO,sBAAsB,MAAM,IAAI;AAEzC,KAAI,OAAO,MAAM,aAAa,SAC5B,QAAO,sBAAsB,MAAM,SAAS;AAG9C,KAAI,MAAM,eAAe,MAAM,eAC7B,QAAO,sBAAsB,IAAI;AAGnC,QAAO,sBAAsB,IAAI;;AAGnC,MAAM,8BAA8B;AAEpC,MAAM,wBAAwB;AAE9B,SAAS,gCACP,kBACA,mBACqB;CACrB,MAAM,2CAA2B,IAAI,KAAqB;AAE1D,MAAK,MAAM,mBAAmB,kBAAkB;EAC9C,MAAM,sBAAsB,cAAc,gBAAgB,KAAK,GAC3D,gBAAgB,OAChB;EACJ,MAAM,wBACJ,uBAAuB,OAAO,oBAAoB,0BAA0B,WACxE,oBAAoB,wBACpB;AACN,MAAI,CAAC,yBAAyB,yBAAyB,IAAI,sBAAsB,CAC/E;AAGF,2BAAyB,IACvB,uBACA,oBAAoB,0BAA0B,sBAAsB,sBAAsB,CAC3F;;AAGH,QAAO;;AAGT,SAAS,+BAA+B,kBAAyD;CAC/F,MAAM,4BAAY,IAAI,KAAa;AACnC,yCAAwC,kBAAkB,UAAU;AACpE,QAAO;;AAGT,SAAS,wCAAwC,OAAgB,WAA8B;AAC7F,KAAI,OAAO,UAAU,UAAU;AAC7B,OAAK,MAAM,SAAS,MAAM,SAAS,sBAAsB,CACvD,KAAI,MAAM,GACR,WAAU,IAAI,MAAM,GAAG;AAG3B,OAAK,MAAM,SAAS,MAAM,SAAS,8BAA8B,CAC/D,KAAI,MAAM,GACR,WAAU,IAAI,MAAM,GAAG;AAG3B;;AAGF,KAAI,MAAM,QAAQ,MAAM,EAAE;AACxB,OAAK,MAAM,SAAS,MAClB,yCAAwC,OAAO,UAAU;AAE3D;;AAGF,KAAI,cAAc,MAAM,CACtB,MAAK,MAAM,cAAc,OAAO,OAAO,MAAM,CAC3C,yCAAwC,YAAY,UAAU;;AAKpE,SAAS,8CACP,WACqC;AACrC,KAAI,CAAC,UAAU,KACb;CAGF,MAAM,aAAsC,EAAE;AAC9C,MAAK,MAAM,YAAY,CAAC,GAAG,UAAU,CAAC,MAAM,CAC1C,YAAW,YAAY,EAAE,MAAM,UAAU;AAG3C,QAAO;EACL,MAAM;EACN;EACA,UAAU,CAAC,GAAG,UAAU,CAAC,MAAM;EAC/B,sBAAsB;EACvB"}

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

@@ -1,22 +1,25 @@
-import { addValueToObject, createFactoryDefinition, toCamelCase } from "../utils.js";
+import { addValueToObject, createFactoryDefinition, toCredentialReferenceName } from "../utils.js";
+import { FullProjectDefinitionSchema } from "@inkeep/agents-core";
 import { z } from "zod";
 
 //#region src/commands/pull-v4/generators/credential-generator.ts
-const CredentialSchema = z.object({
+const MySchema = FullProjectDefinitionSchema.shape.credentialReferences.unwrap().valueType.omit({
+	id: true,
+	createdBy: true,
+	toolId: true,
+	userId: true
+});
+const CredentialSchema = z.strictObject({
 	credentialId: z.string().nonempty(),
-	name: z.string().nonempty(),
-	type: z.string().nonempty(),
-	credentialStoreId: z.string().nonempty(),
-	description: z.string().optional(),
-	retrievalParams: z.unknown().optional()
+	...MySchema.shape
 });
-function generateCredentialDefinition(data) {
+function generateCredentialDefinition({ tenantId, id, projectId, createdBy, createdAt, updatedAt, toolId, userId, ...data }) {
 	const result = CredentialSchema.safeParse(data);
 	if (!result.success) throw new Error(`Validation failed for credential:\n${z.prettifyError(result.error)}`);
 	const parsed = result.data;
 	const { sourceFile, configObject } = createFactoryDefinition({
 		importName: "credential",
-		variableName: toCamelCase(parsed.credentialId)
+		variableName: toCredentialReferenceName(parsed.name)
 	});
 	const { credentialId, ...rest } = parsed;
 	for (const [k, v] of Object.entries({
@@ -27,4 +30,5 @@ function generateCredentialDefinition(data) {
 }
 
 //#endregion
-export { generateCredentialDefinition };
+export { generateCredentialDefinition };
+//# sourceMappingURL=credential-generator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"credential-generator.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/credential-generator.ts"],"sourcesContent":["import { FullProjectDefinitionSchema } from '@inkeep/agents-core';\nimport type { SourceFile } from 'ts-morph';\nimport { z } from 'zod';\nimport { addValueToObject, createFactoryDefinition, toCredentialReferenceName } from '../utils';\n\nconst MySchema = FullProjectDefinitionSchema.shape.credentialReferences.unwrap().valueType.omit({\n  id: true,\n  createdBy: true,\n  toolId: true,\n  userId: true,\n});\n\nconst CredentialSchema = z.strictObject({\n  credentialId: z.string().nonempty(),\n  ...MySchema.shape,\n});\n\ntype CredentialInput = z.input<typeof CredentialSchema>;\n\nexport function generateCredentialDefinition({\n  tenantId,\n  id,\n  projectId,\n  createdBy,\n  createdAt,\n  updatedAt,\n  toolId,\n  userId,\n  ...data\n}: CredentialInput & Record<string, unknown>): SourceFile {\n  const result = CredentialSchema.safeParse(data);\n  if (!result.success) {\n    throw new Error(`Validation failed for credential:\\n${z.prettifyError(result.error)}`);\n  }\n\n  const parsed = result.data;\n  const credentialVarName = toCredentialReferenceName(parsed.name);\n  const { sourceFile, configObject } = createFactoryDefinition({\n    importName: 'credential',\n    variableName: credentialVarName,\n  });\n\n  const { credentialId, ...rest } = parsed;\n\n  for (const [k, v] of Object.entries({ id: credentialId, ...rest })) {\n    addValueToObject(configObject, k, v);\n  }\n  return sourceFile;\n}\n"],"mappings":";;;;;AAKA,MAAM,WAAW,4BAA4B,MAAM,qBAAqB,QAAQ,CAAC,UAAU,KAAK;CAC9F,IAAI;CACJ,WAAW;CACX,QAAQ;CACR,QAAQ;CACT,CAAC;AAEF,MAAM,mBAAmB,EAAE,aAAa;CACtC,cAAc,EAAE,QAAQ,CAAC,UAAU;CACnC,GAAG,SAAS;CACb,CAAC;AAIF,SAAgB,6BAA6B,EAC3C,UACA,IACA,WACA,WACA,WACA,WACA,QACA,QACA,GAAG,QACqD;CACxD,MAAM,SAAS,iBAAiB,UAAU,KAAK;AAC/C,KAAI,CAAC,OAAO,QACV,OAAM,IAAI,MAAM,sCAAsC,EAAE,cAAc,OAAO,MAAM,GAAG;CAGxF,MAAM,SAAS,OAAO;CAEtB,MAAM,EAAE,YAAY,iBAAiB,wBAAwB;EAC3D,YAAY;EACZ,cAHwB,0BAA0B,OAAO,KAAK;EAI/D,CAAC;CAEF,MAAM,EAAE,cAAc,GAAG,SAAS;AAElC,MAAK,MAAM,CAAC,GAAG,MAAM,OAAO,QAAQ;EAAE,IAAI;EAAc,GAAG;EAAM,CAAC,CAChE,kBAAiB,cAAc,GAAG,EAAE;AAEtC,QAAO"}

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

@@ -1,23 +1,20 @@
 import { addStringProperty, addValueToObject, convertJsonSchemaToZodSafe, createFactoryDefinition, toCamelCase } from "../utils.js";
+import { FullProjectDefinitionSchema } from "@inkeep/agents-core";
 import { z } from "zod";
 
 //#region src/commands/pull-v4/generators/data-component-generator.ts
-const DataComponentSchema = z.object({
+const MySchema = FullProjectDefinitionSchema.shape.dataComponents.unwrap().valueType.omit({ id: true });
+const DataComponentSchema = z.strictObject({
 	dataComponentId: z.string().nonempty(),
-	name: z.string().nonempty(),
-	description: z.string().nullable().optional(),
-	props: z.unknown().optional(),
-	schema: z.unknown().optional(),
-	render: z.looseObject({
-		component: z.string().optional(),
-		mockData: z.looseObject({}).optional()
-	}).nullable().optional()
+	...MySchema.shape,
+	description: z.preprocess((v) => v || void 0, MySchema.shape.description),
+	props: z.unknown()
 });
-function generateDataComponentDefinition(data) {
+function generateDataComponentDefinition({ tenantId, id, projectId, createdAt, updatedAt, ...data }) {
 	const result = DataComponentSchema.safeParse(data);
 	if (!result.success) throw new Error(`Validation failed for data component:\n${z.prettifyError(result.error)}`);
 	const parsed = result.data;
-	const props = parsed.props !== void 0 ? parsed.props : parsed.schema;
+	const props = parsed.props;
 	const { sourceFile, configObject } = createFactoryDefinition({
 		importName: "dataComponent",
 		variableName: toCamelCase(parsed.dataComponentId)
@@ -47,4 +44,5 @@ function addRenderProperty(configObject, render) {
 }
 
 //#endregion
-export { generateDataComponentDefinition };
+export { generateDataComponentDefinition };
+//# sourceMappingURL=data-component-generator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"data-component-generator.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/data-component-generator.ts"],"sourcesContent":["import { FullProjectDefinitionSchema } from '@inkeep/agents-core';\nimport type { ObjectLiteralExpression, SourceFile } from 'ts-morph';\nimport { z } from 'zod';\nimport {\n  addStringProperty,\n  addValueToObject,\n  convertJsonSchemaToZodSafe,\n  createFactoryDefinition,\n  toCamelCase,\n} from '../utils';\n\nconst MySchema = FullProjectDefinitionSchema.shape.dataComponents.unwrap().valueType.omit({\n  id: true,\n});\n\nconst DataComponentSchema = z.strictObject({\n  dataComponentId: z.string().nonempty(),\n  ...MySchema.shape,\n  description: z.preprocess((v) => v || undefined, MySchema.shape.description),\n  // Each property must have a \"description\" for LLM compatibility\n  props: z.unknown(),\n});\n\ntype DataComponentInput = z.input<typeof DataComponentSchema>;\ntype DataComponentOutput = z.output<typeof DataComponentSchema>;\n\nexport function generateDataComponentDefinition({\n  tenantId,\n  id,\n  projectId,\n  createdAt,\n  updatedAt,\n  ...data\n}: DataComponentInput & Record<string, unknown>): SourceFile {\n  const result = DataComponentSchema.safeParse(data);\n  if (!result.success) {\n    throw new Error(`Validation failed for data component:\\n${z.prettifyError(result.error)}`);\n  }\n\n  const parsed = result.data;\n  const props = parsed.props;\n  const { sourceFile, configObject } = createFactoryDefinition({\n    importName: 'dataComponent',\n    variableName: toCamelCase(parsed.dataComponentId),\n  });\n\n  if (props !== undefined) {\n    sourceFile.addImportDeclaration({\n      namedImports: ['z'],\n      moduleSpecifier: 'zod',\n    });\n  }\n\n  writeDataComponentConfig(configObject, parsed, props);\n\n  return sourceFile;\n}\n\nfunction writeDataComponentConfig(\n  configObject: ObjectLiteralExpression,\n  data: DataComponentOutput,\n  props: unknown\n): void {\n  addStringProperty(configObject, 'id', data.dataComponentId);\n  addStringProperty(configObject, 'name', data.name);\n\n  if (typeof data.description === 'string') {\n    addStringProperty(configObject, 'description', data.description);\n  }\n\n  if (props !== undefined) {\n    configObject.addPropertyAssignment({\n      name: 'props',\n      initializer: convertJsonSchemaToZodSafe(props),\n    });\n  }\n\n  if (data.render) {\n    addRenderProperty(configObject, data.render);\n  }\n}\n\nfunction addRenderProperty(\n  configObject: ObjectLiteralExpression,\n  render: NonNullable<DataComponentOutput['render']>\n): void {\n  if (render.component) {\n    addValueToObject(configObject, 'render', {\n      component: render.component,\n      mockData: render.mockData,\n    });\n  }\n}\n"],"mappings":";;;;;AAWA,MAAM,WAAW,4BAA4B,MAAM,eAAe,QAAQ,CAAC,UAAU,KAAK,EACxF,IAAI,MACL,CAAC;AAEF,MAAM,sBAAsB,EAAE,aAAa;CACzC,iBAAiB,EAAE,QAAQ,CAAC,UAAU;CACtC,GAAG,SAAS;CACZ,aAAa,EAAE,YAAY,MAAM,KAAK,QAAW,SAAS,MAAM,YAAY;CAE5E,OAAO,EAAE,SAAS;CACnB,CAAC;AAKF,SAAgB,gCAAgC,EAC9C,UACA,IACA,WACA,WACA,WACA,GAAG,QACwD;CAC3D,MAAM,SAAS,oBAAoB,UAAU,KAAK;AAClD,KAAI,CAAC,OAAO,QACV,OAAM,IAAI,MAAM,0CAA0C,EAAE,cAAc,OAAO,MAAM,GAAG;CAG5F,MAAM,SAAS,OAAO;CACtB,MAAM,QAAQ,OAAO;CACrB,MAAM,EAAE,YAAY,iBAAiB,wBAAwB;EAC3D,YAAY;EACZ,cAAc,YAAY,OAAO,gBAAgB;EAClD,CAAC;AAEF,KAAI,UAAU,OACZ,YAAW,qBAAqB;EAC9B,cAAc,CAAC,IAAI;EACnB,iBAAiB;EAClB,CAAC;AAGJ,0BAAyB,cAAc,QAAQ,MAAM;AAErD,QAAO;;AAGT,SAAS,yBACP,cACA,MACA,OACM;AACN,mBAAkB,cAAc,MAAM,KAAK,gBAAgB;AAC3D,mBAAkB,cAAc,QAAQ,KAAK,KAAK;AAElD,KAAI,OAAO,KAAK,gBAAgB,SAC9B,mBAAkB,cAAc,eAAe,KAAK,YAAY;AAGlE,KAAI,UAAU,OACZ,cAAa,sBAAsB;EACjC,MAAM;EACN,aAAa,2BAA2B,MAAM;EAC/C,CAAC;AAGJ,KAAI,KAAK,OACP,mBAAkB,cAAc,KAAK,OAAO;;AAIhD,SAAS,kBACP,cACA,QACM;AACN,KAAI,OAAO,UACT,kBAAiB,cAAc,UAAU;EACvC,WAAW,OAAO;EAClB,UAAU,OAAO;EAClB,CAAC"}

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

@@ -1,22 +1,15 @@
-import { addStringProperty, createFactoryDefinition, formatInlineLiteral, formatPropertyName, isPlainObject } from "../utils.js";
+import { addValueToObject, createFactoryDefinition } from "../utils.js";
+import { FullProjectDefinitionSchema } from "@inkeep/agents-core";
 import { z } from "zod";
-import { SyntaxKind } from "ts-morph";
 
 //#region src/commands/pull-v4/generators/environment-generator.ts
-const EnvironmentSettingsSchema = z.looseObject({ credentials: z.record(z.string(), z.unknown()).nullable().optional() });
+const EnvironmentCredentialSchema = FullProjectDefinitionSchema.shape.credentialReferences.unwrap().valueType.omit({
+	createdBy: true,
+	toolId: true,
+	userId: true
+});
+const EnvironmentSettingsSchema = z.looseObject({ credentials: z.record(z.string(), EnvironmentCredentialSchema).nullable().optional() });
 const EnvironmentIndexSchema = z.array(z.string());
-function generateEnvironmentSettingsImports(environmentData) {
-	const imports = [`import { registerEnvironmentSettings } from '@inkeep/agents-sdk';`];
-	if (needsCredentialStoreType(environmentData)) imports.push(`import { CredentialStoreType } from '@inkeep/agents-core';`);
-	return imports;
-}
-function generateEnvironmentIndexImports(environments) {
-	const result = EnvironmentIndexSchema.safeParse(environments);
-	if (!result.success) throw new Error(`Validation failed for environments index:\n${z.prettifyError(result.error)}`);
-	const imports = [`import { createEnvironmentSettings } from '@inkeep/agents-sdk';`];
-	for (const environmentName of result.data) imports.push(`import { ${environmentName} } from './${environmentName}.env';`);
-	return imports;
-}
 function generateEnvironmentSettingsDefinition(environmentName, environmentData) {
 	const environmentNameResult = z.string().nonempty().safeParse(environmentName);
 	if (!environmentNameResult.success) throw new Error(`Validation failed for environment name:\n${z.prettifyError(environmentNameResult.error)}`);
@@ -27,12 +20,7 @@ function generateEnvironmentSettingsDefinition(environmentName, environmentData)
 		importName: "registerEnvironmentSettings",
 		variableName: environmentNameResult.data
 	});
-	const hasCredentialStoreType = needsCredentialStoreType(parsed);
-	if (hasCredentialStoreType) sourceFile.addImportDeclaration({
-		namedImports: ["CredentialStoreType"],
-		moduleSpecifier: "@inkeep/agents-core"
-	});
-	addCredentialsProperty(configObject, parsed.credentials, hasCredentialStoreType);
+	addValueToObject(configObject, "credentials", parsed.credentials ?? {});
 	return sourceFile;
 }
 function generateEnvironmentIndexDefinition(environments) {
@@ -49,75 +37,7 @@ function generateEnvironmentIndexDefinition(environments) {
 	for (const environmentName of result.data) configObject.addShorthandPropertyAssignment({ name: environmentName });
 	return sourceFile;
 }
-function generateEnvironmentSettingsFile(environmentName, environmentData) {
-	return generateEnvironmentSettingsDefinition(environmentName, environmentData);
-}
-function generateEnvironmentIndexFile(environments) {
-	return generateEnvironmentIndexDefinition(environments);
-}
-function needsCredentialStoreType(environmentData) {
-	if (!isPlainObject(environmentData.credentials)) return false;
-	return Object.values(environmentData.credentials).some((credential) => isPlainObject(credential) && typeof credential.type === "string" && [
-		"memory",
-		"env",
-		"keychain"
-	].includes(credential.type));
-}
-function addCredentialsProperty(configObject, credentials, hasCredentialStoreType) {
-	if (!isPlainObject(credentials) || !Object.keys(credentials).length) {
-		configObject.addPropertyAssignment({
-			name: "credentials",
-			initializer: "{}"
-		});
-		return;
-	}
-	const credentialsObject = configObject.addPropertyAssignment({
-		name: "credentials",
-		initializer: "{}"
-	}).getInitializerIfKindOrThrow(SyntaxKind.ObjectLiteralExpression);
-	for (const [credentialKey, credentialValue] of Object.entries(credentials)) {
-		if (!isPlainObject(credentialValue)) continue;
-		writeCredential(credentialsObject.addPropertyAssignment({
-			name: formatPropertyName(credentialKey),
-			initializer: "{}"
-		}).getInitializerIfKindOrThrow(SyntaxKind.ObjectLiteralExpression), credentialValue, hasCredentialStoreType);
-	}
-}
-function writeCredential(credentialObject, credentialData, hasCredentialStoreType) {
-	if (credentialData.id !== void 0) addStringProperty(credentialObject, "id", credentialData.id);
-	if (credentialData.name !== void 0 && credentialData.name !== null) addStringProperty(credentialObject, "name", credentialData.name);
-	if (credentialData.type !== void 0) if (hasCredentialStoreType && typeof credentialData.type === "string" && [
-		"memory",
-		"env",
-		"keychain"
-	].includes(credentialData.type)) credentialObject.addPropertyAssignment({
-		name: "type",
-		initializer: `CredentialStoreType.${credentialData.type}`
-	});
-	else credentialObject.addPropertyAssignment({
-		name: "type",
-		initializer: formatInlineLiteral(credentialData.type)
-	});
-	if (credentialData.credentialStoreId !== void 0) addStringProperty(credentialObject, "credentialStoreId", credentialData.credentialStoreId);
-	if (credentialData.description !== void 0 && credentialData.description !== null) addStringProperty(credentialObject, "description", credentialData.description);
-	addRetrievalParams(credentialObject, credentialData.retrievalParams);
-}
-function addRetrievalParams(credentialObject, retrievalParams) {
-	if (!isPlainObject(retrievalParams)) return;
-	const retrievalParamsProperty = credentialObject.addPropertyAssignment({
-		name: "retrievalParams",
-		initializer: "{}"
-	});
-	const retrievalParamsObject = retrievalParamsProperty.getInitializerIfKindOrThrow(SyntaxKind.ObjectLiteralExpression);
-	for (const [key, value] of Object.entries(retrievalParams)) {
-		if (value === void 0 || value === null) continue;
-		retrievalParamsObject.addPropertyAssignment({
-			name: formatPropertyName(key),
-			initializer: formatInlineLiteral(value)
-		});
-	}
-	if (!retrievalParamsObject.getProperties().length) retrievalParamsProperty.remove();
-}
 
 //#endregion
-export { generateEnvironmentIndexDefinition, generateEnvironmentIndexFile, generateEnvironmentIndexImports, generateEnvironmentSettingsDefinition, generateEnvironmentSettingsFile, generateEnvironmentSettingsImports };
+export { generateEnvironmentIndexDefinition, generateEnvironmentSettingsDefinition };
+//# sourceMappingURL=environment-generator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"environment-generator.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/environment-generator.ts"],"sourcesContent":["import { FullProjectDefinitionSchema } from '@inkeep/agents-core';\nimport type { SourceFile } from 'ts-morph';\nimport { z } from 'zod';\nimport { addValueToObject, createFactoryDefinition } from '../utils';\n\nconst EnvironmentCredentialSchema = FullProjectDefinitionSchema.shape.credentialReferences\n  .unwrap()\n  .valueType.omit({\n    createdBy: true,\n    toolId: true,\n    userId: true,\n  });\n\nconst EnvironmentSettingsSchema = z.looseObject({\n  credentials: z.record(z.string(), EnvironmentCredentialSchema).nullable().optional(),\n});\n\nconst EnvironmentIndexSchema = z.array(z.string());\n\ntype EnvironmentSettingsInput = z.input<typeof EnvironmentSettingsSchema>;\ntype EnvironmentIndexInput = z.input<typeof EnvironmentIndexSchema>;\n\nexport function generateEnvironmentSettingsDefinition(\n  environmentName: string,\n  environmentData: EnvironmentSettingsInput\n): SourceFile {\n  const environmentNameSchema = z.string().nonempty();\n  const environmentNameResult = environmentNameSchema.safeParse(environmentName);\n  if (!environmentNameResult.success) {\n    throw new Error(\n      `Validation failed for environment name:\\n${z.prettifyError(environmentNameResult.error)}`\n    );\n  }\n\n  const result = EnvironmentSettingsSchema.safeParse(environmentData);\n  if (!result.success) {\n    throw new Error(\n      `Validation failed for environment settings:\\n${z.prettifyError(result.error)}`\n    );\n  }\n\n  const parsed = result.data;\n  const { sourceFile, configObject } = createFactoryDefinition({\n    importName: 'registerEnvironmentSettings',\n    variableName: environmentNameResult.data,\n  });\n\n  addValueToObject(configObject, 'credentials', parsed.credentials ?? {});\n\n  return sourceFile;\n}\n\nexport function generateEnvironmentIndexDefinition(\n  environments: EnvironmentIndexInput\n): SourceFile {\n  const result = EnvironmentIndexSchema.safeParse(environments);\n  if (!result.success) {\n    throw new Error(`Validation failed for environments index:\\n${z.prettifyError(result.error)}`);\n  }\n\n  const { sourceFile, configObject } = createFactoryDefinition({\n    importName: 'createEnvironmentSettings',\n    variableName: 'envSettings',\n  });\n\n  for (const environmentName of result.data) {\n    sourceFile.addImportDeclaration({\n      namedImports: [environmentName],\n      moduleSpecifier: `./${environmentName}.env`,\n    });\n  }\n\n  for (const environmentName of result.data) {\n    configObject.addShorthandPropertyAssignment({ name: environmentName });\n  }\n\n  return sourceFile;\n}\n"],"mappings":";;;;;AAKA,MAAM,8BAA8B,4BAA4B,MAAM,qBACnE,QAAQ,CACR,UAAU,KAAK;CACd,WAAW;CACX,QAAQ;CACR,QAAQ;CACT,CAAC;AAEJ,MAAM,4BAA4B,EAAE,YAAY,EAC9C,aAAa,EAAE,OAAO,EAAE,QAAQ,EAAE,4BAA4B,CAAC,UAAU,CAAC,UAAU,EACrF,CAAC;AAEF,MAAM,yBAAyB,EAAE,MAAM,EAAE,QAAQ,CAAC;AAKlD,SAAgB,sCACd,iBACA,iBACY;CAEZ,MAAM,wBADwB,EAAE,QAAQ,CAAC,UAAU,CACC,UAAU,gBAAgB;AAC9E,KAAI,CAAC,sBAAsB,QACzB,OAAM,IAAI,MACR,4CAA4C,EAAE,cAAc,sBAAsB,MAAM,GACzF;CAGH,MAAM,SAAS,0BAA0B,UAAU,gBAAgB;AACnE,KAAI,CAAC,OAAO,QACV,OAAM,IAAI,MACR,gDAAgD,EAAE,cAAc,OAAO,MAAM,GAC9E;CAGH,MAAM,SAAS,OAAO;CACtB,MAAM,EAAE,YAAY,iBAAiB,wBAAwB;EAC3D,YAAY;EACZ,cAAc,sBAAsB;EACrC,CAAC;AAEF,kBAAiB,cAAc,eAAe,OAAO,eAAe,EAAE,CAAC;AAEvE,QAAO;;AAGT,SAAgB,mCACd,cACY;CACZ,MAAM,SAAS,uBAAuB,UAAU,aAAa;AAC7D,KAAI,CAAC,OAAO,QACV,OAAM,IAAI,MAAM,8CAA8C,EAAE,cAAc,OAAO,MAAM,GAAG;CAGhG,MAAM,EAAE,YAAY,iBAAiB,wBAAwB;EAC3D,YAAY;EACZ,cAAc;EACf,CAAC;AAEF,MAAK,MAAM,mBAAmB,OAAO,KACnC,YAAW,qBAAqB;EAC9B,cAAc,CAAC,gBAAgB;EAC/B,iBAAiB,KAAK,gBAAgB;EACvC,CAAC;AAGJ,MAAK,MAAM,mBAAmB,OAAO,KACnC,cAAa,+BAA+B,EAAE,MAAM,iBAAiB,CAAC;AAGxE,QAAO"}

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

@@ -1,56 +1,39 @@
-import { addStringProperty, createFactoryDefinition, toCamelCase } from "../utils.js";
+import { addValueToObject, createFactoryDefinition, toCamelCase } from "../utils.js";
+import { FullProjectDefinitionSchema } from "@inkeep/agents-core";
 import { z } from "zod";
-import { SyntaxKind } from "ts-morph";
 
 //#region src/commands/pull-v4/generators/external-agent-generator.ts
-const ExternalAgentSchema = z.looseObject({
+const MySchema = FullProjectDefinitionSchema.shape.externalAgents.unwrap().valueType.omit({ id: true });
+const ExternalAgentSchema = z.strictObject({
 	externalAgentId: z.string().nonempty(),
-	name: z.string().nonempty(),
-	description: z.string().nullable().optional(),
-	baseUrl: z.string().nonempty(),
-	credentialReference: z.union([z.string(), z.looseObject({
-		id: z.string().optional(),
-		name: z.string().optional(),
-		description: z.string().optional()
-	})]).optional()
+	externalAgentReferenceName: z.string().optional(),
+	...MySchema.shape
 });
-function generateExternalAgentDefinition(data) {
+function generateExternalAgentDefinition({ id, tenantId, projectId, createdAt, updatedAt, ...data }) {
 	const result = ExternalAgentSchema.safeParse(data);
 	if (!result.success) throw new Error(`Validation failed for external agent:\n${z.prettifyError(result.error)}`);
-	const parsed = result.data;
+	const { externalAgentReferenceName, externalAgentId, credentialReferenceId, ...parsed } = result.data;
 	const { sourceFile, configObject } = createFactoryDefinition({
 		importName: "externalAgent",
-		variableName: toCamelCase(parsed.externalAgentId)
+		variableName: externalAgentReferenceName ?? toCamelCase(externalAgentId)
 	});
-	if (typeof parsed.credentialReference === "string") sourceFile.addImportDeclaration({
-		namedImports: [toCamelCase(parsed.credentialReference)],
-		moduleSpecifier: `../credentials/${parsed.credentialReference}`
-	});
-	writeExternalAgentConfig(configObject, parsed);
-	return sourceFile;
-}
-function writeExternalAgentConfig(configObject, data) {
-	addStringProperty(configObject, "id", data.externalAgentId);
-	addStringProperty(configObject, "name", data.name);
-	if (data.description !== null && data.description !== void 0) addStringProperty(configObject, "description", data.description);
-	else addStringProperty(configObject, "description", `External agent ${data.externalAgentId}`);
-	addStringProperty(configObject, "baseUrl", data.baseUrl);
-	if (typeof data.credentialReference === "string") {
+	for (const [key, value] of Object.entries({
+		id: externalAgentId,
+		...parsed
+	})) addValueToObject(configObject, key, value);
+	if (credentialReferenceId) {
+		sourceFile.addImportDeclaration({
+			namedImports: [toCamelCase(credentialReferenceId)],
+			moduleSpecifier: `../credentials/${credentialReferenceId}`
+		});
 		configObject.addPropertyAssignment({
 			name: "credentialReference",
-			initializer: toCamelCase(data.credentialReference)
+			initializer: toCamelCase(credentialReferenceId)
 		});
-		return;
 	}
-	if (!data.credentialReference) return;
-	const credentialReferenceObject = configObject.addPropertyAssignment({
-		name: "credentialReference",
-		initializer: "{}"
-	}).getInitializerIfKindOrThrow(SyntaxKind.ObjectLiteralExpression);
-	if (data.credentialReference.id !== void 0) addStringProperty(credentialReferenceObject, "id", data.credentialReference.id);
-	if (data.credentialReference.name !== void 0) addStringProperty(credentialReferenceObject, "name", data.credentialReference.name);
-	if (data.credentialReference.description !== void 0) addStringProperty(credentialReferenceObject, "description", data.credentialReference.description);
+	return sourceFile;
 }
 
 //#endregion
-export { generateExternalAgentDefinition };
+export { generateExternalAgentDefinition };
+//# sourceMappingURL=external-agent-generator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"external-agent-generator.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/external-agent-generator.ts"],"sourcesContent":["import { FullProjectDefinitionSchema } from '@inkeep/agents-core';\nimport type { SourceFile } from 'ts-morph';\nimport { z } from 'zod';\nimport { addValueToObject, createFactoryDefinition, toCamelCase } from '../utils';\n\nconst MySchema = FullProjectDefinitionSchema.shape.externalAgents.unwrap().valueType.omit({\n  id: true,\n});\n\nconst ExternalAgentSchema = z.strictObject({\n  externalAgentId: z.string().nonempty(),\n  externalAgentReferenceName: z.string().optional(),\n  ...MySchema.shape,\n});\n\ntype ExternalAgentInput = z.input<typeof ExternalAgentSchema>;\n\nexport function generateExternalAgentDefinition({\n  id,\n  tenantId,\n  projectId,\n  createdAt,\n  updatedAt,\n  ...data\n}: ExternalAgentInput & Record<string, unknown>): SourceFile {\n  const result = ExternalAgentSchema.safeParse(data);\n  if (!result.success) {\n    throw new Error(`Validation failed for external agent:\\n${z.prettifyError(result.error)}`);\n  }\n\n  const { externalAgentReferenceName, externalAgentId, credentialReferenceId, ...parsed } =\n    result.data;\n  const { sourceFile, configObject } = createFactoryDefinition({\n    importName: 'externalAgent',\n    variableName: externalAgentReferenceName ?? toCamelCase(externalAgentId),\n  });\n\n  for (const [key, value] of Object.entries({\n    id: externalAgentId,\n    ...parsed,\n  })) {\n    addValueToObject(configObject, key, value);\n  }\n  if (credentialReferenceId) {\n    sourceFile.addImportDeclaration({\n      namedImports: [toCamelCase(credentialReferenceId)],\n      moduleSpecifier: `../credentials/${credentialReferenceId}`,\n    });\n    configObject.addPropertyAssignment({\n      name: 'credentialReference',\n      initializer: toCamelCase(credentialReferenceId),\n    });\n  }\n  return sourceFile;\n}\n"],"mappings":";;;;;AAKA,MAAM,WAAW,4BAA4B,MAAM,eAAe,QAAQ,CAAC,UAAU,KAAK,EACxF,IAAI,MACL,CAAC;AAEF,MAAM,sBAAsB,EAAE,aAAa;CACzC,iBAAiB,EAAE,QAAQ,CAAC,UAAU;CACtC,4BAA4B,EAAE,QAAQ,CAAC,UAAU;CACjD,GAAG,SAAS;CACb,CAAC;AAIF,SAAgB,gCAAgC,EAC9C,IACA,UACA,WACA,WACA,WACA,GAAG,QACwD;CAC3D,MAAM,SAAS,oBAAoB,UAAU,KAAK;AAClD,KAAI,CAAC,OAAO,QACV,OAAM,IAAI,MAAM,0CAA0C,EAAE,cAAc,OAAO,MAAM,GAAG;CAG5F,MAAM,EAAE,4BAA4B,iBAAiB,uBAAuB,GAAG,WAC7E,OAAO;CACT,MAAM,EAAE,YAAY,iBAAiB,wBAAwB;EAC3D,YAAY;EACZ,cAAc,8BAA8B,YAAY,gBAAgB;EACzE,CAAC;AAEF,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ;EACxC,IAAI;EACJ,GAAG;EACJ,CAAC,CACA,kBAAiB,cAAc,KAAK,MAAM;AAE5C,KAAI,uBAAuB;AACzB,aAAW,qBAAqB;GAC9B,cAAc,CAAC,YAAY,sBAAsB,CAAC;GAClD,iBAAiB,kBAAkB;GACpC,CAAC;AACF,eAAa,sBAAsB;GACjC,MAAM;GACN,aAAa,YAAY,sBAAsB;GAChD,CAAC;;AAEJ,QAAO"}