@llmindset/hf-mcp 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/docs-search/doc-fetch.d.ts +1 -0
- package/dist/docs-search/doc-fetch.d.ts.map +1 -1
- package/dist/docs-search/doc-fetch.js +9 -12
- package/dist/docs-search/doc-fetch.js.map +1 -1
- package/dist/docs-search/doc-fetch.test.js +56 -11
- package/dist/docs-search/doc-fetch.test.js.map +1 -1
- package/dist/docs-search/docs-semantic-search.d.ts.map +1 -1
- package/dist/docs-search/docs-semantic-search.js +7 -1
- package/dist/docs-search/docs-semantic-search.js.map +1 -1
- package/dist/file-icons.d.ts +3 -0
- package/dist/file-icons.d.ts.map +1 -0
- package/dist/file-icons.js +38 -0
- package/dist/file-icons.js.map +1 -0
- package/dist/gradio-files.d.ts +0 -1
- package/dist/gradio-files.d.ts.map +1 -1
- package/dist/gradio-files.js +2 -35
- package/dist/gradio-files.js.map +1 -1
- package/dist/hf-api-call.d.ts.map +1 -1
- package/dist/hf-api-call.js +7 -7
- package/dist/hf-api-call.js.map +1 -1
- package/dist/hub-inspect.d.ts +2 -2
- package/dist/hub-inspect.d.ts.map +1 -1
- package/dist/hub-inspect.js +1 -1
- package/dist/hub-inspect.js.map +1 -1
- package/dist/index.browser.d.ts +48 -0
- package/dist/index.browser.d.ts.map +1 -0
- package/dist/index.browser.js +153 -0
- package/dist/index.browser.js.map +1 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +2 -0
- package/dist/index.js.map +1 -1
- package/dist/jobs/commands/uv-utils.d.ts +0 -3
- package/dist/jobs/commands/uv-utils.d.ts.map +1 -1
- package/dist/jobs/commands/uv-utils.js +2 -2
- package/dist/jobs/commands/uv-utils.js.map +1 -1
- package/dist/jobs/jobs-tool.d.ts.map +1 -1
- package/dist/jobs/jobs-tool.js +11 -12
- package/dist/jobs/jobs-tool.js.map +1 -1
- package/dist/jobs/schema-help.d.ts +2 -9
- package/dist/jobs/schema-help.d.ts.map +1 -1
- package/dist/jobs/schema-help.js +3 -3
- package/dist/jobs/schema-help.js.map +1 -1
- package/dist/jobs/sse-handler.d.ts +3 -2
- package/dist/jobs/sse-handler.d.ts.map +1 -1
- package/dist/jobs/sse-handler.js +8 -4
- package/dist/jobs/sse-handler.js.map +1 -1
- package/dist/jobs/types.d.ts +1 -1
- package/dist/logger.d.ts +2 -2
- package/dist/logger.d.ts.map +1 -1
- package/dist/network/fetch-profile.d.ts +24 -0
- package/dist/network/fetch-profile.d.ts.map +1 -0
- package/dist/network/fetch-profile.js +80 -0
- package/dist/network/fetch-profile.js.map +1 -0
- package/dist/network/index.d.ts +5 -0
- package/dist/network/index.d.ts.map +1 -0
- package/dist/network/index.js +5 -0
- package/dist/network/index.js.map +1 -0
- package/dist/network/ip-policy.d.ts +6 -0
- package/dist/network/ip-policy.d.ts.map +1 -0
- package/dist/network/ip-policy.js +166 -0
- package/dist/network/ip-policy.js.map +1 -0
- package/dist/network/ip-policy.test.d.ts +2 -0
- package/dist/network/ip-policy.test.d.ts.map +1 -0
- package/dist/network/ip-policy.test.js +26 -0
- package/dist/network/ip-policy.test.js.map +1 -0
- package/dist/network/safe-fetch.d.ts +16 -0
- package/dist/network/safe-fetch.d.ts.map +1 -0
- package/dist/network/safe-fetch.js +124 -0
- package/dist/network/safe-fetch.js.map +1 -0
- package/dist/network/safe-fetch.test.d.ts +2 -0
- package/dist/network/safe-fetch.test.d.ts.map +1 -0
- package/dist/network/safe-fetch.test.js +136 -0
- package/dist/network/safe-fetch.test.js.map +1 -0
- package/dist/network/url-policy.d.ts +32 -0
- package/dist/network/url-policy.d.ts.map +1 -0
- package/dist/network/url-policy.js +230 -0
- package/dist/network/url-policy.js.map +1 -0
- package/dist/network/url-policy.test.d.ts +2 -0
- package/dist/network/url-policy.test.d.ts.map +1 -0
- package/dist/network/url-policy.test.js +57 -0
- package/dist/network/url-policy.test.js.map +1 -0
- package/dist/readme-utils.d.ts.map +1 -1
- package/dist/readme-utils.js +3 -4
- package/dist/readme-utils.js.map +1 -1
- package/dist/repo-search.d.ts +46 -0
- package/dist/repo-search.d.ts.map +1 -0
- package/dist/repo-search.js +310 -0
- package/dist/repo-search.js.map +1 -0
- package/dist/repo-search.test.d.ts +2 -0
- package/dist/repo-search.test.d.ts.map +1 -0
- package/dist/repo-search.test.js +130 -0
- package/dist/repo-search.test.js.map +1 -0
- package/dist/space/commands/discover.d.ts +0 -5
- package/dist/space/commands/discover.d.ts.map +1 -1
- package/dist/space/commands/discover.js +9 -2
- package/dist/space/commands/discover.js.map +1 -1
- package/dist/space/commands/invoke.js +1 -59
- package/dist/space/commands/invoke.js.map +1 -1
- package/dist/space/commands/view-parameters.d.ts.map +1 -1
- package/dist/space/commands/view-parameters.js +3 -98
- package/dist/space/commands/view-parameters.js.map +1 -1
- package/dist/space/dynamic-space-tool.d.ts.map +1 -1
- package/dist/space/dynamic-space-tool.js +5 -2
- package/dist/space/dynamic-space-tool.js.map +1 -1
- package/dist/space/utils/gradio-caller.d.ts.map +1 -1
- package/dist/space/utils/gradio-caller.js +13 -6
- package/dist/space/utils/gradio-caller.js.map +1 -1
- package/dist/space/utils/space-http.d.ts +8 -0
- package/dist/space/utils/space-http.d.ts.map +1 -0
- package/dist/space/utils/space-http.js +49 -0
- package/dist/space/utils/space-http.js.map +1 -0
- package/dist/space-files.d.ts +0 -1
- package/dist/space-files.d.ts.map +1 -1
- package/dist/space-files.js +3 -36
- package/dist/space-files.js.map +1 -1
- package/dist/tool-ids.d.ts +6 -5
- package/dist/tool-ids.d.ts.map +1 -1
- package/dist/tool-ids.js +9 -14
- package/dist/tool-ids.js.map +1 -1
- package/package.json +7 -3
- package/src/docs-search/doc-fetch.test.ts +98 -28
- package/src/docs-search/doc-fetch.ts +9 -16
- package/src/docs-search/docs-semantic-search.ts +8 -1
- package/src/file-icons.ts +39 -0
- package/src/gradio-files.ts +2 -40
- package/src/hf-api-call.ts +8 -10
- package/src/hub-inspect.ts +2 -2
- package/src/index.browser.ts +183 -0
- package/src/index.ts +2 -0
- package/src/jobs/commands/uv-utils.ts +2 -2
- package/src/jobs/jobs-tool.ts +13 -12
- package/src/jobs/schema-help.ts +4 -4
- package/src/jobs/sse-handler.ts +12 -7
- package/src/logger.ts +2 -2
- package/src/network/fetch-profile.ts +112 -0
- package/src/network/index.ts +4 -0
- package/src/network/ip-policy.test.ts +29 -0
- package/src/network/ip-policy.ts +206 -0
- package/src/network/safe-fetch.test.ts +181 -0
- package/src/network/safe-fetch.ts +174 -0
- package/src/network/url-policy.test.ts +100 -0
- package/src/network/url-policy.ts +304 -0
- package/src/readme-utils.ts +11 -10
- package/src/repo-search.test.ts +155 -0
- package/src/repo-search.ts +414 -0
- package/src/space/commands/discover.ts +10 -2
- package/src/space/commands/invoke.ts +1 -88
- package/src/space/commands/view-parameters.ts +3 -136
- package/src/space/dynamic-space-tool.ts +6 -2
- package/src/space/utils/gradio-caller.ts +25 -12
- package/src/space/utils/space-http.ts +75 -0
- package/src/space-files.ts +3 -41
- package/src/tool-ids.ts +10 -14
- package/test/fetch-guard.spec.ts +70 -0
- package/test/jobs/sse-handler.spec.ts +60 -0
- package/dist/space/utils/result-formatter.d.ts +0 -4
- package/dist/space/utils/result-formatter.d.ts.map +0 -1
- package/dist/space/utils/result-formatter.js +0 -146
- package/dist/space/utils/result-formatter.js.map +0 -1
- package/src/space/utils/result-formatter.ts +0 -226
|
@@ -0,0 +1,310 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { HfApiCall } from './hf-api-call.js';
|
|
3
|
+
import { formatDate, formatNumber } from './utilities.js';
|
|
4
|
+
const TAGS_TO_RETURN = 20;
|
|
5
|
+
const TOKEN_CAP = 12_500;
|
|
6
|
+
const CHARS_PER_TOKEN = 3;
|
|
7
|
+
const MAX_OUTPUT_CHARS = TOKEN_CAP * CHARS_PER_TOKEN;
|
|
8
|
+
const REPO_TYPES = ['model', 'dataset', 'space'];
|
|
9
|
+
const REPO_TYPE_LABELS = {
|
|
10
|
+
model: 'Models',
|
|
11
|
+
dataset: 'Datasets',
|
|
12
|
+
space: 'Spaces',
|
|
13
|
+
};
|
|
14
|
+
const DEFAULT_REPO_TYPES = ['model', 'dataset'];
|
|
15
|
+
export const REPO_SEARCH_TOOL_CONFIG = {
|
|
16
|
+
name: 'hub_repo_search',
|
|
17
|
+
description: 'Search Hugging Face repositories with a shared query interface. ' +
|
|
18
|
+
'You can target models, datasets, spaces, or aggregate across multiple repo types in one call. ' +
|
|
19
|
+
'Use space_search for semantic-first discovery of Spaces. ' +
|
|
20
|
+
'Include links to repositories in your response.',
|
|
21
|
+
schema: z.object({
|
|
22
|
+
query: z
|
|
23
|
+
.string()
|
|
24
|
+
.optional()
|
|
25
|
+
.describe('Search term. Leave blank and specify sort + limit to browse trending or recent repositories.'),
|
|
26
|
+
repo_types: z
|
|
27
|
+
.array(z.enum(REPO_TYPES))
|
|
28
|
+
.min(1)
|
|
29
|
+
.max(3)
|
|
30
|
+
.optional()
|
|
31
|
+
.default(DEFAULT_REPO_TYPES)
|
|
32
|
+
.describe('Repository types to search. Defaults to ["model", "dataset"]. space uses keyword search via /api/spaces.'),
|
|
33
|
+
author: z
|
|
34
|
+
.string()
|
|
35
|
+
.optional()
|
|
36
|
+
.describe("Organization or user namespace to filter by (e.g. 'google', 'meta-llama', 'huggingface')."),
|
|
37
|
+
filters: z
|
|
38
|
+
.array(z.string())
|
|
39
|
+
.optional()
|
|
40
|
+
.describe('Optional hub filter tags. Applied to each selected repo type (e.g. ["text-generation"], ["language:en"], ["mcp-server"]).'),
|
|
41
|
+
sort: z
|
|
42
|
+
.enum(['trendingScore', 'downloads', 'likes', 'createdAt', 'lastModified'])
|
|
43
|
+
.optional()
|
|
44
|
+
.describe('Sort order (descending): trendingScore, downloads, likes, createdAt, lastModified'),
|
|
45
|
+
limit: z
|
|
46
|
+
.number()
|
|
47
|
+
.min(1)
|
|
48
|
+
.max(100)
|
|
49
|
+
.optional()
|
|
50
|
+
.default(20)
|
|
51
|
+
.describe('Maximum number of results to return per selected repo type'),
|
|
52
|
+
}),
|
|
53
|
+
annotations: {
|
|
54
|
+
title: 'Repo Search',
|
|
55
|
+
destructiveHint: false,
|
|
56
|
+
readOnlyHint: true,
|
|
57
|
+
openWorldHint: true,
|
|
58
|
+
},
|
|
59
|
+
};
|
|
60
|
+
export class RepoSearchTool extends HfApiCall {
|
|
61
|
+
constructor(hfToken) {
|
|
62
|
+
super('https://huggingface.co/api', hfToken);
|
|
63
|
+
}
|
|
64
|
+
async searchWithParams(params) {
|
|
65
|
+
try {
|
|
66
|
+
const repoTypes = normalizeRepoTypes(params.repo_types);
|
|
67
|
+
const apiParams = this.toApiParams(params);
|
|
68
|
+
const searchBatches = await Promise.all(repoTypes.map(async (repoType) => {
|
|
69
|
+
const results = await this.searchByType(repoType, apiParams);
|
|
70
|
+
return { repoType, results };
|
|
71
|
+
}));
|
|
72
|
+
const totalResults = searchBatches.reduce((sum, batch) => sum + batch.results.length, 0);
|
|
73
|
+
if (totalResults === 0) {
|
|
74
|
+
return {
|
|
75
|
+
formatted: `No repositories found for the given criteria.`,
|
|
76
|
+
totalResults: 0,
|
|
77
|
+
resultsShared: 0,
|
|
78
|
+
};
|
|
79
|
+
}
|
|
80
|
+
return formatSearchResults(searchBatches, params, repoTypes);
|
|
81
|
+
}
|
|
82
|
+
catch (error) {
|
|
83
|
+
if (error instanceof Error) {
|
|
84
|
+
throw new Error(`Failed to search repositories: ${error.message}`);
|
|
85
|
+
}
|
|
86
|
+
throw error;
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
toApiParams(params) {
|
|
90
|
+
const apiParams = {};
|
|
91
|
+
if (params.query) {
|
|
92
|
+
apiParams.search = params.query;
|
|
93
|
+
}
|
|
94
|
+
if (params.author) {
|
|
95
|
+
apiParams.author = params.author;
|
|
96
|
+
}
|
|
97
|
+
if (params.filters && params.filters.length > 0) {
|
|
98
|
+
apiParams.filter = params.filters.join(',');
|
|
99
|
+
}
|
|
100
|
+
if (params.sort) {
|
|
101
|
+
apiParams.sort = params.sort;
|
|
102
|
+
apiParams.direction = '-1';
|
|
103
|
+
}
|
|
104
|
+
if (params.limit) {
|
|
105
|
+
apiParams.limit = params.limit.toString();
|
|
106
|
+
}
|
|
107
|
+
return apiParams;
|
|
108
|
+
}
|
|
109
|
+
async searchByType(repoType, params) {
|
|
110
|
+
const endpoint = repoType === 'model' ? 'models' : repoType === 'dataset' ? 'datasets' : 'spaces';
|
|
111
|
+
const url = new URL(`${this.apiUrl}/${endpoint}`);
|
|
112
|
+
if (params.search !== undefined) {
|
|
113
|
+
url.searchParams.set('search', params.search);
|
|
114
|
+
}
|
|
115
|
+
if (params.author !== undefined) {
|
|
116
|
+
url.searchParams.set('author', params.author);
|
|
117
|
+
}
|
|
118
|
+
if (params.filter !== undefined) {
|
|
119
|
+
url.searchParams.set('filter', params.filter);
|
|
120
|
+
}
|
|
121
|
+
if (params.sort !== undefined) {
|
|
122
|
+
url.searchParams.set('sort', params.sort);
|
|
123
|
+
}
|
|
124
|
+
if (params.direction !== undefined) {
|
|
125
|
+
url.searchParams.set('direction', params.direction);
|
|
126
|
+
}
|
|
127
|
+
if (params.limit !== undefined) {
|
|
128
|
+
url.searchParams.set('limit', params.limit);
|
|
129
|
+
}
|
|
130
|
+
return this.fetchFromApi(url);
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
function normalizeRepoTypes(repoTypes) {
|
|
134
|
+
if (!repoTypes || repoTypes.length === 0) {
|
|
135
|
+
return [...DEFAULT_REPO_TYPES];
|
|
136
|
+
}
|
|
137
|
+
const seen = new Set();
|
|
138
|
+
const normalized = [];
|
|
139
|
+
for (const repoType of repoTypes) {
|
|
140
|
+
if (!seen.has(repoType)) {
|
|
141
|
+
seen.add(repoType);
|
|
142
|
+
normalized.push(repoType);
|
|
143
|
+
}
|
|
144
|
+
}
|
|
145
|
+
return normalized.length > 0 ? normalized : [...DEFAULT_REPO_TYPES];
|
|
146
|
+
}
|
|
147
|
+
function formatSearchResults(searchBatches, params, repoTypes) {
|
|
148
|
+
const lines = [];
|
|
149
|
+
const totalResults = searchBatches.reduce((sum, batch) => sum + batch.results.length, 0);
|
|
150
|
+
let resultsShared = 0;
|
|
151
|
+
let truncated = false;
|
|
152
|
+
const tryAppendLines = (nextLines) => {
|
|
153
|
+
const candidate = [...lines, ...nextLines].join('\n');
|
|
154
|
+
if (candidate.length > MAX_OUTPUT_CHARS) {
|
|
155
|
+
return false;
|
|
156
|
+
}
|
|
157
|
+
lines.push(...nextLines);
|
|
158
|
+
return true;
|
|
159
|
+
};
|
|
160
|
+
const searchTerms = [];
|
|
161
|
+
if (params.query)
|
|
162
|
+
searchTerms.push(`query "${params.query}"`);
|
|
163
|
+
if (params.author)
|
|
164
|
+
searchTerms.push(`author "${params.author}"`);
|
|
165
|
+
if (params.filters && params.filters.length > 0)
|
|
166
|
+
searchTerms.push(`filters [${params.filters.join(', ')}]`);
|
|
167
|
+
if (params.sort)
|
|
168
|
+
searchTerms.push(`sorted by ${params.sort} (descending)`);
|
|
169
|
+
const repoTypesText = repoTypes.map((repoType) => REPO_TYPE_LABELS[repoType].toLowerCase()).join(', ');
|
|
170
|
+
const searchDesc = searchTerms.length > 0 ? ` matching ${searchTerms.join(', ')}` : '';
|
|
171
|
+
if (!tryAppendLines([`Found ${totalResults.toString()} repositories across ${repoTypesText}${searchDesc}.`, ''])) {
|
|
172
|
+
truncated = true;
|
|
173
|
+
}
|
|
174
|
+
outer: for (const batch of searchBatches) {
|
|
175
|
+
if (truncated) {
|
|
176
|
+
break;
|
|
177
|
+
}
|
|
178
|
+
const sectionLabel = REPO_TYPE_LABELS[batch.repoType];
|
|
179
|
+
if (!tryAppendLines([`## ${sectionLabel} (${batch.results.length.toString()})`, ''])) {
|
|
180
|
+
truncated = true;
|
|
181
|
+
break;
|
|
182
|
+
}
|
|
183
|
+
if (batch.results.length === 0) {
|
|
184
|
+
if (!tryAppendLines([`No ${sectionLabel.toLowerCase()} matched this query.`, ''])) {
|
|
185
|
+
truncated = true;
|
|
186
|
+
break;
|
|
187
|
+
}
|
|
188
|
+
continue;
|
|
189
|
+
}
|
|
190
|
+
for (const result of batch.results) {
|
|
191
|
+
const repoLines = [];
|
|
192
|
+
appendRepoResult(repoLines, batch.repoType, result);
|
|
193
|
+
if (!tryAppendLines(repoLines)) {
|
|
194
|
+
truncated = true;
|
|
195
|
+
break outer;
|
|
196
|
+
}
|
|
197
|
+
resultsShared += 1;
|
|
198
|
+
}
|
|
199
|
+
}
|
|
200
|
+
if (truncated) {
|
|
201
|
+
const truncationLines = [
|
|
202
|
+
'',
|
|
203
|
+
`⚠️ Results truncated at approximately ${TOKEN_CAP.toLocaleString()} tokens (${MAX_OUTPUT_CHARS.toLocaleString()} characters).`,
|
|
204
|
+
`Included ${resultsShared.toString()} of ${totalResults.toString()} repositories. Narrow the query, reduce limit, or filter repo_types to see more.`,
|
|
205
|
+
];
|
|
206
|
+
while (lines.length > 0 && [...lines, ...truncationLines].join('\n').length > MAX_OUTPUT_CHARS) {
|
|
207
|
+
lines.pop();
|
|
208
|
+
}
|
|
209
|
+
if ([...lines, ...truncationLines].join('\n').length <= MAX_OUTPUT_CHARS) {
|
|
210
|
+
lines.push(...truncationLines);
|
|
211
|
+
}
|
|
212
|
+
}
|
|
213
|
+
return {
|
|
214
|
+
formatted: lines.join('\n'),
|
|
215
|
+
totalResults,
|
|
216
|
+
resultsShared,
|
|
217
|
+
};
|
|
218
|
+
}
|
|
219
|
+
function appendRepoResult(lines, repoType, result) {
|
|
220
|
+
const heading = repoType === 'space' ? getSpaceHeading(result) : result.id;
|
|
221
|
+
lines.push(`### ${heading}`);
|
|
222
|
+
lines.push('');
|
|
223
|
+
if (repoType === 'dataset') {
|
|
224
|
+
const dataset = result;
|
|
225
|
+
if (dataset.description) {
|
|
226
|
+
const trimmed = dataset.description.substring(0, 200);
|
|
227
|
+
lines.push(`${trimmed}${dataset.description.length > 200 ? '...' : ''}`);
|
|
228
|
+
lines.push('');
|
|
229
|
+
}
|
|
230
|
+
}
|
|
231
|
+
if (repoType === 'space') {
|
|
232
|
+
const space = result;
|
|
233
|
+
const description = space.shortDescription || space.ai_short_description;
|
|
234
|
+
if (description) {
|
|
235
|
+
lines.push(description);
|
|
236
|
+
lines.push('');
|
|
237
|
+
}
|
|
238
|
+
}
|
|
239
|
+
const info = [];
|
|
240
|
+
if (result.author)
|
|
241
|
+
info.push(`**Author:** ${result.author}`);
|
|
242
|
+
if (repoType === 'model') {
|
|
243
|
+
const model = result;
|
|
244
|
+
if (model.pipeline_tag)
|
|
245
|
+
info.push(`**Task:** ${model.pipeline_tag}`);
|
|
246
|
+
if (model.library_name)
|
|
247
|
+
info.push(`**Library:** ${model.library_name}`);
|
|
248
|
+
}
|
|
249
|
+
if (repoType === 'space') {
|
|
250
|
+
const space = result;
|
|
251
|
+
if (space.sdk)
|
|
252
|
+
info.push(`**SDK:** ${space.sdk}`);
|
|
253
|
+
}
|
|
254
|
+
if (typeof result.downloads === 'number')
|
|
255
|
+
info.push(`**Downloads:** ${formatNumber(result.downloads)}`);
|
|
256
|
+
if (typeof result.likes === 'number')
|
|
257
|
+
info.push(`**Likes:** ${result.likes.toString()}`);
|
|
258
|
+
if (typeof result.trendingScore === 'number')
|
|
259
|
+
info.push(`**Trending Score:** ${result.trendingScore.toString()}`);
|
|
260
|
+
if (info.length > 0) {
|
|
261
|
+
lines.push(info.join(' | '));
|
|
262
|
+
lines.push('');
|
|
263
|
+
}
|
|
264
|
+
if (result.tags && result.tags.length > 0) {
|
|
265
|
+
lines.push(`**Tags:** ${result.tags.slice(0, TAGS_TO_RETURN).join(', ')}`);
|
|
266
|
+
if (result.tags.length > TAGS_TO_RETURN) {
|
|
267
|
+
lines.push(`*and ${(result.tags.length - TAGS_TO_RETURN).toString()} more...*`);
|
|
268
|
+
}
|
|
269
|
+
lines.push('');
|
|
270
|
+
}
|
|
271
|
+
const status = [];
|
|
272
|
+
if (result.private)
|
|
273
|
+
status.push('🔐 Private');
|
|
274
|
+
if (repoType === 'dataset' && result.gated)
|
|
275
|
+
status.push('🔒 Gated');
|
|
276
|
+
if (repoType === 'space' && result.disabled)
|
|
277
|
+
status.push('⛔ Disabled');
|
|
278
|
+
if (status.length > 0) {
|
|
279
|
+
lines.push(status.join(' | '));
|
|
280
|
+
lines.push('');
|
|
281
|
+
}
|
|
282
|
+
if (result.createdAt) {
|
|
283
|
+
lines.push(`**Created:** ${formatDate(result.createdAt)}`);
|
|
284
|
+
}
|
|
285
|
+
if (result.lastModified && result.lastModified !== result.createdAt) {
|
|
286
|
+
lines.push(`**Last Modified:** ${formatDate(result.lastModified)}`);
|
|
287
|
+
}
|
|
288
|
+
lines.push(`**Link:** [${getRepoLink(repoType, result.id)}](${getRepoLink(repoType, result.id)})`);
|
|
289
|
+
lines.push('');
|
|
290
|
+
lines.push('---');
|
|
291
|
+
lines.push('');
|
|
292
|
+
}
|
|
293
|
+
function getSpaceHeading(result) {
|
|
294
|
+
const space = result;
|
|
295
|
+
const title = space.title?.trim();
|
|
296
|
+
if (title && title.length > 0) {
|
|
297
|
+
return `${title} (\`${result.id}\`)`;
|
|
298
|
+
}
|
|
299
|
+
return result.id;
|
|
300
|
+
}
|
|
301
|
+
function getRepoLink(repoType, id) {
|
|
302
|
+
if (repoType === 'dataset') {
|
|
303
|
+
return `https://hf.co/datasets/${id}`;
|
|
304
|
+
}
|
|
305
|
+
if (repoType === 'space') {
|
|
306
|
+
return `https://hf.co/spaces/${id}`;
|
|
307
|
+
}
|
|
308
|
+
return `https://hf.co/${id}`;
|
|
309
|
+
}
|
|
310
|
+
//# sourceMappingURL=repo-search.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"repo-search.js","sourceRoot":"","sources":["../src/repo-search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAG1D,MAAM,cAAc,GAAG,EAAE,CAAC;AAC1B,MAAM,SAAS,GAAG,MAAM,CAAC;AACzB,MAAM,eAAe,GAAG,CAAC,CAAC;AAC1B,MAAM,gBAAgB,GAAG,SAAS,GAAG,eAAe,CAAC;AAErD,MAAM,UAAU,GAAG,CAAC,OAAO,EAAE,SAAS,EAAE,OAAO,CAAU,CAAC;AAG1D,MAAM,gBAAgB,GAA6B;IAClD,KAAK,EAAE,QAAQ;IACf,OAAO,EAAE,UAAU;IACnB,KAAK,EAAE,QAAQ;CACf,CAAC;AAEF,MAAM,kBAAkB,GAAe,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;AAE5D,MAAM,CAAC,MAAM,uBAAuB,GAAG;IACtC,IAAI,EAAE,iBAAiB;IACvB,WAAW,EACV,kEAAkE;QAClE,gGAAgG;QAChG,2DAA2D;QAC3D,iDAAiD;IAClD,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,KAAK,EAAE,CAAC;aACN,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CAAC,8FAA8F,CAAC;QAC1G,UAAU,EAAE,CAAC;aACX,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;aACzB,GAAG,CAAC,CAAC,CAAC;aACN,GAAG,CAAC,CAAC,CAAC;aACN,QAAQ,EAAE;aACV,OAAO,CAAC,kBAAkB,CAAC;aAC3B,QAAQ,CACR,0GAA0G,CAC1G;QACF,MAAM,EAAE,CAAC;aACP,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CAAC,2FAA2F,CAAC;QACvG,OAAO,EAAE,CAAC;aACR,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;aACjB,QAAQ,EAAE;aACV,QAAQ,CACR,2HAA2H,CAC3H;QACF,IAAI,EAAE,CAAC;aACL,IAAI,CAAC,CAAC,eAAe,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;aAC1E,QAAQ,EAAE;aACV,QAAQ,CAAC,mFAAmF,CAAC;QAC/F,KAAK,EAAE,CAAC;aACN,MAAM,EAAE;aACR,GAAG,CAAC,CAAC,CAAC;aACN,GAAG,CAAC,GAAG,CAAC;aACR,QAAQ,EAAE;aACV,OAAO,CAAC,EAAE,CAAC;aACX,QAAQ,CAAC,4DAA4D,CAAC;KACxE,CAAC;IACF,WAAW,EAAE;QACZ,KAAK,EAAE,aAAa;QACpB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACQ,CAAC;AA2DX,MAAM,OAAO,cAAe,SAAQ,SAA0C;IAC7E,YAAY,OAAgB;QAC3B,KAAK,CAAC,4BAA4B,EAAE,OAAO,CAAC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,MAAiC;QACvD,IAAI,CAAC;YACJ,MAAM,SAAS,GAAG,kBAAkB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACxD,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAE3C,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,GAAG,CACtC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAkC,EAAE;gBAChE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;gBAC7D,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;YAC9B,CAAC,CAAC,CACF,CAAC;YAEF,MAAM,YAAY,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACzF,IAAI,YAAY,KAAK,CAAC,EAAE,CAAC;gBACxB,OAAO;oBACN,SAAS,EAAE,+CAA+C;oBAC1D,YAAY,EAAE,CAAC;oBACf,aAAa,EAAE,CAAC;iBAChB,CAAC;YACH,CAAC;YAED,OAAO,mBAAmB,CAAC,aAAa,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;QAC9D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC5B,MAAM,IAAI,KAAK,CAAC,kCAAkC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACpE,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAEO,WAAW,CAAC,MAAiC;QACpD,MAAM,SAAS,GAAkB,EAAE,CAAC;QAEpC,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,SAAS,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC;QACjC,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YACnB,SAAS,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAClC,CAAC;QACD,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjD,SAAS,CAAC,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,SAAS,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;YAC7B,SAAS,CAAC,SAAS,GAAG,IAAI,CAAC;QAC5B,CAAC;QACD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,SAAS,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC3C,CAAC;QAED,OAAO,SAAS,CAAC;IAClB,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,QAAkB,EAAE,MAAqB;QACnE,MAAM,QAAQ,GAAG,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC;QAClG,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,IAAI,QAAQ,EAAE,CAAC,CAAC;QAElD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YACjC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/C,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YACjC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/C,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YACjC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/C,CAAC;QACD,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC/B,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;QAC3C,CAAC;QACD,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACpC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,WAAW,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAChC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAC7C,CAAC;QAED,OAAO,IAAI,CAAC,YAAY,CAAqB,GAAG,CAAC,CAAC;IACnD,CAAC;CACD;AAED,SAAS,kBAAkB,CAAC,SAAiC;IAC5D,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1C,OAAO,CAAC,GAAG,kBAAkB,CAAC,CAAC;IAChC,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,GAAG,EAAY,CAAC;IACjC,MAAM,UAAU,GAAe,EAAE,CAAC;IAElC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACzB,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACnB,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3B,CAAC;IACF,CAAC;IAED,OAAO,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,kBAAkB,CAAC,CAAC;AACrE,CAAC;AAED,SAAS,mBAAmB,CAC3B,aAAsC,EACtC,MAAiC,EACjC,SAAqB;IAErB,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,YAAY,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACzF,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,SAAS,GAAG,KAAK,CAAC;IAEtB,MAAM,cAAc,GAAG,CAAC,SAAmB,EAAW,EAAE;QACvD,MAAM,SAAS,GAAG,CAAC,GAAG,KAAK,EAAE,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,IAAI,SAAS,CAAC,MAAM,GAAG,gBAAgB,EAAE,CAAC;YACzC,OAAO,KAAK,CAAC;QACd,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;QACzB,OAAO,IAAI,CAAC;IACb,CAAC,CAAC;IAEF,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,IAAI,MAAM,CAAC,KAAK;QAAE,WAAW,CAAC,IAAI,CAAC,UAAU,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC;IAC9D,IAAI,MAAM,CAAC,MAAM;QAAE,WAAW,CAAC,IAAI,CAAC,WAAW,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;IACjE,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC;QAAE,WAAW,CAAC,IAAI,CAAC,YAAY,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC5G,IAAI,MAAM,CAAC,IAAI;QAAE,WAAW,CAAC,IAAI,CAAC,aAAa,MAAM,CAAC,IAAI,eAAe,CAAC,CAAC;IAE3E,MAAM,aAAa,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvG,MAAM,UAAU,GAAG,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACvF,IAAI,CAAC,cAAc,CAAC,CAAC,SAAS,YAAY,CAAC,QAAQ,EAAE,wBAAwB,aAAa,GAAG,UAAU,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC;QAClH,SAAS,GAAG,IAAI,CAAC;IAClB,CAAC;IAED,KAAK,EAAE,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;QAC1C,IAAI,SAAS,EAAE,CAAC;YACf,MAAM;QACP,CAAC;QAED,MAAM,YAAY,GAAG,gBAAgB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM,YAAY,KAAK,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC;YACtF,SAAS,GAAG,IAAI,CAAC;YACjB,MAAM;QACP,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM,YAAY,CAAC,WAAW,EAAE,sBAAsB,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC;gBACnF,SAAS,GAAG,IAAI,CAAC;gBACjB,MAAM;YACP,CAAC;YACD,SAAS;QACV,CAAC;QAED,KAAK,MAAM,MAAM,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YACpC,MAAM,SAAS,GAAa,EAAE,CAAC;YAC/B,gBAAgB,CAAC,SAAS,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAEpD,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,EAAE,CAAC;gBAChC,SAAS,GAAG,IAAI,CAAC;gBACjB,MAAM,KAAK,CAAC;YACb,CAAC;YAED,aAAa,IAAI,CAAC,CAAC;QACpB,CAAC;IACF,CAAC;IAED,IAAI,SAAS,EAAE,CAAC;QACf,MAAM,eAAe,GAAG;YACvB,EAAE;YACF,yCAAyC,SAAS,CAAC,cAAc,EAAE,YAAY,gBAAgB,CAAC,cAAc,EAAE,eAAe;YAC/H,YAAY,aAAa,CAAC,QAAQ,EAAE,OAAO,YAAY,CAAC,QAAQ,EAAE,kFAAkF;SACpJ,CAAC;QAEF,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,GAAG,KAAK,EAAE,GAAG,eAAe,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,gBAAgB,EAAE,CAAC;YAChG,KAAK,CAAC,GAAG,EAAE,CAAC;QACb,CAAC;QAED,IAAI,CAAC,GAAG,KAAK,EAAE,GAAG,eAAe,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,IAAI,gBAAgB,EAAE,CAAC;YAC1E,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,CAAC;QAChC,CAAC;IACF,CAAC;IAED,OAAO;QACN,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;QAC3B,YAAY;QACZ,aAAa;KACb,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAe,EAAE,QAAkB,EAAE,MAAwB;IACtF,MAAM,OAAO,GAAG,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IAC3E,KAAK,CAAC,IAAI,CAAC,OAAO,OAAO,EAAE,CAAC,CAAC;IAC7B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC5B,MAAM,OAAO,GAAG,MAA2B,CAAC;QAC5C,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YACzB,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YACtD,KAAK,CAAC,IAAI,CAAC,GAAG,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACzE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChB,CAAC;IACF,CAAC;IAED,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QAC1B,MAAM,KAAK,GAAG,MAAyB,CAAC;QACxC,MAAM,WAAW,GAAG,KAAK,CAAC,gBAAgB,IAAI,KAAK,CAAC,oBAAoB,CAAC;QACzE,IAAI,WAAW,EAAE,CAAC;YACjB,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACxB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChB,CAAC;IACF,CAAC;IAED,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,IAAI,MAAM,CAAC,MAAM;QAAE,IAAI,CAAC,IAAI,CAAC,eAAe,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IAE7D,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QAC1B,MAAM,KAAK,GAAG,MAAyB,CAAC;QACxC,IAAI,KAAK,CAAC,YAAY;YAAE,IAAI,CAAC,IAAI,CAAC,aAAa,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;QACrE,IAAI,KAAK,CAAC,YAAY;YAAE,IAAI,CAAC,IAAI,CAAC,gBAAgB,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;IACzE,CAAC;IAED,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QAC1B,MAAM,KAAK,GAAG,MAAyB,CAAC;QACxC,IAAI,KAAK,CAAC,GAAG;YAAE,IAAI,CAAC,IAAI,CAAC,YAAY,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;IACnD,CAAC;IAED,IAAI,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ;QAAE,IAAI,CAAC,IAAI,CAAC,kBAAkB,YAAY,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACxG,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;QAAE,IAAI,CAAC,IAAI,CAAC,cAAc,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IACzF,IAAI,OAAO,MAAM,CAAC,aAAa,KAAK,QAAQ;QAAE,IAAI,CAAC,IAAI,CAAC,uBAAuB,MAAM,CAAC,aAAa,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IAElH,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAC7B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3C,KAAK,CAAC,IAAI,CAAC,aAAa,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3E,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,cAAc,EAAE,CAAC;YACzC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QACjF,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;IAED,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,MAAM,CAAC,OAAO;QAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC9C,IAAI,QAAQ,KAAK,SAAS,IAAK,MAA4B,CAAC,KAAK;QAAE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC3F,IAAI,QAAQ,KAAK,OAAO,IAAK,MAA0B,CAAC,QAAQ;QAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAE5F,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAC/B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;IAED,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;QACtB,KAAK,CAAC,IAAI,CAAC,gBAAgB,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IAC5D,CAAC;IACD,IAAI,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC,YAAY,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC;QACrE,KAAK,CAAC,IAAI,CAAC,sBAAsB,UAAU,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;IACrE,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,cAAc,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC,KAAK,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;IACnG,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAChB,CAAC;AAED,SAAS,eAAe,CAAC,MAAwB;IAChD,MAAM,KAAK,GAAG,MAAyB,CAAC;IACxC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC;IAClC,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,OAAO,GAAG,KAAK,OAAO,MAAM,CAAC,EAAE,KAAK,CAAC;IACtC,CAAC;IACD,OAAO,MAAM,CAAC,EAAE,CAAC;AAClB,CAAC;AAED,SAAS,WAAW,CAAC,QAAkB,EAAE,EAAU;IAClD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC5B,OAAO,0BAA0B,EAAE,EAAE,CAAC;IACvC,CAAC;IACD,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QAC1B,OAAO,wBAAwB,EAAE,EAAE,CAAC;IACrC,CAAC;IACD,OAAO,iBAAiB,EAAE,EAAE,CAAC;AAC9B,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"repo-search.test.d.ts","sourceRoot":"","sources":["../src/repo-search.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,130 @@
|
|
|
1
|
+
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
|
|
2
|
+
import { RepoSearchTool } from './repo-search.js';
|
|
3
|
+
describe('RepoSearchTool', () => {
|
|
4
|
+
const originalFetch = globalThis.fetch;
|
|
5
|
+
let calls = [];
|
|
6
|
+
beforeEach(() => {
|
|
7
|
+
calls = [];
|
|
8
|
+
vi.stubGlobal('fetch', (input, init) => {
|
|
9
|
+
const inputString = stringifyRequestInput(input);
|
|
10
|
+
calls.push({ input: inputString, init });
|
|
11
|
+
if (inputString.includes('/api/models')) {
|
|
12
|
+
return Promise.resolve(jsonResponse([
|
|
13
|
+
{
|
|
14
|
+
id: 'meta-llama/Llama-3.1-8B-Instruct',
|
|
15
|
+
pipeline_tag: 'text-generation',
|
|
16
|
+
library_name: 'transformers',
|
|
17
|
+
downloads: 123,
|
|
18
|
+
likes: 10,
|
|
19
|
+
tags: ['text-generation'],
|
|
20
|
+
},
|
|
21
|
+
]));
|
|
22
|
+
}
|
|
23
|
+
if (inputString.includes('/api/datasets')) {
|
|
24
|
+
return Promise.resolve(jsonResponse([
|
|
25
|
+
{
|
|
26
|
+
id: 'openbmb/UltraData-Math',
|
|
27
|
+
description: 'Large-scale mathematical dataset',
|
|
28
|
+
downloads: 50,
|
|
29
|
+
likes: 3,
|
|
30
|
+
tags: ['math'],
|
|
31
|
+
},
|
|
32
|
+
]));
|
|
33
|
+
}
|
|
34
|
+
if (inputString.includes('/api/spaces')) {
|
|
35
|
+
return Promise.resolve(jsonResponse([
|
|
36
|
+
{
|
|
37
|
+
id: 'mrfakename/Z-Image-Turbo',
|
|
38
|
+
title: 'Z Image Turbo',
|
|
39
|
+
sdk: 'gradio',
|
|
40
|
+
likes: 20,
|
|
41
|
+
},
|
|
42
|
+
]));
|
|
43
|
+
}
|
|
44
|
+
return Promise.resolve(jsonResponse([]));
|
|
45
|
+
});
|
|
46
|
+
});
|
|
47
|
+
afterEach(() => {
|
|
48
|
+
vi.unstubAllGlobals();
|
|
49
|
+
globalThis.fetch = originalFetch;
|
|
50
|
+
});
|
|
51
|
+
it('aggregates model and dataset results in one response', async () => {
|
|
52
|
+
const tool = new RepoSearchTool('token');
|
|
53
|
+
const result = await tool.searchWithParams({
|
|
54
|
+
query: 'llama',
|
|
55
|
+
repo_types: ['model', 'dataset'],
|
|
56
|
+
limit: 5,
|
|
57
|
+
});
|
|
58
|
+
expect(calls).toHaveLength(2);
|
|
59
|
+
expect(calls[0]?.input).toContain('/api/models');
|
|
60
|
+
expect(calls[1]?.input).toContain('/api/datasets');
|
|
61
|
+
expect(result.totalResults).toBe(2);
|
|
62
|
+
expect(result.formatted).toContain('## Models (1)');
|
|
63
|
+
expect(result.formatted).toContain('## Datasets (1)');
|
|
64
|
+
expect(result.formatted).toContain('[https://hf.co/meta-llama/Llama-3.1-8B-Instruct]');
|
|
65
|
+
expect(result.formatted).toContain('[https://hf.co/datasets/openbmb/UltraData-Math]');
|
|
66
|
+
});
|
|
67
|
+
it('supports searching spaces through the same interface', async () => {
|
|
68
|
+
const tool = new RepoSearchTool('token');
|
|
69
|
+
const result = await tool.searchWithParams({
|
|
70
|
+
query: 'image generation',
|
|
71
|
+
repo_types: ['space'],
|
|
72
|
+
limit: 3,
|
|
73
|
+
});
|
|
74
|
+
expect(calls).toHaveLength(1);
|
|
75
|
+
expect(calls[0]?.input).toContain('/api/spaces');
|
|
76
|
+
expect(result.totalResults).toBe(1);
|
|
77
|
+
expect(result.formatted).toContain('## Spaces (1)');
|
|
78
|
+
expect(result.formatted).toContain('[https://hf.co/spaces/mrfakename/Z-Image-Turbo]');
|
|
79
|
+
});
|
|
80
|
+
it('applies an overall output length guard with truncation notice', async () => {
|
|
81
|
+
vi.stubGlobal('fetch', (input, init) => {
|
|
82
|
+
const inputString = stringifyRequestInput(input);
|
|
83
|
+
calls.push({ input: inputString, init });
|
|
84
|
+
if (inputString.includes('/api/models')) {
|
|
85
|
+
const longTag = 'very-long-tag-name-for-output-growth';
|
|
86
|
+
const models = Array.from({ length: 100 }, (_, index) => ({
|
|
87
|
+
id: `org/super-long-model-name-${index.toString().padStart(3, '0')}-with-extra-context`,
|
|
88
|
+
pipeline_tag: 'text-generation',
|
|
89
|
+
library_name: 'transformers',
|
|
90
|
+
downloads: 100000 - index,
|
|
91
|
+
likes: 1000 - index,
|
|
92
|
+
tags: Array.from({ length: 30 }, (_unused, tagIndex) => `${longTag}-${tagIndex.toString()}`),
|
|
93
|
+
}));
|
|
94
|
+
return Promise.resolve(jsonResponse(models));
|
|
95
|
+
}
|
|
96
|
+
return Promise.resolve(jsonResponse([]));
|
|
97
|
+
});
|
|
98
|
+
const tool = new RepoSearchTool('token');
|
|
99
|
+
const result = await tool.searchWithParams({
|
|
100
|
+
repo_types: ['model'],
|
|
101
|
+
limit: 100,
|
|
102
|
+
});
|
|
103
|
+
expect(result.totalResults).toBe(100);
|
|
104
|
+
expect(result.resultsShared).toBeLessThan(result.totalResults);
|
|
105
|
+
expect(result.formatted.length).toBeLessThanOrEqual(12_500 * 3);
|
|
106
|
+
expect(result.formatted).toContain('Results truncated at approximately 12,500 tokens');
|
|
107
|
+
expect(result.formatted).toContain('Included');
|
|
108
|
+
});
|
|
109
|
+
});
|
|
110
|
+
function jsonResponse(payload) {
|
|
111
|
+
return new Response(JSON.stringify(payload), {
|
|
112
|
+
status: 200,
|
|
113
|
+
headers: {
|
|
114
|
+
'content-type': 'application/json',
|
|
115
|
+
},
|
|
116
|
+
});
|
|
117
|
+
}
|
|
118
|
+
function stringifyRequestInput(input) {
|
|
119
|
+
if (typeof input === 'string') {
|
|
120
|
+
return input;
|
|
121
|
+
}
|
|
122
|
+
if (input instanceof URL) {
|
|
123
|
+
return input.toString();
|
|
124
|
+
}
|
|
125
|
+
if (input instanceof Request) {
|
|
126
|
+
return input.url;
|
|
127
|
+
}
|
|
128
|
+
return input;
|
|
129
|
+
}
|
|
130
|
+
//# sourceMappingURL=repo-search.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"repo-search.test.js","sourceRoot":"","sources":["../src/repo-search.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAOlD,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC/B,MAAM,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC;IACvC,IAAI,KAAK,GAAoB,EAAE,CAAC;IAEhC,UAAU,CAAC,GAAG,EAAE;QACf,KAAK,GAAG,EAAE,CAAC;QACX,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,KAAwB,EAAE,IAAkB,EAAqB,EAAE;YAC1F,MAAM,WAAW,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;YACjD,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;YAEzC,IAAI,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;gBACzC,OAAO,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC;oBACnC;wBACC,EAAE,EAAE,kCAAkC;wBACtC,YAAY,EAAE,iBAAiB;wBAC/B,YAAY,EAAE,cAAc;wBAC5B,SAAS,EAAE,GAAG;wBACd,KAAK,EAAE,EAAE;wBACT,IAAI,EAAE,CAAC,iBAAiB,CAAC;qBACzB;iBACD,CAAC,CAAC,CAAC;YACL,CAAC;YAED,IAAI,WAAW,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;gBAC3C,OAAO,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC;oBACnC;wBACC,EAAE,EAAE,wBAAwB;wBAC5B,WAAW,EAAE,kCAAkC;wBAC/C,SAAS,EAAE,EAAE;wBACb,KAAK,EAAE,CAAC;wBACR,IAAI,EAAE,CAAC,MAAM,CAAC;qBACd;iBACD,CAAC,CAAC,CAAC;YACL,CAAC;YAED,IAAI,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;gBACzC,OAAO,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC;oBACnC;wBACC,EAAE,EAAE,0BAA0B;wBAC9B,KAAK,EAAE,eAAe;wBACtB,GAAG,EAAE,QAAQ;wBACb,KAAK,EAAE,EAAE;qBACT;iBACD,CAAC,CAAC,CAAC;YACL,CAAC;YAED,OAAO,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,UAAU,CAAC,KAAK,GAAG,aAAa,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC;YAC1C,KAAK,EAAE,OAAO;YACd,UAAU,EAAE,CAAC,OAAO,EAAE,SAAS,CAAC;YAChC,KAAK,EAAE,CAAC;SACR,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACjD,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,kDAAkD,CAAC,CAAC;QACvF,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,iDAAiD,CAAC,CAAC;IACvF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC;YAC1C,KAAK,EAAE,kBAAkB;YACzB,UAAU,EAAE,CAAC,OAAO,CAAC;YACrB,KAAK,EAAE,CAAC;SACR,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,iDAAiD,CAAC,CAAC;IACvF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC9E,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,KAAwB,EAAE,IAAkB,EAAqB,EAAE;YAC1F,MAAM,WAAW,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;YACjD,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;YAEzC,IAAI,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;gBACzC,MAAM,OAAO,GAAG,sCAAsC,CAAC;gBACvD,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;oBACzD,EAAE,EAAE,6BAA6B,KAAK,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,qBAAqB;oBACvF,YAAY,EAAE,iBAAiB;oBAC/B,YAAY,EAAE,cAAc;oBAC5B,SAAS,EAAE,MAAM,GAAG,KAAK;oBACzB,KAAK,EAAE,IAAI,GAAG,KAAK;oBACnB,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,GAAG,OAAO,IAAI,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC;iBAC5F,CAAC,CAAC,CAAC;gBACJ,OAAO,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;YAC9C,CAAC;YAED,OAAO,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC;YAC1C,UAAU,EAAE,CAAC,OAAO,CAAC;YACrB,KAAK,EAAE,GAAG;SACV,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,kDAAkD,CAAC,CAAC;QACvF,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,SAAS,YAAY,CAAC,OAAgB;IACrC,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE;QAC5C,MAAM,EAAE,GAAG;QACX,OAAO,EAAE;YACR,cAAc,EAAE,kBAAkB;SAClC;KACD,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAwB;IACtD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,KAAK,CAAC;IACd,CAAC;IAED,IAAI,KAAK,YAAY,GAAG,EAAE,CAAC;QAC1B,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;IACzB,CAAC;IAED,IAAI,KAAK,YAAY,OAAO,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC,GAAG,CAAC;IAClB,CAAC;IAED,OAAO,KAAK,CAAC;AACd,CAAC"}
|
|
@@ -1,8 +1,3 @@
|
|
|
1
1
|
import type { ToolResult } from '../../types/tool-result.js';
|
|
2
|
-
export declare const DISCOVER_PROMPTS: {
|
|
3
|
-
RESULTS_HEADER: string;
|
|
4
|
-
NO_RESULTS: string;
|
|
5
|
-
FETCH_ERROR: (url: string, error: string) => string;
|
|
6
|
-
};
|
|
7
2
|
export declare function discoverSpaces(): Promise<ToolResult>;
|
|
8
3
|
//# sourceMappingURL=discover.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"discover.d.ts","sourceRoot":"","sources":["../../../src/space/commands/discover.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;
|
|
1
|
+
{"version":3,"file":"discover.d.ts","sourceRoot":"","sources":["../../../src/space/commands/discover.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAgF7D,wBAAsB,cAAc,IAAI,OAAO,CAAC,UAAU,CAAC,CAgD1D"}
|
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
import { escapeMarkdown } from '../../utilities.js';
|
|
2
2
|
import { VIEW_PARAMETERS } from '../types.js';
|
|
3
|
-
|
|
3
|
+
import { fetchWithProfile, NETWORK_FETCH_PROFILES } from '../../network/fetch-profile.js';
|
|
4
|
+
const DISCOVER_PROMPTS = {
|
|
4
5
|
RESULTS_HEADER: `**Available Spaces:**
|
|
5
6
|
|
|
6
7
|
These spaces can be invoked using the \`dynamic_space\` tool.
|
|
@@ -53,7 +54,13 @@ export async function discoverSpaces() {
|
|
|
53
54
|
};
|
|
54
55
|
}
|
|
55
56
|
try {
|
|
56
|
-
const
|
|
57
|
+
const allowPermissiveUrls = process.env.ALLOW_PERMISSIVE_URLS === 'true';
|
|
58
|
+
const profile = allowPermissiveUrls
|
|
59
|
+
? NETWORK_FETCH_PROFILES.httpOrHttpsPermissive()
|
|
60
|
+
: NETWORK_FETCH_PROFILES.externalHttps();
|
|
61
|
+
const { response } = await fetchWithProfile(url, profile, {
|
|
62
|
+
timeoutMs: 10000,
|
|
63
|
+
});
|
|
57
64
|
if (!response.ok) {
|
|
58
65
|
return {
|
|
59
66
|
formatted: DISCOVER_PROMPTS.FETCH_ERROR(url, `HTTP ${response.status}`),
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"discover.js","sourceRoot":"","sources":["../../../src/space/commands/discover.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;
|
|
1
|
+
{"version":3,"file":"discover.js","sourceRoot":"","sources":["../../../src/space/commands/discover.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,gCAAgC,CAAC;AAM1F,MAAM,gBAAgB,GAAG;IAExB,cAAc,EAAE;;;sBAGK,eAAe;;CAEpC;IAGA,UAAU,EAAE,6CAA6C;IAGzD,WAAW,EAAE,CAAC,GAAW,EAAE,KAAa,EAAU,EAAE,CAAC,kCAAkC,GAAG,KAAK,KAAK,EAAE;CACtG,CAAC;AAMF,SAAS,eAAe,CAAC,OAAe;IACvC,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACzC,MAAM,OAAO,GAAiE,EAAE,CAAC;IAEjF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QAC1B,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;YAAE,SAAS;QAG3B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,6BAA6B,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAEhG,IAAI,KAAK,IAAI,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YAC/C,OAAO,CAAC,IAAI,CAAC;gBACZ,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE;gBACnB,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE;gBACzB,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE;aAC5B,CAAC,CAAC;QACJ,CAAC;IACF,CAAC;IAED,OAAO,OAAO,CAAC;AAChB,CAAC;AAKD,SAAS,qBAAqB,CAAC,OAAqE;IACnG,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,gBAAgB,CAAC,UAAU,CAAC;IACpC,CAAC;IAED,IAAI,QAAQ,GAAG,gBAAgB,CAAC,cAAc,CAAC;IAG/C,QAAQ,IAAI,yCAAyC,CAAC;IACtD,QAAQ,IAAI,yCAAyC,CAAC;IAGtD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC9B,QAAQ;YACP,OAAO,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK;gBACrC,KAAK,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG;gBACvC,KAAK,cAAc,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC;IAChD,CAAC;IAED,OAAO,QAAQ,CAAC;AACjB,CAAC;AAMD,MAAM,CAAC,KAAK,UAAU,cAAc;IACnC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;IAE3C,IAAI,CAAC,GAAG,EAAE,CAAC;QACV,OAAO;YACN,SAAS,EAAE,4DAA4D;YACvE,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,OAAO,EAAE,IAAI;SACb,CAAC;IACH,CAAC;IAED,IAAI,CAAC;QACJ,MAAM,mBAAmB,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,KAAK,MAAM,CAAC;QACzE,MAAM,OAAO,GAAG,mBAAmB;YAClC,CAAC,CAAC,sBAAsB,CAAC,qBAAqB,EAAE;YAChD,CAAC,CAAC,sBAAsB,CAAC,aAAa,EAAE,CAAC;QAE1C,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE,OAAO,EAAE;YACzD,SAAS,EAAE,KAAK;SAChB,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAClB,OAAO;gBACN,SAAS,EAAE,gBAAgB,CAAC,WAAW,CAAC,GAAG,EAAE,QAAQ,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACvE,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;gBAChB,OAAO,EAAE,IAAI;aACb,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACtC,MAAM,OAAO,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;QAEzC,OAAO;YACN,SAAS,EAAE,qBAAqB,CAAC,OAAO,CAAC;YACzC,YAAY,EAAE,OAAO,CAAC,MAAM;YAC5B,aAAa,EAAE,OAAO,CAAC,MAAM;SAC7B,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,OAAO;YACN,SAAS,EAAE,gBAAgB,CAAC,WAAW,CAAC,GAAG,EAAE,YAAY,CAAC;YAC1D,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,OAAO,EAAE,IAAI;SACb,CAAC;IACH,CAAC;AACF,CAAC"}
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import { analyzeSchemaComplexity, validateParameters, applyDefaults } from '../utils/schema-validator.js';
|
|
2
2
|
import { formatComplexSchemaError, formatValidationError } from '../utils/parameter-formatter.js';
|
|
3
3
|
import { callGradioToolWithHeaders } from '../utils/gradio-caller.js';
|
|
4
|
-
import {
|
|
4
|
+
import { fetchGradioSchema, fetchSpaceMetadata } from '../utils/space-http.js';
|
|
5
5
|
export async function invokeSpace(spaceName, parametersJson, hfToken, extra) {
|
|
6
6
|
try {
|
|
7
7
|
let inputParameters;
|
|
@@ -79,62 +79,4 @@ export async function invokeSpace(spaceName, parametersJson, hfToken, extra) {
|
|
|
79
79
|
};
|
|
80
80
|
}
|
|
81
81
|
}
|
|
82
|
-
async function fetchSpaceMetadata(spaceName, hfToken) {
|
|
83
|
-
const url = `https://huggingface.co/api/spaces/${spaceName}`;
|
|
84
|
-
const headers = {};
|
|
85
|
-
if (hfToken) {
|
|
86
|
-
headers['Authorization'] = `Bearer ${hfToken}`;
|
|
87
|
-
}
|
|
88
|
-
const controller = new AbortController();
|
|
89
|
-
const timeoutId = setTimeout(() => controller.abort(), 10000);
|
|
90
|
-
try {
|
|
91
|
-
const response = await fetch(url, {
|
|
92
|
-
headers,
|
|
93
|
-
signal: controller.signal,
|
|
94
|
-
});
|
|
95
|
-
clearTimeout(timeoutId);
|
|
96
|
-
if (!response.ok) {
|
|
97
|
-
throw new Error(`HTTP ${response.status}: ${response.statusText}`);
|
|
98
|
-
}
|
|
99
|
-
const info = (await response.json());
|
|
100
|
-
if (!info.subdomain) {
|
|
101
|
-
throw new Error('Space does not have a subdomain');
|
|
102
|
-
}
|
|
103
|
-
return {
|
|
104
|
-
subdomain: info.subdomain,
|
|
105
|
-
private: info.private || false,
|
|
106
|
-
};
|
|
107
|
-
}
|
|
108
|
-
finally {
|
|
109
|
-
clearTimeout(timeoutId);
|
|
110
|
-
}
|
|
111
|
-
}
|
|
112
|
-
async function fetchGradioSchema(subdomain, isPrivate, hfToken) {
|
|
113
|
-
const schemaUrl = `https://${subdomain}.hf.space/gradio_api/mcp/schema`;
|
|
114
|
-
const headers = {
|
|
115
|
-
'Content-Type': 'application/json',
|
|
116
|
-
};
|
|
117
|
-
if (isPrivate && hfToken) {
|
|
118
|
-
headers['X-HF-Authorization'] = `Bearer ${hfToken}`;
|
|
119
|
-
}
|
|
120
|
-
const controller = new AbortController();
|
|
121
|
-
const timeoutId = setTimeout(() => controller.abort(), 10000);
|
|
122
|
-
try {
|
|
123
|
-
const response = await fetch(schemaUrl, {
|
|
124
|
-
method: 'GET',
|
|
125
|
-
headers,
|
|
126
|
-
signal: controller.signal,
|
|
127
|
-
});
|
|
128
|
-
clearTimeout(timeoutId);
|
|
129
|
-
if (!response.ok) {
|
|
130
|
-
throw new Error(`HTTP ${response.status}: ${response.statusText}`);
|
|
131
|
-
}
|
|
132
|
-
const schemaResponse = (await response.json());
|
|
133
|
-
const parsed = parseGradioSchemaResponse(schemaResponse);
|
|
134
|
-
return normalizeParsedTools(parsed);
|
|
135
|
-
}
|
|
136
|
-
finally {
|
|
137
|
-
clearTimeout(timeoutId);
|
|
138
|
-
}
|
|
139
|
-
}
|
|
140
82
|
//# sourceMappingURL=invoke.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"invoke.js","sourceRoot":"","sources":["../../../src/space/commands/invoke.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,uBAAuB,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC1G,OAAO,EAAE,wBAAwB,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AAClG,OAAO,EAAE,yBAAyB,EAAE,MAAM,2BAA2B,CAAC;AACtE,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"invoke.js","sourceRoot":"","sources":["../../../src/space/commands/invoke.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,uBAAuB,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC1G,OAAO,EAAE,wBAAwB,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AAClG,OAAO,EAAE,yBAAyB,EAAE,MAAM,2BAA2B,CAAC;AACtE,OAAO,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAM/E,MAAM,CAAC,KAAK,UAAU,WAAW,CAChC,SAAiB,EACjB,cAAsB,EACtB,OAAgB,EAChB,KAA8D;IAE9D,IAAI,CAAC;QAEJ,IAAI,eAAwC,CAAC;QAC7C,IAAI,CAAC;YACJ,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YACnD,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC5E,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACrD,CAAC;YACD,eAAe,GAAG,MAAiC,CAAC;QACrD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO;gBACN,SAAS,EAAE,8IAA8I,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;gBACjN,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;gBAChB,OAAO,EAAE,IAAI;aACb,CAAC;QACH,CAAC;QAGD,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAG9D,MAAM,KAAK,GAAG,MAAM,iBAAiB,CAAC,QAAQ,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAErF,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO;gBACN,SAAS,EAAE,oCAAoC,SAAS,IAAI;gBAC5D,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;gBAChB,OAAO,EAAE,IAAI;aACb,CAAC;QACH,CAAC;QAED,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAS,CAAC;QAG9B,MAAM,YAAY,GAAG,uBAAuB,CAAC,IAAI,CAAC,CAAC;QAEnD,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC;YAC5B,OAAO;gBACN,SAAS,EAAE,wBAAwB,CAAC,SAAS,EAAE,YAAY,CAAC,MAAM,IAAI,gBAAgB,CAAC;gBACvF,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;gBAChB,OAAO,EAAE,IAAI;aACb,CAAC;QACH,CAAC;QAGD,MAAM,UAAU,GAAG,kBAAkB,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC;QACrE,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACvB,OAAO;gBACN,SAAS,EAAE,qBAAqB,CAAC,UAAU,CAAC,MAAM,EAAE,SAAS,CAAC;gBAC9D,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;gBAChB,OAAO,EAAE,IAAI;aACb,CAAC;QACH,CAAC;QAGD,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5E,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;YAChD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC/B,QAAQ,CAAC,IAAI,CAAC,uBAAuB,GAAG,4BAA4B,CAAC,CAAC;YACvE,CAAC;QACF,CAAC;QAGD,MAAM,eAAe,GAAG,aAAa,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC;QAGrE,MAAM,MAAM,GAAG,WAAW,QAAQ,CAAC,SAAS,2BAA2B,CAAC;QACxE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,yBAAyB,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,KAAK,EAAE;YACtG,iBAAiB,EAAE,IAAI;SACvB,CAAC,CAAC;QAIH,OAAO;YACN,MAAM;YACN,QAAQ;YACR,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,OAAO,EAAE,MAAM,CAAC,OAAO;SACvB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,OAAO;YACN,SAAS,EAAE,yBAAyB,SAAS,MAAM,YAAY,EAAE;YACjE,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,OAAO,EAAE,IAAI;SACb,CAAC;IACH,CAAC;AACF,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"view-parameters.d.ts","sourceRoot":"","sources":["../../../src/space/commands/view-parameters.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;
|
|
1
|
+
{"version":3,"file":"view-parameters.d.ts","sourceRoot":"","sources":["../../../src/space/commands/view-parameters.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAS7D,wBAAsB,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CA6D7F"}
|