@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,75 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { extractQueryParamsToHeaders } from './query-params.js';
|
|
3
|
+
describe('extractQueryParamsToHeaders', () => {
|
|
4
|
+
it('should extract bouquet query parameter to header', () => {
|
|
5
|
+
const req = {
|
|
6
|
+
query: { bouquet: 'search' },
|
|
7
|
+
};
|
|
8
|
+
const headers = {};
|
|
9
|
+
extractQueryParamsToHeaders(req, headers);
|
|
10
|
+
expect(headers['x-mcp-bouquet']).toBe('search');
|
|
11
|
+
expect(headers['x-mcp-mix']).toBeUndefined();
|
|
12
|
+
});
|
|
13
|
+
it('should extract mix query parameter to header', () => {
|
|
14
|
+
const req = {
|
|
15
|
+
query: { mix: 'all' },
|
|
16
|
+
};
|
|
17
|
+
const headers = {};
|
|
18
|
+
extractQueryParamsToHeaders(req, headers);
|
|
19
|
+
expect(headers['x-mcp-mix']).toBe('all');
|
|
20
|
+
expect(headers['x-mcp-bouquet']).toBeUndefined();
|
|
21
|
+
});
|
|
22
|
+
it('should extract both bouquet and mix query parameters', () => {
|
|
23
|
+
const req = {
|
|
24
|
+
query: { bouquet: 'search', mix: 'hf_api' },
|
|
25
|
+
};
|
|
26
|
+
const headers = {};
|
|
27
|
+
extractQueryParamsToHeaders(req, headers);
|
|
28
|
+
expect(headers['x-mcp-bouquet']).toBe('search');
|
|
29
|
+
expect(headers['x-mcp-mix']).toBe('hf_api');
|
|
30
|
+
});
|
|
31
|
+
it('should handle empty query parameters', () => {
|
|
32
|
+
const req = {
|
|
33
|
+
query: {},
|
|
34
|
+
};
|
|
35
|
+
const headers = {};
|
|
36
|
+
extractQueryParamsToHeaders(req, headers);
|
|
37
|
+
expect(headers['x-mcp-bouquet']).toBeUndefined();
|
|
38
|
+
expect(headers['x-mcp-mix']).toBeUndefined();
|
|
39
|
+
});
|
|
40
|
+
it('should handle undefined query parameters', () => {
|
|
41
|
+
const req = {
|
|
42
|
+
query: { bouquet: undefined, mix: undefined },
|
|
43
|
+
};
|
|
44
|
+
const headers = {};
|
|
45
|
+
extractQueryParamsToHeaders(req, headers);
|
|
46
|
+
expect(headers['x-mcp-bouquet']).toBeUndefined();
|
|
47
|
+
expect(headers['x-mcp-mix']).toBeUndefined();
|
|
48
|
+
});
|
|
49
|
+
it('should not overwrite existing headers if query params are undefined', () => {
|
|
50
|
+
const req = {
|
|
51
|
+
query: { bouquet: undefined, mix: undefined },
|
|
52
|
+
};
|
|
53
|
+
const headers = {
|
|
54
|
+
'x-mcp-bouquet': 'existing',
|
|
55
|
+
'x-mcp-mix': 'existing',
|
|
56
|
+
};
|
|
57
|
+
extractQueryParamsToHeaders(req, headers);
|
|
58
|
+
expect(headers['x-mcp-bouquet']).toBe('existing');
|
|
59
|
+
expect(headers['x-mcp-mix']).toBe('existing');
|
|
60
|
+
});
|
|
61
|
+
it('should preserve other headers', () => {
|
|
62
|
+
const req = {
|
|
63
|
+
query: { bouquet: 'search' },
|
|
64
|
+
};
|
|
65
|
+
const headers = {
|
|
66
|
+
authorization: 'Bearer token123',
|
|
67
|
+
'content-type': 'application/json',
|
|
68
|
+
};
|
|
69
|
+
extractQueryParamsToHeaders(req, headers);
|
|
70
|
+
expect(headers['x-mcp-bouquet']).toBe('search');
|
|
71
|
+
expect(headers['authorization']).toBe('Bearer token123');
|
|
72
|
+
expect(headers['content-type']).toBe('application/json');
|
|
73
|
+
});
|
|
74
|
+
});
|
|
75
|
+
//# sourceMappingURL=query-params.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"query-params.test.js","sourceRoot":"","sources":["../../../src/server/utils/query-params.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,2BAA2B,EAAE,MAAM,mBAAmB,CAAC;AAGhE,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC5C,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC3D,MAAM,GAAG,GAAG;YACX,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE;SACN,CAAC;QAExB,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,2BAA2B,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACvD,MAAM,GAAG,GAAG;YACX,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE;SACC,CAAC;QAExB,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,2BAA2B,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC/D,MAAM,GAAG,GAAG;YACX,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,EAAE,QAAQ,EAAE;SACrB,CAAC;QAExB,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,2BAA2B,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC/C,MAAM,GAAG,GAAG;YACX,KAAK,EAAE,EAAE;SACa,CAAC;QAExB,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,2BAA2B,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QACnD,MAAM,GAAG,GAAG;YACX,KAAK,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,SAAS,EAAE;SACvB,CAAC;QAExB,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,2BAA2B,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC9E,MAAM,GAAG,GAAG;YACX,KAAK,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,SAAS,EAAE;SACvB,CAAC;QAExB,MAAM,OAAO,GAA2B;YACvC,eAAe,EAAE,UAAU;YAC3B,WAAW,EAAE,UAAU;SACvB,CAAC;QACF,2BAA2B,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACxC,MAAM,GAAG,GAAG;YACX,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE;SACN,CAAC;QAExB,MAAM,OAAO,GAA2B;YACvC,aAAa,EAAE,iBAAiB;YAChC,cAAc,EAAE,kBAAkB;SAClC,CAAC;QACF,2BAA2B,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAE1C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACzD,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
import { type Express } from 'express';
|
|
2
|
+
import type { TransportInfo } from '../shared/transport-info.js';
|
|
3
|
+
import type { BaseTransport } from './transport/base-transport.js';
|
|
4
|
+
import type { McpApiClient } from './lib/mcp-api-client.js';
|
|
5
|
+
export declare class WebServer {
|
|
6
|
+
private app;
|
|
7
|
+
private server;
|
|
8
|
+
private transportInfo;
|
|
9
|
+
private localSharedToolStates;
|
|
10
|
+
private transport?;
|
|
11
|
+
private apiClient?;
|
|
12
|
+
constructor();
|
|
13
|
+
private setupMiddleware;
|
|
14
|
+
getApp(): Express;
|
|
15
|
+
setTransportInfo(info: TransportInfo): void;
|
|
16
|
+
setClientInfo(clientInfo: {
|
|
17
|
+
name: string;
|
|
18
|
+
version: string;
|
|
19
|
+
} | null): void;
|
|
20
|
+
initializeToolStates(): void;
|
|
21
|
+
setTransport(transport: BaseTransport): void;
|
|
22
|
+
setApiClient(apiClient: McpApiClient): void;
|
|
23
|
+
getTransportInfo(): TransportInfo;
|
|
24
|
+
start(port: number): Promise<void>;
|
|
25
|
+
stop(): Promise<void>;
|
|
26
|
+
setupStaticFiles(isDevelopment: boolean): Promise<void>;
|
|
27
|
+
setupApiRoutes(): void;
|
|
28
|
+
}
|
|
29
|
+
//# sourceMappingURL=web-server.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"web-server.d.ts","sourceRoot":"","sources":["../../src/server/web-server.ts"],"names":[],"mappings":"AAAA,OAAgB,EAAE,KAAK,OAAO,EAAE,MAAM,SAAS,CAAC;AAIhD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAGjE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAQ5D,qBAAa,SAAS;IACrB,OAAO,CAAC,GAAG,CAAU;IACrB,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,aAAa,CAKnB;IACF,OAAO,CAAC,qBAAqB,CAAmC;IAChE,OAAO,CAAC,SAAS,CAAC,CAAgB;IAClC,OAAO,CAAC,SAAS,CAAC,CAAe;;IAOjC,OAAO,CAAC,eAAe;IAIhB,MAAM,IAAI,OAAO;IAIjB,gBAAgB,CAAC,IAAI,EAAE,aAAa,GAAG,IAAI;IAI3C,aAAa,CAAC,UAAU,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,GAAG,IAAI;IAIzE,oBAAoB,IAAI,IAAI;IAS5B,YAAY,CAAC,SAAS,EAAE,aAAa,GAAG,IAAI;IAI5C,YAAY,CAAC,SAAS,EAAE,YAAY,GAAG,IAAI;IAI3C,gBAAgB,IAAI,aAAa;IAI3B,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAelC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAiBrB,gBAAgB,CAAC,aAAa,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAyC7D,cAAc,IAAI,IAAI;CA8N7B"}
|
|
@@ -0,0 +1,273 @@
|
|
|
1
|
+
import express, {} from 'express';
|
|
2
|
+
import path from 'node:path';
|
|
3
|
+
import { fileURLToPath } from 'node:url';
|
|
4
|
+
import { settingsService } from '../shared/settings.js';
|
|
5
|
+
import { logger } from './lib/logger.js';
|
|
6
|
+
import { formatMetricsForAPI } from '../shared/transport-metrics.js';
|
|
7
|
+
import { ALL_BUILTIN_TOOL_IDS } from '@llmindset/hf-mcp';
|
|
8
|
+
import { apiMetrics } from './utils/api-metrics.js';
|
|
9
|
+
import { gradioMetrics } from './utils/gradio-metrics.js';
|
|
10
|
+
const __filename = fileURLToPath(import.meta.url);
|
|
11
|
+
const __dirname = path.dirname(__filename);
|
|
12
|
+
export class WebServer {
|
|
13
|
+
app;
|
|
14
|
+
server = null;
|
|
15
|
+
transportInfo = {
|
|
16
|
+
transport: 'unknown',
|
|
17
|
+
defaultHfTokenSet: false,
|
|
18
|
+
externalApiMode: false,
|
|
19
|
+
stdioClient: null,
|
|
20
|
+
};
|
|
21
|
+
localSharedToolStates = new Map();
|
|
22
|
+
transport;
|
|
23
|
+
apiClient;
|
|
24
|
+
constructor() {
|
|
25
|
+
this.app = express();
|
|
26
|
+
this.setupMiddleware();
|
|
27
|
+
}
|
|
28
|
+
setupMiddleware() {
|
|
29
|
+
this.app.use(express.json());
|
|
30
|
+
}
|
|
31
|
+
getApp() {
|
|
32
|
+
return this.app;
|
|
33
|
+
}
|
|
34
|
+
setTransportInfo(info) {
|
|
35
|
+
this.transportInfo = info;
|
|
36
|
+
}
|
|
37
|
+
setClientInfo(clientInfo) {
|
|
38
|
+
this.transportInfo.stdioClient = clientInfo;
|
|
39
|
+
}
|
|
40
|
+
initializeToolStates() {
|
|
41
|
+
const currentSettings = settingsService.getSettings();
|
|
42
|
+
for (const toolId of ALL_BUILTIN_TOOL_IDS) {
|
|
43
|
+
const isEnabled = currentSettings.builtInTools.includes(toolId);
|
|
44
|
+
this.localSharedToolStates.set(toolId, isEnabled);
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
setTransport(transport) {
|
|
48
|
+
this.transport = transport;
|
|
49
|
+
}
|
|
50
|
+
setApiClient(apiClient) {
|
|
51
|
+
this.apiClient = apiClient;
|
|
52
|
+
}
|
|
53
|
+
getTransportInfo() {
|
|
54
|
+
return this.transportInfo;
|
|
55
|
+
}
|
|
56
|
+
async start(port) {
|
|
57
|
+
if (this.server) {
|
|
58
|
+
throw new Error('Server is already running');
|
|
59
|
+
}
|
|
60
|
+
return new Promise((resolve, reject) => {
|
|
61
|
+
this.server = this.app
|
|
62
|
+
.listen(port, () => {
|
|
63
|
+
this.transportInfo.port = port;
|
|
64
|
+
resolve();
|
|
65
|
+
})
|
|
66
|
+
.on('error', reject);
|
|
67
|
+
});
|
|
68
|
+
}
|
|
69
|
+
async stop() {
|
|
70
|
+
if (!this.server) {
|
|
71
|
+
return;
|
|
72
|
+
}
|
|
73
|
+
return new Promise((resolve, reject) => {
|
|
74
|
+
this.server?.close((err) => {
|
|
75
|
+
if (err) {
|
|
76
|
+
reject(err);
|
|
77
|
+
}
|
|
78
|
+
else {
|
|
79
|
+
this.server = null;
|
|
80
|
+
resolve();
|
|
81
|
+
}
|
|
82
|
+
});
|
|
83
|
+
});
|
|
84
|
+
}
|
|
85
|
+
async setupStaticFiles(isDevelopment) {
|
|
86
|
+
if (isDevelopment) {
|
|
87
|
+
try {
|
|
88
|
+
const { createServer: createViteServer } = await import('vite');
|
|
89
|
+
const rootDir = path.resolve(__dirname, '..', '..', '..', 'app', 'src', 'web');
|
|
90
|
+
const vite = await createViteServer({
|
|
91
|
+
configFile: path.resolve(__dirname, '..', '..', '..', 'app', 'vite.config.ts'),
|
|
92
|
+
server: {
|
|
93
|
+
middlewareMode: true,
|
|
94
|
+
hmr: true,
|
|
95
|
+
},
|
|
96
|
+
appType: 'spa',
|
|
97
|
+
root: rootDir,
|
|
98
|
+
});
|
|
99
|
+
this.app.use(vite.middlewares);
|
|
100
|
+
logger.info('Using Vite middleware in development mode with HMR enabled');
|
|
101
|
+
logger.info({ rootDir }, 'Vite root directory');
|
|
102
|
+
}
|
|
103
|
+
catch (err) {
|
|
104
|
+
logger.error({ err }, 'Error setting up Vite middleware');
|
|
105
|
+
throw err;
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
else {
|
|
109
|
+
const staticPath = path.join(__dirname, '..', 'web');
|
|
110
|
+
this.app.use(express.static(staticPath));
|
|
111
|
+
this.app.get('*', (req, res) => {
|
|
112
|
+
if (!req.path.startsWith('/api/')) {
|
|
113
|
+
res.sendFile(path.join(staticPath, 'index.html'));
|
|
114
|
+
}
|
|
115
|
+
});
|
|
116
|
+
}
|
|
117
|
+
}
|
|
118
|
+
setupApiRoutes() {
|
|
119
|
+
this.app.get('/api/transport', (_req, res) => {
|
|
120
|
+
res.json(this.transportInfo);
|
|
121
|
+
});
|
|
122
|
+
this.app.get('/api/sessions', (_req, res) => {
|
|
123
|
+
if (!this.transport) {
|
|
124
|
+
res.json([]);
|
|
125
|
+
return;
|
|
126
|
+
}
|
|
127
|
+
const sessions = this.transport.getSessions();
|
|
128
|
+
if (this.transportInfo.transport === 'stdio' && sessions.length > 0) {
|
|
129
|
+
const stdioSession = sessions[0];
|
|
130
|
+
if (stdioSession?.clientInfo && !this.transportInfo.stdioClient) {
|
|
131
|
+
this.transportInfo.stdioClient = {
|
|
132
|
+
name: stdioSession.clientInfo.name,
|
|
133
|
+
version: stdioSession.clientInfo.version,
|
|
134
|
+
};
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
res.json(sessions);
|
|
138
|
+
});
|
|
139
|
+
this.app.get('/api/transport-metrics', (_req, res) => {
|
|
140
|
+
if (!this.transport) {
|
|
141
|
+
res.status(503).json({ error: 'Transport not initialized' });
|
|
142
|
+
return;
|
|
143
|
+
}
|
|
144
|
+
try {
|
|
145
|
+
const metrics = this.transport.getMetrics();
|
|
146
|
+
const isStateless = this.transportInfo.transport === 'streamableHttpJson';
|
|
147
|
+
const config = this.transport.getConfiguration();
|
|
148
|
+
const sessions = this.transport.getSessions().map((session) => {
|
|
149
|
+
const fiveMinutesAgo = new Date(Date.now() - 5 * 60 * 1000);
|
|
150
|
+
const hasRecentActivity = session.lastActivity > fiveMinutesAgo;
|
|
151
|
+
const hasPingFailures = (session.pingFailures || 0) >= 1;
|
|
152
|
+
let connectionStatus;
|
|
153
|
+
if (!hasRecentActivity) {
|
|
154
|
+
connectionStatus = 'Disconnected';
|
|
155
|
+
}
|
|
156
|
+
else if (hasPingFailures) {
|
|
157
|
+
connectionStatus = 'Distressed';
|
|
158
|
+
}
|
|
159
|
+
else {
|
|
160
|
+
connectionStatus = 'Connected';
|
|
161
|
+
}
|
|
162
|
+
return {
|
|
163
|
+
id: session.id,
|
|
164
|
+
connectedAt: session.connectedAt.toISOString(),
|
|
165
|
+
lastActivity: session.lastActivity.toISOString(),
|
|
166
|
+
requestCount: session.requestCount,
|
|
167
|
+
clientInfo: session.clientInfo,
|
|
168
|
+
isConnected: hasRecentActivity,
|
|
169
|
+
connectionStatus,
|
|
170
|
+
pingFailures: session.pingFailures || 0,
|
|
171
|
+
lastPingAttempt: session.lastPingAttempt?.toISOString(),
|
|
172
|
+
};
|
|
173
|
+
});
|
|
174
|
+
const formattedMetrics = formatMetricsForAPI(metrics, this.transportInfo.transport, isStateless, sessions);
|
|
175
|
+
if (!isStateless && config.staleCheckInterval && config.staleTimeout) {
|
|
176
|
+
formattedMetrics.configuration = {
|
|
177
|
+
heartbeatInterval: config.heartbeatInterval || 30000,
|
|
178
|
+
staleCheckInterval: config.staleCheckInterval,
|
|
179
|
+
staleTimeout: config.staleTimeout,
|
|
180
|
+
pingEnabled: config.pingEnabled,
|
|
181
|
+
pingInterval: config.pingInterval,
|
|
182
|
+
pingFailureThreshold: config.pingFailureThreshold || 1,
|
|
183
|
+
};
|
|
184
|
+
}
|
|
185
|
+
if (this.transportInfo.externalApiMode) {
|
|
186
|
+
formattedMetrics.apiMetrics = apiMetrics.getMetrics();
|
|
187
|
+
}
|
|
188
|
+
formattedMetrics.gradioMetrics = gradioMetrics.getMetrics();
|
|
189
|
+
res.json(formattedMetrics);
|
|
190
|
+
}
|
|
191
|
+
catch (error) {
|
|
192
|
+
logger.error({ error }, 'Error retrieving transport metrics');
|
|
193
|
+
res.status(500).json({ error: 'Failed to retrieve transport metrics' });
|
|
194
|
+
}
|
|
195
|
+
});
|
|
196
|
+
this.app.get('/api/settings', (_req, res) => {
|
|
197
|
+
res.json(settingsService.getSettings());
|
|
198
|
+
});
|
|
199
|
+
this.app.post('/api/settings', express.json(), (req, res) => {
|
|
200
|
+
const { builtInTools, spaceTools } = req.body;
|
|
201
|
+
let updatedSettings = settingsService.getSettings();
|
|
202
|
+
if (builtInTools !== undefined) {
|
|
203
|
+
updatedSettings = settingsService.updateBuiltInTools(builtInTools);
|
|
204
|
+
}
|
|
205
|
+
if (spaceTools !== undefined) {
|
|
206
|
+
updatedSettings = settingsService.updateSpaceTools(spaceTools);
|
|
207
|
+
}
|
|
208
|
+
if (builtInTools !== undefined) {
|
|
209
|
+
for (const toolId of ALL_BUILTIN_TOOL_IDS) {
|
|
210
|
+
const shouldBeEnabled = builtInTools.includes(toolId);
|
|
211
|
+
const currentlyEnabled = this.localSharedToolStates.get(toolId) ?? false;
|
|
212
|
+
if (currentlyEnabled !== shouldBeEnabled) {
|
|
213
|
+
this.localSharedToolStates.set(toolId, shouldBeEnabled);
|
|
214
|
+
if (this.apiClient) {
|
|
215
|
+
this.apiClient.emit('toolStateChange', toolId, shouldBeEnabled);
|
|
216
|
+
}
|
|
217
|
+
logger.info(`Tool ${toolId} has been ${shouldBeEnabled ? 'enabled' : 'disabled'} via API`);
|
|
218
|
+
}
|
|
219
|
+
}
|
|
220
|
+
}
|
|
221
|
+
res.json(updatedSettings);
|
|
222
|
+
});
|
|
223
|
+
this.app.get('/api/gradio-endpoints', (_req, res) => {
|
|
224
|
+
if (!this.apiClient) {
|
|
225
|
+
res.json([]);
|
|
226
|
+
return;
|
|
227
|
+
}
|
|
228
|
+
res.json(this.apiClient.getGradioEndpoints());
|
|
229
|
+
});
|
|
230
|
+
this.app.post('/api/gradio-endpoints/:index', express.json(), (req, res) => {
|
|
231
|
+
const index = parseInt(req.params.index);
|
|
232
|
+
const { enabled } = req.body;
|
|
233
|
+
if (!this.apiClient) {
|
|
234
|
+
res.status(500).json({ error: 'API client not initialized' });
|
|
235
|
+
return;
|
|
236
|
+
}
|
|
237
|
+
const endpoints = this.apiClient.getGradioEndpoints();
|
|
238
|
+
if (index < 0 || index >= endpoints.length) {
|
|
239
|
+
res.status(404).json({ error: 'Endpoint not found' });
|
|
240
|
+
return;
|
|
241
|
+
}
|
|
242
|
+
this.apiClient.updateGradioEndpointState(index, enabled);
|
|
243
|
+
const endpoint = endpoints[index];
|
|
244
|
+
if (endpoint) {
|
|
245
|
+
const toolId = `gradio_${endpoint.subdomain}`;
|
|
246
|
+
this.apiClient.emit('toolStateChange', toolId, enabled);
|
|
247
|
+
}
|
|
248
|
+
const updatedEndpoint = endpoints[index];
|
|
249
|
+
res.json(updatedEndpoint);
|
|
250
|
+
});
|
|
251
|
+
this.app.put('/api/gradio-endpoints/:index', express.json(), (req, res) => {
|
|
252
|
+
const index = parseInt(req.params.index);
|
|
253
|
+
const { name, subdomain, id, emoji } = req.body;
|
|
254
|
+
if (!this.apiClient) {
|
|
255
|
+
res.status(500).json({ error: 'API client not initialized' });
|
|
256
|
+
return;
|
|
257
|
+
}
|
|
258
|
+
const endpoints = this.apiClient.getGradioEndpoints();
|
|
259
|
+
if (index < 0 || index >= endpoints.length) {
|
|
260
|
+
res.status(404).json({ error: 'Endpoint not found' });
|
|
261
|
+
return;
|
|
262
|
+
}
|
|
263
|
+
if (!name || !subdomain) {
|
|
264
|
+
res.status(400).json({ error: 'Name and subdomain are required' });
|
|
265
|
+
return;
|
|
266
|
+
}
|
|
267
|
+
const updatedEndpoint = { name, subdomain, id, emoji };
|
|
268
|
+
this.apiClient.updateGradioEndpoint(index, updatedEndpoint);
|
|
269
|
+
res.json(updatedEndpoint);
|
|
270
|
+
});
|
|
271
|
+
}
|
|
272
|
+
}
|
|
273
|
+
//# sourceMappingURL=web-server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"web-server.js","sourceRoot":"","sources":["../../src/server/web-server.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,EAAE,EAAgB,MAAM,SAAS,CAAC;AAChD,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAGzC,OAAO,EAAE,eAAe,EAAkB,MAAM,uBAAuB,CAAC;AACxE,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAGzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAC;AACrE,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;AAE3C,MAAM,OAAO,SAAS;IACb,GAAG,CAAU;IACb,MAAM,GAAkB,IAAI,CAAC;IAC7B,aAAa,GAAkB;QACtC,SAAS,EAAE,SAAS;QACpB,iBAAiB,EAAE,KAAK;QACxB,eAAe,EAAE,KAAK;QACtB,WAAW,EAAE,IAAI;KACjB,CAAC;IACM,qBAAqB,GAAyB,IAAI,GAAG,EAAE,CAAC;IACxD,SAAS,CAAiB;IAC1B,SAAS,CAAgB;IAEjC;QACC,IAAI,CAAC,GAAG,GAAG,OAAO,EAAa,CAAC;QAChC,IAAI,CAAC,eAAe,EAAE,CAAC;IACxB,CAAC;IAEO,eAAe;QACtB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAC9B,CAAC;IAEM,MAAM;QACZ,OAAO,IAAI,CAAC,GAAG,CAAC;IACjB,CAAC;IAEM,gBAAgB,CAAC,IAAmB;QAC1C,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;IAC3B,CAAC;IAEM,aAAa,CAAC,UAAoD;QACxE,IAAI,CAAC,aAAa,CAAC,WAAW,GAAG,UAAU,CAAC;IAC7C,CAAC;IAEM,oBAAoB;QAE1B,MAAM,eAAe,GAAG,eAAe,CAAC,WAAW,EAAE,CAAC;QACtD,KAAK,MAAM,MAAM,IAAI,oBAAoB,EAAE,CAAC;YAC3C,MAAM,SAAS,GAAG,eAAe,CAAC,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAChE,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QACnD,CAAC;IACF,CAAC;IAEM,YAAY,CAAC,SAAwB;QAC3C,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC5B,CAAC;IAEM,YAAY,CAAC,SAAuB;QAC1C,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC5B,CAAC;IAEM,gBAAgB;QACtB,OAAO,IAAI,CAAC,aAAa,CAAC;IAC3B,CAAC;IAEM,KAAK,CAAC,KAAK,CAAC,IAAY;QAC9B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC9C,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACtC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG;iBACpB,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;gBAClB,IAAI,CAAC,aAAa,CAAC,IAAI,GAAG,IAAI,CAAC;gBAC/B,OAAO,EAAE,CAAC;YACX,CAAC,CAAC;iBACD,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACvB,CAAC,CAAC,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,IAAI;QAChB,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YAClB,OAAO;QACR,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACtC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC1B,IAAI,GAAG,EAAE,CAAC;oBACT,MAAM,CAAC,GAAG,CAAC,CAAC;gBACb,CAAC;qBAAM,CAAC;oBACP,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;oBACnB,OAAO,EAAE,CAAC;gBACX,CAAC;YACF,CAAC,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,aAAsB;QACnD,IAAI,aAAa,EAAE,CAAC;YAEnB,IAAI,CAAC;gBACJ,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,CAAC;gBAChE,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;gBAG/E,MAAM,IAAI,GAAG,MAAM,gBAAgB,CAAC;oBACnC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,CAAC;oBAC9E,MAAM,EAAE;wBACP,cAAc,EAAE,IAAI;wBACpB,GAAG,EAAE,IAAI;qBACT;oBACD,OAAO,EAAE,KAAK;oBACd,IAAI,EAAE,OAAO;iBACb,CAAC,CAAC;gBAGH,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBAE/B,MAAM,CAAC,IAAI,CAAC,4DAA4D,CAAC,CAAC;gBAC1E,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,EAAE,qBAAqB,CAAC,CAAC;YACjD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACd,MAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,kCAAkC,CAAC,CAAC;gBAC1D,MAAM,GAAG,CAAC;YACX,CAAC;QACF,CAAC;aAAM,CAAC;YAEP,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;YACrD,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;YAGzC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;gBAC9B,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;oBACnC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC,CAAC;gBACnD,CAAC;YACF,CAAC,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAEM,cAAc;QAEpB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YAC5C,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAGH,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YAC3C,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACrB,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACb,OAAO;YACR,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC;YAG9C,IAAI,IAAI,CAAC,aAAa,CAAC,SAAS,KAAK,OAAO,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrE,MAAM,YAAY,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;gBACjC,IAAI,YAAY,EAAE,UAAU,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;oBACjE,IAAI,CAAC,aAAa,CAAC,WAAW,GAAG;wBAChC,IAAI,EAAE,YAAY,CAAC,UAAU,CAAC,IAAI;wBAClC,OAAO,EAAE,YAAY,CAAC,UAAU,CAAC,OAAO;qBACxC,CAAC;gBACH,CAAC;YACF,CAAC;YAED,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpB,CAAC,CAAC,CAAC;QAGH,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,wBAAwB,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YACpD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACrB,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,2BAA2B,EAAE,CAAC,CAAC;gBAC7D,OAAO;YACR,CAAC;YAED,IAAI,CAAC;gBAEJ,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;gBAG5C,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,KAAK,oBAAoB,CAAC;gBAG1E,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC;gBAGjD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;oBAE7D,MAAM,cAAc,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;oBAC5D,MAAM,iBAAiB,GAAG,OAAO,CAAC,YAAY,GAAG,cAAc,CAAC;oBAChE,MAAM,eAAe,GAAG,CAAC,OAAO,CAAC,YAAY,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC;oBAIzD,IAAI,gBAA6D,CAAC;oBAClE,IAAI,CAAC,iBAAiB,EAAE,CAAC;wBACxB,gBAAgB,GAAG,cAAc,CAAC;oBACnC,CAAC;yBAAM,IAAI,eAAe,EAAE,CAAC;wBAC5B,gBAAgB,GAAG,YAAY,CAAC;oBACjC,CAAC;yBAAM,CAAC;wBACP,gBAAgB,GAAG,WAAW,CAAC;oBAChC,CAAC;oBAED,OAAO;wBACN,EAAE,EAAE,OAAO,CAAC,EAAE;wBACd,WAAW,EAAE,OAAO,CAAC,WAAW,CAAC,WAAW,EAAE;wBAC9C,YAAY,EAAE,OAAO,CAAC,YAAY,CAAC,WAAW,EAAE;wBAChD,YAAY,EAAE,OAAO,CAAC,YAAY;wBAClC,UAAU,EAAE,OAAO,CAAC,UAAU;wBAC9B,WAAW,EAAE,iBAAiB;wBAC9B,gBAAgB;wBAChB,YAAY,EAAE,OAAO,CAAC,YAAY,IAAI,CAAC;wBACvC,eAAe,EAAE,OAAO,CAAC,eAAe,EAAE,WAAW,EAAE;qBACvD,CAAC;gBACH,CAAC,CAAC,CAAC;gBAGH,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;gBAG3G,IAAI,CAAC,WAAW,IAAI,MAAM,CAAC,kBAAkB,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;oBACtE,gBAAgB,CAAC,aAAa,GAAG;wBAChC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,KAAK;wBACpD,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;wBAC7C,YAAY,EAAE,MAAM,CAAC,YAAY;wBACjC,WAAW,EAAE,MAAM,CAAC,WAAW;wBAC/B,YAAY,EAAE,MAAM,CAAC,YAAY;wBACjC,oBAAoB,EAAE,MAAM,CAAC,oBAAoB,IAAI,CAAC;qBACtD,CAAC;gBACH,CAAC;gBAGD,IAAI,IAAI,CAAC,aAAa,CAAC,eAAe,EAAE,CAAC;oBACxC,gBAAgB,CAAC,UAAU,GAAG,UAAU,CAAC,UAAU,EAAE,CAAC;gBACvD,CAAC;gBAGD,gBAAgB,CAAC,aAAa,GAAG,aAAa,CAAC,UAAU,EAAE,CAAC;gBAE5D,GAAG,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAC5B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,oCAAoC,CAAC,CAAC;gBAC9D,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,sCAAsC,EAAE,CAAC,CAAC;YACzE,CAAC;QACF,CAAC,CAAC,CAAC;QAGH,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YAC3C,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAGH,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YAC3D,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,GAAG,CAAC,IAA6D,CAAC;YAEvG,IAAI,eAAe,GAAG,eAAe,CAAC,WAAW,EAAE,CAAC;YAEpD,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBAChC,eAAe,GAAG,eAAe,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;YACpE,CAAC;YAED,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;gBAC9B,eAAe,GAAG,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;YAChE,CAAC;YAGD,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBAChC,KAAK,MAAM,MAAM,IAAI,oBAAoB,EAAE,CAAC;oBAC3C,MAAM,eAAe,GAAG,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;oBACtD,MAAM,gBAAgB,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC;oBAGzE,IAAI,gBAAgB,KAAK,eAAe,EAAE,CAAC;wBAC1C,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;wBAExD,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;4BACpB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,iBAAiB,EAAE,MAAM,EAAE,eAAe,CAAC,CAAC;wBACjE,CAAC;wBACD,MAAM,CAAC,IAAI,CAAC,QAAQ,MAAM,aAAa,eAAe,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,UAAU,CAAC,CAAC;oBAC5F,CAAC;gBACF,CAAC;YACF,CAAC;YAED,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;QAGH,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,uBAAuB,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YACnD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACrB,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACb,OAAO;YACR,CAAC;YACD,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAGH,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,8BAA8B,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YAC1E,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzC,MAAM,EAAE,OAAO,EAAE,GAAG,GAAG,CAAC,IAA4B,CAAC;YAErD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACrB,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,4BAA4B,EAAE,CAAC,CAAC;gBAC9D,OAAO;YACR,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,CAAC;YACtD,IAAI,KAAK,GAAG,CAAC,IAAI,KAAK,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;gBAC5C,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC,CAAC;gBACtD,OAAO;YACR,CAAC;YAGD,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAGzD,MAAM,QAAQ,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;YAClC,IAAI,QAAQ,EAAE,CAAC;gBACd,MAAM,MAAM,GAAG,UAAU,QAAQ,CAAC,SAAS,EAAE,CAAC;gBAC9C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,iBAAiB,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YACzD,CAAC;YAGD,MAAM,eAAe,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;YAEzC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;QAGH,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,8BAA8B,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACzE,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzC,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,GAAG,CAAC,IAK1C,CAAC;YAEF,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACrB,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,4BAA4B,EAAE,CAAC,CAAC;gBAC9D,OAAO;YACR,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,kBAAkB,EAAE,CAAC;YACtD,IAAI,KAAK,GAAG,CAAC,IAAI,KAAK,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;gBAC5C,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC,CAAC;gBACtD,OAAO;YACR,CAAC;YAGD,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACzB,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,iCAAiC,EAAE,CAAC,CAAC;gBACnE,OAAO;YACR,CAAC;YAGD,MAAM,eAAe,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;YACvD,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;YAE5D,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;IACJ,CAAC;CACD"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
export type TransportType = 'stdio' | 'sse' | 'streamableHttp' | 'streamableHttpJson' | 'unknown';
|
|
2
|
+
export declare const DEFAULT_WEB_APP_PORT = 3000;
|
|
3
|
+
export declare const GRADIO_PREFIX = "gr";
|
|
4
|
+
export declare const GRADIO_PRIVATE_PREFIX = "grp";
|
|
5
|
+
//# sourceMappingURL=constants.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../src/shared/constants.ts"],"names":[],"mappings":"AAKA,MAAM,MAAM,aAAa,GAAG,OAAO,GAAG,KAAK,GAAG,gBAAgB,GAAG,oBAAoB,GAAG,SAAS,CAAC;AAGlG,eAAO,MAAM,oBAAoB,OAAO,CAAC;AAGzC,eAAO,MAAM,aAAa,OAAO,CAAC;AAClC,eAAO,MAAM,qBAAqB,QAAQ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/shared/constants.ts"],"names":[],"mappings":"AAQA,MAAM,CAAC,MAAM,oBAAoB,GAAG,IAAI,CAAC;AAGzC,MAAM,CAAC,MAAM,aAAa,GAAG,IAAI,CAAC;AAClC,MAAM,CAAC,MAAM,qBAAqB,GAAG,KAAK,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
export interface SpaceTool {
|
|
2
|
+
_id: string;
|
|
3
|
+
name: string;
|
|
4
|
+
subdomain: string;
|
|
5
|
+
emoji: string;
|
|
6
|
+
}
|
|
7
|
+
export interface AppSettings {
|
|
8
|
+
builtInTools: string[];
|
|
9
|
+
spaceTools: SpaceTool[];
|
|
10
|
+
}
|
|
11
|
+
export declare const DEFAULT_SPACE_TOOLS: SpaceTool[];
|
|
12
|
+
export declare const settingsService: {
|
|
13
|
+
getSettings(): AppSettings;
|
|
14
|
+
updateBuiltInTools(builtInTools: string[]): AppSettings;
|
|
15
|
+
updateSpaceTools(spaceTools: SpaceTool[]): AppSettings;
|
|
16
|
+
resetSettings(): AppSettings;
|
|
17
|
+
isToolEnabled(toolId: string): boolean;
|
|
18
|
+
};
|
|
19
|
+
//# sourceMappingURL=settings.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"settings.d.ts","sourceRoot":"","sources":["../../src/shared/settings.ts"],"names":[],"mappings":"AAQA,MAAM,WAAW,SAAS;IACzB,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,WAAW;IAC3B,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,UAAU,EAAE,SAAS,EAAE,CAAC;CACxB;AAGD,eAAO,MAAM,mBAAmB,EAAE,SAAS,EAa1C,CAAC;AAYF,eAAO,MAAM,eAAe;mBAIZ,WAAW;qCAOO,MAAM,EAAE,GAAG,WAAW;iCAW1B,SAAS,EAAE,GAAG,WAAW;qBAWrC,WAAW;0BAQN,MAAM,GAAG,OAAO;CAGtC,CAAC"}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
import { ALL_BUILTIN_TOOL_IDS } from '@llmindset/hf-mcp';
|
|
2
|
+
export const DEFAULT_SPACE_TOOLS = [
|
|
3
|
+
{
|
|
4
|
+
_id: '6755d0d9e0ea01e11fa2a38a',
|
|
5
|
+
name: 'evalstate/flux1_schnell',
|
|
6
|
+
subdomain: 'evalstate-flux1-schnell',
|
|
7
|
+
emoji: '🏎️💨',
|
|
8
|
+
},
|
|
9
|
+
{
|
|
10
|
+
_id: '680be03dc38b7fa7d6855910',
|
|
11
|
+
name: 'abidlabs/EasyGhibli',
|
|
12
|
+
subdomain: 'abidlabs-easyghibli',
|
|
13
|
+
emoji: '🦀',
|
|
14
|
+
},
|
|
15
|
+
];
|
|
16
|
+
const defaultSettings = {
|
|
17
|
+
builtInTools: [...ALL_BUILTIN_TOOL_IDS],
|
|
18
|
+
spaceTools: [...DEFAULT_SPACE_TOOLS],
|
|
19
|
+
};
|
|
20
|
+
let settings = { ...defaultSettings };
|
|
21
|
+
export const settingsService = {
|
|
22
|
+
getSettings() {
|
|
23
|
+
return { ...settings };
|
|
24
|
+
},
|
|
25
|
+
updateBuiltInTools(builtInTools) {
|
|
26
|
+
settings = {
|
|
27
|
+
...settings,
|
|
28
|
+
builtInTools: [...builtInTools],
|
|
29
|
+
};
|
|
30
|
+
return { ...settings };
|
|
31
|
+
},
|
|
32
|
+
updateSpaceTools(spaceTools) {
|
|
33
|
+
settings = {
|
|
34
|
+
...settings,
|
|
35
|
+
spaceTools: [...spaceTools],
|
|
36
|
+
};
|
|
37
|
+
return { ...settings };
|
|
38
|
+
},
|
|
39
|
+
resetSettings() {
|
|
40
|
+
settings = { ...defaultSettings };
|
|
41
|
+
return { ...settings };
|
|
42
|
+
},
|
|
43
|
+
isToolEnabled(toolId) {
|
|
44
|
+
return settings.builtInTools.includes(toolId);
|
|
45
|
+
},
|
|
46
|
+
};
|
|
47
|
+
//# sourceMappingURL=settings.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"settings.js","sourceRoot":"","sources":["../../src/shared/settings.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAgBzD,MAAM,CAAC,MAAM,mBAAmB,GAAgB;IAC/C;QACC,GAAG,EAAE,0BAA0B;QAC/B,IAAI,EAAE,yBAAyB;QAC/B,SAAS,EAAE,yBAAyB;QACpC,KAAK,EAAE,OAAO;KACd;IACD;QACC,GAAG,EAAE,0BAA0B;QAC/B,IAAI,EAAE,qBAAqB;QAC3B,SAAS,EAAE,qBAAqB;QAChC,KAAK,EAAE,IAAI;KACX;CACD,CAAC;AAGF,MAAM,eAAe,GAAgB;IACpC,YAAY,EAAE,CAAC,GAAG,oBAAoB,CAAC;IACvC,UAAU,EAAE,CAAC,GAAG,mBAAmB,CAAC;CACpC,CAAC;AAGF,IAAI,QAAQ,GAAgB,EAAE,GAAG,eAAe,EAAE,CAAC;AAGnD,MAAM,CAAC,MAAM,eAAe,GAAG;IAI9B,WAAW;QACV,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;IACxB,CAAC;IAKD,kBAAkB,CAAC,YAAsB;QACxC,QAAQ,GAAG;YACV,GAAG,QAAQ;YACX,YAAY,EAAE,CAAC,GAAG,YAAY,CAAC;SAC/B,CAAC;QACF,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;IACxB,CAAC;IAKD,gBAAgB,CAAC,UAAuB;QACvC,QAAQ,GAAG;YACV,GAAG,QAAQ;YACX,UAAU,EAAE,CAAC,GAAG,UAAU,CAAC;SAC3B,CAAC;QACF,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;IACxB,CAAC;IAKD,aAAa;QACZ,QAAQ,GAAG,EAAE,GAAG,eAAe,EAAE,CAAC;QAClC,OAAO,EAAE,GAAG,QAAQ,EAAE,CAAC;IACxB,CAAC;IAKD,aAAa,CAAC,MAAc;QAC3B,OAAO,QAAQ,CAAC,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC/C,CAAC;CACD,CAAC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { TransportType } from './constants.js';
|
|
2
|
+
export interface TransportInfo {
|
|
3
|
+
transport: TransportType;
|
|
4
|
+
port?: number;
|
|
5
|
+
defaultHfTokenSet: boolean;
|
|
6
|
+
hfTokenMasked?: string;
|
|
7
|
+
jsonResponseEnabled?: boolean;
|
|
8
|
+
externalApiMode?: boolean;
|
|
9
|
+
stdioClient?: {
|
|
10
|
+
name: string;
|
|
11
|
+
version: string;
|
|
12
|
+
} | null;
|
|
13
|
+
}
|
|
14
|
+
export declare function getTokenDisplayText(info: TransportInfo): {
|
|
15
|
+
text: string;
|
|
16
|
+
isWarning: boolean;
|
|
17
|
+
};
|
|
18
|
+
//# sourceMappingURL=transport-info.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"transport-info.d.ts","sourceRoot":"","sources":["../../src/shared/transport-info.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAKpD,MAAM,WAAW,aAAa;IAC7B,SAAS,EAAE,aAAa,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,iBAAiB,EAAE,OAAO,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,WAAW,CAAC,EAAE;QACb,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;KAChB,GAAG,IAAI,CAAC;CACT;AAQD,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,aAAa,GAAG;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,OAAO,CAAA;CAAE,CAmB7F"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
export function getTokenDisplayText(info) {
|
|
2
|
+
if (info.transport === 'stdio') {
|
|
3
|
+
if (info.defaultHfTokenSet && info.hfTokenMasked) {
|
|
4
|
+
return { text: info.hfTokenMasked, isWarning: false };
|
|
5
|
+
}
|
|
6
|
+
else {
|
|
7
|
+
return { text: 'Warning: No token set', isWarning: true };
|
|
8
|
+
}
|
|
9
|
+
}
|
|
10
|
+
else {
|
|
11
|
+
if (!info.defaultHfTokenSet) {
|
|
12
|
+
return { text: 'Using Authorization Headers', isWarning: false };
|
|
13
|
+
}
|
|
14
|
+
else {
|
|
15
|
+
return {
|
|
16
|
+
text: `⚠️ Using ${info.hfTokenMasked || 'token'} as default`,
|
|
17
|
+
isWarning: true,
|
|
18
|
+
};
|
|
19
|
+
}
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
//# sourceMappingURL=transport-info.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"transport-info.js","sourceRoot":"","sources":["../../src/shared/transport-info.ts"],"names":[],"mappings":"AAwBA,MAAM,UAAU,mBAAmB,CAAC,IAAmB;IACtD,IAAI,IAAI,CAAC,SAAS,KAAK,OAAO,EAAE,CAAC;QAEhC,IAAI,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YAClD,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,aAAa,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QACvD,CAAC;aAAM,CAAC;YACP,OAAO,EAAE,IAAI,EAAE,uBAAuB,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;QAC3D,CAAC;IACF,CAAC;SAAM,CAAC;QAEP,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC7B,OAAO,EAAE,IAAI,EAAE,6BAA6B,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAClE,CAAC;aAAM,CAAC;YACP,OAAO;gBACN,IAAI,EAAE,YAAY,IAAI,CAAC,aAAa,IAAI,OAAO,aAAa;gBAC5D,SAAS,EAAE,IAAI;aACf,CAAC;QACH,CAAC;IACF,CAAC;AACF,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"transport-info.test.d.ts","sourceRoot":"","sources":["../../src/shared/transport-info.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,90 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { getTokenDisplayText } from './transport-info.js';
|
|
3
|
+
describe('getTokenDisplayText', () => {
|
|
4
|
+
describe('STDIO mode', () => {
|
|
5
|
+
it('should show masked token when token is set', () => {
|
|
6
|
+
const info = {
|
|
7
|
+
transport: 'stdio',
|
|
8
|
+
defaultHfTokenSet: true,
|
|
9
|
+
hfTokenMasked: 'hf_ab...xyz',
|
|
10
|
+
stdioClient: null,
|
|
11
|
+
};
|
|
12
|
+
const result = getTokenDisplayText(info);
|
|
13
|
+
expect(result.text).toBe('hf_ab...xyz');
|
|
14
|
+
expect(result.isWarning).toBe(false);
|
|
15
|
+
});
|
|
16
|
+
it('should show warning when token is not set', () => {
|
|
17
|
+
const info = {
|
|
18
|
+
transport: 'stdio',
|
|
19
|
+
defaultHfTokenSet: false,
|
|
20
|
+
stdioClient: null,
|
|
21
|
+
};
|
|
22
|
+
const result = getTokenDisplayText(info);
|
|
23
|
+
expect(result.text).toBe('Warning: No token set');
|
|
24
|
+
expect(result.isWarning).toBe(true);
|
|
25
|
+
});
|
|
26
|
+
it('should show warning when token is set but mask is missing', () => {
|
|
27
|
+
const info = {
|
|
28
|
+
transport: 'stdio',
|
|
29
|
+
defaultHfTokenSet: true,
|
|
30
|
+
hfTokenMasked: undefined,
|
|
31
|
+
stdioClient: null,
|
|
32
|
+
};
|
|
33
|
+
const result = getTokenDisplayText(info);
|
|
34
|
+
expect(result.text).toBe('Warning: No token set');
|
|
35
|
+
expect(result.isWarning).toBe(true);
|
|
36
|
+
});
|
|
37
|
+
});
|
|
38
|
+
describe('Non-STDIO modes (SSE, HTTP)', () => {
|
|
39
|
+
const nonStdioModes = ['sse', 'streamableHttp', 'streamableHttpJson'];
|
|
40
|
+
nonStdioModes.forEach((transport) => {
|
|
41
|
+
describe(`${transport} mode`, () => {
|
|
42
|
+
it('should show "Using Authorization Headers" when token is not set', () => {
|
|
43
|
+
const info = {
|
|
44
|
+
transport,
|
|
45
|
+
defaultHfTokenSet: false,
|
|
46
|
+
stdioClient: null,
|
|
47
|
+
};
|
|
48
|
+
const result = getTokenDisplayText(info);
|
|
49
|
+
expect(result.text).toBe('Using Authorization Headers');
|
|
50
|
+
expect(result.isWarning).toBe(false);
|
|
51
|
+
});
|
|
52
|
+
it('should show warning with masked token when token is set', () => {
|
|
53
|
+
const info = {
|
|
54
|
+
transport,
|
|
55
|
+
defaultHfTokenSet: true,
|
|
56
|
+
hfTokenMasked: 'hf_ab...xyz',
|
|
57
|
+
stdioClient: null,
|
|
58
|
+
};
|
|
59
|
+
const result = getTokenDisplayText(info);
|
|
60
|
+
expect(result.text).toBe('⚠️ Using hf_ab...xyz as default');
|
|
61
|
+
expect(result.isWarning).toBe(true);
|
|
62
|
+
});
|
|
63
|
+
it('should show warning with fallback when token is set but mask is missing', () => {
|
|
64
|
+
const info = {
|
|
65
|
+
transport,
|
|
66
|
+
defaultHfTokenSet: true,
|
|
67
|
+
hfTokenMasked: undefined,
|
|
68
|
+
stdioClient: null,
|
|
69
|
+
};
|
|
70
|
+
const result = getTokenDisplayText(info);
|
|
71
|
+
expect(result.text).toBe('⚠️ Using token as default');
|
|
72
|
+
expect(result.isWarning).toBe(true);
|
|
73
|
+
});
|
|
74
|
+
});
|
|
75
|
+
});
|
|
76
|
+
});
|
|
77
|
+
describe('Unknown transport mode', () => {
|
|
78
|
+
it('should handle unknown transport as non-STDIO mode', () => {
|
|
79
|
+
const info = {
|
|
80
|
+
transport: 'unknown',
|
|
81
|
+
defaultHfTokenSet: false,
|
|
82
|
+
stdioClient: null,
|
|
83
|
+
};
|
|
84
|
+
const result = getTokenDisplayText(info);
|
|
85
|
+
expect(result.text).toBe('Using Authorization Headers');
|
|
86
|
+
expect(result.isWarning).toBe(false);
|
|
87
|
+
});
|
|
88
|
+
});
|
|
89
|
+
});
|
|
90
|
+
//# sourceMappingURL=transport-info.test.js.map
|