@llmindset/hf-mcp-server 0.3.1 → 0.3.3
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/dist/server/application.d.ts +2 -2
- package/dist/server/application.d.ts.map +1 -1
- package/dist/server/application.js +1 -1
- package/dist/server/application.js.map +1 -1
- package/dist/server/gradio-endpoint-connector.d.ts.map +1 -1
- package/dist/server/gradio-endpoint-connector.js +18 -11
- package/dist/server/gradio-endpoint-connector.js.map +1 -1
- package/dist/server/mcp-proxy.d.ts.map +1 -1
- package/dist/server/mcp-proxy.js +36 -3
- package/dist/server/mcp-proxy.js.map +1 -1
- package/dist/server/mcp-server.d.ts +0 -1
- package/dist/server/mcp-server.d.ts.map +1 -1
- package/dist/server/mcp-server.js +64 -16
- package/dist/server/mcp-server.js.map +1 -1
- package/dist/server/resources/gradio-widget-content.d.ts +1 -1
- package/dist/server/resources/gradio-widget-content.d.ts.map +1 -1
- package/dist/server/resources/gradio-widget-content.js +1 -1
- package/dist/server/resources/gradio-widget-resource.d.ts +1 -1
- package/dist/server/resources/gradio-widget-resource.d.ts.map +1 -1
- package/dist/server/run-application.d.ts +7 -0
- package/dist/server/run-application.d.ts.map +1 -0
- package/dist/server/run-application.js +42 -0
- package/dist/server/run-application.js.map +1 -0
- package/dist/server/stdio.js +3 -32
- package/dist/server/stdio.js.map +1 -1
- package/dist/server/streamableHttp.js +3 -32
- package/dist/server/streamableHttp.js.map +1 -1
- package/dist/server/transport/json-rpc-errors.d.ts +4 -4
- package/dist/server/transport/json-rpc-errors.d.ts.map +1 -1
- package/dist/server/transport/json-rpc-errors.js +2 -2
- package/dist/server/transport/json-rpc-errors.js.map +1 -1
- package/dist/server/transport/stateless-http-transport.d.ts.map +1 -1
- package/dist/server/transport/stateless-http-transport.js +6 -1
- package/dist/server/transport/stateless-http-transport.js.map +1 -1
- package/dist/server/transport/stdio-transport.d.ts.map +1 -1
- package/dist/server/transport/stdio-transport.js +9 -0
- package/dist/server/transport/stdio-transport.js.map +1 -1
- package/dist/server/transport/streamable-http-transport.d.ts.map +1 -1
- package/dist/server/transport/streamable-http-transport.js +6 -1
- package/dist/server/transport/streamable-http-transport.js.map +1 -1
- package/dist/server/utils/api-metrics.d.ts +1 -1
- package/dist/server/utils/api-metrics.d.ts.map +1 -1
- package/dist/server/utils/capability-utils.d.ts +2 -1
- package/dist/server/utils/capability-utils.d.ts.map +1 -1
- package/dist/server/utils/gradio-cache.d.ts +1 -1
- package/dist/server/utils/gradio-cache.d.ts.map +1 -1
- package/dist/server/utils/gradio-discovery.d.ts +3 -2
- package/dist/server/utils/gradio-discovery.d.ts.map +1 -1
- package/dist/server/utils/gradio-discovery.js +87 -97
- package/dist/server/utils/gradio-discovery.js.map +1 -1
- package/dist/server/utils/gradio-metrics.d.ts +3 -2
- package/dist/server/utils/gradio-metrics.d.ts.map +1 -1
- package/dist/server/utils/gradio-metrics.js +1 -1
- package/dist/server/utils/gradio-metrics.js.map +1 -1
- package/dist/server/utils/gradio-result-processor.d.ts +2 -1
- package/dist/server/utils/gradio-result-processor.d.ts.map +1 -1
- package/dist/server/utils/gradio-utils.d.ts +3 -3
- package/dist/server/utils/gradio-utils.d.ts.map +1 -1
- package/dist/server/utils/gradio-utils.js +1 -1
- package/dist/server/utils/gradio-utils.js.map +1 -1
- package/dist/server/utils/hf-dataset-transport.d.ts +2 -1
- package/dist/server/utils/hf-dataset-transport.d.ts.map +1 -1
- package/dist/server/utils/mcp-api-client.d.ts +2 -1
- package/dist/server/utils/mcp-api-client.d.ts.map +1 -1
- package/dist/server/utils/mcp-api-client.js +29 -8
- package/dist/server/utils/mcp-api-client.js.map +1 -1
- package/dist/server/utils/proxy-tools-config.d.ts +0 -5
- package/dist/server/utils/proxy-tools-config.d.ts.map +1 -1
- package/dist/server/utils/proxy-tools-config.js +8 -10
- package/dist/server/utils/proxy-tools-config.js.map +1 -1
- package/dist/server/utils/query-logger.d.ts +0 -21
- package/dist/server/utils/query-logger.d.ts.map +1 -1
- package/dist/server/utils/query-logger.js +6 -29
- package/dist/server/utils/query-logger.js.map +1 -1
- package/dist/server/utils/repo-search-shim.d.ts +8 -0
- package/dist/server/utils/repo-search-shim.d.ts.map +1 -0
- package/dist/server/utils/repo-search-shim.js +116 -0
- package/dist/server/utils/repo-search-shim.js.map +1 -0
- package/dist/server/utils/streamable-http-tool-caller.d.ts.map +1 -1
- package/dist/server/utils/streamable-http-tool-caller.js +10 -1
- package/dist/server/utils/streamable-http-tool-caller.js.map +1 -1
- package/dist/server/utils/tool-selection-strategy.d.ts +2 -1
- package/dist/server/utils/tool-selection-strategy.d.ts.map +1 -1
- package/dist/server/utils/ui/audio-player.d.ts +2 -2
- package/dist/server/utils/ui/audio-player.d.ts.map +1 -1
- package/dist/server/utils/ui/audio-player.js +1 -1
- package/dist/server/utils/ui/audio-player.js.map +1 -1
- package/dist/shared/bouquet-presets.d.ts +6 -5
- package/dist/shared/bouquet-presets.d.ts.map +1 -1
- package/dist/shared/bouquet-presets.js +9 -14
- package/dist/shared/bouquet-presets.js.map +1 -1
- package/dist/shared/constants.d.ts.map +1 -1
- package/dist/shared/constants.js.map +1 -1
- package/dist/shared/repo-search-migration.d.ts +5 -0
- package/dist/shared/repo-search-migration.d.ts.map +1 -0
- package/dist/shared/repo-search-migration.js +22 -0
- package/dist/shared/repo-search-migration.js.map +1 -0
- package/dist/shared/tool-normalizer.d.ts.map +1 -1
- package/dist/shared/tool-normalizer.js +10 -8
- package/dist/shared/tool-normalizer.js.map +1 -1
- package/dist/shared/transport-metrics.d.ts +2 -3
- package/dist/shared/transport-metrics.d.ts.map +1 -1
- package/dist/shared/transport-metrics.js +2 -2
- package/dist/shared/transport-metrics.js.map +1 -1
- package/dist/web/assets/{gradioWidgetDev-BTbDFx4A.js → gradioWidgetDev-DKjSxsth.js} +1 -1
- package/dist/web/assets/index-C82JscI3.css +1 -0
- package/dist/web/assets/main-VwOEai0c.js +145 -0
- package/dist/web/gradio-widget-dev.html +3 -3
- package/dist/web/gradio-widget.html +1 -1
- package/dist/web/index.html +3 -3
- package/dist/web/mcp-welcome.html +44 -128
- package/package.json +3 -4
- package/dist/web/assets/index-DA_tuFd-.css +0 -1
- package/dist/web/assets/main-BajWSVcC.js +0 -229
- /package/dist/web/assets/{index-BVX5KR1A.js → index-CZS9IJ_2.js} +0 -0
|
@@ -1,10 +1,13 @@
|
|
|
1
1
|
import { readFile } from 'node:fs/promises';
|
|
2
2
|
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
|
|
3
3
|
import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
|
|
4
|
+
import { fetchWithProfile, NETWORK_FETCH_PROFILES, parseAndValidateUrl } from '@llmindset/hf-mcp/network';
|
|
4
5
|
import { logger } from './logger.js';
|
|
5
6
|
const PROXY_TOOLS_ENV_VAR = 'PROXY_TOOLS_CSV';
|
|
6
7
|
const VALID_RESPONSE_TYPES = new Set(['JSON', 'SSE']);
|
|
7
8
|
const PROXY_SCHEMA_TIMEOUT_MS = 10_000;
|
|
9
|
+
const PROXY_CSV_SOURCE_PROFILE = NETWORK_FETCH_PROFILES.externalHttps();
|
|
10
|
+
const PROXY_TOOL_URL_POLICY = NETWORK_FETCH_PROFILES.httpOrHttpsPermissive().urlPolicy;
|
|
8
11
|
let cachedTools = null;
|
|
9
12
|
let cachedConfigPromise = null;
|
|
10
13
|
let cachedToolsByName = new Map();
|
|
@@ -32,7 +35,9 @@ export async function loadProxyToolsConfig() {
|
|
|
32
35
|
let content = null;
|
|
33
36
|
if (source.startsWith('https://')) {
|
|
34
37
|
try {
|
|
35
|
-
const response = await
|
|
38
|
+
const { response } = await fetchWithProfile(source, PROXY_CSV_SOURCE_PROFILE, {
|
|
39
|
+
timeoutMs: PROXY_SCHEMA_TIMEOUT_MS,
|
|
40
|
+
});
|
|
36
41
|
if (!response.ok) {
|
|
37
42
|
logger.error({ status: response.status, source }, 'Failed to fetch proxy tools CSV');
|
|
38
43
|
cachedTools = [];
|
|
@@ -77,10 +82,7 @@ async function loadProxyToolSchemas(sources) {
|
|
|
77
82
|
}
|
|
78
83
|
const shouldPrefix = sources.length > 1;
|
|
79
84
|
const hfToken = process.env.LOGGING_HF_TOKEN || process.env.DEFAULT_HF_TOKEN;
|
|
80
|
-
const schemaTasks = sources.map((source) => Promise.race([
|
|
81
|
-
fetchProxyToolSchemas(source, shouldPrefix, hfToken),
|
|
82
|
-
createTimeout(PROXY_SCHEMA_TIMEOUT_MS),
|
|
83
|
-
])
|
|
85
|
+
const schemaTasks = sources.map((source) => Promise.race([fetchProxyToolSchemas(source, shouldPrefix, hfToken), createTimeout(PROXY_SCHEMA_TIMEOUT_MS)])
|
|
84
86
|
.then((tools) => ({ source, tools }))
|
|
85
87
|
.catch((error) => {
|
|
86
88
|
logger.error({ error, proxyId: source.proxyId, url: source.url }, 'Failed to fetch proxy tool schemas');
|
|
@@ -186,11 +188,7 @@ function parseProxyToolsCsv(content) {
|
|
|
186
188
|
const url = urlRaw.trim();
|
|
187
189
|
let parsedUrl;
|
|
188
190
|
try {
|
|
189
|
-
parsedUrl =
|
|
190
|
-
if (parsedUrl.protocol !== 'https:' && parsedUrl.protocol !== 'http:') {
|
|
191
|
-
logger.warn({ proxyId, url }, 'Skipping proxy tool with unsupported URL protocol');
|
|
192
|
-
continue;
|
|
193
|
-
}
|
|
191
|
+
parsedUrl = parseAndValidateUrl(url, PROXY_TOOL_URL_POLICY);
|
|
194
192
|
}
|
|
195
193
|
catch (error) {
|
|
196
194
|
logger.warn({ proxyId, url, error }, 'Skipping proxy tool with invalid URL');
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"proxy-tools-config.js","sourceRoot":"","sources":["../../../src/server/utils/proxy-tools-config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AAEnG,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAmCrC,MAAM,mBAAmB,GAAG,iBAAiB,CAAC;AAC9C,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAwB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;AAC7E,MAAM,uBAAuB,GAAG,MAAM,CAAC;AAEvC,IAAI,WAAW,GAAiC,IAAI,CAAC;AACrD,IAAI,mBAAmB,GAA0C,IAAI,CAAC;AACtE,IAAI,iBAAiB,GAAqC,IAAI,GAAG,EAAE,CAAC;AAEpE,MAAM,UAAU,mBAAmB;IAClC,OAAO,WAAW,IAAI,EAAE,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,QAAgB;IACtD,OAAO,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AACxC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB;IACzC,IAAI,WAAW,EAAE,CAAC;QACjB,OAAO,WAAW,CAAC;IACpB,CAAC;IACD,IAAI,mBAAmB,EAAE,CAAC;QACzB,OAAO,mBAAmB,CAAC;IAC5B,CAAC;IAED,mBAAmB,GAAG,CAAC,KAAK,IAAI,EAAE;QACjC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,EAAE,IAAI,EAAE,CAAC;QACxD,IAAI,CAAC,MAAM,EAAE,CAAC;YACb,WAAW,GAAG,EAAE,CAAC;YACjB,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;YAC9B,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,mBAAmB,EAAE,EAAE,gCAAgC,CAAC,CAAC;YAChF,OAAO,WAAW,CAAC;QACpB,CAAC;QAED,IAAI,OAAO,GAAkB,IAAI,CAAC;QAClC,IAAI,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YACnC,IAAI,CAAC;gBACJ,MAAM,QAAQ,GAAG,MAAM,
|
|
1
|
+
{"version":3,"file":"proxy-tools-config.js","sourceRoot":"","sources":["../../../src/server/utils/proxy-tools-config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AAEnG,OAAO,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAC1G,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAmCrC,MAAM,mBAAmB,GAAG,iBAAiB,CAAC;AAC9C,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAwB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;AAC7E,MAAM,uBAAuB,GAAG,MAAM,CAAC;AAEvC,MAAM,wBAAwB,GAAG,sBAAsB,CAAC,aAAa,EAAE,CAAC;AACxE,MAAM,qBAAqB,GAAG,sBAAsB,CAAC,qBAAqB,EAAE,CAAC,SAAS,CAAC;AAEvF,IAAI,WAAW,GAAiC,IAAI,CAAC;AACrD,IAAI,mBAAmB,GAA0C,IAAI,CAAC;AACtE,IAAI,iBAAiB,GAAqC,IAAI,GAAG,EAAE,CAAC;AAEpE,MAAM,UAAU,mBAAmB;IAClC,OAAO,WAAW,IAAI,EAAE,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,QAAgB;IACtD,OAAO,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AACxC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB;IACzC,IAAI,WAAW,EAAE,CAAC;QACjB,OAAO,WAAW,CAAC;IACpB,CAAC;IACD,IAAI,mBAAmB,EAAE,CAAC;QACzB,OAAO,mBAAmB,CAAC;IAC5B,CAAC;IAED,mBAAmB,GAAG,CAAC,KAAK,IAAI,EAAE;QACjC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,EAAE,IAAI,EAAE,CAAC;QACxD,IAAI,CAAC,MAAM,EAAE,CAAC;YACb,WAAW,GAAG,EAAE,CAAC;YACjB,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;YAC9B,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,mBAAmB,EAAE,EAAE,gCAAgC,CAAC,CAAC;YAChF,OAAO,WAAW,CAAC;QACpB,CAAC;QAED,IAAI,OAAO,GAAkB,IAAI,CAAC;QAClC,IAAI,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YACnC,IAAI,CAAC;gBACJ,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,gBAAgB,CAAC,MAAM,EAAE,wBAAwB,EAAE;oBAC7E,SAAS,EAAE,uBAAuB;iBAClC,CAAC,CAAC;gBACH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBAClB,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,iCAAiC,CAAC,CAAC;oBACrF,WAAW,GAAG,EAAE,CAAC;oBACjB,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;oBAC9B,OAAO,WAAW,CAAC;gBACpB,CAAC;gBACD,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACjC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,gCAAgC,CAAC,CAAC;gBAClE,WAAW,GAAG,EAAE,CAAC;gBACjB,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;gBAC9B,OAAO,WAAW,CAAC;YACpB,CAAC;QACF,CAAC;aAAM,CAAC;YACP,IAAI,CAAC;gBACJ,OAAO,GAAG,MAAM,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAC1C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,gCAAgC,CAAC,CAAC;gBAClE,WAAW,GAAG,EAAE,CAAC;gBACjB,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;gBAC9B,OAAO,WAAW,CAAC;YACpB,CAAC;QACF,CAAC;QAED,MAAM,aAAa,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,eAAe,GAAG,MAAM,oBAAoB,CAAC,aAAa,CAAC,CAAC;QAClE,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9D,MAAM,CAAC,KAAK,CAAC,wDAAwD,CAAC,CAAC;QACxE,CAAC;QACD,WAAW,GAAG,eAAe,CAAC;QAC9B,iBAAiB,GAAG,IAAI,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;QACrF,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,eAAe,CAAC,MAAM,EAAE,EAAE,kCAAkC,CAAC,CAAC;QACvF,OAAO,eAAe,CAAC;IACxB,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,mBAAmB,CAAC;AAC5B,CAAC;AAED,KAAK,UAAU,oBAAoB,CAAC,OAA0B;IAC7D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,EAAE,CAAC;IACX,CAAC;IAED,MAAM,YAAY,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;IACxC,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAC7E,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAC1C,OAAO,CAAC,IAAI,CAAC,CAAC,qBAAqB,CAAC,MAAM,EAAE,YAAY,EAAE,OAAO,CAAC,EAAE,aAAa,CAAC,uBAAuB,CAAC,CAAC,CAAC;SAC1G,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;SACpC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;QACzB,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,EAAE,oCAAoC,CAAC,CAAC;QACxG,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,EAA2B,EAAE,CAAC;IACvD,CAAC,CAAC,CACH,CAAC;IAEF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC/C,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAClD,CAAC;AAED,KAAK,UAAU,qBAAqB,CACnC,MAAuB,EACvB,YAAqB,EACrB,OAA2B;IAE3B,MAAM,MAAM,GAAG,IAAI,MAAM,CACxB;QACC,IAAI,EAAE,qBAAqB;QAC3B,OAAO,EAAE,OAAO;KAChB,EACD,EAAE,YAAY,EAAE,EAAE,EAAE,CACpB,CAAC;IAEF,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAC1C,MAAM,SAAS,GAAG,IAAI,6BAA6B,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE;QACxE,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS;KAC9C,CAAC,CAAC;IAEH,IAAI,CAAC;QACJ,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,OAAO,EAAE,uBAAuB,EAAE,CAAC,CAAC;QACtE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,uBAAuB,EAAE,CAAC,CAAC;QAChF,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;QAEjC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,EAAE,qCAAqC,CAAC,CAAC;YAClG,OAAO,EAAE,CAAC;QACX,CAAC;QAED,OAAO,KAAK;aACV,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,wBAAwB,CAAC,MAAM,EAAE,IAAI,EAAE,YAAY,CAAC,CAAC;aACnE,MAAM,CAAC,CAAC,IAAI,EAA+B,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IAChE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,EAAE,gCAAgC,CAAC,CAAC;QACpG,OAAO,EAAE,CAAC;IACX,CAAC;YAAS,CAAC;QACV,IAAI,CAAC;YACJ,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,EAAE,mCAAmC,CAAC,CAAC;QACxG,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAS,wBAAwB,CAChC,MAAuB,EACvB,IAAU,EACV,YAAqB;IAErB,MAAM,WAAW,GAAG,YAAY,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;IAChF,MAAM,WAAW,GAAG,IAAI,CAAC,WAA+C,CAAC;IACzE,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QACnD,MAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,sCAAsC,CAAC,CAAC;QACvG,OAAO,IAAI,CAAC;IACb,CAAC;IAED,OAAO;QACN,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,QAAQ,EAAE,WAAW;QACrB,gBAAgB,EAAE,IAAI,CAAC,IAAI;QAC3B,GAAG,EAAE,MAAM,CAAC,GAAG;QACf,YAAY,EAAE,MAAM,CAAC,YAAY;QACjC,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,WAAW;KACX,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAgB;IACzC,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,OAAO;QACN,aAAa,EAAE,UAAU,OAAO,EAAE;QAClC,oBAAoB,EAAE,UAAU,OAAO,EAAE;KACzC,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CAAC,EAAU;IAChC,OAAO,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;QAChC,UAAU,CAAC,GAAG,EAAE;YACf,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,EAAE,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC;QAClE,CAAC,EAAE,EAAE,CAAC,CAAC;IACR,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,OAAe;IAC1C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACrC,MAAM,OAAO,GAAsB,EAAE,CAAC;IACtC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAE/B,KAAK,MAAM,OAAO,IAAI,KAAK,EAAE,CAAC;QAC7B,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;QAC5B,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACnC,SAAS;QACV,CAAC;QAED,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;QAClC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAE,uDAAuD,CAAC,CAAC;YAC/E,SAAS;QACV,CAAC;QAED,MAAM,CAAC,UAAU,EAAE,MAAM,EAAE,eAAe,CAAC,GAAG,MAAM,CAAC;QACrD,IAAI,CAAC,UAAU,IAAI,CAAC,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAChD,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAE,kDAAkD,CAAC,CAAC;YAC1E,SAAS;QACV,CAAC;QAED,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,EAAE,CAAC;QAClC,IAAI,OAAO,CAAC,WAAW,EAAE,KAAK,UAAU,EAAE,CAAC;YAC1C,SAAS;QACV,CAAC;QAED,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,EAAE,0CAA0C,CAAC,CAAC;YACrE,SAAS;QACV,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;QAC1B,IAAI,SAAc,CAAC;QACnB,IAAI,CAAC;YACJ,SAAS,GAAG,mBAAmB,CAAC,GAAG,EAAE,qBAAqB,CAAC,CAAC;QAC7D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,sCAAsC,CAAC,CAAC;YAC7E,SAAS;QACV,CAAC;QAED,MAAM,YAAY,GAAG,eAAe,CAAC,IAAI,EAAE,CAAC,WAAW,EAA2B,CAAC;QACnF,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;YAC7C,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,EAAE,gDAAgD,CAAC,CAAC;YACzF,SAAS;QACV,CAAC;QAED,OAAO,CAAC,IAAI,CAAC;YACZ,OAAO;YACP,GAAG,EAAE,SAAS,CAAC,QAAQ,EAAE;YACzB,YAAY;SACZ,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACnB,CAAC;IAED,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,SAAS,YAAY,CAAC,IAAY;IACjC,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,OAAO,GAAG,EAAE,CAAC;IACjB,IAAI,QAAQ,GAAG,KAAK,CAAC;IAErB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QACzC,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACrB,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;YAClB,IAAI,QAAQ,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,GAAG,EAAE,CAAC;gBACrC,OAAO,IAAI,GAAG,CAAC;gBACf,CAAC,IAAI,CAAC,CAAC;gBACP,SAAS;YACV,CAAC;YACD,QAAQ,GAAG,CAAC,QAAQ,CAAC;YACrB,SAAS;QACV,CAAC;QAED,IAAI,IAAI,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;YAC5B,OAAO,GAAG,EAAE,CAAC;YACb,SAAS;QACV,CAAC;QAED,OAAO,IAAI,IAAI,CAAC;IACjB,CAAC;IAED,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC"}
|
|
@@ -1,21 +1,3 @@
|
|
|
1
|
-
import { type Logger } from 'pino';
|
|
2
|
-
export interface QueryLogEntry {
|
|
3
|
-
mcpServerSessionId: string;
|
|
4
|
-
clientSessionId?: string | null;
|
|
5
|
-
name?: string | null;
|
|
6
|
-
version?: string | null;
|
|
7
|
-
methodName: string;
|
|
8
|
-
query: string;
|
|
9
|
-
parameters: string;
|
|
10
|
-
isAuthenticated?: boolean;
|
|
11
|
-
totalResults?: number;
|
|
12
|
-
resultsShared?: number;
|
|
13
|
-
responseCharCount?: number;
|
|
14
|
-
requestJson?: string;
|
|
15
|
-
durationMs?: number;
|
|
16
|
-
success?: boolean;
|
|
17
|
-
errorMessage?: string | null;
|
|
18
|
-
}
|
|
19
1
|
export interface QueryLoggerOptions {
|
|
20
2
|
clientSessionId?: string;
|
|
21
3
|
isAuthenticated?: boolean;
|
|
@@ -28,8 +10,6 @@ export interface QueryLoggerOptions {
|
|
|
28
10
|
success?: boolean;
|
|
29
11
|
error?: unknown;
|
|
30
12
|
}
|
|
31
|
-
declare const queryLogger: Logger | null;
|
|
32
|
-
export declare function logQuery(entry: QueryLogEntry): void;
|
|
33
13
|
export declare function logSearchQuery(methodName: string, query: string, data: Record<string, unknown>, options?: QueryLoggerOptions): void;
|
|
34
14
|
export declare function logPromptQuery(methodName: string, query: string, data: Record<string, unknown>, options?: QueryLoggerOptions): void;
|
|
35
15
|
export declare function logSystemEvent(methodName: string, sessionId: string, options?: {
|
|
@@ -52,5 +32,4 @@ export declare function logGradioEvent(endpointName: string, sessionId: string,
|
|
|
52
32
|
notificationCount?: number;
|
|
53
33
|
isDynamic?: boolean;
|
|
54
34
|
}): void;
|
|
55
|
-
export { queryLogger };
|
|
56
35
|
//# sourceMappingURL=query-logger.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"query-logger.d.ts","sourceRoot":"","sources":["../../../src/server/utils/query-logger.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"query-logger.d.ts","sourceRoot":"","sources":["../../../src/server/utils/query-logger.ts"],"names":[],"mappings":"AAoCA,MAAM,WAAW,kBAAkB;IAClC,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,OAAO,CAAC;CAChB;AAgMD,wBAAgB,cAAc,CAC7B,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,CAAC,EAAE,kBAAkB,GAC1B,IAAI,CAEN;AAKD,wBAAgB,cAAc,CAC7B,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,CAAC,EAAE,kBAAkB,GAC1B,IAAI,CAEN;AAKD,wBAAgB,cAAc,CAC7B,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM,EACjB,OAAO,CAAC,EAAE;IACT,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,CAAC;CACnB,GACC,IAAI,CA6CN;AAKD,wBAAgB,cAAc,CAC7B,YAAY,EAAE,MAAM,EACpB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE;IACR,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,SAAS,CAAC,EAAE,OAAO,CAAC;CACpB,GACC,IAAI,CAwCN"}
|
|
@@ -114,13 +114,13 @@ const mcpServerSessionId = crypto.randomUUID();
|
|
|
114
114
|
function getMcpServerSessionId() {
|
|
115
115
|
return mcpServerSessionId;
|
|
116
116
|
}
|
|
117
|
-
|
|
117
|
+
function logQuery(entry) {
|
|
118
118
|
if (!queryLogger) {
|
|
119
119
|
return;
|
|
120
120
|
}
|
|
121
121
|
queryLogger.info(entry);
|
|
122
122
|
}
|
|
123
|
-
|
|
123
|
+
function logQueryEvent(methodName, query, data, options) {
|
|
124
124
|
const mcpServerSessionId = getMcpServerSessionId();
|
|
125
125
|
const normalizedDurationMs = options?.durationMs !== undefined ? Math.round(options.durationMs) : undefined;
|
|
126
126
|
const serializedParameters = JSON.stringify(data);
|
|
@@ -148,33 +148,11 @@ export function logSearchQuery(methodName, query, data, options) {
|
|
|
148
148
|
errorMessage: normalizedError,
|
|
149
149
|
});
|
|
150
150
|
}
|
|
151
|
+
export function logSearchQuery(methodName, query, data, options) {
|
|
152
|
+
logQueryEvent(methodName, query, data, options);
|
|
153
|
+
}
|
|
151
154
|
export function logPromptQuery(methodName, query, data, options) {
|
|
152
|
-
|
|
153
|
-
const normalizedDurationMs = options?.durationMs !== undefined ? Math.round(options.durationMs) : undefined;
|
|
154
|
-
const serializedParameters = JSON.stringify(data);
|
|
155
|
-
const requestPayload = {
|
|
156
|
-
methodName,
|
|
157
|
-
query,
|
|
158
|
-
parameters: data,
|
|
159
|
-
};
|
|
160
|
-
const normalizedError = options?.error !== undefined && options?.error !== null ? normalizeError(options.error) : null;
|
|
161
|
-
logQuery({
|
|
162
|
-
query,
|
|
163
|
-
methodName,
|
|
164
|
-
parameters: serializedParameters,
|
|
165
|
-
requestJson: JSON.stringify(requestPayload),
|
|
166
|
-
mcpServerSessionId,
|
|
167
|
-
clientSessionId: options?.clientSessionId || null,
|
|
168
|
-
isAuthenticated: options?.isAuthenticated ?? false,
|
|
169
|
-
name: options?.clientName || null,
|
|
170
|
-
version: options?.clientVersion || null,
|
|
171
|
-
totalResults: options?.totalResults,
|
|
172
|
-
resultsShared: options?.resultsShared,
|
|
173
|
-
responseCharCount: options?.responseCharCount,
|
|
174
|
-
durationMs: normalizedDurationMs,
|
|
175
|
-
success: options?.success ?? true,
|
|
176
|
-
errorMessage: normalizedError,
|
|
177
|
-
});
|
|
155
|
+
logQueryEvent(methodName, query, data, options);
|
|
178
156
|
}
|
|
179
157
|
export function logSystemEvent(methodName, sessionId, options) {
|
|
180
158
|
if (!systemLogger) {
|
|
@@ -243,7 +221,6 @@ export function logGradioEvent(endpointName, sessionId, options) {
|
|
|
243
221
|
mcpServerSessionId,
|
|
244
222
|
}, 'Gradio event logged');
|
|
245
223
|
}
|
|
246
|
-
export { queryLogger };
|
|
247
224
|
function normalizeError(error) {
|
|
248
225
|
if (error instanceof Error) {
|
|
249
226
|
return `${error.name}: ${error.message}`;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"query-logger.js","sourceRoot":"","sources":["../../../src/server/utils/query-logger.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,EAAE,EAAmC,MAAM,MAAM,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAGrC,MAAM,kBAAkB,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,MAAM,CAAC,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC;AAC7F,MAAM,mBAAmB,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,MAAM,CAAC,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC;AAC/F,MAAM,kBAAkB,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;AAG5D,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAsCtC,SAAS,iBAAiB;IAEzB,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;QACtE,OAAO,IAAI,CAAC;IACb,CAAC;IAED,IAAI,CAAC,kBAAkB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAChD,OAAO,IAAI,CAAC;IACb,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;IACjD,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAE7E,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,CAAC,IAAI,CAAC,qGAAqG,CAAC,CAAC;QACpH,OAAO,IAAI,CAAC;IACb,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,qDAAqD,SAAS,EAAE,CAAC,CAAC;IAE9E,IAAI,CAAC;QACJ,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,EAAE,yBAAyB,CAAC,CAAC;QAEjE,MAAM,WAAW,GAAkB;YAClC,KAAK,EAAE,MAAM;YACb,SAAS,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO;SACxC,CAAC;QAGF,OAAO,IAAI,CAAC;YACX,GAAG,WAAW;YACd,SAAS,EAAE;gBACV,MAAM,EAAE,aAAa;gBACrB,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE;aAC1C;SACD,CAAC,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,yDAAyD,EAAE,KAAK,CAAC,CAAC;QAChF,OAAO,IAAI,CAAC;IACb,CAAC;AACF,CAAC;AAED,MAAM,WAAW,GAAkB,iBAAiB,EAAE,CAAC;AAEvD,SAAS,kBAAkB;IAE1B,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;QACtE,OAAO,IAAI,CAAC;IACb,CAAC;IAGD,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC;IACb,CAAC;IAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAC7E,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,CAAC,IAAI,CACX,uGAAuG,CACvG,CAAC;QACF,OAAO,IAAI,CAAC;IACb,CAAC;IAED,IAAI,CAAC;QACJ,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,EAAE,yBAAyB,CAAC,CAAC;QACjE,MAAM,WAAW,GAAkB;YAClC,KAAK,EAAE,MAAM;YACb,SAAS,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO;SACxC,CAAC;QACF,OAAO,IAAI,CAAC;YACX,GAAG,WAAW;YACd,SAAS,EAAE;gBACV,MAAM,EAAE,aAAa;gBACrB,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE;aAC3C;SACD,CAAC,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,2DAA2D,EAAE,KAAK,CAAC,CAAC;QAClF,OAAO,IAAI,CAAC;IACb,CAAC;AACF,CAAC;AAED,MAAM,YAAY,GAAkB,kBAAkB,EAAE,CAAC;AAEzD,SAAS,kBAAkB;IAE1B,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;QACtE,OAAO,IAAI,CAAC;IACb,CAAC;IAGD,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC;IACb,CAAC;IAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAC7E,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,CAAC,IAAI,CACX,uGAAuG,CACvG,CAAC;QACF,OAAO,IAAI,CAAC;IACb,CAAC;IAED,IAAI,CAAC;QACJ,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,EAAE,yBAAyB,CAAC,CAAC;QACjE,MAAM,WAAW,GAAkB;YAClC,KAAK,EAAE,MAAM;YACb,SAAS,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO;SACxC,CAAC;QACF,OAAO,IAAI,CAAC;YACX,GAAG,WAAW;YACd,SAAS,EAAE;gBACV,MAAM,EAAE,aAAa;gBACrB,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE;aAC3C;SACD,CAAC,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,2DAA2D,EAAE,KAAK,CAAC,CAAC;QAClF,OAAO,IAAI,CAAC;IACb,CAAC;AACF,CAAC;AAED,MAAM,YAAY,GAAkB,kBAAkB,EAAE,CAAC;AAGzD,MAAM,kBAAkB,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;AAE/C,SAAS,qBAAqB;IAC7B,OAAO,kBAAkB,CAAC;AAC3B,CAAC;AAKD,
|
|
1
|
+
{"version":3,"file":"query-logger.js","sourceRoot":"","sources":["../../../src/server/utils/query-logger.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,EAAE,EAAmC,MAAM,MAAM,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAGrC,MAAM,kBAAkB,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,MAAM,CAAC,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC;AAC7F,MAAM,mBAAmB,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,MAAM,CAAC,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC;AAC/F,MAAM,kBAAkB,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;AAG5D,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAsCtC,SAAS,iBAAiB;IAEzB,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;QACtE,OAAO,IAAI,CAAC;IACb,CAAC;IAED,IAAI,CAAC,kBAAkB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAChD,OAAO,IAAI,CAAC;IACb,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;IACjD,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAE7E,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,CAAC,IAAI,CAAC,qGAAqG,CAAC,CAAC;QACpH,OAAO,IAAI,CAAC;IACb,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,qDAAqD,SAAS,EAAE,CAAC,CAAC;IAE9E,IAAI,CAAC;QACJ,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,EAAE,yBAAyB,CAAC,CAAC;QAEjE,MAAM,WAAW,GAAkB;YAClC,KAAK,EAAE,MAAM;YACb,SAAS,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO;SACxC,CAAC;QAGF,OAAO,IAAI,CAAC;YACX,GAAG,WAAW;YACd,SAAS,EAAE;gBACV,MAAM,EAAE,aAAa;gBACrB,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE;aAC1C;SACD,CAAC,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,yDAAyD,EAAE,KAAK,CAAC,CAAC;QAChF,OAAO,IAAI,CAAC;IACb,CAAC;AACF,CAAC;AAED,MAAM,WAAW,GAAkB,iBAAiB,EAAE,CAAC;AAEvD,SAAS,kBAAkB;IAE1B,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;QACtE,OAAO,IAAI,CAAC;IACb,CAAC;IAGD,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC;IACb,CAAC;IAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAC7E,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,CAAC,IAAI,CACX,uGAAuG,CACvG,CAAC;QACF,OAAO,IAAI,CAAC;IACb,CAAC;IAED,IAAI,CAAC;QACJ,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,EAAE,yBAAyB,CAAC,CAAC;QACjE,MAAM,WAAW,GAAkB;YAClC,KAAK,EAAE,MAAM;YACb,SAAS,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO;SACxC,CAAC;QACF,OAAO,IAAI,CAAC;YACX,GAAG,WAAW;YACd,SAAS,EAAE;gBACV,MAAM,EAAE,aAAa;gBACrB,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE;aAC3C;SACD,CAAC,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,2DAA2D,EAAE,KAAK,CAAC,CAAC;QAClF,OAAO,IAAI,CAAC;IACb,CAAC;AACF,CAAC;AAED,MAAM,YAAY,GAAkB,kBAAkB,EAAE,CAAC;AAEzD,SAAS,kBAAkB;IAE1B,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;QACtE,OAAO,IAAI,CAAC;IACb,CAAC;IAGD,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC1B,OAAO,IAAI,CAAC;IACb,CAAC;IAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAC7E,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,CAAC,IAAI,CACX,uGAAuG,CACvG,CAAC;QACF,OAAO,IAAI,CAAC;IACb,CAAC;IAED,IAAI,CAAC;QACJ,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,EAAE,yBAAyB,CAAC,CAAC;QACjE,MAAM,WAAW,GAAkB;YAClC,KAAK,EAAE,MAAM;YACb,SAAS,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO;SACxC,CAAC;QACF,OAAO,IAAI,CAAC;YACX,GAAG,WAAW;YACd,SAAS,EAAE;gBACV,MAAM,EAAE,aAAa;gBACrB,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE;aAC3C;SACD,CAAC,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,2DAA2D,EAAE,KAAK,CAAC,CAAC;QAClF,OAAO,IAAI,CAAC;IACb,CAAC;AACF,CAAC;AAED,MAAM,YAAY,GAAkB,kBAAkB,EAAE,CAAC;AAGzD,MAAM,kBAAkB,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;AAE/C,SAAS,qBAAqB;IAC7B,OAAO,kBAAkB,CAAC;AAC3B,CAAC;AAKD,SAAS,QAAQ,CAAC,KAAoB;IACrC,IAAI,CAAC,WAAW,EAAE,CAAC;QAClB,OAAO;IACR,CAAC;IAED,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACzB,CAAC;AAED,SAAS,aAAa,CACrB,UAAkB,EAClB,KAAa,EACb,IAA6B,EAC7B,OAA4B;IAG5B,MAAM,kBAAkB,GAAG,qBAAqB,EAAE,CAAC;IACnD,MAAM,oBAAoB,GAAG,OAAO,EAAE,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC5G,MAAM,oBAAoB,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAClD,MAAM,cAAc,GAAG;QACtB,UAAU;QACV,KAAK;QACL,UAAU,EAAE,IAAI;KAChB,CAAC;IACF,MAAM,eAAe,GACpB,OAAO,EAAE,KAAK,KAAK,SAAS,IAAI,OAAO,EAAE,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAEhG,QAAQ,CAAC;QACR,KAAK;QACL,UAAU;QACV,UAAU,EAAE,oBAAoB;QAChC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC;QAC3C,kBAAkB;QAClB,eAAe,EAAE,OAAO,EAAE,eAAe,IAAI,IAAI;QACjD,eAAe,EAAE,OAAO,EAAE,eAAe,IAAI,KAAK;QAClD,IAAI,EAAE,OAAO,EAAE,UAAU,IAAI,IAAI;QACjC,OAAO,EAAE,OAAO,EAAE,aAAa,IAAI,IAAI;QACvC,YAAY,EAAE,OAAO,EAAE,YAAY;QACnC,aAAa,EAAE,OAAO,EAAE,aAAa;QACrC,iBAAiB,EAAE,OAAO,EAAE,iBAAiB;QAC7C,UAAU,EAAE,oBAAoB;QAChC,OAAO,EAAE,OAAO,EAAE,OAAO,IAAI,IAAI;QACjC,YAAY,EAAE,eAAe;KAC7B,CAAC,CAAC;AACJ,CAAC;AAKD,MAAM,UAAU,cAAc,CAC7B,UAAkB,EAClB,KAAa,EACb,IAA6B,EAC7B,OAA4B;IAE5B,aAAa,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AACjD,CAAC;AAKD,MAAM,UAAU,cAAc,CAC7B,UAAkB,EAClB,KAAa,EACb,IAA6B,EAC7B,OAA4B;IAE5B,aAAa,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;AACjD,CAAC;AAKD,MAAM,UAAU,cAAc,CAC7B,UAAkB,EAClB,SAAiB,EACjB,OAQC;IAED,IAAI,CAAC,YAAY,EAAE,CAAC;QACnB,OAAO;IACR,CAAC;IAED,MAAM,kBAAkB,GAAG,qBAAqB,EAAE,CAAC;IAGnD,IAAI,gBAAgB,GAAG,IAAI,CAAC;IAC5B,IAAI,mBAAmB,GAAG,IAAI,CAAC;IAC/B,IAAI,OAAO,EAAE,YAAY,IAAI,OAAO,OAAO,CAAC,YAAY,KAAK,QAAQ,IAAI,OAAO,CAAC,YAAY,KAAK,IAAI,EAAE,CAAC;QACxG,MAAM,IAAI,GAAG,OAAO,CAAC,YAAuC,CAAC;QAC7D,IAAI,IAAI,CAAC,UAAU,IAAI,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI,EAAE,CAAC;YACxF,MAAM,UAAU,GAAG,IAAI,CAAC,UAAqC,CAAC;YAC9D,gBAAgB,GAAG,OAAO,UAAU,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;YAChF,mBAAmB,GAAG,OAAO,UAAU,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;QAC1F,CAAC;IACF,CAAC;IAED,YAAY,CAAC,IAAI,CAChB;QAEC,SAAS;QACT,UAAU;QAKV,IAAI,EAAE,OAAO,EAAE,UAAU,IAAI,gBAAgB,IAAI,IAAI;QACrD,OAAO,EAAE,OAAO,EAAE,aAAa,IAAI,mBAAmB,IAAI,IAAI;QAC9D,UAAU,EAAE,OAAO,EAAE,eAAe,IAAI,KAAK;QAG7C,SAAS,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI;QAGrC,YAAY,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI;QACjF,eAAe,EAAE,OAAO,EAAE,eAAe,IAAI,IAAI;QACjD,WAAW,EAAE,OAAO,EAAE,WAAW;YAChC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,WAAW,CAAC;YACrC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;QAC5C,kBAAkB;KAClB,EACD,qBAAqB,CACrB,CAAC;AACH,CAAC;AAKD,MAAM,UAAU,cAAc,CAC7B,YAAoB,EACpB,SAAiB,EACjB,OAUC;IAED,IAAI,CAAC,YAAY,EAAE,CAAC;QACnB,OAAO;IACR,CAAC;IAED,MAAM,kBAAkB,GAAG,qBAAqB,EAAE,CAAC;IAGnD,IAAI,WAAW,GAAkB,IAAI,CAAC;IACtC,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,IAAI,OAAO,CAAC,KAAK,KAAK,IAAI,EAAE,CAAC;QAC3D,IAAI,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;YACvC,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC;QAC7B,CAAC;aAAM,IAAI,OAAO,CAAC,KAAK,YAAY,KAAK,EAAE,CAAC;YAC3C,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC;QACrC,CAAC;aAAM,CAAC;YACP,IAAI,CAAC;gBACJ,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC7C,CAAC;YAAC,MAAM,CAAC;gBACR,WAAW,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACrC,CAAC;QACF,CAAC;IACF,CAAC;IAED,YAAY,CAAC,IAAI,CAChB;QACC,SAAS;QACT,YAAY;QACZ,IAAI,EAAE,OAAO,CAAC,UAAU,IAAI,IAAI;QAChC,OAAO,EAAE,OAAO,CAAC,aAAa,IAAI,IAAI;QACtC,UAAU,EAAE,OAAO,CAAC,eAAe,IAAI,KAAK;QAC5C,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,KAAK,EAAE,WAAW;QAClB,iBAAiB,EAAE,OAAO,CAAC,iBAAiB,IAAI,IAAI;QACpD,iBAAiB,EAAE,OAAO,CAAC,iBAAiB,IAAI,CAAC;QACjD,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,KAAK;QACrC,kBAAkB;KAClB,EACD,qBAAqB,CACrB,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACrC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC5B,OAAO,GAAG,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC;IAC1C,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,KAAK,CAAC;IACd,CAAC;IACD,IAAI,CAAC;QACJ,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC;AACF,CAAC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
interface RewriteResult {
|
|
2
|
+
rewrittenBody: unknown;
|
|
3
|
+
legacyToolName?: string;
|
|
4
|
+
rewrittenToolName?: string;
|
|
5
|
+
}
|
|
6
|
+
export declare function rewriteLegacySearchToolCallRequest(requestBody: unknown): RewriteResult;
|
|
7
|
+
export {};
|
|
8
|
+
//# sourceMappingURL=repo-search-shim.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"repo-search-shim.d.ts","sourceRoot":"","sources":["../../../src/server/utils/repo-search-shim.ts"],"names":[],"mappings":"AAOA,UAAU,aAAa;IACtB,aAAa,EAAE,OAAO,CAAC;IACvB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAsED,wBAAgB,kCAAkC,CAAC,WAAW,EAAE,OAAO,GAAG,aAAa,CAsEtF"}
|
|
@@ -0,0 +1,116 @@
|
|
|
1
|
+
import { REPO_SEARCH_TOOL_ID } from '@llmindset/hf-mcp';
|
|
2
|
+
import { isLegacyDatasetSearchTool, isLegacyModelSearchTool, isLegacyRepoSearchTool, } from '../../shared/repo-search-migration.js';
|
|
3
|
+
function isRecord(value) {
|
|
4
|
+
return typeof value === 'object' && value !== null;
|
|
5
|
+
}
|
|
6
|
+
function toStringArray(value) {
|
|
7
|
+
if (!Array.isArray(value)) {
|
|
8
|
+
return [];
|
|
9
|
+
}
|
|
10
|
+
return value.filter((item) => typeof item === 'string');
|
|
11
|
+
}
|
|
12
|
+
function uniq(values) {
|
|
13
|
+
return [...new Set(values)];
|
|
14
|
+
}
|
|
15
|
+
function rewriteModelSearchArguments(argumentsInput) {
|
|
16
|
+
if (!isRecord(argumentsInput)) {
|
|
17
|
+
return { repo_types: ['model'] };
|
|
18
|
+
}
|
|
19
|
+
const nextArgs = { ...argumentsInput, repo_types: ['model'] };
|
|
20
|
+
const existingFilters = toStringArray(nextArgs.filters);
|
|
21
|
+
const task = typeof nextArgs.task === 'string' ? nextArgs.task : undefined;
|
|
22
|
+
const library = typeof nextArgs.library === 'string' ? nextArgs.library : undefined;
|
|
23
|
+
const mergedFilters = uniq([
|
|
24
|
+
...existingFilters,
|
|
25
|
+
...(task ? [task] : []),
|
|
26
|
+
...(library ? [library] : []),
|
|
27
|
+
]);
|
|
28
|
+
delete nextArgs.task;
|
|
29
|
+
delete nextArgs.library;
|
|
30
|
+
if (mergedFilters.length > 0) {
|
|
31
|
+
nextArgs.filters = mergedFilters;
|
|
32
|
+
}
|
|
33
|
+
else {
|
|
34
|
+
delete nextArgs.filters;
|
|
35
|
+
}
|
|
36
|
+
return nextArgs;
|
|
37
|
+
}
|
|
38
|
+
function rewriteDatasetSearchArguments(argumentsInput) {
|
|
39
|
+
if (!isRecord(argumentsInput)) {
|
|
40
|
+
return { repo_types: ['dataset'] };
|
|
41
|
+
}
|
|
42
|
+
const nextArgs = { ...argumentsInput, repo_types: ['dataset'] };
|
|
43
|
+
const existingFilters = toStringArray(nextArgs.filters);
|
|
44
|
+
const tags = toStringArray(nextArgs.tags);
|
|
45
|
+
const mergedFilters = uniq([...existingFilters, ...tags]);
|
|
46
|
+
delete nextArgs.tags;
|
|
47
|
+
if (mergedFilters.length > 0) {
|
|
48
|
+
nextArgs.filters = mergedFilters;
|
|
49
|
+
}
|
|
50
|
+
else {
|
|
51
|
+
delete nextArgs.filters;
|
|
52
|
+
}
|
|
53
|
+
return nextArgs;
|
|
54
|
+
}
|
|
55
|
+
export function rewriteLegacySearchToolCallRequest(requestBody) {
|
|
56
|
+
if (!isRecord(requestBody)) {
|
|
57
|
+
return { rewrittenBody: requestBody };
|
|
58
|
+
}
|
|
59
|
+
if (requestBody.method !== 'tools/call') {
|
|
60
|
+
return { rewrittenBody: requestBody };
|
|
61
|
+
}
|
|
62
|
+
const params = requestBody.params;
|
|
63
|
+
if (!isRecord(params)) {
|
|
64
|
+
return { rewrittenBody: requestBody };
|
|
65
|
+
}
|
|
66
|
+
const toolName = params.name;
|
|
67
|
+
if (typeof toolName !== 'string') {
|
|
68
|
+
return { rewrittenBody: requestBody };
|
|
69
|
+
}
|
|
70
|
+
if (isLegacyModelSearchTool(toolName)) {
|
|
71
|
+
const rewrittenBody = {
|
|
72
|
+
...requestBody,
|
|
73
|
+
params: {
|
|
74
|
+
...params,
|
|
75
|
+
name: REPO_SEARCH_TOOL_ID,
|
|
76
|
+
arguments: rewriteModelSearchArguments(params.arguments),
|
|
77
|
+
},
|
|
78
|
+
};
|
|
79
|
+
return {
|
|
80
|
+
rewrittenBody,
|
|
81
|
+
legacyToolName: toolName,
|
|
82
|
+
rewrittenToolName: REPO_SEARCH_TOOL_ID,
|
|
83
|
+
};
|
|
84
|
+
}
|
|
85
|
+
if (isLegacyDatasetSearchTool(toolName)) {
|
|
86
|
+
const rewrittenBody = {
|
|
87
|
+
...requestBody,
|
|
88
|
+
params: {
|
|
89
|
+
...params,
|
|
90
|
+
name: REPO_SEARCH_TOOL_ID,
|
|
91
|
+
arguments: rewriteDatasetSearchArguments(params.arguments),
|
|
92
|
+
},
|
|
93
|
+
};
|
|
94
|
+
return {
|
|
95
|
+
rewrittenBody,
|
|
96
|
+
legacyToolName: toolName,
|
|
97
|
+
rewrittenToolName: REPO_SEARCH_TOOL_ID,
|
|
98
|
+
};
|
|
99
|
+
}
|
|
100
|
+
if (isLegacyRepoSearchTool(toolName)) {
|
|
101
|
+
const rewrittenBody = {
|
|
102
|
+
...requestBody,
|
|
103
|
+
params: {
|
|
104
|
+
...params,
|
|
105
|
+
name: REPO_SEARCH_TOOL_ID,
|
|
106
|
+
},
|
|
107
|
+
};
|
|
108
|
+
return {
|
|
109
|
+
rewrittenBody,
|
|
110
|
+
legacyToolName: toolName,
|
|
111
|
+
rewrittenToolName: REPO_SEARCH_TOOL_ID,
|
|
112
|
+
};
|
|
113
|
+
}
|
|
114
|
+
return { rewrittenBody: requestBody };
|
|
115
|
+
}
|
|
116
|
+
//# sourceMappingURL=repo-search-shim.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"repo-search-shim.js","sourceRoot":"","sources":["../../../src/server/utils/repo-search-shim.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EACN,yBAAyB,EACzB,uBAAuB,EACvB,sBAAsB,GACtB,MAAM,uCAAuC,CAAC;AAQ/C,SAAS,QAAQ,CAAC,KAAc;IAC/B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACpD,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACpC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,EAAE,CAAC;IACX,CAAC;IAED,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC;AACzE,CAAC;AAED,SAAS,IAAI,CAAC,MAAgB;IAC7B,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;AAC7B,CAAC;AAED,SAAS,2BAA2B,CAAC,cAAuB;IAC3D,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;QAC/B,OAAO,EAAE,UAAU,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;IAClC,CAAC;IAED,MAAM,QAAQ,GAA4B,EAAE,GAAG,cAAc,EAAE,UAAU,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;IACvF,MAAM,eAAe,GAAG,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IACxD,MAAM,IAAI,GAAG,OAAO,QAAQ,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3E,MAAM,OAAO,GAAG,OAAO,QAAQ,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;IACpF,MAAM,aAAa,GAAG,IAAI,CAAC;QAC1B,GAAG,eAAe;QAClB,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACvB,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;KAC7B,CAAC,CAAC;IAEH,OAAO,QAAQ,CAAC,IAAI,CAAC;IACrB,OAAO,QAAQ,CAAC,OAAO,CAAC;IAExB,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9B,QAAQ,CAAC,OAAO,GAAG,aAAa,CAAC;IAClC,CAAC;SAAM,CAAC;QACP,OAAO,QAAQ,CAAC,OAAO,CAAC;IACzB,CAAC;IAED,OAAO,QAAQ,CAAC;AACjB,CAAC;AAED,SAAS,6BAA6B,CAAC,cAAuB;IAC7D,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;QAC/B,OAAO,EAAE,UAAU,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC;IACpC,CAAC;IAED,MAAM,QAAQ,GAA4B,EAAE,GAAG,cAAc,EAAE,UAAU,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC;IACzF,MAAM,eAAe,GAAG,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IACxD,MAAM,IAAI,GAAG,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC1C,MAAM,aAAa,GAAG,IAAI,CAAC,CAAC,GAAG,eAAe,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;IAE1D,OAAO,QAAQ,CAAC,IAAI,CAAC;IAErB,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9B,QAAQ,CAAC,OAAO,GAAG,aAAa,CAAC;IAClC,CAAC;SAAM,CAAC;QACP,OAAO,QAAQ,CAAC,OAAO,CAAC;IACzB,CAAC;IAED,OAAO,QAAQ,CAAC;AACjB,CAAC;AAMD,MAAM,UAAU,kCAAkC,CAAC,WAAoB;IACtE,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;QAC5B,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,CAAC;IACvC,CAAC;IAED,IAAI,WAAW,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;QACzC,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,CAAC;IACvC,CAAC;IAED,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC;IAClC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QACvB,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,CAAC;IACvC,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC;IAC7B,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAClC,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,CAAC;IACvC,CAAC;IAED,IAAI,uBAAuB,CAAC,QAAQ,CAAC,EAAE,CAAC;QACvC,MAAM,aAAa,GAA4B;YAC9C,GAAG,WAAW;YACd,MAAM,EAAE;gBACP,GAAG,MAAM;gBACT,IAAI,EAAE,mBAAmB;gBACzB,SAAS,EAAE,2BAA2B,CAAC,MAAM,CAAC,SAAS,CAAC;aACxD;SACD,CAAC;QAEF,OAAO;YACN,aAAa;YACb,cAAc,EAAE,QAAQ;YACxB,iBAAiB,EAAE,mBAAmB;SACtC,CAAC;IACH,CAAC;IAED,IAAI,yBAAyB,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzC,MAAM,aAAa,GAA4B;YAC9C,GAAG,WAAW;YACd,MAAM,EAAE;gBACP,GAAG,MAAM;gBACT,IAAI,EAAE,mBAAmB;gBACzB,SAAS,EAAE,6BAA6B,CAAC,MAAM,CAAC,SAAS,CAAC;aAC1D;SACD,CAAC;QAEF,OAAO;YACN,aAAa;YACb,cAAc,EAAE,QAAQ;YACxB,iBAAiB,EAAE,mBAAmB;SACtC,CAAC;IACH,CAAC;IAED,IAAI,sBAAsB,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtC,MAAM,aAAa,GAA4B;YAC9C,GAAG,WAAW;YACd,MAAM,EAAE;gBACP,GAAG,MAAM;gBACT,IAAI,EAAE,mBAAmB;aACzB;SACD,CAAC;QAEF,OAAO;YACN,aAAa;YACb,cAAc,EAAE,QAAQ;YACxB,iBAAiB,EAAE,mBAAmB;SACtC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,CAAC;AACvC,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"streamable-http-tool-caller.d.ts","sourceRoot":"","sources":["../../../src/server/utils/streamable-http-tool-caller.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AAEzE,OAAO,KAAK,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,oCAAoC,CAAC;AAC5F,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAC;
|
|
1
|
+
{"version":3,"file":"streamable-http-tool-caller.d.ts","sourceRoot":"","sources":["../../../src/server/utils/streamable-http-tool-caller.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AAEzE,OAAO,KAAK,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,oCAAoC,CAAC;AAC5F,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAC;AAuBxF,wBAAsB,sBAAsB,CAC3C,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnC,OAAO,EAAE,MAAM,GAAG,SAAS,EAC3B,KAAK,EAAE,mBAAmB,CAAC,aAAa,EAAE,kBAAkB,CAAC,GAAG,SAAS,GACvE,OAAO,CAAC,cAAc,CAAC,CAsIzB"}
|
|
@@ -1,7 +1,9 @@
|
|
|
1
1
|
import { CallToolResultSchema } from '@modelcontextprotocol/sdk/types.js';
|
|
2
2
|
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
|
|
3
3
|
import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
|
|
4
|
+
import { fetchWithProfile, NETWORK_FETCH_PROFILES, parseAndValidateUrl } from '@llmindset/hf-mcp/network';
|
|
4
5
|
import { logger } from './logger.js';
|
|
6
|
+
const PROXY_STREAMABLE_PROFILE = NETWORK_FETCH_PROFILES.streamableProxy();
|
|
5
7
|
function buildAuthHeaders(hfToken) {
|
|
6
8
|
if (!hfToken) {
|
|
7
9
|
return undefined;
|
|
@@ -19,6 +21,7 @@ export async function callStreamableHttpTool(serverUrl, toolName, parameters, hf
|
|
|
19
21
|
paramKeys: Object.keys(parameters ?? {}),
|
|
20
22
|
}, 'Streamable proxy calling upstream tool');
|
|
21
23
|
logger.info({ serverUrl, toolName, params: parameters }, 'Calling Streamable HTTP tool');
|
|
24
|
+
const validatedServerUrl = parseAndValidateUrl(serverUrl, PROXY_STREAMABLE_PROFILE.urlPolicy);
|
|
22
25
|
const client = new Client({
|
|
23
26
|
name: 'hf-mcp-streamable-client',
|
|
24
27
|
version: '1.0.0',
|
|
@@ -26,8 +29,14 @@ export async function callStreamableHttpTool(serverUrl, toolName, parameters, hf
|
|
|
26
29
|
capabilities: {},
|
|
27
30
|
});
|
|
28
31
|
const headers = buildAuthHeaders(hfToken);
|
|
29
|
-
const transport = new StreamableHTTPClientTransport(
|
|
32
|
+
const transport = new StreamableHTTPClientTransport(validatedServerUrl, {
|
|
30
33
|
requestInit: headers ? { headers } : undefined,
|
|
34
|
+
fetch: async (url, init) => {
|
|
35
|
+
const { response } = await fetchWithProfile(url.toString(), PROXY_STREAMABLE_PROFILE, {
|
|
36
|
+
requestInit: init,
|
|
37
|
+
});
|
|
38
|
+
return response;
|
|
39
|
+
},
|
|
31
40
|
});
|
|
32
41
|
await client.connect(transport);
|
|
33
42
|
logger.trace({ serverUrl }, 'Streamable proxy connected upstream');
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"streamable-http-tool-caller.js","sourceRoot":"","sources":["../../../src/server/utils/streamable-http-tool-caller.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,MAAM,oCAAoC,CAAC;AAG1E,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AACnG,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErC,SAAS,gBAAgB,CAAC,OAAgB;IACzC,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,OAAO;QACN,aAAa,EAAE,UAAU,OAAO,EAAE;QAClC,oBAAoB,EAAE,UAAU,OAAO,EAAE;KACzC,CAAC;AACH,CAAC;AAMD,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC3C,SAAiB,EACjB,QAAgB,EAChB,UAAmC,EACnC,OAA2B,EAC3B,KAAyE;IAEzE,MAAM,CAAC,KAAK,CACX;QACC,SAAS;QACT,QAAQ;QACR,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC;QAC1B,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;KACxC,EACD,wCAAwC,CACxC,CAAC;IACF,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,8BAA8B,CAAC,CAAC;IAEzF,MAAM,MAAM,GAAG,IAAI,MAAM,CACxB;QACC,IAAI,EAAE,0BAA0B;QAChC,OAAO,EAAE,OAAO;KAChB,EACD;QACC,YAAY,EAAE,EAAE;KAChB,CACD,CAAC;IAEF,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAC1C,MAAM,SAAS,GAAG,IAAI,6BAA6B,CAAC,
|
|
1
|
+
{"version":3,"file":"streamable-http-tool-caller.js","sourceRoot":"","sources":["../../../src/server/utils/streamable-http-tool-caller.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,MAAM,oCAAoC,CAAC;AAG1E,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AACnG,OAAO,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAC1G,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErC,MAAM,wBAAwB,GAAG,sBAAsB,CAAC,eAAe,EAAE,CAAC;AAE1E,SAAS,gBAAgB,CAAC,OAAgB;IACzC,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,OAAO;QACN,aAAa,EAAE,UAAU,OAAO,EAAE;QAClC,oBAAoB,EAAE,UAAU,OAAO,EAAE;KACzC,CAAC;AACH,CAAC;AAMD,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC3C,SAAiB,EACjB,QAAgB,EAChB,UAAmC,EACnC,OAA2B,EAC3B,KAAyE;IAEzE,MAAM,CAAC,KAAK,CACX;QACC,SAAS;QACT,QAAQ;QACR,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC;QAC1B,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;KACxC,EACD,wCAAwC,CACxC,CAAC;IACF,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,8BAA8B,CAAC,CAAC;IAEzF,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,SAAS,EAAE,wBAAwB,CAAC,SAAS,CAAC,CAAC;IAE9F,MAAM,MAAM,GAAG,IAAI,MAAM,CACxB;QACC,IAAI,EAAE,0BAA0B;QAChC,OAAO,EAAE,OAAO;KAChB,EACD;QACC,YAAY,EAAE,EAAE;KAChB,CACD,CAAC;IAEF,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAC1C,MAAM,SAAS,GAAG,IAAI,6BAA6B,CAAC,kBAAkB,EAAE;QACvE,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS;QAC9C,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAC1B,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,gBAAgB,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,wBAAwB,EAAE;gBACrF,WAAW,EAAE,IAAI;aACjB,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC;QACjB,CAAC;KACD,CAAC,CAAC;IAEH,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAChC,MAAM,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,EAAE,qCAAqC,CAAC,CAAC;IAEnE,IAAI,CAAC;QACJ,MAAM,aAAa,GAAG,KAAK,EAAE,KAAK,EAAE,aAAa,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,EAAE,aAAa,EAAE,aAAa,IAAI,IAAI,EAAE,EAAE,6CAA6C,CAAC,CAAC;QACtG,IAAI,qBAAqB,GAAG,KAAK,CAAC;QAElC,MAAM,wBAAwB,GAAG,KAAK,EAAE,QAAiE,EAAE,EAAE;YAC5G,IAAI,CAAC,KAAK,IAAI,qBAAqB,EAAE,CAAC;gBACrC,OAAO;YACR,CAAC;YACD,IAAI,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;gBAC3B,qBAAqB,GAAG,IAAI,CAAC;gBAC7B,OAAO;YACR,CAAC;YACD,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;gBACjC,OAAO;YACR,CAAC;YACD,MAAM,CAAC,KAAK,CACX;gBACC,aAAa;gBACb,QAAQ;aACR,EACD,0CAA0C,CAC1C,CAAC;YAEF,IAAI,CAAC;gBACJ,MAAM,MAAM,GAKR;oBACH,aAAa;oBACb,QAAQ,EAAE,QAAQ,CAAC,QAAQ,IAAI,CAAC;iBAChC,CAAC;gBACF,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;oBAClC,MAAM,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;gBAC/B,CAAC;gBACD,IAAI,QAAQ,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;oBACpC,MAAM,CAAC,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;gBACnC,CAAC;gBACD,MAAM,KAAK,CAAC,gBAAgB,CAAC;oBAC5B,MAAM,EAAE,wBAAwB;oBAChC,MAAM;iBACN,CAAC,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,qBAAqB,GAAG,IAAI,CAAC;gBAC7B,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,wCAAwC,CAAC,CAAC;gBAClE,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,uDAAuD,CAAC,CAAC;YAClF,CAAC;QACF,CAAC,CAAC;QAEF,MAAM,cAAc,GAGhB,EAAE,CAAC;QAEP,IAAI,aAAa,KAAK,SAAS,IAAI,KAAK,EAAE,CAAC;YAC1C,cAAc,CAAC,UAAU,GAAG,CAAC,QAAQ,EAAE,EAAE;gBACxC,KAAK,wBAAwB,CAAC,QAAQ,CAAC,CAAC;YACzC,CAAC,CAAC;YACF,cAAc,CAAC,sBAAsB,GAAG,IAAI,CAAC;QAC9C,CAAC;aAAM,CAAC;YACP,MAAM,CAAC,KAAK,CACX;gBACC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC;gBACxB,aAAa,EAAE,aAAa,IAAI,IAAI;aACpC,EACD,0CAA0C,CAC1C,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAClC;YACC,MAAM,EAAE,YAAY;YACpB,MAAM,EAAE;gBACP,IAAI,EAAE,QAAQ;gBACd,SAAS,EAAE,UAAU;gBACrB,KAAK,EAAE,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,SAAS;aAClE;SACD,EACD,oBAAoB,EACpB,cAAc,CACd,CAAC;QAEF,MAAM,CAAC,KAAK,CACX;YACC,SAAS;YACT,QAAQ;SACR,EACD,gDAAgD,CAChD,CAAC;QAEF,OAAO,MAAM,CAAC;IACf,CAAC;YAAS,CAAC;QACV,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;AACF,CAAC"}
|
|
@@ -12,7 +12,7 @@ export interface ToolSelectionContext {
|
|
|
12
12
|
userSettings?: AppSettings;
|
|
13
13
|
hfToken?: string;
|
|
14
14
|
}
|
|
15
|
-
|
|
15
|
+
interface ToolSelectionResult {
|
|
16
16
|
mode: ToolSelectionMode;
|
|
17
17
|
enabledToolIds: string[];
|
|
18
18
|
reason: string;
|
|
@@ -30,4 +30,5 @@ export declare class ToolSelectionStrategy {
|
|
|
30
30
|
selectTools(context: ToolSelectionContext): Promise<ToolSelectionResult>;
|
|
31
31
|
private getUserSettings;
|
|
32
32
|
}
|
|
33
|
+
export {};
|
|
33
34
|
//# sourceMappingURL=tool-selection-strategy.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tool-selection-strategy.d.ts","sourceRoot":"","sources":["../../../src/server/utils/tool-selection-strategy.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAEvE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAOxD,oBAAY,iBAAiB;IAC5B,gBAAgB,qBAAqB;IACrC,GAAG,QAAQ;IACX,YAAY,iBAAiB;IAC7B,YAAY,iBAAiB;IAC7B,QAAQ,aAAa;CACrB;AAED,MAAM,WAAW,oBAAoB;IACpC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC;IACvC,YAAY,CAAC,EAAE,WAAW,CAAC;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,
|
|
1
|
+
{"version":3,"file":"tool-selection-strategy.d.ts","sourceRoot":"","sources":["../../../src/server/utils/tool-selection-strategy.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAEvE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAOxD,oBAAY,iBAAiB;IAC5B,gBAAgB,qBAAqB;IACrC,GAAG,QAAQ;IACX,YAAY,iBAAiB;IAC7B,YAAY,iBAAiB;IAC7B,QAAQ,aAAa;CACrB;AAED,MAAM,WAAW,oBAAoB;IACpC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC;IACvC,YAAY,CAAC,EAAE,WAAW,CAAC;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,UAAU,mBAAmB;IAC5B,IAAI,EAAE,iBAAiB,CAAC;IACxB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,WAAW,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,gBAAgB,CAAC,EAAE,SAAS,EAAE,CAAC;CAC/B;AAuCD,qBAAa,qBAAqB;IACjC,OAAO,CAAC,SAAS,CAAe;gBAEpB,SAAS,EAAE,YAAY;IAYnC,OAAO,CAAC,oBAAoB;IAuB5B,OAAO,CAAC,uBAAuB;IAU/B,OAAO,CAAC,iBAAiB;IAIzB,OAAO,CAAC,gBAAgB;IAmBlB,WAAW,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,mBAAmB,CAAC;YAgIhE,eAAe;CAiC7B"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { type UIResource } from '@mcp-ui/server';
|
|
2
|
-
|
|
2
|
+
interface AudioPlayerOptions {
|
|
3
3
|
title: string;
|
|
4
4
|
subtitle?: string;
|
|
5
5
|
base64Audio?: string;
|
|
@@ -11,6 +11,6 @@ export interface AudioPlayerOptions {
|
|
|
11
11
|
size?: string;
|
|
12
12
|
};
|
|
13
13
|
}
|
|
14
|
-
export declare function buildAudioPlayerHTML({ title, subtitle, base64Audio, srcUrl, mimeType, metadata, }: AudioPlayerOptions): string;
|
|
15
14
|
export declare function createAudioPlayerUIResource(uri: `ui://${string}`, options: AudioPlayerOptions): UIResource;
|
|
15
|
+
export {};
|
|
16
16
|
//# sourceMappingURL=audio-player.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"audio-player.d.ts","sourceRoot":"","sources":["../../../../src/server/utils/ui/audio-player.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoB,KAAK,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAEnE,
|
|
1
|
+
{"version":3,"file":"audio-player.d.ts","sourceRoot":"","sources":["../../../../src/server/utils/ui/audio-player.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoB,KAAK,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAEnE,UAAU,kBAAkB;IAE3B,KAAK,EAAE,MAAM,CAAC;IAEd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,QAAQ,CAAC,EAAE;QACV,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,IAAI,CAAC,EAAE,MAAM,CAAC;KACd,CAAC;CACF;AAgJD,wBAAgB,2BAA2B,CAAC,GAAG,EAAE,QAAQ,MAAM,EAAE,EAAE,OAAO,EAAE,kBAAkB,GAAG,UAAU,CAM1G"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { createUIResource } from '@mcp-ui/server';
|
|
2
|
-
|
|
2
|
+
function buildAudioPlayerHTML({ title, subtitle, base64Audio, srcUrl, mimeType = 'audio/wav', metadata, }) {
|
|
3
3
|
const safeTitle = escapeHtml(title);
|
|
4
4
|
const audioSrc = base64Audio ? `data:${mimeType};base64,${base64Audio}` : srcUrl ? escapeHtml(srcUrl) : '';
|
|
5
5
|
return `
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"audio-player.js","sourceRoot":"","sources":["../../../../src/server/utils/ui/audio-player.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAmB,MAAM,gBAAgB,CAAC;AA0BnE,
|
|
1
|
+
{"version":3,"file":"audio-player.js","sourceRoot":"","sources":["../../../../src/server/utils/ui/audio-player.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAmB,MAAM,gBAAgB,CAAC;AA0BnE,SAAS,oBAAoB,CAAC,EAC7B,KAAK,EACL,QAAQ,EACR,WAAW,EACX,MAAM,EACN,QAAQ,GAAG,WAAW,EACtB,QAAQ,GACY;IACpB,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;IACpC,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,QAAQ,QAAQ,WAAW,WAAW,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAI3G,OAAO;;;;WAIG,SAAS;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gCA6FY,SAAS,KAAK,SAAS;;OAEhD,SAAS,EAAE;;;KAGb,QAAQ,CAAC,CAAC,CAAC,yBAAyB,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE;;;0DAGhB,QAAQ;;MAG7D,QAAQ;QACP,CAAC,CAAC;OACD,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,+EAA+E,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE;OAChJ,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,iFAAiF,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE;OACtJ,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,6EAA6E,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE;WACtI;QACL,CAAC,CAAC,EACJ;;;SAGK,CAAC;AACV,CAAC;AAKD,MAAM,UAAU,2BAA2B,CAAC,GAAqB,EAAE,OAA2B;IAC7F,OAAO,gBAAgB,CAAC;QACvB,GAAG;QACH,QAAQ,EAAE,MAAM;QAChB,OAAO,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,UAAU,EAAE,oBAAoB,CAAC,OAAO,CAAC,EAAE;KACvE,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,KAAa;IAChC,OAAO,KAAK;SACV,UAAU,CAAC,GAAG,EAAE,OAAO,CAAC;SACxB,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC;SACvB,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC;SACvB,UAAU,CAAC,GAAG,EAAE,QAAQ,CAAC;SACzB,UAAU,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;AAC5B,CAAC;AAID,SAAS,SAAS;IACjB,OAAO;;;;;;;;;CASP,CAAC;AACF,CAAC"}
|
|
@@ -1,12 +1,12 @@
|
|
|
1
1
|
import type { AppSettings } from './settings.js';
|
|
2
2
|
export declare const BOUQUETS: Record<string, AppSettings>;
|
|
3
|
-
|
|
4
|
-
|
|
3
|
+
type BouquetKey = keyof typeof BOUQUETS;
|
|
4
|
+
interface DirectParamOption {
|
|
5
5
|
label: string;
|
|
6
6
|
param: string;
|
|
7
7
|
description?: string;
|
|
8
8
|
}
|
|
9
|
-
|
|
9
|
+
interface BouquetPreset {
|
|
10
10
|
key: BouquetKey;
|
|
11
11
|
label: string;
|
|
12
12
|
description: string;
|
|
@@ -17,12 +17,13 @@ export interface BouquetPreset {
|
|
|
17
17
|
directParams?: DirectParamOption[];
|
|
18
18
|
}
|
|
19
19
|
export declare const BOUQUET_PRESETS: BouquetPreset[];
|
|
20
|
-
|
|
21
|
-
|
|
20
|
+
type ConfigEntryKind = 'tool' | 'behavior-flag';
|
|
21
|
+
interface ConfigEntryDescription {
|
|
22
22
|
id: string;
|
|
23
23
|
label: string;
|
|
24
24
|
description?: string;
|
|
25
25
|
kind: ConfigEntryKind;
|
|
26
26
|
}
|
|
27
27
|
export declare function describeConfigEntry(id: string): ConfigEntryDescription;
|
|
28
|
+
export {};
|
|
28
29
|
//# sourceMappingURL=bouquet-presets.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"bouquet-presets.d.ts","sourceRoot":"","sources":["../../src/shared/bouquet-presets.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"bouquet-presets.d.ts","sourceRoot":"","sources":["../../src/shared/bouquet-presets.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAGjD,eAAO,MAAM,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CA4DhD,CAAC;AAEF,KAAK,UAAU,GAAG,MAAM,OAAO,QAAQ,CAAC;AAExC,UAAU,iBAAiB;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,UAAU,aAAa;IACtB,GAAG,EAAE,UAAU,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,GAAG,UAAU,CAAC;IAC9B,eAAe,EAAE,OAAO,CAAC;IACzB,WAAW,EAAE,OAAO,CAAC;IACrB,YAAY,EAAE,SAAS,MAAM,EAAE,CAAC;IAChC,YAAY,CAAC,EAAE,iBAAiB,EAAE,CAAC;CACnC;AA8GD,eAAO,MAAM,eAAe,EAAE,aAAa,EAOzC,CAAC;AAEH,KAAK,eAAe,GAAG,MAAM,GAAG,eAAe,CAAC;AAEhD,UAAU,sBAAsB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,EAAE,eAAe,CAAC;CACtB;AAoED,wBAAgB,mBAAmB,CAAC,EAAE,EAAE,MAAM,GAAG,sBAAsB,CAsBtE"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { ALL_BUILTIN_TOOL_IDS, TOOL_ID_GROUPS,
|
|
1
|
+
import { ALL_BUILTIN_TOOL_IDS, TOOL_ID_GROUPS, HUB_REPO_DETAILS_TOOL_ID, USE_SPACE_TOOL_ID, HF_JOBS_TOOL_ID, DYNAMIC_SPACE_TOOL_ID, REPO_SEARCH_TOOL_ID, DOCS_SEMANTIC_SEARCH_TOOL_ID, } from '@llmindset/hf-mcp';
|
|
2
2
|
import { README_INCLUDE_FLAG, GRADIO_IMAGE_FILTER_FLAG } from './behavior-flags.js';
|
|
3
3
|
export const BOUQUETS = {
|
|
4
4
|
hf_api: {
|
|
@@ -19,10 +19,9 @@ export const BOUQUETS = {
|
|
|
19
19
|
},
|
|
20
20
|
skills: {
|
|
21
21
|
builtInTools: [
|
|
22
|
-
|
|
22
|
+
HUB_REPO_DETAILS_TOOL_ID,
|
|
23
23
|
README_INCLUDE_FLAG,
|
|
24
|
-
|
|
25
|
-
DATASET_SEARCH_TOOL_ID,
|
|
24
|
+
REPO_SEARCH_TOOL_ID,
|
|
26
25
|
DOCS_SEMANTIC_SEARCH_TOOL_ID,
|
|
27
26
|
HF_JOBS_TOOL_ID,
|
|
28
27
|
],
|
|
@@ -33,11 +32,11 @@ export const BOUQUETS = {
|
|
|
33
32
|
spaceTools: [],
|
|
34
33
|
},
|
|
35
34
|
hub_repo_details_readme: {
|
|
36
|
-
builtInTools: [
|
|
35
|
+
builtInTools: [HUB_REPO_DETAILS_TOOL_ID, README_INCLUDE_FLAG],
|
|
37
36
|
spaceTools: [],
|
|
38
37
|
},
|
|
39
38
|
hub_repo_details: {
|
|
40
|
-
builtInTools: [
|
|
39
|
+
builtInTools: [HUB_REPO_DETAILS_TOOL_ID],
|
|
41
40
|
spaceTools: [],
|
|
42
41
|
},
|
|
43
42
|
no_gradio_images: {
|
|
@@ -81,7 +80,7 @@ const PRESET_META = [
|
|
|
81
80
|
{
|
|
82
81
|
key: 'search',
|
|
83
82
|
label: 'Search Tools',
|
|
84
|
-
description: '
|
|
83
|
+
description: 'Search across models, datasets, spaces, papers, and docs.',
|
|
85
84
|
category: 'core',
|
|
86
85
|
supportsBouquet: true,
|
|
87
86
|
supportsMix: true,
|
|
@@ -155,9 +154,9 @@ const TOOL_DESCRIPTIONS = {
|
|
|
155
154
|
label: 'Space Search',
|
|
156
155
|
description: 'Semantic search across public Spaces on the Hugging Face Hub.',
|
|
157
156
|
},
|
|
158
|
-
|
|
159
|
-
label: '
|
|
160
|
-
description: '
|
|
157
|
+
hub_repo_search: {
|
|
158
|
+
label: 'Repo Search',
|
|
159
|
+
description: 'Search models, datasets, and optional spaces with one shared query.',
|
|
161
160
|
},
|
|
162
161
|
model_details: {
|
|
163
162
|
label: 'Model Details',
|
|
@@ -167,10 +166,6 @@ const TOOL_DESCRIPTIONS = {
|
|
|
167
166
|
label: 'Paper Search',
|
|
168
167
|
description: 'Discover research papers relevant to your query.',
|
|
169
168
|
},
|
|
170
|
-
dataset_search: {
|
|
171
|
-
label: 'Dataset Search',
|
|
172
|
-
description: 'Explore datasets published on the Hub.',
|
|
173
|
-
},
|
|
174
169
|
dataset_details: {
|
|
175
170
|
label: 'Dataset Details',
|
|
176
171
|
description: 'Inspect dataset metadata and card information.',
|