@llmindset/hf-mcp-server 0.1.16
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/LICENSE +21 -0
- package/components.json +21 -0
- package/dist/server/application.d.ts +30 -0
- package/dist/server/application.d.ts.map +1 -0
- package/dist/server/application.js +131 -0
- package/dist/server/application.js.map +1 -0
- package/dist/server/gradio-endpoint-connector.d.ts +46 -0
- package/dist/server/gradio-endpoint-connector.d.ts.map +1 -0
- package/dist/server/gradio-endpoint-connector.js +348 -0
- package/dist/server/gradio-endpoint-connector.js.map +1 -0
- package/dist/server/gradio-endpoint-connector.test.d.ts +2 -0
- package/dist/server/gradio-endpoint-connector.test.d.ts.map +1 -0
- package/dist/server/gradio-endpoint-connector.test.js +313 -0
- package/dist/server/gradio-endpoint-connector.test.js.map +1 -0
- package/dist/server/lib/logger.d.ts +5 -0
- package/dist/server/lib/logger.d.ts.map +1 -0
- package/dist/server/lib/logger.js +45 -0
- package/dist/server/lib/logger.js.map +1 -0
- package/dist/server/lib/mcp-api-client.d.ts +42 -0
- package/dist/server/lib/mcp-api-client.d.ts.map +1 -0
- package/dist/server/lib/mcp-api-client.js +188 -0
- package/dist/server/lib/mcp-api-client.js.map +1 -0
- package/dist/server/lib/tool-selection-strategy.d.ts +29 -0
- package/dist/server/lib/tool-selection-strategy.d.ts.map +1 -0
- package/dist/server/lib/tool-selection-strategy.js +113 -0
- package/dist/server/lib/tool-selection-strategy.js.map +1 -0
- package/dist/server/lib/tool-selection-strategy.test.d.ts +2 -0
- package/dist/server/lib/tool-selection-strategy.test.d.ts.map +1 -0
- package/dist/server/lib/tool-selection-strategy.test.js +407 -0
- package/dist/server/lib/tool-selection-strategy.test.js.map +1 -0
- package/dist/server/mcp-proxy.d.ts +5 -0
- package/dist/server/mcp-proxy.d.ts.map +1 -0
- package/dist/server/mcp-proxy.js +66 -0
- package/dist/server/mcp-proxy.js.map +1 -0
- package/dist/server/mcp-server.d.ts +8 -0
- package/dist/server/mcp-server.d.ts.map +1 -0
- package/dist/server/mcp-server.js +219 -0
- package/dist/server/mcp-server.js.map +1 -0
- package/dist/server/sse.d.ts +3 -0
- package/dist/server/sse.d.ts.map +1 -0
- package/dist/server/sse.js +47 -0
- package/dist/server/sse.js.map +1 -0
- package/dist/server/stdio.d.ts +3 -0
- package/dist/server/stdio.d.ts.map +1 -0
- package/dist/server/stdio.js +49 -0
- package/dist/server/stdio.js.map +1 -0
- package/dist/server/streamableHttp.d.ts +3 -0
- package/dist/server/streamableHttp.d.ts.map +1 -0
- package/dist/server/streamableHttp.js +53 -0
- package/dist/server/streamableHttp.js.map +1 -0
- package/dist/server/transport/base-transport.d.ts +132 -0
- package/dist/server/transport/base-transport.d.ts.map +1 -0
- package/dist/server/transport/base-transport.js +390 -0
- package/dist/server/transport/base-transport.js.map +1 -0
- package/dist/server/transport/json-rpc-errors.d.ts +36 -0
- package/dist/server/transport/json-rpc-errors.d.ts.map +1 -0
- package/dist/server/transport/json-rpc-errors.js +53 -0
- package/dist/server/transport/json-rpc-errors.js.map +1 -0
- package/dist/server/transport/sse-transport.d.ts +17 -0
- package/dist/server/transport/sse-transport.d.ts.map +1 -0
- package/dist/server/transport/sse-transport.js +170 -0
- package/dist/server/transport/sse-transport.js.map +1 -0
- package/dist/server/transport/stateless-http-transport.d.ts +11 -0
- package/dist/server/transport/stateless-http-transport.d.ts.map +1 -0
- package/dist/server/transport/stateless-http-transport.js +172 -0
- package/dist/server/transport/stateless-http-transport.js.map +1 -0
- package/dist/server/transport/stateless-http-transport.test.d.ts +2 -0
- package/dist/server/transport/stateless-http-transport.test.d.ts.map +1 -0
- package/dist/server/transport/stateless-http-transport.test.js +58 -0
- package/dist/server/transport/stateless-http-transport.test.js.map +1 -0
- package/dist/server/transport/stdio-transport.d.ts +12 -0
- package/dist/server/transport/stdio-transport.d.ts.map +1 -0
- package/dist/server/transport/stdio-transport.js +78 -0
- package/dist/server/transport/stdio-transport.js.map +1 -0
- package/dist/server/transport/streamable-http-transport.d.ts +21 -0
- package/dist/server/transport/streamable-http-transport.d.ts.map +1 -0
- package/dist/server/transport/streamable-http-transport.js +220 -0
- package/dist/server/transport/streamable-http-transport.js.map +1 -0
- package/dist/server/transport/transport-factory.d.ts +5 -0
- package/dist/server/transport/transport-factory.d.ts.map +1 -0
- package/dist/server/transport/transport-factory.js +21 -0
- package/dist/server/transport/transport-factory.js.map +1 -0
- package/dist/server/utils/api-metrics.d.ts +15 -0
- package/dist/server/utils/api-metrics.d.ts.map +1 -0
- package/dist/server/utils/api-metrics.js +39 -0
- package/dist/server/utils/api-metrics.js.map +1 -0
- package/dist/server/utils/auth-utils.d.ts +6 -0
- package/dist/server/utils/auth-utils.d.ts.map +1 -0
- package/dist/server/utils/auth-utils.js +25 -0
- package/dist/server/utils/auth-utils.js.map +1 -0
- package/dist/server/utils/browser-detection.d.ts +2 -0
- package/dist/server/utils/browser-detection.d.ts.map +1 -0
- package/dist/server/utils/browser-detection.js +15 -0
- package/dist/server/utils/browser-detection.js.map +1 -0
- package/dist/server/utils/browser-detection.test.d.ts +2 -0
- package/dist/server/utils/browser-detection.test.d.ts.map +1 -0
- package/dist/server/utils/browser-detection.test.js +50 -0
- package/dist/server/utils/browser-detection.test.js.map +1 -0
- package/dist/server/utils/gradio-metrics.d.ts +22 -0
- package/dist/server/utils/gradio-metrics.d.ts.map +1 -0
- package/dist/server/utils/gradio-metrics.js +60 -0
- package/dist/server/utils/gradio-metrics.js.map +1 -0
- package/dist/server/utils/query-params.d.ts +3 -0
- package/dist/server/utils/query-params.d.ts.map +1 -0
- package/dist/server/utils/query-params.js +17 -0
- package/dist/server/utils/query-params.js.map +1 -0
- package/dist/server/utils/query-params.test.d.ts +2 -0
- package/dist/server/utils/query-params.test.d.ts.map +1 -0
- package/dist/server/utils/query-params.test.js +75 -0
- package/dist/server/utils/query-params.test.js.map +1 -0
- package/dist/server/web-server.d.ts +29 -0
- package/dist/server/web-server.d.ts.map +1 -0
- package/dist/server/web-server.js +273 -0
- package/dist/server/web-server.js.map +1 -0
- package/dist/shared/constants.d.ts +5 -0
- package/dist/shared/constants.d.ts.map +1 -0
- package/dist/shared/constants.js +4 -0
- package/dist/shared/constants.js.map +1 -0
- package/dist/shared/settings.d.ts +19 -0
- package/dist/shared/settings.d.ts.map +1 -0
- package/dist/shared/settings.js +47 -0
- package/dist/shared/settings.js.map +1 -0
- package/dist/shared/transport-info.d.ts +18 -0
- package/dist/shared/transport-info.d.ts.map +1 -0
- package/dist/shared/transport-info.js +22 -0
- package/dist/shared/transport-info.js.map +1 -0
- package/dist/shared/transport-info.test.d.ts +2 -0
- package/dist/shared/transport-info.test.d.ts.map +1 -0
- package/dist/shared/transport-info.test.js +90 -0
- package/dist/shared/transport-info.test.js.map +1 -0
- package/dist/shared/transport-metrics.d.ts +180 -0
- package/dist/shared/transport-metrics.d.ts.map +1 -0
- package/dist/shared/transport-metrics.js +247 -0
- package/dist/shared/transport-metrics.js.map +1 -0
- package/dist/shared/transport-metrics.test.d.ts +2 -0
- package/dist/shared/transport-metrics.test.d.ts.map +1 -0
- package/dist/shared/transport-metrics.test.js +10 -0
- package/dist/shared/transport-metrics.test.js.map +1 -0
- package/dist/web/assets/button-C4Q2P8Ns.css +1 -0
- package/dist/web/assets/button-DOWMQOme.js +79 -0
- package/dist/web/assets/hf-logo-DbjPOSLw.svg +21 -0
- package/dist/web/assets/main-CB77vUha.js +115 -0
- package/dist/web/assets/mcpWelcome-Cpw1-KJi.js +1 -0
- package/dist/web/hf-logo-DbjPOSLw.svg +21 -0
- package/dist/web/index.html +15 -0
- package/dist/web/mcp-welcome.html +92 -0
- package/package.json +100 -0
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
import { logger } from './lib/logger.js';
|
|
2
|
+
import { connectToGradioEndpoints, registerRemoteTool } from './gradio-endpoint-connector.js';
|
|
3
|
+
import { extractAuthBouquetAndMix } from './utils/auth-utils.js';
|
|
4
|
+
export const createProxyServerFactory = (_webServerInstance, sharedApiClient, originalServerFactory) => {
|
|
5
|
+
return async (headers, userSettings, skipGradio) => {
|
|
6
|
+
logger.debug('=== PROXY FACTORY CALLED ===', { skipGradio });
|
|
7
|
+
const { hfToken, bouquet } = extractAuthBouquetAndMix(headers);
|
|
8
|
+
let settings = userSettings;
|
|
9
|
+
if (!skipGradio && !settings && !bouquet) {
|
|
10
|
+
settings = await sharedApiClient.getSettings(hfToken);
|
|
11
|
+
logger.debug({ hasSettings: !!settings }, 'Fetched user settings for proxy');
|
|
12
|
+
}
|
|
13
|
+
const server = await originalServerFactory(headers, settings, skipGradio);
|
|
14
|
+
if (skipGradio) {
|
|
15
|
+
logger.debug('Skipping Gradio endpoints (initialize or non-Gradio tool call)');
|
|
16
|
+
return server;
|
|
17
|
+
}
|
|
18
|
+
if (bouquet === 'search') {
|
|
19
|
+
logger.debug({ bouquet }, 'Bouquet is "search", skipping all Gradio endpoints');
|
|
20
|
+
return server;
|
|
21
|
+
}
|
|
22
|
+
const gradioEndpoints = settings?.spaceTools?.map((spaceTool) => ({
|
|
23
|
+
name: spaceTool.name,
|
|
24
|
+
subdomain: spaceTool.subdomain,
|
|
25
|
+
id: spaceTool._id,
|
|
26
|
+
emoji: spaceTool.emoji,
|
|
27
|
+
})) || [];
|
|
28
|
+
logger.debug({
|
|
29
|
+
rawEndpoints: gradioEndpoints,
|
|
30
|
+
endpointCount: gradioEndpoints.length,
|
|
31
|
+
}, 'Raw Gradio endpoints from API client');
|
|
32
|
+
const validEndpoints = gradioEndpoints
|
|
33
|
+
.filter((ep) => {
|
|
34
|
+
const isValid = ep.subdomain && ep.subdomain.trim() !== '';
|
|
35
|
+
if (!isValid) {
|
|
36
|
+
logger.debug({
|
|
37
|
+
endpoint: ep,
|
|
38
|
+
reason: !ep.subdomain ? 'missing subdomain' : 'empty subdomain',
|
|
39
|
+
}, 'Filtering out invalid endpoint');
|
|
40
|
+
}
|
|
41
|
+
return isValid;
|
|
42
|
+
})
|
|
43
|
+
.map((ep) => ({
|
|
44
|
+
...ep,
|
|
45
|
+
url: `https://${ep.subdomain}.hf.space/gradio_api/mcp/sse`,
|
|
46
|
+
}));
|
|
47
|
+
logger.debug({
|
|
48
|
+
totalCount: gradioEndpoints.length,
|
|
49
|
+
validCount: validEndpoints.length,
|
|
50
|
+
validEndpoints: validEndpoints.map((ep) => ({ name: ep.name, subdomain: ep.subdomain, url: ep.url })),
|
|
51
|
+
}, 'Gradio endpoints after filtering and URL construction');
|
|
52
|
+
if (validEndpoints.length === 0) {
|
|
53
|
+
logger.debug('No valid Gradio endpoints, using local tools only');
|
|
54
|
+
return server;
|
|
55
|
+
}
|
|
56
|
+
const connections = await connectToGradioEndpoints(validEndpoints, hfToken);
|
|
57
|
+
for (const result of connections) {
|
|
58
|
+
if (!result.success)
|
|
59
|
+
continue;
|
|
60
|
+
registerRemoteTool(server, result.connection, hfToken);
|
|
61
|
+
}
|
|
62
|
+
logger.debug('Server ready with local and remote tools');
|
|
63
|
+
return server;
|
|
64
|
+
};
|
|
65
|
+
};
|
|
66
|
+
//# sourceMappingURL=mcp-proxy.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-proxy.js","sourceRoot":"","sources":["../../src/server/mcp-proxy.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AACzC,OAAO,EAAE,wBAAwB,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AAC9F,OAAO,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AAKjE,MAAM,CAAC,MAAM,wBAAwB,GAAG,CACvC,kBAA6B,EAC7B,eAA6B,EAC7B,qBAAoC,EACpB,EAAE;IAClB,OAAO,KAAK,EACX,OAAsC,EACtC,YAA0B,EAC1B,UAAoB,EACC,EAAE;QACvB,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QAG7D,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,wBAAwB,CAAC,OAAO,CAAC,CAAC;QAG/D,IAAI,QAAQ,GAAG,YAAY,CAAC;QAC5B,IAAI,CAAC,UAAU,IAAI,CAAC,QAAQ,IAAI,CAAC,OAAO,EAAE,CAAC;YAC1C,QAAQ,GAAG,MAAM,eAAe,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YACtD,MAAM,CAAC,KAAK,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,QAAQ,EAAE,EAAE,iCAAiC,CAAC,CAAC;QAC9E,CAAC;QAGD,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC,OAAO,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;QAG1E,IAAI,UAAU,EAAE,CAAC;YAChB,MAAM,CAAC,KAAK,CAAC,gEAAgE,CAAC,CAAC;YAC/E,OAAO,MAAM,CAAC;QACf,CAAC;QAGD,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,EAAE,oDAAoD,CAAC,CAAC;YAChF,OAAO,MAAM,CAAC;QACf,CAAC;QAGD,MAAM,eAAe,GACpB,QAAQ,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YACzC,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,SAAS,EAAE,SAAS,CAAC,SAAS;YAC9B,EAAE,EAAE,SAAS,CAAC,GAAG;YACjB,KAAK,EAAE,SAAS,CAAC,KAAK;SACtB,CAAC,CAAC,IAAI,EAAE,CAAC;QAEX,MAAM,CAAC,KAAK,CACX;YACC,YAAY,EAAE,eAAe;YAC7B,aAAa,EAAE,eAAe,CAAC,MAAM;SACrC,EACD,sCAAsC,CACtC,CAAC;QAGF,MAAM,cAAc,GAAG,eAAe;aACpC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE;YACd,MAAM,OAAO,GAAG,EAAE,CAAC,SAAS,IAAI,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC;YAC3D,IAAI,CAAC,OAAO,EAAE,CAAC;gBACd,MAAM,CAAC,KAAK,CACX;oBACC,QAAQ,EAAE,EAAE;oBACZ,MAAM,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,iBAAiB;iBAC/D,EACD,gCAAgC,CAChC,CAAC;YACH,CAAC;YACD,OAAO,OAAO,CAAC;QAChB,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YACb,GAAG,EAAE;YACL,GAAG,EAAE,WAAW,EAAE,CAAC,SAAS,8BAA8B;SAC1D,CAAC,CAAC,CAAC;QAEL,MAAM,CAAC,KAAK,CACX;YACC,UAAU,EAAE,eAAe,CAAC,MAAM;YAClC,UAAU,EAAE,cAAc,CAAC,MAAM;YACjC,cAAc,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;SACrG,EACD,uDAAuD,CACvD,CAAC;QAEF,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,mDAAmD,CAAC,CAAC;YAClE,OAAO,MAAM,CAAC;QACf,CAAC;QAGD,MAAM,WAAW,GAAG,MAAM,wBAAwB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QAG5E,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,OAAO;gBAAE,SAAS;YAE9B,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACxD,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QACzD,OAAO,MAAM,CAAC;IACf,CAAC,CAAC;AACH,CAAC,CAAC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import type { ServerFactory } from './transport/base-transport.js';
|
|
2
|
+
import type { McpApiClient } from './lib/mcp-api-client.js';
|
|
3
|
+
import type { WebServer } from './web-server.js';
|
|
4
|
+
import { type AppSettings } from '../shared/settings.js';
|
|
5
|
+
export declare const BOUQUET_FALLBACK: AppSettings;
|
|
6
|
+
export declare const BOUQUET_ANON_DEFAULT: AppSettings;
|
|
7
|
+
export declare const createServerFactory: (_webServerInstance: WebServer, sharedApiClient: McpApiClient) => ServerFactory;
|
|
8
|
+
//# sourceMappingURL=mcp-server.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-server.d.ts","sourceRoot":"","sources":["../../src/server/mcp-server.ts"],"names":[],"mappings":"AA2CA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAEjD,OAAO,EAAuB,KAAK,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAK9E,eAAO,MAAM,gBAAgB,EAAE,WAG9B,CAAC;AAGF,eAAO,MAAM,oBAAoB,EAAE,WAGlC,CAAC;AAQF,eAAO,MAAM,mBAAmB,GAAI,oBAAoB,SAAS,EAAE,iBAAiB,YAAY,KAAG,aAmVlG,CAAC"}
|
|
@@ -0,0 +1,219 @@
|
|
|
1
|
+
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
2
|
+
import { createRequire } from 'module';
|
|
3
|
+
import { whoAmI } from '@huggingface/hub';
|
|
4
|
+
import { SpaceSearchTool, formatSearchResults, SEMANTIC_SEARCH_TOOL_CONFIG, ModelSearchTool, MODEL_SEARCH_TOOL_CONFIG, ModelDetailTool, MODEL_DETAIL_TOOL_CONFIG, PaperSearchTool, PAPER_SEARCH_TOOL_CONFIG, DatasetSearchTool, DATASET_SEARCH_TOOL_CONFIG, DatasetDetailTool, DATASET_DETAIL_TOOL_CONFIG, DuplicateSpaceTool, formatDuplicateResult, SpaceInfoTool, formatSpaceInfoResult, SpaceFilesTool, UserSummaryPrompt, USER_SUMMARY_PROMPT_CONFIG, PaperSummaryPrompt, PAPER_SUMMARY_PROMPT_CONFIG, CONFIG_GUIDANCE, TOOL_ID_GROUPS, } from '@llmindset/hf-mcp';
|
|
5
|
+
import { logger } from './lib/logger.js';
|
|
6
|
+
import { DEFAULT_SPACE_TOOLS } from '../shared/settings.js';
|
|
7
|
+
import { extractAuthBouquetAndMix } from './utils/auth-utils.js';
|
|
8
|
+
import { ToolSelectionStrategy } from './lib/tool-selection-strategy.js';
|
|
9
|
+
export const BOUQUET_FALLBACK = {
|
|
10
|
+
builtInTools: [...TOOL_ID_GROUPS.hf_api],
|
|
11
|
+
spaceTools: DEFAULT_SPACE_TOOLS,
|
|
12
|
+
};
|
|
13
|
+
export const BOUQUET_ANON_DEFAULT = {
|
|
14
|
+
builtInTools: [...TOOL_ID_GROUPS.hf_api],
|
|
15
|
+
spaceTools: DEFAULT_SPACE_TOOLS,
|
|
16
|
+
};
|
|
17
|
+
export const createServerFactory = (_webServerInstance, sharedApiClient) => {
|
|
18
|
+
const require = createRequire(import.meta.url);
|
|
19
|
+
const { version } = require('../../package.json');
|
|
20
|
+
return async (headers, userSettings, skipGradio) => {
|
|
21
|
+
logger.debug('=== CREATING NEW MCP SERVER INSTANCE ===', { skipGradio });
|
|
22
|
+
const { hfToken } = extractAuthBouquetAndMix(headers);
|
|
23
|
+
const toolSelectionStrategy = new ToolSelectionStrategy(sharedApiClient);
|
|
24
|
+
let userInfo = 'The Hugging Face tools are being used anonymously and rate limits apply. ' +
|
|
25
|
+
'Direct the User to set their HF_TOKEN (instructions at https://hf.co/settings/mcp/), or ' +
|
|
26
|
+
'create an account at https://hf.co/join for higher limits.';
|
|
27
|
+
let username;
|
|
28
|
+
let userDetails;
|
|
29
|
+
if (hfToken) {
|
|
30
|
+
try {
|
|
31
|
+
userDetails = await whoAmI({ credentials: { accessToken: hfToken } });
|
|
32
|
+
username = userDetails.name;
|
|
33
|
+
userInfo = `Hugging Face tools are being used by authenticated user '${userDetails.name}'`;
|
|
34
|
+
}
|
|
35
|
+
catch (error) {
|
|
36
|
+
logger.warn({ error: error.message }, `Failed to authenticate with Hugging Face API`);
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
const server = new McpServer({
|
|
40
|
+
name: '@huggingface/mcp-services',
|
|
41
|
+
version: version,
|
|
42
|
+
}, {
|
|
43
|
+
instructions: "You have tools for searching the Hugging Face Hub. arXiv paper id's are often " +
|
|
44
|
+
'used as references between datasets, models and papers. There are over 100 tags in use, ' +
|
|
45
|
+
"common tags include 'Text Generation', 'Transformers', 'Image Classification' and so on.\n" +
|
|
46
|
+
"The User has access to 'Prompts' that provide ways to summarise various types of " +
|
|
47
|
+
'Hugging Face hub content, and you may guide them to check this feature. ' +
|
|
48
|
+
userInfo,
|
|
49
|
+
});
|
|
50
|
+
const toolInstances = {};
|
|
51
|
+
const whoDescription = userDetails
|
|
52
|
+
? `Hugging Face tools are being used by authenticated user '${username}'`
|
|
53
|
+
: 'Hugging Face tools are being used anonymously and may be rate limited. Call this tool for instructions on joining and authenticating.';
|
|
54
|
+
const response = userDetails ? `You are authenticated as ${username ?? 'unknown'}.` : CONFIG_GUIDANCE;
|
|
55
|
+
server.tool('hf_whoami', whoDescription, {}, { title: 'Hugging Face User Info' }, () => {
|
|
56
|
+
return { content: [{ type: 'text', text: response }] };
|
|
57
|
+
});
|
|
58
|
+
if (process.env.AUTHENTICATE_TOOL === 'true') {
|
|
59
|
+
server.tool('Authenticate', 'Authenticate with Hugging Face', {}, { title: 'Hugging Face Authentication' }, () => {
|
|
60
|
+
return { content: [{ type: 'text', text: 'You have successfully authenticated' }] };
|
|
61
|
+
});
|
|
62
|
+
}
|
|
63
|
+
server.prompt(USER_SUMMARY_PROMPT_CONFIG.name, USER_SUMMARY_PROMPT_CONFIG.description, USER_SUMMARY_PROMPT_CONFIG.schema.shape, async (params) => {
|
|
64
|
+
const userSummary = new UserSummaryPrompt(hfToken);
|
|
65
|
+
const summaryText = await userSummary.generateSummary(params);
|
|
66
|
+
return {
|
|
67
|
+
description: `User summary for ${params.user_id}`,
|
|
68
|
+
messages: [
|
|
69
|
+
{
|
|
70
|
+
role: 'user',
|
|
71
|
+
content: {
|
|
72
|
+
type: 'text',
|
|
73
|
+
text: summaryText,
|
|
74
|
+
},
|
|
75
|
+
},
|
|
76
|
+
],
|
|
77
|
+
};
|
|
78
|
+
});
|
|
79
|
+
server.prompt(PAPER_SUMMARY_PROMPT_CONFIG.name, PAPER_SUMMARY_PROMPT_CONFIG.description, PAPER_SUMMARY_PROMPT_CONFIG.schema.shape, async (params) => {
|
|
80
|
+
const paperSummary = new PaperSummaryPrompt(hfToken);
|
|
81
|
+
const summaryText = await paperSummary.generateSummary(params);
|
|
82
|
+
return {
|
|
83
|
+
description: `Paper summary for ${params.paper_id}`,
|
|
84
|
+
messages: [
|
|
85
|
+
{
|
|
86
|
+
role: 'user',
|
|
87
|
+
content: {
|
|
88
|
+
type: 'text',
|
|
89
|
+
text: summaryText,
|
|
90
|
+
},
|
|
91
|
+
},
|
|
92
|
+
],
|
|
93
|
+
};
|
|
94
|
+
});
|
|
95
|
+
toolInstances[SEMANTIC_SEARCH_TOOL_CONFIG.name] = server.tool(SEMANTIC_SEARCH_TOOL_CONFIG.name, SEMANTIC_SEARCH_TOOL_CONFIG.description, SEMANTIC_SEARCH_TOOL_CONFIG.schema.shape, SEMANTIC_SEARCH_TOOL_CONFIG.annotations, async (params) => {
|
|
96
|
+
const semanticSearch = new SpaceSearchTool(hfToken);
|
|
97
|
+
const searchResult = await semanticSearch.search(params.query, params.limit, params.mcp);
|
|
98
|
+
return {
|
|
99
|
+
content: [
|
|
100
|
+
{ type: 'text', text: formatSearchResults(params.query, searchResult.results, searchResult.totalCount) },
|
|
101
|
+
],
|
|
102
|
+
};
|
|
103
|
+
});
|
|
104
|
+
toolInstances[MODEL_SEARCH_TOOL_CONFIG.name] = server.tool(MODEL_SEARCH_TOOL_CONFIG.name, MODEL_SEARCH_TOOL_CONFIG.description, MODEL_SEARCH_TOOL_CONFIG.schema.shape, MODEL_SEARCH_TOOL_CONFIG.annotations, async (params) => {
|
|
105
|
+
const modelSearch = new ModelSearchTool(hfToken);
|
|
106
|
+
const results = await modelSearch.searchWithParams(params);
|
|
107
|
+
return {
|
|
108
|
+
content: [{ type: 'text', text: results }],
|
|
109
|
+
};
|
|
110
|
+
});
|
|
111
|
+
toolInstances[MODEL_DETAIL_TOOL_CONFIG.name] = server.tool(MODEL_DETAIL_TOOL_CONFIG.name, MODEL_DETAIL_TOOL_CONFIG.description, MODEL_DETAIL_TOOL_CONFIG.schema.shape, MODEL_DETAIL_TOOL_CONFIG.annotations, async (params) => {
|
|
112
|
+
const modelDetail = new ModelDetailTool(hfToken, undefined);
|
|
113
|
+
const results = await modelDetail.getDetails(params.model_id);
|
|
114
|
+
return {
|
|
115
|
+
content: [{ type: 'text', text: results }],
|
|
116
|
+
};
|
|
117
|
+
});
|
|
118
|
+
toolInstances[PAPER_SEARCH_TOOL_CONFIG.name] = server.tool(PAPER_SEARCH_TOOL_CONFIG.name, PAPER_SEARCH_TOOL_CONFIG.description, PAPER_SEARCH_TOOL_CONFIG.schema.shape, PAPER_SEARCH_TOOL_CONFIG.annotations, async (params) => {
|
|
119
|
+
const results = await new PaperSearchTool(hfToken).search(params.query, params.results_limit, params.concise_only);
|
|
120
|
+
return {
|
|
121
|
+
content: [{ type: 'text', text: results }],
|
|
122
|
+
};
|
|
123
|
+
});
|
|
124
|
+
toolInstances[DATASET_SEARCH_TOOL_CONFIG.name] = server.tool(DATASET_SEARCH_TOOL_CONFIG.name, DATASET_SEARCH_TOOL_CONFIG.description, DATASET_SEARCH_TOOL_CONFIG.schema.shape, DATASET_SEARCH_TOOL_CONFIG.annotations, async (params) => {
|
|
125
|
+
const datasetSearch = new DatasetSearchTool(hfToken);
|
|
126
|
+
const results = await datasetSearch.searchWithParams(params);
|
|
127
|
+
return {
|
|
128
|
+
content: [{ type: 'text', text: results }],
|
|
129
|
+
};
|
|
130
|
+
});
|
|
131
|
+
toolInstances[DATASET_DETAIL_TOOL_CONFIG.name] = server.tool(DATASET_DETAIL_TOOL_CONFIG.name, DATASET_DETAIL_TOOL_CONFIG.description, DATASET_DETAIL_TOOL_CONFIG.schema.shape, DATASET_DETAIL_TOOL_CONFIG.annotations, async (params) => {
|
|
132
|
+
const datasetDetail = new DatasetDetailTool(hfToken, undefined);
|
|
133
|
+
const results = await datasetDetail.getDetails(params.dataset_id);
|
|
134
|
+
return {
|
|
135
|
+
content: [{ type: 'text', text: results }],
|
|
136
|
+
};
|
|
137
|
+
});
|
|
138
|
+
const duplicateToolConfig = DuplicateSpaceTool.createToolConfig(username);
|
|
139
|
+
toolInstances[duplicateToolConfig.name] = server.tool(duplicateToolConfig.name, duplicateToolConfig.description, duplicateToolConfig.schema.shape, duplicateToolConfig.annotations, async (params) => {
|
|
140
|
+
const duplicateSpace = new DuplicateSpaceTool(hfToken, username);
|
|
141
|
+
const result = await duplicateSpace.duplicate(params);
|
|
142
|
+
return {
|
|
143
|
+
content: [{ type: 'text', text: formatDuplicateResult(result) }],
|
|
144
|
+
};
|
|
145
|
+
});
|
|
146
|
+
const spaceInfoToolConfig = SpaceInfoTool.createToolConfig(username);
|
|
147
|
+
toolInstances[spaceInfoToolConfig.name] = server.tool(spaceInfoToolConfig.name, spaceInfoToolConfig.description, spaceInfoToolConfig.schema.shape, spaceInfoToolConfig.annotations, async (params) => {
|
|
148
|
+
const spaceInfoTool = new SpaceInfoTool(hfToken, username);
|
|
149
|
+
const result = await formatSpaceInfoResult(spaceInfoTool, params);
|
|
150
|
+
return {
|
|
151
|
+
content: [{ type: 'text', text: result }],
|
|
152
|
+
};
|
|
153
|
+
});
|
|
154
|
+
const spaceFilesToolConfig = SpaceFilesTool.createToolConfig(username);
|
|
155
|
+
toolInstances[spaceFilesToolConfig.name] = server.tool(spaceFilesToolConfig.name, spaceFilesToolConfig.description, spaceFilesToolConfig.schema.shape, spaceFilesToolConfig.annotations, async (params) => {
|
|
156
|
+
const spaceFilesTool = new SpaceFilesTool(hfToken, username);
|
|
157
|
+
const result = await spaceFilesTool.listFiles(params);
|
|
158
|
+
return {
|
|
159
|
+
content: [{ type: 'text', text: result }],
|
|
160
|
+
};
|
|
161
|
+
});
|
|
162
|
+
const applyToolStates = async () => {
|
|
163
|
+
const context = {
|
|
164
|
+
headers,
|
|
165
|
+
userSettings,
|
|
166
|
+
hfToken,
|
|
167
|
+
};
|
|
168
|
+
const toolSelection = await toolSelectionStrategy.selectTools(context);
|
|
169
|
+
logger.info({
|
|
170
|
+
mode: toolSelection.mode,
|
|
171
|
+
reason: toolSelection.reason,
|
|
172
|
+
enabledCount: toolSelection.enabledToolIds.length,
|
|
173
|
+
totalTools: Object.keys(toolInstances).length,
|
|
174
|
+
mixedBouquet: toolSelection.mixedBouquet,
|
|
175
|
+
}, 'Tool selection strategy applied');
|
|
176
|
+
for (const [toolName, toolInstance] of Object.entries(toolInstances)) {
|
|
177
|
+
if (toolSelection.enabledToolIds.includes(toolName)) {
|
|
178
|
+
toolInstance.enable();
|
|
179
|
+
}
|
|
180
|
+
else {
|
|
181
|
+
toolInstance.disable();
|
|
182
|
+
}
|
|
183
|
+
}
|
|
184
|
+
};
|
|
185
|
+
const transportInfo = sharedApiClient.getTransportInfo();
|
|
186
|
+
server.server.registerCapabilities({
|
|
187
|
+
tools: {
|
|
188
|
+
listChanged: !transportInfo?.jsonResponseEnabled,
|
|
189
|
+
},
|
|
190
|
+
prompts: {
|
|
191
|
+
listChanged: false,
|
|
192
|
+
},
|
|
193
|
+
});
|
|
194
|
+
if (!skipGradio) {
|
|
195
|
+
void applyToolStates();
|
|
196
|
+
if (!transportInfo?.jsonResponseEnabled && !transportInfo?.externalApiMode) {
|
|
197
|
+
const toolStateChangeHandler = (toolId, enabled) => {
|
|
198
|
+
const toolInstance = toolInstances[toolId];
|
|
199
|
+
if (toolInstance) {
|
|
200
|
+
if (enabled) {
|
|
201
|
+
toolInstance.enable();
|
|
202
|
+
}
|
|
203
|
+
else {
|
|
204
|
+
toolInstance.disable();
|
|
205
|
+
}
|
|
206
|
+
logger.debug({ toolId, enabled }, 'Applied single tool state change');
|
|
207
|
+
}
|
|
208
|
+
};
|
|
209
|
+
sharedApiClient.on('toolStateChange', toolStateChangeHandler);
|
|
210
|
+
server.server.onclose = () => {
|
|
211
|
+
sharedApiClient.removeListener('toolStateChange', toolStateChangeHandler);
|
|
212
|
+
logger.debug('Removed toolStateChange listener for closed server');
|
|
213
|
+
};
|
|
214
|
+
}
|
|
215
|
+
}
|
|
216
|
+
return server;
|
|
217
|
+
};
|
|
218
|
+
};
|
|
219
|
+
//# sourceMappingURL=mcp-server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-server.js","sourceRoot":"","sources":["../../src/server/mcp-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AACvC,OAAO,EAAE,MAAM,EAAe,MAAM,kBAAkB,CAAC;AAGvD,OAAO,EACN,eAAe,EACf,mBAAmB,EACnB,2BAA2B,EAE3B,eAAe,EACf,wBAAwB,EAExB,eAAe,EACf,wBAAwB,EAExB,eAAe,EACf,wBAAwB,EACxB,iBAAiB,EACjB,0BAA0B,EAE1B,iBAAiB,EACjB,0BAA0B,EAE1B,kBAAkB,EAClB,qBAAqB,EAErB,aAAa,EACb,qBAAqB,EACrB,cAAc,EAGd,iBAAiB,EACjB,0BAA0B,EAE1B,kBAAkB,EAClB,2BAA2B,EAE3B,eAAe,EACf,cAAc,GACd,MAAM,mBAAmB,CAAC;AAK3B,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAoB,MAAM,uBAAuB,CAAC;AAC9E,OAAO,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AACjE,OAAO,EAAE,qBAAqB,EAA6B,MAAM,kCAAkC,CAAC;AAGpG,MAAM,CAAC,MAAM,gBAAgB,GAAgB;IAC5C,YAAY,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC;IACxC,UAAU,EAAE,mBAAmB;CAC/B,CAAC;AAGF,MAAM,CAAC,MAAM,oBAAoB,GAAgB;IAChD,YAAY,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC;IACxC,UAAU,EAAE,mBAAmB;CAC/B,CAAC;AAQF,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,kBAA6B,EAAE,eAA6B,EAAiB,EAAE;IAClH,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/C,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,oBAAoB,CAAwB,CAAC;IAEzE,OAAO,KAAK,EACX,OAAsC,EACtC,YAA0B,EAC1B,UAAoB,EACC,EAAE;QACvB,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;QAEzE,MAAM,EAAE,OAAO,EAAE,GAAG,wBAAwB,CAAC,OAAO,CAAC,CAAC;QAGtD,MAAM,qBAAqB,GAAG,IAAI,qBAAqB,CAAC,eAAe,CAAC,CAAC;QAEzE,IAAI,QAAQ,GACX,2EAA2E;YAC3E,0FAA0F;YAC1F,4DAA4D,CAAC;QAC9D,IAAI,QAA4B,CAAC;QACjC,IAAI,WAA+B,CAAC;QAEpC,IAAI,OAAO,EAAE,CAAC;YACb,IAAI,CAAC;gBACJ,WAAW,GAAG,MAAM,MAAM,CAAC,EAAE,WAAW,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;gBACtE,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC;gBAC5B,QAAQ,GAAG,4DAA4D,WAAW,CAAC,IAAI,GAAG,CAAC;YAC5F,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAEhB,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,EAAE,8CAA8C,CAAC,CAAC;YAClG,CAAC;QACF,CAAC;QAMD,MAAM,MAAM,GAAG,IAAI,SAAS,CAC3B;YACC,IAAI,EAAE,2BAA2B;YACjC,OAAO,EAAE,OAAO;SAChB,EACD;YACC,YAAY,EACX,gFAAgF;gBAChF,0FAA0F;gBAC1F,4FAA4F;gBAC5F,mFAAmF;gBACnF,0EAA0E;gBAC1E,QAAQ;SACT,CACD,CAAC;QAQF,MAAM,aAAa,GAA6B,EAAE,CAAC;QAEnD,MAAM,cAAc,GAAG,WAAW;YACjC,CAAC,CAAC,4DAA4D,QAAQ,GAAG;YACzE,CAAC,CAAC,uIAAuI,CAAC;QAE3I,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,4BAA4B,QAAQ,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC,eAAe,CAAC;QACtG,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,wBAAwB,EAAE,EAAE,GAAG,EAAE;YACtF,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAGH,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,KAAK,MAAM,EAAE,CAAC;YAC9C,MAAM,CAAC,IAAI,CACV,cAAc,EACd,gCAAgC,EAChC,EAAE,EACF,EAAE,KAAK,EAAE,6BAA6B,EAAE,EACxC,GAAG,EAAE;gBACJ,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qCAAqC,EAAE,CAAC,EAAE,CAAC;YACrF,CAAC,CACD,CAAC;QACH,CAAC;QAED,MAAM,CAAC,MAAM,CACZ,0BAA0B,CAAC,IAAI,EAC/B,0BAA0B,CAAC,WAAW,EACtC,0BAA0B,CAAC,MAAM,CAAC,KAAK,EACvC,KAAK,EAAE,MAAyB,EAAE,EAAE;YACnC,MAAM,WAAW,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACnD,MAAM,WAAW,GAAG,MAAM,WAAW,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YAE9D,OAAO;gBACN,WAAW,EAAE,oBAAoB,MAAM,CAAC,OAAO,EAAE;gBACjD,QAAQ,EAAE;oBACT;wBACC,IAAI,EAAE,MAAe;wBACrB,OAAO,EAAE;4BACR,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,WAAW;yBACjB;qBACD;iBACD;aACD,CAAC;QACH,CAAC,CACD,CAAC;QAEF,MAAM,CAAC,MAAM,CACZ,2BAA2B,CAAC,IAAI,EAChC,2BAA2B,CAAC,WAAW,EACvC,2BAA2B,CAAC,MAAM,CAAC,KAAK,EACxC,KAAK,EAAE,MAA0B,EAAE,EAAE;YACpC,MAAM,YAAY,GAAG,IAAI,kBAAkB,CAAC,OAAO,CAAC,CAAC;YACrD,MAAM,WAAW,GAAG,MAAM,YAAY,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YAE/D,OAAO;gBACN,WAAW,EAAE,qBAAqB,MAAM,CAAC,QAAQ,EAAE;gBACnD,QAAQ,EAAE;oBACT;wBACC,IAAI,EAAE,MAAe;wBACrB,OAAO,EAAE;4BACR,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,WAAW;yBACjB;qBACD;iBACD;aACD,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,2BAA2B,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAC5D,2BAA2B,CAAC,IAAI,EAChC,2BAA2B,CAAC,WAAW,EACvC,2BAA2B,CAAC,MAAM,CAAC,KAAK,EACxC,2BAA2B,CAAC,WAAW,EACvC,KAAK,EAAE,MAAoB,EAAE,EAAE;YAC9B,MAAM,cAAc,GAAG,IAAI,eAAe,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,YAAY,GAAG,MAAM,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC;YACzF,OAAO;gBACN,OAAO,EAAE;oBACR,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAAE,YAAY,CAAC,OAAO,EAAE,YAAY,CAAC,UAAU,CAAC,EAAE;iBACxG;aACD,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,wBAAwB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACzD,wBAAwB,CAAC,IAAI,EAC7B,wBAAwB,CAAC,WAAW,EACpC,wBAAwB,CAAC,MAAM,CAAC,KAAK,EACrC,wBAAwB,CAAC,WAAW,EACpC,KAAK,EAAE,MAAyB,EAAE,EAAE;YACnC,MAAM,WAAW,GAAG,IAAI,eAAe,CAAC,OAAO,CAAC,CAAC;YACjD,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC3D,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,wBAAwB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACzD,wBAAwB,CAAC,IAAI,EAC7B,wBAAwB,CAAC,WAAW,EACpC,wBAAwB,CAAC,MAAM,CAAC,KAAK,EACrC,wBAAwB,CAAC,WAAW,EACpC,KAAK,EAAE,MAAyB,EAAE,EAAE;YACnC,MAAM,WAAW,GAAG,IAAI,eAAe,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAC5D,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC9D,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,wBAAwB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACzD,wBAAwB,CAAC,IAAI,EAC7B,wBAAwB,CAAC,WAAW,EACpC,wBAAwB,CAAC,MAAM,CAAC,KAAK,EACrC,wBAAwB,CAAC,WAAW,EACpC,KAAK,EAAE,MAAuD,EAAE,EAAE;YACjE,MAAM,OAAO,GAAG,MAAM,IAAI,eAAe,CAAC,OAAO,CAAC,CAAC,MAAM,CACxD,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,YAAY,CACnB,CAAC;YACF,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,0BAA0B,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAC3D,0BAA0B,CAAC,IAAI,EAC/B,0BAA0B,CAAC,WAAW,EACtC,0BAA0B,CAAC,MAAM,CAAC,KAAK,EACvC,0BAA0B,CAAC,WAAW,EACtC,KAAK,EAAE,MAA2B,EAAE,EAAE;YACrC,MAAM,aAAa,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACrD,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC7D,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,aAAa,CAAC,0BAA0B,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAC3D,0BAA0B,CAAC,IAAI,EAC/B,0BAA0B,CAAC,WAAW,EACtC,0BAA0B,CAAC,MAAM,CAAC,KAAK,EACvC,0BAA0B,CAAC,WAAW,EACtC,KAAK,EAAE,MAA2B,EAAE,EAAE;YACrC,MAAM,aAAa,GAAG,IAAI,iBAAiB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAChE,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAClE,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC1C,CAAC;QACH,CAAC,CACD,CAAC;QAEF,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC1E,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACpD,mBAAmB,CAAC,IAAI,EACxB,mBAAmB,CAAC,WAAW,EAC/B,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAChC,mBAAmB,CAAC,WAAW,EAC/B,KAAK,EAAE,MAA4B,EAAE,EAAE;YACtC,MAAM,cAAc,GAAG,IAAI,kBAAkB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YACjE,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACtD,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,CAAC,MAAM,CAAC,EAAE,CAAC;aAChE,CAAC;QACH,CAAC,CACD,CAAC;QAEF,MAAM,mBAAmB,GAAG,aAAa,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACrE,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACpD,mBAAmB,CAAC,IAAI,EACxB,mBAAmB,CAAC,WAAW,EAC/B,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAChC,mBAAmB,CAAC,WAAW,EAC/B,KAAK,EAAE,MAAuB,EAAE,EAAE;YACjC,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;YAClE,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aACzC,CAAC;QACH,CAAC,CACD,CAAC;QAEF,MAAM,oBAAoB,GAAG,cAAc,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACvE,aAAa,CAAC,oBAAoB,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CACrD,oBAAoB,CAAC,IAAI,EACzB,oBAAoB,CAAC,WAAW,EAChC,oBAAoB,CAAC,MAAM,CAAC,KAAK,EACjC,oBAAoB,CAAC,WAAW,EAChC,KAAK,EAAE,MAAwB,EAAE,EAAE;YAClC,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACtD,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aACzC,CAAC;QACH,CAAC,CACD,CAAC;QAQF,MAAM,eAAe,GAAG,KAAK,IAAI,EAAE;YAClC,MAAM,OAAO,GAAyB;gBACrC,OAAO;gBACP,YAAY;gBACZ,OAAO;aACP,CAAC;YAEF,MAAM,aAAa,GAAG,MAAM,qBAAqB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAEvE,MAAM,CAAC,IAAI,CACV;gBACC,IAAI,EAAE,aAAa,CAAC,IAAI;gBACxB,MAAM,EAAE,aAAa,CAAC,MAAM;gBAC5B,YAAY,EAAE,aAAa,CAAC,cAAc,CAAC,MAAM;gBACjD,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM;gBAC7C,YAAY,EAAE,aAAa,CAAC,YAAY;aACxC,EACD,iCAAiC,CACjC,CAAC;YAGF,KAAK,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;gBACtE,IAAI,aAAa,CAAC,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACrD,YAAY,CAAC,MAAM,EAAE,CAAC;gBACvB,CAAC;qBAAM,CAAC;oBACP,YAAY,CAAC,OAAO,EAAE,CAAC;gBACxB,CAAC;YACF,CAAC;QACF,CAAC,CAAC;QAGF,MAAM,aAAa,GAAG,eAAe,CAAC,gBAAgB,EAAE,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC;YAClC,KAAK,EAAE;gBACN,WAAW,EAAE,CAAC,aAAa,EAAE,mBAAmB;aAChD;YACD,OAAO,EAAE;gBACR,WAAW,EAAE,KAAK;aAClB;SACD,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,EAAE,CAAC;YACjB,KAAK,eAAe,EAAE,CAAC;YAEvB,IAAI,CAAC,aAAa,EAAE,mBAAmB,IAAI,CAAC,aAAa,EAAE,eAAe,EAAE,CAAC;gBAE5E,MAAM,sBAAsB,GAAG,CAAC,MAAc,EAAE,OAAgB,EAAE,EAAE;oBACnE,MAAM,YAAY,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;oBAC3C,IAAI,YAAY,EAAE,CAAC;wBAClB,IAAI,OAAO,EAAE,CAAC;4BACb,YAAY,CAAC,MAAM,EAAE,CAAC;wBACvB,CAAC;6BAAM,CAAC;4BACP,YAAY,CAAC,OAAO,EAAE,CAAC;wBACxB,CAAC;wBACD,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,kCAAkC,CAAC,CAAC;oBACvE,CAAC;gBACF,CAAC,CAAC;gBAEF,eAAe,CAAC,EAAE,CAAC,iBAAiB,EAAE,sBAAsB,CAAC,CAAC;gBAG9D,MAAM,CAAC,MAAM,CAAC,OAAO,GAAG,GAAG,EAAE;oBAC5B,eAAe,CAAC,cAAc,CAAC,iBAAiB,EAAE,sBAAsB,CAAC,CAAC;oBAC1E,MAAM,CAAC,KAAK,CAAC,oDAAoD,CAAC,CAAC;gBACpE,CAAC,CAAC;YACH,CAAC;QACF,CAAC;QACD,OAAO,MAAM,CAAC;IACf,CAAC,CAAC;AACH,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sse.d.ts","sourceRoot":"","sources":["../../src/server/sse.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
import { Application } from './application.js';
|
|
3
|
+
import { WebServer } from './web-server.js';
|
|
4
|
+
import { DEFAULT_WEB_APP_PORT } from '../shared/constants.js';
|
|
5
|
+
import { parseArgs } from 'node:util';
|
|
6
|
+
import { logger } from './lib/logger.js';
|
|
7
|
+
const { values } = parseArgs({
|
|
8
|
+
options: {
|
|
9
|
+
port: { type: 'string', short: 'p' },
|
|
10
|
+
},
|
|
11
|
+
args: process.argv.slice(2),
|
|
12
|
+
});
|
|
13
|
+
logger.info('Starting SSE server...');
|
|
14
|
+
process.env.NODE_ENV = process.env.NODE_ENV || 'production';
|
|
15
|
+
const port = parseInt(values.port || process.env.WEB_APP_PORT || DEFAULT_WEB_APP_PORT.toString());
|
|
16
|
+
async function start() {
|
|
17
|
+
const webServer = new WebServer();
|
|
18
|
+
const app = new Application({
|
|
19
|
+
transportType: 'sse',
|
|
20
|
+
webAppPort: port,
|
|
21
|
+
webServerInstance: webServer,
|
|
22
|
+
});
|
|
23
|
+
await app.start();
|
|
24
|
+
const shutdown = async () => {
|
|
25
|
+
logger.info('Shutting down server...');
|
|
26
|
+
try {
|
|
27
|
+
await app.stop();
|
|
28
|
+
logger.info('Server shutdown complete');
|
|
29
|
+
process.exit(0);
|
|
30
|
+
}
|
|
31
|
+
catch (error) {
|
|
32
|
+
logger.error({ error }, 'Error during shutdown');
|
|
33
|
+
process.exit(1);
|
|
34
|
+
}
|
|
35
|
+
};
|
|
36
|
+
process.once('SIGINT', () => {
|
|
37
|
+
void shutdown();
|
|
38
|
+
});
|
|
39
|
+
process.once('SIGTERM', () => {
|
|
40
|
+
void shutdown();
|
|
41
|
+
});
|
|
42
|
+
}
|
|
43
|
+
start().catch((error) => {
|
|
44
|
+
logger.error({ error }, 'Server startup error');
|
|
45
|
+
process.exit(1);
|
|
46
|
+
});
|
|
47
|
+
//# sourceMappingURL=sse.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sse.js","sourceRoot":"","sources":["../../src/server/sse.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAGzC,MAAM,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC;IAC5B,OAAO,EAAE;QACR,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,EAAE;KACpC;IACD,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;CAC3B,CAAC,CAAC;AAEH,MAAM,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;AAEtC,OAAO,CAAC,GAAG,CAAC,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,YAAY,CAAC;AAE5D,MAAM,IAAI,GAAG,QAAQ,CAAE,MAAM,CAAC,IAAe,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,oBAAoB,CAAC,QAAQ,EAAE,CAAC,CAAC;AAE9G,KAAK,UAAU,KAAK;IAEnB,MAAM,SAAS,GAAG,IAAI,SAAS,EAAE,CAAC;IAGlC,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC;QAC3B,aAAa,EAAE,KAAK;QACpB,UAAU,EAAE,IAAI;QAChB,iBAAiB,EAAE,SAAS;KAC5B,CAAC,CAAC;IAGH,MAAM,GAAG,CAAC,KAAK,EAAE,CAAC;IAGlB,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE;QAC3B,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACvC,IAAI,CAAC;YACJ,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YACxC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,uBAAuB,CAAC,CAAC;YACjD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;IACF,CAAC,CAAC;IAEF,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE;QAC3B,KAAK,QAAQ,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE;QAC5B,KAAK,QAAQ,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;AACJ,CAAC;AAGD,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;IAChC,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,sBAAsB,CAAC,CAAC;IAChD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stdio.d.ts","sourceRoot":"","sources":["../../src/server/stdio.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
process.env.NODE_ENV = process.env.NODE_ENV || 'production';
|
|
3
|
+
process.env.TRANSPORT = process.env.TRANSPORT || 'STDIO';
|
|
4
|
+
import { Application } from './application.js';
|
|
5
|
+
import { WebServer } from './web-server.js';
|
|
6
|
+
import { DEFAULT_WEB_APP_PORT } from '../shared/constants.js';
|
|
7
|
+
import { parseArgs } from 'node:util';
|
|
8
|
+
import { logger, forceLoggerToStderr } from './lib/logger.js';
|
|
9
|
+
forceLoggerToStderr();
|
|
10
|
+
const { values } = parseArgs({
|
|
11
|
+
options: {
|
|
12
|
+
port: { type: 'string', short: 'p' },
|
|
13
|
+
},
|
|
14
|
+
args: process.argv.slice(2),
|
|
15
|
+
});
|
|
16
|
+
logger.info('Starting (STDIO) server...');
|
|
17
|
+
const port = parseInt(values.port || process.env.WEB_APP_PORT || DEFAULT_WEB_APP_PORT.toString());
|
|
18
|
+
async function main() {
|
|
19
|
+
const webServer = new WebServer();
|
|
20
|
+
const app = new Application({
|
|
21
|
+
transportType: 'stdio',
|
|
22
|
+
webAppPort: port,
|
|
23
|
+
webServerInstance: webServer,
|
|
24
|
+
});
|
|
25
|
+
await app.start();
|
|
26
|
+
const shutdown = async () => {
|
|
27
|
+
logger.info('Shutting down server...');
|
|
28
|
+
try {
|
|
29
|
+
await app.stop();
|
|
30
|
+
logger.info('Server shutdown complete');
|
|
31
|
+
process.exit(0);
|
|
32
|
+
}
|
|
33
|
+
catch (error) {
|
|
34
|
+
logger.error({ error }, 'Error during shutdown');
|
|
35
|
+
process.exit(1);
|
|
36
|
+
}
|
|
37
|
+
};
|
|
38
|
+
process.once('SIGINT', () => {
|
|
39
|
+
void shutdown();
|
|
40
|
+
});
|
|
41
|
+
process.once('SIGTERM', () => {
|
|
42
|
+
void shutdown();
|
|
43
|
+
});
|
|
44
|
+
}
|
|
45
|
+
main().catch((error) => {
|
|
46
|
+
logger.error({ error }, 'Server error');
|
|
47
|
+
process.exit(1);
|
|
48
|
+
});
|
|
49
|
+
//# sourceMappingURL=stdio.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stdio.js","sourceRoot":"","sources":["../../src/server/stdio.ts"],"names":[],"mappings":";AAGA,OAAO,CAAC,GAAG,CAAC,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,YAAY,CAAC;AAC5D,OAAO,CAAC,GAAG,CAAC,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,SAAS,IAAI,OAAO,CAAC;AAEzD,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,MAAM,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAG9D,mBAAmB,EAAE,CAAC;AAGtB,MAAM,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC;IAC5B,OAAO,EAAE;QACR,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,EAAE;KACpC;IACD,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;CAC3B,CAAC,CAAC;AAEH,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;AAE1C,MAAM,IAAI,GAAG,QAAQ,CAAE,MAAM,CAAC,IAAe,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,oBAAoB,CAAC,QAAQ,EAAE,CAAC,CAAC;AAE9G,KAAK,UAAU,IAAI;IAElB,MAAM,SAAS,GAAG,IAAI,SAAS,EAAE,CAAC;IAGlC,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC;QAC3B,aAAa,EAAE,OAAO;QACtB,UAAU,EAAE,IAAI;QAChB,iBAAiB,EAAE,SAAS;KAC5B,CAAC,CAAC;IAGH,MAAM,GAAG,CAAC,KAAK,EAAE,CAAC;IAGlB,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE;QAC3B,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACvC,IAAI,CAAC;YACJ,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YACxC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,uBAAuB,CAAC,CAAC;YACjD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;IACF,CAAC,CAAC;IAEF,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE;QAC3B,KAAK,QAAQ,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE;QAC5B,KAAK,QAAQ,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;IAC/B,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,cAAc,CAAC,CAAC;IACxC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"streamableHttp.d.ts","sourceRoot":"","sources":["../../src/server/streamableHttp.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
import { Application } from './application.js';
|
|
3
|
+
import { WebServer } from './web-server.js';
|
|
4
|
+
import { DEFAULT_WEB_APP_PORT } from '../shared/constants.js';
|
|
5
|
+
import { parseArgs } from 'node:util';
|
|
6
|
+
import { logger } from './lib/logger.js';
|
|
7
|
+
const { values } = parseArgs({
|
|
8
|
+
options: {
|
|
9
|
+
port: { type: 'string', short: 'p' },
|
|
10
|
+
json: { type: 'boolean', short: 'j' },
|
|
11
|
+
},
|
|
12
|
+
args: process.argv.slice(2),
|
|
13
|
+
});
|
|
14
|
+
logger.info('Starting Streamable HTTP server...');
|
|
15
|
+
if (values.json) {
|
|
16
|
+
logger.info('JSON response mode enabled');
|
|
17
|
+
}
|
|
18
|
+
process.env.NODE_ENV = process.env.NODE_ENV || 'production';
|
|
19
|
+
const port = parseInt(values.port || process.env.WEB_APP_PORT || DEFAULT_WEB_APP_PORT.toString());
|
|
20
|
+
async function start() {
|
|
21
|
+
const useJsonMode = values.json || false;
|
|
22
|
+
const transportType = useJsonMode ? 'streamableHttpJson' : 'streamableHttp';
|
|
23
|
+
const webServer = new WebServer();
|
|
24
|
+
const app = new Application({
|
|
25
|
+
transportType,
|
|
26
|
+
webAppPort: port,
|
|
27
|
+
webServerInstance: webServer,
|
|
28
|
+
});
|
|
29
|
+
await app.start();
|
|
30
|
+
const shutdown = async () => {
|
|
31
|
+
logger.info('Shutting down server...');
|
|
32
|
+
try {
|
|
33
|
+
await app.stop();
|
|
34
|
+
logger.info('Server shutdown complete');
|
|
35
|
+
process.exit(0);
|
|
36
|
+
}
|
|
37
|
+
catch (error) {
|
|
38
|
+
logger.error({ error }, 'Error during shutdown');
|
|
39
|
+
process.exit(1);
|
|
40
|
+
}
|
|
41
|
+
};
|
|
42
|
+
process.once('SIGINT', () => {
|
|
43
|
+
void shutdown();
|
|
44
|
+
});
|
|
45
|
+
process.once('SIGTERM', () => {
|
|
46
|
+
void shutdown();
|
|
47
|
+
});
|
|
48
|
+
}
|
|
49
|
+
start().catch((error) => {
|
|
50
|
+
logger.error({ error }, 'Server startup error');
|
|
51
|
+
process.exit(1);
|
|
52
|
+
});
|
|
53
|
+
//# sourceMappingURL=streamableHttp.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"streamableHttp.js","sourceRoot":"","sources":["../../src/server/streamableHttp.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAGzC,MAAM,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC;IAC5B,OAAO,EAAE;QACR,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,EAAE;QACpC,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE;KACrC;IACD,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;CAC3B,CAAC,CAAC;AAEH,MAAM,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;AAClD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;IACjB,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;AAC3C,CAAC;AAGD,OAAO,CAAC,GAAG,CAAC,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,YAAY,CAAC;AAG5D,MAAM,IAAI,GAAG,QAAQ,CAAE,MAAM,CAAC,IAAe,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,oBAAoB,CAAC,QAAQ,EAAE,CAAC,CAAC;AAE9G,KAAK,UAAU,KAAK;IACnB,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC;IAGzC,MAAM,aAAa,GAAG,WAAW,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,gBAAgB,CAAC;IAG5E,MAAM,SAAS,GAAG,IAAI,SAAS,EAAE,CAAC;IAGlC,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC;QAC3B,aAAa;QACb,UAAU,EAAE,IAAI;QAChB,iBAAiB,EAAE,SAAS;KAC5B,CAAC,CAAC;IAGH,MAAM,GAAG,CAAC,KAAK,EAAE,CAAC;IAGlB,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE;QAC3B,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACvC,IAAI,CAAC;YACJ,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YACxC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,uBAAuB,CAAC,CAAC;YACjD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;IACF,CAAC,CAAC;IAEF,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE;QAC3B,KAAK,QAAQ,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE;QAC5B,KAAK,QAAQ,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;AACJ,CAAC;AAGD,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;IAChC,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,sBAAsB,CAAC,CAAC;IAChD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC,CAAC,CAAC"}
|