@juspay/neurolink 9.32.0 → 9.33.0
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/anthropicOAuth.js +1 -1
- package/dist/cli/commands/proxy.js +18 -5
- package/dist/client/aiSdkAdapter.js +1 -1
- package/dist/client/index.js +137 -501
- package/dist/core/factory.js +0 -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/lib/neurolink.d.ts +10 -0
- package/dist/lib/neurolink.js +41 -7
- package/dist/lib/server/routes/claudeProxyRoutes.js +45 -9
- package/dist/lib/types/generateTypes.d.ts +16 -0
- package/dist/lib/types/streamTypes.d.ts +15 -0
- package/dist/mcp/elicitationProtocol.js +1 -1
- package/dist/mcp/servers/agent/directToolsServer.js +0 -1
- package/dist/neurolink.d.ts +10 -0
- package/dist/neurolink.js +41 -7
- package/dist/providers/azureOpenai.js +1 -1
- package/dist/providers/huggingFace.js +0 -1
- package/dist/providers/openaiCompatible.js +0 -1
- package/dist/sdk/toolRegistration.js +0 -1
- package/dist/server/openapi/generator.js +1 -1
- package/dist/server/routes/claudeProxyRoutes.js +45 -9
- package/dist/types/configTypes.js +0 -5
- package/dist/types/generateTypes.d.ts +16 -0
- package/dist/types/modelTypes.js +0 -1
- package/dist/types/streamTypes.d.ts +15 -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 +2 -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/AuthProviderFactory.js +0 -111
- package/dist/client/auth/AuthProviderRegistry.js +0 -190
- package/dist/client/auth/RequestContext.js +0 -78
- package/dist/client/auth/accountPool.js +0 -178
- package/dist/client/auth/anthropicOAuth.js +0 -974
- package/dist/client/auth/authContext.js +0 -314
- package/dist/client/auth/errors.js +0 -39
- package/dist/client/auth/index.js +0 -61
- package/dist/client/auth/middleware/AuthMiddleware.js +0 -519
- package/dist/client/auth/middleware/rateLimitByUser.js +0 -554
- package/dist/client/auth/providers/BaseAuthProvider.js +0 -723
- package/dist/client/auth/providers/CognitoProvider.js +0 -304
- package/dist/client/auth/providers/KeycloakProvider.js +0 -393
- package/dist/client/auth/providers/auth0.js +0 -274
- package/dist/client/auth/providers/betterAuth.js +0 -182
- package/dist/client/auth/providers/clerk.js +0 -317
- package/dist/client/auth/providers/custom.js +0 -112
- package/dist/client/auth/providers/firebase.js +0 -226
- package/dist/client/auth/providers/jwt.js +0 -212
- package/dist/client/auth/providers/oauth2.js +0 -303
- package/dist/client/auth/providers/supabase.js +0 -259
- package/dist/client/auth/providers/workos.js +0 -284
- package/dist/client/auth/serverBridge.js +0 -25
- package/dist/client/auth/sessionManager.js +0 -437
- 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 -53
- 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 -729
- 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 -8233
- 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 -189
- 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 -9
- 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,708 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* MCP Client Factory
|
|
3
|
-
* Creates and manages MCP clients for external servers
|
|
4
|
-
* Supports stdio, SSE, WebSocket, and HTTP transports
|
|
5
|
-
* Enhanced with retry, rate limiting, and OAuth 2.1 support
|
|
6
|
-
*/
|
|
7
|
-
import { Client } from "@modelcontextprotocol/sdk/client/index.js";
|
|
8
|
-
import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
|
|
9
|
-
import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
|
|
10
|
-
import { WebSocketClientTransport } from "@modelcontextprotocol/sdk/client/websocket.js";
|
|
11
|
-
import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
|
|
12
|
-
import { spawn } from "child_process";
|
|
13
|
-
import { mcpLogger } from "../utils/logger.js";
|
|
14
|
-
import { globalCircuitBreakerManager } from "./mcpCircuitBreaker.js";
|
|
15
|
-
import { CircuitBreakerOpenError } from "../types/circuitBreakerErrors.js";
|
|
16
|
-
import { withHTTPRetry, DEFAULT_HTTP_RETRY_CONFIG, } from "./httpRetryHandler.js";
|
|
17
|
-
import { globalRateLimiterManager } from "./httpRateLimiter.js";
|
|
18
|
-
import { NeuroLinkOAuthProvider, InMemoryTokenStorage } from "./auth/index.js";
|
|
19
|
-
import { SpanSerializer, SpanType, SpanStatus, } from "../observability/index.js";
|
|
20
|
-
import { getMetricsAggregator } from "../observability/index.js";
|
|
21
|
-
/**
|
|
22
|
-
* Default timeout for MCP client creation in milliseconds.
|
|
23
|
-
* Configurable via MCP_CLIENT_TIMEOUT env var.
|
|
24
|
-
* Covers process spawn, transport setup, connection, and handshake.
|
|
25
|
-
* Set to 60s to accommodate stdio servers that may be slow to start,
|
|
26
|
-
* especially when multiple MCP servers are started concurrently.
|
|
27
|
-
*/
|
|
28
|
-
const DEFAULT_CLIENT_TIMEOUT = Math.max(5000, Number(process.env.MCP_CLIENT_TIMEOUT) || 60000);
|
|
29
|
-
/**
|
|
30
|
-
* MCPClientFactory
|
|
31
|
-
* Factory class for creating MCP clients with different transports
|
|
32
|
-
*/
|
|
33
|
-
export class MCPClientFactory {
|
|
34
|
-
static NEUROLINK_IMPLEMENTATION = {
|
|
35
|
-
name: "neurolink-sdk",
|
|
36
|
-
version: "1.0.0",
|
|
37
|
-
};
|
|
38
|
-
static DEFAULT_CAPABILITIES = {
|
|
39
|
-
sampling: {},
|
|
40
|
-
roots: {
|
|
41
|
-
listChanged: false,
|
|
42
|
-
},
|
|
43
|
-
};
|
|
44
|
-
/**
|
|
45
|
-
* Create an MCP client for the given server configuration
|
|
46
|
-
* Enhanced with retry logic, rate limiting, and circuit breaker protection
|
|
47
|
-
*/
|
|
48
|
-
static async createClient(config, timeout = DEFAULT_CLIENT_TIMEOUT) {
|
|
49
|
-
const startTime = Date.now();
|
|
50
|
-
const obsSpan = SpanSerializer.createSpan(SpanType.MCP_TRANSPORT, "mcp.connect", {
|
|
51
|
-
"mcp.transport": config.transport,
|
|
52
|
-
"mcp.operation": "connect",
|
|
53
|
-
"mcp.server_id": config.id,
|
|
54
|
-
});
|
|
55
|
-
try {
|
|
56
|
-
mcpLogger.info(`[MCPClientFactory] Creating client for ${config.id}`, {
|
|
57
|
-
transport: config.transport,
|
|
58
|
-
command: config.command,
|
|
59
|
-
hasRetryConfig: !!config.retryConfig,
|
|
60
|
-
hasRateLimiting: !!config.rateLimiting,
|
|
61
|
-
hasAuth: !!config.auth,
|
|
62
|
-
});
|
|
63
|
-
// Acquire rate limit token if rate limiting is configured for HTTP transport
|
|
64
|
-
if ((config.transport === "http" || config.transport === "sse") &&
|
|
65
|
-
config.rateLimiting) {
|
|
66
|
-
const rateLimiter = globalRateLimiterManager.getLimiter(config.id, {
|
|
67
|
-
requestsPerWindow: config.rateLimiting.requestsPerMinute ?? 60,
|
|
68
|
-
windowMs: 60000,
|
|
69
|
-
maxBurst: config.rateLimiting.maxBurst ?? 10,
|
|
70
|
-
useTokenBucket: config.rateLimiting.useTokenBucket ?? true,
|
|
71
|
-
refillRate: (config.rateLimiting.requestsPerMinute ?? 60) / 60,
|
|
72
|
-
});
|
|
73
|
-
await rateLimiter.acquire();
|
|
74
|
-
mcpLogger.debug(`[MCPClientFactory] Rate limit token acquired for ${config.id}`);
|
|
75
|
-
}
|
|
76
|
-
// Create circuit breaker for this server
|
|
77
|
-
const circuitBreaker = globalCircuitBreakerManager.getBreaker(`mcp-client-${config.id}`, {
|
|
78
|
-
failureThreshold: 3,
|
|
79
|
-
resetTimeout: 30000,
|
|
80
|
-
operationTimeout: timeout,
|
|
81
|
-
});
|
|
82
|
-
// Define the client creation operation
|
|
83
|
-
const createClientOperation = async () => {
|
|
84
|
-
return await circuitBreaker.execute(async () => {
|
|
85
|
-
return await this.createClientInternal(config, timeout);
|
|
86
|
-
});
|
|
87
|
-
};
|
|
88
|
-
// Wrap with retry logic if retry config is provided for HTTP transport
|
|
89
|
-
let result;
|
|
90
|
-
if ((config.transport === "http" || config.transport === "sse") &&
|
|
91
|
-
config.retryConfig) {
|
|
92
|
-
mcpLogger.debug(`[MCPClientFactory] Using retry logic for ${config.id}`, {
|
|
93
|
-
maxAttempts: config.retryConfig.maxAttempts ??
|
|
94
|
-
DEFAULT_HTTP_RETRY_CONFIG.maxAttempts,
|
|
95
|
-
});
|
|
96
|
-
result = await withHTTPRetry(createClientOperation, {
|
|
97
|
-
maxAttempts: config.retryConfig.maxAttempts ??
|
|
98
|
-
DEFAULT_HTTP_RETRY_CONFIG.maxAttempts,
|
|
99
|
-
initialDelay: config.retryConfig.initialDelay ??
|
|
100
|
-
DEFAULT_HTTP_RETRY_CONFIG.initialDelay,
|
|
101
|
-
maxDelay: config.retryConfig.maxDelay ?? DEFAULT_HTTP_RETRY_CONFIG.maxDelay,
|
|
102
|
-
backoffMultiplier: config.retryConfig.backoffMultiplier ??
|
|
103
|
-
DEFAULT_HTTP_RETRY_CONFIG.backoffMultiplier,
|
|
104
|
-
});
|
|
105
|
-
}
|
|
106
|
-
else {
|
|
107
|
-
result = await createClientOperation();
|
|
108
|
-
}
|
|
109
|
-
mcpLogger.info(`[MCPClientFactory] Client created successfully for ${config.id}`, {
|
|
110
|
-
duration: Date.now() - startTime,
|
|
111
|
-
capabilities: result.capabilities,
|
|
112
|
-
});
|
|
113
|
-
obsSpan.durationMs = Date.now() - startTime;
|
|
114
|
-
const endedObsSpan = SpanSerializer.endSpan(obsSpan, SpanStatus.OK);
|
|
115
|
-
getMetricsAggregator().recordSpan(endedObsSpan);
|
|
116
|
-
return {
|
|
117
|
-
...result,
|
|
118
|
-
success: true,
|
|
119
|
-
duration: Date.now() - startTime,
|
|
120
|
-
};
|
|
121
|
-
}
|
|
122
|
-
catch (error) {
|
|
123
|
-
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
124
|
-
// Circuit breaker open: log at warn (not error) since this is expected
|
|
125
|
-
// protection behavior, and preserve the structured metadata.
|
|
126
|
-
if (error instanceof CircuitBreakerOpenError) {
|
|
127
|
-
mcpLogger.warn(`[MCPClientFactory] Client creation blocked by circuit breaker for ${config.id}`, {
|
|
128
|
-
serverId: config.id,
|
|
129
|
-
breakerState: error.breakerState,
|
|
130
|
-
retryAfter: error.retryAfter,
|
|
131
|
-
retryAfterMs: error.retryAfterMs,
|
|
132
|
-
failureCount: error.failureCount,
|
|
133
|
-
});
|
|
134
|
-
obsSpan.durationMs = Date.now() - startTime;
|
|
135
|
-
const endedObsSpan = SpanSerializer.endSpan(obsSpan, SpanStatus.ERROR);
|
|
136
|
-
endedObsSpan.statusMessage = `Circuit breaker open: ${errorMessage}`;
|
|
137
|
-
getMetricsAggregator().recordSpan(endedObsSpan);
|
|
138
|
-
return {
|
|
139
|
-
success: false,
|
|
140
|
-
error: errorMessage,
|
|
141
|
-
duration: Date.now() - startTime,
|
|
142
|
-
};
|
|
143
|
-
}
|
|
144
|
-
mcpLogger.error(`[MCPClientFactory] Failed to create client for ${config.id}:`, error);
|
|
145
|
-
obsSpan.durationMs = Date.now() - startTime;
|
|
146
|
-
const endedObsSpan = SpanSerializer.endSpan(obsSpan, SpanStatus.ERROR);
|
|
147
|
-
endedObsSpan.statusMessage = errorMessage;
|
|
148
|
-
getMetricsAggregator().recordSpan(endedObsSpan);
|
|
149
|
-
return {
|
|
150
|
-
success: false,
|
|
151
|
-
error: errorMessage,
|
|
152
|
-
duration: Date.now() - startTime,
|
|
153
|
-
};
|
|
154
|
-
}
|
|
155
|
-
}
|
|
156
|
-
/**
|
|
157
|
-
* Internal client creation logic
|
|
158
|
-
*/
|
|
159
|
-
static async createClientInternal(config, timeout) {
|
|
160
|
-
// Create transport
|
|
161
|
-
const transportResult = await this.createTransport(config);
|
|
162
|
-
// Extract transport and process with necessary type assertions
|
|
163
|
-
// Note: Type assertions required due to TransportResult using 'unknown' to avoid circular imports
|
|
164
|
-
const transport = transportResult.transport;
|
|
165
|
-
const process = transportResult.process;
|
|
166
|
-
try {
|
|
167
|
-
// Create client
|
|
168
|
-
const client = new Client(this.NEUROLINK_IMPLEMENTATION, {
|
|
169
|
-
capabilities: this.DEFAULT_CAPABILITIES,
|
|
170
|
-
});
|
|
171
|
-
// Connect with timeout
|
|
172
|
-
await Promise.race([
|
|
173
|
-
client.connect(transport),
|
|
174
|
-
this.createTimeoutPromise(timeout, `Client connection timeout for ${config.id}`),
|
|
175
|
-
]);
|
|
176
|
-
// Perform handshake to get server capabilities
|
|
177
|
-
const serverCapabilities = await this.performHandshake(client, timeout);
|
|
178
|
-
mcpLogger.debug(`[MCPClientFactory] Handshake completed for ${config.id}`, {
|
|
179
|
-
capabilities: serverCapabilities,
|
|
180
|
-
});
|
|
181
|
-
return {
|
|
182
|
-
client,
|
|
183
|
-
transport,
|
|
184
|
-
process,
|
|
185
|
-
capabilities: serverCapabilities,
|
|
186
|
-
};
|
|
187
|
-
}
|
|
188
|
-
catch (error) {
|
|
189
|
-
// Clean up on failure
|
|
190
|
-
try {
|
|
191
|
-
await transport.close();
|
|
192
|
-
}
|
|
193
|
-
catch (closeError) {
|
|
194
|
-
mcpLogger.debug(`[MCPClientFactory] Error closing transport during cleanup:`, closeError);
|
|
195
|
-
}
|
|
196
|
-
if (process && !process.killed) {
|
|
197
|
-
process.kill("SIGTERM");
|
|
198
|
-
}
|
|
199
|
-
throw error;
|
|
200
|
-
}
|
|
201
|
-
}
|
|
202
|
-
/**
|
|
203
|
-
* Create transport based on configuration
|
|
204
|
-
*/
|
|
205
|
-
static async createTransport(config) {
|
|
206
|
-
switch (config.transport) {
|
|
207
|
-
case "stdio":
|
|
208
|
-
return this.createStdioTransport(config);
|
|
209
|
-
case "sse":
|
|
210
|
-
return this.createSSETransport(config);
|
|
211
|
-
case "websocket":
|
|
212
|
-
return this.createWebSocketTransport(config);
|
|
213
|
-
case "http":
|
|
214
|
-
return this.createHTTPTransport(config);
|
|
215
|
-
default:
|
|
216
|
-
throw new Error(`Unsupported transport type: ${config.transport}`);
|
|
217
|
-
}
|
|
218
|
-
}
|
|
219
|
-
/**
|
|
220
|
-
* Create stdio transport with process spawning
|
|
221
|
-
*/
|
|
222
|
-
static async createStdioTransport(config) {
|
|
223
|
-
mcpLogger.debug(`[MCPClientFactory] Creating stdio transport for ${config.id}`, {
|
|
224
|
-
command: config.command,
|
|
225
|
-
args: config.args,
|
|
226
|
-
});
|
|
227
|
-
// Validate command is present
|
|
228
|
-
if (!config.command) {
|
|
229
|
-
throw new Error(`Command is required for stdio transport`);
|
|
230
|
-
}
|
|
231
|
-
// Spawn the process
|
|
232
|
-
const childProcess = spawn(config.command, config.args || [], {
|
|
233
|
-
stdio: ["pipe", "pipe", "pipe"],
|
|
234
|
-
env: Object.fromEntries(Object.entries({
|
|
235
|
-
...process.env,
|
|
236
|
-
...config.env,
|
|
237
|
-
})
|
|
238
|
-
.filter(([, value]) => value !== undefined)
|
|
239
|
-
.map(([k, v]) => [k, String(v)])),
|
|
240
|
-
cwd: config.cwd,
|
|
241
|
-
});
|
|
242
|
-
// Handle process errors
|
|
243
|
-
const processErrorPromise = new Promise((_, reject) => {
|
|
244
|
-
childProcess.on("error", (error) => {
|
|
245
|
-
reject(new Error(`Process spawn error: ${error.message}`));
|
|
246
|
-
});
|
|
247
|
-
childProcess.on("exit", (code, signal) => {
|
|
248
|
-
if (code !== 0) {
|
|
249
|
-
reject(new Error(`Process exited with code ${code}, signal ${signal}`));
|
|
250
|
-
}
|
|
251
|
-
});
|
|
252
|
-
});
|
|
253
|
-
// Wait for process to be ready or fail using AbortController for better async patterns
|
|
254
|
-
const processStartupController = new AbortController();
|
|
255
|
-
const processStartupTimeout = setTimeout(() => {
|
|
256
|
-
processStartupController.abort();
|
|
257
|
-
}, 1000);
|
|
258
|
-
try {
|
|
259
|
-
await Promise.race([
|
|
260
|
-
new Promise((resolve) => {
|
|
261
|
-
const checkReady = () => {
|
|
262
|
-
if (processStartupController.signal.aborted) {
|
|
263
|
-
resolve(); // Timeout reached, continue
|
|
264
|
-
}
|
|
265
|
-
else {
|
|
266
|
-
setTimeout(checkReady, 100);
|
|
267
|
-
}
|
|
268
|
-
};
|
|
269
|
-
checkReady();
|
|
270
|
-
}),
|
|
271
|
-
processErrorPromise,
|
|
272
|
-
]);
|
|
273
|
-
}
|
|
274
|
-
finally {
|
|
275
|
-
clearTimeout(processStartupTimeout);
|
|
276
|
-
}
|
|
277
|
-
// Check if process is still running
|
|
278
|
-
if (childProcess.killed || childProcess.exitCode !== null) {
|
|
279
|
-
throw new Error("Process failed to start or exited immediately");
|
|
280
|
-
}
|
|
281
|
-
// Create transport
|
|
282
|
-
if (!config.command) {
|
|
283
|
-
throw new Error(`Command is required for stdio transport`);
|
|
284
|
-
}
|
|
285
|
-
const transport = new StdioClientTransport({
|
|
286
|
-
command: config.command,
|
|
287
|
-
args: config.args || [],
|
|
288
|
-
env: Object.fromEntries(Object.entries({
|
|
289
|
-
...process.env,
|
|
290
|
-
...config.env,
|
|
291
|
-
})
|
|
292
|
-
.filter(([, value]) => value !== undefined)
|
|
293
|
-
.map(([key, value]) => [key, String(value)])),
|
|
294
|
-
cwd: config.cwd,
|
|
295
|
-
stderr: "ignore", // Suppress MCP server startup messages
|
|
296
|
-
});
|
|
297
|
-
return { transport, process: childProcess };
|
|
298
|
-
}
|
|
299
|
-
/**
|
|
300
|
-
* Create SSE transport
|
|
301
|
-
*/
|
|
302
|
-
static async createSSETransport(config) {
|
|
303
|
-
if (!config.url) {
|
|
304
|
-
throw new Error("URL is required for SSE transport");
|
|
305
|
-
}
|
|
306
|
-
mcpLogger.debug(`[MCPClientFactory] Creating SSE transport for ${config.id}`, {
|
|
307
|
-
url: config.url,
|
|
308
|
-
});
|
|
309
|
-
try {
|
|
310
|
-
const url = new URL(config.url);
|
|
311
|
-
const transport = new SSEClientTransport(url);
|
|
312
|
-
return { transport };
|
|
313
|
-
}
|
|
314
|
-
catch (error) {
|
|
315
|
-
throw new Error(`Invalid SSE URL: ${error instanceof Error ? error.message : String(error)}`, { cause: error });
|
|
316
|
-
}
|
|
317
|
-
}
|
|
318
|
-
/**
|
|
319
|
-
* Create WebSocket transport
|
|
320
|
-
*/
|
|
321
|
-
static async createWebSocketTransport(config) {
|
|
322
|
-
if (!config.url) {
|
|
323
|
-
throw new Error("URL is required for WebSocket transport");
|
|
324
|
-
}
|
|
325
|
-
mcpLogger.debug(`[MCPClientFactory] Creating WebSocket transport for ${config.id}`, {
|
|
326
|
-
url: config.url,
|
|
327
|
-
});
|
|
328
|
-
try {
|
|
329
|
-
const url = new URL(config.url);
|
|
330
|
-
const transport = new WebSocketClientTransport(url);
|
|
331
|
-
return { transport };
|
|
332
|
-
}
|
|
333
|
-
catch (error) {
|
|
334
|
-
throw new Error(`Invalid WebSocket URL: ${error instanceof Error ? error.message : String(error)}`, { cause: error });
|
|
335
|
-
}
|
|
336
|
-
}
|
|
337
|
-
/**
|
|
338
|
-
* Create HTTP transport (Streamable HTTP)
|
|
339
|
-
* Enhanced with OAuth 2.1, rate limiting, and configurable timeouts
|
|
340
|
-
*/
|
|
341
|
-
static async createHTTPTransport(config) {
|
|
342
|
-
if (!config.url) {
|
|
343
|
-
throw new Error("URL is required for HTTP transport");
|
|
344
|
-
}
|
|
345
|
-
// Extract HTTP options with defaults
|
|
346
|
-
const httpOptions = {
|
|
347
|
-
connectionTimeout: config.httpOptions?.connectionTimeout ?? 30000,
|
|
348
|
-
requestTimeout: config.httpOptions?.requestTimeout ?? 60000,
|
|
349
|
-
idleTimeout: config.httpOptions?.idleTimeout ?? 120000,
|
|
350
|
-
keepAliveTimeout: config.httpOptions?.keepAliveTimeout ?? 30000,
|
|
351
|
-
};
|
|
352
|
-
mcpLogger.debug(`[MCPClientFactory] Creating HTTP transport for ${config.id}`, {
|
|
353
|
-
url: config.url,
|
|
354
|
-
hasHeaders: !!config.headers,
|
|
355
|
-
hasAuth: !!config.auth,
|
|
356
|
-
authType: config.auth?.type,
|
|
357
|
-
httpOptions,
|
|
358
|
-
});
|
|
359
|
-
try {
|
|
360
|
-
const url = new URL(config.url);
|
|
361
|
-
// Set up OAuth provider if configured
|
|
362
|
-
const oauthProvider = await this.setupAuthProvider(config);
|
|
363
|
-
// Build headers including authentication
|
|
364
|
-
const headers = {
|
|
365
|
-
...(config.headers ?? {}),
|
|
366
|
-
};
|
|
367
|
-
// Add authentication headers based on auth type
|
|
368
|
-
if (config.auth) {
|
|
369
|
-
const authHeader = await this.getAuthorizationHeader(config, oauthProvider);
|
|
370
|
-
if (authHeader) {
|
|
371
|
-
headers["Authorization"] = authHeader;
|
|
372
|
-
}
|
|
373
|
-
}
|
|
374
|
-
// Create custom fetch wrapper with timeout and rate limiting support
|
|
375
|
-
const fetchWithEnhancements = this.createEnhancedFetch(config, httpOptions.requestTimeout, oauthProvider);
|
|
376
|
-
// Create request init with custom headers
|
|
377
|
-
const requestInit = {
|
|
378
|
-
headers: Object.keys(headers).length > 0 ? headers : undefined,
|
|
379
|
-
};
|
|
380
|
-
const transport = new StreamableHTTPClientTransport(url, {
|
|
381
|
-
requestInit,
|
|
382
|
-
fetch: fetchWithEnhancements,
|
|
383
|
-
});
|
|
384
|
-
return { transport };
|
|
385
|
-
}
|
|
386
|
-
catch (error) {
|
|
387
|
-
throw new Error(`Invalid HTTP URL: ${error instanceof Error ? error.message : String(error)}`, { cause: error });
|
|
388
|
-
}
|
|
389
|
-
}
|
|
390
|
-
/**
|
|
391
|
-
* Create a fetch wrapper with timeout support
|
|
392
|
-
*/
|
|
393
|
-
static createFetchWithTimeout(timeoutMs) {
|
|
394
|
-
return async (input, init) => {
|
|
395
|
-
const controller = new AbortController();
|
|
396
|
-
const timeoutId = setTimeout(() => controller.abort(), timeoutMs);
|
|
397
|
-
try {
|
|
398
|
-
return await fetch(input, { ...init, signal: controller.signal });
|
|
399
|
-
}
|
|
400
|
-
finally {
|
|
401
|
-
clearTimeout(timeoutId);
|
|
402
|
-
}
|
|
403
|
-
};
|
|
404
|
-
}
|
|
405
|
-
/**
|
|
406
|
-
* Create an enhanced fetch function with timeout and optional retry
|
|
407
|
-
*/
|
|
408
|
-
static createEnhancedFetch(config, timeoutMs, oauthProvider) {
|
|
409
|
-
const fetchWithTimeout = this.createFetchWithTimeout(timeoutMs);
|
|
410
|
-
return async (input, init) => {
|
|
411
|
-
// If OAuth is configured, ensure we have valid tokens
|
|
412
|
-
if (oauthProvider && config.auth?.type === "oauth2") {
|
|
413
|
-
try {
|
|
414
|
-
const authHeader = await oauthProvider.getAuthorizationHeader(config.id);
|
|
415
|
-
if (authHeader) {
|
|
416
|
-
const existingHeaders = init?.headers ?? {};
|
|
417
|
-
const headers = new Headers(existingHeaders);
|
|
418
|
-
headers.set("Authorization", authHeader);
|
|
419
|
-
init = {
|
|
420
|
-
...init,
|
|
421
|
-
headers,
|
|
422
|
-
};
|
|
423
|
-
}
|
|
424
|
-
}
|
|
425
|
-
catch (error) {
|
|
426
|
-
mcpLogger.warn(`[MCPClientFactory] OAuth token refresh failed for ${config.id}:`, error instanceof Error ? error.message : String(error));
|
|
427
|
-
// Continue without auth - let the request fail naturally
|
|
428
|
-
}
|
|
429
|
-
}
|
|
430
|
-
return fetchWithTimeout(input, init);
|
|
431
|
-
};
|
|
432
|
-
}
|
|
433
|
-
/**
|
|
434
|
-
* Set up OAuth provider if configured
|
|
435
|
-
*/
|
|
436
|
-
static async setupAuthProvider(config) {
|
|
437
|
-
if (config.auth?.type === "oauth2" && config.auth.oauth) {
|
|
438
|
-
const tokenStorage = new InMemoryTokenStorage();
|
|
439
|
-
const oauthConfig = {
|
|
440
|
-
clientId: config.auth.oauth.clientId,
|
|
441
|
-
clientSecret: config.auth.oauth.clientSecret,
|
|
442
|
-
authorizationUrl: config.auth.oauth.authorizationUrl,
|
|
443
|
-
tokenUrl: config.auth.oauth.tokenUrl,
|
|
444
|
-
redirectUrl: config.auth.oauth.redirectUrl,
|
|
445
|
-
scope: config.auth.oauth.scope,
|
|
446
|
-
usePKCE: config.auth.oauth.usePKCE ?? true,
|
|
447
|
-
};
|
|
448
|
-
const provider = new NeuroLinkOAuthProvider(oauthConfig, tokenStorage);
|
|
449
|
-
mcpLogger.debug(`[MCPClientFactory] OAuth provider created for ${config.id}`, {
|
|
450
|
-
clientId: oauthConfig.clientId,
|
|
451
|
-
usePKCE: oauthConfig.usePKCE,
|
|
452
|
-
});
|
|
453
|
-
return provider;
|
|
454
|
-
}
|
|
455
|
-
return undefined;
|
|
456
|
-
}
|
|
457
|
-
/**
|
|
458
|
-
* Get authorization header based on auth configuration
|
|
459
|
-
*/
|
|
460
|
-
static async getAuthorizationHeader(config, oauthProvider) {
|
|
461
|
-
if (!config.auth) {
|
|
462
|
-
return undefined;
|
|
463
|
-
}
|
|
464
|
-
switch (config.auth.type) {
|
|
465
|
-
case "oauth2":
|
|
466
|
-
if (oauthProvider) {
|
|
467
|
-
const header = await oauthProvider.getAuthorizationHeader(config.id);
|
|
468
|
-
return header ?? undefined;
|
|
469
|
-
}
|
|
470
|
-
return undefined;
|
|
471
|
-
case "bearer":
|
|
472
|
-
if (config.auth.token) {
|
|
473
|
-
return `Bearer ${config.auth.token}`;
|
|
474
|
-
}
|
|
475
|
-
return undefined;
|
|
476
|
-
case "api-key":
|
|
477
|
-
// API key is typically sent as a custom header, not Authorization
|
|
478
|
-
// But if needed, we can return it here
|
|
479
|
-
return undefined;
|
|
480
|
-
default:
|
|
481
|
-
return undefined;
|
|
482
|
-
}
|
|
483
|
-
}
|
|
484
|
-
/**
|
|
485
|
-
* Perform MCP handshake and get server capabilities
|
|
486
|
-
*/
|
|
487
|
-
static async performHandshake(client, timeout) {
|
|
488
|
-
try {
|
|
489
|
-
// The MCP SDK handles the handshake automatically during connect()
|
|
490
|
-
// We can request server info to verify the connection
|
|
491
|
-
const serverInfo = await Promise.race([
|
|
492
|
-
this.getServerInfo(client),
|
|
493
|
-
this.createTimeoutPromise(timeout, "Handshake timeout"),
|
|
494
|
-
]);
|
|
495
|
-
// Extract capabilities from server info
|
|
496
|
-
return this.extractCapabilities(serverInfo);
|
|
497
|
-
}
|
|
498
|
-
catch (error) {
|
|
499
|
-
mcpLogger.warn("[MCPClientFactory] Handshake failed, but connection may still be valid:", error);
|
|
500
|
-
// Return default capabilities if handshake fails
|
|
501
|
-
// The connection might still work for basic operations
|
|
502
|
-
return this.DEFAULT_CAPABILITIES;
|
|
503
|
-
}
|
|
504
|
-
}
|
|
505
|
-
/**
|
|
506
|
-
* Get server information
|
|
507
|
-
*/
|
|
508
|
-
static async getServerInfo(client) {
|
|
509
|
-
try {
|
|
510
|
-
// Try to list tools to verify server is responding
|
|
511
|
-
const toolsResult = await client.listTools();
|
|
512
|
-
return {
|
|
513
|
-
tools: toolsResult.tools || [],
|
|
514
|
-
capabilities: this.DEFAULT_CAPABILITIES,
|
|
515
|
-
};
|
|
516
|
-
}
|
|
517
|
-
catch {
|
|
518
|
-
// If listing tools fails, try a simpler ping
|
|
519
|
-
mcpLogger.debug("[MCPClientFactory] Tool listing failed, server may not support tools yet");
|
|
520
|
-
return {
|
|
521
|
-
tools: [],
|
|
522
|
-
capabilities: this.DEFAULT_CAPABILITIES,
|
|
523
|
-
};
|
|
524
|
-
}
|
|
525
|
-
}
|
|
526
|
-
/**
|
|
527
|
-
* Extract capabilities from server info
|
|
528
|
-
*/
|
|
529
|
-
static extractCapabilities(serverInfo) {
|
|
530
|
-
// For now, return default capabilities
|
|
531
|
-
// This can be enhanced when MCP servers provide more detailed capability info
|
|
532
|
-
return {
|
|
533
|
-
...this.DEFAULT_CAPABILITIES,
|
|
534
|
-
...(serverInfo.tools ? { tools: {} } : {}),
|
|
535
|
-
};
|
|
536
|
-
}
|
|
537
|
-
/**
|
|
538
|
-
* Create a timeout promise with AbortController support
|
|
539
|
-
* Provides consistent async timeout patterns across the factory
|
|
540
|
-
*/
|
|
541
|
-
static createTimeoutPromise(timeout, message, abortSignal) {
|
|
542
|
-
return new Promise((_, reject) => {
|
|
543
|
-
const timeoutId = setTimeout(() => {
|
|
544
|
-
reject(new Error(message));
|
|
545
|
-
}, timeout);
|
|
546
|
-
// Support abortion for better async cleanup
|
|
547
|
-
if (abortSignal) {
|
|
548
|
-
abortSignal.addEventListener("abort", () => {
|
|
549
|
-
clearTimeout(timeoutId);
|
|
550
|
-
reject(new Error(`Operation aborted: ${message}`));
|
|
551
|
-
});
|
|
552
|
-
}
|
|
553
|
-
});
|
|
554
|
-
}
|
|
555
|
-
/**
|
|
556
|
-
* Close an MCP client and clean up resources
|
|
557
|
-
*/
|
|
558
|
-
static async closeClient(client, transport, process) {
|
|
559
|
-
const errors = [];
|
|
560
|
-
// Close client
|
|
561
|
-
try {
|
|
562
|
-
await client.close();
|
|
563
|
-
}
|
|
564
|
-
catch (error) {
|
|
565
|
-
errors.push(`Client close error: ${error instanceof Error ? error.message : String(error)}`);
|
|
566
|
-
}
|
|
567
|
-
// Close transport
|
|
568
|
-
try {
|
|
569
|
-
await transport.close();
|
|
570
|
-
}
|
|
571
|
-
catch (error) {
|
|
572
|
-
errors.push(`Transport close error: ${error instanceof Error ? error.message : String(error)}`);
|
|
573
|
-
}
|
|
574
|
-
// Kill process if exists with proper async cleanup
|
|
575
|
-
if (process && !process.killed) {
|
|
576
|
-
try {
|
|
577
|
-
process.kill("SIGTERM");
|
|
578
|
-
// Use Promise-based approach for force kill timeout
|
|
579
|
-
await new Promise((resolve) => {
|
|
580
|
-
const forceKillTimeout = setTimeout(() => {
|
|
581
|
-
if (!process.killed) {
|
|
582
|
-
mcpLogger.warn("[MCPClientFactory] Force killing process");
|
|
583
|
-
try {
|
|
584
|
-
process.kill("SIGKILL");
|
|
585
|
-
}
|
|
586
|
-
catch (killError) {
|
|
587
|
-
mcpLogger.debug("[MCPClientFactory] Error in force kill:", killError);
|
|
588
|
-
}
|
|
589
|
-
}
|
|
590
|
-
resolve();
|
|
591
|
-
}, 5000);
|
|
592
|
-
// If process exits gracefully before timeout, clear the force kill
|
|
593
|
-
process.on("exit", () => {
|
|
594
|
-
clearTimeout(forceKillTimeout);
|
|
595
|
-
resolve();
|
|
596
|
-
});
|
|
597
|
-
});
|
|
598
|
-
}
|
|
599
|
-
catch (error) {
|
|
600
|
-
errors.push(`Process kill error: ${error instanceof Error ? error.message : String(error)}`);
|
|
601
|
-
}
|
|
602
|
-
}
|
|
603
|
-
if (errors.length > 0) {
|
|
604
|
-
mcpLogger.warn("[MCPClientFactory] Errors during client cleanup:", errors);
|
|
605
|
-
}
|
|
606
|
-
}
|
|
607
|
-
/**
|
|
608
|
-
* Test connection to an MCP server
|
|
609
|
-
*/
|
|
610
|
-
static async testConnection(config, timeout = 5000) {
|
|
611
|
-
let client;
|
|
612
|
-
let transport;
|
|
613
|
-
let process;
|
|
614
|
-
try {
|
|
615
|
-
const result = await this.createClient(config, timeout);
|
|
616
|
-
if (!result.success) {
|
|
617
|
-
return { success: false, error: result.error };
|
|
618
|
-
}
|
|
619
|
-
client = result.client;
|
|
620
|
-
transport = result.transport;
|
|
621
|
-
process = result.process;
|
|
622
|
-
// Try to list tools as a connectivity test
|
|
623
|
-
if (client) {
|
|
624
|
-
try {
|
|
625
|
-
await client.listTools();
|
|
626
|
-
}
|
|
627
|
-
catch {
|
|
628
|
-
// Tool listing failure doesn't necessarily mean connection failure
|
|
629
|
-
mcpLogger.debug("[MCPClientFactory] Tool listing failed during test, but connection may be valid");
|
|
630
|
-
}
|
|
631
|
-
}
|
|
632
|
-
return {
|
|
633
|
-
success: true,
|
|
634
|
-
capabilities: result.capabilities,
|
|
635
|
-
};
|
|
636
|
-
}
|
|
637
|
-
catch (error) {
|
|
638
|
-
return {
|
|
639
|
-
success: false,
|
|
640
|
-
error: error instanceof Error ? error.message : String(error),
|
|
641
|
-
};
|
|
642
|
-
}
|
|
643
|
-
finally {
|
|
644
|
-
// Clean up test connection
|
|
645
|
-
if (client && transport) {
|
|
646
|
-
try {
|
|
647
|
-
await this.closeClient(client, transport, process);
|
|
648
|
-
}
|
|
649
|
-
catch (error) {
|
|
650
|
-
mcpLogger.debug("[MCPClientFactory] Error cleaning up test connection:", error);
|
|
651
|
-
}
|
|
652
|
-
}
|
|
653
|
-
}
|
|
654
|
-
}
|
|
655
|
-
/**
|
|
656
|
-
* Validate MCP server configuration for client creation
|
|
657
|
-
*/
|
|
658
|
-
static validateClientConfig(config) {
|
|
659
|
-
const errors = [];
|
|
660
|
-
// Basic validation
|
|
661
|
-
if (!config.command) {
|
|
662
|
-
errors.push("Command is required");
|
|
663
|
-
}
|
|
664
|
-
if (!config.transport) {
|
|
665
|
-
errors.push("Transport is required");
|
|
666
|
-
}
|
|
667
|
-
if (!["stdio", "sse", "websocket", "http"].includes(config.transport)) {
|
|
668
|
-
errors.push("Transport must be stdio, sse, websocket, or http");
|
|
669
|
-
}
|
|
670
|
-
// Transport-specific validation
|
|
671
|
-
if (config.transport === "sse" ||
|
|
672
|
-
config.transport === "websocket" ||
|
|
673
|
-
config.transport === "http") {
|
|
674
|
-
if (!config.url) {
|
|
675
|
-
errors.push(`URL is required for ${config.transport} transport`);
|
|
676
|
-
}
|
|
677
|
-
else {
|
|
678
|
-
try {
|
|
679
|
-
new URL(config.url);
|
|
680
|
-
}
|
|
681
|
-
catch {
|
|
682
|
-
errors.push(`Invalid URL for ${config.transport} transport`);
|
|
683
|
-
}
|
|
684
|
-
}
|
|
685
|
-
}
|
|
686
|
-
if (config.transport === "stdio") {
|
|
687
|
-
if (!Array.isArray(config.args)) {
|
|
688
|
-
errors.push("Args array is required for stdio transport");
|
|
689
|
-
}
|
|
690
|
-
}
|
|
691
|
-
return {
|
|
692
|
-
isValid: errors.length === 0,
|
|
693
|
-
errors,
|
|
694
|
-
};
|
|
695
|
-
}
|
|
696
|
-
/**
|
|
697
|
-
* Get supported transport types
|
|
698
|
-
*/
|
|
699
|
-
static getSupportedTransports() {
|
|
700
|
-
return ["stdio", "sse", "websocket", "http"];
|
|
701
|
-
}
|
|
702
|
-
/**
|
|
703
|
-
* Get default client capabilities
|
|
704
|
-
*/
|
|
705
|
-
static getDefaultCapabilities() {
|
|
706
|
-
return { ...this.DEFAULT_CAPABILITIES };
|
|
707
|
-
}
|
|
708
|
-
}
|