@inkeep/agents-cli 0.59.3 → 0.59.4

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

@@ -1,7 +1,16 @@
-import { addFactoryConfigVariable, addStringProperty, addValueToObject, convertJsonSchemaToZodSafe, createInMemoryProject, formatPropertyName, isPlainObject, toCamelCase } from "../utils.js";
+import { isPlainObject } from "../utils/shared.js";
+import { codeExpression, codeReference } from "../utils/code-values.js";
+import { toCamelCase } from "../utils/naming.js";
+import { addFactoryConfigVariable, addValueToObject, createInMemoryProject } from "../utils/factory-writer.js";
+import { convertJsonSchemaToZodSafe } from "../utils/schema-rendering.js";
+import "../utils/index.js";
+import { applyPromptHeaderTemplateSchema, asRecord, collectHeaderTemplateVariablesFromAgentPrompts } from "../collector-common.js";
+import { collectContextConfigCredentialReferenceOverrides, collectContextConfigCredentialReferencePathOverrides, collectContextConfigHeadersReferenceOverride } from "../collector-reference-helpers.js";
+import { generateValidatedSourceFile } from "../simple-factory-generator.js";
+import { addNamedImports, applyImportPlan, createImportPlan } from "../import-plan.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/context-config-generator.ts
 const MySchema = FullProjectDefinitionSchema.shape.agents.valueType.shape.contextConfig.unwrap();
@@ -21,91 +30,88 @@ const ContextConfigSchema = z.strictObject({
 	fetchConfig: z.record(z.string(), z.unknown()).optional(),
 	defaultValue: z.unknown(),
 	responseSchema: z.unknown()
-});
-function generateContextConfigDefinition(data) {
-	const result = ContextConfigSchema.safeParse(data);
-	if (!result.success) throw new Error(`Validation failed for context config:\n${z.prettifyError(result.error)}`);
-	const project = createInMemoryProject();
-	const parsed = result.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 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(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");
-	sourceFile.addImportDeclaration({
-		namedImports: coreImports,
-		moduleSpecifier: "@inkeep/agents-core"
-	});
-	const hasResponseSchemas = fetchDefinitions.some((definition) => isPlainObject(definition.data.responseSchema));
-	if (shouldDefineHeadersInFile || hasResponseSchemas) sourceFile.addImportDeclaration({
-		namedImports: ["z"],
-		moduleSpecifier: "zod"
-	});
-	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
-		});
-		headersObject.addPropertyAssignment({
-			name: "schema",
-			initializer: convertJsonSchemaToZod(headersSchema)
-		});
-	}
-	for (const fetchDefinition of fetchDefinitions) {
-		const { configObject: fetchConfigObject } = addFactoryConfigVariable({
-			sourceFile,
-			importName: "fetchDefinition",
-			variableName: fetchDefinition.variableName
-		});
-		writeFetchDefinition(fetchConfigObject, fetchDefinition.data, credentialReferenceNames, headersReference);
+}).transform((data) => ({
+	...data,
+	normalizedHeadersReference: extractHeadersReference(data.headers),
+	normalizedContextVariables: normalizeContextVariables(data.contextVariables)
+}));
+function normalizeContextVariables(contextVariables) {
+	if (!contextVariables) return {};
+	const normalizedVariables = {};
+	for (const [key, value] of Object.entries(contextVariables)) {
+		const referenceName = extractContextVariableReference(key, value);
+		if (!referenceName) continue;
+		normalizedVariables[key] = {
+			referenceName,
+			rawValue: value,
+			...isPlainObject(value) && isFetchDefinitionData(value) ? { fetchDefinitionData: value } : {}
+		};
 	}
-	const { configObject } = addFactoryConfigVariable({
-		sourceFile,
+	return normalizedVariables;
+}
+function generateContextConfigDefinition(data) {
+	return generateValidatedSourceFile(data, {
+		schema: ContextConfigSchema,
 		importName: "contextConfig",
-		variableName: toContextConfigVariableName(parsed.contextConfigId),
-		isExported: true
+		render(parsed) {
+			const sourceFile = createInMemoryProject().createSourceFile("context-config-definition.ts", "", { overwrite: true });
+			if (isHeadersDefinitionData(parsed)) return generateStandaloneHeadersDefinition(sourceFile, parsed);
+			if (isFetchDefinitionData(parsed)) return generateStandaloneFetchDefinition(sourceFile, parsed);
+			const explicitHeadersReference = (typeof parsed.headersReference === "string" && parsed.headersReference.length > 0 ? parsed.headersReference : void 0) ?? parsed.normalizedHeadersReference;
+			const templateHeaderVariables = collectTemplateHeaderVariables(parsed.normalizedContextVariables);
+			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(parsed.normalizedContextVariables);
+			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");
+			const importPlan = createImportPlan();
+			addNamedImports(importPlan, "@inkeep/agents-core", coreImports);
+			const hasResponseSchemas = fetchDefinitions.some((definition) => isPlainObject(definition.data.responseSchema));
+			if (shouldDefineHeadersInFile || hasResponseSchemas) addNamedImports(importPlan, "zod", "z");
+			for (const [credentialId, credentialReferenceName] of credentialReferenceNames) addNamedImports(importPlan, `../credentials/${parsed.referencePathOverrides?.credentialReferences?.[credentialId] ?? credentialId}`, credentialReferenceName);
+			applyImportPlan(sourceFile, importPlan);
+			if (shouldDefineHeadersInFile && headersReference && headersSchema) {
+				const { configObject: headersObject } = addFactoryConfigVariable({
+					sourceFile,
+					isExported: true,
+					importName: "headers",
+					variableName: headersReference
+				});
+				addValueToObject(headersObject, "schema", createSchemaExpression(headersSchema));
+			}
+			for (const fetchDefinition of fetchDefinitions) {
+				const { configObject: fetchConfigObject } = addFactoryConfigVariable({
+					sourceFile,
+					importName: "fetchDefinition",
+					variableName: fetchDefinition.variableName
+				});
+				writeFetchDefinition(fetchConfigObject, fetchDefinition.data, credentialReferenceNames, headersReference);
+			}
+			const { configObject } = addFactoryConfigVariable({
+				sourceFile,
+				importName: "contextConfig",
+				variableName: toContextConfigVariableName(parsed.contextConfigId),
+				isExported: true
+			});
+			writeContextConfig(configObject, parsed, headersReference);
+			return sourceFile;
+		}
 	});
-	writeContextConfig(configObject, parsed, headersReference);
-	return sourceFile;
 }
 function writeContextConfig(configObject, data, headersReference) {
-	if (data.id !== void 0) addStringProperty(configObject, "id", data.id);
-	if (headersReference) configObject.addPropertyAssignment({
-		name: "headers",
-		initializer: headersReference
-	});
-	if (data.contextVariables && Object.keys(data.contextVariables).length > 0) {
-		const contextVariablesObject = configObject.addPropertyAssignment({
-			name: "contextVariables",
-			initializer: "{}"
-		}).getInitializerIfKindOrThrow(SyntaxKind.ObjectLiteralExpression);
-		for (const [key, value] of Object.entries(data.contextVariables)) {
-			const reference = extractContextVariableReference(key, value);
-			if (!reference) continue;
-			contextVariablesObject.addPropertyAssignment({
-				name: formatPropertyName(key),
-				initializer: reference
-			});
-		}
+	const contextConfig = {};
+	if (data.id !== void 0) contextConfig.id = data.id;
+	if (headersReference) contextConfig.headers = codeReference(headersReference);
+	if (Object.keys(data.normalizedContextVariables).length > 0) {
+		const contextVariables = {};
+		for (const [key, value] of Object.entries(data.normalizedContextVariables)) contextVariables[key] = codeReference(value.referenceName);
+		contextConfig.contextVariables = contextVariables;
 	}
+	for (const [key, value] of Object.entries(contextConfig)) addValueToObject(configObject, key, value);
 }
 function extractHeadersReference(headers) {
 	if (!headers) return;
@@ -124,34 +130,25 @@ function isFetchDefinitionData(value) {
 	return value.fetchConfig !== void 0 || value.responseSchema !== void 0;
 }
 function collectFetchDefinitionEntries(contextVariables) {
-	if (!contextVariables) return [];
-	return Object.entries(contextVariables).filter(([, value]) => isFetchDefinitionData(value)).map(([key, value]) => {
-		return {
-			key,
-			variableName: extractContextVariableReference(key, value) ?? toReferenceIdentifier(key),
-			data: value
-		};
-	});
+	if (!contextVariables || Object.keys(contextVariables).length === 0) return [];
+	return Object.entries(contextVariables).filter(([, value]) => Boolean(value.fetchDefinitionData)).map(([key, value]) => ({
+		key,
+		variableName: value.referenceName,
+		data: value.fetchDefinitionData
+	}));
 }
 function writeFetchDefinition(configObject, fetchDefinitionData, credentialReferenceNames, headersReference) {
-	const { contextConfigId, responseSchema, credentialReferenceId, ...rest } = isPlainObject(fetchDefinitionData) ? fetchDefinitionData : {};
+	const { contextConfigId, responseSchema, credentialReferenceId, normalizedHeadersReference: _normalizedHeadersReference, normalizedContextVariables: _normalizedContextVariables, ...rest } = isPlainObject(fetchDefinitionData) ? fetchDefinitionData : {};
 	const normalizedRest = rewriteHeaderTemplates(rest, headersReference);
+	const fetchDefinition = {};
 	for (const [k, v] of Object.entries({
 		id: contextConfigId,
 		...normalizedRest
-	})) if (v !== null) addValueToObject(configObject, k, v);
-	if (responseSchema) configObject.addPropertyAssignment({
-		name: "responseSchema",
-		initializer: convertJsonSchemaToZod(responseSchema)
-	});
-	if (typeof credentialReferenceId === "string" && credentialReferenceNames?.has(credentialReferenceId)) {
-		configObject.addPropertyAssignment({
-			name: "credentialReference",
-			initializer: credentialReferenceNames.get(credentialReferenceId)
-		});
-		return;
-	}
-	if (typeof credentialReferenceId === "string") addStringProperty(configObject, "credentialReferenceId", credentialReferenceId);
+	})) if (v !== null) fetchDefinition[k] = v;
+	if (responseSchema) fetchDefinition.responseSchema = createSchemaExpression(responseSchema);
+	if (typeof credentialReferenceId === "string" && credentialReferenceNames?.has(credentialReferenceId)) fetchDefinition.credentialReference = codeReference(credentialReferenceNames.get(credentialReferenceId));
+	else if (typeof credentialReferenceId === "string") fetchDefinition.credentialReferenceId = credentialReferenceId;
+	for (const [key, value] of Object.entries(fetchDefinition)) addValueToObject(configObject, key, value);
 }
 const HEADER_TEMPLATE_REGEX = /\{\{headers\.([^}]+)\}\}/g;
 const HEADER_TO_TEMPLATE_CALL_REGEX = /\$\{\s*(headersSchema|headers)\.toTemplate\((['"])([^'"`]+)\2\)\s*\}/g;
@@ -168,23 +165,16 @@ function rewriteHeaderTemplates(value, headersReference) {
 }
 function generateStandaloneHeadersDefinition(sourceFile, data) {
 	const importName = "headers";
-	sourceFile.addImportDeclaration({
-		namedImports: [importName],
-		moduleSpecifier: "@inkeep/agents-core"
-	});
-	sourceFile.addImportDeclaration({
-		namedImports: ["z"],
-		moduleSpecifier: "zod"
-	});
+	const importPlan = createImportPlan();
+	addNamedImports(importPlan, "@inkeep/agents-core", importName);
+	addNamedImports(importPlan, "zod", "z");
+	applyImportPlan(sourceFile, importPlan);
 	const { configObject } = addFactoryConfigVariable({
 		sourceFile,
 		importName,
 		variableName: toContextConfigVariableName(data.contextConfigId)
 	});
-	configObject.addPropertyAssignment({
-		name: "schema",
-		initializer: convertJsonSchemaToZod(data.schema)
-	});
+	addValueToObject(configObject, "schema", createSchemaExpression(data.schema));
 	return sourceFile;
 }
 function isHeadersDefinitionData(value) {
@@ -192,25 +182,18 @@ function isHeadersDefinitionData(value) {
 }
 function generateStandaloneFetchDefinition(sourceFile, data) {
 	const importName = "fetchDefinition";
-	sourceFile.addImportDeclaration({
-		namedImports: [importName],
-		moduleSpecifier: "@inkeep/agents-core"
-	});
-	if (isPlainObject(data.responseSchema)) sourceFile.addImportDeclaration({
-		namedImports: ["z"],
-		moduleSpecifier: "zod"
-	});
+	const importPlan = createImportPlan();
+	addNamedImports(importPlan, "@inkeep/agents-core", importName);
+	if (isPlainObject(data.responseSchema)) addNamedImports(importPlan, "zod", "z");
 	const credentialReferenceNames = /* @__PURE__ */ new Map();
 	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/${credentialReferencePath}`
-		});
+		addNamedImports(importPlan, `../credentials/${credentialReferencePath}`, credentialReferenceName);
 	}
+	applyImportPlan(sourceFile, importPlan);
 	const { configObject } = addFactoryConfigVariable({
 		sourceFile,
 		importName,
@@ -222,6 +205,9 @@ function generateStandaloneFetchDefinition(sourceFile, data) {
 function convertJsonSchemaToZod(schema) {
 	return convertJsonSchemaToZodSafe(schema, { conversionOptions: { module: "none" } });
 }
+function createSchemaExpression(schema) {
+	return codeExpression(convertJsonSchemaToZod(schema));
+}
 function extractContextVariableReference(key, value) {
 	if (typeof value === "string") return toReferenceIdentifier(value);
 	if (!isPlainObject(value)) return;
@@ -246,7 +232,7 @@ function collectCredentialReferenceNames(fetchDefinitions, overrideNamesById) {
 }
 function collectTemplateHeaderVariables(contextVariables) {
 	const variables = /* @__PURE__ */ new Set();
-	collectTemplateHeaderVariablesFromValue(contextVariables, variables);
+	for (const value of Object.values(contextVariables ?? {})) collectTemplateHeaderVariablesFromValue(value.rawValue, variables);
 	return variables;
 }
 function collectTemplateHeaderVariablesFromValue(value, variables) {
@@ -272,7 +258,39 @@ function inferHeadersSchemaFromTemplateHeaderVariables(variables) {
 		additionalProperties: false
 	};
 }
+const task = {
+	type: "context-config",
+	collect(context) {
+		if (!context.project.agents) return [];
+		const contextConfigRecordsById = /* @__PURE__ */ new Map();
+		for (const agentId of context.completeAgentIds) {
+			const agentData = context.project.agents[agentId];
+			const contextConfig = agentData ? asRecord(agentData.contextConfig) : void 0;
+			if (!agentData || !contextConfig) continue;
+			const normalizedContextConfig = applyPromptHeaderTemplateSchema(contextConfig, collectHeaderTemplateVariablesFromAgentPrompts(agentData));
+			const contextConfigId = typeof normalizedContextConfig.id === "string" ? normalizedContextConfig.id : "";
+			if (!contextConfigId || contextConfigRecordsById.has(contextConfigId)) continue;
+			const contextConfigFilePath = context.resolver.resolveOutputFilePath("contextConfigs", contextConfigId, join(context.paths.contextConfigsDir, `${contextConfigId}.ts`));
+			const credentialReferenceOverrides = collectContextConfigCredentialReferenceOverrides(context, normalizedContextConfig);
+			const credentialReferencePathOverrides = collectContextConfigCredentialReferencePathOverrides(context, normalizedContextConfig);
+			const headersReferenceOverride = collectContextConfigHeadersReferenceOverride(context, contextConfigId, contextConfigFilePath);
+			contextConfigRecordsById.set(contextConfigId, {
+				id: contextConfigId,
+				filePath: contextConfigFilePath,
+				payload: {
+					contextConfigId,
+					...normalizedContextConfig,
+					...headersReferenceOverride && { headersReference: headersReferenceOverride },
+					...credentialReferenceOverrides && { referenceOverrides: { credentialReferences: credentialReferenceOverrides } },
+					...credentialReferencePathOverrides && { referencePathOverrides: { credentialReferences: credentialReferencePathOverrides } }
+				}
+			});
+		}
+		return [...contextConfigRecordsById.values()];
+	},
+	generate: generateContextConfigDefinition
+};
 
 //#endregion
-export { generateContextConfigDefinition };
+export { generateContextConfigDefinition, task };
 //# sourceMappingURL=context-config-generator.js.map

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

@@ -1 +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"}
+{"version":3,"file":"context-config-generator.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/context-config-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 {\n  applyPromptHeaderTemplateSchema,\n  asRecord,\n  collectHeaderTemplateVariablesFromAgentPrompts,\n} from '../collector-common';\nimport {\n  collectContextConfigCredentialReferenceOverrides,\n  collectContextConfigCredentialReferencePathOverrides,\n  collectContextConfigHeadersReferenceOverride,\n} from '../collector-reference-helpers';\nimport type { GenerationTask } from '../generation-types';\nimport { addNamedImports, applyImportPlan, createImportPlan } from '../import-plan';\nimport { generateValidatedSourceFile } from '../simple-factory-generator';\nimport {\n  addFactoryConfigVariable,\n  addValueToObject,\n  codeExpression,\n  codeReference,\n  convertJsonSchemaToZodSafe,\n  createInMemoryProject,\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\ninterface NormalizedContextVariableEntry {\n  referenceName: string;\n  rawValue: unknown;\n  fetchDefinitionData?: Record<string, unknown>;\n}\n\ntype NormalizedContextVariableMap = Record<string, NormalizedContextVariableEntry>;\n\nconst BaseContextConfigSchema = 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\nconst ContextConfigSchema = BaseContextConfigSchema.transform((data) => ({\n  ...data,\n  normalizedHeadersReference: extractHeadersReference(data.headers),\n  normalizedContextVariables: normalizeContextVariables(data.contextVariables),\n}));\n\ntype ContextConfigInput = z.input<typeof ContextConfigSchema>;\ntype ContextConfigOutput = z.output<typeof ContextConfigSchema>;\n\nfunction normalizeContextVariables(\n  contextVariables?: Record<string, unknown>\n): NormalizedContextVariableMap {\n  if (!contextVariables) {\n    return {};\n  }\n\n  const normalizedVariables: NormalizedContextVariableMap = {};\n  for (const [key, value] of Object.entries(contextVariables)) {\n    const referenceName = extractContextVariableReference(key, value);\n    if (!referenceName) {\n      continue;\n    }\n\n    normalizedVariables[key] = {\n      referenceName,\n      rawValue: value,\n      ...(isPlainObject(value) && isFetchDefinitionData(value)\n        ? { fetchDefinitionData: value }\n        : {}),\n    };\n  }\n\n  return normalizedVariables;\n}\n\nexport function generateContextConfigDefinition(data: ContextConfigInput): SourceFile {\n  return generateValidatedSourceFile(data, {\n    schema: ContextConfigSchema,\n    importName: 'contextConfig',\n    render(parsed) {\n      const project = createInMemoryProject();\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) ?? parsed.normalizedHeadersReference;\n      const templateHeaderVariables = collectTemplateHeaderVariables(\n        parsed.normalizedContextVariables\n      );\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.normalizedContextVariables);\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      const importPlan = createImportPlan();\n      addNamedImports(importPlan, '@inkeep/agents-core', coreImports);\n\n      const hasResponseSchemas = fetchDefinitions.some((definition) =>\n        isPlainObject(definition.data.responseSchema)\n      );\n      if (shouldDefineHeadersInFile || hasResponseSchemas) {\n        addNamedImports(importPlan, 'zod', 'z');\n      }\n\n      for (const [credentialId, credentialReferenceName] of credentialReferenceNames) {\n        const credentialReferencePath =\n          parsed.referencePathOverrides?.credentialReferences?.[credentialId] ?? credentialId;\n        addNamedImports(\n          importPlan,\n          `../credentials/${credentialReferencePath}`,\n          credentialReferenceName\n        );\n      }\n      applyImportPlan(sourceFile, importPlan);\n      if (shouldDefineHeadersInFile && headersReference && headersSchema) {\n        const { configObject: headersObject } = addFactoryConfigVariable({\n          sourceFile,\n          isExported: true,\n          importName: 'headers',\n          variableName: headersReference,\n        });\n\n        addValueToObject(headersObject, 'schema', createSchemaExpression(headersSchema));\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  });\n}\n\nfunction writeContextConfig(\n  configObject: ObjectLiteralExpression,\n  data: ContextConfigOutput,\n  headersReference?: string\n) {\n  const contextConfig: Record<string, unknown> = {};\n  if (data.id !== undefined) {\n    contextConfig.id = data.id;\n  }\n  if (headersReference) {\n    contextConfig.headers = codeReference(headersReference);\n  }\n  if (Object.keys(data.normalizedContextVariables).length > 0) {\n    const contextVariables: Record<string, unknown> = {};\n\n    for (const [key, value] of Object.entries(data.normalizedContextVariables)) {\n      contextVariables[key] = codeReference(value.referenceName);\n    }\n    contextConfig.contextVariables = contextVariables;\n  }\n\n  for (const [key, value] of Object.entries(contextConfig)) {\n    addValueToObject(configObject, key, value);\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(\n  contextVariables?: NormalizedContextVariableMap\n): Array<{ key: string; variableName: string; data: Record<string, unknown> }> {\n  if (!contextVariables || Object.keys(contextVariables).length === 0) {\n    return [];\n  }\n\n  return Object.entries(contextVariables)\n    .filter(([, value]) => Boolean(value.fetchDefinitionData))\n    .map(([key, value]) => ({\n      key,\n      variableName: value.referenceName,\n      data: value.fetchDefinitionData as Record<string, unknown>,\n    }));\n}\n\nfunction writeFetchDefinition(\n  configObject: ObjectLiteralExpression,\n  fetchDefinitionData: unknown,\n  credentialReferenceNames?: Map<string, string>,\n  headersReference?: string\n) {\n  const {\n    contextConfigId,\n    responseSchema,\n    credentialReferenceId,\n    normalizedHeadersReference: _normalizedHeadersReference,\n    normalizedContextVariables: _normalizedContextVariables,\n    ...rest\n  } = isPlainObject(fetchDefinitionData) ? fetchDefinitionData : {};\n  const normalizedRest = rewriteHeaderTemplates(rest, headersReference);\n  const fetchDefinition: Record<string, unknown> = {};\n  for (const [k, v] of Object.entries({\n    id: contextConfigId,\n    ...normalizedRest,\n  })) {\n    if (v !== null) {\n      fetchDefinition[k] = v;\n    }\n  }\n  if (responseSchema) {\n    fetchDefinition.responseSchema = createSchemaExpression(\n      responseSchema as Record<string, unknown>\n    );\n  }\n\n  if (\n    typeof credentialReferenceId === 'string' &&\n    credentialReferenceNames?.has(credentialReferenceId)\n  ) {\n    fetchDefinition.credentialReference = codeReference(\n      credentialReferenceNames.get(credentialReferenceId) as string\n    );\n  } else if (typeof credentialReferenceId === 'string') {\n    fetchDefinition.credentialReferenceId = credentialReferenceId;\n  }\n\n  for (const [key, value] of Object.entries(fetchDefinition)) {\n    addValueToObject(configObject, key, value);\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  const importPlan = createImportPlan();\n  addNamedImports(importPlan, '@inkeep/agents-core', importName);\n  addNamedImports(importPlan, 'zod', 'z');\n  applyImportPlan(sourceFile, importPlan);\n\n  const headersVarName = toContextConfigVariableName(data.contextConfigId);\n  const { configObject } = addFactoryConfigVariable({\n    sourceFile,\n    importName,\n    variableName: headersVarName,\n  });\n\n  addValueToObject(configObject, 'schema', createSchemaExpression(data.schema));\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  const importPlan = createImportPlan();\n  addNamedImports(importPlan, '@inkeep/agents-core', importName);\n\n  if (isPlainObject(data.responseSchema)) {\n    addNamedImports(importPlan, 'zod', 'z');\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    addNamedImports(\n      importPlan,\n      `../credentials/${credentialReferencePath}`,\n      credentialReferenceName\n    );\n  }\n  applyImportPlan(sourceFile, importPlan);\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 createSchemaExpression(schema: Record<string, unknown>) {\n  return codeExpression(convertJsonSchemaToZod(schema));\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(\n  contextVariables?: NormalizedContextVariableMap\n): Set<string> {\n  const variables = new Set<string>();\n  for (const value of Object.values(contextVariables ?? {})) {\n    collectTemplateHeaderVariablesFromValue(value.rawValue, variables);\n  }\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\nexport const task = {\n  type: 'context-config',\n  collect(context) {\n    if (!context.project.agents) {\n      return [];\n    }\n\n    const contextConfigRecordsById = new Map<\n      string,\n      ReturnType<\n        GenerationTask<Parameters<typeof generateContextConfigDefinition>[0]>['collect']\n      >[number]\n    >();\n\n    for (const agentId of context.completeAgentIds) {\n      const agentData = context.project.agents[agentId];\n      const contextConfig = agentData ? asRecord(agentData.contextConfig) : undefined;\n      if (!agentData || !contextConfig) {\n        continue;\n      }\n\n      const normalizedContextConfig = applyPromptHeaderTemplateSchema(\n        contextConfig,\n        collectHeaderTemplateVariablesFromAgentPrompts(agentData)\n      );\n      const contextConfigId =\n        typeof normalizedContextConfig.id === 'string' ? normalizedContextConfig.id : '';\n      if (!contextConfigId || contextConfigRecordsById.has(contextConfigId)) {\n        continue;\n      }\n\n      const contextConfigFilePath = context.resolver.resolveOutputFilePath(\n        'contextConfigs',\n        contextConfigId,\n        join(context.paths.contextConfigsDir, `${contextConfigId}.ts`)\n      );\n      const credentialReferenceOverrides = collectContextConfigCredentialReferenceOverrides(\n        context,\n        normalizedContextConfig\n      );\n      const credentialReferencePathOverrides = collectContextConfigCredentialReferencePathOverrides(\n        context,\n        normalizedContextConfig\n      );\n      const headersReferenceOverride = collectContextConfigHeadersReferenceOverride(\n        context,\n        contextConfigId,\n        contextConfigFilePath\n      );\n\n      contextConfigRecordsById.set(contextConfigId, {\n        id: contextConfigId,\n        filePath: contextConfigFilePath,\n        payload: {\n          contextConfigId,\n          ...normalizedContextConfig,\n          ...(headersReferenceOverride && {\n            headersReference: headersReferenceOverride,\n          }),\n          ...(credentialReferenceOverrides && {\n            referenceOverrides: {\n              credentialReferences: credentialReferenceOverrides,\n            },\n          }),\n          ...(credentialReferencePathOverrides && {\n            referencePathOverrides: {\n              credentialReferences: credentialReferencePathOverrides,\n            },\n          }),\n        } as Parameters<typeof generateContextConfigDefinition>[0],\n      });\n    }\n\n    return [...contextConfigRecordsById.values()];\n  },\n  generate: generateContextConfigDefinition,\n} satisfies GenerationTask<Parameters<typeof generateContextConfigDefinition>[0]>;\n"],"mappings":";;;;;;;;;;;;;;;AA4BA,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;AA2BF,MAAM,sBAjB0B,EAAE,aAAa;CAC7C,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,CAEkD,WAAW,UAAU;CACvE,GAAG;CACH,4BAA4B,wBAAwB,KAAK,QAAQ;CACjE,4BAA4B,0BAA0B,KAAK,iBAAiB;CAC7E,EAAE;AAKH,SAAS,0BACP,kBAC8B;AAC9B,KAAI,CAAC,iBACH,QAAO,EAAE;CAGX,MAAM,sBAAoD,EAAE;AAC5D,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,iBAAiB,EAAE;EAC3D,MAAM,gBAAgB,gCAAgC,KAAK,MAAM;AACjE,MAAI,CAAC,cACH;AAGF,sBAAoB,OAAO;GACzB;GACA,UAAU;GACV,GAAI,cAAc,MAAM,IAAI,sBAAsB,MAAM,GACpD,EAAE,qBAAqB,OAAO,GAC9B,EAAE;GACP;;AAGH,QAAO;;AAGT,SAAgB,gCAAgC,MAAsC;AACpF,QAAO,4BAA4B,MAAM;EACvC,QAAQ;EACR,YAAY;EACZ,OAAO,QAAQ;GAEb,MAAM,aADU,uBAAuB,CACZ,iBAAiB,gCAAgC,IAAI,EAC9E,WAAW,MACZ,CAAC;AAEF,OAAI,wBAAwB,OAAO,CACjC,QAAO,oCAAoC,YAAY,OAAO;AAGhE,OAAI,sBAAsB,OAAO,CAC/B,QAAO,kCAAkC,YAAY,OAAO;GAG9D,MAAM,4BACH,OAAO,OAAO,qBAAqB,YAAY,OAAO,iBAAiB,SAAS,IAC7E,OAAO,mBACP,WAAc,OAAO;GAC3B,MAAM,0BAA0B,+BAC9B,OAAO,2BACR;GACD,MAAM,wBACJ,CAAC,cAAc,OAAO,cAAc,IAAI,CAAC,2BACrC,8CAA8C,wBAAwB,GACtE;GACN,MAAM,gBAAgB,cAAc,OAAO,cAAc,GACrD,OAAO,gBACP;GACJ,MAAM,mBAAmB,wBAAwB,QAAQ,QAAQ,cAAc,CAAC;GAChF,MAAM,4BAA4B,QAAQ,iBAAiB,IAAI,cAAc,cAAc;GAC3F,MAAM,mBAAmB,8BAA8B,OAAO,2BAA2B;GACzF,MAAM,2BAA2B,gCAC/B,kBACA,OAAO,oBAAoB,qBAC5B;GACD,MAAM,cAAc,CAAC,gBAAgB;AACrC,OAAI,0BACF,aAAY,QAAQ,UAAU;AAEhC,OAAI,iBAAiB,SAAS,EAC5B,aAAY,OAAO,YAAY,SAAS,GAAG,GAAG,kBAAkB;GAGlE,MAAM,aAAa,kBAAkB;AACrC,mBAAgB,YAAY,uBAAuB,YAAY;GAE/D,MAAM,qBAAqB,iBAAiB,MAAM,eAChD,cAAc,WAAW,KAAK,eAAe,CAC9C;AACD,OAAI,6BAA6B,mBAC/B,iBAAgB,YAAY,OAAO,IAAI;AAGzC,QAAK,MAAM,CAAC,cAAc,4BAA4B,yBAGpD,iBACE,YACA,kBAHA,OAAO,wBAAwB,uBAAuB,iBAAiB,gBAIvE,wBACD;AAEH,mBAAgB,YAAY,WAAW;AACvC,OAAI,6BAA6B,oBAAoB,eAAe;IAClE,MAAM,EAAE,cAAc,kBAAkB,yBAAyB;KAC/D;KACA,YAAY;KACZ,YAAY;KACZ,cAAc;KACf,CAAC;AAEF,qBAAiB,eAAe,UAAU,uBAAuB,cAAc,CAAC;;AAGlF,QAAK,MAAM,mBAAmB,kBAAkB;IAC9C,MAAM,EAAE,cAAc,sBAAsB,yBAAyB;KACnE;KACA,YAAY;KACZ,cAAc,gBAAgB;KAC/B,CAAC;AACF,yBACE,mBACA,gBAAgB,MAChB,0BACA,iBACD;;GAGH,MAAM,EAAE,iBAAiB,yBAAyB;IAChD;IACA,YAAY;IACZ,cAJ2B,4BAA4B,OAAO,gBAAgB;IAK9E,YAAY;IACb,CAAC;AAEF,sBAAmB,cAAc,QAAQ,iBAAiB;AAE1D,UAAO;;EAEV,CAAC;;AAGJ,SAAS,mBACP,cACA,MACA,kBACA;CACA,MAAM,gBAAyC,EAAE;AACjD,KAAI,KAAK,OAAO,OACd,eAAc,KAAK,KAAK;AAE1B,KAAI,iBACF,eAAc,UAAU,cAAc,iBAAiB;AAEzD,KAAI,OAAO,KAAK,KAAK,2BAA2B,CAAC,SAAS,GAAG;EAC3D,MAAM,mBAA4C,EAAE;AAEpD,OAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,KAAK,2BAA2B,CACxE,kBAAiB,OAAO,cAAc,MAAM,cAAc;AAE5D,gBAAc,mBAAmB;;AAGnC,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,cAAc,CACtD,kBAAiB,cAAc,KAAK,MAAM;;AAI9C,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,8BACP,kBAC6E;AAC7E,KAAI,CAAC,oBAAoB,OAAO,KAAK,iBAAiB,CAAC,WAAW,EAChE,QAAO,EAAE;AAGX,QAAO,OAAO,QAAQ,iBAAiB,CACpC,QAAQ,GAAG,WAAW,QAAQ,MAAM,oBAAoB,CAAC,CACzD,KAAK,CAAC,KAAK,YAAY;EACtB;EACA,cAAc,MAAM;EACpB,MAAM,MAAM;EACb,EAAE;;AAGP,SAAS,qBACP,cACA,qBACA,0BACA,kBACA;CACA,MAAM,EACJ,iBACA,gBACA,uBACA,4BAA4B,6BAC5B,4BAA4B,6BAC5B,GAAG,SACD,cAAc,oBAAoB,GAAG,sBAAsB,EAAE;CACjE,MAAM,iBAAiB,uBAAuB,MAAM,iBAAiB;CACrE,MAAM,kBAA2C,EAAE;AACnD,MAAK,MAAM,CAAC,GAAG,MAAM,OAAO,QAAQ;EAClC,IAAI;EACJ,GAAG;EACJ,CAAC,CACA,KAAI,MAAM,KACR,iBAAgB,KAAK;AAGzB,KAAI,eACF,iBAAgB,iBAAiB,uBAC/B,eACD;AAGH,KACE,OAAO,0BAA0B,YACjC,0BAA0B,IAAI,sBAAsB,CAEpD,iBAAgB,sBAAsB,cACpC,yBAAyB,IAAI,sBAAsB,CACpD;UACQ,OAAO,0BAA0B,SAC1C,iBAAgB,wBAAwB;AAG1C,MAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,gBAAgB,CACxD,kBAAiB,cAAc,KAAK,MAAM;;AAI9C,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;CACnB,MAAM,aAAa,kBAAkB;AACrC,iBAAgB,YAAY,uBAAuB,WAAW;AAC9D,iBAAgB,YAAY,OAAO,IAAI;AACvC,iBAAgB,YAAY,WAAW;CAGvC,MAAM,EAAE,iBAAiB,yBAAyB;EAChD;EACA;EACA,cAJqB,4BAA4B,KAAK,gBAAgB;EAKvE,CAAC;AAEF,kBAAiB,cAAc,UAAU,uBAAuB,KAAK,OAAO,CAAC;AAC7E,QAAO;;AAGT,SAAS,wBACP,OACoE;AACpE,QAAO,cAAc,MAAM,OAAO;;AAGpC,SAAS,kCACP,YACA,MACY;CACZ,MAAM,aAAa;CACnB,MAAM,aAAa,kBAAkB;AACrC,iBAAgB,YAAY,uBAAuB,WAAW;AAE9D,KAAI,cAAc,KAAK,eAAe,CACpC,iBAAgB,YAAY,OAAO,IAAI;CAGzC,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,kBACE,YACA,kBAAkB,2BAClB,wBACD;;AAEH,iBAAgB,YAAY,WAAW;CAGvC,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,uBAAuB,QAAiC;AAC/D,QAAO,eAAe,uBAAuB,OAAO,CAAC;;AAGvD,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,+BACP,kBACa;CACb,MAAM,4BAAY,IAAI,KAAa;AACnC,MAAK,MAAM,SAAS,OAAO,OAAO,oBAAoB,EAAE,CAAC,CACvD,yCAAwC,MAAM,UAAU,UAAU;AAEpE,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;;AAGH,MAAa,OAAO;CAClB,MAAM;CACN,QAAQ,SAAS;AACf,MAAI,CAAC,QAAQ,QAAQ,OACnB,QAAO,EAAE;EAGX,MAAM,2CAA2B,IAAI,KAKlC;AAEH,OAAK,MAAM,WAAW,QAAQ,kBAAkB;GAC9C,MAAM,YAAY,QAAQ,QAAQ,OAAO;GACzC,MAAM,gBAAgB,YAAY,SAAS,UAAU,cAAc,GAAG;AACtE,OAAI,CAAC,aAAa,CAAC,cACjB;GAGF,MAAM,0BAA0B,gCAC9B,eACA,+CAA+C,UAAU,CAC1D;GACD,MAAM,kBACJ,OAAO,wBAAwB,OAAO,WAAW,wBAAwB,KAAK;AAChF,OAAI,CAAC,mBAAmB,yBAAyB,IAAI,gBAAgB,CACnE;GAGF,MAAM,wBAAwB,QAAQ,SAAS,sBAC7C,kBACA,iBACA,KAAK,QAAQ,MAAM,mBAAmB,GAAG,gBAAgB,KAAK,CAC/D;GACD,MAAM,+BAA+B,iDACnC,SACA,wBACD;GACD,MAAM,mCAAmC,qDACvC,SACA,wBACD;GACD,MAAM,2BAA2B,6CAC/B,SACA,iBACA,sBACD;AAED,4BAAyB,IAAI,iBAAiB;IAC5C,IAAI;IACJ,UAAU;IACV,SAAS;KACP;KACA,GAAG;KACH,GAAI,4BAA4B,EAC9B,kBAAkB,0BACnB;KACD,GAAI,gCAAgC,EAClC,oBAAoB,EAClB,sBAAsB,8BACvB,EACF;KACD,GAAI,oCAAoC,EACtC,wBAAwB,EACtB,sBAAsB,kCACvB,EACF;KACF;IACF,CAAC;;AAGJ,SAAO,CAAC,GAAG,yBAAyB,QAAQ,CAAC;;CAE/C,UAAU;CACX"}

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

@@ -1,6 +1,10 @@
-import { addValueToObject, createFactoryDefinition, toCredentialReferenceName } from "../utils.js";
+import { toCredentialReferenceName } from "../utils/naming.js";
+import "../utils/index.js";
+import { buildSequentialNameFileNames } from "../generation-resolver.js";
+import { generateSimpleFactoryDefinition } 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/credential-generator.ts
 const MySchema = FullProjectDefinitionSchema.shape.credentialReferences.unwrap().valueType.omit({
@@ -14,21 +18,39 @@ const CredentialSchema = z.strictObject({
 	...MySchema.shape
 });
 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: toCredentialReferenceName(parsed.name)
+	return generateSimpleFactoryDefinition(data, {
+		schema: CredentialSchema,
+		factory: {
+			importName: "credential",
+			variableName: (parsed) => toCredentialReferenceName(parsed.name)
+		},
+		buildConfig(parsed) {
+			const { credentialId, ...rest } = parsed;
+			return {
+				id: credentialId,
+				...rest
+			};
+		}
 	});
-	const { credentialId, ...rest } = parsed;
-	for (const [k, v] of Object.entries({
-		id: credentialId,
-		...rest
-	})) addValueToObject(configObject, k, v);
-	return sourceFile;
 }
+const task = {
+	type: "credential",
+	collect(context) {
+		if (!context.project.credentialReferences) return [];
+		const credentialEntries = Object.entries(context.project.credentialReferences);
+		const fileNamesByCredentialId = buildSequentialNameFileNames(credentialEntries);
+		return credentialEntries.map(([credentialId, credentialData]) => ({
+			id: credentialId,
+			filePath: context.resolver.resolveOutputFilePath("credentials", credentialId, join(context.paths.credentialsDir, fileNamesByCredentialId[credentialId])),
+			payload: {
+				credentialId,
+				...credentialData
+			}
+		}));
+	},
+	generate: generateCredentialDefinition
+};
 
 //#endregion
-export { generateCredentialDefinition };
+export { generateCredentialDefinition, task };
 //# sourceMappingURL=credential-generator.js.map

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

@@ -1 +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"}
+{"version":3,"file":"credential-generator.js","names":[],"sources":["../../../../src/commands/pull-v4/generators/credential-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 { buildSequentialNameFileNames } from '../generation-resolver';\nimport type { GenerationTask } from '../generation-types';\nimport { generateSimpleFactoryDefinition } from '../simple-factory-generator';\nimport { 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  return generateSimpleFactoryDefinition(data, {\n    schema: CredentialSchema,\n    factory: {\n      importName: 'credential',\n      variableName: (parsed) => toCredentialReferenceName(parsed.name),\n    },\n    buildConfig(parsed) {\n      const { credentialId, ...rest } = parsed;\n      return {\n        id: credentialId,\n        ...rest,\n      };\n    },\n  });\n}\n\nexport const task = {\n  type: 'credential',\n  collect(context) {\n    if (!context.project.credentialReferences) {\n      return [];\n    }\n\n    const credentialEntries = Object.entries(context.project.credentialReferences);\n    const fileNamesByCredentialId = buildSequentialNameFileNames(credentialEntries);\n\n    return credentialEntries.map(([credentialId, credentialData]) => ({\n      id: credentialId,\n      filePath: context.resolver.resolveOutputFilePath(\n        'credentials',\n        credentialId,\n        join(context.paths.credentialsDir, fileNamesByCredentialId[credentialId])\n      ),\n      payload: {\n        credentialId,\n        ...credentialData,\n      } as Parameters<typeof generateCredentialDefinition>[0],\n    }));\n  },\n  generate: generateCredentialDefinition,\n} satisfies GenerationTask<Parameters<typeof generateCredentialDefinition>[0]>;\n"],"mappings":";;;;;;;;;AASA,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;AACxD,QAAO,gCAAgC,MAAM;EAC3C,QAAQ;EACR,SAAS;GACP,YAAY;GACZ,eAAe,WAAW,0BAA0B,OAAO,KAAK;GACjE;EACD,YAAY,QAAQ;GAClB,MAAM,EAAE,cAAc,GAAG,SAAS;AAClC,UAAO;IACL,IAAI;IACJ,GAAG;IACJ;;EAEJ,CAAC;;AAGJ,MAAa,OAAO;CAClB,MAAM;CACN,QAAQ,SAAS;AACf,MAAI,CAAC,QAAQ,QAAQ,qBACnB,QAAO,EAAE;EAGX,MAAM,oBAAoB,OAAO,QAAQ,QAAQ,QAAQ,qBAAqB;EAC9E,MAAM,0BAA0B,6BAA6B,kBAAkB;AAE/E,SAAO,kBAAkB,KAAK,CAAC,cAAc,qBAAqB;GAChE,IAAI;GACJ,UAAU,QAAQ,SAAS,sBACzB,eACA,cACA,KAAK,QAAQ,MAAM,gBAAgB,wBAAwB,cAAc,CAC1E;GACD,SAAS;IACP;IACA,GAAG;IACJ;GACF,EAAE;;CAEL,UAAU;CACX"}