@llmindset/hf-mcp-server 0.3.0 → 0.3.2
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/mcp-server.d.ts +0 -1
- package/dist/server/mcp-server.d.ts.map +1 -1
- package/dist/server/mcp-server.js +50 -14
- package/dist/server/mcp-server.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/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/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/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/web/assets/{main-BajWSVcC.js → main-CdA2xGil.js} +35 -35
- package/dist/web/index.html +1 -1
- package/dist/web/mcp-welcome.html +38 -38
- package/package.json +3 -3
|
@@ -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":"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,MAAM,MAAM,UAAU,GAAG,MAAM,OAAO,QAAQ,CAAC;AAE/C,MAAM,WAAW,iBAAiB;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC7B,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,MAAM,MAAM,eAAe,GAAG,MAAM,GAAG,eAAe,CAAC;AAEvD,MAAM,WAAW,sBAAsB;IACtC,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.',
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"bouquet-presets.js","sourceRoot":"","sources":["../../src/shared/bouquet-presets.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,oBAAoB,EACpB,cAAc,EACd,
|
|
1
|
+
{"version":3,"file":"bouquet-presets.js","sourceRoot":"","sources":["../../src/shared/bouquet-presets.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,oBAAoB,EACpB,cAAc,EACd,wBAAwB,EACxB,iBAAiB,EACjB,eAAe,EACf,qBAAqB,EACrB,mBAAmB,EACnB,4BAA4B,GAC5B,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EAAE,mBAAmB,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAEpF,MAAM,CAAC,MAAM,QAAQ,GAAgC;IACpD,MAAM,EAAE;QACP,YAAY,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC;QACxC,UAAU,EAAE,EAAE;KACd;IACD,MAAM,EAAE;QACP,YAAY,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC;QACxC,UAAU,EAAE,EAAE;KACd;IACD,MAAM,EAAE;QACP,YAAY,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC;QACxC,UAAU,EAAE,EAAE;KACd;IACD,IAAI,EAAE;QACL,YAAY,EAAE,CAAC,GAAG,cAAc,CAAC,IAAI,CAAC;QACtC,UAAU,EAAE,EAAE;KACd;IACD,MAAM,EAAE;QACP,YAAY,EAAE;YACb,wBAAwB;YACxB,mBAAmB;YACnB,mBAAmB;YACnB,4BAA4B;YAC5B,eAAe;SACf;QACD,UAAU,EAAE,EAAE;KACd;IACD,GAAG,EAAE;QACJ,YAAY,EAAE,CAAC,GAAG,oBAAoB,CAAC;QACvC,UAAU,EAAE,EAAE;KACd;IAED,uBAAuB,EAAE;QACxB,YAAY,EAAE,CAAC,wBAAwB,EAAE,mBAAmB,CAAC;QAC7D,UAAU,EAAE,EAAE;KACd;IACD,gBAAgB,EAAE;QACjB,YAAY,EAAE,CAAC,wBAAwB,CAAC;QACxC,UAAU,EAAE,EAAE;KACd;IACD,gBAAgB,EAAE;QACjB,YAAY,EAAE,CAAC,wBAAwB,CAAC;QACxC,UAAU,EAAE,EAAE;KACd;IACD,MAAM,EAAE;QACP,YAAY,EAAE,CAAC,iBAAiB,CAAC;QACjC,UAAU,EAAE,EAAE;KACd;IACD,IAAI,EAAE;QACL,YAAY,EAAE,CAAC,eAAe,CAAC;QAC/B,UAAU,EAAE,EAAE;KACd;IACD,aAAa,EAAE;QACd,YAAY,EAAE,CAAC,qBAAqB,CAAC;QACrC,UAAU,EAAE,EAAE;KACd;IACD,KAAK,EAAE;QACN,YAAY,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC;QACxC,UAAU,EAAE,EAAE;KACd;CACD,CAAC;AAqBF,MAAM,WAAW,GAA+C;IAC/D;QACC,GAAG,EAAE,QAAQ;QACb,KAAK,EAAE,gBAAgB;QACvB,WAAW,EAAE,wFAAwF;QACrG,QAAQ,EAAE,MAAM;QAChB,eAAe,EAAE,IAAI;QACrB,WAAW,EAAE,IAAI;KACjB;IACD;QACC,GAAG,EAAE,QAAQ;QACb,KAAK,EAAE,gBAAgB;QACvB,WAAW,EAAE,yDAAyD;QACtE,QAAQ,EAAE,MAAM;QAChB,eAAe,EAAE,IAAI;QACrB,WAAW,EAAE,IAAI;KACjB;IACD;QACC,GAAG,EAAE,QAAQ;QACb,KAAK,EAAE,cAAc;QACrB,WAAW,EAAE,2DAA2D;QACxE,QAAQ,EAAE,MAAM;QAChB,eAAe,EAAE,IAAI;QACrB,WAAW,EAAE,IAAI;KACjB;IACD;QACC,GAAG,EAAE,MAAM;QACX,KAAK,EAAE,4BAA4B;QACnC,WAAW,EAAE,uCAAuC;QACpD,QAAQ,EAAE,MAAM;QAChB,eAAe,EAAE,IAAI;QACrB,WAAW,EAAE,IAAI;KACjB;IACD;QACC,GAAG,EAAE,QAAQ;QACb,KAAK,EAAE,2BAA2B;QAClC,WAAW,EAAE,2EAA2E;QACxF,QAAQ,EAAE,MAAM;QAChB,eAAe,EAAE,IAAI;QACrB,WAAW,EAAE,IAAI;KACjB;IA0BD;QACC,GAAG,EAAE,kBAAkB;QACvB,KAAK,EAAE,kBAAkB;QACzB,WAAW,EAAE,6DAA6D;QAC1E,QAAQ,EAAE,UAAU;QACpB,eAAe,EAAE,KAAK;QACtB,WAAW,EAAE,KAAK;QAClB,YAAY,EAAE;YACb;gBACC,KAAK,EAAE,uBAAuB;gBAC9B,KAAK,EAAE,uBAAuB;gBAC9B,WAAW,EAAE,mDAAmD;aAChE;SACD;KACD;IACD;QACC,GAAG,EAAE,QAAQ;QACb,KAAK,EAAE,gBAAgB;QACvB,WAAW,EACV,+GAA+G;QAChH,QAAQ,EAAE,UAAU;QACpB,eAAe,EAAE,IAAI;QACrB,WAAW,EAAE,IAAI;KACjB;IACD;QACC,GAAG,EAAE,MAAM;QACX,KAAK,EAAE,qBAAqB;QAC5B,WAAW,EAAE,gEAAgE;QAC7E,QAAQ,EAAE,UAAU;QACpB,eAAe,EAAE,IAAI;QACrB,WAAW,EAAE,IAAI;KACjB;IACD;QACC,GAAG,EAAE,OAAO;QACZ,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE,6DAA6D;QAC1E,QAAQ,EAAE,UAAU;QACpB,eAAe,EAAE,IAAI;QACrB,WAAW,EAAE,IAAI;KACjB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,eAAe,GAAoB,WAAW,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;IAC1E,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACpC,OAAO;QACN,GAAG,MAAM;QACT,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE;QACpD,YAAY,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS;KACpG,CAAC;AACH,CAAC,CAAC,CAAC;AAWH,MAAM,iBAAiB,GAAgE;IACtF,YAAY,EAAE;QACb,KAAK,EAAE,cAAc;QACrB,WAAW,EAAE,+DAA+D;KAC5E;IACD,eAAe,EAAE;QAChB,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE,qEAAqE;KAClF;IACD,aAAa,EAAE;QACd,KAAK,EAAE,eAAe;QACtB,WAAW,EAAE,6DAA6D;KAC1E;IACD,YAAY,EAAE;QACb,KAAK,EAAE,cAAc;QACrB,WAAW,EAAE,kDAAkD;KAC/D;IACD,eAAe,EAAE;QAChB,KAAK,EAAE,iBAAiB;QACxB,WAAW,EAAE,gDAAgD;KAC7D;IACD,eAAe,EAAE;QAChB,KAAK,EAAE,iBAAiB;QACxB,WAAW,EAAE,sDAAsD;KACnE;IACD,UAAU,EAAE;QACX,KAAK,EAAE,YAAY;QACnB,WAAW,EAAE,6CAA6C;KAC1D;IACD,WAAW,EAAE;QACZ,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE,kDAAkD;KAC/D;IACD,SAAS,EAAE;QACV,KAAK,EAAE,WAAW;QAClB,WAAW,EAAE,qDAAqD;KAClE;IACD,aAAa,EAAE;QACd,KAAK,EAAE,aAAa;QACpB,WAAW,EAAE,6CAA6C;KAC1D;IACD,YAAY,EAAE;QACb,KAAK,EAAE,YAAY;QACnB,WAAW,EAAE,2DAA2D;KACxE;IACD,gBAAgB,EAAE;QACjB,KAAK,EAAE,kBAAkB;QACzB,WAAW,EAAE,8DAA8D;KAC3E;IACD,OAAO,EAAE;QACR,KAAK,EAAE,SAAS;QAChB,WAAW,EAAE,gEAAgE;KAC7E;CACD,CAAC;AAEF,MAAM,0BAA0B,GAAgE;IAC/F,CAAC,mBAAmB,CAAC,EAAE;QACtB,KAAK,EAAE,sBAAsB;QAC7B,WAAW,EAAE,gEAAgE;KAC7E;IACD,CAAC,wBAAwB,CAAC,EAAE;QAC3B,KAAK,EAAE,oBAAoB;QAC3B,WAAW,EAAE,2DAA2D;KACxE;CACD,CAAC;AAEF,MAAM,UAAU,mBAAmB,CAAC,EAAU;IAC7C,MAAM,IAAI,GAAG,iBAAiB,CAAC,EAAE,CAAC,CAAC;IACnC,IAAI,IAAI,EAAE,CAAC;QACV,OAAO;YACN,EAAE;YACF,IAAI,EAAE,MAAM;YACZ,GAAG,IAAI;SACP,CAAC;IACH,CAAC;IACD,MAAM,IAAI,GAAG,0BAA0B,CAAC,EAAE,CAAC,CAAC;IAC5C,IAAI,IAAI,EAAE,CAAC;QACV,OAAO;YACN,EAAE;YACF,IAAI,EAAE,eAAe;YACrB,GAAG,IAAI;SACP,CAAC;IACH,CAAC;IACD,OAAO;QACN,EAAE;QACF,KAAK,EAAE,EAAE;QACT,IAAI,EAAE,MAAM;KACZ,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
export declare function mapLegacySearchToolId(toolId: string): string;
|
|
2
|
+
export declare function isLegacyModelSearchTool(toolName: string): boolean;
|
|
3
|
+
export declare function isLegacyDatasetSearchTool(toolName: string): boolean;
|
|
4
|
+
export declare function isLegacyRepoSearchTool(toolName: string): boolean;
|
|
5
|
+
//# sourceMappingURL=repo-search-migration.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"repo-search-migration.d.ts","sourceRoot":"","sources":["../../src/shared/repo-search-migration.ts"],"names":[],"mappings":"AAgBA,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAU5D;AAED,wBAAgB,uBAAuB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAEjE;AAED,wBAAgB,yBAAyB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAEnE;AAED,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAEhE"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import { DATASET_SEARCH_TOOL_ID, MODEL_SEARCH_TOOL_ID, REPO_SEARCH_TOOL_ID } from '@llmindset/hf-mcp';
|
|
2
|
+
const LEGACY_MODEL_SEARCH_ALIASES = new Set([MODEL_SEARCH_TOOL_ID, 'model-search', 'hf_model_search']);
|
|
3
|
+
const LEGACY_DATASET_SEARCH_ALIASES = new Set([DATASET_SEARCH_TOOL_ID, 'dataset-search', 'hf_dataset_search']);
|
|
4
|
+
const LEGACY_REPO_SEARCH_ALIASES = new Set([REPO_SEARCH_TOOL_ID, 'repo_search', 'repo-search', 'hf_repo_search']);
|
|
5
|
+
export function mapLegacySearchToolId(toolId) {
|
|
6
|
+
if (LEGACY_MODEL_SEARCH_ALIASES.has(toolId) ||
|
|
7
|
+
LEGACY_DATASET_SEARCH_ALIASES.has(toolId) ||
|
|
8
|
+
LEGACY_REPO_SEARCH_ALIASES.has(toolId)) {
|
|
9
|
+
return REPO_SEARCH_TOOL_ID;
|
|
10
|
+
}
|
|
11
|
+
return toolId;
|
|
12
|
+
}
|
|
13
|
+
export function isLegacyModelSearchTool(toolName) {
|
|
14
|
+
return LEGACY_MODEL_SEARCH_ALIASES.has(toolName);
|
|
15
|
+
}
|
|
16
|
+
export function isLegacyDatasetSearchTool(toolName) {
|
|
17
|
+
return LEGACY_DATASET_SEARCH_ALIASES.has(toolName);
|
|
18
|
+
}
|
|
19
|
+
export function isLegacyRepoSearchTool(toolName) {
|
|
20
|
+
return LEGACY_REPO_SEARCH_ALIASES.has(toolName) && toolName !== REPO_SEARCH_TOOL_ID;
|
|
21
|
+
}
|
|
22
|
+
//# sourceMappingURL=repo-search-migration.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"repo-search-migration.js","sourceRoot":"","sources":["../../src/shared/repo-search-migration.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAMtG,MAAM,2BAA2B,GAAG,IAAI,GAAG,CAAC,CAAC,oBAAoB,EAAE,cAAc,EAAE,iBAAiB,CAAC,CAAC,CAAC;AACvG,MAAM,6BAA6B,GAAG,IAAI,GAAG,CAAC,CAAC,sBAAsB,EAAE,gBAAgB,EAAE,mBAAmB,CAAC,CAAC,CAAC;AAC/G,MAAM,0BAA0B,GAAG,IAAI,GAAG,CAAC,CAAC,mBAAmB,EAAE,aAAa,EAAE,aAAa,EAAE,gBAAgB,CAAC,CAAC,CAAC;AAQlH,MAAM,UAAU,qBAAqB,CAAC,MAAc;IACnD,IACC,2BAA2B,CAAC,GAAG,CAAC,MAAM,CAAC;QACvC,6BAA6B,CAAC,GAAG,CAAC,MAAM,CAAC;QACzC,0BAA0B,CAAC,GAAG,CAAC,MAAM,CAAC,EACrC,CAAC;QACF,OAAO,mBAAmB,CAAC;IAC5B,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,QAAgB;IACvD,OAAO,2BAA2B,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,QAAgB;IACzD,OAAO,6BAA6B,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,QAAgB;IACtD,OAAO,0BAA0B,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,QAAQ,KAAK,mBAAmB,CAAC;AACrF,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tool-normalizer.d.ts","sourceRoot":"","sources":["../../src/shared/tool-normalizer.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"tool-normalizer.d.ts","sourceRoot":"","sources":["../../src/shared/tool-normalizer.ts"],"names":[],"mappings":"AAQA,wBAAgB,qBAAqB,CAAC,GAAG,EAAE,SAAS,MAAM,EAAE,GAAG,MAAM,EAAE,CAyBtE"}
|
|
@@ -1,21 +1,23 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { HUB_REPO_DETAILS_TOOL_ID, MODEL_DETAIL_TOOL_ID, DATASET_DETAIL_TOOL_ID } from '@llmindset/hf-mcp';
|
|
2
|
+
import { mapLegacySearchToolId } from './repo-search-migration.js';
|
|
2
3
|
export function normalizeBuiltInTools(ids) {
|
|
3
4
|
const seen = new Set();
|
|
4
5
|
const normalized = [];
|
|
5
6
|
let addHubInspect = false;
|
|
6
7
|
for (const rawId of ids) {
|
|
7
|
-
|
|
8
|
+
const normalizedToolId = mapLegacySearchToolId(rawId);
|
|
9
|
+
if (normalizedToolId === MODEL_DETAIL_TOOL_ID || normalizedToolId === DATASET_DETAIL_TOOL_ID) {
|
|
8
10
|
addHubInspect = true;
|
|
9
11
|
continue;
|
|
10
12
|
}
|
|
11
|
-
if (!seen.has(
|
|
12
|
-
seen.add(
|
|
13
|
-
normalized.push(
|
|
13
|
+
if (!seen.has(normalizedToolId)) {
|
|
14
|
+
seen.add(normalizedToolId);
|
|
15
|
+
normalized.push(normalizedToolId);
|
|
14
16
|
}
|
|
15
17
|
}
|
|
16
|
-
if (addHubInspect && !seen.has(
|
|
17
|
-
seen.add(
|
|
18
|
-
normalized.push(
|
|
18
|
+
if (addHubInspect && !seen.has(HUB_REPO_DETAILS_TOOL_ID)) {
|
|
19
|
+
seen.add(HUB_REPO_DETAILS_TOOL_ID);
|
|
20
|
+
normalized.push(HUB_REPO_DETAILS_TOOL_ID);
|
|
19
21
|
}
|
|
20
22
|
return normalized;
|
|
21
23
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tool-normalizer.js","sourceRoot":"","sources":["../../src/shared/tool-normalizer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"tool-normalizer.js","sourceRoot":"","sources":["../../src/shared/tool-normalizer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,wBAAwB,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAC3G,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAOnE,MAAM,UAAU,qBAAqB,CAAC,GAAsB;IAC3D,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,IAAI,aAAa,GAAG,KAAK,CAAC;IAE1B,KAAK,MAAM,KAAK,IAAI,GAAG,EAAE,CAAC;QACzB,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QAEtD,IAAI,gBAAgB,KAAK,oBAAoB,IAAI,gBAAgB,KAAK,sBAAsB,EAAE,CAAC;YAC9F,aAAa,GAAG,IAAI,CAAC;YACrB,SAAS;QACV,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YAC3B,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACnC,CAAC;IACF,CAAC;IAED,IAAI,aAAa,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,wBAAwB,CAAC,EAAE,CAAC;QAC1D,IAAI,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QACnC,UAAU,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC3C,CAAC;IAED,OAAO,UAAU,CAAC;AACnB,CAAC"}
|