/**
* Knowledge base and analytics handlers
* Extracted from main server file for better organization
*/
import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js';
import { logger } from '../utils/logger.js';
import SecurityMiddleware from '../security/middleware.js';
import {
QueryClusterDataSchema,
GetClusterInsightsSchema,
GetJobAnalyticsSchema,
QueryKnowledgeSchema,
} from '../validation/schemas.js';
import { KnowledgeIndexer } from '../services/knowledge-indexer.js';
import { SemanticQueryService } from '../services/semantic-query.js';
import { TemplatingIntegration } from '../services/templating-integration.js';
import { DefaultParameterManager } from '../services/default-params.js';
export interface KnowledgeHandlerDependencies {
knowledgeIndexer?: KnowledgeIndexer;
semanticQueryService?: SemanticQueryService;
templatingIntegration?: TemplatingIntegration;
defaultParamManager?: DefaultParameterManager;
}
export async function handleQueryClusterData(args: any, deps: KnowledgeHandlerDependencies) {
// Apply security middleware
SecurityMiddleware.checkRateLimit(`query_cluster_data:${JSON.stringify(args)}`);
// Sanitize input
const sanitizedArgs = SecurityMiddleware.sanitizeObject(args);
// Validate input with Zod schema
let validatedArgs;
try {
validatedArgs = SecurityMiddleware.validateInput(QueryClusterDataSchema, sanitizedArgs);
} catch (error) {
SecurityMiddleware.auditLog(
'Input validation failed',
{
tool: 'query_cluster_data',
error: error instanceof Error ? error.message : 'Unknown error',
args: SecurityMiddleware.sanitizeForLogging(args),
},
'warn'
);
throw new McpError(
ErrorCode.InvalidParams,
error instanceof Error ? error.message : 'Invalid input'
);
}
// Get default parameters if not provided
let { projectId, region } = validatedArgs;
const { query, clusterName, limit } = validatedArgs;
if (!projectId && deps.defaultParamManager) {
try {
projectId = deps.defaultParamManager.getParameterValue('projectId');
} catch (error) {
// Ignore error, parameter is optional
}
}
if (!region && deps.defaultParamManager) {
try {
region = deps.defaultParamManager.getParameterValue('region');
} catch (error) {
// Ignore error, parameter is optional
}
}
// Audit log the operation
SecurityMiddleware.auditLog('Cluster data query initiated', {
tool: 'query_cluster_data',
queryLength: query.length,
projectId,
region,
clusterName,
limit,
});
if (!deps.knowledgeIndexer) {
return {
content: [
{
type: 'text',
text: 'š **Cluster Data Query**\n\nā ļø Knowledge indexer not available. Semantic search requires Qdrant vector database.\n\nš” **To enable semantic search:**\n- Start Qdrant: `docker run -p 6334:6333 qdrant/qdrant`\n- Restart the MCP server',
},
],
};
}
try {
const results = await deps.knowledgeIndexer.queryKnowledge(query, {
type: 'cluster',
projectId,
region,
limit,
});
SecurityMiddleware.auditLog('Cluster data query completed', {
tool: 'query_cluster_data',
queryLength: query.length,
resultCount: results.length,
});
if (results.length === 0) {
return {
content: [
{
type: 'text',
text: `š **Cluster Data Query**: "${query}"\n\nā **No results found**\n\nTry:\n- Different keywords\n- Broader search terms\n- Check if clusters have been indexed`,
},
],
};
}
let response = `š **Cluster Data Query**: "${query}"\n`;
response += `š **Found**: ${results.length} matching results\n\n`;
results.forEach((result, index) => {
const data = result.data as any;
response += `**${index + 1}. ${data.clusterName || 'Unknown'}** (${data.projectId || 'Unknown'}/${data.region || 'Unknown'})\n`;
response += ` šÆ Confidence: ${(result.confidence * 100).toFixed(1)}%\n`;
if (data.configurations?.machineTypes?.length > 0) {
response += ` š„ļø Machine types: ${data.configurations.machineTypes.join(', ')}\n`;
}
if (data.configurations?.components?.length > 0) {
response += ` š§ Components: ${data.configurations.components.join(', ')}\n`;
}
response += '\n';
});
return {
content: [
{
type: 'text',
text: response,
},
],
};
} catch (error) {
SecurityMiddleware.auditLog(
'Cluster data query failed',
{
tool: 'query_cluster_data',
query,
error: error instanceof Error ? error.message : 'Unknown error',
},
'error'
);
logger.error('MCP query_cluster_data: Error:', error);
throw error;
}
}
export async function handleGetClusterInsights(args: any, deps: KnowledgeHandlerDependencies) {
// Apply security middleware
SecurityMiddleware.checkRateLimit(`get_cluster_insights:${JSON.stringify(args)}`);
// Sanitize input
const sanitizedArgs = SecurityMiddleware.sanitizeObject(args);
// Validate input with Zod schema
let validatedArgs;
try {
validatedArgs = SecurityMiddleware.validateInput(GetClusterInsightsSchema, sanitizedArgs);
} catch (error) {
SecurityMiddleware.auditLog(
'Input validation failed',
{
tool: 'get_cluster_insights',
error: error instanceof Error ? error.message : 'Unknown error',
args: SecurityMiddleware.sanitizeForLogging(args),
},
'warn'
);
throw new McpError(
ErrorCode.InvalidParams,
error instanceof Error ? error.message : 'Invalid input'
);
}
// Get default parameters if not provided
let { projectId, region } = validatedArgs;
if (!projectId && deps.defaultParamManager) {
try {
projectId = deps.defaultParamManager.getParameterValue('projectId');
} catch (error) {
// Ignore error, parameter is optional
}
}
if (!region && deps.defaultParamManager) {
try {
region = deps.defaultParamManager.getParameterValue('region');
} catch (error) {
// Ignore error, parameter is optional
}
}
// Audit log the operation
SecurityMiddleware.auditLog('Cluster insights requested', {
tool: 'get_cluster_insights',
projectId,
region,
});
if (!deps.knowledgeIndexer) {
return {
content: [
{
type: 'text',
text: 'š **Cluster Insights**\n\nā ļø Knowledge indexer not available. Insights require Qdrant vector database.\n\nš” **To enable insights:**\n- Start Qdrant: `docker run -p 6334:6333 qdrant/qdrant`\n- Restart the MCP server',
},
],
};
}
try {
// Try dynamic insights first, fallback to legacy
let response = 'š **Cluster Discovery Insights**\n\n';
// Use dynamic insights
const dynamicInsights = await deps.knowledgeIndexer.getDynamicClusterInsights();
response += `šļø **Total Clusters**: ${dynamicInsights.totalDocuments}\n\n`;
// Show top field analysis
if (dynamicInsights.fieldAnalysis.length > 0) {
response += `š **Dynamic Field Analysis**:\n`;
dynamicInsights.fieldAnalysis.slice(0, 5).forEach((field) => {
response += ` ā¢ ${field.fieldName} (${field.fieldType}): ${field.uniqueCount} unique values\n`;
});
response += '\n';
}
// Show patterns
if (dynamicInsights.patterns.length > 0) {
response += `š **Detected Patterns**:\n`;
dynamicInsights.patterns.forEach((pattern) => {
response += ` **${pattern.category}** (${(pattern.confidence * 100).toFixed(0)}% confidence):\n`;
pattern.insights.forEach((insight) => {
response += ` ā¢ ${insight}\n`;
});
});
response += '\n';
}
// Show recommendations
if (dynamicInsights.recommendations.length > 0) {
response += `š” **Recommendations**:\n`;
dynamicInsights.recommendations.forEach((rec) => {
response += ` ā¢ ${rec}\n`;
});
response += '\n';
}
response += `š
**Last Updated**: ${new Date().toISOString()}`;
SecurityMiddleware.auditLog('Cluster insights completed', {
tool: 'get_cluster_insights',
responseLength: response.length,
});
return {
content: [
{
type: 'text',
text: response,
},
],
};
} catch (error) {
SecurityMiddleware.auditLog(
'Cluster insights failed',
{
tool: 'get_cluster_insights',
error: error instanceof Error ? error.message : 'Unknown error',
},
'error'
);
logger.error('MCP get_cluster_insights: Error:', error);
// Fallback response
return {
content: [
{
type: 'text',
text: 'š **Cluster Insights**\n\nā ļø Unable to retrieve insights at this time. This may be due to:\n- Qdrant not running\n- No clusters indexed yet\n- Service initialization in progress',
},
],
};
}
}
export async function handleGetJobAnalytics(args: any, deps: KnowledgeHandlerDependencies) {
// Apply security middleware
SecurityMiddleware.checkRateLimit(`get_job_analytics:${JSON.stringify(args)}`);
// Sanitize input
const sanitizedArgs = SecurityMiddleware.sanitizeObject(args);
// Validate input with Zod schema
let validatedArgs;
try {
validatedArgs = SecurityMiddleware.validateInput(GetJobAnalyticsSchema, sanitizedArgs);
} catch (error) {
SecurityMiddleware.auditLog(
'Input validation failed',
{
tool: 'get_job_analytics',
error: error instanceof Error ? error.message : 'Unknown error',
args: SecurityMiddleware.sanitizeForLogging(args),
},
'warn'
);
throw new McpError(
ErrorCode.InvalidParams,
error instanceof Error ? error.message : 'Invalid input'
);
}
// Get default parameters if not provided
let { projectId, region } = validatedArgs;
if (!projectId && deps.defaultParamManager) {
try {
projectId = deps.defaultParamManager.getParameterValue('projectId');
} catch (error) {
// Ignore error, parameter is optional
}
}
if (!region && deps.defaultParamManager) {
try {
region = deps.defaultParamManager.getParameterValue('region');
} catch (error) {
// Ignore error, parameter is optional
}
}
// Audit log the operation
SecurityMiddleware.auditLog('Job analytics requested', {
tool: 'get_job_analytics',
projectId,
region,
});
if (!deps.knowledgeIndexer) {
return {
content: [
{
type: 'text',
text: 'š **Job Analytics**\n\nā ļø Knowledge indexer not available. Analytics require Qdrant vector database.\n\nš” **To enable analytics:**\n- Start Qdrant: `docker run -p 6334:6333 qdrant/qdrant`\n- Restart the MCP server',
},
],
};
}
try {
// Use dynamic analytics
let response = 'š **Job Submission Analytics**\n\n';
const dynamicAnalytics = await deps.knowledgeIndexer.getDynamicJobAnalytics();
response += `š **Total Jobs**: ${dynamicAnalytics.totalDocuments}\n\n`;
// Show top field analysis
if (dynamicAnalytics.fieldAnalysis.length > 0) {
response += `š **Dynamic Field Analysis**:\n`;
dynamicAnalytics.fieldAnalysis.slice(0, 5).forEach((field) => {
response += ` ā¢ ${field.fieldName} (${field.fieldType}): ${field.uniqueCount} unique values\n`;
if (field.statistics?.avg) {
response += ` Average: ${field.statistics.avg.toFixed(2)}\n`;
}
});
response += '\n';
}
// Show patterns
if (dynamicAnalytics.patterns.length > 0) {
response += `š **Detected Patterns**:\n`;
dynamicAnalytics.patterns.forEach((pattern) => {
response += ` **${pattern.category}** (${(pattern.confidence * 100).toFixed(0)}% confidence):\n`;
pattern.insights.forEach((insight) => {
response += ` ā¢ ${insight}\n`;
});
});
response += '\n';
}
// Show recommendations
if (dynamicAnalytics.recommendations.length > 0) {
response += `š” **Recommendations**:\n`;
dynamicAnalytics.recommendations.forEach((rec) => {
response += ` ā¢ ${rec}\n`;
});
response += '\n';
}
response += `š
**Last Updated**: ${new Date().toISOString()}`;
SecurityMiddleware.auditLog('Job analytics completed', {
tool: 'get_job_analytics',
responseLength: response.length,
});
return {
content: [
{
type: 'text',
text: response,
},
],
};
} catch (error) {
SecurityMiddleware.auditLog(
'Job analytics failed',
{
tool: 'get_job_analytics',
error: error instanceof Error ? error.message : 'Unknown error',
},
'error'
);
logger.error('MCP get_job_analytics: Error:', error);
// Fallback response
return {
content: [
{
type: 'text',
text: 'š **Job Analytics**\n\nā ļø Unable to retrieve analytics at this time. This may be due to:\n- Qdrant not running\n- No jobs indexed yet\n- Service initialization in progress',
},
],
};
}
}
export async function handleQueryKnowledge(args: any, deps: KnowledgeHandlerDependencies) {
// Apply security middleware
SecurityMiddleware.checkRateLimit(`query_knowledge:${JSON.stringify(args)}`);
// Sanitize input
const sanitizedArgs = SecurityMiddleware.sanitizeObject(args);
// Validate input with Zod schema
let validatedArgs;
try {
validatedArgs = SecurityMiddleware.validateInput(QueryKnowledgeSchema, sanitizedArgs);
} catch (error) {
SecurityMiddleware.auditLog(
'Input validation failed',
{
tool: 'query_knowledge',
error: error instanceof Error ? error.message : 'Unknown error',
args: SecurityMiddleware.sanitizeForLogging(args),
},
'warn'
);
throw new McpError(
ErrorCode.InvalidParams,
error instanceof Error ? error.message : 'Invalid input'
);
}
// Get default parameters if not provided
let { projectId, region } = validatedArgs;
const { query, type, limit, includeRawDocument } = validatedArgs;
if (!projectId && deps.defaultParamManager) {
try {
projectId = deps.defaultParamManager.getParameterValue('projectId');
} catch (error) {
// Ignore error, parameter is optional
}
}
if (!region && deps.defaultParamManager) {
try {
region = deps.defaultParamManager.getParameterValue('region');
} catch (error) {
// Ignore error, parameter is optional
}
}
// Audit log the operation
SecurityMiddleware.auditLog('Knowledge query initiated', {
tool: 'query_knowledge',
queryLength: query.length,
type,
projectId,
region,
limit,
includeRawDocument,
});
if (!deps.knowledgeIndexer) {
return {
content: [
{
type: 'text',
text: 'š§ **Knowledge Base Query**\n\nā ļø Knowledge indexer not available. Semantic search requires Qdrant vector database.\n\nš” **To enable knowledge search:**\n- Start Qdrant: `docker run -p 6334:6333 qdrant/qdrant`\n- Restart the MCP server',
},
],
};
}
try {
const results = await deps.knowledgeIndexer.queryKnowledge(query, {
type: type as any,
projectId,
region,
limit,
});
SecurityMiddleware.auditLog('Knowledge query completed', {
tool: 'query_knowledge',
queryLength: query.length,
type,
resultCount: results.length,
});
if (results.length === 0) {
return {
content: [
{
type: 'text',
text: `š§ **Knowledge Query**: "${query}"\nš **Type**: ${type}\n\nā **No results found**\n\nTry:\n- Different keywords\n- Broader search terms\n- Different type filter\n- Check if data has been indexed`,
},
],
};
}
let response = `š§ **Knowledge Query**: "${query}"\n`;
response += `š **Type**: ${type}\n`;
response += `š **Found**: ${results.length} matching results\n\n`;
results.forEach((result, index) => {
const data = result.data as any;
if (!data) {
response += `**${index + 1}. [No Data]**\n`;
response += ` šÆ Confidence: ${(result.confidence * 100).toFixed(1)}%\n`;
response += ` ā ļø Warning: Result has null data\n`;
return;
}
response += `**${index + 1}. ${data.clusterName || data.jobId || 'Unknown'}**\n`;
response += ` šÆ Confidence: ${(result.confidence * 100).toFixed(1)}%\n`;
if (data.projectId) response += ` š¢ Project: ${data.projectId}\n`;
if (data.region) response += ` š Region: ${data.region}\n`;
if (data.jobType) response += ` š§ Job Type: ${data.jobType}\n`;
if (data.status) response += ` š Status: ${data.status}\n`;
response += '\n';
});
if (includeRawDocument && results.length > 0) {
response += `\nš **Raw Document Sample** (first result):\n`;
response += `\`\`\`json\n${JSON.stringify(results[0], null, 2)}\n\`\`\``;
}
return {
content: [
{
type: 'text',
text: response,
},
],
};
} catch (error) {
SecurityMiddleware.auditLog(
'Knowledge query failed',
{
tool: 'query_knowledge',
query,
type,
error: error instanceof Error ? error.message : 'Unknown error',
},
'error'
);
logger.error('MCP query_knowledge: Error:', error);
throw error;
}
}
