@gotza02/seq-thinking 1.1.0

package/src/index.ts

@@ -0,0 +1,118 @@
+/**
+ * MCP Sequential Thinking with Swarm Agents
+ * Main entry point for the library
+ * @module mcp-sequential-thinking
+ * @version 1.0.0
+ */
+
+// Export MCP Server
+export { MCPServer } from './mcp-server.js';
+
+// Export Sequential Thinking components
+export {
+  SessionManager,
+  ThoughtGraph,
+  MetaReasoningEngine,
+  SelfCorrectionEngine,
+  AdaptiveGranularityEngine,
+  ConfidenceCalibrator
+} from './sequential-thinking.js';
+
+// Export Swarm Coordinator components
+export {
+  SwarmCoordinator,
+  AgentRegistry,
+  MessageBroker,
+  TaskQueueManager,
+  TaskAssigner,
+  ConsensusEngine,
+  ConflictResolver
+} from './swarm-coordinator.js';
+
+// Export Agents
+export {
+  BaseAgent,
+  ReasonerAgent,
+  CriticAgent,
+  SynthesizerAgent,
+  MetaReasoningAgent
+} from './agents/index.js';
+
+// Export all types
+export * from './types/index.js';
+
+// Export utilities (avoiding Result conflict)
+export {
+  generateId,
+  generateUUID,
+  generateShortId,
+  clamp,
+  clampConfidence,
+  lerp,
+  roundTo,
+  extractWords,
+  calculateJaccardSimilarity,
+  calculateSimilarity,
+  calculateCosineSimilarity,
+  extractNGrams,
+  calculateNGramSimilarity,
+  calculateMean,
+  calculateWeightedAverage,
+  calculateMedian,
+  calculateStandardDeviation,
+  calculateVariance,
+  findMin,
+  findMax,
+  normalizeValues,
+  calculatePercentile,
+  formatDuration,
+  formatTimestamp,
+  formatRelativeTime,
+  deepClone,
+  deepCloneWithDates,
+  deepMerge,
+  deepEqual,
+  pick,
+  omit,
+  validateUUID,
+  isNonEmptyString,
+  isPositiveNumber,
+  isInRange,
+  isNonEmptyArray,
+  isPlainObject,
+  sanitizeString,
+  truncateString,
+  toCamelCase,
+  toPascalCase,
+  toSnakeCase,
+  toKebabCase,
+  escapeRegExp,
+  unique,
+  uniqueBy,
+  groupBy,
+  partition,
+  shuffle,
+  sample,
+  chunk,
+  flatten,
+  deepFlatten,
+  sleep,
+  createTimeoutPromise,
+  withTimeout,
+  retry,
+  debounce,
+  throttle,
+  tryCatch,
+  tryCatchAsync,
+  createError
+} from './utils/index.js';
+
+/**
+ * Library version
+ */
+export const VERSION = '1.0.0';
+
+/**
+ * Default export - MCPServer
+ */
+export { MCPServer as default } from './mcp-server.js';

package/src/mcp-server.ts

@@ -0,0 +1,387 @@
+/**
+ * MCP Server
+ */
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
+import { SessionManager, SelfCorrectionEngine, AdaptiveGranularityEngine } from './sequential-thinking.js';
+import { SwarmCoordinator } from './swarm-coordinator.js';
+import { ThoughtType, AgentType, ReasoningStrategy, CriticType, SynthesizerType, ConsensusAlgorithm, ConflictResolutionStrategy } from './types/index.js';
+import { ReasonerAgent, CriticAgent, SynthesizerAgent, MetaReasoningAgent } from './agents/index.js';
+const SEQUENTIAL_THINKING_TOOL = {
+    name: 'advanced_sequential_thinking',
+    description: 'Advanced sequential thinking with self-correction, parallel hypotheses, meta-reasoning',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            operation: {
+                type: 'string',
+                enum: [
+                    'initialize', 'add_thought', 'revise_thought', 'branch_thought',
+                    'merge_branches', 'prune_branch', 'self_correct', 'add_meta_thought',
+                    'get_thinking_path', 'get_branch_comparison', 'calibrate_confidence',
+                    'adjust_granularity', 'get_session_state', 'complete_session',
+                    'delegate_to_swarm'
+                ]
+            },
+            sessionId: { type: 'string' },
+            topic: { type: 'string' },
+            content: { type: 'string' },
+            thoughtType: { type: 'string', enum: Object.values(ThoughtType) },
+            thoughtId: { type: 'string' },
+            parentThoughtId: { type: 'string' },
+            branchId: { type: 'string' },
+            confidence: { type: 'number' },
+            revisionReason: { type: 'string' },
+            branchName: { type: 'string' },
+            branchDescription: { type: 'string' },
+            tags: { type: 'array', items: { type: 'string' } },
+            assumptions: { type: 'array', items: { type: 'string' } },
+            dependencies: { type: 'array', items: { type: 'string' } },
+            settings: { type: 'object' }
+        },
+        required: ['operation']
+    }
+};
+const SWARM_COORDINATION_TOOL = {
+    name: 'swarm_agent_coordination',
+    description: 'Coordinate multiple AI agents in a swarm',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            operation: {
+                type: 'string',
+                enum: [
+                    'register_agent', 'unregister_agent', 'list_agents', 'get_agent_status',
+                    'assign_task', 'get_task_status', 'reach_consensus', 'identify_conflict',
+                    'resolve_conflict', 'broadcast_message', 'send_direct_message', 'get_swarm_stats'
+                ]
+            },
+            name: { type: 'string' },
+            agentType: { type: 'string', enum: Object.values(AgentType) },
+            subtype: { type: 'string' },
+            capabilities: { type: 'array' },
+            agentId: { type: 'string' },
+            taskType: { type: 'string' },
+            description: { type: 'string' },
+            input: {},
+            requiredCapabilities: { type: 'array', items: { type: 'string' } },
+            strategy: { type: 'string', enum: ['round_robin', 'capability_based', 'load_balanced', 'adaptive'] },
+            proposal: {},
+            options: { type: 'array' },
+            algorithm: { type: 'string', enum: Object.values(ConsensusAlgorithm) },
+            participatingAgents: { type: 'array', items: { type: 'string' } },
+            minAgreementRatio: { type: 'number' },
+            maxRounds: { type: 'number' },
+            conflictType: { type: 'string' },
+            positions: { type: 'array' },
+            conflictId: { type: 'string' },
+            resolutionStrategy: { type: 'string', enum: Object.values(ConflictResolutionStrategy) },
+            resolverAgentId: { type: 'string' },
+            senderId: { type: 'string' },
+            recipientId: { type: 'string' },
+            payload: {},
+            taskId: { type: 'string' }
+        },
+        required: ['operation']
+    }
+};
+export class MCPServer {
+    server;
+    sessionManager;
+    swarmCoordinator;
+    registeredAgents = new Map();
+    private dataDir = './data';
+
+    constructor() {
+        this.server = new Server({ name: 'mcp-sequential-thinking', version: '1.0.0' }, { capabilities: { tools: {} } });
+        this.sessionManager = new SessionManager(this.dataDir);
+        this.swarmCoordinator = new SwarmCoordinator(this.dataDir);
+        this.setupHandlers();
+    }
+
+    async initialize() {
+        await this.sessionManager.initialize();
+        await this.swarmCoordinator.initialize();
+    }
+    setupHandlers() {
+        this.server.setRequestHandler(ListToolsRequestSchema, async () => ({
+            tools: [SEQUENTIAL_THINKING_TOOL, SWARM_COORDINATION_TOOL]
+        }));
+        this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+            const { name, arguments: args } = request.params;
+            try {
+                if (name === 'advanced_sequential_thinking') {
+                    const result = await this.handleSequentialThinking(args);
+                    return { content: [{ type: 'text', text: JSON.stringify(result, null, 2) }] };
+                }
+                if (name === 'swarm_agent_coordination') {
+                    const result = await this.handleSwarmCoordination(args);
+                    return { content: [{ type: 'text', text: JSON.stringify(result, null, 2) }] };
+                }
+                throw new Error(`Unknown tool: ${name}`);
+            }
+            catch (error) {
+                return {
+                    content: [{ type: 'text', text: JSON.stringify({ success: false, error: error instanceof Error ? error.message : String(error) }, null, 2) }],
+                    isError: true
+                };
+            }
+        });
+    }
+    async handleSequentialThinking(args: any) {
+        const operation = String(args.operation);
+        switch (operation) {
+            case 'initialize': {
+                const session = await this.sessionManager.createSession(String(args.topic), args.settings || {});
+                return { success: true, sessionId: session.id, topic: session.topic, status: session.status, settings: session.settings };
+            }
+            case 'add_thought': {
+                const { thought, metaNote, issues } = await this.sessionManager.addThought(String(args.sessionId), String(args.content), {
+                    thoughtType: args.thoughtType,
+                    parentThoughtId: args.parentThoughtId,
+                    branchId: args.branchId,
+                    confidence: args.confidence,
+                    tags: args.tags,
+                    assumptions: args.assumptions,
+                    dependencies: args.dependencies
+                });
+                return {
+                    success: true,
+                    thought: { id: thought.id, stepNumber: thought.stepNumber, thoughtType: thought.thoughtType, confidence: thought.confidence, branchId: thought.branchId },
+                    metaNote: metaNote ? { id: metaNote.id, type: metaNote.observationType, content: metaNote.content, triggeredActions: metaNote.triggeredActions } : undefined,
+                    issuesDetected: issues
+                };
+            }
+            case 'revise_thought': {
+                const graph = this.sessionManager.getThoughtGraph();
+                const revision = graph.reviseThought(String(args.thoughtId), String(args.newContent), String(args.revisionReason));
+                return { success: true, revision: { id: revision.id, originalThoughtId: args.thoughtId, content: revision.content, revisionReason: args.revisionReason } };
+            }
+            case 'branch_thought': {
+                const graph = this.sessionManager.getThoughtGraph();
+                const { branch, thought } = graph.createBranch(String(args.sessionId), String(args.parentThoughtId), String(args.branchName), String(args.branchDescription), String(args.initialContent));
+                return { success: true, branch: { id: branch.id, name: branch.name, description: branch.description, parentBranchId: branch.parentBranchId }, rootThought: { id: thought.id, content: thought.content } };
+            }
+            case 'merge_branches': {
+                const graph = this.sessionManager.getThoughtGraph();
+                const mergeThought = graph.mergeBranches(String(args.targetBranchId), String(args.sourceBranchId), String(args.mergeContent));
+                return { success: true, mergeThought: { id: mergeThought.id, content: mergeThought.content, mergedBranches: [args.sourceBranchId] } };
+            }
+            case 'prune_branch': {
+                const graph = this.sessionManager.getThoughtGraph();
+                graph.pruneBranch(String(args.branchId), String(args.reason));
+                return { success: true, message: `Branch ${args.branchId} pruned: ${args.reason}` };
+            }
+            case 'self_correct': {
+                const session = this.sessionManager.getSession(String(args.sessionId));
+                if (!session)
+                    throw new Error(`Session not found: ${args.sessionId}`);
+                const graph = this.sessionManager.getThoughtGraph();
+                const thought = graph.getThoughts().get(String(args.thoughtId));
+                if (!thought)
+                    throw new Error(`Thought not found: ${args.thoughtId}`);
+                const correctionEngine = new SelfCorrectionEngine();
+                const issues = correctionEngine.detectIssues(thought, graph);
+                return { success: true, issuesDetected: issues, recommendations: issues.map((i: any) => i.suggestion) };
+            }
+            case 'add_meta_thought': {
+                const graph = this.sessionManager.getThoughtGraph();
+                const metaEngine = this.sessionManager.getMetaEngine();
+                const metaNote = metaEngine.generateMetaObservation(String(args.thoughtId), graph);
+                return { success: true, metaNote: metaNote ? { id: metaNote.id, type: metaNote.observationType, content: metaNote.content, triggeredActions: metaNote.triggeredActions } : null };
+            }
+            case 'get_thinking_path': {
+                const graph = this.sessionManager.getThoughtGraph();
+                const path = graph.getThinkingPath(String(args.thoughtId));
+                return { success: true, path: path.map((t: any) => ({ id: t.id, stepNumber: t.stepNumber, thoughtType: t.thoughtType, content: t.content.substring(0, 200), confidence: t.confidence.overall })) };
+            }
+            case 'get_branch_comparison': {
+                const graph = this.sessionManager.getThoughtGraph();
+                const branchIds = Array.isArray(args.branchIds) ? args.branchIds : [];
+                const comparison = graph.getBranchComparison(branchIds);
+                return { success: true, comparison };
+            }
+            case 'calibrate_confidence': {
+                const graph = this.sessionManager.getThoughtGraph();
+                const thought = graph.getThoughts().get(String(args.thoughtId));
+                if (!thought)
+                    throw new Error(`Thought not found: ${args.thoughtId}`);
+                return { success: true, originalConfidence: thought.confidence, message: 'Confidence calibration would update based on feedback' };
+            }
+            case 'adjust_granularity': {
+                const session = this.sessionManager.getSession(String(args.sessionId));
+                if (!session)
+                    throw new Error(`Session not found: ${args.sessionId}`);
+                const engine = new AdaptiveGranularityEngine();
+                const granularity = engine.calculateGranularity(session.topic, session.context, session.thoughts.size);
+                return { success: true, recommendedDetail: granularity.detail, reasoning: granularity.reasoning };
+            }
+            case 'get_session_state': {
+                const session = this.sessionManager.getSession(String(args.sessionId));
+                if (!session)
+                    throw new Error(`Session not found: ${args.sessionId}`);
+                const graph = this.sessionManager.getThoughtGraph();
+                return {
+                    success: true,
+                    session: { id: session.id, topic: session.topic, status: session.status, totalThoughts: session.metadata.totalThoughts, totalBranches: session.metadata.totalBranches, lastActivity: session.metadata.lastActivityAt },
+                    branches: Array.from(graph.getBranches().values()).map(b => ({ id: b.id, name: b.name, status: b.status, thoughtCount: b.thoughtIds.length, confidence: b.confidence })),
+                    recentThoughts: Array.from(session.thoughts.values()).slice(-5).map((t: any) => ({ id: t.id, stepNumber: t.stepNumber, thoughtType: t.thoughtType, content: t.content.substring(0, 100) }))
+                };
+            }
+            case 'complete_session': {
+                await this.sessionManager.completeSession(String(args.sessionId));
+                return { success: true, message: `Session ${args.sessionId} completed` };
+            }
+            case 'delegate_to_swarm': {
+                const session = this.sessionManager.getSession(String(args.sessionId));
+                if (!session) throw new Error(`Session not found: ${args.sessionId}`);
+                
+                const graph = this.sessionManager.getThoughtGraph();
+                const thought = graph.getThoughts().get(String(args.thoughtId));
+                if (!thought) throw new Error(`Thought not found: ${args.thoughtId}`);
+
+                const taskType = String(args.taskType || 'reasoning');
+                const { task, assignedAgent } = this.swarmCoordinator.submitTask(
+                    taskType,
+                    `Analyze and expand on this thought: ${thought.content}`,
+                    { thought: thought.content, context: session.context },
+                    { requiredCapabilities: args.requiredCapabilities || [], minConfidence: 0.6, maxTimeMs: 30000 }
+                );
+
+                if (!assignedAgent) {
+                    return { success: false, error: 'No suitable agent available in swarm' };
+                }
+
+                const result = await this.swarmCoordinator.executeTask(task.id);
+                
+                if (result.success && result.output) {
+                    const swarmOutput = typeof result.output === 'string' ? result.output : JSON.stringify(result.output, null, 2);
+                    const newThought = await this.sessionManager.addThought(session.id, swarmOutput, {
+                        thoughtType: ThoughtType.SYNTHESIS,
+                        parentThoughtId: thought.id,
+                        branchId: thought.branchId,
+                        confidence: result.confidence,
+                        tags: ['swarm-result', assignedAgent.config.name]
+                    });
+                    return { success: true, swarmResult: result, newThought: newThought.thought };
+                } else {
+                    return { success: false, error: 'Swarm execution failed', details: result };
+                }
+            }
+            default:
+                throw new Error(`Unknown operation: ${operation}`);
+        }
+    }
+    async handleSwarmCoordination(args: any) {
+        const operation = String(args.operation);
+        switch (operation) {
+            case 'register_agent': {
+                const agentType = String(args.agentType);
+                let agent;
+                switch (agentType) {
+                    case AgentType.REASONER:
+                        agent = new ReasonerAgent({ name: String(args.name), strategy: args.subtype || ReasoningStrategy.CHAIN_OF_THOUGHT });
+                        break;
+                    case AgentType.CRITIC:
+                        agent = new CriticAgent({ name: String(args.name), criticType: args.subtype || CriticType.LOGICAL });
+                        break;
+                    case AgentType.SYNTHESIZER:
+                        agent = new SynthesizerAgent({ name: String(args.name), synthesizerType: args.subtype || SynthesizerType.CONSENSUS });
+                        break;
+                    case AgentType.META_REASONING:
+                        agent = new MetaReasoningAgent({ name: String(args.name) });
+                        break;
+                    default:
+                        throw new Error(`Unsupported agent type: ${agentType}`);
+                }
+                this.registeredAgents.set(agent.id, agent);
+                await this.swarmCoordinator.registerAgent(agent);
+                return { success: true, agentId: agent.id, name: agent.config.name, type: agent.config.type, capabilities: agent.config.capabilities.map(c => c.name) };
+            }
+            case 'unregister_agent': {
+                const success = await this.swarmCoordinator.unregisterAgent(String(args.agentId));
+                this.registeredAgents.delete(String(args.agentId));
+                return { success, message: success ? `Agent ${args.agentId} unregistered` : 'Agent not found' };
+            }
+            case 'list_agents': {
+                const agents = this.swarmCoordinator.registry.getAllAgents();
+                return { success: true, agents: agents.map(a => ({ id: a.config.id, name: a.config.name, type: a.config.type, subtype: a.config.subtype, status: a.status, capabilities: a.config.capabilities.map(c => c.name) })) };
+            }
+            case 'get_agent_status': {
+                const agent = this.swarmCoordinator.registry.getAgent(String(args.agentId));
+                if (!agent)
+                    throw new Error(`Agent not found: ${args.agentId}`);
+                return { success: true, agent: { id: agent.config.id, name: agent.config.name, type: agent.config.type, status: agent.status, currentTasks: agent.currentTasks, performanceMetrics: { tasksCompleted: agent.performanceHistory.length, averageQuality: agent.performanceHistory.length > 0 ? agent.performanceHistory.reduce((sum, p) => sum + p.quality, 0) / agent.performanceHistory.length : 0, averageTimeMs: agent.performanceHistory.length > 0 ? agent.performanceHistory.reduce((sum, p) => sum + p.timeMs, 0) / agent.performanceHistory.length : 0 } } };
+            }
+            case 'assign_task': {
+                const { task, assignedAgent } = this.swarmCoordinator.submitTask(String(args.taskType), String(args.description), args.input, { requiredCapabilities: Array.isArray(args.requiredCapabilities) ? args.requiredCapabilities : [], minConfidence: 0.6, maxTimeMs: 30000 }, args.strategy || 'adaptive');
+                
+                let result = null;
+                if (assignedAgent) {
+                    result = await this.swarmCoordinator.executeTask(task.id);
+                }
+
+                return { 
+                    success: true, 
+                    task: { 
+                        id: task.id, 
+                        type: task.type, 
+                        description: task.description, 
+                        status: task.status, 
+                        assignedAgentId: task.assignedAgentId 
+                    }, 
+                    assignedAgent: assignedAgent ? { 
+                        id: assignedAgent.config.id, 
+                        name: assignedAgent.config.name, 
+                        type: assignedAgent.config.type 
+                    } : null,
+                    result
+                };
+            }
+            case 'get_task_status': {
+                const task = this.swarmCoordinator.taskQueue.getTask(String(args.taskId));
+                if (!task)
+                    throw new Error(`Task not found: ${args.taskId}`);
+                return { success: true, task: { id: task.id, type: task.type, description: task.description, status: task.status, assignedAgentId: task.assignedAgentId, result: task.result, createdAt: task.metadata.createdAt, completedAt: task.metadata.completedAt } };
+            }
+            case 'reach_consensus': {
+                const result = await this.swarmCoordinator.reachConsensus(args.proposal, Array.isArray(args.options) ? args.options : [], args.algorithm || ConsensusAlgorithm.WEIGHTED_VOTE, Array.isArray(args.participatingAgents) ? args.participatingAgents : undefined, typeof args.minAgreementRatio === 'number' ? args.minAgreementRatio : 0.67, typeof args.maxRounds === 'number' ? args.maxRounds : 3);
+                return { success: true, consensus: { proposalId: result.proposalId, algorithm: result.algorithm, winningOption: result.winningOption, confidence: result.confidence, agreementRatio: result.agreementRatio, rounds: result.metadata.rounds, dissentingOpinions: result.dissentingOpinions } };
+            }
+            case 'identify_conflict': {
+                const conflict = this.swarmCoordinator.identifyConflict(String(args.conflictType), String(args.description), Array.isArray(args.positions) ? args.positions : []);
+                return { success: true, conflict: { id: conflict.id, type: conflict.type, description: conflict.description, involvedAgents: conflict.involvedAgents, positions: conflict.conflictingPositions, status: conflict.status } };
+            }
+            case 'resolve_conflict': {
+                const { conflict, resolution } = this.swarmCoordinator.resolveConflict(String(args.conflictId), args.resolutionStrategy || ConflictResolutionStrategy.VOTING, args.resolverAgentId);
+                return { success: true, conflict: { id: conflict.id, status: conflict.status, resolutionStrategy: conflict.metadata.resolutionStrategy }, resolution };
+            }
+            case 'broadcast_message': {
+                const message = this.swarmCoordinator.broadcastMessage(String(args.senderId), args.payload);
+                return { success: true, message: { id: message.id, type: message.type, senderId: message.senderId, timestamp: message.timestamp } };
+            }
+            case 'send_direct_message': {
+                const message = this.swarmCoordinator.sendDirectMessage(String(args.senderId), String(args.recipientId), args.payload);
+                return { success: true, message: { id: message.id, type: message.type, senderId: message.senderId, recipientId: message.recipientId, timestamp: message.timestamp } };
+            }
+            case 'get_swarm_stats': {
+                const stats = this.swarmCoordinator.getSwarmStats();
+                return { success: true, stats };
+            }
+            default:
+                throw new Error(`Unknown operation: ${operation}`);
+        }
+    }
+    async start() {
+        const transport = new StdioServerTransport();
+        await this.server.connect(transport);
+        console.error('MCP Sequential Thinking Server running on stdio');
+    }
+}
+const isMainModule = import.meta.url === `file://${process.argv[1]}`;
+if (isMainModule) {
+    const server = new MCPServer();
+    server.start().catch(console.error);
+}
+//# sourceMappingURL=mcp-server.js.map