mcp-prompt-optimizer 1.5.0 → 2.2.3

package/index.js

@@ -5,7 +5,7 @@
  * Production-grade with Bayesian optimization, AG-UI real-time features, enhanced network resilience, 
  * development mode, and complete backend alignment
  * 
- * Version: 2.1.0 - Aligned with FastAPI Backend production-v2.1.0-bayesian
+ * Version: 2.2.0 - Aligned with FastAPI Backend production-v2.2.0-stable
  */
 
 const { Server } = require('@modelcontextprotocol/sdk/server/index.js');
@@ -15,6 +15,53 @@ const https = require('https');
 const CloudApiKeyManager = require('./lib/api-key-manager');
 const packageJson = require('./package.json');
 
+const API_KEYS_PREFIX = '/api/v1/api-keys';
+
+const ENDPOINTS = {
+  /** Detect AI context (POST)  */
+  DETECT_CONTEXT:      '/api/v1/templates/detect-context',
+
+  /** Prompt optimization (POST) */
+  OPTIMIZE:            '/api/v1/optimize',
+
+  /** CRUD on templates */
+  TEMPLATE: {
+    /** Create (POST) */
+    CREATE:            '/api/v1/templates/',
+
+    /** Read (GET) */
+    GET:               (id) => `/api/v1/templates/${id}`,
+
+    /** Update (PATCH) */
+    UPDATE:            (id) => `/api/v1/templates/${id}`,
+
+    /** Delete (DELETE) */
+    DELETE:            (id) => `/api/v1/templates/${id}`,
+  },
+
+  /**
+   * Search templates
+   *  (In the current code‑base this is *not* a dedicated `/search` path –
+   *   the original implementation appends query‑params to `/api/v1/templates/`.
+   *   We keep the original path here so that the existing `handleSearchTemplates`
+   *   handler does not need any change other than using the constant.
+   */
+  SEARCH_TEMPLATES:    '/api/v1/templates/',
+
+  /** Quota status (GET) – note the plural “api‑keys” */
+  QUOTA_STATUS:        `${API_KEYS_PREFIX}/quota-status`,
+
+  /** Validate API key (POST) – also under the plural router */
+  VALIDATE_KEY:        `${API_KEYS_PREFIX}/validate`,
+
+  /** Bayesian insights (GET) */
+  ANALYTICS_BAYESIAN_INSIGHTS:
+    '/api/v1/analytics/bayesian-insights',
+
+  /** AG‑UI status (GET) */
+  AGUI_STATUS:         '/api/v1/agui/status',
+};
+
 class MCPPromptOptimizer {
   constructor() {
     this.server = new Server(
@@ -35,8 +82,8 @@ class MCPPromptOptimizer {
     this.requestTimeout = parseInt(process.env.OPTIMIZER_REQUEST_TIMEOUT) || 30000;
     
     // NEW: Feature flags aligned with backend
-    this.bayesianOptimizationEnabled = process.env.ENABLE_BAYESIAN_OPTIMIZATION !== 'false';
-    this.aguiFeatures = process.env.ENABLE_AGUI_FEATURES !== 'false';
+    this.bayesianOptimizationEnabled = process.env.ENABLE_BAYESIAN_OPTIMIZATION === 'true';
+    this.aguiFeatures = process.env.ENABLE_AGUI_FEATURES === 'true';
     
     this.setupMCPHandlers();
   }
@@ -82,6 +129,60 @@ class MCPPromptOptimizer {
           description: "📊 Check subscription status, quota usage, and account information with detailed insights and Bayesian optimization metrics",
           inputSchema: { type: "object", properties: {}, additionalProperties: false }
         },
+        {
+          name: "create_template",
+          description: "➕ Create a new optimization template.",
+          inputSchema: {
+            type: "object",
+            properties: {
+              title: { type: "string", description: "Title of the template" },
+              description: { type: "string", description: "Description of the template" },
+              original_prompt: { type: "string", description: "The original prompt text" },
+              optimized_prompt: { type: "string", description: "The optimized prompt text" },
+              optimization_goals: { type: "array", items: { type: "string" }, description: "Goals for this optimization (e.g., 'clarity', 'conciseness', 'creativity', 'technical_accuracy', 'analytical_depth', 'creative_enhancement')" },
+              confidence_score: { type: "number", description: "Confidence score of the optimization (0.0-1.0)" },
+              model_used: { type: "string", description: "Model used for optimization" },
+              optimization_tier: { type: "string", description: "Tier of optimization (e.g., rules, llm, hybrid)" },
+              ai_context_detected: { type: "string", description: "Detected AI context (e.g., code_generation, image_generation)" },
+              is_public: { type: "boolean", default: false, description: "Whether the template is public" },
+              tags: { type: "array", items: { type: "string" }, description: "Tags for the template" }
+            },
+            required: ["title", "original_prompt", "optimized_prompt", "confidence_score"]
+          }
+        },
+        {
+          name: "get_template",
+          description: "🔍 Retrieve a specific template by its ID.",
+          inputSchema: {
+            type: "object",
+            properties: {
+              template_id: { type: "string", description: "The ID of the template to retrieve" }
+            },
+            required: ["template_id"]
+          }
+        },
+        {
+          name: "update_template",
+          description: "✏️ Update an existing optimization template.",
+          inputSchema: {
+            type: "object",
+            properties: {
+              template_id: { type: "string", description: "The ID of the template to update" },
+              title: { type: "string", description: "New title for the template" },
+              description: { type: "string", description: "New description for the template" },
+              original_prompt: { type: "string", description: "New original prompt text" },
+              optimized_prompt: { type: "string", description: "New optimized prompt text" },
+              optimization_goals: { type: "array", items: { type: "string" }, description: "New optimization goals" },
+              confidence_score: { type: "number", description: "New confidence score (0.0-1.0)" },
+              model_used: { type: "string", description: "New model used for optimization" },
+              optimization_tier: { type: "string", description: "New tier of optimization" },
+              ai_context_detected: { type: "string", description: "New detected AI context" },
+              is_public: { type: "boolean", description: "Whether the template is public" },
+              tags: { type: "array", items: { type: "string" }, description: "New tags for the template" }
+            },
+            required: ["template_id"]
+          }
+        },
         {
           name: "search_templates",
           description: "🔍 Search your saved template library with AI-aware filtering, context-based search, and sophisticated template matching",
@@ -130,7 +231,21 @@ class MCPPromptOptimizer {
               }
             }
           }
-        }
+        },
+        {
+          name: "detect_ai_context",
+          description: "🧠 Detects the AI context for a given prompt using advanced backend analysis.",
+          inputSchema: {
+            type: "object",
+            properties: {
+              prompt: {
+                type: "string",
+                description: "The prompt text for which to detect the AI context"
+              }
+            },
+            required: ["prompt"]
+          }
+        },
       ];
 
       // Add advanced tools if Bayesian optimization is enabled
@@ -176,6 +291,10 @@ class MCPPromptOptimizer {
           case "optimize_prompt": return await this.handleOptimizePrompt(args);
           case "get_quota_status": return await this.handleGetQuotaStatus();
           case "search_templates": return await this.handleSearchTemplates(args);
+          case "detect_ai_context": return await this.handleDetectAIContext(args);
+          case "create_template": return await this.handleCreateTemplate(args);
+          case "get_template": return await this.handleGetTemplate(args);
+          case "update_template": return await this.handleUpdateTemplate(args);
           case "get_optimization_insights": return await this.handleGetOptimizationInsights(args);
           case "get_real_time_status": return await this.handleGetRealTimeStatus();
           default: throw new Error(`Unknown tool: ${name}`);
@@ -186,95 +305,6 @@ class MCPPromptOptimizer {
     });
   }
 
-  detectAIContext(prompt) {
-    const p = prompt.toLowerCase();
-    
-    // Enhanced AI context detection with weighted scoring system - aligned with backend
-    const contextScores = {
-      image_generation: 0,
-      code_generation: 0,
-      api_automation: 0,
-      technical_automation: 0,
-      structured_output: 0,
-      llm_interaction: 0,
-      human_communication: 0
-    };
-
-    const contextKeywords = {
-      image_generation: {
-        "photorealistic": 3.0, "photography": 3.0, "create image": 3.0, "generate picture": 3.0, "--ar": 3.5,
-        "gothic": 2.5, "aspect ratio": 2.5, "16:9": 2.5, "render": 2.5, "make visual": 2.5, "digital art": 2.5,
-        "edits": 2.0, "style": 2.0, "high resolution": 2.0, "cinematic": 2.0, "draw": 2.0, "painting": 2.0, "illustration": 2.0,
-        "image": 1.5, "visual": 1.5, "artwork": 1.5, "picture": 1.5, "design": 1.5, "sketch": 1.5, "logo": 1.5
-      },
-      code_generation: {
-        "```": 3.5, "fibonacci": 3.0, "def ": 3.0, "class ": 3.0, "algorithm": 2.5, "debug": 2.5, "recursive": 2.5,
-        "script": 2.0, "programming": 2.0, "python": 2.0, "javascript": 2.0, "function": 2.0, "c++": 2.0, "import": 2.0,
-        "java": 1.5, "sql": 1.5, "loop": 1.5, "code": 1.5, "variable": 1.5, "return": 1.5, "if ": 1.0, "else": 1.0,
-        "photography": -3.0, "photorealistic": -3.0, "gothic": -2.5, "style": -2.0, "cinematic": -2.0, "visual design": -2.0, "artwork": -2.0
-      },
-      api_automation: {
-        "api": 3.0, "curl": 2.5, "endpoint": 2.5, "get request": 2.5, "post request": 2.5, "webhook": 2.5, "swagger": 2.5, "openapi": 2.5,
-        "json": 2.0, "rest": 2.0, "http": 2.0, "put request": 2.0, "delete request": 2.0, "oauth": 2.0, "postman": 2.0, "bearer token": 2.0, "axios": 2.0, "api key": 2.0, "status code": 2.0,
-        "authentication": 1.5, "request": 1.5, "response": 1.5, "xml": 1.5, "insomnia": 1.5, "fetch": 1.5, "requests": 1.5, "headers": 1.5, "payload": 1.5
-      },
-      technical_automation: {
-        "automate": 3.0, "ci/cd": 3.0, "docker": 2.5, "kubernetes": 2.5, "deploy": 2.5, "deployment": 2.5, "automation": 2.5, "pipeline": 2.5, "github actions": 2.5, "shell script": 2.5,
-        "jenkins": 2.0, "workflow": 2.0, "gitlab ci": 2.0, "azure devops": 2.0, "k8s": 2.0, "terraform": 2.0, "ansible": 2.0, "bash": 2.0, "powershell": 2.0, "cron": 2.0, "infrastructure": 2.0,
-        "helm": 1.5, "chef": 1.5, "puppet": 1.5, "batch": 1.5, "scheduled": 1.5, "monitoring": 1.5
-      },
-      structured_output: {
-        "json": 2.5, "csv": 2.5, "schema": 2.5, "format data": 2.5, "data structure": 2.5,
-        "xml": 2.0, "yaml": 2.0, "format": 2.0, "table": 2.0, "spreadsheet": 2.0, "parse": 2.0, "serialize": 2.0, "deserialize": 2.0,
-        "toml": 1.5, "structure": 1.5, "template": 1.5, "database": 1.5, "export": 1.5, "extract": 1.5, "transform": 1.5, "convert": 1.5, "organize": 1.5, "layout": 1.5, "report": 1.5, "dashboard": 1.5, "chart": 1.5
-      },
-      llm_interaction: {
-        "analyze": 2.5, "analysis": 2.5, "evaluate": 2.0, "assess": 2.0, "research": 2.0, "compare": 2.0, "explain": 2.0, "summarize": 2.0, "pros and cons": 2.0,
-        "study": 1.5, "contrast": 1.5, "examine": 1.5, "investigate": 1.5, "review": 1.5, "critique": 1.5, "describe": 1.5, "interpret": 1.5, "comprehensive": 1.5, "detailed": 1.5, "thorough": 1.5, "in-depth": 1.5, "advantages": 1.5, "disadvantages": 1.5, "opinion": 1.5, "perspective": 1.5, "viewpoint": 1.5
-      },
-      human_communication: {
-        "email": 2.5, "message": 2.0, "letter": 2.0, "compose": 2.0, "communicate": 2.0, "correspondence": 2.0,
-        "memo": 1.5, "write": 1.5, "draft": 1.5, "send": 1.5, "reply": 1.5, "respond": 1.5, "formal": 1.5, "informal": 1.5, "professional": 1.5, "business": 1.5, "greeting": 1.5, "introduction": 1.5, "conclusion": 1.5, "signature": 1.5, "meeting": 1.5, "appointment": 1.5, "schedule": 1.5, "call": 1.5, "conversation": 1.5, "discussion": 1.5, "chat": 1.5,
-        "create": -0.5
-      }
-    };
-
-    // Calculate scores based on keyword presence
-    for (const [context, keywords] of Object.entries(contextKeywords)) {
-      for (const [keyword, weight] of Object.entries(keywords)) {
-        if (p.includes(keyword)) {
-          contextScores[context] += weight;
-        }
-      }
-    }
-
-    // Find the highest scoring context
-    const maxScore = Math.max(...Object.values(contextScores));
-    if (maxScore <= 0) {
-      return 'human_communication';
-    }
-    
-    const detectedContext = Object.keys(contextScores).find(key => contextScores[key] === maxScore) || 'human_communication';
-    return detectedContext;
-  }
-
-  enhanceGoalsForContext(originalGoals, aiContext) {
-    const goals = new Set(originalGoals);
-    
-    const contextGoals = {
-      image_generation: ['creative_enhancement', 'parameter_preservation', 'technical_accuracy'],
-      llm_interaction: ['analytical_depth', 'context_specificity', 'token_efficiency'],
-      technical_automation: ['technical_accuracy', 'parameter_preservation', 'specificity'],
-      code_generation: ['syntax_clarity', 'best_practices', 'maintainability', 'technical_accuracy'],
-      structured_output: ['format_compliance', 'data_validation', 'schema_adherence'],
-      api_automation: ['endpoint_clarity', 'parameter_specification', 'error_handling'],
-      human_communication: ['clarity', 'professionalism', 'engagement']
-    };
-
-    (contextGoals[aiContext] || ['clarity', 'actionability']).forEach(g => goals.add(g));
-    return Array.from(goals);
-  }
-
   generateMockOptimization(prompt, goals, aiContext, enableBayesian = false) {
     const optimized_prompt = `Enhanced for ${aiContext}: ${prompt}`;
     const baseResult = {
@@ -328,47 +358,98 @@ class MCPPromptOptimizer {
     return baseResult;
   }
 
+  generateMockContextDetection(prompt) {
+    let primary_context = 'human_communication'; // Default context
+    const lc = prompt.toLowerCase();   // one‑off lower‑case copy
+
+    /* 1️⃣  Code / programming   – now includes `def` / `return`. */
+    if (lc.match(/def\b|return\b|import\b|class\b|for\b|while\b|if\b|else\b|elif\b|function\b|code\b|python|javascript|java|c\+\+/i)) {
+      primary_context = 'code_generation';
+
+    /* 2️⃣  Image / art   – unchanged. */
+    } else if (lc.match(/image|generate|dall-e|midjourney/i)) {
+      primary_context = 'image_generation';
+
+    /* 3️⃣  Automation   – unchanged. */
+    } else if (lc.match(/automate|script|api/i)) {
+      primary_context = 'technical_automation';
+
+    /* 4️⃣  LLM / analysis – newly added keyword “analyze”. */
+    } else if (lc.match(/analyze|explain|evaluate|summary|research|paper|analysis|interpret|discussion|assessment|compare|contrast/i)) {
+      primary_context = 'llm_interaction';
+    }
+ 
+    return {
+      primary_context: primary_context,
+      confidence: 0.75,
+      secondary_contexts: ['llm_interaction'],
+      detected_parameters: [],
+      mock_mode: true,
+      reason: 'Running in development/fallback mode.'
+    };
+  }
+
   async handleOptimizePrompt(args) {
     if (!args.prompt) throw new Error('Prompt is required');
     
-    const detectedContext = args.ai_context || this.detectAIContext(args.prompt);
-    const enhancedGoals = this.enhanceGoalsForContext(args.goals || ['clarity'], detectedContext);
-    const enableBayesian = args.enable_bayesian !== false && this.bayesianOptimizationEnabled;
-    
     const manager = new CloudApiKeyManager(this.apiKey, { developmentMode: this.developmentMode });
     
     try {
       const validation = await manager.validateApiKey();
       
       if (validation.mock_mode || this.developmentMode) {
-        const mockResult = this.generateMockOptimization(args.prompt, enhancedGoals, detectedContext, enableBayesian);
-        const formatted = this.formatOptimizationResult(mockResult, { detectedContext, enableBayesian });
+        // In mock/dev mode, we still need a context for mock generation
+        const mockContext = args.ai_context || 'human_communication'; 
+        const mockGoals = args.goals || ['clarity'];
+        const mockEnableBayesian = args.enable_bayesian !== false && this.bayesianOptimizationEnabled;
+        const mockResult = this.generateMockOptimization(args.prompt, mockGoals, mockContext, mockEnableBayesian);
+        const formatted = this.formatOptimizationResult(mockResult, { detectedContext: mockContext, enableBayesian: mockEnableBayesian });
         return { content: [{ type: "text", text: formatted }] };
       }
 
-      // Use the correct backend endpoint
-      const result = await this.callBackendAPI('/api/v1/mcp/optimize', { 
+      // 1. Detect AI Context from backend
+      let detectedContext = args.ai_context;
+      if (!detectedContext) {
+        try {
+          const contextDetectionResult = await this.callBackendAPI(ENDPOINTS.DETECT_CONTEXT, { prompt: args.prompt });
+          detectedContext = contextDetectionResult.primary_context;
+          console.error(`Detected AI Context from backend: ${detectedContext}`);
+        } catch (contextError) {
+          console.error(`Failed to detect AI context from backend, falling back to default: ${contextError.message}`);
+          detectedContext = 'human_communication'; // Fallback
+        }
+      }
+
+      // 2. Call the main optimization endpoint
+      const enableBayesian = args.enable_bayesian !== false && this.bayesianOptimizationEnabled;
+      const optimizationPayload = { 
         prompt: args.prompt, 
-        goals: enhancedGoals, 
-        ai_context: detectedContext,
+        goals: args.goals || ['clarity'], // Pass original goals, backend will enhance
+        context: {
+          ai_context: detectedContext
+        },
+        enable_bayesian: enableBayesian,
         metadata: {
-          enable_bayesian: enableBayesian,
           mcp_version: packageJson.version,
           feature_flags: {
             bayesian_optimization: this.bayesianOptimizationEnabled,
             agui_features: this.aguiFeatures
           }
         }
-      });
+      };
+
+      const result = await this.callBackendAPI(ENDPOINTS.OPTIMIZE, optimizationPayload);
       
       return { content: [{ type: "text", text: this.formatOptimizationResult(result, { detectedContext, enableBayesian }) }] };
       
     } catch (error) {
-      if (error.message.includes('Network') || error.message.includes('DNS') || error.message.includes('timeout')) {
-        const fallbackResult = this.generateMockOptimization(args.prompt, enhancedGoals, detectedContext, enableBayesian);
+      if (error.message.includes('Network') || error.message.includes('DNS') || error.message.includes('timeout') || error.message.includes('Connection')) {
+        const fallbackContext = args.ai_context || 'human_communication';
+        const fallbackEnableBayesian = args.enable_bayesian !== false && this.bayesianOptimizationEnabled;
+        const fallbackResult = this.generateMockOptimization(args.prompt, args.goals || ['clarity'], fallbackContext, fallbackEnableBayesian);
         fallbackResult.fallback_mode = true;
         fallbackResult.error_reason = error.message;
-        const formatted = this.formatOptimizationResult(fallbackResult, { detectedContext, enableBayesian });
+        const formatted = this.formatOptimizationResult(fallbackResult, { detectedContext: fallbackContext, enableBayesian: fallbackEnableBayesian });
         return { content: [{ type: "text", text: formatted }] };
       }
       throw new Error(`Optimization failed: ${error.message}`);
@@ -396,7 +477,7 @@ class MCPPromptOptimizer {
       if (args.complexity_level) params.append('complexity_level', args.complexity_level);
       if (args.optimization_strategy) params.append('optimization_strategy', args.optimization_strategy);
 
-      const endpoint = `/api/v1/templates/?${params.toString()}`;
+      const endpoint = `${ENDPOINTS.SEARCH_TEMPLATES}?${params.toString()}`;
       const result = await this.callBackendAPI(endpoint, null, 'GET');
       
       const searchResult = {
@@ -432,7 +513,7 @@ class MCPPromptOptimizer {
 
     try {
       // Try to get insights from backend
-      const endpoint = `/api/v1/analytics/bayesian-insights?depth=${args.analysis_depth || 'detailed'}&recommendations=${args.include_recommendations !== false}`;
+      const endpoint = `${ENDPOINTS.ANALYTICS_BAYESIAN_INSIGHTS}?depth=${args.analysis_depth || 'detailed'}&recommendations=${args.include_recommendations !== false}`;
       const result = await this.callBackendAPI(endpoint, null, 'GET');
       
       return { content: [{ type: "text", text: this.formatOptimizationInsights(result) }] };
@@ -472,8 +553,7 @@ class MCPPromptOptimizer {
     }
 
     try {
-      const endpoint = `/api/v1/agui/status`;
-      const result = await this.callBackendAPI(endpoint, null, 'GET');
+      const result = await this.callBackendAPI(ENDPOINTS.AGUI_STATUS, null, 'GET');
       
       return { content: [{ type: "text", text: this.formatRealTimeStatus(result) }] };
       
@@ -497,9 +577,110 @@ class MCPPromptOptimizer {
     }
   }
 
+  async handleDetectAIContext(args) {
+    if (!args.prompt) throw new Error('Prompt is required');
+
+    const formatResult = (result) => {
+        let output = `# 🧠 AI Context Detection Result\n\n`;
+        output += `**Primary Context:** ${result.primary_context}\n`;
+        output += `**Confidence:** ${(result.confidence * 100).toFixed(1)}%\n`;
+        if (result.secondary_contexts && result.secondary_contexts.length > 0) {
+            output += `**Secondary Contexts:** ${result.secondary_contexts.join(', ')}\n`;
+        }
+        
+        const detections = result.detected_parameters ?? [];
+        const safeDetections = detections.filter(d => d && d.name);
+
+        if (safeDetections.length > 0) {
+            output += `**Detected Parameters:** ${safeDetections.map(d => d.name).join(', ')}\n`;
+        }
+
+        if (result.mock_mode) {
+            output += `\n⚠️ **Fallback Mode Active:** Using mock data due to development mode or network issues.\n`;
+        }
+        return { content: [{ type: "text", text: output }] };
+    };
+
+    try {
+        const manager = new CloudApiKeyManager(this.apiKey, { developmentMode: this.developmentMode });
+        const validation = await manager.validateApiKey();
+
+        if (validation.mock_mode || this.developmentMode) {
+            const mockResult = this.generateMockContextDetection(args.prompt);
+            return formatResult(mockResult);
+        }
+
+        const result = await this.callBackendAPI(ENDPOINTS.DETECT_CONTEXT, { prompt: args.prompt });
+        return formatResult(result);
+
+    } catch (error) {
+        // Fallback for ANY error during the process (missing key, network, etc.)
+        const fallbackResult = this.generateMockContextDetection(args.prompt);
+        return formatResult(fallbackResult);
+    }
+  }
+
+  async handleCreateTemplate(args) {
+    try {
+      const result = await this.callBackendAPI(ENDPOINTS.TEMPLATE.CREATE, args);
+      let output = `# ✅ Template Created Successfully!\n\n`;
+      output += `**Title:** ${result.title}\n`;
+      output += `**ID:** \`${result.id}\`\n`;
+      output += `**Confidence Score:** ${(result.confidence_score * 100).toFixed(1)}%\n`;
+      output += `**AI Context:** ${result.ai_context_detected || 'N/A'}\n`;
+      output += `**Public:** ${result.is_public ? 'Yes' : 'No'}\n`;
+      output += `\n**Optimized Prompt Preview:**\n\`\`\`\n${result.optimized_prompt.substring(0, 150)}...\n\`\`\`\n`;
+      return { content: [{ type: "text", text: output }] };
+    } catch (error) {
+      throw new Error(`Failed to create template: ${error.message}`);
+    }
+  }
+
+  async handleGetTemplate(args) {
+    if (!args.template_id) throw new Error('Template ID is required');
+    try {
+      const result = await this.callBackendAPI(ENDPOINTS.TEMPLATE.GET(args.template_id), null, 'GET');
+      let output = `# 📄 Template Details\n\n`;
+      output += `**Title:** ${result.title}\n`;
+      output += `**ID:** \`${result.id}\`\n`;
+      output += `**Description:** ${result.description || 'N/A'}\n`;
+      output += `**AI Context:** ${result.ai_context_detected || 'N/A'}\n`;
+      output += `**Confidence Score:** ${(result.confidence_score * 100).toFixed(1)}%\n`;
+      output += `**Public:** ${result.is_public ? 'Yes' : 'No'}\n`;
+      output += `**Tags:** ${result.tags ? result.tags.join(', ') : 'None'}\n\n`;
+      output += `**Original Prompt:**\n\`\`\`\n${result.original_prompt}\n\`\`\`\n\n`;
+      output += `**Optimized Prompt:**\n\`\`\`\n${result.optimized_prompt}\n\`\`\`\n`;
+      return { content: [{ type: "text", text: output }] };
+    } catch (error) {
+      throw new Error(`Failed to retrieve template: ${error.message}`);
+    }
+  }
+
+  async handleUpdateTemplate(args) {
+    if (!args.template_id) throw new Error('Template ID is required');
+    try {
+      const { template_id, ...updateData } = args;
+      // Filter out undefined values so we only send fields that are being updated
+      Object.keys(updateData).forEach(key => updateData[key] === undefined && delete updateData[key]);
+      
+      const result = await this.callBackendAPI(ENDPOINTS.TEMPLATE.UPDATE(template_id), updateData, 'PATCH'); // PATCH is better for partial updates
+      let output = `# ✅ Template Updated Successfully!\n\n`;
+      output += `**ID:** \`${result.id}\`\n`;
+      output += `**Title:** ${result.title}\n\n`;
+      output += `Use 'get_template' with the ID to see the full updated template.`;
+      return { content: [{ type: "text", text: output }] };
+    } catch (error) {
+      throw new Error(`Failed to update template: ${error.message}`);
+    }
+  }
+
+  async _buildUrl(path) {
+    return `${this.backendUrl}${path}`;
+  }
+
   async callBackendAPI(endpoint, data, method = 'POST') {
     return new Promise((resolve, reject) => {
-      const url = `${this.backendUrl}${endpoint}`;
+      const url = this._buildUrl(endpoint);
       
       const options = {
         method: method,
@@ -561,8 +742,6 @@ class MCPPromptOptimizer {
         reject(new Error('Request timeout - backend may be unavailable'));
       });
 
-      req.setTimeout(this.requestTimeout);
-      
       if (method !== 'GET' && data) {
         req.write(JSON.stringify(data));
       }
@@ -588,10 +767,12 @@ class MCPPromptOptimizer {
     
     if (result.optimization_insights) {
       const metrics = result.optimization_insights.improvement_metrics;
-      output += `\n📊 **Optimization Insights**\n`;
-      if (metrics.clarity_improvement) output += `- Clarity: +${(metrics.clarity_improvement * 100).toFixed(1)}%\n`;
-      if (metrics.specificity_improvement) output += `- Specificity: +${(metrics.specificity_improvement * 100).toFixed(1)}%\n`;
-      if (metrics.context_alignment) output += `- Context Alignment: +${(metrics.context_alignment * 100).toFixed(1)}%\n`;
+      if (metrics) {
+        output += `\n📊 **Optimization Insights**\n`;
+        if (metrics.clarity_improvement) output += `- Clarity: +${(metrics.clarity_improvement * 100).toFixed(1)}%\n`;
+        if (metrics.specificity_improvement) output += `- Specificity: +${(metrics.specificity_improvement * 100).toFixed(1)}%\n`;
+        if (metrics.context_alignment) output += `- Context Alignment: +${(metrics.context_alignment * 100).toFixed(1)}%\n`;
+      }
       
       if (result.optimization_insights.recommendations?.length) {
         output += `\n💡 **Recommendations:**\n`;
@@ -826,9 +1007,9 @@ class MCPPromptOptimizer {
 }
 
 async function startValidatedMCPServer() {
-  console.error(`🚀 MCP Prompt Optimizer - Professional Cloud Server v${packageJson.version}\n`);
-  console.error(`🧠 Bayesian Optimization: ${process.env.ENABLE_BAYESIAN_OPTIMIZATION !== 'false' ? 'Enabled' : 'Disabled'}`);
-  console.error(`⚡ AG-UI Features: ${process.env.ENABLE_AGUI_FEATURES !== 'false' ? 'Enabled' : 'Disabled'}\n`);
+  console.log(`🚀 MCP Prompt Optimizer - Professional Cloud Server v${packageJson.version}\n`);
+  console.log(`🧠 Bayesian Optimization: ${process.env.ENABLE_BAYESIAN_OPTIMIZATION === 'true' ? 'Enabled' : 'Disabled'}`);
+  console.log(`⚡ AG-UI Features: ${process.env.ENABLE_AGUI_FEATURES === 'true' ? 'Enabled' : 'Disabled'}\n`);
   
   try {
     const apiKey = process.env.OPTIMIZER_API_KEY;
@@ -838,24 +1019,24 @@ async function startValidatedMCPServer() {
     }
     
     const manager = new CloudApiKeyManager(apiKey, { developmentMode: process.env.OPTIMIZER_DEV_MODE === 'true' });
-    console.error('🔧 Validating API key...\n');
+    console.log('🔧 Validating API key...\n');
     const validation = await manager.validateAndPrepare();
     
-    console.error('🔧 Starting MCP server...\n');
+    console.log('🔧 Starting MCP server...\n');
     const mcpServer = new MCPPromptOptimizer();
-    console.error('✅ MCP server ready for connections');
+    console.log('✅ MCP server ready for connections');
     
     // Enhanced status display
     const quotaDisplay = validation.quotaStatus.unlimited ? 
       'Unlimited' : 
       `${validation.quotaStatus.remaining}/${validation.quotaStatus.limit} remaining`;
     
-    console.error(`📊 Plan: ${validation.tier} | Quota: ${quotaDisplay}`);
+    console.log(`📊 Plan: ${validation.tier} | Quota: ${quotaDisplay}`);
     
-    if (validation.mode.mock) console.error('🎭 Running in mock mode');
-    if (validation.mode.development) console.error('⚙️ Development mode active');
-    if (validation.mode.fallback) console.error('🔄 Fallback mode active');
-    if (validation.mode.offline) console.error('📱 Offline mode active');
+    if (validation.mode.mock) console.log('🎭 Running in mock mode');
+    if (validation.mode.development) console.log('⚙️ Development mode active');
+    if (validation.mode.fallback) console.log('🔄 Fallback mode active');
+    if (validation.mode.offline) console.log('📱 Offline mode active');
     
     await mcpServer.run();
   } catch (error) {