@juspay/neurolink 9.31.2 → 9.32.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +12 -0
- package/dist/auth/AuthProviderFactory.d.ts +71 -0
- package/dist/auth/AuthProviderFactory.js +111 -0
- package/dist/auth/AuthProviderRegistry.d.ts +33 -0
- package/dist/auth/AuthProviderRegistry.js +190 -0
- package/dist/auth/RequestContext.d.ts +23 -0
- package/dist/auth/RequestContext.js +78 -0
- package/dist/auth/anthropicOAuth.js +1 -1
- package/dist/auth/authContext.d.ts +198 -0
- package/dist/auth/authContext.js +314 -0
- package/dist/auth/errors.d.ts +63 -0
- package/dist/auth/errors.js +39 -0
- package/dist/auth/index.d.ts +20 -8
- package/dist/auth/index.js +35 -7
- package/dist/auth/middleware/AuthMiddleware.d.ts +181 -0
- package/dist/auth/middleware/AuthMiddleware.js +519 -0
- package/dist/auth/middleware/rateLimitByUser.d.ts +282 -0
- package/dist/auth/middleware/rateLimitByUser.js +554 -0
- package/dist/auth/providers/BaseAuthProvider.d.ts +259 -0
- package/dist/auth/providers/BaseAuthProvider.js +723 -0
- package/dist/auth/providers/CognitoProvider.d.ts +61 -0
- package/dist/auth/providers/CognitoProvider.js +304 -0
- package/dist/auth/providers/KeycloakProvider.d.ts +61 -0
- package/dist/auth/providers/KeycloakProvider.js +393 -0
- package/dist/auth/providers/auth0.d.ts +59 -0
- package/dist/auth/providers/auth0.js +274 -0
- package/dist/auth/providers/betterAuth.d.ts +51 -0
- package/dist/auth/providers/betterAuth.js +182 -0
- package/dist/auth/providers/clerk.d.ts +65 -0
- package/dist/auth/providers/clerk.js +317 -0
- package/dist/auth/providers/custom.d.ts +64 -0
- package/dist/auth/providers/custom.js +112 -0
- package/dist/auth/providers/firebase.d.ts +63 -0
- package/dist/auth/providers/firebase.js +226 -0
- package/dist/auth/providers/jwt.d.ts +68 -0
- package/dist/auth/providers/jwt.js +212 -0
- package/dist/auth/providers/oauth2.d.ts +73 -0
- package/dist/auth/providers/oauth2.js +303 -0
- package/dist/auth/providers/supabase.d.ts +63 -0
- package/dist/auth/providers/supabase.js +259 -0
- package/dist/auth/providers/workos.d.ts +61 -0
- package/dist/auth/providers/workos.js +284 -0
- package/dist/auth/serverBridge.d.ts +14 -0
- package/dist/auth/serverBridge.js +25 -0
- package/dist/auth/sessionManager.d.ts +142 -0
- package/dist/auth/sessionManager.js +437 -0
- package/dist/cli/commands/authProviders.d.ts +43 -0
- package/dist/cli/commands/authProviders.js +399 -0
- package/dist/cli/commands/proxy.js +18 -5
- package/dist/cli/factories/authCommandFactory.d.ts +23 -5
- package/dist/cli/factories/authCommandFactory.js +108 -5
- package/dist/cli/parser.js +1 -1
- package/dist/client/aiSdkAdapter.js +1 -1
- package/dist/client/index.js +138 -477
- package/dist/core/factory.js +0 -1
- package/dist/core/infrastructure/baseRegistry.d.ts +3 -1
- package/dist/core/infrastructure/baseRegistry.js +5 -1
- package/dist/core/redisConversationMemoryManager.js +1 -1
- package/dist/features/ppt/slideGenerator.js +0 -1
- package/dist/features/ppt/utils.js +0 -1
- package/dist/index.d.ts +1 -0
- package/dist/index.js +25 -0
- package/dist/lib/auth/AuthProviderFactory.d.ts +71 -0
- package/dist/lib/auth/AuthProviderFactory.js +112 -0
- package/dist/lib/auth/AuthProviderRegistry.d.ts +33 -0
- package/dist/lib/auth/AuthProviderRegistry.js +191 -0
- package/dist/lib/auth/RequestContext.d.ts +23 -0
- package/dist/lib/auth/RequestContext.js +79 -0
- package/dist/lib/auth/authContext.d.ts +198 -0
- package/dist/lib/auth/authContext.js +315 -0
- package/dist/lib/auth/errors.d.ts +63 -0
- package/dist/lib/auth/errors.js +40 -0
- package/dist/lib/auth/index.d.ts +20 -8
- package/dist/lib/auth/index.js +35 -7
- package/dist/lib/auth/middleware/AuthMiddleware.d.ts +181 -0
- package/dist/lib/auth/middleware/AuthMiddleware.js +520 -0
- package/dist/lib/auth/middleware/rateLimitByUser.d.ts +282 -0
- package/dist/lib/auth/middleware/rateLimitByUser.js +555 -0
- package/dist/lib/auth/providers/BaseAuthProvider.d.ts +259 -0
- package/dist/lib/auth/providers/BaseAuthProvider.js +724 -0
- package/dist/lib/auth/providers/CognitoProvider.d.ts +61 -0
- package/dist/lib/auth/providers/CognitoProvider.js +305 -0
- package/dist/lib/auth/providers/KeycloakProvider.d.ts +61 -0
- package/dist/lib/auth/providers/KeycloakProvider.js +394 -0
- package/dist/lib/auth/providers/auth0.d.ts +59 -0
- package/dist/lib/auth/providers/auth0.js +275 -0
- package/dist/lib/auth/providers/betterAuth.d.ts +51 -0
- package/dist/lib/auth/providers/betterAuth.js +183 -0
- package/dist/lib/auth/providers/clerk.d.ts +65 -0
- package/dist/lib/auth/providers/clerk.js +318 -0
- package/dist/lib/auth/providers/custom.d.ts +64 -0
- package/dist/lib/auth/providers/custom.js +113 -0
- package/dist/lib/auth/providers/firebase.d.ts +63 -0
- package/dist/lib/auth/providers/firebase.js +227 -0
- package/dist/lib/auth/providers/jwt.d.ts +68 -0
- package/dist/lib/auth/providers/jwt.js +213 -0
- package/dist/lib/auth/providers/oauth2.d.ts +73 -0
- package/dist/lib/auth/providers/oauth2.js +304 -0
- package/dist/lib/auth/providers/supabase.d.ts +63 -0
- package/dist/lib/auth/providers/supabase.js +260 -0
- package/dist/lib/auth/providers/workos.d.ts +61 -0
- package/dist/lib/auth/providers/workos.js +285 -0
- package/dist/lib/auth/serverBridge.d.ts +14 -0
- package/dist/lib/auth/serverBridge.js +26 -0
- package/dist/lib/auth/sessionManager.d.ts +142 -0
- package/dist/lib/auth/sessionManager.js +438 -0
- package/dist/lib/core/infrastructure/baseRegistry.d.ts +3 -1
- package/dist/lib/core/infrastructure/baseRegistry.js +5 -1
- package/dist/lib/index.d.ts +1 -0
- package/dist/lib/index.js +25 -0
- package/dist/lib/mcp/toolRegistry.js +11 -1
- package/dist/lib/neurolink.d.ts +42 -1
- package/dist/lib/neurolink.js +218 -0
- package/dist/lib/rag/ChunkerRegistry.js +2 -2
- package/dist/lib/rag/metadata/MetadataExtractorRegistry.js +2 -2
- package/dist/lib/rag/reranker/RerankerRegistry.js +2 -2
- package/dist/lib/server/routes/agentRoutes.js +20 -2
- package/dist/lib/server/routes/claudeProxyRoutes.js +45 -9
- package/dist/lib/types/authTypes.d.ts +937 -1
- package/dist/lib/types/authTypes.js +2 -1
- package/dist/lib/types/configTypes.d.ts +46 -0
- package/dist/lib/types/generateTypes.d.ts +6 -0
- package/dist/lib/types/index.d.ts +1 -0
- package/dist/lib/types/streamTypes.d.ts +6 -0
- package/dist/mcp/elicitationProtocol.js +1 -1
- package/dist/mcp/servers/agent/directToolsServer.js +0 -1
- package/dist/mcp/toolRegistry.js +11 -1
- package/dist/neurolink.d.ts +42 -1
- package/dist/neurolink.js +218 -0
- package/dist/providers/azureOpenai.js +1 -1
- package/dist/providers/huggingFace.js +0 -1
- package/dist/providers/openaiCompatible.js +0 -1
- package/dist/rag/ChunkerRegistry.js +2 -2
- package/dist/rag/metadata/MetadataExtractorRegistry.js +2 -2
- package/dist/rag/reranker/RerankerRegistry.js +2 -2
- package/dist/sdk/toolRegistration.js +0 -1
- package/dist/server/openapi/generator.js +1 -1
- package/dist/server/routes/agentRoutes.js +20 -2
- package/dist/server/routes/claudeProxyRoutes.js +45 -9
- package/dist/types/authTypes.d.ts +937 -1
- package/dist/types/authTypes.js +2 -1
- package/dist/types/configTypes.d.ts +46 -0
- package/dist/types/configTypes.js +0 -5
- package/dist/types/generateTypes.d.ts +6 -0
- package/dist/types/index.d.ts +1 -0
- package/dist/types/modelTypes.js +0 -1
- package/dist/types/streamTypes.d.ts +6 -0
- package/dist/types/tools.js +0 -1
- package/dist/types/typeAliases.js +0 -1
- package/dist/types/utilities.js +1 -1
- package/dist/types/workflowTypes.js +0 -1
- package/dist/utils/providerRetry.js +0 -1
- package/dist/utils/providerUtils.js +0 -1
- package/package.json +3 -2
- package/dist/client/adapters/providerImageAdapter.js +0 -588
- package/dist/client/adapters/tts/googleTTSHandler.js +0 -344
- package/dist/client/adapters/video/directorPipeline.js +0 -516
- package/dist/client/adapters/video/ffmpegAdapter.js +0 -206
- package/dist/client/adapters/video/frameExtractor.js +0 -143
- package/dist/client/adapters/video/vertexVideoHandler.js +0 -763
- package/dist/client/adapters/video/videoAnalyzer.js +0 -238
- package/dist/client/adapters/video/videoMerger.js +0 -171
- package/dist/client/agent/directTools.js +0 -840
- package/dist/client/auth/anthropicOAuth.js +0 -974
- package/dist/client/auth/tokenStore.js +0 -799
- package/dist/client/client/aiSdkAdapter.js +0 -487
- package/dist/client/client/auth.js +0 -473
- package/dist/client/client/errors.js +0 -552
- package/dist/client/client/httpClient.js +0 -837
- package/dist/client/client/index.js +0 -172
- package/dist/client/client/interceptors.js +0 -601
- package/dist/client/client/sseClient.js +0 -545
- package/dist/client/client/streamingClient.js +0 -917
- package/dist/client/client/wsClient.js +0 -369
- package/dist/client/config/configManager.js +0 -303
- package/dist/client/config/conversationMemory.js +0 -86
- package/dist/client/config/taskClassificationConfig.js +0 -148
- package/dist/client/constants/contextWindows.js +0 -295
- package/dist/client/constants/enums.js +0 -853
- package/dist/client/constants/index.js +0 -207
- package/dist/client/constants/performance.js +0 -389
- package/dist/client/constants/retry.js +0 -266
- package/dist/client/constants/timeouts.js +0 -182
- package/dist/client/constants/tokens.js +0 -380
- package/dist/client/constants/videoErrors.js +0 -46
- package/dist/client/context/budgetChecker.js +0 -98
- package/dist/client/context/contextCompactor.js +0 -205
- package/dist/client/context/emergencyTruncation.js +0 -88
- package/dist/client/context/errorDetection.js +0 -171
- package/dist/client/context/errors.js +0 -21
- package/dist/client/context/fileTokenBudget.js +0 -127
- package/dist/client/context/prompts/summarizationPrompt.js +0 -117
- package/dist/client/context/stages/fileReadDeduplicator.js +0 -66
- package/dist/client/context/stages/slidingWindowTruncator.js +0 -190
- package/dist/client/context/stages/structuredSummarizer.js +0 -99
- package/dist/client/context/stages/toolOutputPruner.js +0 -52
- package/dist/client/context/summarizationEngine.js +0 -136
- package/dist/client/context/toolOutputLimits.js +0 -78
- package/dist/client/context/toolPairRepair.js +0 -66
- package/dist/client/core/analytics.js +0 -88
- package/dist/client/core/baseProvider.js +0 -1385
- package/dist/client/core/constants.js +0 -140
- package/dist/client/core/conversationMemoryFactory.js +0 -141
- package/dist/client/core/conversationMemoryInitializer.js +0 -128
- package/dist/client/core/conversationMemoryManager.js +0 -344
- package/dist/client/core/dynamicModels.js +0 -358
- package/dist/client/core/evaluation.js +0 -309
- package/dist/client/core/evaluationProviders.js +0 -248
- package/dist/client/core/factory.js +0 -412
- package/dist/client/core/infrastructure/baseError.js +0 -22
- package/dist/client/core/infrastructure/baseFactory.js +0 -54
- package/dist/client/core/infrastructure/baseRegistry.js +0 -49
- package/dist/client/core/infrastructure/index.js +0 -5
- package/dist/client/core/infrastructure/retry.js +0 -20
- package/dist/client/core/infrastructure/typedEventEmitter.js +0 -23
- package/dist/client/core/modelConfiguration.js +0 -851
- package/dist/client/core/modules/GenerationHandler.js +0 -588
- package/dist/client/core/modules/MessageBuilder.js +0 -273
- package/dist/client/core/modules/StreamHandler.js +0 -185
- package/dist/client/core/modules/TelemetryHandler.js +0 -203
- package/dist/client/core/modules/ToolsManager.js +0 -499
- package/dist/client/core/modules/Utilities.js +0 -331
- package/dist/client/core/redisConversationMemoryManager.js +0 -1435
- package/dist/client/core/streamAnalytics.js +0 -131
- package/dist/client/evaluation/contextBuilder.js +0 -134
- package/dist/client/evaluation/index.js +0 -61
- package/dist/client/evaluation/prompts.js +0 -73
- package/dist/client/evaluation/ragasEvaluator.js +0 -110
- package/dist/client/evaluation/retryManager.js +0 -78
- package/dist/client/evaluation/scoring.js +0 -61
- package/dist/client/factories/providerFactory.js +0 -166
- package/dist/client/factories/providerRegistry.js +0 -166
- package/dist/client/features/ppt/constants.js +0 -896
- package/dist/client/features/ppt/contentPlanner.js +0 -529
- package/dist/client/features/ppt/presentationOrchestrator.js +0 -236
- package/dist/client/features/ppt/slideGenerator.js +0 -532
- package/dist/client/features/ppt/slideRenderers.js +0 -2383
- package/dist/client/features/ppt/slideTypeInference.js +0 -405
- package/dist/client/features/ppt/types.js +0 -13
- package/dist/client/features/ppt/utils.js +0 -443
- package/dist/client/files/fileReferenceRegistry.js +0 -1543
- package/dist/client/files/fileTools.js +0 -450
- package/dist/client/files/streamingReader.js +0 -321
- package/dist/client/files/types.js +0 -23
- package/dist/client/hitl/hitlErrors.js +0 -54
- package/dist/client/hitl/hitlManager.js +0 -460
- package/dist/client/mcp/agentExposure.js +0 -356
- package/dist/client/mcp/auth/index.js +0 -11
- package/dist/client/mcp/auth/oauthClientProvider.js +0 -325
- package/dist/client/mcp/auth/tokenStorage.js +0 -134
- package/dist/client/mcp/batching/index.js +0 -10
- package/dist/client/mcp/batching/requestBatcher.js +0 -441
- package/dist/client/mcp/caching/index.js +0 -10
- package/dist/client/mcp/caching/toolCache.js +0 -433
- package/dist/client/mcp/elicitation/elicitationManager.js +0 -376
- package/dist/client/mcp/elicitation/index.js +0 -11
- package/dist/client/mcp/elicitation/types.js +0 -10
- package/dist/client/mcp/elicitationProtocol.js +0 -375
- package/dist/client/mcp/enhancedToolDiscovery.js +0 -481
- package/dist/client/mcp/externalServerManager.js +0 -1478
- package/dist/client/mcp/factory.js +0 -161
- package/dist/client/mcp/flexibleToolValidator.js +0 -161
- package/dist/client/mcp/httpRateLimiter.js +0 -391
- package/dist/client/mcp/httpRetryHandler.js +0 -178
- package/dist/client/mcp/index.js +0 -74
- package/dist/client/mcp/mcpCircuitBreaker.js +0 -427
- package/dist/client/mcp/mcpClientFactory.js +0 -708
- package/dist/client/mcp/mcpRegistryClient.js +0 -488
- package/dist/client/mcp/mcpServerBase.js +0 -373
- package/dist/client/mcp/multiServerManager.js +0 -579
- package/dist/client/mcp/registry.js +0 -158
- package/dist/client/mcp/routing/index.js +0 -10
- package/dist/client/mcp/routing/toolRouter.js +0 -416
- package/dist/client/mcp/serverCapabilities.js +0 -502
- package/dist/client/mcp/servers/agent/directToolsServer.js +0 -150
- package/dist/client/mcp/toolAnnotations.js +0 -239
- package/dist/client/mcp/toolConverter.js +0 -258
- package/dist/client/mcp/toolDiscoveryService.js +0 -798
- package/dist/client/mcp/toolIntegration.js +0 -334
- package/dist/client/mcp/toolRegistry.js +0 -719
- package/dist/client/memory/hippocampusInitializer.js +0 -19
- package/dist/client/memory/memoryRetrievalTools.js +0 -166
- package/dist/client/middleware/builtin/analytics.js +0 -132
- package/dist/client/middleware/builtin/autoEvaluation.js +0 -203
- package/dist/client/middleware/builtin/guardrails.js +0 -109
- package/dist/client/middleware/builtin/lifecycle.js +0 -168
- package/dist/client/middleware/factory.js +0 -327
- package/dist/client/middleware/registry.js +0 -295
- package/dist/client/middleware/utils/guardrailsUtils.js +0 -396
- package/dist/client/models/anthropicModels.js +0 -527
- package/dist/client/neurolink.js +0 -8015
- package/dist/client/observability/exporterRegistry.js +0 -413
- package/dist/client/observability/exporters/arizeExporter.js +0 -138
- package/dist/client/observability/exporters/baseExporter.js +0 -190
- package/dist/client/observability/exporters/braintrustExporter.js +0 -154
- package/dist/client/observability/exporters/datadogExporter.js +0 -196
- package/dist/client/observability/exporters/laminarExporter.js +0 -302
- package/dist/client/observability/exporters/langfuseExporter.js +0 -209
- package/dist/client/observability/exporters/langsmithExporter.js +0 -143
- package/dist/client/observability/exporters/otelExporter.js +0 -164
- package/dist/client/observability/exporters/posthogExporter.js +0 -287
- package/dist/client/observability/exporters/sentryExporter.js +0 -165
- package/dist/client/observability/index.js +0 -31
- package/dist/client/observability/metricsAggregator.js +0 -556
- package/dist/client/observability/otelBridge.js +0 -131
- package/dist/client/observability/retryPolicy.js +0 -383
- package/dist/client/observability/sampling/samplers.js +0 -216
- package/dist/client/observability/spanProcessor.js +0 -303
- package/dist/client/observability/tokenTracker.js +0 -413
- package/dist/client/observability/types/exporterTypes.js +0 -5
- package/dist/client/observability/types/index.js +0 -4
- package/dist/client/observability/types/spanTypes.js +0 -92
- package/dist/client/observability/utils/safeMetadata.js +0 -25
- package/dist/client/observability/utils/spanSerializer.js +0 -292
- package/dist/client/processors/archive/ArchiveProcessor.js +0 -1308
- package/dist/client/processors/base/BaseFileProcessor.js +0 -614
- package/dist/client/processors/base/types.js +0 -82
- package/dist/client/processors/config/fileTypes.js +0 -520
- package/dist/client/processors/config/index.js +0 -92
- package/dist/client/processors/config/languageMap.js +0 -410
- package/dist/client/processors/config/mimeTypes.js +0 -363
- package/dist/client/processors/config/sizeLimits.js +0 -258
- package/dist/client/processors/document/ExcelProcessor.js +0 -590
- package/dist/client/processors/document/OpenDocumentProcessor.js +0 -212
- package/dist/client/processors/document/PptxProcessor.js +0 -157
- package/dist/client/processors/document/RtfProcessor.js +0 -361
- package/dist/client/processors/document/WordProcessor.js +0 -353
- package/dist/client/processors/errors/FileErrorCode.js +0 -255
- package/dist/client/processors/errors/errorHelpers.js +0 -386
- package/dist/client/processors/errors/errorSerializer.js +0 -507
- package/dist/client/processors/errors/index.js +0 -49
- package/dist/client/processors/markup/SvgProcessor.js +0 -240
- package/dist/client/processors/media/AudioProcessor.js +0 -707
- package/dist/client/processors/media/VideoProcessor.js +0 -1045
- package/dist/client/providers/amazonBedrock.js +0 -1512
- package/dist/client/providers/amazonSagemaker.js +0 -162
- package/dist/client/providers/anthropic.js +0 -831
- package/dist/client/providers/azureOpenai.js +0 -143
- package/dist/client/providers/googleAiStudio.js +0 -1200
- package/dist/client/providers/googleNativeGemini3.js +0 -543
- package/dist/client/providers/googleVertex.js +0 -2936
- package/dist/client/providers/huggingFace.js +0 -315
- package/dist/client/providers/litellm.js +0 -488
- package/dist/client/providers/mistral.js +0 -157
- package/dist/client/providers/ollama.js +0 -1579
- package/dist/client/providers/openAI.js +0 -627
- package/dist/client/providers/openRouter.js +0 -543
- package/dist/client/providers/openaiCompatible.js +0 -290
- package/dist/client/providers/providerTypeUtils.js +0 -46
- package/dist/client/providers/sagemaker/adaptive-semaphore.js +0 -215
- package/dist/client/providers/sagemaker/client.js +0 -472
- package/dist/client/providers/sagemaker/config.js +0 -317
- package/dist/client/providers/sagemaker/detection.js +0 -606
- package/dist/client/providers/sagemaker/error-constants.js +0 -227
- package/dist/client/providers/sagemaker/errors.js +0 -299
- package/dist/client/providers/sagemaker/language-model.js +0 -775
- package/dist/client/providers/sagemaker/parsers.js +0 -634
- package/dist/client/providers/sagemaker/streaming.js +0 -331
- package/dist/client/providers/sagemaker/structured-parser.js +0 -625
- package/dist/client/proxy/accountQuota.js +0 -162
- package/dist/client/proxy/claudeFormat.js +0 -595
- package/dist/client/proxy/modelRouter.js +0 -29
- package/dist/client/proxy/oauthFetch.js +0 -367
- package/dist/client/proxy/proxyFetch.js +0 -586
- package/dist/client/proxy/requestLogger.js +0 -207
- package/dist/client/proxy/tokenRefresh.js +0 -124
- package/dist/client/proxy/usageStats.js +0 -74
- package/dist/client/proxy/utils/noProxyUtils.js +0 -149
- package/dist/client/rag/ChunkerFactory.js +0 -320
- package/dist/client/rag/ChunkerRegistry.js +0 -421
- package/dist/client/rag/chunkers/BaseChunker.js +0 -143
- package/dist/client/rag/chunkers/CharacterChunker.js +0 -28
- package/dist/client/rag/chunkers/HTMLChunker.js +0 -38
- package/dist/client/rag/chunkers/JSONChunker.js +0 -68
- package/dist/client/rag/chunkers/LaTeXChunker.js +0 -63
- package/dist/client/rag/chunkers/MarkdownChunker.js +0 -306
- package/dist/client/rag/chunkers/RecursiveChunker.js +0 -139
- package/dist/client/rag/chunkers/SemanticMarkdownChunker.js +0 -138
- package/dist/client/rag/chunkers/SentenceChunker.js +0 -66
- package/dist/client/rag/chunkers/TokenChunker.js +0 -61
- package/dist/client/rag/chunkers/index.js +0 -15
- package/dist/client/rag/chunking/characterChunker.js +0 -142
- package/dist/client/rag/chunking/chunkerRegistry.js +0 -194
- package/dist/client/rag/chunking/htmlChunker.js +0 -247
- package/dist/client/rag/chunking/index.js +0 -17
- package/dist/client/rag/chunking/jsonChunker.js +0 -281
- package/dist/client/rag/chunking/latexChunker.js +0 -251
- package/dist/client/rag/chunking/markdownChunker.js +0 -373
- package/dist/client/rag/chunking/recursiveChunker.js +0 -148
- package/dist/client/rag/chunking/semanticChunker.js +0 -306
- package/dist/client/rag/chunking/sentenceChunker.js +0 -230
- package/dist/client/rag/chunking/tokenChunker.js +0 -183
- package/dist/client/rag/document/MDocument.js +0 -392
- package/dist/client/rag/document/index.js +0 -5
- package/dist/client/rag/document/loaders.js +0 -500
- package/dist/client/rag/errors/RAGError.js +0 -274
- package/dist/client/rag/errors/index.js +0 -6
- package/dist/client/rag/graphRag/graphRAG.js +0 -401
- package/dist/client/rag/graphRag/index.js +0 -4
- package/dist/client/rag/index.js +0 -141
- package/dist/client/rag/metadata/MetadataExtractorFactory.js +0 -418
- package/dist/client/rag/metadata/MetadataExtractorRegistry.js +0 -362
- package/dist/client/rag/metadata/index.js +0 -9
- package/dist/client/rag/metadata/metadataExtractor.js +0 -280
- package/dist/client/rag/pipeline/RAGPipeline.js +0 -436
- package/dist/client/rag/pipeline/contextAssembly.js +0 -341
- package/dist/client/rag/pipeline/index.js +0 -5
- package/dist/client/rag/ragIntegration.js +0 -321
- package/dist/client/rag/reranker/RerankerFactory.js +0 -430
- package/dist/client/rag/reranker/RerankerRegistry.js +0 -402
- package/dist/client/rag/reranker/index.js +0 -9
- package/dist/client/rag/reranker/reranker.js +0 -277
- package/dist/client/rag/resilience/CircuitBreaker.js +0 -431
- package/dist/client/rag/resilience/RetryHandler.js +0 -304
- package/dist/client/rag/resilience/index.js +0 -7
- package/dist/client/rag/retrieval/hybridSearch.js +0 -335
- package/dist/client/rag/retrieval/index.js +0 -5
- package/dist/client/rag/retrieval/vectorQueryTool.js +0 -307
- package/dist/client/rag/types.js +0 -8
- package/dist/client/sdk/toolRegistration.js +0 -377
- package/dist/client/server/abstract/baseServerAdapter.js +0 -575
- package/dist/client/server/adapters/expressAdapter.js +0 -486
- package/dist/client/server/adapters/fastifyAdapter.js +0 -472
- package/dist/client/server/adapters/honoAdapter.js +0 -632
- package/dist/client/server/adapters/koaAdapter.js +0 -510
- package/dist/client/server/errors.js +0 -486
- package/dist/client/server/factory/serverAdapterFactory.js +0 -160
- package/dist/client/server/index.js +0 -108
- package/dist/client/server/middleware/abortSignal.js +0 -111
- package/dist/client/server/middleware/auth.js +0 -388
- package/dist/client/server/middleware/cache.js +0 -359
- package/dist/client/server/middleware/common.js +0 -281
- package/dist/client/server/middleware/deprecation.js +0 -190
- package/dist/client/server/middleware/mcpBodyAttachment.js +0 -63
- package/dist/client/server/middleware/rateLimit.js +0 -227
- package/dist/client/server/middleware/validation.js +0 -388
- package/dist/client/server/openapi/generator.js +0 -398
- package/dist/client/server/openapi/index.js +0 -36
- package/dist/client/server/openapi/schemas.js +0 -695
- package/dist/client/server/openapi/templates.js +0 -374
- package/dist/client/server/routes/agentRoutes.js +0 -171
- package/dist/client/server/routes/claudeProxyRoutes.js +0 -1600
- package/dist/client/server/routes/healthRoutes.js +0 -187
- package/dist/client/server/routes/index.js +0 -57
- package/dist/client/server/routes/mcpRoutes.js +0 -342
- package/dist/client/server/routes/memoryRoutes.js +0 -350
- package/dist/client/server/routes/openApiRoutes.js +0 -126
- package/dist/client/server/routes/toolRoutes.js +0 -199
- package/dist/client/server/streaming/dataStream.js +0 -486
- package/dist/client/server/streaming/index.js +0 -11
- package/dist/client/server/types.js +0 -67
- package/dist/client/server/utils/redaction.js +0 -334
- package/dist/client/server/utils/validation.js +0 -243
- package/dist/client/server/websocket/WebSocketHandler.js +0 -383
- package/dist/client/server/websocket/index.js +0 -4
- package/dist/client/services/server/ai/observability/instrumentation.js +0 -808
- package/dist/client/telemetry/attributes.js +0 -100
- package/dist/client/telemetry/index.js +0 -26
- package/dist/client/telemetry/telemetryService.js +0 -308
- package/dist/client/telemetry/tracers.js +0 -17
- package/dist/client/telemetry/withSpan.js +0 -34
- package/dist/client/types/actionTypes.js +0 -6
- package/dist/client/types/analytics.js +0 -5
- package/dist/client/types/authTypes.js +0 -8
- package/dist/client/types/circuitBreakerErrors.js +0 -34
- package/dist/client/types/cli.js +0 -21
- package/dist/client/types/clientTypes.js +0 -10
- package/dist/client/types/common.js +0 -51
- package/dist/client/types/configTypes.js +0 -49
- package/dist/client/types/content.js +0 -19
- package/dist/client/types/contextTypes.js +0 -400
- package/dist/client/types/conversation.js +0 -47
- package/dist/client/types/conversationMemoryInterface.js +0 -6
- package/dist/client/types/domainTypes.js +0 -5
- package/dist/client/types/errors.js +0 -167
- package/dist/client/types/evaluation.js +0 -5
- package/dist/client/types/evaluationProviders.js +0 -5
- package/dist/client/types/evaluationTypes.js +0 -1
- package/dist/client/types/externalMcp.js +0 -6
- package/dist/client/types/fileReferenceTypes.js +0 -8
- package/dist/client/types/fileTypes.js +0 -4
- package/dist/client/types/generateTypes.js +0 -1
- package/dist/client/types/guardrails.js +0 -1
- package/dist/client/types/hitlTypes.js +0 -8
- package/dist/client/types/index.js +0 -57
- package/dist/client/types/mcpTypes.js +0 -5
- package/dist/client/types/middlewareTypes.js +0 -1
- package/dist/client/types/modelTypes.js +0 -30
- package/dist/client/types/multimodal.js +0 -135
- package/dist/client/types/observability.js +0 -6
- package/dist/client/types/pptTypes.js +0 -82
- package/dist/client/types/providers.js +0 -111
- package/dist/client/types/proxyTypes.js +0 -16
- package/dist/client/types/ragTypes.js +0 -7
- package/dist/client/types/sdkTypes.js +0 -8
- package/dist/client/types/serviceTypes.js +0 -5
- package/dist/client/types/streamTypes.js +0 -1
- package/dist/client/types/subscriptionTypes.js +0 -9
- package/dist/client/types/taskClassificationTypes.js +0 -5
- package/dist/client/types/tools.js +0 -24
- package/dist/client/types/ttsTypes.js +0 -57
- package/dist/client/types/typeAliases.js +0 -48
- package/dist/client/types/utilities.js +0 -4
- package/dist/client/types/workflowTypes.js +0 -30
- package/dist/client/utils/async/withTimeout.js +0 -98
- package/dist/client/utils/asyncMutex.js +0 -60
- package/dist/client/utils/conversationMemory.js +0 -431
- package/dist/client/utils/csvProcessor.js +0 -846
- package/dist/client/utils/errorHandling.js +0 -936
- package/dist/client/utils/evaluationUtils.js +0 -131
- package/dist/client/utils/factoryProcessing.js +0 -589
- package/dist/client/utils/fileDetector.js +0 -2161
- package/dist/client/utils/imageCache.js +0 -376
- package/dist/client/utils/imageProcessor.js +0 -704
- package/dist/client/utils/logger.js +0 -491
- package/dist/client/utils/mcpDefaults.js +0 -134
- package/dist/client/utils/messageBuilder.js +0 -1653
- package/dist/client/utils/modelAliasResolver.js +0 -54
- package/dist/client/utils/modelDetection.js +0 -80
- package/dist/client/utils/modelRouter.js +0 -292
- package/dist/client/utils/multimodalOptionsBuilder.js +0 -65
- package/dist/client/utils/observabilityHelpers.js +0 -47
- package/dist/client/utils/parameterValidation.js +0 -966
- package/dist/client/utils/pdfProcessor.js +0 -410
- package/dist/client/utils/performance.js +0 -222
- package/dist/client/utils/pricing.js +0 -340
- package/dist/client/utils/promptRedaction.js +0 -62
- package/dist/client/utils/providerConfig.js +0 -1009
- package/dist/client/utils/providerHealth.js +0 -1237
- package/dist/client/utils/providerRetry.js +0 -112
- package/dist/client/utils/providerUtils.js +0 -434
- package/dist/client/utils/rateLimiter.js +0 -200
- package/dist/client/utils/redis.js +0 -368
- package/dist/client/utils/retryHandler.js +0 -269
- package/dist/client/utils/retryability.js +0 -22
- package/dist/client/utils/sanitizers/svg.js +0 -481
- package/dist/client/utils/schemaConversion.js +0 -255
- package/dist/client/utils/taskClassificationUtils.js +0 -149
- package/dist/client/utils/taskClassifier.js +0 -94
- package/dist/client/utils/thinkingConfig.js +0 -104
- package/dist/client/utils/timeout.js +0 -359
- package/dist/client/utils/tokenEstimation.js +0 -142
- package/dist/client/utils/tokenLimits.js +0 -125
- package/dist/client/utils/tokenUtils.js +0 -239
- package/dist/client/utils/toolUtils.js +0 -75
- package/dist/client/utils/transformationUtils.js +0 -554
- package/dist/client/utils/ttsProcessor.js +0 -286
- package/dist/client/utils/typeUtils.js +0 -97
- package/dist/client/utils/videoAnalysisProcessor.js +0 -67
- package/dist/client/workflow/config.js +0 -398
- package/dist/client/workflow/core/ensembleExecutor.js +0 -407
- package/dist/client/workflow/core/judgeScorer.js +0 -544
- package/dist/client/workflow/core/responseConditioner.js +0 -225
- package/dist/client/workflow/core/types/conditionerTypes.js +0 -7
- package/dist/client/workflow/core/types/ensembleTypes.js +0 -7
- package/dist/client/workflow/core/types/index.js +0 -7
- package/dist/client/workflow/core/types/judgeTypes.js +0 -7
- package/dist/client/workflow/core/types/layerTypes.js +0 -7
- package/dist/client/workflow/core/types/registryTypes.js +0 -7
- package/dist/client/workflow/core/workflowRegistry.js +0 -304
- package/dist/client/workflow/core/workflowRunner.js +0 -586
- package/dist/client/workflow/index.js +0 -50
- package/dist/client/workflow/types.js +0 -9
- package/dist/client/workflow/utils/types/index.js +0 -7
- package/dist/client/workflow/utils/workflowMetrics.js +0 -311
- package/dist/client/workflow/utils/workflowValidation.js +0 -420
- package/dist/client/workflow/workflows/adaptiveWorkflow.js +0 -366
- package/dist/client/workflow/workflows/consensusWorkflow.js +0 -192
- package/dist/client/workflow/workflows/fallbackWorkflow.js +0 -225
- package/dist/client/workflow/workflows/multiJudgeWorkflow.js +0 -351
- /package/dist/client/{client/reactHooks.js → reactHooks.js} +0 -0
|
@@ -1,1478 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* External MCP Server Manager
|
|
3
|
-
* Handles lifecycle management of external MCP servers including:
|
|
4
|
-
* - Process spawning and management
|
|
5
|
-
* - Health monitoring and automatic restart
|
|
6
|
-
* - Connection management and cleanup
|
|
7
|
-
* - Tool discovery and registration
|
|
8
|
-
*/
|
|
9
|
-
import { EventEmitter } from "events";
|
|
10
|
-
import { mcpLogger } from "../utils/logger.js";
|
|
11
|
-
import { MCPClientFactory } from "./mcpClientFactory.js";
|
|
12
|
-
import { ToolDiscoveryService } from "./toolDiscoveryService.js";
|
|
13
|
-
import { toolRegistry } from "./toolRegistry.js";
|
|
14
|
-
import { HITLUserRejectedError, HITLTimeoutError } from "../hitl/hitlErrors.js";
|
|
15
|
-
import { detectCategory } from "../utils/mcpDefaults.js";
|
|
16
|
-
import { isObject, isNonNullObject } from "../utils/typeUtils.js";
|
|
17
|
-
import { TelemetryService } from "../telemetry/telemetryService.js";
|
|
18
|
-
import { tracers } from "../telemetry/tracers.js";
|
|
19
|
-
import { SpanStatusCode } from "@opentelemetry/api";
|
|
20
|
-
/**
|
|
21
|
-
* Recursively substitute environment variables in strings
|
|
22
|
-
* Replaces ${VAR_NAME} with the value from process.env.VAR_NAME
|
|
23
|
-
* @param value - Value to process (string, object, array, or primitive)
|
|
24
|
-
* @returns Processed value with environment variables substituted
|
|
25
|
-
*/
|
|
26
|
-
function substituteEnvVariables(value) {
|
|
27
|
-
if (typeof value === "string") {
|
|
28
|
-
// Replace ${VAR_NAME} with process.env.VAR_NAME
|
|
29
|
-
return value.replace(/\$\{([^}]+)\}/g, (match, varName) => {
|
|
30
|
-
const envValue = process.env[varName.trim()];
|
|
31
|
-
if (envValue === undefined) {
|
|
32
|
-
mcpLogger.warn(`[ExternalServerManager] Environment variable ${varName} is not defined, using empty string`);
|
|
33
|
-
return "";
|
|
34
|
-
}
|
|
35
|
-
return envValue;
|
|
36
|
-
});
|
|
37
|
-
}
|
|
38
|
-
if (Array.isArray(value)) {
|
|
39
|
-
return value.map((item) => substituteEnvVariables(item));
|
|
40
|
-
}
|
|
41
|
-
if (isNonNullObject(value)) {
|
|
42
|
-
const result = {};
|
|
43
|
-
for (const [key, val] of Object.entries(value)) {
|
|
44
|
-
result[key] = substituteEnvVariables(val);
|
|
45
|
-
}
|
|
46
|
-
return result;
|
|
47
|
-
}
|
|
48
|
-
return value;
|
|
49
|
-
}
|
|
50
|
-
/**
|
|
51
|
-
* Sensitive CLI flag patterns whose following value should be masked in logs.
|
|
52
|
-
*/
|
|
53
|
-
const SENSITIVE_ARG_PATTERNS = /^--(api-key|token|secret|password|key|figma-api-key|access-token|auth|credential)$/i;
|
|
54
|
-
/**
|
|
55
|
-
* Redact values that follow sensitive flags in a CLI args array.
|
|
56
|
-
* Handles: "--api-key sk-abc", "--api-key=sk-abc", consecutive flags.
|
|
57
|
-
* E.g. ["--api-key", "sk-abc123"] → ["--api-key", "[REDACTED]"]
|
|
58
|
-
* ["--api-key=sk-abc123"] → ["--api-key=[REDACTED]"]
|
|
59
|
-
*/
|
|
60
|
-
function redactSensitiveArgs(args) {
|
|
61
|
-
if (!args || args.length === 0) {
|
|
62
|
-
return args;
|
|
63
|
-
}
|
|
64
|
-
const redacted = [...args];
|
|
65
|
-
for (let i = 0; i < redacted.length; i++) {
|
|
66
|
-
// Handle --flag=value inline form
|
|
67
|
-
const eqIdx = redacted[i].indexOf("=");
|
|
68
|
-
if (eqIdx !== -1) {
|
|
69
|
-
const flag = redacted[i].substring(0, eqIdx);
|
|
70
|
-
if (SENSITIVE_ARG_PATTERNS.test(flag)) {
|
|
71
|
-
redacted[i] = `${flag}=[REDACTED]`;
|
|
72
|
-
}
|
|
73
|
-
continue;
|
|
74
|
-
}
|
|
75
|
-
// Handle --flag value (two-part form)
|
|
76
|
-
if (SENSITIVE_ARG_PATTERNS.test(redacted[i]) &&
|
|
77
|
-
i + 1 < redacted.length &&
|
|
78
|
-
!redacted[i + 1].startsWith("--")) {
|
|
79
|
-
redacted[i + 1] = "[REDACTED]";
|
|
80
|
-
i++; // skip the redacted value
|
|
81
|
-
}
|
|
82
|
-
}
|
|
83
|
-
return redacted;
|
|
84
|
-
}
|
|
85
|
-
/**
|
|
86
|
-
* Type guard to validate if an object can be safely used as Record<string, JsonValue>
|
|
87
|
-
*/
|
|
88
|
-
function isValidJsonRecord(value) {
|
|
89
|
-
if (!isObject(value)) {
|
|
90
|
-
return false;
|
|
91
|
-
}
|
|
92
|
-
const record = value;
|
|
93
|
-
return Object.values(record).every((val) => {
|
|
94
|
-
// JsonValue = string | number | boolean | null | JsonValue[] | { [key: string]: JsonValue }
|
|
95
|
-
if (val === null ||
|
|
96
|
-
typeof val === "string" ||
|
|
97
|
-
typeof val === "number" ||
|
|
98
|
-
typeof val === "boolean") {
|
|
99
|
-
return true;
|
|
100
|
-
}
|
|
101
|
-
if (Array.isArray(val)) {
|
|
102
|
-
return val.every((item) => isValidJsonRecord(item) ||
|
|
103
|
-
typeof item === "string" ||
|
|
104
|
-
typeof item === "number" ||
|
|
105
|
-
typeof item === "boolean" ||
|
|
106
|
-
item === null);
|
|
107
|
-
}
|
|
108
|
-
if (isNonNullObject(val)) {
|
|
109
|
-
return isValidJsonRecord(val);
|
|
110
|
-
}
|
|
111
|
-
return false;
|
|
112
|
-
});
|
|
113
|
-
}
|
|
114
|
-
/**
|
|
115
|
-
* Safely converts unknown metadata to Record<string, JsonValue> or returns undefined
|
|
116
|
-
*/
|
|
117
|
-
function safeMetadataConversion(metadata) {
|
|
118
|
-
return isValidJsonRecord(metadata) ? metadata : undefined;
|
|
119
|
-
}
|
|
120
|
-
/**
|
|
121
|
-
* Type guard to validate external MCP server configuration
|
|
122
|
-
* Supports both stdio transport (requires command) and HTTP transport (requires url)
|
|
123
|
-
*/
|
|
124
|
-
function isValidExternalMCPServerConfig(config) {
|
|
125
|
-
if (!isNonNullObject(config)) {
|
|
126
|
-
return false;
|
|
127
|
-
}
|
|
128
|
-
const record = config;
|
|
129
|
-
// Validate blockedTools array contains only strings
|
|
130
|
-
if (record.blockedTools !== undefined) {
|
|
131
|
-
if (!Array.isArray(record.blockedTools)) {
|
|
132
|
-
return false;
|
|
133
|
-
}
|
|
134
|
-
if (!record.blockedTools.every((item) => typeof item === "string")) {
|
|
135
|
-
return false;
|
|
136
|
-
}
|
|
137
|
-
}
|
|
138
|
-
// Must have either command (for stdio) or url (for HTTP/SSE transport)
|
|
139
|
-
const hasCommand = typeof record.command === "string";
|
|
140
|
-
const hasUrl = typeof record.url === "string";
|
|
141
|
-
if (!hasCommand && !hasUrl) {
|
|
142
|
-
return false;
|
|
143
|
-
}
|
|
144
|
-
return ((record.command === undefined || typeof record.command === "string") &&
|
|
145
|
-
(record.args === undefined || Array.isArray(record.args)) &&
|
|
146
|
-
(record.env === undefined || isNonNullObject(record.env)) &&
|
|
147
|
-
(record.transport === undefined || typeof record.transport === "string") &&
|
|
148
|
-
(record.timeout === undefined || typeof record.timeout === "number") &&
|
|
149
|
-
(record.retries === undefined || typeof record.retries === "number") &&
|
|
150
|
-
(record.healthCheckInterval === undefined ||
|
|
151
|
-
typeof record.healthCheckInterval === "number") &&
|
|
152
|
-
(record.autoRestart === undefined ||
|
|
153
|
-
typeof record.autoRestart === "boolean") &&
|
|
154
|
-
(record.cwd === undefined || typeof record.cwd === "string") &&
|
|
155
|
-
(record.url === undefined || typeof record.url === "string") &&
|
|
156
|
-
(record.headers === undefined || isNonNullObject(record.headers)) &&
|
|
157
|
-
(record.httpOptions === undefined || isNonNullObject(record.httpOptions)) &&
|
|
158
|
-
(record.retryConfig === undefined || isNonNullObject(record.retryConfig)) &&
|
|
159
|
-
(record.rateLimiting === undefined ||
|
|
160
|
-
isNonNullObject(record.rateLimiting)) &&
|
|
161
|
-
(record.metadata === undefined || isNonNullObject(record.metadata)));
|
|
162
|
-
}
|
|
163
|
-
/**
|
|
164
|
-
* ExternalServerManager
|
|
165
|
-
* Core class for managing external MCP servers
|
|
166
|
-
*/
|
|
167
|
-
export class ExternalServerManager extends EventEmitter {
|
|
168
|
-
servers = new Map();
|
|
169
|
-
config;
|
|
170
|
-
isShuttingDown = false;
|
|
171
|
-
toolDiscovery;
|
|
172
|
-
enableMainRegistryIntegration;
|
|
173
|
-
hitlManager; // Optional HITL manager for safety mechanisms
|
|
174
|
-
constructor(config = {}, options = {}) {
|
|
175
|
-
super();
|
|
176
|
-
// Set defaults for configuration
|
|
177
|
-
// Default timeout increased to 60s and made configurable via MCP_CLIENT_TIMEOUT
|
|
178
|
-
// to accommodate MCP server startup latency (especially concurrent stdio servers)
|
|
179
|
-
const defaultMcpTimeout = Math.max(5000, Number(process.env.MCP_CLIENT_TIMEOUT) || 60000);
|
|
180
|
-
this.config = {
|
|
181
|
-
maxServers: config.maxServers ?? 10,
|
|
182
|
-
defaultTimeout: config.defaultTimeout ?? defaultMcpTimeout,
|
|
183
|
-
defaultHealthCheckInterval: config.defaultHealthCheckInterval ?? 30000,
|
|
184
|
-
enableAutoRestart: config.enableAutoRestart ?? true,
|
|
185
|
-
maxRestartAttempts: config.maxRestartAttempts ?? 3,
|
|
186
|
-
restartBackoffMultiplier: config.restartBackoffMultiplier ?? 2,
|
|
187
|
-
enablePerformanceMonitoring: config.enablePerformanceMonitoring ?? true,
|
|
188
|
-
logLevel: config.logLevel ?? "info",
|
|
189
|
-
};
|
|
190
|
-
// Disable main tool registry integration by default to prevent automatic tool execution
|
|
191
|
-
this.enableMainRegistryIntegration =
|
|
192
|
-
options.enableMainRegistryIntegration ?? false;
|
|
193
|
-
// Initialize tool discovery service
|
|
194
|
-
this.toolDiscovery = new ToolDiscoveryService();
|
|
195
|
-
// Forward tool discovery events
|
|
196
|
-
this.toolDiscovery.on("toolRegistered", (event) => {
|
|
197
|
-
this.emit("toolDiscovered", event);
|
|
198
|
-
});
|
|
199
|
-
this.toolDiscovery.on("toolUnregistered", (event) => {
|
|
200
|
-
this.emit("toolRemoved", event);
|
|
201
|
-
});
|
|
202
|
-
// Handle process cleanup
|
|
203
|
-
process.on("SIGINT", () => this.shutdown());
|
|
204
|
-
process.on("SIGTERM", () => this.shutdown());
|
|
205
|
-
process.on("beforeExit", () => this.shutdown());
|
|
206
|
-
}
|
|
207
|
-
/**
|
|
208
|
-
* Set HITL manager for human-in-the-loop safety mechanisms
|
|
209
|
-
* @param hitlManager - HITL manager instance (optional, can be undefined to disable)
|
|
210
|
-
*/
|
|
211
|
-
setHITLManager(hitlManager) {
|
|
212
|
-
this.hitlManager = hitlManager;
|
|
213
|
-
if (hitlManager && hitlManager.isEnabled()) {
|
|
214
|
-
mcpLogger.info("[ExternalServerManager] HITL safety mechanisms enabled for external tool execution");
|
|
215
|
-
}
|
|
216
|
-
else {
|
|
217
|
-
mcpLogger.debug("[ExternalServerManager] HITL safety mechanisms disabled or not configured");
|
|
218
|
-
}
|
|
219
|
-
}
|
|
220
|
-
/**
|
|
221
|
-
* Get current HITL manager
|
|
222
|
-
*/
|
|
223
|
-
getHITLManager() {
|
|
224
|
-
return this.hitlManager;
|
|
225
|
-
}
|
|
226
|
-
/**
|
|
227
|
-
* Load MCP server configurations from .mcp-config.json file with parallel loading support
|
|
228
|
-
* Automatically registers servers found in the configuration
|
|
229
|
-
* @param configPath Optional path to config file (defaults to .mcp-config.json in cwd)
|
|
230
|
-
* @param options Loading options including parallel support
|
|
231
|
-
* @returns Promise resolving to { serversLoaded, errors }
|
|
232
|
-
*/
|
|
233
|
-
async loadMCPConfiguration(configPath, options = {}) {
|
|
234
|
-
if (options.parallel) {
|
|
235
|
-
return this.loadMCPConfigurationParallel(configPath);
|
|
236
|
-
}
|
|
237
|
-
return this.loadMCPConfigurationSequential(configPath);
|
|
238
|
-
}
|
|
239
|
-
/**
|
|
240
|
-
* Load MCP servers in parallel for improved performance
|
|
241
|
-
* @param configPath Optional path to config file (defaults to .mcp-config.json in cwd)
|
|
242
|
-
* @returns Promise resolving to batch operation result
|
|
243
|
-
*/
|
|
244
|
-
async loadMCPConfigurationParallel(configPath) {
|
|
245
|
-
const fs = await import("fs");
|
|
246
|
-
const path = await import("path");
|
|
247
|
-
const finalConfigPath = configPath || path.join(process.cwd(), ".mcp-config.json");
|
|
248
|
-
if (!fs.existsSync(finalConfigPath)) {
|
|
249
|
-
mcpLogger.debug(`[ExternalServerManager] No MCP config found at ${finalConfigPath}`);
|
|
250
|
-
return { serversLoaded: 0, errors: [] };
|
|
251
|
-
}
|
|
252
|
-
mcpLogger.debug(`[ExternalServerManager] Loading MCP configuration in PARALLEL mode from ${finalConfigPath}`);
|
|
253
|
-
try {
|
|
254
|
-
const configContent = fs.readFileSync(finalConfigPath, "utf8");
|
|
255
|
-
const config = JSON.parse(configContent);
|
|
256
|
-
if (!config.mcpServers || typeof config.mcpServers !== "object") {
|
|
257
|
-
mcpLogger.debug("[ExternalServerManager] No mcpServers found in configuration");
|
|
258
|
-
return { serversLoaded: 0, errors: [] };
|
|
259
|
-
}
|
|
260
|
-
// Create promises for all servers to start them concurrently
|
|
261
|
-
const serverPromises = Object.entries(config.mcpServers).map(async ([serverId, serverConfig]) => {
|
|
262
|
-
try {
|
|
263
|
-
// Validate and convert config format to MCPServerInfo
|
|
264
|
-
if (!isValidExternalMCPServerConfig(serverConfig)) {
|
|
265
|
-
throw new Error(`Invalid server config for ${serverId}: missing required properties or wrong types`);
|
|
266
|
-
}
|
|
267
|
-
const externalConfig = {
|
|
268
|
-
id: serverId,
|
|
269
|
-
name: serverId,
|
|
270
|
-
description: typeof serverConfig.description === "string"
|
|
271
|
-
? serverConfig.description
|
|
272
|
-
: `External MCP server: ${serverId}`,
|
|
273
|
-
transport: typeof serverConfig.transport === "string"
|
|
274
|
-
? serverConfig.transport
|
|
275
|
-
: "stdio",
|
|
276
|
-
status: "initializing",
|
|
277
|
-
tools: [],
|
|
278
|
-
command: typeof serverConfig.command === "string"
|
|
279
|
-
? serverConfig.command
|
|
280
|
-
: undefined,
|
|
281
|
-
args: Array.isArray(serverConfig.args)
|
|
282
|
-
? serverConfig.args
|
|
283
|
-
: [],
|
|
284
|
-
env: isNonNullObject(serverConfig.env)
|
|
285
|
-
? substituteEnvVariables(serverConfig.env)
|
|
286
|
-
: {},
|
|
287
|
-
timeout: typeof serverConfig.timeout === "number"
|
|
288
|
-
? serverConfig.timeout
|
|
289
|
-
: undefined,
|
|
290
|
-
retries: typeof serverConfig.retries === "number"
|
|
291
|
-
? serverConfig.retries
|
|
292
|
-
: undefined,
|
|
293
|
-
healthCheckInterval: typeof serverConfig.healthCheckInterval === "number"
|
|
294
|
-
? serverConfig.healthCheckInterval
|
|
295
|
-
: undefined,
|
|
296
|
-
autoRestart: typeof serverConfig.autoRestart === "boolean"
|
|
297
|
-
? serverConfig.autoRestart
|
|
298
|
-
: undefined,
|
|
299
|
-
cwd: typeof serverConfig.cwd === "string"
|
|
300
|
-
? serverConfig.cwd
|
|
301
|
-
: undefined,
|
|
302
|
-
url: typeof serverConfig.url === "string"
|
|
303
|
-
? serverConfig.url
|
|
304
|
-
: undefined,
|
|
305
|
-
// HTTP transport-specific fields
|
|
306
|
-
headers: isNonNullObject(serverConfig.headers)
|
|
307
|
-
? substituteEnvVariables(serverConfig.headers)
|
|
308
|
-
: undefined,
|
|
309
|
-
httpOptions: isNonNullObject(serverConfig.httpOptions)
|
|
310
|
-
? serverConfig.httpOptions
|
|
311
|
-
: undefined,
|
|
312
|
-
retryConfig: isNonNullObject(serverConfig.retryConfig)
|
|
313
|
-
? serverConfig.retryConfig
|
|
314
|
-
: undefined,
|
|
315
|
-
rateLimiting: isNonNullObject(serverConfig.rateLimiting)
|
|
316
|
-
? serverConfig.rateLimiting
|
|
317
|
-
: undefined,
|
|
318
|
-
blockedTools: Array.isArray(serverConfig.blockedTools)
|
|
319
|
-
? serverConfig.blockedTools
|
|
320
|
-
: undefined,
|
|
321
|
-
metadata: safeMetadataConversion(serverConfig.metadata),
|
|
322
|
-
};
|
|
323
|
-
const result = await this.addServer(serverId, externalConfig);
|
|
324
|
-
return { serverId, result };
|
|
325
|
-
}
|
|
326
|
-
catch (error) {
|
|
327
|
-
const errorMsg = `Failed to load MCP server ${serverId}: ${error instanceof Error ? error.message : String(error)}`;
|
|
328
|
-
mcpLogger.warn(`[ExternalServerManager] ${errorMsg}`);
|
|
329
|
-
return { serverId, error: errorMsg };
|
|
330
|
-
}
|
|
331
|
-
});
|
|
332
|
-
// Start all servers concurrently and wait for completion
|
|
333
|
-
const results = await Promise.allSettled(serverPromises);
|
|
334
|
-
// Process results to count successes and collect errors
|
|
335
|
-
let serversLoaded = 0;
|
|
336
|
-
const errors = [];
|
|
337
|
-
for (const result of results) {
|
|
338
|
-
if (result.status === "fulfilled") {
|
|
339
|
-
const { serverId, result: serverResult, error } = result.value;
|
|
340
|
-
if (serverResult && serverResult.success) {
|
|
341
|
-
serversLoaded++;
|
|
342
|
-
mcpLogger.debug(`[ExternalServerManager] Successfully loaded MCP server in parallel: ${serverId}`);
|
|
343
|
-
}
|
|
344
|
-
else if (error) {
|
|
345
|
-
errors.push(error);
|
|
346
|
-
}
|
|
347
|
-
else if (serverResult && !serverResult.success) {
|
|
348
|
-
const errorMsg = `Failed to load server ${serverId}: ${serverResult.error}`;
|
|
349
|
-
errors.push(errorMsg);
|
|
350
|
-
mcpLogger.warn(`[ExternalServerManager] ${errorMsg}`);
|
|
351
|
-
}
|
|
352
|
-
}
|
|
353
|
-
else {
|
|
354
|
-
// Promise.allSettled rejected - this shouldn't happen with our error handling
|
|
355
|
-
const errorMsg = `Unexpected error during parallel loading: ${result.reason}`;
|
|
356
|
-
errors.push(errorMsg);
|
|
357
|
-
mcpLogger.error(`[ExternalServerManager] ${errorMsg}`);
|
|
358
|
-
}
|
|
359
|
-
}
|
|
360
|
-
mcpLogger.info(`[ExternalServerManager] PARALLEL MCP configuration loading complete: ${serversLoaded} servers loaded, ${errors.length} errors`);
|
|
361
|
-
return { serversLoaded, errors };
|
|
362
|
-
}
|
|
363
|
-
catch (error) {
|
|
364
|
-
const errorMsg = `Failed to load MCP configuration in parallel mode: ${error instanceof Error ? error.message : String(error)}`;
|
|
365
|
-
mcpLogger.error(`[ExternalServerManager] ${errorMsg}`);
|
|
366
|
-
return { serversLoaded: 0, errors: [errorMsg] };
|
|
367
|
-
}
|
|
368
|
-
}
|
|
369
|
-
/**
|
|
370
|
-
* Load MCP servers sequentially (original implementation for backward compatibility)
|
|
371
|
-
* @param configPath Optional path to config file (defaults to .mcp-config.json in cwd)
|
|
372
|
-
* @returns Promise resolving to batch operation result
|
|
373
|
-
*/
|
|
374
|
-
async loadMCPConfigurationSequential(configPath) {
|
|
375
|
-
const fs = await import("fs");
|
|
376
|
-
const path = await import("path");
|
|
377
|
-
const finalConfigPath = configPath || path.join(process.cwd(), ".mcp-config.json");
|
|
378
|
-
if (!fs.existsSync(finalConfigPath)) {
|
|
379
|
-
mcpLogger.debug(`[ExternalServerManager] No MCP config found at ${finalConfigPath}`);
|
|
380
|
-
return { serversLoaded: 0, errors: [] };
|
|
381
|
-
}
|
|
382
|
-
mcpLogger.debug(`[ExternalServerManager] Loading MCP configuration from ${finalConfigPath}`);
|
|
383
|
-
try {
|
|
384
|
-
const configContent = fs.readFileSync(finalConfigPath, "utf8");
|
|
385
|
-
const config = JSON.parse(configContent);
|
|
386
|
-
if (!config.mcpServers || typeof config.mcpServers !== "object") {
|
|
387
|
-
mcpLogger.debug("[ExternalServerManager] No mcpServers found in configuration");
|
|
388
|
-
return { serversLoaded: 0, errors: [] };
|
|
389
|
-
}
|
|
390
|
-
let serversLoaded = 0;
|
|
391
|
-
const errors = [];
|
|
392
|
-
for (const [serverId, serverConfig] of Object.entries(config.mcpServers)) {
|
|
393
|
-
try {
|
|
394
|
-
// Validate and convert config format to MCPServerInfo
|
|
395
|
-
if (!isValidExternalMCPServerConfig(serverConfig)) {
|
|
396
|
-
throw new Error(`Invalid server config for ${serverId}: missing required properties or wrong types`);
|
|
397
|
-
}
|
|
398
|
-
const externalConfig = {
|
|
399
|
-
id: serverId,
|
|
400
|
-
name: serverId,
|
|
401
|
-
description: typeof serverConfig.description === "string"
|
|
402
|
-
? serverConfig.description
|
|
403
|
-
: `External MCP server: ${serverId}`,
|
|
404
|
-
transport: typeof serverConfig.transport === "string"
|
|
405
|
-
? serverConfig.transport
|
|
406
|
-
: "stdio",
|
|
407
|
-
status: "initializing",
|
|
408
|
-
tools: [],
|
|
409
|
-
command: typeof serverConfig.command === "string"
|
|
410
|
-
? serverConfig.command
|
|
411
|
-
: undefined,
|
|
412
|
-
args: Array.isArray(serverConfig.args)
|
|
413
|
-
? serverConfig.args
|
|
414
|
-
: [],
|
|
415
|
-
env: isNonNullObject(serverConfig.env)
|
|
416
|
-
? substituteEnvVariables(serverConfig.env)
|
|
417
|
-
: {},
|
|
418
|
-
timeout: typeof serverConfig.timeout === "number"
|
|
419
|
-
? serverConfig.timeout
|
|
420
|
-
: undefined,
|
|
421
|
-
retries: typeof serverConfig.retries === "number"
|
|
422
|
-
? serverConfig.retries
|
|
423
|
-
: undefined,
|
|
424
|
-
healthCheckInterval: typeof serverConfig.healthCheckInterval === "number"
|
|
425
|
-
? serverConfig.healthCheckInterval
|
|
426
|
-
: undefined,
|
|
427
|
-
autoRestart: typeof serverConfig.autoRestart === "boolean"
|
|
428
|
-
? serverConfig.autoRestart
|
|
429
|
-
: undefined,
|
|
430
|
-
cwd: typeof serverConfig.cwd === "string"
|
|
431
|
-
? serverConfig.cwd
|
|
432
|
-
: undefined,
|
|
433
|
-
url: typeof serverConfig.url === "string"
|
|
434
|
-
? serverConfig.url
|
|
435
|
-
: undefined,
|
|
436
|
-
// HTTP transport-specific fields
|
|
437
|
-
headers: isNonNullObject(serverConfig.headers)
|
|
438
|
-
? substituteEnvVariables(serverConfig.headers)
|
|
439
|
-
: undefined,
|
|
440
|
-
httpOptions: isNonNullObject(serverConfig.httpOptions)
|
|
441
|
-
? serverConfig.httpOptions
|
|
442
|
-
: undefined,
|
|
443
|
-
retryConfig: isNonNullObject(serverConfig.retryConfig)
|
|
444
|
-
? serverConfig.retryConfig
|
|
445
|
-
: undefined,
|
|
446
|
-
rateLimiting: isNonNullObject(serverConfig.rateLimiting)
|
|
447
|
-
? serverConfig.rateLimiting
|
|
448
|
-
: undefined,
|
|
449
|
-
blockedTools: Array.isArray(serverConfig.blockedTools)
|
|
450
|
-
? serverConfig.blockedTools
|
|
451
|
-
: undefined,
|
|
452
|
-
metadata: safeMetadataConversion(serverConfig.metadata),
|
|
453
|
-
};
|
|
454
|
-
const result = await this.addServer(serverId, externalConfig);
|
|
455
|
-
if (result.success) {
|
|
456
|
-
serversLoaded++;
|
|
457
|
-
mcpLogger.debug(`[ExternalServerManager] Successfully loaded MCP server: ${serverId}`);
|
|
458
|
-
}
|
|
459
|
-
else {
|
|
460
|
-
const error = `Failed to load server ${serverId}: ${result.error}`;
|
|
461
|
-
errors.push(error);
|
|
462
|
-
mcpLogger.warn(`[ExternalServerManager] ${error}`);
|
|
463
|
-
}
|
|
464
|
-
}
|
|
465
|
-
catch (error) {
|
|
466
|
-
const errorMsg = `Failed to load MCP server ${serverId}: ${error instanceof Error ? error.message : String(error)}`;
|
|
467
|
-
errors.push(errorMsg);
|
|
468
|
-
mcpLogger.warn(`[ExternalServerManager] ${errorMsg}`);
|
|
469
|
-
// Continue with other servers - don't let one failure break everything
|
|
470
|
-
}
|
|
471
|
-
}
|
|
472
|
-
mcpLogger.info(`[ExternalServerManager] MCP configuration loading complete: ${serversLoaded} servers loaded, ${errors.length} errors`);
|
|
473
|
-
return { serversLoaded, errors };
|
|
474
|
-
}
|
|
475
|
-
catch (error) {
|
|
476
|
-
const errorMsg = `Failed to load MCP configuration: ${error instanceof Error ? error.message : String(error)}`;
|
|
477
|
-
mcpLogger.error(`[ExternalServerManager] ${errorMsg}`);
|
|
478
|
-
return { serversLoaded: 0, errors: [errorMsg] };
|
|
479
|
-
}
|
|
480
|
-
}
|
|
481
|
-
/**
|
|
482
|
-
* Validate external MCP server configuration
|
|
483
|
-
*/
|
|
484
|
-
validateConfig(config) {
|
|
485
|
-
const errors = [];
|
|
486
|
-
const warnings = [];
|
|
487
|
-
const suggestions = [];
|
|
488
|
-
// Required fields validation
|
|
489
|
-
if (!config.id || typeof config.id !== "string") {
|
|
490
|
-
errors.push("Server ID is required and must be a string");
|
|
491
|
-
}
|
|
492
|
-
if (!["stdio", "sse", "websocket", "http"].includes(config.transport)) {
|
|
493
|
-
errors.push("Transport must be one of: stdio, sse, websocket, http");
|
|
494
|
-
}
|
|
495
|
-
// Transport-specific validation
|
|
496
|
-
if (config.transport === "stdio") {
|
|
497
|
-
// stdio transport requires command
|
|
498
|
-
if (!config.command || typeof config.command !== "string") {
|
|
499
|
-
errors.push("Command is required and must be a string for stdio transport");
|
|
500
|
-
}
|
|
501
|
-
if (!Array.isArray(config.args)) {
|
|
502
|
-
errors.push("Args must be an array");
|
|
503
|
-
}
|
|
504
|
-
}
|
|
505
|
-
else if (config.transport === "sse" ||
|
|
506
|
-
config.transport === "websocket" ||
|
|
507
|
-
config.transport === "http") {
|
|
508
|
-
// HTTP-based transports require URL
|
|
509
|
-
if (!config.url || typeof config.url !== "string") {
|
|
510
|
-
errors.push(`URL is required for ${config.transport} transport`);
|
|
511
|
-
}
|
|
512
|
-
}
|
|
513
|
-
// Warnings for common issues
|
|
514
|
-
if (config.timeout && config.timeout < 5000) {
|
|
515
|
-
warnings.push("Timeout less than 5 seconds may cause connection issues");
|
|
516
|
-
}
|
|
517
|
-
if (config.retries && config.retries > 5) {
|
|
518
|
-
warnings.push("High retry count may slow down error recovery");
|
|
519
|
-
}
|
|
520
|
-
// Suggestions for optimization
|
|
521
|
-
if (!config.healthCheckInterval) {
|
|
522
|
-
suggestions.push("Consider setting a health check interval for better reliability");
|
|
523
|
-
}
|
|
524
|
-
if (config.autoRestart === undefined) {
|
|
525
|
-
suggestions.push("Consider enabling auto-restart for production use");
|
|
526
|
-
}
|
|
527
|
-
return {
|
|
528
|
-
isValid: errors.length === 0,
|
|
529
|
-
errors,
|
|
530
|
-
warnings,
|
|
531
|
-
suggestions,
|
|
532
|
-
};
|
|
533
|
-
}
|
|
534
|
-
/**
|
|
535
|
-
* Convert MCPServerInfo format (keeping for backward compatibility)
|
|
536
|
-
* Helper function for transitioning to zero-conversion architecture
|
|
537
|
-
*/
|
|
538
|
-
convertConfigToMCPServerInfo(serverId, config) {
|
|
539
|
-
return {
|
|
540
|
-
id: serverId,
|
|
541
|
-
name: String(config.metadata?.title || serverId),
|
|
542
|
-
description: `External MCP server (${config.transport})`,
|
|
543
|
-
status: "initializing",
|
|
544
|
-
transport: config.transport,
|
|
545
|
-
command: config.command,
|
|
546
|
-
args: config.args,
|
|
547
|
-
env: config.env,
|
|
548
|
-
tools: [], // Will be populated after server connection
|
|
549
|
-
blockedTools: config.blockedTools,
|
|
550
|
-
// Preserve top-level operational fields so startServer can read them
|
|
551
|
-
timeout: config.timeout,
|
|
552
|
-
retries: config.retries,
|
|
553
|
-
healthCheckInterval: config.healthCheckInterval,
|
|
554
|
-
autoRestart: config.autoRestart,
|
|
555
|
-
cwd: config.cwd,
|
|
556
|
-
url: config.url,
|
|
557
|
-
metadata: {
|
|
558
|
-
category: "external",
|
|
559
|
-
...(safeMetadataConversion(config.metadata) || {}),
|
|
560
|
-
},
|
|
561
|
-
};
|
|
562
|
-
}
|
|
563
|
-
async addServer(serverId, configOrServerInfo) {
|
|
564
|
-
const startTime = Date.now();
|
|
565
|
-
try {
|
|
566
|
-
// Use MCPServerInfo directly (zero-conversion architecture)
|
|
567
|
-
const serverInfo = "transport" in configOrServerInfo &&
|
|
568
|
-
"command" in configOrServerInfo &&
|
|
569
|
-
!("tools" in configOrServerInfo)
|
|
570
|
-
? this.convertConfigToMCPServerInfo(serverId, configOrServerInfo)
|
|
571
|
-
: configOrServerInfo;
|
|
572
|
-
// Check server limit
|
|
573
|
-
if (this.servers.size >= this.config.maxServers) {
|
|
574
|
-
return {
|
|
575
|
-
success: false,
|
|
576
|
-
error: `Maximum number of servers (${this.config.maxServers}) reached`,
|
|
577
|
-
serverId,
|
|
578
|
-
duration: Date.now() - startTime,
|
|
579
|
-
};
|
|
580
|
-
}
|
|
581
|
-
// Validate configuration (for backward compatibility, create temporary config)
|
|
582
|
-
const tempConfig = {
|
|
583
|
-
id: serverId,
|
|
584
|
-
name: serverInfo.name,
|
|
585
|
-
description: serverInfo.description,
|
|
586
|
-
transport: serverInfo.transport,
|
|
587
|
-
status: serverInfo.status,
|
|
588
|
-
tools: serverInfo.tools,
|
|
589
|
-
command: serverInfo.command || "",
|
|
590
|
-
args: serverInfo.args || [],
|
|
591
|
-
env: serverInfo.env || {},
|
|
592
|
-
timeout: serverInfo.timeout,
|
|
593
|
-
retries: serverInfo.retries,
|
|
594
|
-
healthCheckInterval: serverInfo.healthCheckInterval,
|
|
595
|
-
autoRestart: serverInfo.autoRestart,
|
|
596
|
-
cwd: serverInfo.cwd,
|
|
597
|
-
url: serverInfo.url,
|
|
598
|
-
blockedTools: serverInfo.blockedTools,
|
|
599
|
-
metadata: safeMetadataConversion(serverInfo.metadata),
|
|
600
|
-
// HTTP transport-specific fields
|
|
601
|
-
headers: serverInfo.headers,
|
|
602
|
-
httpOptions: serverInfo.httpOptions,
|
|
603
|
-
retryConfig: serverInfo.retryConfig,
|
|
604
|
-
rateLimiting: serverInfo.rateLimiting,
|
|
605
|
-
};
|
|
606
|
-
const validation = this.validateConfig(tempConfig);
|
|
607
|
-
if (!validation.isValid) {
|
|
608
|
-
return {
|
|
609
|
-
success: false,
|
|
610
|
-
error: `Configuration validation failed: ${validation.errors.join(", ")}`,
|
|
611
|
-
serverId,
|
|
612
|
-
duration: Date.now() - startTime,
|
|
613
|
-
};
|
|
614
|
-
}
|
|
615
|
-
// Check for duplicate server ID
|
|
616
|
-
if (this.servers.has(serverId)) {
|
|
617
|
-
return {
|
|
618
|
-
success: false,
|
|
619
|
-
error: `Server with ID '${serverId}' already exists`,
|
|
620
|
-
serverId,
|
|
621
|
-
duration: Date.now() - startTime,
|
|
622
|
-
};
|
|
623
|
-
}
|
|
624
|
-
mcpLogger.info(`[ExternalServerManager] Adding server: ${serverId}`, {
|
|
625
|
-
command: serverInfo.command,
|
|
626
|
-
transport: serverInfo.transport,
|
|
627
|
-
});
|
|
628
|
-
// Create server instance as RuntimeMCPServerInfo (transition to zero-conversion)
|
|
629
|
-
const instance = {
|
|
630
|
-
...serverInfo,
|
|
631
|
-
process: null,
|
|
632
|
-
client: null,
|
|
633
|
-
transportInstance: null,
|
|
634
|
-
status: "initializing",
|
|
635
|
-
reconnectAttempts: 0,
|
|
636
|
-
maxReconnectAttempts: this.config.maxRestartAttempts,
|
|
637
|
-
toolsMap: new Map(),
|
|
638
|
-
metrics: {
|
|
639
|
-
totalConnections: 0,
|
|
640
|
-
totalDisconnections: 0,
|
|
641
|
-
totalErrors: 0,
|
|
642
|
-
totalToolCalls: 0,
|
|
643
|
-
averageResponseTime: 0,
|
|
644
|
-
lastResponseTime: 0,
|
|
645
|
-
},
|
|
646
|
-
config: tempConfig,
|
|
647
|
-
};
|
|
648
|
-
// Store the instance
|
|
649
|
-
this.servers.set(serverId, instance);
|
|
650
|
-
// Start the server
|
|
651
|
-
await this.startServer(serverId);
|
|
652
|
-
const finalInstance = this.servers.get(serverId);
|
|
653
|
-
if (!finalInstance) {
|
|
654
|
-
throw new Error(`Server ${serverId} not found after registration`);
|
|
655
|
-
}
|
|
656
|
-
// Convert RuntimeMCPServerInfo to ExternalMCPServerInstance for return
|
|
657
|
-
const convertedInstance = {
|
|
658
|
-
config: finalInstance.config,
|
|
659
|
-
process: finalInstance.process,
|
|
660
|
-
client: finalInstance.client,
|
|
661
|
-
transport: finalInstance.transportInstance,
|
|
662
|
-
status: finalInstance.status,
|
|
663
|
-
lastError: finalInstance.lastError,
|
|
664
|
-
startTime: finalInstance.startTime,
|
|
665
|
-
lastHealthCheck: finalInstance.lastHealthCheck,
|
|
666
|
-
reconnectAttempts: finalInstance.reconnectAttempts,
|
|
667
|
-
maxReconnectAttempts: finalInstance.maxReconnectAttempts,
|
|
668
|
-
tools: finalInstance.toolsMap,
|
|
669
|
-
toolsArray: finalInstance.toolsArray,
|
|
670
|
-
capabilities: finalInstance.capabilities,
|
|
671
|
-
healthTimer: finalInstance.healthTimer,
|
|
672
|
-
restartTimer: finalInstance.restartTimer,
|
|
673
|
-
metrics: finalInstance.metrics,
|
|
674
|
-
};
|
|
675
|
-
return {
|
|
676
|
-
success: true,
|
|
677
|
-
data: convertedInstance,
|
|
678
|
-
serverId,
|
|
679
|
-
duration: Date.now() - startTime,
|
|
680
|
-
metadata: {
|
|
681
|
-
timestamp: Date.now(),
|
|
682
|
-
operation: "addServer",
|
|
683
|
-
toolsDiscovered: finalInstance.tools.length,
|
|
684
|
-
},
|
|
685
|
-
};
|
|
686
|
-
}
|
|
687
|
-
catch (error) {
|
|
688
|
-
mcpLogger.error(`[ExternalServerManager] Failed to add server ${serverId}:`, error);
|
|
689
|
-
// Clean up if instance was created
|
|
690
|
-
this.servers.delete(serverId);
|
|
691
|
-
return {
|
|
692
|
-
success: false,
|
|
693
|
-
error: error instanceof Error ? error.message : String(error),
|
|
694
|
-
serverId,
|
|
695
|
-
duration: Date.now() - startTime,
|
|
696
|
-
};
|
|
697
|
-
}
|
|
698
|
-
}
|
|
699
|
-
/**
|
|
700
|
-
* Remove an external MCP server
|
|
701
|
-
*/
|
|
702
|
-
async removeServer(serverId) {
|
|
703
|
-
const startTime = Date.now();
|
|
704
|
-
try {
|
|
705
|
-
const instance = this.servers.get(serverId);
|
|
706
|
-
if (!instance) {
|
|
707
|
-
return {
|
|
708
|
-
success: false,
|
|
709
|
-
error: `Server '${serverId}' not found`,
|
|
710
|
-
serverId,
|
|
711
|
-
duration: Date.now() - startTime,
|
|
712
|
-
};
|
|
713
|
-
}
|
|
714
|
-
mcpLogger.info(`[ExternalServerManager] Removing server: ${serverId}`);
|
|
715
|
-
// Stop the server
|
|
716
|
-
await this.stopServer(serverId);
|
|
717
|
-
// Remove from registry
|
|
718
|
-
this.servers.delete(serverId);
|
|
719
|
-
// Emit event
|
|
720
|
-
this.emit("disconnected", {
|
|
721
|
-
serverId,
|
|
722
|
-
reason: "Manually removed",
|
|
723
|
-
timestamp: new Date(),
|
|
724
|
-
});
|
|
725
|
-
return {
|
|
726
|
-
success: true,
|
|
727
|
-
serverId,
|
|
728
|
-
duration: Date.now() - startTime,
|
|
729
|
-
metadata: {
|
|
730
|
-
timestamp: Date.now(),
|
|
731
|
-
operation: "removeServer",
|
|
732
|
-
},
|
|
733
|
-
};
|
|
734
|
-
}
|
|
735
|
-
catch (error) {
|
|
736
|
-
mcpLogger.error(`[ExternalServerManager] Failed to remove server ${serverId}:`, error);
|
|
737
|
-
return {
|
|
738
|
-
success: false,
|
|
739
|
-
error: error instanceof Error ? error.message : String(error),
|
|
740
|
-
serverId,
|
|
741
|
-
duration: Date.now() - startTime,
|
|
742
|
-
};
|
|
743
|
-
}
|
|
744
|
-
}
|
|
745
|
-
/**
|
|
746
|
-
* Start an external MCP server
|
|
747
|
-
*/
|
|
748
|
-
async startServer(serverId) {
|
|
749
|
-
const instance = this.servers.get(serverId);
|
|
750
|
-
if (!instance) {
|
|
751
|
-
throw new Error(`Server '${serverId}' not found`);
|
|
752
|
-
}
|
|
753
|
-
const config = instance.config;
|
|
754
|
-
const span = tracers.mcp.startSpan("neurolink.mcp.server.start", {
|
|
755
|
-
attributes: {
|
|
756
|
-
"mcp.server_id": serverId,
|
|
757
|
-
"mcp.transport": config.transport,
|
|
758
|
-
"mcp.command_name": config.command
|
|
759
|
-
? config.command.split(/[\\/]/).pop() || ""
|
|
760
|
-
: "",
|
|
761
|
-
"mcp.command_present": Boolean(config.command),
|
|
762
|
-
},
|
|
763
|
-
});
|
|
764
|
-
try {
|
|
765
|
-
this.updateServerStatus(serverId, "connecting");
|
|
766
|
-
mcpLogger.debug(`[ExternalServerManager] Starting server: ${serverId}`, {
|
|
767
|
-
command: config.command,
|
|
768
|
-
args: redactSensitiveArgs(config.args),
|
|
769
|
-
transport: config.transport,
|
|
770
|
-
});
|
|
771
|
-
// Create MCP client using the factory
|
|
772
|
-
const clientResult = await MCPClientFactory.createClient(config, config.timeout || this.config.defaultTimeout);
|
|
773
|
-
if (!clientResult.success ||
|
|
774
|
-
!clientResult.client ||
|
|
775
|
-
!clientResult.transport) {
|
|
776
|
-
throw new Error(`Failed to create MCP client: ${clientResult.error}`);
|
|
777
|
-
}
|
|
778
|
-
// Store client components
|
|
779
|
-
instance.client = clientResult.client;
|
|
780
|
-
instance.transportInstance = clientResult.transport;
|
|
781
|
-
instance.process = clientResult.process || null;
|
|
782
|
-
instance.capabilities = safeMetadataConversion(clientResult.capabilities);
|
|
783
|
-
instance.startTime = new Date();
|
|
784
|
-
instance.lastHealthCheck = new Date();
|
|
785
|
-
instance.metrics.totalConnections++;
|
|
786
|
-
// Handle process events if there's a process
|
|
787
|
-
if (instance.process) {
|
|
788
|
-
instance.process.on("error", (error) => {
|
|
789
|
-
mcpLogger.error(`[ExternalServerManager] Process error for ${serverId}:`, error);
|
|
790
|
-
this.handleServerError(serverId, error);
|
|
791
|
-
});
|
|
792
|
-
instance.process.on("exit", (code, signal) => {
|
|
793
|
-
mcpLogger.warn(`[ExternalServerManager] Process exited for ${serverId}`, {
|
|
794
|
-
code,
|
|
795
|
-
signal,
|
|
796
|
-
});
|
|
797
|
-
this.handleServerDisconnection(serverId, `Process exited with code ${code}`);
|
|
798
|
-
});
|
|
799
|
-
// Log stderr for debugging
|
|
800
|
-
instance.process.stderr?.on("data", (data) => {
|
|
801
|
-
const message = data.toString().trim();
|
|
802
|
-
if (message) {
|
|
803
|
-
mcpLogger.debug(`[ExternalServerManager] ${serverId} stderr:`, message);
|
|
804
|
-
}
|
|
805
|
-
});
|
|
806
|
-
}
|
|
807
|
-
this.updateServerStatus(serverId, "connected");
|
|
808
|
-
// Discover tools from the server
|
|
809
|
-
await this.discoverServerTools(serverId);
|
|
810
|
-
// Register tools with main registry if integration is enabled
|
|
811
|
-
if (this.enableMainRegistryIntegration) {
|
|
812
|
-
await this.registerServerToolsWithMainRegistry(serverId);
|
|
813
|
-
}
|
|
814
|
-
// Start health monitoring
|
|
815
|
-
this.startHealthMonitoring(serverId);
|
|
816
|
-
// Emit connected event
|
|
817
|
-
this.emit("connected", {
|
|
818
|
-
serverId,
|
|
819
|
-
toolCount: instance.toolsMap.size,
|
|
820
|
-
timestamp: new Date(),
|
|
821
|
-
});
|
|
822
|
-
span.setAttribute("mcp.tool_count", instance.toolsMap.size);
|
|
823
|
-
span.setStatus({ code: SpanStatusCode.OK });
|
|
824
|
-
mcpLogger.info(`[ExternalServerManager] Server started successfully: ${serverId}`);
|
|
825
|
-
}
|
|
826
|
-
catch (error) {
|
|
827
|
-
mcpLogger.error(`[ExternalServerManager] Failed to start server ${serverId}:`, error);
|
|
828
|
-
this.updateServerStatus(serverId, "failed");
|
|
829
|
-
instance.lastError =
|
|
830
|
-
error instanceof Error ? error.message : String(error);
|
|
831
|
-
span.recordException(error instanceof Error ? error : new Error(String(error)));
|
|
832
|
-
span.setStatus({
|
|
833
|
-
code: SpanStatusCode.ERROR,
|
|
834
|
-
message: error instanceof Error ? error.message : String(error),
|
|
835
|
-
});
|
|
836
|
-
throw error;
|
|
837
|
-
}
|
|
838
|
-
finally {
|
|
839
|
-
span.end();
|
|
840
|
-
}
|
|
841
|
-
}
|
|
842
|
-
/**
|
|
843
|
-
* Stop an external MCP server
|
|
844
|
-
*/
|
|
845
|
-
async stopServer(serverId) {
|
|
846
|
-
const instance = this.servers.get(serverId);
|
|
847
|
-
if (!instance) {
|
|
848
|
-
return;
|
|
849
|
-
}
|
|
850
|
-
const span = tracers.mcp.startSpan("neurolink.mcp.server.stop", {
|
|
851
|
-
attributes: {
|
|
852
|
-
"mcp.server_id": serverId,
|
|
853
|
-
},
|
|
854
|
-
});
|
|
855
|
-
try {
|
|
856
|
-
this.updateServerStatus(serverId, "stopping");
|
|
857
|
-
// Clear timers
|
|
858
|
-
if (instance.healthTimer) {
|
|
859
|
-
clearInterval(instance.healthTimer);
|
|
860
|
-
instance.healthTimer = undefined;
|
|
861
|
-
}
|
|
862
|
-
if (instance.restartTimer) {
|
|
863
|
-
clearTimeout(instance.restartTimer);
|
|
864
|
-
instance.restartTimer = undefined;
|
|
865
|
-
}
|
|
866
|
-
// Unregister tools from main registry if integration is enabled
|
|
867
|
-
if (this.enableMainRegistryIntegration) {
|
|
868
|
-
this.unregisterServerToolsFromMainRegistry(serverId);
|
|
869
|
-
}
|
|
870
|
-
// Clear server tools from discovery service
|
|
871
|
-
this.toolDiscovery.clearServerTools(serverId);
|
|
872
|
-
// Close MCP client using factory cleanup
|
|
873
|
-
if (instance.client && instance.transportInstance) {
|
|
874
|
-
try {
|
|
875
|
-
await MCPClientFactory.closeClient(instance.client, instance.transportInstance, instance.process || undefined);
|
|
876
|
-
}
|
|
877
|
-
catch (error) {
|
|
878
|
-
mcpLogger.debug(`[ExternalServerManager] Error closing client for ${serverId}:`, error);
|
|
879
|
-
}
|
|
880
|
-
instance.client = null;
|
|
881
|
-
instance.transportInstance = null;
|
|
882
|
-
instance.process = null;
|
|
883
|
-
}
|
|
884
|
-
this.updateServerStatus(serverId, "stopped");
|
|
885
|
-
span.setStatus({ code: SpanStatusCode.OK });
|
|
886
|
-
mcpLogger.info(`[ExternalServerManager] Server stopped: ${serverId}`);
|
|
887
|
-
}
|
|
888
|
-
catch (error) {
|
|
889
|
-
mcpLogger.error(`[ExternalServerManager] Error stopping server ${serverId}:`, error);
|
|
890
|
-
this.updateServerStatus(serverId, "failed");
|
|
891
|
-
span.recordException(error instanceof Error ? error : new Error(String(error)));
|
|
892
|
-
span.setStatus({
|
|
893
|
-
code: SpanStatusCode.ERROR,
|
|
894
|
-
message: error instanceof Error ? error.message : String(error),
|
|
895
|
-
});
|
|
896
|
-
}
|
|
897
|
-
finally {
|
|
898
|
-
span.end();
|
|
899
|
-
}
|
|
900
|
-
}
|
|
901
|
-
/**
|
|
902
|
-
* Update server status and emit events
|
|
903
|
-
*/
|
|
904
|
-
updateServerStatus(serverId, newStatus) {
|
|
905
|
-
const instance = this.servers.get(serverId);
|
|
906
|
-
if (!instance) {
|
|
907
|
-
return;
|
|
908
|
-
}
|
|
909
|
-
const oldStatus = instance.status;
|
|
910
|
-
// Map ExternalMCPServerStatus to MCPServerInfo status
|
|
911
|
-
const mappedStatus = newStatus === "connecting" || newStatus === "restarting"
|
|
912
|
-
? "initializing"
|
|
913
|
-
: newStatus === "stopping" || newStatus === "stopped"
|
|
914
|
-
? "stopping"
|
|
915
|
-
: newStatus === "connected"
|
|
916
|
-
? "connected"
|
|
917
|
-
: newStatus === "disconnected"
|
|
918
|
-
? "disconnected"
|
|
919
|
-
: "failed";
|
|
920
|
-
instance.status = mappedStatus;
|
|
921
|
-
// Emit status change event
|
|
922
|
-
this.emit("statusChanged", {
|
|
923
|
-
serverId,
|
|
924
|
-
oldStatus,
|
|
925
|
-
newStatus,
|
|
926
|
-
timestamp: new Date(),
|
|
927
|
-
});
|
|
928
|
-
mcpLogger.debug(`[ExternalServerManager] Status changed for ${serverId}: ${oldStatus} -> ${newStatus}`);
|
|
929
|
-
}
|
|
930
|
-
/**
|
|
931
|
-
* Handle server errors
|
|
932
|
-
*/
|
|
933
|
-
handleServerError(serverId, error) {
|
|
934
|
-
const instance = this.servers.get(serverId);
|
|
935
|
-
if (!instance) {
|
|
936
|
-
return;
|
|
937
|
-
}
|
|
938
|
-
instance.lastError = error.message;
|
|
939
|
-
instance.metrics.totalErrors++;
|
|
940
|
-
mcpLogger.error(`[ExternalServerManager] Server error for ${serverId}:`, error);
|
|
941
|
-
// Emit failed event
|
|
942
|
-
this.emit("failed", {
|
|
943
|
-
serverId,
|
|
944
|
-
error: error.message,
|
|
945
|
-
timestamp: new Date(),
|
|
946
|
-
});
|
|
947
|
-
// Attempt restart if enabled
|
|
948
|
-
if (this.config.enableAutoRestart && !this.isShuttingDown) {
|
|
949
|
-
this.scheduleRestart(serverId);
|
|
950
|
-
}
|
|
951
|
-
else {
|
|
952
|
-
this.updateServerStatus(serverId, "failed");
|
|
953
|
-
}
|
|
954
|
-
}
|
|
955
|
-
/**
|
|
956
|
-
* Handle server disconnection
|
|
957
|
-
*/
|
|
958
|
-
handleServerDisconnection(serverId, reason) {
|
|
959
|
-
const instance = this.servers.get(serverId);
|
|
960
|
-
if (!instance) {
|
|
961
|
-
return;
|
|
962
|
-
}
|
|
963
|
-
instance.metrics.totalDisconnections++;
|
|
964
|
-
mcpLogger.warn(`[ExternalServerManager] Server disconnected ${serverId}: ${reason}`);
|
|
965
|
-
// Emit disconnected event
|
|
966
|
-
this.emit("disconnected", {
|
|
967
|
-
serverId,
|
|
968
|
-
reason,
|
|
969
|
-
timestamp: new Date(),
|
|
970
|
-
});
|
|
971
|
-
// Attempt restart if enabled — prefer server-specific setting, fall back to global
|
|
972
|
-
if ((instance.config.autoRestart ?? this.config.enableAutoRestart) &&
|
|
973
|
-
!this.isShuttingDown) {
|
|
974
|
-
this.scheduleRestart(serverId);
|
|
975
|
-
}
|
|
976
|
-
else {
|
|
977
|
-
this.updateServerStatus(serverId, "disconnected");
|
|
978
|
-
}
|
|
979
|
-
}
|
|
980
|
-
/**
|
|
981
|
-
* Schedule server restart with exponential backoff
|
|
982
|
-
*/
|
|
983
|
-
scheduleRestart(serverId) {
|
|
984
|
-
const instance = this.servers.get(serverId);
|
|
985
|
-
if (!instance) {
|
|
986
|
-
return;
|
|
987
|
-
}
|
|
988
|
-
if (instance.reconnectAttempts >= instance.maxReconnectAttempts) {
|
|
989
|
-
mcpLogger.error(`[ExternalServerManager] Max restart attempts reached for ${serverId}`);
|
|
990
|
-
this.updateServerStatus(serverId, "failed");
|
|
991
|
-
return;
|
|
992
|
-
}
|
|
993
|
-
instance.reconnectAttempts++;
|
|
994
|
-
this.updateServerStatus(serverId, "restarting");
|
|
995
|
-
const delay = Math.min(1000 *
|
|
996
|
-
Math.pow(this.config.restartBackoffMultiplier, instance.reconnectAttempts - 1), 30000);
|
|
997
|
-
mcpLogger.info(`[ExternalServerManager] Scheduling restart for ${serverId} in ${delay}ms (attempt ${instance.reconnectAttempts})`);
|
|
998
|
-
if (instance.restartTimer) {
|
|
999
|
-
return;
|
|
1000
|
-
} // already scheduled
|
|
1001
|
-
instance.restartTimer = setTimeout(async () => {
|
|
1002
|
-
const restartSpan = tracers.mcp.startSpan("neurolink.mcp.server.restart", {
|
|
1003
|
-
attributes: {
|
|
1004
|
-
"mcp.server_id": serverId,
|
|
1005
|
-
"mcp.restart_attempt": instance.reconnectAttempts,
|
|
1006
|
-
"mcp.restart_delay_ms": delay,
|
|
1007
|
-
},
|
|
1008
|
-
});
|
|
1009
|
-
try {
|
|
1010
|
-
await this.stopServer(serverId);
|
|
1011
|
-
await this.startServer(serverId);
|
|
1012
|
-
// Reset restart attempts on successful restart
|
|
1013
|
-
instance.reconnectAttempts = 0;
|
|
1014
|
-
restartSpan.setStatus({ code: SpanStatusCode.OK });
|
|
1015
|
-
}
|
|
1016
|
-
catch (error) {
|
|
1017
|
-
mcpLogger.error(`[ExternalServerManager] Restart failed for ${serverId}:`, error);
|
|
1018
|
-
restartSpan.recordException(error instanceof Error ? error : new Error(String(error)));
|
|
1019
|
-
restartSpan.setStatus({
|
|
1020
|
-
code: SpanStatusCode.ERROR,
|
|
1021
|
-
message: error instanceof Error ? error.message : String(error),
|
|
1022
|
-
});
|
|
1023
|
-
this.scheduleRestart(serverId); // Try again
|
|
1024
|
-
}
|
|
1025
|
-
finally {
|
|
1026
|
-
restartSpan.end();
|
|
1027
|
-
}
|
|
1028
|
-
}, delay);
|
|
1029
|
-
}
|
|
1030
|
-
/**
|
|
1031
|
-
* Start health monitoring for a server
|
|
1032
|
-
*/
|
|
1033
|
-
startHealthMonitoring(serverId) {
|
|
1034
|
-
const instance = this.servers.get(serverId);
|
|
1035
|
-
if (!instance || !this.config.enablePerformanceMonitoring) {
|
|
1036
|
-
return;
|
|
1037
|
-
}
|
|
1038
|
-
const interval = instance.config.healthCheckInterval ??
|
|
1039
|
-
this.config.defaultHealthCheckInterval;
|
|
1040
|
-
instance.healthTimer = setInterval(async () => {
|
|
1041
|
-
await this.performHealthCheck(serverId);
|
|
1042
|
-
}, interval);
|
|
1043
|
-
}
|
|
1044
|
-
/**
|
|
1045
|
-
* Perform health check on a server
|
|
1046
|
-
*/
|
|
1047
|
-
async performHealthCheck(serverId) {
|
|
1048
|
-
const instance = this.servers.get(serverId);
|
|
1049
|
-
if (!instance || instance.status !== "connected") {
|
|
1050
|
-
return;
|
|
1051
|
-
}
|
|
1052
|
-
const startTime = Date.now();
|
|
1053
|
-
try {
|
|
1054
|
-
// For now, simple process check
|
|
1055
|
-
let isHealthy = true;
|
|
1056
|
-
const issues = [];
|
|
1057
|
-
if (instance.process && instance.process.killed) {
|
|
1058
|
-
isHealthy = false;
|
|
1059
|
-
issues.push("Process is killed");
|
|
1060
|
-
}
|
|
1061
|
-
const responseTime = Date.now() - startTime;
|
|
1062
|
-
instance.lastHealthCheck = new Date();
|
|
1063
|
-
const health = {
|
|
1064
|
-
serverId,
|
|
1065
|
-
isHealthy,
|
|
1066
|
-
status: instance.status,
|
|
1067
|
-
checkedAt: new Date(),
|
|
1068
|
-
responseTime,
|
|
1069
|
-
toolCount: instance.toolsMap.size,
|
|
1070
|
-
issues,
|
|
1071
|
-
performance: {
|
|
1072
|
-
uptime: instance.startTime
|
|
1073
|
-
? Date.now() - instance.startTime.getTime()
|
|
1074
|
-
: 0,
|
|
1075
|
-
averageResponseTime: instance.metrics.averageResponseTime,
|
|
1076
|
-
},
|
|
1077
|
-
};
|
|
1078
|
-
// Emit health check event
|
|
1079
|
-
this.emit("healthCheck", {
|
|
1080
|
-
serverId,
|
|
1081
|
-
health,
|
|
1082
|
-
timestamp: new Date(),
|
|
1083
|
-
});
|
|
1084
|
-
if (!isHealthy) {
|
|
1085
|
-
mcpLogger.warn(`[ExternalServerManager] Health check failed for ${serverId}:`, issues);
|
|
1086
|
-
this.handleServerError(serverId, new Error(`Health check failed: ${issues.join(", ")}`));
|
|
1087
|
-
}
|
|
1088
|
-
}
|
|
1089
|
-
catch (error) {
|
|
1090
|
-
mcpLogger.error(`[ExternalServerManager] Health check error for ${serverId}:`, error);
|
|
1091
|
-
this.handleServerError(serverId, error instanceof Error ? error : new Error(String(error)));
|
|
1092
|
-
}
|
|
1093
|
-
}
|
|
1094
|
-
/**
|
|
1095
|
-
* Get server instance - converted to ExternalMCPServerInstance for compatibility
|
|
1096
|
-
*/
|
|
1097
|
-
getServer(serverId) {
|
|
1098
|
-
const runtime = this.servers.get(serverId);
|
|
1099
|
-
if (!runtime) {
|
|
1100
|
-
return undefined;
|
|
1101
|
-
}
|
|
1102
|
-
return {
|
|
1103
|
-
config: runtime.config,
|
|
1104
|
-
process: runtime.process,
|
|
1105
|
-
client: runtime.client,
|
|
1106
|
-
transport: runtime.transportInstance,
|
|
1107
|
-
status: runtime.status,
|
|
1108
|
-
lastError: runtime.lastError,
|
|
1109
|
-
startTime: runtime.startTime,
|
|
1110
|
-
lastHealthCheck: runtime.lastHealthCheck,
|
|
1111
|
-
reconnectAttempts: runtime.reconnectAttempts,
|
|
1112
|
-
maxReconnectAttempts: runtime.maxReconnectAttempts,
|
|
1113
|
-
tools: runtime.toolsMap,
|
|
1114
|
-
toolsArray: runtime.toolsArray,
|
|
1115
|
-
capabilities: runtime.capabilities,
|
|
1116
|
-
healthTimer: runtime.healthTimer,
|
|
1117
|
-
restartTimer: runtime.restartTimer,
|
|
1118
|
-
metrics: runtime.metrics,
|
|
1119
|
-
};
|
|
1120
|
-
}
|
|
1121
|
-
/**
|
|
1122
|
-
* Get all servers - converted to ExternalMCPServerInstance for compatibility
|
|
1123
|
-
*/
|
|
1124
|
-
getAllServers() {
|
|
1125
|
-
const converted = new Map();
|
|
1126
|
-
for (const [serverId, runtime] of this.servers.entries()) {
|
|
1127
|
-
converted.set(serverId, {
|
|
1128
|
-
config: runtime.config,
|
|
1129
|
-
process: runtime.process,
|
|
1130
|
-
client: runtime.client,
|
|
1131
|
-
transport: runtime.transportInstance,
|
|
1132
|
-
status: runtime.status,
|
|
1133
|
-
lastError: runtime.lastError,
|
|
1134
|
-
startTime: runtime.startTime,
|
|
1135
|
-
lastHealthCheck: runtime.lastHealthCheck,
|
|
1136
|
-
reconnectAttempts: runtime.reconnectAttempts,
|
|
1137
|
-
maxReconnectAttempts: runtime.maxReconnectAttempts,
|
|
1138
|
-
tools: runtime.toolsMap,
|
|
1139
|
-
toolsArray: runtime.toolsArray,
|
|
1140
|
-
capabilities: runtime.capabilities,
|
|
1141
|
-
healthTimer: runtime.healthTimer,
|
|
1142
|
-
restartTimer: runtime.restartTimer,
|
|
1143
|
-
metrics: runtime.metrics,
|
|
1144
|
-
});
|
|
1145
|
-
}
|
|
1146
|
-
return converted;
|
|
1147
|
-
}
|
|
1148
|
-
/**
|
|
1149
|
-
* List servers as MCPServerInfo - ZERO conversion needed
|
|
1150
|
-
*/
|
|
1151
|
-
listServers() {
|
|
1152
|
-
return Array.from(this.servers.values());
|
|
1153
|
-
}
|
|
1154
|
-
/**
|
|
1155
|
-
* Get server statuses
|
|
1156
|
-
*/
|
|
1157
|
-
getServerStatuses() {
|
|
1158
|
-
const statuses = [];
|
|
1159
|
-
for (const [serverId, instance] of Array.from(this.servers.entries())) {
|
|
1160
|
-
const uptime = instance.startTime
|
|
1161
|
-
? Date.now() - instance.startTime.getTime()
|
|
1162
|
-
: 0;
|
|
1163
|
-
statuses.push({
|
|
1164
|
-
serverId,
|
|
1165
|
-
isHealthy: instance.status === "connected",
|
|
1166
|
-
status: instance.status,
|
|
1167
|
-
checkedAt: instance.lastHealthCheck || new Date(),
|
|
1168
|
-
toolCount: instance.toolsMap.size,
|
|
1169
|
-
issues: instance.lastError ? [instance.lastError] : [],
|
|
1170
|
-
performance: {
|
|
1171
|
-
uptime,
|
|
1172
|
-
averageResponseTime: instance.metrics.averageResponseTime,
|
|
1173
|
-
},
|
|
1174
|
-
});
|
|
1175
|
-
}
|
|
1176
|
-
return statuses;
|
|
1177
|
-
}
|
|
1178
|
-
/**
|
|
1179
|
-
* Shutdown all servers and clean up resources
|
|
1180
|
-
* This method should be called during application shutdown to prevent memory leaks
|
|
1181
|
-
*/
|
|
1182
|
-
async shutdown() {
|
|
1183
|
-
if (this.isShuttingDown) {
|
|
1184
|
-
return;
|
|
1185
|
-
}
|
|
1186
|
-
this.isShuttingDown = true;
|
|
1187
|
-
mcpLogger.info("[ExternalServerManager] Shutting down all servers...");
|
|
1188
|
-
const shutdownPromises = Array.from(this.servers.keys()).map((serverId) => this.stopServer(serverId).catch((error) => {
|
|
1189
|
-
mcpLogger.error(`[ExternalServerManager] Error shutting down ${serverId}:`, error);
|
|
1190
|
-
}));
|
|
1191
|
-
await Promise.all(shutdownPromises);
|
|
1192
|
-
this.servers.clear();
|
|
1193
|
-
// Clean up the tool discovery service to prevent memory leaks
|
|
1194
|
-
// from accumulated event listeners
|
|
1195
|
-
this.toolDiscovery.destroy();
|
|
1196
|
-
// Remove all event listeners from this manager
|
|
1197
|
-
this.removeAllListeners();
|
|
1198
|
-
mcpLogger.info("[ExternalServerManager] All servers shut down and resources cleaned up");
|
|
1199
|
-
}
|
|
1200
|
-
/**
|
|
1201
|
-
* Destroy the manager and all associated resources
|
|
1202
|
-
* Alias for shutdown() to match the pattern used by other components
|
|
1203
|
-
*/
|
|
1204
|
-
async destroy() {
|
|
1205
|
-
return this.shutdown();
|
|
1206
|
-
}
|
|
1207
|
-
/**
|
|
1208
|
-
* Get manager statistics
|
|
1209
|
-
*/
|
|
1210
|
-
getStatistics() {
|
|
1211
|
-
let connectedServers = 0;
|
|
1212
|
-
let failedServers = 0;
|
|
1213
|
-
let totalTools = 0;
|
|
1214
|
-
let totalConnections = 0;
|
|
1215
|
-
let totalErrors = 0;
|
|
1216
|
-
for (const instance of Array.from(this.servers.values())) {
|
|
1217
|
-
if (instance.status === "connected") {
|
|
1218
|
-
connectedServers++;
|
|
1219
|
-
}
|
|
1220
|
-
else if (instance.status === "failed") {
|
|
1221
|
-
failedServers++;
|
|
1222
|
-
}
|
|
1223
|
-
totalTools += instance.toolsMap.size;
|
|
1224
|
-
totalConnections += instance.metrics.totalConnections;
|
|
1225
|
-
totalErrors += instance.metrics.totalErrors;
|
|
1226
|
-
}
|
|
1227
|
-
return {
|
|
1228
|
-
totalServers: this.servers.size,
|
|
1229
|
-
connectedServers,
|
|
1230
|
-
failedServers,
|
|
1231
|
-
totalTools,
|
|
1232
|
-
totalConnections,
|
|
1233
|
-
totalErrors,
|
|
1234
|
-
};
|
|
1235
|
-
}
|
|
1236
|
-
/**
|
|
1237
|
-
* Discover tools from a server
|
|
1238
|
-
*/
|
|
1239
|
-
async discoverServerTools(serverId) {
|
|
1240
|
-
const instance = this.servers.get(serverId);
|
|
1241
|
-
if (!instance || !instance.client) {
|
|
1242
|
-
throw new Error(`Server '${serverId}' not found or not connected`);
|
|
1243
|
-
}
|
|
1244
|
-
try {
|
|
1245
|
-
mcpLogger.debug(`[ExternalServerManager] Discovering tools for server: ${serverId}`);
|
|
1246
|
-
const discoveryResult = await this.toolDiscovery.discoverTools(serverId, instance.client, instance.config.timeout || this.config.defaultTimeout);
|
|
1247
|
-
if (discoveryResult.success) {
|
|
1248
|
-
instance.toolsMap.clear();
|
|
1249
|
-
instance.toolsArray = undefined;
|
|
1250
|
-
instance.tools = [];
|
|
1251
|
-
const blockedTools = instance.blockedTools || [];
|
|
1252
|
-
let blockedCount = 0;
|
|
1253
|
-
for (const tool of discoveryResult.tools) {
|
|
1254
|
-
// Check if tool is blocked
|
|
1255
|
-
if (blockedTools.includes(tool.name)) {
|
|
1256
|
-
mcpLogger.info(`[ExternalServerManager] Blocking tool '${tool.name}' from server '${serverId}' (configured in blockedTools)`);
|
|
1257
|
-
blockedCount++;
|
|
1258
|
-
continue; // Skip blocked tools
|
|
1259
|
-
}
|
|
1260
|
-
instance.toolsMap.set(tool.name, tool);
|
|
1261
|
-
instance.tools.push({
|
|
1262
|
-
name: tool.name,
|
|
1263
|
-
description: tool.description,
|
|
1264
|
-
inputSchema: tool.inputSchema,
|
|
1265
|
-
});
|
|
1266
|
-
}
|
|
1267
|
-
mcpLogger.info(`[ExternalServerManager] Discovered ${discoveryResult.toolCount} tools for ${serverId} (${blockedCount} blocked, ${instance.toolsMap.size} available)`);
|
|
1268
|
-
}
|
|
1269
|
-
else {
|
|
1270
|
-
mcpLogger.warn(`[ExternalServerManager] Tool discovery failed for ${serverId}: ${discoveryResult.error}`);
|
|
1271
|
-
}
|
|
1272
|
-
}
|
|
1273
|
-
catch (error) {
|
|
1274
|
-
mcpLogger.error(`[ExternalServerManager] Tool discovery error for ${serverId}:`, error);
|
|
1275
|
-
}
|
|
1276
|
-
}
|
|
1277
|
-
/**
|
|
1278
|
-
* Register server tools with main tool registry for unified access
|
|
1279
|
-
* This enables external MCP tools to be accessed via the main toolRegistry.executeTool()
|
|
1280
|
-
*/
|
|
1281
|
-
async registerServerToolsWithMainRegistry(serverId) {
|
|
1282
|
-
const instance = this.servers.get(serverId);
|
|
1283
|
-
if (!instance) {
|
|
1284
|
-
throw new Error(`Server '${serverId}' not found`);
|
|
1285
|
-
}
|
|
1286
|
-
try {
|
|
1287
|
-
mcpLogger.debug(`[ExternalServerManager] Registering ${instance.toolsMap.size} tools with main registry for server: ${serverId}`);
|
|
1288
|
-
const registrations = [];
|
|
1289
|
-
for (const [toolName, tool] of instance.toolsMap.entries()) {
|
|
1290
|
-
const toolId = `${serverId}.${toolName}`;
|
|
1291
|
-
const toolInfo = {
|
|
1292
|
-
name: toolName,
|
|
1293
|
-
description: tool.description || toolName,
|
|
1294
|
-
inputSchema: tool.inputSchema || {},
|
|
1295
|
-
serverId: serverId,
|
|
1296
|
-
category: detectCategory({ isExternal: true, serverId }),
|
|
1297
|
-
};
|
|
1298
|
-
try {
|
|
1299
|
-
registrations.push(toolRegistry.registerTool(toolId, toolInfo, {
|
|
1300
|
-
execute: async (params, _context) => {
|
|
1301
|
-
// Execute tool via ExternalServerManager for proper lifecycle management
|
|
1302
|
-
return await this.executeTool(serverId, toolName, params, {
|
|
1303
|
-
timeout: instance.config.timeout || this.config.defaultTimeout,
|
|
1304
|
-
});
|
|
1305
|
-
},
|
|
1306
|
-
}));
|
|
1307
|
-
mcpLogger.debug(`[ExternalServerManager] Registered tool with main registry: ${toolId}`);
|
|
1308
|
-
}
|
|
1309
|
-
catch (registrationError) {
|
|
1310
|
-
mcpLogger.warn(`[ExternalServerManager] Failed to register tool ${toolId} with main registry:`, registrationError);
|
|
1311
|
-
}
|
|
1312
|
-
}
|
|
1313
|
-
const results = await Promise.allSettled(registrations);
|
|
1314
|
-
const ok = results.filter((r) => r.status === "fulfilled").length;
|
|
1315
|
-
const failed = results.length - ok;
|
|
1316
|
-
mcpLogger.info(`[ExternalServerManager] Registered ${ok}/${results.length} tools with main registry for ${serverId}${failed ? ` (${failed} failed)` : ""}`);
|
|
1317
|
-
}
|
|
1318
|
-
catch (error) {
|
|
1319
|
-
mcpLogger.error(`[ExternalServerManager] Failed to register tools with main registry for ${serverId}:`, error);
|
|
1320
|
-
}
|
|
1321
|
-
}
|
|
1322
|
-
/**
|
|
1323
|
-
* Unregister server tools from main tool registry
|
|
1324
|
-
*/
|
|
1325
|
-
unregisterServerToolsFromMainRegistry(serverId) {
|
|
1326
|
-
const instance = this.servers.get(serverId);
|
|
1327
|
-
if (!instance || !this.enableMainRegistryIntegration) {
|
|
1328
|
-
return;
|
|
1329
|
-
}
|
|
1330
|
-
try {
|
|
1331
|
-
mcpLogger.debug(`[ExternalServerManager] Unregistering tools from main registry for server: ${serverId}`);
|
|
1332
|
-
for (const [toolName] of instance.toolsMap.entries()) {
|
|
1333
|
-
const toolId = `${serverId}.${toolName}`;
|
|
1334
|
-
try {
|
|
1335
|
-
toolRegistry.removeTool(toolId);
|
|
1336
|
-
mcpLogger.debug(`[ExternalServerManager] Unregistered tool from main registry: ${toolId}`);
|
|
1337
|
-
}
|
|
1338
|
-
catch (error) {
|
|
1339
|
-
mcpLogger.debug(`[ExternalServerManager] Failed to unregister tool ${toolId}:`, error);
|
|
1340
|
-
}
|
|
1341
|
-
}
|
|
1342
|
-
mcpLogger.debug(`[ExternalServerManager] Completed unregistering tools from main registry for ${serverId}`);
|
|
1343
|
-
}
|
|
1344
|
-
catch (error) {
|
|
1345
|
-
mcpLogger.error(`[ExternalServerManager] Error unregistering tools from main registry for ${serverId}:`, error);
|
|
1346
|
-
}
|
|
1347
|
-
}
|
|
1348
|
-
/**
|
|
1349
|
-
* Execute a tool on a specific server
|
|
1350
|
-
*/
|
|
1351
|
-
async executeTool(serverId, toolName, parameters, options) {
|
|
1352
|
-
const instance = this.servers.get(serverId);
|
|
1353
|
-
if (!instance) {
|
|
1354
|
-
throw new Error(`Server '${serverId}' not found`);
|
|
1355
|
-
}
|
|
1356
|
-
if (!instance.client) {
|
|
1357
|
-
throw new Error(`Server '${serverId}' is not connected`);
|
|
1358
|
-
}
|
|
1359
|
-
if (instance.status !== "connected") {
|
|
1360
|
-
throw new Error(`Server '${serverId}' is not in connected state: ${instance.status}`);
|
|
1361
|
-
}
|
|
1362
|
-
// Check if tool is blocked
|
|
1363
|
-
const blockedTools = instance.blockedTools || [];
|
|
1364
|
-
if (blockedTools.includes(toolName)) {
|
|
1365
|
-
throw new Error(`Tool '${toolName}' is blocked on server '${serverId}' by configuration`);
|
|
1366
|
-
}
|
|
1367
|
-
const startTime = Date.now();
|
|
1368
|
-
try {
|
|
1369
|
-
// HITL Safety Check: Request confirmation if required
|
|
1370
|
-
let finalParameters = parameters;
|
|
1371
|
-
if (this.hitlManager && this.hitlManager.isEnabled()) {
|
|
1372
|
-
const requiresConfirmation = this.hitlManager.requiresConfirmation(toolName, parameters);
|
|
1373
|
-
if (requiresConfirmation) {
|
|
1374
|
-
mcpLogger.info(`[ExternalServerManager] External tool '${toolName}' on server '${serverId}' requires HITL confirmation`);
|
|
1375
|
-
try {
|
|
1376
|
-
const confirmationResult = await this.hitlManager.requestConfirmation(toolName, parameters, {
|
|
1377
|
-
serverId: serverId,
|
|
1378
|
-
sessionId: `external-${serverId}-${Date.now()}`,
|
|
1379
|
-
userId: undefined, // External tools don't have user context by default
|
|
1380
|
-
});
|
|
1381
|
-
if (!confirmationResult.approved) {
|
|
1382
|
-
// User rejected the tool execution
|
|
1383
|
-
throw new HITLUserRejectedError(`External tool execution rejected by user: ${confirmationResult.reason || "No reason provided"}`, toolName, confirmationResult.reason);
|
|
1384
|
-
}
|
|
1385
|
-
// User approved - use modified arguments if provided
|
|
1386
|
-
if (confirmationResult.modifiedArguments !== undefined) {
|
|
1387
|
-
finalParameters =
|
|
1388
|
-
confirmationResult.modifiedArguments;
|
|
1389
|
-
mcpLogger.info(`[ExternalServerManager] External tool '${toolName}' arguments modified by user`);
|
|
1390
|
-
}
|
|
1391
|
-
mcpLogger.info(`[ExternalServerManager] External tool '${toolName}' approved for execution (response time: ${confirmationResult.responseTime}ms)`);
|
|
1392
|
-
}
|
|
1393
|
-
catch (error) {
|
|
1394
|
-
if (error instanceof HITLTimeoutError) {
|
|
1395
|
-
// Timeout occurred - user didn't respond in time
|
|
1396
|
-
mcpLogger.warn(`[ExternalServerManager] External tool '${toolName}' execution timed out waiting for user confirmation`);
|
|
1397
|
-
throw error;
|
|
1398
|
-
}
|
|
1399
|
-
else if (error instanceof HITLUserRejectedError) {
|
|
1400
|
-
// User explicitly rejected
|
|
1401
|
-
mcpLogger.info(`[ExternalServerManager] External tool '${toolName}' execution rejected by user`);
|
|
1402
|
-
throw error;
|
|
1403
|
-
}
|
|
1404
|
-
else {
|
|
1405
|
-
// Other HITL error (configuration, system error, etc.)
|
|
1406
|
-
mcpLogger.error(`[ExternalServerManager] HITL confirmation failed for external tool '${toolName}':`, error);
|
|
1407
|
-
throw new Error(`HITL confirmation failed: ${error instanceof Error ? error.message : String(error)}`, { cause: error });
|
|
1408
|
-
}
|
|
1409
|
-
}
|
|
1410
|
-
}
|
|
1411
|
-
else {
|
|
1412
|
-
mcpLogger.debug(`[ExternalServerManager] External tool '${toolName}' does not require HITL confirmation`);
|
|
1413
|
-
}
|
|
1414
|
-
}
|
|
1415
|
-
// Execute tool through discovery service (with potentially modified parameters)
|
|
1416
|
-
const result = await this.toolDiscovery.executeTool(toolName, serverId, instance.client, finalParameters, {
|
|
1417
|
-
timeout: options?.timeout ||
|
|
1418
|
-
instance.config.timeout ||
|
|
1419
|
-
this.config.defaultTimeout,
|
|
1420
|
-
});
|
|
1421
|
-
const duration = Date.now() - startTime;
|
|
1422
|
-
// Update metrics
|
|
1423
|
-
instance.metrics.totalToolCalls++;
|
|
1424
|
-
instance.metrics.lastResponseTime = duration;
|
|
1425
|
-
// Update average response time
|
|
1426
|
-
const totalTime = instance.metrics.averageResponseTime *
|
|
1427
|
-
(instance.metrics.totalToolCalls - 1) +
|
|
1428
|
-
duration;
|
|
1429
|
-
instance.metrics.averageResponseTime =
|
|
1430
|
-
totalTime / instance.metrics.totalToolCalls;
|
|
1431
|
-
if (result.success) {
|
|
1432
|
-
mcpLogger.debug(`[ExternalServerManager] Tool executed successfully: ${toolName} on ${serverId}`, {
|
|
1433
|
-
duration,
|
|
1434
|
-
});
|
|
1435
|
-
try {
|
|
1436
|
-
TelemetryService.getInstance()?.recordMCPToolCall(toolName, duration, true);
|
|
1437
|
-
}
|
|
1438
|
-
catch {
|
|
1439
|
-
/* telemetry should not break execution */
|
|
1440
|
-
}
|
|
1441
|
-
return result.data;
|
|
1442
|
-
}
|
|
1443
|
-
else {
|
|
1444
|
-
throw new Error(result.error || "Tool execution failed");
|
|
1445
|
-
}
|
|
1446
|
-
}
|
|
1447
|
-
catch (error) {
|
|
1448
|
-
instance.metrics.totalErrors++;
|
|
1449
|
-
try {
|
|
1450
|
-
const errorDuration = Date.now() - startTime;
|
|
1451
|
-
TelemetryService.getInstance()?.recordMCPToolCall(toolName, errorDuration, false);
|
|
1452
|
-
}
|
|
1453
|
-
catch {
|
|
1454
|
-
/* telemetry should not break execution */
|
|
1455
|
-
}
|
|
1456
|
-
mcpLogger.error(`[ExternalServerManager] Tool execution failed: ${toolName} on ${serverId}`, error);
|
|
1457
|
-
throw error;
|
|
1458
|
-
}
|
|
1459
|
-
}
|
|
1460
|
-
/**
|
|
1461
|
-
* Get all tools from all servers
|
|
1462
|
-
*/
|
|
1463
|
-
getAllTools() {
|
|
1464
|
-
return this.toolDiscovery.getAllTools();
|
|
1465
|
-
}
|
|
1466
|
-
/**
|
|
1467
|
-
* Get tools for a specific server
|
|
1468
|
-
*/
|
|
1469
|
-
getServerTools(serverId) {
|
|
1470
|
-
return this.toolDiscovery.getServerTools(serverId);
|
|
1471
|
-
}
|
|
1472
|
-
/**
|
|
1473
|
-
* Get tool discovery service
|
|
1474
|
-
*/
|
|
1475
|
-
getToolDiscovery() {
|
|
1476
|
-
return this.toolDiscovery;
|
|
1477
|
-
}
|
|
1478
|
-
}
|