@inkeep/agents-cli 0.58.10 → 0.58.13

package/dist/commands/pull-v4/component-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"component-registry.js","names":[],"sources":["../../../src/commands/pull-v4/component-registry.ts"],"sourcesContent":["/**\n * Unified Component Registry - Handles both unique variable names AND file paths\n *\n * This combines the functionality of ComponentTracker and ComponentMapper\n * into a single, clean system that handles:\n * 1. Unique variable name generation (with collision resolution)\n * 2. File path tracking for imports\n * 3. Reference resolution for code generation\n */\nimport type { FullProjectDefinition } from '@inkeep/agents-core';\n\nexport type ComponentType =\n  | 'agents'\n  | 'subAgents'\n  | 'tools'\n  | 'functionTools'\n  | 'functions'\n  | 'dataComponents'\n  | 'artifactComponents'\n  | 'statusComponents'\n  | 'environments'\n  | 'externalAgents'\n  | 'credentials'\n  | 'contextConfigs'\n  | 'fetchDefinitions'\n  | 'headers'\n  | 'models'\n  | 'skills'\n  | 'triggers'\n  | 'scheduledTriggers'\n  | 'project';\n\nexport interface ComponentInfo {\n  id: string; // Original component ID\n  name: string; // Unique variable name (guaranteed unique across project)\n  type: ComponentType; // Component type\n  filePath: string; // Relative file path (e.g., \"agents/foo.ts\")\n  exportName: string; // Export name (usually same as name)\n  isInline: boolean; // Whether component was defined inline (true) or exported (false)\n}\n\nexport class ComponentRegistry {\n  private components = new Map<string, ComponentInfo>();\n  private componentsByTypeAndId = new Map<string, ComponentInfo>(); // Type-aware lookup\n  private usedNames = new Set<string>(); // Global name registry for uniqueness\n\n  /**\n   * Register a component with both unique name and file path\n   */\n  register(\n    id: string,\n    type: ComponentType,\n    filePath: string,\n    exportName?: string,\n    isInline: boolean = false\n  ): ComponentInfo {\n    const typeKey = `${type}:${id}`;\n\n    // Check if this component already exists - if so, return the existing one\n    const existing = this.componentsByTypeAndId.get(typeKey);\n    if (existing) {\n      return existing;\n    }\n\n    let name: string;\n    let actualExportName: string;\n\n    // If exportName is provided (real discovered name), use it directly\n    // Otherwise, generate and ensure uniqueness for assumed names\n    if (exportName) {\n      // Real export name discovered from file - use it as-is\n      name = exportName;\n      actualExportName = exportName;\n    } else {\n      // No real export name - generate unique name with prefixes if needed\n      const baseName = this.toCamelCase(id);\n      const uniqueName = this.ensureUniqueName(baseName, type);\n      name = uniqueName;\n      actualExportName = uniqueName;\n    }\n\n    const info: ComponentInfo = {\n      id,\n      name,\n      type,\n      filePath,\n      exportName: actualExportName,\n      isInline,\n    };\n\n    // Store with both ID-only key (for backward compatibility) and type+ID key (for collision handling)\n    this.components.set(id, info);\n    this.componentsByTypeAndId.set(typeKey, info);\n\n    this.usedNames.add(name);\n    if (actualExportName && actualExportName !== name) {\n      this.usedNames.add(actualExportName);\n    }\n\n    return info;\n  }\n\n  /**\n   * Override a credential component with environment settings key\n   * This allows env settings registration to take precedence over standalone credentials\n   */\n  overrideCredentialWithEnvKey(id: string, filePath: string, envKey: string): ComponentInfo {\n    const typeKey = `credentials:${id}`;\n\n    const info: ComponentInfo = {\n      id,\n      type: 'credentials',\n      name: envKey, // Use the environment settings key\n      filePath,\n      exportName: envKey,\n      isInline: true,\n    };\n\n    // Override existing registration\n    this.componentsByTypeAndId.set(typeKey, info);\n    this.components.set(`${filePath}:${envKey}`, info);\n\n    return info;\n  }\n\n  /**\n   * Get component info by ID and type\n   */\n  get(id: string, type: ComponentType): ComponentInfo | undefined {\n    const typeKey = `${type}:${id}`;\n    return this.componentsByTypeAndId.get(typeKey);\n  }\n\n  /**\n   * Get component info by variable name (since variable names are globally unique)\n   */\n  getByVariableName(variableName: string): ComponentInfo | undefined {\n    for (const component of this.componentsByTypeAndId.values()) {\n      if (component.name === variableName) {\n        return component;\n      }\n    }\n    return undefined;\n  }\n\n  /**\n   * Get unique variable name for a component by ID and type\n   */\n  getVariableName(id: string, type: ComponentType): string | undefined {\n    const typeKey = `${type}:${id}`;\n    const result = this.componentsByTypeAndId.get(typeKey)?.name;\n\n    return result;\n  }\n\n  /**\n   * Get import statement for a component\n   */\n  getImportStatement(\n    fromFilePath: string,\n    componentId: string,\n    componentType: ComponentType\n  ): string | undefined {\n    const component = this.get(componentId, componentType);\n    if (!component) return undefined;\n\n    // Normalize both paths to the same format for comparison\n    const normalizedFrom = this.normalizeToProjectPath(fromFilePath);\n    const normalizedTo = this.normalizeToProjectPath(component.filePath);\n\n    // Skip import if same file\n    if (normalizedFrom === normalizedTo) {\n      return undefined;\n    }\n\n    const relativePath = this.calculateRelativeImport(fromFilePath, component.filePath);\n    const importStmt = `import { ${component.exportName} } from '${relativePath}';`;\n\n    return importStmt;\n  }\n\n  /**\n   * Format an array of references for code generation\n   */\n  formatReferencesForCode(\n    references: any[],\n    componentType: ComponentType,\n    style: { quotes: 'single' | 'double'; indentation: string },\n    indentLevel: number\n  ): string {\n    if (!Array.isArray(references) || references.length === 0) {\n      return '[]';\n    }\n\n    const variableNames: string[] = [];\n\n    for (const ref of references) {\n      const id = this.extractIdFromReference(ref);\n      if (id) {\n        const component = this.get(id, componentType);\n        if (component) {\n          variableNames.push(component.name);\n        } else {\n          console.warn(`ComponentRegistry: Component not found: ${id} (type: ${componentType})`);\n          variableNames.push(this.toCamelCase(id));\n        }\n      }\n    }\n\n    if (variableNames.length === 0) {\n      return '[]';\n    }\n\n    if (variableNames.length === 1) {\n      return `[${variableNames[0]}]`;\n    }\n\n    // Multi-line format\n    const { indentation } = style;\n    const indent = indentation.repeat(indentLevel);\n    const lines = ['['];\n\n    for (let i = 0; i < variableNames.length; i++) {\n      const isLast = i === variableNames.length - 1;\n      lines.push(`${indent}${variableNames[i]}${isLast ? '' : ','}`);\n    }\n\n    lines.push(`${indentation.repeat(indentLevel - 1)}]`);\n    return lines.join('\\n');\n  }\n\n  /**\n   * Get all import statements needed for a file\n   */\n  getImportsForFile(\n    fromFilePath: string,\n    referencedComponents: Array<{ id: string; type: ComponentType }>\n  ): string[] {\n    const imports: string[] = [];\n    const seenImports = new Set<string>(); // Deduplicate imports\n\n    for (const { id, type } of referencedComponents) {\n      const importStatement = this.getImportStatement(fromFilePath, id, type);\n      if (importStatement && !seenImports.has(importStatement)) {\n        imports.push(importStatement);\n        seenImports.add(importStatement);\n      }\n    }\n\n    return imports;\n  }\n\n  /**\n   * Get all component IDs referenced in arrays for import generation\n   */\n  getReferencedComponentIds(referenceArrays: any[][]): string[] {\n    const componentIds: string[] = [];\n\n    for (const refArray of referenceArrays) {\n      if (Array.isArray(refArray)) {\n        for (const ref of refArray) {\n          const id = this.extractIdFromReference(ref);\n          if (id) {\n            componentIds.push(id);\n          }\n        }\n      }\n    }\n\n    return componentIds;\n  }\n\n  /**\n   * Extract ID from a reference (string or object) based on component type\n   */\n  private extractIdFromReference(ref: any): string | null {\n    if (typeof ref === 'string') {\n      return ref;\n    }\n    if (typeof ref === 'object' && ref) {\n      // Handle different component types by their specific ID fields (confirmed from debug output)\n\n      // Tool references (MCP tools and function tools)\n      if (ref.toolId) return ref.toolId;\n\n      // Agent references (main agents and sub-agents)\n      if (ref.agentId) return ref.agentId;\n\n      // External agent references\n      if (ref.externalAgentId) return ref.externalAgentId;\n\n      // Credential store references (found in generated files)\n      if (ref.credentialStoreId) return ref.credentialStoreId;\n\n      // Status component references using type field (confirmed from debug output)\n      if (ref.type && !ref.agentId && !ref.toolId && !ref.externalAgentId) return ref.type;\n\n      // Generic ID field (fallback)\n      if (ref.id) return ref.id;\n\n      // Name field (fallback)\n      if (ref.name) return ref.name;\n\n      // For objects without recognized ID fields, warn and skip\n      console.warn('ComponentRegistry: Reference without recognized ID field:', ref);\n      return null;\n    }\n\n    return null;\n  }\n\n  /**\n   * Convert string to camelCase and ensure it's a valid JavaScript identifier\n   */\n  private toCamelCase(str: string): string {\n    return str\n      .toLowerCase()\n      .replace(/[-_](.)/g, (_, char) => char.toUpperCase())\n      .replace(/[^a-zA-Z0-9]/g, '')\n      .replace(/^[0-9]/, '_$&');\n  }\n\n  /**\n   * Ensure a name is unique by adding prefixes/suffixes if needed\n   */\n  private ensureUniqueName(baseName: string, type: ComponentType): string {\n    let uniqueName = baseName;\n    let counter = 1;\n\n    while (this.usedNames.has(uniqueName)) {\n      // Try with type prefix first\n      if (counter === 1) {\n        const typePrefix = this.getTypePrefix(type);\n        uniqueName = `${typePrefix}${baseName.charAt(0).toUpperCase() + baseName.slice(1)}`;\n      } else {\n        // Then try with counter\n        uniqueName = `${baseName}${counter}`;\n      }\n      counter++;\n\n      // Safety net to prevent infinite loops\n      if (counter > 100) {\n        uniqueName = `${baseName}_${Date.now()}`;\n        break;\n      }\n    }\n\n    return uniqueName;\n  }\n\n  /**\n   * Get type prefix for uniqueness resolution\n   */\n  private getTypePrefix(type: ComponentType): string {\n    switch (type) {\n      case 'agents':\n        return 'agent';\n      case 'subAgents':\n        return 'sub';\n      case 'externalAgents':\n        return 'ext';\n      case 'tools':\n        return 'tool';\n      case 'functionTools':\n        return 'func';\n      case 'functions':\n        return 'func';\n      case 'dataComponents':\n        return 'data';\n      case 'artifactComponents':\n        return 'artifact';\n      case 'statusComponents':\n        return 'status';\n      case 'environments':\n        return 'env';\n      case 'credentials':\n        return 'cred';\n      case 'contextConfigs':\n        return 'context';\n      case 'fetchDefinitions':\n        return 'fetch';\n      case 'headers':\n        return 'header';\n      case 'models':\n        return 'model';\n      case 'triggers':\n        return 'trigger';\n      case 'project':\n        return 'project';\n      default:\n        return 'comp';\n    }\n  }\n\n  /**\n   * Normalize path to project-relative format using existing registry paths as templates\n   */\n  private normalizeToProjectPath(filePath: string): string {\n    // If already in project format (no slashes before known directories), return as-is\n    if (!filePath.includes('/') || filePath.match(/^(agents|tools|environments|data-components)/)) {\n      return filePath;\n    }\n\n    // Find project directories by looking at existing registry paths\n    const knownDirs = new Set<string>();\n    for (const component of this.components.values()) {\n      const firstDir = component.filePath.split('/')[0];\n      if (firstDir) knownDirs.add(firstDir);\n    }\n\n    // Find the last occurrence of any known project directory\n    const segments = filePath.split('/');\n    for (let i = segments.length - 1; i >= 0; i--) {\n      if (knownDirs.has(segments[i])) {\n        return segments.slice(i).join('/');\n      }\n    }\n\n    // Fallback: return the filename\n    return segments[segments.length - 1] || filePath;\n  }\n\n  /**\n   * Calculate relative import path between files\n   */\n  private calculateRelativeImport(fromPath: string, toPath: string): string {\n    // Remove .ts extensions for calculation\n    const fromParts = fromPath.replace('.ts', '').split('/');\n    const toParts = toPath.replace('.ts', '').split('/');\n\n    // Remove filename from fromPath (keep directory only)\n    const fromDir = fromParts.slice(0, -1);\n    const toFile = toParts;\n\n    // Find common path prefix\n    let commonLength = 0;\n    while (\n      commonLength < fromDir.length &&\n      commonLength < toFile.length - 1 &&\n      fromDir[commonLength] === toFile[commonLength]\n    ) {\n      commonLength++;\n    }\n\n    // Calculate relative path\n    let relativePath = '';\n\n    // Go up for remaining directories in fromDir after common path\n    const upLevels = fromDir.length - commonLength;\n    for (let i = 0; i < upLevels; i++) {\n      relativePath += '../';\n    }\n\n    // Add remaining path from toFile after common path\n    const remainingPath = toFile.slice(commonLength).join('/');\n    relativePath += remainingPath;\n\n    // Clean up path format\n    if (relativePath.startsWith('../')) {\n      return relativePath;\n    }\n    return `./${relativePath}`;\n  }\n\n  /**\n   * Get all components for debugging\n   */\n  getAllComponents(): ComponentInfo[] {\n    return Array.from(this.componentsByTypeAndId.values());\n  }\n\n  /**\n   * Get all components in a specific file\n   */\n  getComponentsInFile(filePath: string): ComponentInfo[] {\n    return Array.from(this.componentsByTypeAndId.values()).filter(\n      (component) => component.filePath === filePath\n    );\n  }\n\n  /**\n   * Remove a component from the registry\n   */\n  removeComponent(type: string, id: string): void {\n    const key = `${type}:${id}`;\n    const component = this.componentsByTypeAndId.get(key);\n\n    if (component) {\n      // Remove from both maps\n      this.componentsByTypeAndId.delete(key);\n      this.components.delete(component.name);\n\n      // Remove from used names if no other component uses it\n      const nameStillUsed = Array.from(this.componentsByTypeAndId.values()).some(\n        (comp) => comp.name === component.name\n      );\n      if (!nameStillUsed) {\n        this.usedNames.delete(component.name);\n      }\n    }\n  }\n\n  /**\n   * Clear all components (for testing)\n   */\n  clear(): void {\n    this.components.clear();\n    this.componentsByTypeAndId.clear();\n    this.usedNames.clear();\n  }\n}\n\n/**\n * Register all components from a project with their file paths\n */\nexport function registerAllComponents(\n  project: FullProjectDefinition,\n  registry: ComponentRegistry\n): void {\n  // Register project\n  registry.register(project.id, 'project', 'index.ts');\n\n  // Register credentials\n  if (project.credentialReferences) {\n    for (const credId of Object.keys(project.credentialReferences)) {\n      registry.register(credId, 'credentials', `credentials/${credId}.ts`);\n    }\n  }\n\n  // Register tools\n  if (project.tools) {\n    for (const toolId of Object.keys(project.tools)) {\n      registry.register(toolId, 'tools', `tools/${toolId}.ts`);\n    }\n  }\n\n  // Register function tools - functionTools has the name/description, functions has the code\n  // We generate files based on functionTools IDs since that's the user-facing entity\n  const registeredFunctionToolIds = new Set<string>();\n\n  // Register functionTools first (they have the name that identifies the tool)\n  if (project.functionTools) {\n    for (const funcToolId of Object.keys(project.functionTools)) {\n      registry.register(funcToolId, 'functionTools', `tools/functions/${funcToolId}.ts`);\n      registeredFunctionToolIds.add(funcToolId);\n    }\n  }\n\n  // Also check agent-level functionTools\n  if (project.agents) {\n    for (const agentData of Object.values(project.agents)) {\n      const agentFunctionTools = (agentData as any).functionTools;\n      if (agentFunctionTools) {\n        for (const funcToolId of Object.keys(agentFunctionTools)) {\n          if (!registeredFunctionToolIds.has(funcToolId)) {\n            registry.register(funcToolId, 'functionTools', `tools/functions/${funcToolId}.ts`);\n            registeredFunctionToolIds.add(funcToolId);\n          }\n        }\n      }\n    }\n  }\n\n  // Register data components\n  if (project.dataComponents) {\n    for (const componentId of Object.keys(project.dataComponents)) {\n      registry.register(componentId, 'dataComponents', `data-components/${componentId}.ts`);\n    }\n  }\n\n  // Register artifact components\n  if (project.artifactComponents) {\n    for (const componentId of Object.keys(project.artifactComponents)) {\n      registry.register(componentId, 'artifactComponents', `artifact-components/${componentId}.ts`);\n    }\n  }\n\n  // Register external agents\n  if (project.externalAgents) {\n    for (const extAgentId of Object.keys(project.externalAgents)) {\n      registry.register(extAgentId, 'externalAgents', `external-agents/${extAgentId}.ts`);\n    }\n  }\n\n  // Register skills\n  if (project.skills) {\n    for (const skillId of Object.keys(project.skills)) {\n      registry.register(skillId, 'skills', `skills/${skillId}.md`);\n    }\n  }\n\n  // Register extracted status components\n  const statusComponents = extractStatusComponents(project);\n  for (const statusId of Object.keys(statusComponents)) {\n    registry.register(statusId, 'statusComponents', `status-components/${statusId}.ts`);\n  }\n\n  // Register agents\n  if (project.agents) {\n    for (const agentId of Object.keys(project.agents)) {\n      registry.register(agentId, 'agents', `agents/${agentId}.ts`);\n    }\n  }\n\n  // Register triggers (agent-scoped)\n  if (project.agents) {\n    for (const agentData of Object.values(project.agents)) {\n      const agentTriggers = (agentData as any).triggers;\n      if (agentTriggers) {\n        for (const triggerId of Object.keys(agentTriggers)) {\n          registry.register(triggerId, 'triggers', `agents/triggers/${triggerId}.ts`);\n        }\n      }\n    }\n  }\n\n  // Register extracted sub-agents\n  const subAgents = extractSubAgents(project);\n  for (const subAgentId of Object.keys(subAgents)) {\n    registry.register(subAgentId, 'subAgents', `agents/sub-agents/${subAgentId}.ts`);\n  }\n\n  // Register extracted context configs\n  const contextConfigs = extractContextConfigs(project);\n  for (const contextId of Object.keys(contextConfigs)) {\n    registry.register(contextId, 'contextConfigs', `context-configs/${contextId}.ts`);\n  }\n}\n\n/**\n * Extract status components from project agents\n */\nfunction extractStatusComponents(project: FullProjectDefinition): Record<string, any> {\n  const statusComponents: Record<string, any> = {};\n\n  if (project.agents) {\n    for (const agentData of Object.values(project.agents)) {\n      if (agentData.statusUpdates?.statusComponents) {\n        // statusComponents is an array that can contain strings or objects\n        for (const statusComp of agentData.statusUpdates.statusComponents) {\n          let statusId: string;\n\n          if (typeof statusComp === 'string') {\n            // Direct string reference to status component ID\n            statusId = statusComp;\n          } else if (typeof statusComp === 'object' && statusComp) {\n            // Object with id, type, or name field\n            statusId = statusComp.type;\n          } else {\n            continue;\n          }\n\n          if (statusId && !statusComponents[statusId]) {\n            // Use the actual status component data instead of creating dummy data\n            statusComponents[statusId] = {\n              // Include any other properties from the actual data first\n              ...statusComp,\n              id: statusId,\n              type: statusComp.type || statusId,\n              description: statusComp.description || `Status component for ${statusId}`,\n              detailsSchema: statusComp.detailsSchema,\n            };\n          }\n        }\n      }\n    }\n  }\n\n  return statusComponents;\n}\n\n/**\n * Extract sub-agents from project agents\n */\nexport function extractSubAgents(project: FullProjectDefinition): Record<string, any> {\n  const subAgents: Record<string, any> = {};\n\n  if (project.agents) {\n    for (const agentData of Object.values(project.agents)) {\n      if (agentData.subAgents) {\n        for (const [subAgentId, subAgentData] of Object.entries(agentData.subAgents)) {\n          subAgents[subAgentId] = subAgentData;\n        }\n      }\n    }\n  }\n\n  return subAgents;\n}\n\n/**\n * Extract context configs from project agents\n */\nfunction extractContextConfigs(project: FullProjectDefinition): Record<string, any> {\n  const contextConfigs: Record<string, any> = {};\n\n  if (project.agents) {\n    for (const [agentId, agentData] of Object.entries(project.agents)) {\n      if (agentData.contextConfig) {\n        // Use the actual contextConfig.id (now required)\n        const contextConfigId = agentData.contextConfig.id;\n        if (contextConfigId) {\n          contextConfigs[contextConfigId] = agentData.contextConfig;\n        } else {\n          console.warn(`contextConfig for agent ${agentId} is missing required 'id' field`);\n        }\n      }\n    }\n  }\n\n  return contextConfigs;\n}\n\n/**\n * Find sub-agent data with parent agent info for contextConfig resolution\n */\nexport function findSubAgentWithParent(\n  project: FullProjectDefinition,\n  subAgentId: string\n): { subAgentData: any; parentAgentId: string; contextConfigData?: any } | undefined {\n  if (project.agents) {\n    for (const [agentId, agentData] of Object.entries(project.agents)) {\n      if (agentData.subAgents?.[subAgentId]) {\n        // Get contextConfig data if parent agent has one with an ID\n        const contextConfigData = agentData.contextConfig?.id ? agentData.contextConfig : undefined;\n\n        return {\n          subAgentData: agentData.subAgents[subAgentId],\n          parentAgentId: agentId,\n          contextConfigData,\n        };\n      }\n    }\n  }\n  return undefined;\n}\n"],"mappings":";AAyCA,IAAa,oBAAb,MAA+B;CAC7B,AAAQ,6BAAa,IAAI,KAA4B;CACrD,AAAQ,wCAAwB,IAAI,KAA4B;CAChE,AAAQ,4BAAY,IAAI,KAAa;;;;CAKrC,SACE,IACA,MACA,UACA,YACA,WAAoB,OACL;EACf,MAAM,UAAU,GAAG,KAAK,GAAG;EAG3B,MAAM,WAAW,KAAK,sBAAsB,IAAI,QAAQ;AACxD,MAAI,SACF,QAAO;EAGT,IAAI;EACJ,IAAI;AAIJ,MAAI,YAAY;AAEd,UAAO;AACP,sBAAmB;SACd;GAEL,MAAM,WAAW,KAAK,YAAY,GAAG;GACrC,MAAM,aAAa,KAAK,iBAAiB,UAAU,KAAK;AACxD,UAAO;AACP,sBAAmB;;EAGrB,MAAM,OAAsB;GAC1B;GACA;GACA;GACA;GACA,YAAY;GACZ;GACD;AAGD,OAAK,WAAW,IAAI,IAAI,KAAK;AAC7B,OAAK,sBAAsB,IAAI,SAAS,KAAK;AAE7C,OAAK,UAAU,IAAI,KAAK;AACxB,MAAI,oBAAoB,qBAAqB,KAC3C,MAAK,UAAU,IAAI,iBAAiB;AAGtC,SAAO;;;;;;CAOT,6BAA6B,IAAY,UAAkB,QAA+B;EACxF,MAAM,UAAU,eAAe;EAE/B,MAAM,OAAsB;GAC1B;GACA,MAAM;GACN,MAAM;GACN;GACA,YAAY;GACZ,UAAU;GACX;AAGD,OAAK,sBAAsB,IAAI,SAAS,KAAK;AAC7C,OAAK,WAAW,IAAI,GAAG,SAAS,GAAG,UAAU,KAAK;AAElD,SAAO;;;;;CAMT,IAAI,IAAY,MAAgD;EAC9D,MAAM,UAAU,GAAG,KAAK,GAAG;AAC3B,SAAO,KAAK,sBAAsB,IAAI,QAAQ;;;;;CAMhD,kBAAkB,cAAiD;AACjE,OAAK,MAAM,aAAa,KAAK,sBAAsB,QAAQ,CACzD,KAAI,UAAU,SAAS,aACrB,QAAO;;;;;CASb,gBAAgB,IAAY,MAAyC;EACnE,MAAM,UAAU,GAAG,KAAK,GAAG;AAG3B,SAFe,KAAK,sBAAsB,IAAI,QAAQ,EAAE;;;;;CAQ1D,mBACE,cACA,aACA,eACoB;EACpB,MAAM,YAAY,KAAK,IAAI,aAAa,cAAc;AACtD,MAAI,CAAC,UAAW,QAAO;AAOvB,MAJuB,KAAK,uBAAuB,aAAa,KAC3C,KAAK,uBAAuB,UAAU,SAAS,CAIlE;EAGF,MAAM,eAAe,KAAK,wBAAwB,cAAc,UAAU,SAAS;AAGnF,SAFmB,YAAY,UAAU,WAAW,WAAW,aAAa;;;;;CAQ9E,wBACE,YACA,eACA,OACA,aACQ;AACR,MAAI,CAAC,MAAM,QAAQ,WAAW,IAAI,WAAW,WAAW,EACtD,QAAO;EAGT,MAAM,gBAA0B,EAAE;AAElC,OAAK,MAAM,OAAO,YAAY;GAC5B,MAAM,KAAK,KAAK,uBAAuB,IAAI;AAC3C,OAAI,IAAI;IACN,MAAM,YAAY,KAAK,IAAI,IAAI,cAAc;AAC7C,QAAI,UACF,eAAc,KAAK,UAAU,KAAK;SAC7B;AACL,aAAQ,KAAK,2CAA2C,GAAG,UAAU,cAAc,GAAG;AACtF,mBAAc,KAAK,KAAK,YAAY,GAAG,CAAC;;;;AAK9C,MAAI,cAAc,WAAW,EAC3B,QAAO;AAGT,MAAI,cAAc,WAAW,EAC3B,QAAO,IAAI,cAAc,GAAG;EAI9B,MAAM,EAAE,gBAAgB;EACxB,MAAM,SAAS,YAAY,OAAO,YAAY;EAC9C,MAAM,QAAQ,CAAC,IAAI;AAEnB,OAAK,IAAI,IAAI,GAAG,IAAI,cAAc,QAAQ,KAAK;GAC7C,MAAM,SAAS,MAAM,cAAc,SAAS;AAC5C,SAAM,KAAK,GAAG,SAAS,cAAc,KAAK,SAAS,KAAK,MAAM;;AAGhE,QAAM,KAAK,GAAG,YAAY,OAAO,cAAc,EAAE,CAAC,GAAG;AACrD,SAAO,MAAM,KAAK,KAAK;;;;;CAMzB,kBACE,cACA,sBACU;EACV,MAAM,UAAoB,EAAE;EAC5B,MAAM,8BAAc,IAAI,KAAa;AAErC,OAAK,MAAM,EAAE,IAAI,UAAU,sBAAsB;GAC/C,MAAM,kBAAkB,KAAK,mBAAmB,cAAc,IAAI,KAAK;AACvE,OAAI,mBAAmB,CAAC,YAAY,IAAI,gBAAgB,EAAE;AACxD,YAAQ,KAAK,gBAAgB;AAC7B,gBAAY,IAAI,gBAAgB;;;AAIpC,SAAO;;;;;CAMT,0BAA0B,iBAAoC;EAC5D,MAAM,eAAyB,EAAE;AAEjC,OAAK,MAAM,YAAY,gBACrB,KAAI,MAAM,QAAQ,SAAS,CACzB,MAAK,MAAM,OAAO,UAAU;GAC1B,MAAM,KAAK,KAAK,uBAAuB,IAAI;AAC3C,OAAI,GACF,cAAa,KAAK,GAAG;;AAM7B,SAAO;;;;;CAMT,AAAQ,uBAAuB,KAAyB;AACtD,MAAI,OAAO,QAAQ,SACjB,QAAO;AAET,MAAI,OAAO,QAAQ,YAAY,KAAK;AAIlC,OAAI,IAAI,OAAQ,QAAO,IAAI;AAG3B,OAAI,IAAI,QAAS,QAAO,IAAI;AAG5B,OAAI,IAAI,gBAAiB,QAAO,IAAI;AAGpC,OAAI,IAAI,kBAAmB,QAAO,IAAI;AAGtC,OAAI,IAAI,QAAQ,CAAC,IAAI,WAAW,CAAC,IAAI,UAAU,CAAC,IAAI,gBAAiB,QAAO,IAAI;AAGhF,OAAI,IAAI,GAAI,QAAO,IAAI;AAGvB,OAAI,IAAI,KAAM,QAAO,IAAI;AAGzB,WAAQ,KAAK,6DAA6D,IAAI;AAC9E,UAAO;;AAGT,SAAO;;;;;CAMT,AAAQ,YAAY,KAAqB;AACvC,SAAO,IACJ,aAAa,CACb,QAAQ,aAAa,GAAG,SAAS,KAAK,aAAa,CAAC,CACpD,QAAQ,iBAAiB,GAAG,CAC5B,QAAQ,UAAU,MAAM;;;;;CAM7B,AAAQ,iBAAiB,UAAkB,MAA6B;EACtE,IAAI,aAAa;EACjB,IAAI,UAAU;AAEd,SAAO,KAAK,UAAU,IAAI,WAAW,EAAE;AAErC,OAAI,YAAY,EAEd,cAAa,GADM,KAAK,cAAc,KAAK,GACd,SAAS,OAAO,EAAE,CAAC,aAAa,GAAG,SAAS,MAAM,EAAE;OAGjF,cAAa,GAAG,WAAW;AAE7B;AAGA,OAAI,UAAU,KAAK;AACjB,iBAAa,GAAG,SAAS,GAAG,KAAK,KAAK;AACtC;;;AAIJ,SAAO;;;;;CAMT,AAAQ,cAAc,MAA6B;AACjD,UAAQ,MAAR;GACE,KAAK,SACH,QAAO;GACT,KAAK,YACH,QAAO;GACT,KAAK,iBACH,QAAO;GACT,KAAK,QACH,QAAO;GACT,KAAK,gBACH,QAAO;GACT,KAAK,YACH,QAAO;GACT,KAAK,iBACH,QAAO;GACT,KAAK,qBACH,QAAO;GACT,KAAK,mBACH,QAAO;GACT,KAAK,eACH,QAAO;GACT,KAAK,cACH,QAAO;GACT,KAAK,iBACH,QAAO;GACT,KAAK,mBACH,QAAO;GACT,KAAK,UACH,QAAO;GACT,KAAK,SACH,QAAO;GACT,KAAK,WACH,QAAO;GACT,KAAK,UACH,QAAO;GACT,QACE,QAAO;;;;;;CAOb,AAAQ,uBAAuB,UAA0B;AAEvD,MAAI,CAAC,SAAS,SAAS,IAAI,IAAI,SAAS,MAAM,+CAA+C,CAC3F,QAAO;EAIT,MAAM,4BAAY,IAAI,KAAa;AACnC,OAAK,MAAM,aAAa,KAAK,WAAW,QAAQ,EAAE;GAChD,MAAM,WAAW,UAAU,SAAS,MAAM,IAAI,CAAC;AAC/C,OAAI,SAAU,WAAU,IAAI,SAAS;;EAIvC,MAAM,WAAW,SAAS,MAAM,IAAI;AACpC,OAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,IACxC,KAAI,UAAU,IAAI,SAAS,GAAG,CAC5B,QAAO,SAAS,MAAM,EAAE,CAAC,KAAK,IAAI;AAKtC,SAAO,SAAS,SAAS,SAAS,MAAM;;;;;CAM1C,AAAQ,wBAAwB,UAAkB,QAAwB;EAExE,MAAM,YAAY,SAAS,QAAQ,OAAO,GAAG,CAAC,MAAM,IAAI;EACxD,MAAM,UAAU,OAAO,QAAQ,OAAO,GAAG,CAAC,MAAM,IAAI;EAGpD,MAAM,UAAU,UAAU,MAAM,GAAG,GAAG;EACtC,MAAM,SAAS;EAGf,IAAI,eAAe;AACnB,SACE,eAAe,QAAQ,UACvB,eAAe,OAAO,SAAS,KAC/B,QAAQ,kBAAkB,OAAO,cAEjC;EAIF,IAAI,eAAe;EAGnB,MAAM,WAAW,QAAQ,SAAS;AAClC,OAAK,IAAI,IAAI,GAAG,IAAI,UAAU,IAC5B,iBAAgB;EAIlB,MAAM,gBAAgB,OAAO,MAAM,aAAa,CAAC,KAAK,IAAI;AAC1D,kBAAgB;AAGhB,MAAI,aAAa,WAAW,MAAM,CAChC,QAAO;AAET,SAAO,KAAK;;;;;CAMd,mBAAoC;AAClC,SAAO,MAAM,KAAK,KAAK,sBAAsB,QAAQ,CAAC;;;;;CAMxD,oBAAoB,UAAmC;AACrD,SAAO,MAAM,KAAK,KAAK,sBAAsB,QAAQ,CAAC,CAAC,QACpD,cAAc,UAAU,aAAa,SACvC;;;;;CAMH,gBAAgB,MAAc,IAAkB;EAC9C,MAAM,MAAM,GAAG,KAAK,GAAG;EACvB,MAAM,YAAY,KAAK,sBAAsB,IAAI,IAAI;AAErD,MAAI,WAAW;AAEb,QAAK,sBAAsB,OAAO,IAAI;AACtC,QAAK,WAAW,OAAO,UAAU,KAAK;AAMtC,OAAI,CAHkB,MAAM,KAAK,KAAK,sBAAsB,QAAQ,CAAC,CAAC,MACnE,SAAS,KAAK,SAAS,UAAU,KACnC,CAEC,MAAK,UAAU,OAAO,UAAU,KAAK;;;;;;CAQ3C,QAAc;AACZ,OAAK,WAAW,OAAO;AACvB,OAAK,sBAAsB,OAAO;AAClC,OAAK,UAAU,OAAO;;;;;;AAO1B,SAAgB,sBACd,SACA,UACM;AAEN,UAAS,SAAS,QAAQ,IAAI,WAAW,WAAW;AAGpD,KAAI,QAAQ,qBACV,MAAK,MAAM,UAAU,OAAO,KAAK,QAAQ,qBAAqB,CAC5D,UAAS,SAAS,QAAQ,eAAe,eAAe,OAAO,KAAK;AAKxE,KAAI,QAAQ,MACV,MAAK,MAAM,UAAU,OAAO,KAAK,QAAQ,MAAM,CAC7C,UAAS,SAAS,QAAQ,SAAS,SAAS,OAAO,KAAK;CAM5D,MAAM,4CAA4B,IAAI,KAAa;AAGnD,KAAI,QAAQ,cACV,MAAK,MAAM,cAAc,OAAO,KAAK,QAAQ,cAAc,EAAE;AAC3D,WAAS,SAAS,YAAY,iBAAiB,mBAAmB,WAAW,KAAK;AAClF,4BAA0B,IAAI,WAAW;;AAK7C,KAAI,QAAQ,OACV,MAAK,MAAM,aAAa,OAAO,OAAO,QAAQ,OAAO,EAAE;EACrD,MAAM,qBAAsB,UAAkB;AAC9C,MAAI,oBACF;QAAK,MAAM,cAAc,OAAO,KAAK,mBAAmB,CACtD,KAAI,CAAC,0BAA0B,IAAI,WAAW,EAAE;AAC9C,aAAS,SAAS,YAAY,iBAAiB,mBAAmB,WAAW,KAAK;AAClF,8BAA0B,IAAI,WAAW;;;;AAQnD,KAAI,QAAQ,eACV,MAAK,MAAM,eAAe,OAAO,KAAK,QAAQ,eAAe,CAC3D,UAAS,SAAS,aAAa,kBAAkB,mBAAmB,YAAY,KAAK;AAKzF,KAAI,QAAQ,mBACV,MAAK,MAAM,eAAe,OAAO,KAAK,QAAQ,mBAAmB,CAC/D,UAAS,SAAS,aAAa,sBAAsB,uBAAuB,YAAY,KAAK;AAKjG,KAAI,QAAQ,eACV,MAAK,MAAM,cAAc,OAAO,KAAK,QAAQ,eAAe,CAC1D,UAAS,SAAS,YAAY,kBAAkB,mBAAmB,WAAW,KAAK;AAKvF,KAAI,QAAQ,OACV,MAAK,MAAM,WAAW,OAAO,KAAK,QAAQ,OAAO,CAC/C,UAAS,SAAS,SAAS,UAAU,UAAU,QAAQ,KAAK;CAKhE,MAAM,mBAAmB,wBAAwB,QAAQ;AACzD,MAAK,MAAM,YAAY,OAAO,KAAK,iBAAiB,CAClD,UAAS,SAAS,UAAU,oBAAoB,qBAAqB,SAAS,KAAK;AAIrF,KAAI,QAAQ,OACV,MAAK,MAAM,WAAW,OAAO,KAAK,QAAQ,OAAO,CAC/C,UAAS,SAAS,SAAS,UAAU,UAAU,QAAQ,KAAK;AAKhE,KAAI,QAAQ,OACV,MAAK,MAAM,aAAa,OAAO,OAAO,QAAQ,OAAO,EAAE;EACrD,MAAM,gBAAiB,UAAkB;AACzC,MAAI,cACF,MAAK,MAAM,aAAa,OAAO,KAAK,cAAc,CAChD,UAAS,SAAS,WAAW,YAAY,mBAAmB,UAAU,KAAK;;CAOnF,MAAM,YAAY,iBAAiB,QAAQ;AAC3C,MAAK,MAAM,cAAc,OAAO,KAAK,UAAU,CAC7C,UAAS,SAAS,YAAY,aAAa,qBAAqB,WAAW,KAAK;CAIlF,MAAM,iBAAiB,sBAAsB,QAAQ;AACrD,MAAK,MAAM,aAAa,OAAO,KAAK,eAAe,CACjD,UAAS,SAAS,WAAW,kBAAkB,mBAAmB,UAAU,KAAK;;;;;AAOrF,SAAS,wBAAwB,SAAqD;CACpF,MAAM,mBAAwC,EAAE;AAEhD,KAAI,QAAQ,QACV;OAAK,MAAM,aAAa,OAAO,OAAO,QAAQ,OAAO,CACnD,KAAI,UAAU,eAAe,iBAE3B,MAAK,MAAM,cAAc,UAAU,cAAc,kBAAkB;GACjE,IAAI;AAEJ,OAAI,OAAO,eAAe,SAExB,YAAW;YACF,OAAO,eAAe,YAAY,WAE3C,YAAW,WAAW;OAEtB;AAGF,OAAI,YAAY,CAAC,iBAAiB,UAEhC,kBAAiB,YAAY;IAE3B,GAAG;IACH,IAAI;IACJ,MAAM,WAAW,QAAQ;IACzB,aAAa,WAAW,eAAe,wBAAwB;IAC/D,eAAe,WAAW;IAC3B;;;AAOX,QAAO;;;;;AAMT,SAAgB,iBAAiB,SAAqD;CACpF,MAAM,YAAiC,EAAE;AAEzC,KAAI,QAAQ,QACV;OAAK,MAAM,aAAa,OAAO,OAAO,QAAQ,OAAO,CACnD,KAAI,UAAU,UACZ,MAAK,MAAM,CAAC,YAAY,iBAAiB,OAAO,QAAQ,UAAU,UAAU,CAC1E,WAAU,cAAc;;AAMhC,QAAO;;;;;AAMT,SAAS,sBAAsB,SAAqD;CAClF,MAAM,iBAAsC,EAAE;AAE9C,KAAI,QAAQ,QACV;OAAK,MAAM,CAAC,SAAS,cAAc,OAAO,QAAQ,QAAQ,OAAO,CAC/D,KAAI,UAAU,eAAe;GAE3B,MAAM,kBAAkB,UAAU,cAAc;AAChD,OAAI,gBACF,gBAAe,mBAAmB,UAAU;OAE5C,SAAQ,KAAK,2BAA2B,QAAQ,iCAAiC;;;AAMzF,QAAO;;;;;AAMT,SAAgB,uBACd,SACA,YACmF;AACnF,KAAI,QAAQ,QACV;OAAK,MAAM,CAAC,SAAS,cAAc,OAAO,QAAQ,QAAQ,OAAO,CAC/D,KAAI,UAAU,YAAY,aAAa;GAErC,MAAM,oBAAoB,UAAU,eAAe,KAAK,UAAU,gBAAgB;AAElF,UAAO;IACL,cAAc,UAAU,UAAU;IAClC,eAAe;IACf;IACD"}

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

@@ -0,0 +1,186 @@
+import { collectTemplateVariableNames, createUniqueReferenceName, isPlainObject, toCamelCase, toKebabCase, toTriggerReferenceName } from "../utils.js";
+
+//#region src/commands/pull-v4/generators/agent-generator.helpers.ts
+function collectTemplateVariableNamesFromFields(values) {
+	const variables = [];
+	for (const value of values) {
+		if (typeof value !== "string") continue;
+		variables.push(...collectTemplateVariableNames(value));
+	}
+	return variables;
+}
+function extractIds(value) {
+	if (Array.isArray(value)) return value.map((item) => {
+		if (typeof item === "string") return item;
+		if (isPlainObject(item) && typeof item.id === "string") return item.id;
+		return null;
+	}).filter((id) => Boolean(id));
+	return Object.keys(value);
+}
+function extractContextConfigId(contextConfig) {
+	if (!contextConfig) return;
+	if (typeof contextConfig === "string") return contextConfig;
+	return contextConfig.id;
+}
+function addSubAgentImports(sourceFile, referenceNames, importNames, modulePathOverrides) {
+	for (const [subAgentId, referenceName] of referenceNames) {
+		const importName = importNames.get(subAgentId);
+		if (!importName) continue;
+		sourceFile.addImportDeclaration({
+			namedImports: [importName === referenceName ? importName : {
+				name: importName,
+				alias: referenceName
+			}],
+			moduleSpecifier: `./sub-agents/${modulePathOverrides?.[subAgentId] ?? subAgentId}`
+		});
+	}
+}
+function addTriggerImports(sourceFile, referenceNames, importRefs) {
+	for (const [triggerId, referenceName] of referenceNames) {
+		const importRef = importRefs.get(triggerId);
+		if (!importRef) continue;
+		const { importName, modulePath } = importRef;
+		sourceFile.addImportDeclaration({
+			namedImports: [importName === referenceName ? importName : {
+				name: importName,
+				alias: referenceName
+			}],
+			moduleSpecifier: `./triggers/${modulePath}`
+		});
+	}
+}
+function addScheduledTriggerImports(sourceFile, referenceNames, importRefs) {
+	for (const [scheduledTriggerId, referenceName] of referenceNames) {
+		const importRef = importRefs.get(scheduledTriggerId);
+		if (!importRef) continue;
+		const { importName, modulePath } = importRef;
+		sourceFile.addImportDeclaration({
+			namedImports: [importName === referenceName ? importName : {
+				name: importName,
+				alias: referenceName
+			}],
+			moduleSpecifier: `./scheduled-triggers/${modulePath}`
+		});
+	}
+}
+function extractStatusComponentIds(statusUpdates) {
+	if (!statusUpdates?.statusComponents?.length) return [];
+	const statusComponentIds = statusUpdates.statusComponents.map(resolveStatusComponentId);
+	return [...new Set(statusComponentIds)];
+}
+function resolveStatusComponentId(statusComponent) {
+	const id = typeof statusComponent === "string" ? statusComponent : statusComponent.id || statusComponent.type;
+	if (!id) throw new Error(`Unable to resolve status component with id ${JSON.stringify(statusComponent)}`);
+	return id;
+}
+function addStatusComponentImports(sourceFile, referenceNames) {
+	for (const [statusComponentId, referenceName] of referenceNames) {
+		const importName = toCamelCase(statusComponentId);
+		sourceFile.addImportDeclaration({
+			namedImports: [importName === referenceName ? importName : {
+				name: importName,
+				alias: referenceName
+			}],
+			moduleSpecifier: `../status-components/${statusComponentId}`
+		});
+	}
+}
+function createSubAgentReferenceMaps(ids, reservedNames, conflictSuffix, overrides) {
+	const referenceNames = /* @__PURE__ */ new Map();
+	const importNames = /* @__PURE__ */ new Map();
+	for (const id of ids) {
+		if (referenceNames.has(id)) continue;
+		const override = overrides?.[id];
+		const importName = override?.name ?? toCamelCase(id);
+		const isLocal = override?.local === true;
+		const referenceName = isLocal ? importName : createUniqueReferenceName(importName, reservedNames, conflictSuffix);
+		if (isLocal) reservedNames.add(referenceName);
+		else importNames.set(id, importName);
+		referenceNames.set(id, referenceName);
+	}
+	return {
+		referenceNames,
+		importNames
+	};
+}
+function createReferenceNameMap(ids, reservedNames, conflictSuffix) {
+	const map = /* @__PURE__ */ new Map();
+	for (const id of ids) {
+		if (map.has(id)) continue;
+		map.set(id, createUniqueReferenceName(toCamelCase(id), reservedNames, conflictSuffix));
+	}
+	return map;
+}
+function createScheduledTriggerReferenceMaps(scheduledTriggers, reservedNames) {
+	const referenceNames = /* @__PURE__ */ new Map();
+	const importRefs = /* @__PURE__ */ new Map();
+	if (!scheduledTriggers || !isPlainObject(scheduledTriggers)) return {
+		referenceNames,
+		importRefs
+	};
+	const moduleNameCounts = /* @__PURE__ */ new Map();
+	for (const [scheduledTriggerId, scheduledTriggerData] of Object.entries(scheduledTriggers)) {
+		if (referenceNames.has(scheduledTriggerId)) continue;
+		const scheduledTriggerRecord = isPlainObject(scheduledTriggerData) ? scheduledTriggerData : void 0;
+		const scheduledTriggerName = typeof scheduledTriggerRecord?.name === "string" && scheduledTriggerRecord.name.length > 0 ? scheduledTriggerRecord.name : scheduledTriggerId;
+		const importName = toTriggerReferenceName(scheduledTriggerName);
+		const referenceName = createNumericReferenceName(importName, reservedNames);
+		const baseModuleName = toKebabCase(scheduledTriggerName) || toKebabCase(scheduledTriggerId) || scheduledTriggerId;
+		const moduleCount = moduleNameCounts.get(baseModuleName) ?? 0;
+		moduleNameCounts.set(baseModuleName, moduleCount + 1);
+		const modulePath = moduleCount === 0 ? baseModuleName : `${baseModuleName}-${moduleCount}`;
+		referenceNames.set(scheduledTriggerId, referenceName);
+		importRefs.set(scheduledTriggerId, {
+			importName,
+			modulePath
+		});
+	}
+	return {
+		referenceNames,
+		importRefs
+	};
+}
+function createTriggerReferenceMaps(triggers, reservedNames) {
+	const referenceNames = /* @__PURE__ */ new Map();
+	const importRefs = /* @__PURE__ */ new Map();
+	if (!triggers || !isPlainObject(triggers)) return {
+		referenceNames,
+		importRefs
+	};
+	const moduleNameCounts = /* @__PURE__ */ new Map();
+	for (const [triggerId, triggerData] of Object.entries(triggers)) {
+		if (referenceNames.has(triggerId)) continue;
+		const triggerRecord = isPlainObject(triggerData) ? triggerData : void 0;
+		const triggerName = typeof triggerRecord?.name === "string" && triggerRecord.name.length > 0 ? triggerRecord.name : triggerId;
+		const importName = toTriggerReferenceName(triggerName);
+		const referenceName = createNumericReferenceName(importName, reservedNames);
+		const baseModuleName = toKebabCase(triggerName) || toKebabCase(triggerId) || triggerId;
+		const moduleCount = moduleNameCounts.get(baseModuleName) ?? 0;
+		moduleNameCounts.set(baseModuleName, moduleCount + 1);
+		const modulePath = moduleCount === 0 ? baseModuleName : `${baseModuleName}-${moduleCount}`;
+		referenceNames.set(triggerId, referenceName);
+		importRefs.set(triggerId, {
+			importName,
+			modulePath
+		});
+	}
+	return {
+		referenceNames,
+		importRefs
+	};
+}
+function createNumericReferenceName(baseName, reservedNames) {
+	if (!reservedNames.has(baseName)) {
+		reservedNames.add(baseName);
+		return baseName;
+	}
+	let index = 1;
+	while (reservedNames.has(`${baseName}${index}`)) index += 1;
+	const uniqueName = `${baseName}${index}`;
+	reservedNames.add(uniqueName);
+	return uniqueName;
+}
+
+//#endregion
+export { addScheduledTriggerImports, addStatusComponentImports, addSubAgentImports, addTriggerImports, collectTemplateVariableNamesFromFields, createReferenceNameMap, createScheduledTriggerReferenceMaps, createSubAgentReferenceMaps, createTriggerReferenceMaps, extractContextConfigId, extractIds, extractStatusComponentIds, resolveStatusComponentId };
+//# sourceMappingURL=agent-generator.helpers.js.map

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

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

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

@@ -1,4 +1,6 @@
-import { addObjectEntries, addReferenceGetterProperty, addStringProperty, collectTemplateVariableNames, createFactoryDefinition, createUniqueReferenceName, formatStringLiteral, formatTemplate, isPlainObject, toCamelCase } from "../utils.js";
+import { addReferenceGetterProperty, addValueToObject, createFactoryDefinition, formatTemplate, resolveContextTemplateImports, toCamelCase, toTriggerReferenceName } from "../utils.js";
+import { addScheduledTriggerImports, addStatusComponentImports, addSubAgentImports, addTriggerImports, collectTemplateVariableNamesFromFields, createReferenceNameMap, createScheduledTriggerReferenceMaps, createSubAgentReferenceMaps, createTriggerReferenceMaps, extractContextConfigId, extractIds, extractStatusComponentIds, resolveStatusComponentId } from "./agent-generator.helpers.js";
+import { FullProjectDefinitionSchema } from "@inkeep/agents-core";
 import { z } from "zod";
 import { SyntaxKind } from "ts-morph";
 
@@ -7,40 +9,39 @@ const SubAgentReferenceSchema = z.object({
 	name: z.string().nonempty(),
 	local: z.boolean().optional()
 });
-const AgentSchema = z.looseObject({
+const MySchema = FullProjectDefinitionSchema.shape.agents.valueType.omit({ id: true });
+const SubAgentSchema = MySchema.shape.subAgents.valueType.omit({ type: true });
+const ToolSchema = MySchema.shape.tools.unwrap().valueType;
+const AgentSchema = z.strictObject({
 	agentId: z.string().nonempty(),
-	name: z.string().nonempty(),
-	description: z.string().nullish(),
-	prompt: z.string().optional(),
-	models: z.looseObject({}).optional(),
-	defaultSubAgentId: z.string().nonempty(),
-	subAgents: z.union([z.array(z.string()), z.record(z.string(), z.unknown())]),
-	contextConfig: z.union([z.string(), z.looseObject({ id: z.string().optional() })]).optional(),
-	stopWhen: z.object({ transferCountIs: z.int().optional() }).optional(),
-	statusUpdates: z.strictObject({
-		numEvents: z.int().optional(),
-		timeInSeconds: z.int().optional(),
-		statusComponents: z.array(z.union([z.string(), z.looseObject({
-			id: z.string().optional(),
-			type: z.string(),
-			name: z.string().optional()
-		})])).optional(),
-		prompt: z.string().optional()
-	}).optional(),
-	credentials: z.array(z.union([z.string(), z.strictObject({ id: z.string() })])).optional(),
-	triggers: z.union([z.array(z.string()), z.record(z.string(), z.unknown())]).optional(),
-	scheduledTriggers: z.union([z.array(z.string()), z.record(z.string(), z.unknown())]).optional(),
+	...MySchema.shape,
+	description: z.preprocess((v) => v || void 0, MySchema.shape.description),
+	models: z.preprocess((v) => v ?? void 0, MySchema.shape.models),
+	stopWhen: z.preprocess((v) => v && Object.keys(v).length && v || void 0, MySchema.shape.stopWhen),
+	subAgents: z.record(z.string(), z.strictObject({
+		...SubAgentSchema.shape,
+		models: z.preprocess((v) => v ?? void 0, SubAgentSchema.shape.models),
+		stopWhen: z.preprocess((v) => v ?? void 0, SubAgentSchema.shape.stopWhen),
+		skills: z.unknown(),
+		canDelegateTo: z.unknown()
+	})),
+	tools: z.record(z.string(), z.strictObject({
+		...ToolSchema.shape,
+		imageUrl: z.preprocess((v) => v ?? void 0, ToolSchema.shape.imageUrl)
+	})).optional(),
+	triggers: z.record(z.string(), z.unknown()).optional(),
 	agentVariableName: z.string().nonempty().optional(),
 	subAgentReferences: z.record(z.string(), SubAgentReferenceSchema).optional(),
+	subAgentReferencePathOverrides: z.record(z.string(), z.string().nonempty()).optional(),
 	contextConfigReference: SubAgentReferenceSchema.optional(),
 	contextConfigHeadersReference: SubAgentReferenceSchema.optional()
 });
-function generateAgentDefinition(data) {
+function generateAgentDefinition({ id, createdAt, updatedAt, ...data }) {
 	const result = AgentSchema.safeParse(data);
 	if (!result.success) throw new Error(`Validation failed for agent:\n${z.prettifyError(result.error)}`);
 	const parsed = result.data;
 	const subAgentIds = new Set(extractIds(parsed.subAgents));
-	subAgentIds.add(parsed.defaultSubAgentId);
+	if (parsed.defaultSubAgentId) subAgentIds.add(parsed.defaultSubAgentId);
 	const agentVarName = parsed.agentVariableName || toCamelCase(parsed.agentId);
 	const { sourceFile, configObject } = createFactoryDefinition({
 		importName: "agent",
@@ -48,34 +49,32 @@ function generateAgentDefinition(data) {
 	});
 	const reservedReferenceNames = new Set([agentVarName]);
 	const { referenceNames: subAgentReferenceNames, importNames: subAgentImportNames } = createSubAgentReferenceMaps(subAgentIds, reservedReferenceNames, "SubAgent", parsed.subAgentReferences);
-	addSubAgentImports(sourceFile, subAgentReferenceNames, subAgentImportNames);
+	addSubAgentImports(sourceFile, subAgentReferenceNames, subAgentImportNames, parsed.subAgentReferencePathOverrides);
 	const contextConfigId = extractContextConfigId(parsed.contextConfig);
 	let contextConfigReferenceName;
 	let contextHeadersReferenceName;
 	const hasHeadersTemplateVariables = collectTemplateVariableNamesFromFields([parsed.prompt, parsed.statusUpdates?.prompt]).some((variableName) => variableName.startsWith("headers."));
 	if (contextConfigId) {
-		const contextConfigImportName = parsed.contextConfigReference?.name ?? toCamelCase(contextConfigId);
-		contextConfigReferenceName = createUniqueReferenceName(contextConfigImportName, reservedReferenceNames, "ContextConfig");
-		const contextHeadersImportName = parsed.contextConfigHeadersReference?.name ?? `${toCamelCase(contextConfigId)}Headers`;
-		if (hasHeadersTemplateVariables) contextHeadersReferenceName = createUniqueReferenceName(contextHeadersImportName, reservedReferenceNames, "Headers");
-		const namedImports = [];
-		if (parsed.contextConfigReference?.local !== true) namedImports.push(contextConfigImportName === contextConfigReferenceName ? contextConfigImportName : {
-			name: contextConfigImportName,
-			alias: contextConfigReferenceName
-		});
-		if (hasHeadersTemplateVariables && contextHeadersReferenceName && parsed.contextConfigHeadersReference?.local !== true) namedImports.push(contextHeadersImportName === contextHeadersReferenceName ? contextHeadersImportName : {
-			name: contextHeadersImportName,
-			alias: contextHeadersReferenceName
+		const contextImportResolution = resolveContextTemplateImports({
+			reservedNames: reservedReferenceNames,
+			shouldResolveContextReference: true,
+			shouldResolveHeadersReference: hasHeadersTemplateVariables,
+			contextConfigReference: parsed.contextConfigReference,
+			contextConfigHeadersReference: parsed.contextConfigHeadersReference,
+			defaultContextImportName: toCamelCase(contextConfigId),
+			defaultHeadersImportName: `${toCamelCase(contextConfigId)}Headers`
 		});
-		if (namedImports.length > 0) sourceFile.addImportDeclaration({
-			namedImports,
+		contextConfigReferenceName = contextImportResolution.contextReferenceName;
+		contextHeadersReferenceName = contextImportResolution.headersReferenceName;
+		if (contextImportResolution.namedImports.length > 0) sourceFile.addImportDeclaration({
+			namedImports: contextImportResolution.namedImports,
 			moduleSpecifier: `../context-configs/${contextConfigId}`
 		});
 	}
-	const triggerReferenceNames = createReferenceNameMap(parsed.triggers ? extractIds(parsed.triggers) : [], reservedReferenceNames, "Trigger");
-	addTriggerImports(sourceFile, triggerReferenceNames);
-	const scheduledTriggerReferenceNames = createReferenceNameMap(parsed.scheduledTriggers ? extractIds(parsed.scheduledTriggers) : [], reservedReferenceNames, "ScheduledTrigger");
-	addScheduledTriggerImports(sourceFile, scheduledTriggerReferenceNames);
+	const { referenceNames: triggerReferenceNames, importRefs: triggerImportRefs } = createTriggerReferenceMaps(parsed.triggers, reservedReferenceNames);
+	addTriggerImports(sourceFile, triggerReferenceNames, triggerImportRefs);
+	const { referenceNames: scheduledTriggerReferenceNames, importRefs: scheduledTriggerImportRefs } = createScheduledTriggerReferenceMaps(parsed.scheduledTriggers, reservedReferenceNames);
+	addScheduledTriggerImports(sourceFile, scheduledTriggerReferenceNames, scheduledTriggerImportRefs);
 	const statusComponentReferenceNames = createReferenceNameMap(extractStatusComponentIds(parsed.statusUpdates), reservedReferenceNames, "StatusComponent");
 	addStatusComponentImports(sourceFile, statusComponentReferenceNames);
 	writeAgentConfig(configObject, parsed, {
@@ -89,188 +88,50 @@ function generateAgentDefinition(data) {
 	return sourceFile;
 }
 function writeAgentConfig(configObject, data, referenceNames) {
-	addStringProperty(configObject, "id", data.agentId);
-	addStringProperty(configObject, "name", data.name);
-	if (data.description != null) addStringProperty(configObject, "description", data.description);
-	if (data.prompt !== void 0) {
-		const template = formatTemplate(data.prompt, {
+	for (const [key, value] of Object.entries({
+		id: data.agentId,
+		name: data.name,
+		description: data.description,
+		prompt: data.prompt && formatTemplate(data.prompt, {
 			contextReference: referenceNames.contextConfig,
 			headersReference: referenceNames.contextHeaders
-		});
-		configObject.addPropertyAssignment({
-			name: "prompt",
-			initializer: formatStringLiteral(template)
-		});
-	}
-	if (data.models && Object.keys(data.models).length > 0) addObjectEntries(configObject.addPropertyAssignment({
-		name: "models",
-		initializer: "{}"
-	}).getInitializerIfKindOrThrow(SyntaxKind.ObjectLiteralExpression), data.models);
-	configObject.addPropertyAssignment({
+		}),
+		models: data.models,
+		stopWhen: data.stopWhen
+	})) addValueToObject(configObject, key, value);
+	const { defaultSubAgentId } = data;
+	if (defaultSubAgentId) configObject.addPropertyAssignment({
 		name: "defaultSubAgent",
-		initializer: referenceNames.subAgents.get(data.defaultSubAgentId) ?? toCamelCase(data.defaultSubAgentId)
+		initializer: referenceNames.subAgents.get(defaultSubAgentId) ?? toCamelCase(defaultSubAgentId)
 	});
 	addReferenceGetterProperty(configObject, "subAgents", extractIds(data.subAgents).map((id) => referenceNames.subAgents.get(id) ?? toCamelCase(id)));
 	if (extractContextConfigId(data.contextConfig) && referenceNames.contextConfig) configObject.addPropertyAssignment({
 		name: "contextConfig",
 		initializer: referenceNames.contextConfig
 	});
-	if (data.credentials?.length) {
-		const credentialIds = data.credentials.map((credential) => {
-			if (typeof credential === "string") return credential;
-			return credential.id;
-		}).filter((id) => Boolean(id));
-		if (credentialIds.length > 0) addReferenceGetterProperty(configObject, "credentials", credentialIds.map((id) => toCamelCase(id)));
-	}
 	const triggerIds = data.triggers ? extractIds(data.triggers) : [];
-	if (triggerIds.length > 0) addReferenceGetterProperty(configObject, "triggers", triggerIds.map((id) => referenceNames.triggers.get(id) ?? toCamelCase(id)));
+	if (triggerIds.length) addReferenceGetterProperty(configObject, "triggers", triggerIds.map((id) => referenceNames.triggers.get(id) ?? toTriggerReferenceName(id)));
 	const scheduledTriggerIds = data.scheduledTriggers ? extractIds(data.scheduledTriggers) : [];
-	if (scheduledTriggerIds.length > 0) addReferenceGetterProperty(configObject, "scheduledTriggers", scheduledTriggerIds.map((id) => referenceNames.scheduledTriggers.get(id) ?? toCamelCase(id)));
-	if (data.stopWhen?.transferCountIs !== void 0) configObject.addPropertyAssignment({
-		name: "stopWhen",
-		initializer: "{}"
-	}).getInitializerIfKindOrThrow(SyntaxKind.ObjectLiteralExpression).addPropertyAssignment({
-		name: "transferCountIs",
-		initializer: String(data.stopWhen.transferCountIs)
-	});
+	if (scheduledTriggerIds.length) addReferenceGetterProperty(configObject, "scheduledTriggers", scheduledTriggerIds.map((id) => referenceNames.scheduledTriggers.get(id) ?? toTriggerReferenceName(id)));
 	if (data.statusUpdates) {
 		const statusUpdatesObject = configObject.addPropertyAssignment({
 			name: "statusUpdates",
 			initializer: "{}"
 		}).getInitializerIfKindOrThrow(SyntaxKind.ObjectLiteralExpression);
-		if (data.statusUpdates.numEvents !== void 0) statusUpdatesObject.addPropertyAssignment({
-			name: "numEvents",
-			initializer: String(data.statusUpdates.numEvents)
-		});
-		if (data.statusUpdates.timeInSeconds !== void 0) statusUpdatesObject.addPropertyAssignment({
-			name: "timeInSeconds",
-			initializer: String(data.statusUpdates.timeInSeconds)
-		});
-		if (data.statusUpdates.statusComponents && data.statusUpdates.statusComponents.length > 0) {
-			const statusComponentRefs = data.statusUpdates.statusComponents.map((statusComponent) => `${referenceNames.statusComponents.get(resolveStatusComponentId(statusComponent)) ?? toCamelCase(resolveStatusComponentId(statusComponent))}.config`);
-			if (statusComponentRefs.length > 0) statusUpdatesObject.addPropertyAssignment({
-				name: "statusComponents",
-				initializer: "[]"
-			}).getInitializerIfKindOrThrow(SyntaxKind.ArrayLiteralExpression).addElements(statusComponentRefs);
-		}
-		if (data.statusUpdates.prompt !== void 0) {
-			const template = formatTemplate(data.statusUpdates.prompt, {
-				contextReference: referenceNames.contextConfig,
-				headersReference: referenceNames.contextHeaders
-			});
-			statusUpdatesObject.addPropertyAssignment({
-				name: "prompt",
-				initializer: formatStringLiteral(template)
-			});
-		}
-	}
-}
-function collectTemplateVariableNamesFromFields(values) {
-	const variables = [];
-	for (const value of values) {
-		if (typeof value !== "string") continue;
-		variables.push(...collectTemplateVariableNames(value));
-	}
-	return variables;
-}
-function extractIds(value) {
-	if (Array.isArray(value)) return value.map((item) => {
-		if (typeof item === "string") return item;
-		if (isPlainObject(item) && typeof item.id === "string") return item.id;
-		return null;
-	}).filter((id) => !!id);
-	return Object.keys(value);
-}
-function extractContextConfigId(contextConfig) {
-	if (!contextConfig) return;
-	if (typeof contextConfig === "string") return contextConfig;
-	return contextConfig.id;
-}
-function addSubAgentImports(sourceFile, referenceNames, importNames) {
-	for (const [subAgentId, referenceName] of referenceNames) {
-		const importName = importNames.get(subAgentId);
-		if (!importName) continue;
-		sourceFile.addImportDeclaration({
-			namedImports: [importName === referenceName ? importName : {
-				name: importName,
-				alias: referenceName
-			}],
-			moduleSpecifier: `./sub-agents/${subAgentId}`
-		});
-	}
-}
-function addTriggerImports(sourceFile, referenceNames) {
-	for (const [triggerId, referenceName] of referenceNames) {
-		const importName = toCamelCase(triggerId);
-		sourceFile.addImportDeclaration({
-			namedImports: [importName === referenceName ? importName : {
-				name: importName,
-				alias: referenceName
-			}],
-			moduleSpecifier: `./triggers/${triggerId}`
-		});
-	}
-}
-function addScheduledTriggerImports(sourceFile, referenceNames) {
-	for (const [scheduledTriggerId, referenceName] of referenceNames) {
-		const importName = toCamelCase(scheduledTriggerId);
-		sourceFile.addImportDeclaration({
-			namedImports: [importName === referenceName ? importName : {
-				name: importName,
-				alias: referenceName
-			}],
-			moduleSpecifier: `./scheduled-triggers/${scheduledTriggerId}`
-		});
-	}
-}
-function extractStatusComponentIds(statusUpdates) {
-	if (!statusUpdates?.statusComponents?.length) return [];
-	const statusComponentIds = statusUpdates.statusComponents.map(resolveStatusComponentId);
-	return [...new Set(statusComponentIds)];
-}
-function resolveStatusComponentId(statusComponent) {
-	const id = typeof statusComponent === "string" ? statusComponent : statusComponent.id || statusComponent.type;
-	if (!id) throw new Error(`Unable to resolve status component with id ${JSON.stringify(statusComponent)}`);
-	return id;
-}
-function addStatusComponentImports(sourceFile, referenceNames) {
-	for (const [statusComponentId, referenceName] of referenceNames) {
-		const importName = toCamelCase(statusComponentId);
-		sourceFile.addImportDeclaration({
-			namedImports: [importName === referenceName ? importName : {
-				name: importName,
-				alias: referenceName
-			}],
-			moduleSpecifier: `../status-components/${statusComponentId}`
-		});
-	}
-}
-function createSubAgentReferenceMaps(ids, reservedNames, conflictSuffix, overrides) {
-	const referenceNames = /* @__PURE__ */ new Map();
-	const importNames = /* @__PURE__ */ new Map();
-	for (const id of ids) {
-		if (referenceNames.has(id)) continue;
-		const override = overrides?.[id];
-		const importName = override?.name ?? toCamelCase(id);
-		const isLocal = override?.local === true;
-		const referenceName = isLocal ? importName : createUniqueReferenceName(importName, reservedNames, conflictSuffix);
-		if (isLocal) reservedNames.add(referenceName);
-		else importNames.set(id, importName);
-		referenceNames.set(id, referenceName);
-	}
-	return {
-		referenceNames,
-		importNames
-	};
-}
-function createReferenceNameMap(ids, reservedNames, conflictSuffix) {
-	const map = /* @__PURE__ */ new Map();
-	for (const id of ids) {
-		if (map.has(id)) continue;
-		map.set(id, createUniqueReferenceName(toCamelCase(id), reservedNames, conflictSuffix));
+		addValueToObject(statusUpdatesObject, "numEvents", data.statusUpdates.numEvents);
+		addValueToObject(statusUpdatesObject, "timeInSeconds", data.statusUpdates.timeInSeconds);
+		addValueToObject(statusUpdatesObject, "prompt", data.statusUpdates.prompt && formatTemplate(data.statusUpdates.prompt, {
+			contextReference: referenceNames.contextConfig,
+			headersReference: referenceNames.contextHeaders
+		}));
+		const statusComponentRefs = data.statusUpdates.statusComponents?.map((statusComponent) => `${referenceNames.statusComponents.get(resolveStatusComponentId(statusComponent)) ?? toCamelCase(resolveStatusComponentId(statusComponent))}.config`);
+		if (statusComponentRefs?.length) statusUpdatesObject.addPropertyAssignment({
+			name: "statusComponents",
+			initializer: "[]"
+		}).getInitializerIfKindOrThrow(SyntaxKind.ArrayLiteralExpression).addElements(statusComponentRefs);
 	}
-	return map;
 }
 
 //#endregion
-export { generateAgentDefinition };
+export { generateAgentDefinition };
+//# sourceMappingURL=agent-generator.js.map