fix: add memoryProviders to template render config

Author: notgitika

src/cli/operations/agent/generate/schema-mapper.ts

@@ -8,7 +8,10 @@ import type {
   MemoryStrategy,
   ModelProvider,
 } from '../../../../schema';
-import type { AgentRenderConfig } from '../../../templates/types';
+import type {
+  AgentRenderConfig,
+  MemoryProviderRenderConfig,
+} from '../../../templates/types';
 import {
   DEFAULT_MEMORY_EXPIRY_DAYS,
   DEFAULT_NETWORK_MODE,
@@ -112,6 +115,33 @@ export function mapGenerateConfigToResources(config: GenerateConfig): GenerateCo
   };
 }
 
+/**
+ * Compute the default env var name for a memory.
+ */
+function computeMemoryEnvVarName(memoryName: string): string {
+  return `AGENTCORE_MEMORY_${memoryName.toUpperCase()}`;
+}
+
+/**
+ * Maps memory option to memory providers for template rendering.
+ */
+function mapMemoryOptionToMemoryProviders(
+  memory: MemoryOption,
+  projectName: string
+): MemoryProviderRenderConfig[] {
+  if (memory === 'none') {
+    return [];
+  }
+
+  const memoryName = `${projectName}Memory`;
+  return [
+    {
+      name: memoryName,
+      envVarName: computeMemoryEnvVarName(memoryName),
+    },
+  ];
+}
+
 /**
  * Maps GenerateConfig to AgentRenderConfig for template rendering.
  */
@@ -123,5 +153,6 @@ export function mapGenerateConfigToRenderConfig(config: GenerateConfig): AgentRe
     modelProvider: config.modelProvider,
     hasMemory: config.memory !== 'none',
     hasIdentity: config.modelProvider !== 'Bedrock',
+    memoryProviders: mapMemoryOptionToMemoryProviders(config.memory, config.projectName),
   };
 }

src/cli/templates/types.ts

@@ -1,5 +1,13 @@
 import type { ModelProvider, SDKFramework, TargetLanguage } from '../../schema';
 
+/**
+ * Memory provider config for template rendering.
+ */
+export interface MemoryProviderRenderConfig {
+  name: string;
+  envVarName: string;
+}
+
 /**
  * Configuration needed by template renderers.
  * This is separate from the v2 Agent schema which only stores runtime config.
@@ -11,4 +19,5 @@ export interface AgentRenderConfig {
   modelProvider: ModelProvider;
   hasMemory: boolean;
   hasIdentity: boolean;
+  memoryProviders: MemoryProviderRenderConfig[];
 }