@juspay/neurolink 1.5.3 → 1.9.0

package/dist/lib/mcp/servers/ai-providers/ai-workflow-tools.js

@@ -2,123 +2,184 @@
  * AI Development Workflow Tools
  * Phase 1.2 Implementation - 4 specialized tools for AI development lifecycle
  */
-import { z } from 'zod';
+import { z } from "zod";
+import { AIProviderFactory } from "../../../core/factory.js";
+import { getBestProvider } from "../../../utils/providerUtils.js";
 // Tool-specific schemas with comprehensive validation
 const generateTestCasesSchema = z.object({
-    codeFunction: z.string().min(1).describe('The function or code to generate test cases for'),
-    testTypes: z.array(z.enum(['unit', 'integration', 'edge-cases', 'performance', 'security']))
+    codeFunction: z
+        .string()
         .min(1)
-        .default(['unit', 'edge-cases'])
-        .describe('Types of test cases to generate'),
-    framework: z.enum(['jest', 'mocha', 'vitest', 'pytest', 'unittest', 'rspec'])
-        .default('jest')
-        .describe('Testing framework to target'),
-    coverageTarget: z.number().min(0).max(100).default(80).describe('Target test coverage percentage'),
-    includeAsyncTests: z.boolean().default(true).describe('Whether to include async test cases')
+        .describe("The function or code to generate test cases for"),
+    testTypes: z
+        .array(z.enum(["unit", "integration", "edge-cases", "performance", "security"]))
+        .min(1)
+        .default(["unit", "edge-cases"])
+        .describe("Types of test cases to generate"),
+    framework: z
+        .enum(["jest", "mocha", "vitest", "pytest", "unittest", "rspec"])
+        .default("jest")
+        .describe("Testing framework to target"),
+    coverageTarget: z
+        .number()
+        .min(0)
+        .max(100)
+        .default(80)
+        .describe("Target test coverage percentage"),
+    includeAsyncTests: z
+        .boolean()
+        .default(true)
+        .describe("Whether to include async test cases"),
 });
 const refactorCodeSchema = z.object({
-    code: z.string().min(1).describe('The code to refactor'),
-    language: z.string().default('javascript').describe('Programming language of the code'),
-    objectives: z.array(z.enum([
-        'readability',
-        'performance',
-        'maintainability',
-        'testability',
-        'modularity',
-        'dry-principle',
-        'solid-principles'
-    ])).default(['readability', 'maintainability']).describe('Refactoring objectives'),
-    preserveFunctionality: z.boolean().default(true).describe('Ensure functionality remains identical'),
-    styleGuide: z.string().optional().describe('Optional style guide to follow (e.g., airbnb, google)')
+    code: z.string().min(1).describe("The code to refactor"),
+    language: z
+        .string()
+        .default("javascript")
+        .describe("Programming language of the code"),
+    objectives: z
+        .array(z.enum([
+        "readability",
+        "performance",
+        "maintainability",
+        "testability",
+        "modularity",
+        "dry-principle",
+        "solid-principles",
+    ]))
+        .default(["readability", "maintainability"])
+        .describe("Refactoring objectives"),
+    preserveFunctionality: z
+        .boolean()
+        .default(true)
+        .describe("Ensure functionality remains identical"),
+    styleGuide: z
+        .string()
+        .optional()
+        .describe("Optional style guide to follow (e.g., airbnb, google)"),
 });
 const generateDocumentationSchema = z.object({
-    code: z.string().min(1).describe('The code to document'),
-    language: z.string().default('javascript').describe('Programming language of the code'),
-    documentationType: z.enum(['jsdoc', 'markdown', 'sphinx', 'doxygen', 'readme'])
-        .default('jsdoc')
-        .describe('Type of documentation to generate'),
-    includeExamples: z.boolean().default(true).describe('Whether to include usage examples'),
-    detailLevel: z.enum(['minimal', 'standard', 'comprehensive']).default('standard')
-        .describe('Level of documentation detail')
+    code: z.string().min(1).describe("The code to document"),
+    language: z
+        .string()
+        .default("javascript")
+        .describe("Programming language of the code"),
+    documentationType: z
+        .enum(["jsdoc", "markdown", "sphinx", "doxygen", "readme"])
+        .default("jsdoc")
+        .describe("Type of documentation to generate"),
+    includeExamples: z
+        .boolean()
+        .default(true)
+        .describe("Whether to include usage examples"),
+    detailLevel: z
+        .enum(["minimal", "standard", "comprehensive"])
+        .default("standard")
+        .describe("Level of documentation detail"),
 });
 const debugAIOutputSchema = z.object({
-    aiOutput: z.string().min(1).describe('The AI-generated output to debug'),
-    expectedBehavior: z.string().describe('Description of expected behavior or output'),
-    context: z.string().optional().describe('Additional context about the AI generation'),
-    outputType: z.enum(['code', 'text', 'structured-data', 'conversation'])
-        .default('text')
-        .describe('Type of AI output being debugged'),
-    includeFixSuggestions: z.boolean().default(true).describe('Whether to include fix suggestions')
+    aiOutput: z.string().min(1).describe("The AI-generated output to debug"),
+    expectedBehavior: z
+        .string()
+        .describe("Description of expected behavior or output"),
+    context: z
+        .string()
+        .optional()
+        .describe("Additional context about the AI generation"),
+    outputType: z
+        .enum(["code", "text", "structured-data", "conversation"])
+        .default("text")
+        .describe("Type of AI output being debugged"),
+    includeFixSuggestions: z
+        .boolean()
+        .default(true)
+        .describe("Whether to include fix suggestions"),
 });
 /**
  * Generate test cases for code functions
  */
 export const generateTestCasesTool = {
-    name: 'generate-test-cases',
-    description: 'Generate comprehensive test cases for code functions with various test types and frameworks',
-    category: 'ai-workflow',
+    name: "generate-test-cases",
+    description: "Generate comprehensive test cases for code functions with various test types and frameworks",
+    category: "ai-workflow",
     inputSchema: generateTestCasesSchema,
     isImplemented: true,
-    permissions: ['write'],
-    version: '1.0.0',
+    permissions: ["write"],
+    version: "2.0.0", // Updated version with real AI
     execute: async (params, context) => {
         const startTime = Date.now();
         try {
             const validatedParams = generateTestCasesSchema.parse(params);
-            const { codeFunction, testTypes, framework, coverageTarget, includeAsyncTests } = validatedParams;
-            // Simulate test case generation with realistic data
-            const testCases = [];
-            // Generate test cases based on requested types
-            if (testTypes.includes('unit')) {
-                testCases.push({
-                    name: 'should handle basic input correctly',
-                    type: 'unit',
-                    code: `test('should handle basic input correctly', () => {\n  const result = ${extractFunctionName(codeFunction)}('test');\n  expect(result).toBeDefined();\n  expect(typeof result).toBe('string');\n});`,
-                    description: 'Tests basic functionality with standard input',
-                    assertions: 2
-                });
-            }
-            if (testTypes.includes('edge-cases')) {
-                testCases.push({
-                    name: 'should handle null/undefined gracefully',
-                    type: 'edge-case',
-                    code: `test('should handle null/undefined gracefully', () => {\n  expect(() => ${extractFunctionName(codeFunction)}(null)).not.toThrow();\n  expect(() => ${extractFunctionName(codeFunction)}(undefined)).not.toThrow();\n});`,
-                    description: 'Tests edge cases with null and undefined inputs',
-                    assertions: 2
-                });
+            const { codeFunction, testTypes, framework, coverageTarget, includeAsyncTests, } = validatedParams;
+            // Get AI provider for real test case generation
+            const providerName = await getBestProvider();
+            const provider = await AIProviderFactory.createProvider(providerName);
+            if (!provider) {
+                throw new Error(`Failed to create AI provider: ${providerName}`);
             }
-            if (testTypes.includes('integration') && includeAsyncTests) {
-                testCases.push({
-                    name: 'should integrate with async operations',
-                    type: 'integration',
-                    code: `test('should integrate with async operations', async () => {\n  const result = await ${extractFunctionName(codeFunction)}Async('test');\n  expect(result).toBeDefined();\n  expect(result.status).toBe('success');\n});`,
-                    description: 'Tests integration with asynchronous operations',
-                    assertions: 2
-                });
+            // Create structured prompt for test case generation
+            const prompt = `Generate ${testTypes.join(", ")} test cases for this ${framework} function:
+
+${codeFunction}
+
+Requirements:
+- Test types: ${testTypes.join(", ")}
+- Framework: ${framework}
+- Coverage target: ${coverageTarget}%
+- Include async tests: ${includeAsyncTests}
+- Generate realistic, executable test code
+
+Return ONLY a valid JSON object with this exact structure:
+{
+  "testCases": [
+    {
+      "name": "descriptive test name",
+      "type": "unit|integration|edge-cases|performance|security",
+      "code": "complete executable test code for ${framework}",
+      "description": "what this test validates",
+      "assertions": number_of_assertions
+    }
+  ]
+}
+
+Generate 3-5 comprehensive test cases covering the requested types.`;
+            const result = await provider.generateText({
+                prompt,
+                maxTokens: 1200,
+                temperature: 0.3, // Lower temperature for more consistent structured output
+            });
+            if (!result || !result.text) {
+                throw new Error("AI provider returned no result for test case generation.");
             }
-            const result = {
+            // Parse AI response
+            const aiResponse = JSON.parse(result.text);
+            const testCases = aiResponse.testCases || [];
+            const executionTime = Date.now() - startTime;
+            const responseData = {
                 testCases,
                 framework,
-                coverageEstimate: Math.min(coverageTarget, 85 + Math.random() * 10),
+                coverageEstimate: Math.min(coverageTarget, 80 + Math.random() * 15),
                 totalTests: testCases.length,
-                totalAssertions: testCases.reduce((sum, tc) => sum + tc.assertions, 0),
-                executionTime: Date.now() - startTime
+                totalAssertions: testCases.reduce((sum, tc) => sum + (tc.assertions || 1), 0),
+                generatedAt: new Date().toISOString(),
+                aiProvider: providerName,
             };
             return {
                 success: true,
-                data: result,
+                data: responseData,
                 usage: {
-                    executionTime: Date.now() - startTime,
-                    provider: 'workflow-engine',
-                    model: 'test-generator'
+                    ...result.usage,
+                    executionTime,
+                    provider: providerName,
+                    model: "test-case-generator",
                 },
                 metadata: {
-                    toolName: 'generate-test-cases',
-                    serverId: 'neurolink-ai-core',
+                    toolName: "generate-test-cases",
+                    serverId: "neurolink-ai-core",
                     sessionId: context.sessionId,
                     timestamp: Date.now(),
-                    executionTime: Date.now() - startTime
-                }
+                    executionTime,
+                },
             };
         }
         catch (error) {
@@ -128,64 +189,110 @@ export const generateTestCasesTool = {
                 success: false,
                 error: errorMessage,
                 metadata: {
-                    toolName: 'generate-test-cases',
-                    serverId: 'neurolink-ai-core',
+                    toolName: "generate-test-cases",
+                    serverId: "neurolink-ai-core",
                     sessionId: context.sessionId,
                     timestamp: Date.now(),
-                    executionTime
-                }
+                    executionTime,
+                },
             };
         }
-    }
+    },
 };
 /**
  * Refactor code for improved quality
  */
 export const refactorCodeTool = {
-    name: 'refactor-code',
-    description: 'AI-powered code refactoring for improved readability, performance, and maintainability',
-    category: 'ai-workflow',
+    name: "refactor-code",
+    description: "AI-powered code refactoring for improved readability, performance, and maintainability",
+    category: "ai-workflow",
     inputSchema: refactorCodeSchema,
     isImplemented: true,
-    permissions: ['write'],
-    version: '1.0.0',
+    permissions: ["write"],
+    version: "2.0.0", // Updated version with real AI
     execute: async (params, context) => {
         const startTime = Date.now();
         try {
             const validatedParams = refactorCodeSchema.parse(params);
             const { code, language, objectives, preserveFunctionality, styleGuide } = validatedParams;
-            // Simulate code refactoring with improvements
-            const refactoredCode = simulateRefactoring(code, objectives, styleGuide);
-            const result = {
-                refactoredCode,
-                changes: [
-                    'Extracted magic numbers into named constants',
-                    'Simplified conditional logic using early returns',
-                    'Renamed variables for clarity',
-                    'Added proper error handling'
-                ],
-                improvements: objectives.map(obj => `Improved ${obj}`),
+            // Get AI provider for real code refactoring
+            const providerName = await getBestProvider();
+            const provider = await AIProviderFactory.createProvider(providerName);
+            if (!provider) {
+                throw new Error(`Failed to create AI provider: ${providerName}`);
+            }
+            // Create structured prompt for code refactoring
+            const prompt = `Refactor this ${language} code focusing on: ${objectives.join(", ")}
+
+Original code:
+\`\`\`${language}
+${code}
+\`\`\`
+
+Requirements:
+- Language: ${language}
+- Objectives: ${objectives.join(", ")}
+- Style guide: ${styleGuide || "standard best practices"}
+- Preserve functionality: ${preserveFunctionality}
+- Provide clean, production-ready code
+
+Return ONLY a valid JSON object with this exact structure:
+{
+  "refactoredCode": "improved code here with proper formatting",
+  "changes": ["specific change 1", "specific change 2", "specific change 3"],
+  "improvements": ["improvement achieved 1", "improvement achieved 2"],
+  "metrics": {
+    "linesReduced": positive_number_or_0,
+    "complexityReduction": percentage_number,
+    "readabilityScore": score_out_of_100
+  }
+}
+
+Focus on real, actionable improvements based on the specified objectives.`;
+            const result = await provider.generateText({
+                prompt,
+                maxTokens: 1000,
+                temperature: 0.2, // Very low temperature for consistent refactoring
+            });
+            if (!result || !result.text) {
+                throw new Error("AI provider returned no result for code refactoring.");
+            }
+            // Parse AI response
+            const aiResponse = JSON.parse(result.text);
+            const executionTime = Date.now() - startTime;
+            const responseData = {
+                refactoredCode: aiResponse.refactoredCode || code,
+                changes: aiResponse.changes || ["Code refactored successfully"],
+                improvements: aiResponse.improvements || objectives.map((obj) => `Improved ${obj}`),
                 metrics: {
-                    linesReduced: Math.floor(Math.random() * 10) + 5,
-                    complexityReduction: Math.floor(Math.random() * 20) + 10,
-                    readabilityScore: 85 + Math.floor(Math.random() * 10)
-                }
+                    linesReduced: aiResponse.metrics?.linesReduced || 0,
+                    complexityReduction: aiResponse.metrics?.complexityReduction || 15,
+                    readabilityScore: aiResponse.metrics?.readabilityScore || 85,
+                },
             };
             return {
                 success: true,
-                data: result,
+                data: {
+                    ...responseData,
+                    originalCode: code,
+                    language,
+                    objectives,
+                    generatedAt: new Date().toISOString(),
+                    aiProvider: providerName,
+                },
                 usage: {
-                    executionTime: Date.now() - startTime,
-                    provider: 'workflow-engine',
-                    model: 'refactor-engine'
+                    ...result.usage,
+                    executionTime,
+                    provider: providerName,
+                    model: "code-refactorer",
                 },
                 metadata: {
-                    toolName: 'refactor-code',
-                    serverId: 'neurolink-ai-core',
+                    toolName: "refactor-code",
+                    serverId: "neurolink-ai-core",
                     sessionId: context.sessionId,
                     timestamp: Date.now(),
-                    executionTime: Date.now() - startTime
-                }
+                    executionTime,
+                },
             };
         }
         catch (error) {
@@ -195,89 +302,108 @@ export const refactorCodeTool = {
                 success: false,
                 error: errorMessage,
                 metadata: {
-                    toolName: 'refactor-code',
-                    serverId: 'neurolink-ai-core',
+                    toolName: "refactor-code",
+                    serverId: "neurolink-ai-core",
                     sessionId: context.sessionId,
                     timestamp: Date.now(),
-                    executionTime
-                }
+                    executionTime,
+                },
             };
         }
-    }
+    },
 };
 /**
  * Generate documentation from code
  */
 export const generateDocumentationTool = {
-    name: 'generate-documentation',
-    description: 'Automatically generate comprehensive documentation from code',
-    category: 'ai-workflow',
+    name: "generate-documentation",
+    description: "Automatically generate comprehensive documentation from code",
+    category: "ai-workflow",
     inputSchema: generateDocumentationSchema,
     isImplemented: true,
-    permissions: ['read'],
-    version: '1.0.0',
+    permissions: ["read"],
+    version: "2.0.0", // Updated version with real AI
     execute: async (params, context) => {
         const startTime = Date.now();
         try {
             const validatedParams = generateDocumentationSchema.parse(params);
-            const { code, language, documentationType, includeExamples, detailLevel } = validatedParams;
-            // Generate documentation based on type
-            let documentation = '';
-            const sections = [];
-            const examples = [];
-            if (documentationType === 'jsdoc') {
-                documentation = `/**
- * ${extractFunctionName(code)} - Processes input data and returns formatted result
- *
- * @param {string} input - The input data to process
- * @param {Object} options - Configuration options
- * @param {boolean} options.validate - Whether to validate input
- * @param {number} options.timeout - Operation timeout in milliseconds
- * @returns {Promise<Object>} Processed result object
- * @throws {Error} If input validation fails
- */`;
-                sections.push('Parameters', 'Returns', 'Throws');
+            const { code, language, documentationType, includeExamples, detailLevel, } = validatedParams;
+            // Get AI provider for real documentation generation
+            const providerName = await getBestProvider();
+            const provider = await AIProviderFactory.createProvider(providerName);
+            if (!provider) {
+                throw new Error(`Failed to create AI provider: ${providerName}`);
             }
-            else if (documentationType === 'markdown') {
-                documentation = `# ${extractFunctionName(code)}
+            // Create structured prompt for documentation generation
+            const prompt = `Generate ${documentationType} documentation for this ${language} code:
 
-## Description
-Processes input data and returns formatted result with validation and timeout support.
+\`\`\`${language}
+${code}
+\`\`\`
 
-## Parameters
-- \`input\` (string): The input data to process
-- \`options\` (object): Configuration options
-  - \`validate\` (boolean): Whether to validate input
-  - \`timeout\` (number): Operation timeout in milliseconds
+Requirements:
+- Language: ${language}
+- Documentation type: ${documentationType}
+- Detail level: ${detailLevel}
+- Include examples: ${includeExamples}
+- Generate professional, comprehensive documentation
 
-## Returns
-Promise<Object>: Processed result object`;
-                sections.push('Description', 'Parameters', 'Returns');
-            }
-            if (includeExamples) {
-                examples.push(`// Basic usage\nconst result = await ${extractFunctionName(code)}('data', { validate: true });`, `// With timeout\nconst result = await ${extractFunctionName(code)}('data', { timeout: 5000 });`);
+Return ONLY a valid JSON object with this exact structure:
+{
+  "documentation": "formatted documentation string in ${documentationType} format",
+  "sections": ["list of documentation sections included"],
+  "examples": ${includeExamples ? '["code examples with usage"]' : "[]"},
+  "coverage": percentage_number_representing_documentation_completeness
+}
+
+Focus on creating accurate, useful documentation that explains the code's purpose, parameters, return values, and usage patterns.`;
+            const result = await provider.generateText({
+                prompt,
+                maxTokens: 1200,
+                temperature: 0.3, // Moderate temperature for creative but structured documentation
+            });
+            if (!result || !result.text) {
+                throw new Error("AI provider returned no result for documentation generation.");
             }
-            const result = {
-                documentation,
-                sections,
-                examples,
-                coverage: detailLevel === 'comprehensive' ? 95 : detailLevel === 'standard' ? 80 : 60
+            // Parse AI response
+            const aiResponse = JSON.parse(result.text);
+            const executionTime = Date.now() - startTime;
+            const responseData = {
+                documentation: aiResponse.documentation || "Documentation generated successfully",
+                sections: aiResponse.sections || ["Overview"],
+                examples: aiResponse.examples || [],
+                coverage: aiResponse.coverage ||
+                    (detailLevel === "comprehensive"
+                        ? 95
+                        : detailLevel === "standard"
+                            ? 80
+                            : 60),
             };
             return {
                 success: true,
-                data: result,
+                data: {
+                    ...responseData,
+                    originalCode: code,
+                    language,
+                    documentationType,
+                    detailLevel,
+                    includeExamples,
+                    generatedAt: new Date().toISOString(),
+                    aiProvider: providerName,
+                },
                 usage: {
-                    executionTime: Date.now() - startTime,
-                    provider: 'workflow-engine',
-                    model: 'doc-generator'
+                    ...result.usage,
+                    executionTime,
+                    provider: providerName,
+                    model: "documentation-generator",
                 },
                 metadata: {
-                    toolName: 'generate-documentation',
-                    serverId: 'neurolink-ai-core',
+                    toolName: "generate-documentation",
+                    serverId: "neurolink-ai-core",
                     sessionId: context.sessionId,
                     timestamp: Date.now(),
-                    executionTime: Date.now() - startTime
-                }
+                    executionTime,
+                },
             };
         }
         catch (error) {
@@ -287,96 +413,121 @@ Promise<Object>: Processed result object`;
                 success: false,
                 error: errorMessage,
                 metadata: {
-                    toolName: 'generate-documentation',
-                    serverId: 'neurolink-ai-core',
+                    toolName: "generate-documentation",
+                    serverId: "neurolink-ai-core",
                     sessionId: context.sessionId,
                     timestamp: Date.now(),
-                    executionTime
-                }
+                    executionTime,
+                },
             };
         }
-    }
+    },
 };
 /**
  * Debug AI-generated output
  */
 export const debugAIOutputTool = {
-    name: 'debug-ai-output',
-    description: 'Analyze and debug AI-generated output to identify issues and suggest improvements',
-    category: 'ai-workflow',
+    name: "debug-ai-output",
+    description: "Analyze and debug AI-generated output to identify issues and suggest improvements",
+    category: "ai-workflow",
     inputSchema: debugAIOutputSchema,
     isImplemented: true,
-    permissions: ['read', 'analytics'],
-    version: '1.0.0',
+    permissions: ["read", "analytics"],
+    version: "2.0.0", // Updated version with real AI
     execute: async (params, context) => {
         const startTime = Date.now();
         try {
             const validatedParams = debugAIOutputSchema.parse(params);
-            const { aiOutput, expectedBehavior, context: debugContext, outputType, includeFixSuggestions } = validatedParams;
-            // Analyze AI output for issues
-            const issues = [];
-            const suggestions = [];
-            const possibleCauses = [];
-            // Simulate issue detection based on output type
-            if (outputType === 'code') {
-                if (!aiOutput.includes('error handling')) {
-                    issues.push({
-                        type: 'missing-error-handling',
-                        severity: 'medium',
-                        description: 'Code lacks proper error handling',
-                        location: 'throughout'
-                    });
-                    suggestions.push('Add try-catch blocks for error handling');
-                }
-                if (aiOutput.length < 50) {
-                    issues.push({
-                        type: 'incomplete-implementation',
-                        severity: 'high',
-                        description: 'Code appears incomplete or truncated',
-                        location: 'end of output'
-                    });
-                    possibleCauses.push('Token limit reached', 'Prompt ambiguity');
-                }
-            }
-            else if (outputType === 'text') {
-                if (aiOutput.toLowerCase() !== aiOutput && aiOutput.toUpperCase() !== aiOutput) {
-                    // Mixed case - check for consistency
-                    if (Math.random() > 0.7) {
-                        issues.push({
-                            type: 'inconsistent-formatting',
-                            severity: 'low',
-                            description: 'Inconsistent text formatting detected',
-                            location: 'various'
-                        });
-                    }
-                }
+            const { aiOutput, expectedBehavior, context: debugContext, outputType, includeFixSuggestions, } = validatedParams;
+            // Get AI provider for real output analysis
+            const providerName = await getBestProvider();
+            const provider = await AIProviderFactory.createProvider(providerName);
+            if (!provider) {
+                throw new Error(`Failed to create AI provider: ${providerName}`);
             }
-            // Add general suggestions if requested
-            if (includeFixSuggestions) {
-                suggestions.push('Refine the prompt for clearer instructions', 'Adjust temperature parameter for more consistent output', 'Consider using system prompts for better context');
+            // Create structured prompt for AI output debugging
+            const prompt = `Analyze this AI-generated ${outputType} output for issues and improvements:
+
+AI Output to Debug:
+\`\`\`
+${aiOutput}
+\`\`\`
+
+Expected Behavior:
+${expectedBehavior}
+
+Context: ${debugContext || "None provided"}
+Output Type: ${outputType}
+Include Fix Suggestions: ${includeFixSuggestions}
+
+Analyze the output for:
+1. Quality issues (completeness, accuracy, formatting)
+2. Technical problems (syntax errors, logical flaws)
+3. Content issues (relevance, clarity, consistency)
+4. Improvement opportunities
+
+Return ONLY a valid JSON object with this exact structure:
+{
+  "issues": [
+    {
+      "type": "issue-category",
+      "severity": "low|medium|high",
+      "description": "detailed description of the issue",
+      "location": "where in output this occurs"
+    }
+  ],
+  "suggestions": ["actionable improvement suggestion 1", "suggestion 2"],
+  "possibleCauses": ["potential cause 1", "potential cause 2"],
+  "fixedOutput": ${includeFixSuggestions ? '"corrected version if possible"' : "null"}
+}
+
+Provide thorough, actionable analysis focused on improving AI output quality.`;
+            const result = await provider.generateText({
+                prompt,
+                maxTokens: 1000,
+                temperature: 0.4, // Moderate temperature for analytical thinking
+            });
+            if (!result || !result.text) {
+                throw new Error("AI provider returned no result for output debugging.");
             }
-            const result = {
-                issues,
-                suggestions,
-                possibleCauses: possibleCauses.length > 0 ? possibleCauses : ['Prompt clarity', 'Model limitations'],
-                fixedOutput: issues.length > 0 && includeFixSuggestions ?
-                    `${aiOutput}\n// TODO: Add error handling and validation` : undefined
+            // Parse AI response
+            const aiResponse = JSON.parse(result.text);
+            const executionTime = Date.now() - startTime;
+            const responseData = {
+                issues: aiResponse.issues || [],
+                suggestions: aiResponse.suggestions || [
+                    "Consider refining the prompt for clearer instructions",
+                ],
+                possibleCauses: aiResponse.possibleCauses || [
+                    "Prompt clarity",
+                    "Model limitations",
+                ],
+                fixedOutput: aiResponse.fixedOutput || undefined,
             };
             return {
                 success: true,
-                data: result,
+                data: {
+                    ...responseData,
+                    originalOutput: aiOutput,
+                    expectedBehavior,
+                    outputType,
+                    analysisContext: debugContext,
+                    generatedAt: new Date().toISOString(),
+                    aiProvider: providerName,
+                },
                 usage: {
-                    executionTime: Date.now() - startTime,
-                    provider: 'workflow-engine',
-                    model: 'debug-analyzer'
+                    ...result.usage,
+                    executionTime,
+                    provider: providerName,
+                    model: "ai-output-debugger",
                 },
                 metadata: {
-                    toolName: 'debug-ai-output',
-                    serverId: 'neurolink-ai-core',
+                    toolName: "debug-ai-output",
+                    serverId: "neurolink-ai-core",
                     sessionId: context.sessionId,
                     timestamp: Date.now(),
-                    executionTime: Date.now() - startTime
-                }
+                    executionTime,
+                },
             };
         }
         catch (error) {
@@ -386,28 +537,30 @@ export const debugAIOutputTool = {
                 success: false,
                 error: errorMessage,
                 metadata: {
-                    toolName: 'debug-ai-output',
-                    serverId: 'neurolink-ai-core',
+                    toolName: "debug-ai-output",
+                    serverId: "neurolink-ai-core",
                     sessionId: context.sessionId,
                     timestamp: Date.now(),
-                    executionTime
-                }
+                    executionTime,
+                },
             };
         }
-    }
+    },
 };
 // Helper functions
 function extractFunctionName(code) {
     const match = code.match(/function\s+(\w+)|const\s+(\w+)\s*=|(\w+)\s*\(/);
-    return match ? (match[1] || match[2] || match[3] || 'processData') : 'processData';
+    return match
+        ? match[1] || match[2] || match[3] || "processData"
+        : "processData";
 }
 function simulateRefactoring(code, objectives, styleGuide) {
     // Simulate basic refactoring
     let refactored = code;
-    if (objectives.includes('readability')) {
-        refactored = refactored.replace(/([a-z])([A-Z])/g, '$1 $2');
+    if (objectives.includes("readability")) {
+        refactored = refactored.replace(/([a-z])([A-Z])/g, "$1 $2");
     }
-    if (objectives.includes('dry-principle')) {
+    if (objectives.includes("dry-principle")) {
         refactored = `// Extracted common functionality\nconst CONSTANTS = { MAX_RETRIES: 3, TIMEOUT: 5000 };\n\n${refactored}`;
     }
     return refactored;
@@ -417,12 +570,12 @@ export const aiWorkflowTools = [
     generateTestCasesTool,
     refactorCodeTool,
     generateDocumentationTool,
-    debugAIOutputTool
+    debugAIOutputTool,
 ];
 // Export schemas for external validation
 export const workflowToolSchemas = {
-    'generate-test-cases': generateTestCasesSchema,
-    'refactor-code': refactorCodeSchema,
-    'generate-documentation': generateDocumentationSchema,
-    'debug-ai-output': debugAIOutputSchema
+    "generate-test-cases": generateTestCasesSchema,
+    "refactor-code": refactorCodeSchema,
+    "generate-documentation": generateDocumentationSchema,
+    "debug-ai-output": debugAIOutputSchema,
 };