@aws/ml-container-creator 0.2.1 → 0.2.3

package/src/lib/prompt-runner.js

@@ -60,9 +60,9 @@ export default class PromptRunner {
         const buildTimestamp = new Date().toISOString().replace(/[:.]/g, '-').slice(0, 19);
 
         // Load catalog data via Registry_Loader
-        const registryLoader = new RegistryLoader()
-        this._tritonBackends = await registryLoader.loadTritonBackends()
-        this._instanceAcceleratorMapping = await registryLoader.loadInstanceAcceleratorMapping()
+        const registryLoader = new RegistryLoader();
+        this._tritonBackends = await registryLoader.loadTritonBackends();
+        this._instanceAcceleratorMapping = await registryLoader.loadInstanceAcceleratorMapping();
 
         // Get existing configuration to use as defaults
         const existingConfig = this.baseConfig || {};
@@ -77,8 +77,8 @@ export default class PromptRunner {
 
         // 1a. Query region MCP, then prompt for region + deployment target
         await this._queryMcpForRegion({}, explicitConfig);
-        const bootstrapRegion = existingConfig.awsRegion || explicitConfig.awsRegion
-        const regionPreviousAnswers = bootstrapRegion ? { _bootstrapRegion: bootstrapRegion } : {}
+        const bootstrapRegion = existingConfig.awsRegion || explicitConfig.awsRegion;
+        const regionPreviousAnswers = bootstrapRegion ? { _bootstrapRegion: bootstrapRegion } : {};
         const regionAndTargetAnswers = await this._runPhase(infraRegionAndTargetPrompts, regionPreviousAnswers, explicitConfig, existingConfig);
 
         // 1b. Instance type — query MCP and prompt for managed-inference, async-inference, batch-transform, and hyperpod-eks
@@ -171,18 +171,18 @@ export default class PromptRunner {
         
         // Query base-image-picker MCP server for base image choices
         // Requirements: 5.1, 5.2, 5.3
-        await this._queryMcpForBaseImage(frameworkAnswers, explicitConfig)
+        await this._queryMcpForBaseImage(frameworkAnswers, explicitConfig);
         const baseImagePreviousAnswers = {
             ...frameworkAnswers,
             ...engineAnswers,
             ...(this._mcpBaseImageChoices ? { _mcpBaseImageChoices: this._mcpBaseImageChoices } : {})
-        }
+        };
         const baseImageAnswers = await this._runPhase(
             baseImagePrompts,
             baseImagePreviousAnswers,
             explicitConfig,
             existingConfig
-        )
+        );
 
         // Populate framework version choices from registry
         const frameworkVersionChoices = this._getFrameworkVersionChoices(frameworkAnswers.framework);
@@ -211,10 +211,10 @@ export default class PromptRunner {
         );
         
         // Query model-picker MCP server for model choices
-        this._queryMcpForModels(frameworkAnswers.architecture)
+        this._queryMcpForModels(frameworkAnswers.architecture);
         if (this._mcpModelChoices) {
-            console.log(`   🔍 Querying model-picker...`)
-            console.log(`   ✓ ${this._mcpModelChoices.length} model(s) available from catalog`)
+            console.log('   🔍 Querying model-picker...');
+            console.log(`   ✓ ${this._mcpModelChoices.length} model(s) available from catalog`);
         }
         const modelFormatPreviousAnswers = {
             ...frameworkAnswers,
@@ -222,7 +222,7 @@ export default class PromptRunner {
             ...frameworkVersionAnswers,
             ...frameworkProfileAnswers,
             ...(this._mcpModelChoices ? { _mcpModelChoices: this._mcpModelChoices } : {})
-        }
+        };
         const modelFormatAnswers = await this._runPhase(
             modelFormatPrompts, 
             modelFormatPreviousAnswers, 
@@ -392,7 +392,7 @@ export default class PromptRunner {
         // Registry models — note about InferenceSpecification requirement
         if (combinedAnswers.modelSource === 'registry') {
             if (!combinedAnswers.artifactUri) {
-                console.log(`\n   ⚠️  Model source is 'registry' but no artifact URI was resolved.`);
+                console.log('\n   ⚠️  Model source is \'registry\' but no artifact URI was resolved.');
                 console.log('   The model package must have an InferenceSpecification with a valid');
                 console.log('   ModelDataUrl or S3DataSource for the runtime resolver to work.');
                 console.log('   If your model package was registered without an InferenceSpecification,');
@@ -421,7 +421,7 @@ export default class PromptRunner {
         // Apply auto-set model format for Triton backends with single format
         // Requirements: 3.3, 3.4, 3.5
         if (tritonAutoFormat) {
-            combinedAnswers.modelFormat = tritonAutoFormat
+            combinedAnswers.modelFormat = tritonAutoFormat;
         }
 
         // Handle custom model name for transformers, diffusors, and Triton LLM backends
@@ -464,14 +464,14 @@ export default class PromptRunner {
 
         // Handle custom base image
         if (combinedAnswers.customBaseImage) {
-            combinedAnswers.baseImage = combinedAnswers.customBaseImage
-            combinedAnswers._baseImageSource = 'custom'
-            delete combinedAnswers.customBaseImage
+            combinedAnswers.baseImage = combinedAnswers.customBaseImage;
+            combinedAnswers._baseImageSource = 'custom';
+            delete combinedAnswers.customBaseImage;
         }
 
         // Handle --base-image CLI override
         if (this.options['base-image']) {
-            combinedAnswers.baseImage = this.options['base-image']
+            combinedAnswers.baseImage = this.options['base-image'];
         }
 
         // Map awsRoleArn to roleArn for templates
@@ -576,17 +576,17 @@ export default class PromptRunner {
      * @private
      */
     _getTritonAutoModelFormat(architecture, backend) {
-        if (architecture !== 'triton') return null
+        if (architecture !== 'triton') return null;
 
-        const meta = this._tritonBackends[backend]
-        if (!meta || !meta.modelFormats) return null
+        const meta = this._tritonBackends[backend];
+        if (!meta || !meta.modelFormats) return null;
 
         // Only auto-set if there's exactly one format
         if (meta.modelFormats.length === 1) {
-            return meta.modelFormats[0]
+            return meta.modelFormats[0];
         }
 
-        return null
+        return null;
     }
 
     /**
@@ -725,52 +725,52 @@ export default class PromptRunner {
      * Requirements: 5.1, 5.2, 5.3, 5.4, 9.1, 9.2, 9.3
      * @private
      */
-    async _queryMcpForBaseImage(frameworkAnswers, explicitConfig) {
+    async _queryMcpForBaseImage(frameworkAnswers, _explicitConfig) {
         // Skip if base image provided via CLI --base-image flag
-        if (this.options['base-image']) return
+        if (this.options['base-image']) return;
 
-        const cm = this.configManager
-        if (!cm) return
+        const cm = this.configManager;
+        if (!cm) return;
 
-        const mcpServers = cm.getMcpServerNames()
-        if (!mcpServers.includes('base-image-picker')) return
+        const mcpServers = cm.getMcpServerNames();
+        if (!mcpServers.includes('base-image-picker')) return;
 
-        const smart = this.options.smart === true
-        const discover = this.options.discover === true
-        const framework = frameworkAnswers.framework
-        const modelServer = frameworkAnswers.modelServer
-        const architecture = frameworkAnswers.architecture || frameworkAnswers.deploymentConfig?.split('-')[0]
-        const isTransformer = framework === 'transformers'
-        const isTriton = architecture === 'triton'
-        const isDiffusors = architecture === 'diffusors'
+        const smart = this.options.smart === true;
+        const discover = this.options.discover === true;
+        const framework = frameworkAnswers.framework;
+        const modelServer = frameworkAnswers.modelServer;
+        const architecture = frameworkAnswers.architecture || frameworkAnswers.deploymentConfig?.split('-')[0];
+        const isTransformer = framework === 'transformers';
+        const isTriton = architecture === 'triton';
+        const isDiffusors = architecture === 'diffusors';
 
         // For non-transformer, non-triton, non-diffusors frameworks, prompt for optional search criteria
-        let searchCriteria
+        let searchCriteria;
         if (!isTransformer && !isTriton && !isDiffusors) {
             const searchAnswer = await this._runPrompts(baseImageSearchPrompts.map(p => ({
                 ...p,
                 when: () => true // Always show for non-transformer since we already checked
-            })))
-            searchCriteria = searchAnswer.baseImageSearch
+            })));
+            searchCriteria = searchAnswer.baseImageSearch;
         }
 
-        const modeLabel = [smart && '[smart]', discover && '[discover]'].filter(Boolean).join(' ')
-        console.log(`   🔍 Querying base-image-picker${modeLabel ? ` ${modeLabel}` : ''}...`)
+        const modeLabel = [smart && '[smart]', discover && '[discover]'].filter(Boolean).join(' ');
+        console.log(`   🔍 Querying base-image-picker${modeLabel ? ` ${modeLabel}` : ''}...`);
 
-        const context = { framework, modelServer, architecture }
+        const context = { framework, modelServer, architecture };
         if (searchCriteria && searchCriteria.trim()) {
-            context.searchCriteria = searchCriteria.trim()
+            context.searchCriteria = searchCriteria.trim();
         }
 
-        const result = await cm.queryMcpServer('base-image-picker', context)
+        const result = await cm.queryMcpServer('base-image-picker', context);
 
         if (result && result.metadata?.baseImage?.length > 0) {
-            const entries = result.metadata.baseImage
-            this._mcpBaseImageChoices = formatImageChoices(entries, isTransformer || isTriton || isDiffusors)
-            const count = entries.length
-            console.log(`   ✓ ${count} base image(s) available`)
+            const entries = result.metadata.baseImage;
+            this._mcpBaseImageChoices = formatImageChoices(entries, isTransformer || isTriton || isDiffusors);
+            const count = entries.length;
+            console.log(`   ✓ ${count} base image(s) available`);
         } else {
-            console.log('   ↳ No MCP results, using default image')
+            console.log('   ↳ No MCP results, using default image');
         }
     }
 
@@ -782,47 +782,47 @@ export default class PromptRunner {
      * @private
      */
     _queryMcpForModels(architecture) {
-        const cm = this.configManager
-        if (!cm) return
+        const cm = this.configManager;
+        if (!cm) return;
 
-        const mcpServers = cm.getMcpServerNames()
-        if (!mcpServers.includes('model-picker')) return
+        const mcpServers = cm.getMcpServerNames();
+        if (!mcpServers.includes('model-picker')) return;
 
         try {
-            const mcpConfigPath = path.join(GENERATOR_ROOT, 'config', 'mcp.json')
-            if (!fs.existsSync(mcpConfigPath)) return
+            const mcpConfigPath = path.join(GENERATOR_ROOT, 'config', 'mcp.json');
+            if (!fs.existsSync(mcpConfigPath)) return;
 
-            const mcpConfig = JSON.parse(fs.readFileSync(mcpConfigPath, 'utf8'))
-            const serverConfig = mcpConfig.mcpServers?.['model-picker']
-            if (!serverConfig?.args?.length) return
+            const mcpConfig = JSON.parse(fs.readFileSync(mcpConfigPath, 'utf8'));
+            const serverConfig = mcpConfig.mcpServers?.['model-picker'];
+            if (!serverConfig?.args?.length) return;
 
             // Resolve the server entry point directory from the args
-            const serverEntryPoint = serverConfig.args[serverConfig.args.length - 1]
-            const serverDir = path.dirname(serverEntryPoint)
+            const serverEntryPoint = serverConfig.args[serverConfig.args.length - 1];
+            const serverDir = path.dirname(serverEntryPoint);
 
             // Read manifest to find catalog path
-            const manifestPath = path.join(serverDir, 'manifest.json')
-            if (!fs.existsSync(manifestPath)) return
+            const manifestPath = path.join(serverDir, 'manifest.json');
+            if (!fs.existsSync(manifestPath)) return;
 
-            const manifest = JSON.parse(fs.readFileSync(manifestPath, 'utf8'))
+            const manifest = JSON.parse(fs.readFileSync(manifestPath, 'utf8'));
 
             // Select catalog based on architecture
             const catalogKey = architecture === 'diffusors'
                 ? 'popular-diffusors'
-                : 'popular-transformers'
-            const catalogRelPath = manifest.catalogs?.[catalogKey]
-            if (!catalogRelPath) return
+                : 'popular-transformers';
+            const catalogRelPath = manifest.catalogs?.[catalogKey];
+            if (!catalogRelPath) return;
 
-            const catalogPath = path.resolve(serverDir, catalogRelPath)
-            if (!fs.existsSync(catalogPath)) return
+            const catalogPath = path.resolve(serverDir, catalogRelPath);
+            if (!fs.existsSync(catalogPath)) return;
 
-            const catalog = JSON.parse(fs.readFileSync(catalogPath, 'utf8'))
+            const catalog = JSON.parse(fs.readFileSync(catalogPath, 'utf8'));
 
             // Extract model IDs, filtering out glob patterns (entries with *)
-            const modelIds = Object.keys(catalog).filter(id => !id.includes('*'))
+            const modelIds = Object.keys(catalog).filter(id => !id.includes('*'));
 
             if (modelIds.length > 0) {
-                this._mcpModelChoices = modelIds
+                this._mcpModelChoices = modelIds;
             }
         } catch {
             // Silently fall back to hardcoded defaults
@@ -999,186 +999,186 @@ export default class PromptRunner {
      * @private
      */
     async _fetchAndDisplayModelInfo(modelId) {
-            console.log(`\n   🔍 Querying model-picker [discover]...`);
-
-            const sources = [];
-            let chatTemplate = null;
-            let modelFamily = null;
-            let mcpUsed = false;
-
-            // Try model-picker MCP server in discover mode (queries HuggingFace + merges with catalog)
-            const cm = this.configManager;
-            if (cm) {
-                const mcpServers = cm.getMcpServerNames();
-                if (mcpServers.includes('model-picker')) {
-                    try {
-                        const mcpConfigPath = path.join(GENERATOR_ROOT, 'config', 'mcp.json');
-                        if (fs.existsSync(mcpConfigPath)) {
-                            const mcpConfig = JSON.parse(fs.readFileSync(mcpConfigPath, 'utf8'));
-                            const serverConfig = mcpConfig.mcpServers?.['model-picker'];
-                            if (serverConfig) {
-                                const { McpClient } = await import('./mcp-client.js');
-                                const client = new McpClient(serverConfig, { timeout: 15000 });
-
-                                // Override _buildContext to pass model_id and mode directly
-                                client._getUnboundedParameterNames = () => [];
-                                client._buildContext = () => ({});
-
-                                // Connect and call get_models directly
-                                const { Client } = await import('@modelcontextprotocol/sdk/client/index.js');
-                                const { StdioClientTransport } = await import('@modelcontextprotocol/sdk/client/stdio.js');
-
-                                const transport = new StdioClientTransport({
-                                    command: serverConfig.command,
-                                    args: serverConfig.args || [],
-                                    env: { ...process.env, ...(serverConfig.env || {}) },
-                                    stderr: 'pipe'
-                                });
-
-                                const mcpClient = new Client(
-                                    { name: 'ml-container-creator', version: '1.0.0' },
-                                    { capabilities: {} }
-                                );
-
-                                await mcpClient.connect(transport);
-
-                                const result = await mcpClient.callTool({
-                                    name: 'get_models',
-                                    arguments: { model_id: modelId, mode: 'discover' }
-                                });
-
-                                await mcpClient.close();
-
-                                // Parse the response
-                                const textBlock = result?.content?.find(b => b.type === 'text');
-                                if (textBlock) {
-                                    const parsed = JSON.parse(textBlock.text);
-                                    if (parsed.values && Object.keys(parsed.values).length > 0) {
-                                        mcpUsed = true;
-                                        const vals = parsed.values;
-
-                                        if (vals.chat_template) {
-                                            chatTemplate = vals.chat_template;
-                                        }
-                                        if (vals.family) {
-                                            modelFamily = vals.family;
-                                        }
-
-                                        // Extract model source metadata for loading adapter
-                                        // Requirements: 2.1, 2.2, 2.3, 2.4
-                                        if (vals.provider) {
-                                            this._mcpModelSource = vals.provider;
-                                        }
-                                        if (vals.artifactUri) {
-                                            this._mcpArtifactUri = vals.artifactUri;
-                                        }
-
-                                        // Determine sources based on what was returned
-                                        if (vals.tags || vals.pipeline_tag) {
-                                            sources.push('HuggingFace_Hub_API');
-                                        }
-                                        if (vals.validation_level || vals.framework_compatibility) {
-                                            sources.push('Model_Picker_Catalog');
-                                        }
-                                        if (sources.length === 0) {
-                                            sources.push('model-picker');
-                                        }
-                                        console.log(`   ✓ Resolved: ${modelId}`);
-                                    } else if (parsed.message) {
-                                        console.log(`   ↳ ${parsed.message}`);
+        console.log('\n   🔍 Querying model-picker [discover]...');
+
+        const sources = [];
+        let chatTemplate = null;
+        let modelFamily = null;
+        let mcpUsed = false;
+
+        // Try model-picker MCP server in discover mode (queries HuggingFace + merges with catalog)
+        const cm = this.configManager;
+        if (cm) {
+            const mcpServers = cm.getMcpServerNames();
+            if (mcpServers.includes('model-picker')) {
+                try {
+                    const mcpConfigPath = path.join(GENERATOR_ROOT, 'config', 'mcp.json');
+                    if (fs.existsSync(mcpConfigPath)) {
+                        const mcpConfig = JSON.parse(fs.readFileSync(mcpConfigPath, 'utf8'));
+                        const serverConfig = mcpConfig.mcpServers?.['model-picker'];
+                        if (serverConfig) {
+                            const { McpClient } = await import('./mcp-client.js');
+                            const client = new McpClient(serverConfig, { timeout: 15000 });
+
+                            // Override _buildContext to pass model_id and mode directly
+                            client._getUnboundedParameterNames = () => [];
+                            client._buildContext = () => ({});
+
+                            // Connect and call get_models directly
+                            const { Client } = await import('@modelcontextprotocol/sdk/client/index.js');
+                            const { StdioClientTransport } = await import('@modelcontextprotocol/sdk/client/stdio.js');
+
+                            const transport = new StdioClientTransport({
+                                command: serverConfig.command,
+                                args: serverConfig.args || [],
+                                env: { ...process.env, ...(serverConfig.env || {}) },
+                                stderr: 'pipe'
+                            });
+
+                            const mcpClient = new Client(
+                                { name: 'ml-container-creator', version: '1.0.0' },
+                                { capabilities: {} }
+                            );
+
+                            await mcpClient.connect(transport);
+
+                            const result = await mcpClient.callTool({
+                                name: 'get_models',
+                                arguments: { model_id: modelId, mode: 'discover' }
+                            });
+
+                            await mcpClient.close();
+
+                            // Parse the response
+                            const textBlock = result?.content?.find(b => b.type === 'text');
+                            if (textBlock) {
+                                const parsed = JSON.parse(textBlock.text);
+                                if (parsed.values && Object.keys(parsed.values).length > 0) {
+                                    mcpUsed = true;
+                                    const vals = parsed.values;
+
+                                    if (vals.chat_template) {
+                                        chatTemplate = vals.chat_template;
+                                    }
+                                    if (vals.family) {
+                                        modelFamily = vals.family;
+                                    }
+
+                                    // Extract model source metadata for loading adapter
+                                    // Requirements: 2.1, 2.2, 2.3, 2.4
+                                    if (vals.provider) {
+                                        this._mcpModelSource = vals.provider;
+                                    }
+                                    if (vals.artifactUri) {
+                                        this._mcpArtifactUri = vals.artifactUri;
+                                    }
+
+                                    // Determine sources based on what was returned
+                                    if (vals.tags || vals.pipeline_tag) {
+                                        sources.push('HuggingFace_Hub_API');
+                                    }
+                                    if (vals.validation_level || vals.framework_compatibility) {
+                                        sources.push('Model_Picker_Catalog');
                                     }
+                                    if (sources.length === 0) {
+                                        sources.push('model-picker');
+                                    }
+                                    console.log(`   ✓ Resolved: ${modelId}`);
+                                } else if (parsed.message) {
+                                    console.log(`   ↳ ${parsed.message}`);
                                 }
                             }
                         }
-                    } catch (err) {
-                        console.log('   ↳ model-picker unavailable, using fallback');
                     }
+                } catch (err) {
+                    console.log('   ↳ model-picker unavailable, using fallback');
                 }
             }
+        }
 
-            // Fallback to legacy path if MCP didn't resolve
-            if (!mcpUsed) {
-                const registryConfigManager = this.registryConfigManager;
-                if (registryConfigManager) {
-                    // Only try HuggingFace API for bare model IDs (not prefixed URIs)
-                    const isNonHfUri = modelId.startsWith('jumpstart://') ||
+        // Fallback to legacy path if MCP didn't resolve
+        if (!mcpUsed) {
+            const registryConfigManager = this.registryConfigManager;
+            if (registryConfigManager) {
+                // Only try HuggingFace API for bare model IDs (not prefixed URIs)
+                const isNonHfUri = modelId.startsWith('jumpstart://') ||
                         modelId.startsWith('jumpstart-hub://') ||
                         modelId.startsWith('s3://') ||
                         modelId.startsWith('registry://');
 
-                    if (!isNonHfUri) {
-                        // Try HuggingFace API directly
-                        try {
-                            const hfData = await registryConfigManager._fetchHuggingFaceData(modelId);
-                            if (hfData) {
-                                sources.push('HuggingFace_Hub_API');
-                                if (hfData.chatTemplate) {
-                                    chatTemplate = hfData.chatTemplate;
-                                }
-                                console.log('   ✅ Found on HuggingFace Hub');
-                            } else {
-                                console.log('   ℹ️  Not found on HuggingFace Hub (may be private or offline)');
+                if (!isNonHfUri) {
+                    // Try HuggingFace API directly
+                    try {
+                        const hfData = await registryConfigManager._fetchHuggingFaceData(modelId);
+                        if (hfData) {
+                            sources.push('HuggingFace_Hub_API');
+                            if (hfData.chatTemplate) {
+                                chatTemplate = hfData.chatTemplate;
                             }
-                        } catch (error) {
-                            console.log('   ⚠️  HuggingFace API unavailable');
+                            console.log('   ✅ Found on HuggingFace Hub');
+                        } else {
+                            console.log('   ℹ️  Not found on HuggingFace Hub (may be private or offline)');
                         }
-                    } else {
-                        // Non-HF URI (jumpstart://, s3://, etc.) — skip HF lookup silently
-                        // The summary at the end of this function will report "No additional model information"
+                    } catch (error) {
+                        console.log('   ⚠️  HuggingFace API unavailable');
                     }
+                } else {
+                    // Non-HF URI (jumpstart://, s3://, etc.) — skip HF lookup silently
+                    // The summary at the end of this function will report "No additional model information"
+                }
 
-                    // Check Model Registry for overrides
-                    if (registryConfigManager.modelRegistry) {
-                        let modelConfig = registryConfigManager.modelRegistry[modelId];
-
-                        if (!modelConfig) {
-                            for (const [pattern, config] of Object.entries(registryConfigManager.modelRegistry)) {
-                                if (pattern.includes('*')) {
-                                    const regex = new RegExp('^' + pattern.replace(/\*/g, '.*') + '$');
-                                    if (regex.test(modelId)) {
-                                        modelConfig = config;
-                                        console.log(`   ✅ Matched pattern in Model_Registry: ${pattern}`);
-                                        break;
-                                    }
+                // Check Model Registry for overrides
+                if (registryConfigManager.modelRegistry) {
+                    let modelConfig = registryConfigManager.modelRegistry[modelId];
+
+                    if (!modelConfig) {
+                        for (const [pattern, config] of Object.entries(registryConfigManager.modelRegistry)) {
+                            if (pattern.includes('*')) {
+                                const regex = new RegExp(`^${  pattern.replace(/\*/g, '.*')  }$`);
+                                if (regex.test(modelId)) {
+                                    modelConfig = config;
+                                    console.log(`   ✅ Matched pattern in Model_Registry: ${pattern}`);
+                                    break;
                                 }
                             }
-                        } else {
-                            console.log('   ✅ Found in Model_Registry');
                         }
+                    } else {
+                        console.log('   ✅ Found in Model_Registry');
+                    }
 
-                        if (modelConfig) {
-                            sources.push('Model_Registry');
-                            if (modelConfig.chatTemplate) {
-                                chatTemplate = modelConfig.chatTemplate;
-                            }
-                            if (modelConfig.family) {
-                                modelFamily = modelConfig.family;
-                            }
+                    if (modelConfig) {
+                        sources.push('Model_Registry');
+                        if (modelConfig.chatTemplate) {
+                            chatTemplate = modelConfig.chatTemplate;
+                        }
+                        if (modelConfig.family) {
+                            modelFamily = modelConfig.family;
                         }
                     }
                 }
             }
+        }
 
-            // Display information
-            if (sources.length > 0) {
-                console.log('\n📋 Model Information:');
-                console.log(`   • Model ID: ${modelId}`);
-                if (modelFamily) {
-                    console.log(`   • Family: ${modelFamily}`);
-                }
-                if (chatTemplate) {
-                    console.log('   • Chat Template: ✅ Available');
-                    console.log('     (Will be injected into generated files)');
-                } else {
-                    console.log('   • Chat Template: ❌ Not available');
-                    console.log('     (Chat endpoints may require manual configuration)');
-                }
-                console.log(`   • Sources: ${sources.join(', ')}`);
+        // Display information
+        if (sources.length > 0) {
+            console.log('\n📋 Model Information:');
+            console.log(`   • Model ID: ${modelId}`);
+            if (modelFamily) {
+                console.log(`   • Family: ${modelFamily}`);
+            }
+            if (chatTemplate) {
+                console.log('   • Chat Template: ✅ Available');
+                console.log('     (Will be injected into generated files)');
             } else {
-                console.log('   ℹ️  No additional model information available');
-                console.log('   Proceeding with default configuration');
+                console.log('   • Chat Template: ❌ Not available');
+                console.log('     (Chat endpoints may require manual configuration)');
             }
+            console.log(`   • Sources: ${sources.join(', ')}`);
+        } else {
+            console.log('   ℹ️  No additional model information available');
+            console.log('   Proceeding with default configuration');
         }
+    }