@llmindset/hf-mcp 0.3.2 → 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/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/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 +1 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +1 -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/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/package.json +6 -2
- package/src/docs-search/doc-fetch.test.ts +98 -28
- package/src/docs-search/doc-fetch.ts +9 -16
- package/src/file-icons.ts +39 -0
- package/src/gradio-files.ts +2 -40
- package/src/hf-api-call.ts +8 -10
- package/src/index.browser.ts +183 -0
- package/src/index.ts +1 -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/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/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
|
@@ -22,6 +22,7 @@ export declare const DOC_FETCH_CONFIG: {
|
|
|
22
22
|
export type DocFetchParams = z.infer<typeof DOC_FETCH_CONFIG.schema>;
|
|
23
23
|
export declare class DocFetchTool {
|
|
24
24
|
private turndownService;
|
|
25
|
+
private readonly docsPolicy;
|
|
25
26
|
constructor();
|
|
26
27
|
validateUrl(hfUrl: string): void;
|
|
27
28
|
fetch(params: DocFetchParams): Promise<string>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"doc-fetch.d.ts","sourceRoot":"","sources":["../../src/docs-search/doc-fetch.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;
|
|
1
|
+
{"version":3,"file":"doc-fetch.d.ts","sourceRoot":"","sources":["../../src/docs-search/doc-fetch.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAMxB,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;;;CAkBnB,CAAC;AAEX,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,MAAM,CAAC,CAAC;AAErE,qBAAa,YAAY;IACxB,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAwB;;IA6FnD,WAAW,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAW1B,KAAK,CAAC,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC;IAuCpD,OAAO,CAAC,aAAa;CAoCrB;AAMD,wBAAgB,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAoBrD"}
|
|
@@ -1,6 +1,8 @@
|
|
|
1
1
|
import { z } from 'zod';
|
|
2
2
|
import TurndownService from 'turndown';
|
|
3
3
|
import { estimateTokens } from '../utilities.js';
|
|
4
|
+
import { createHfDocsPolicy, parseAndValidateUrl } from '../network/url-policy.js';
|
|
5
|
+
import { fetchWithProfile, NETWORK_FETCH_PROFILES } from '../network/fetch-profile.js';
|
|
4
6
|
export const DOC_FETCH_CONFIG = {
|
|
5
7
|
name: 'hf_doc_fetch',
|
|
6
8
|
description: 'Fetch a document from the Hugging Face or Gradio documentation library. For large documents, use offset to get subsequent chunks.',
|
|
@@ -21,6 +23,7 @@ export const DOC_FETCH_CONFIG = {
|
|
|
21
23
|
};
|
|
22
24
|
export class DocFetchTool {
|
|
23
25
|
turndownService;
|
|
26
|
+
docsPolicy = createHfDocsPolicy();
|
|
24
27
|
constructor() {
|
|
25
28
|
this.turndownService = new TurndownService({
|
|
26
29
|
headingStyle: 'atx',
|
|
@@ -95,16 +98,7 @@ export class DocFetchTool {
|
|
|
95
98
|
}
|
|
96
99
|
validateUrl(hfUrl) {
|
|
97
100
|
try {
|
|
98
|
-
|
|
99
|
-
if (url.protocol !== 'https:') {
|
|
100
|
-
throw new Error('That was not a valid documentation URL');
|
|
101
|
-
}
|
|
102
|
-
const hostname = url.hostname.toLowerCase();
|
|
103
|
-
const isHfDocs = (hostname === 'huggingface.co' || hostname === 'www.huggingface.co') && url.pathname.startsWith('/docs/');
|
|
104
|
-
const isGradio = hostname === 'gradio.app' || hostname === 'www.gradio.app';
|
|
105
|
-
if (!isHfDocs && !isGradio) {
|
|
106
|
-
throw new Error('That was not a valid documentation URL');
|
|
107
|
-
}
|
|
101
|
+
parseAndValidateUrl(hfUrl, this.docsPolicy);
|
|
108
102
|
}
|
|
109
103
|
catch {
|
|
110
104
|
throw new Error('That was not a valid documentation URL');
|
|
@@ -113,8 +107,11 @@ export class DocFetchTool {
|
|
|
113
107
|
async fetch(params) {
|
|
114
108
|
try {
|
|
115
109
|
const normalizedUrl = normalizeDocUrl(params.doc_url);
|
|
116
|
-
|
|
117
|
-
|
|
110
|
+
const { response } = await fetchWithProfile(normalizedUrl, NETWORK_FETCH_PROFILES.hfDocs(), {
|
|
111
|
+
requestInit: {
|
|
112
|
+
headers: { accept: 'text/markdown' },
|
|
113
|
+
},
|
|
114
|
+
});
|
|
118
115
|
if (!response.ok) {
|
|
119
116
|
throw new Error(`Failed to fetch document: ${response.status} ${response.statusText}`);
|
|
120
117
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"doc-fetch.js","sourceRoot":"","sources":["../../src/docs-search/doc-fetch.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,eAAe,MAAM,UAAU,CAAC;AACvC,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;
|
|
1
|
+
{"version":3,"file":"doc-fetch.js","sourceRoot":"","sources":["../../src/docs-search/doc-fetch.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,eAAe,MAAM,UAAU,CAAC;AACvC,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AACnF,OAAO,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AAEvF,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC/B,IAAI,EAAE,cAAc;IACpB,WAAW,EACV,mIAAmI;IACpI,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,CAAC,CAAC,QAAQ,CAAC,4CAA4C,CAAC;QACrG,MAAM,EAAE,CAAC;aACP,MAAM,EAAE;aACR,GAAG,CAAC,CAAC,CAAC;aACN,QAAQ,EAAE;aACV,QAAQ,CAAC,2EAA2E,CAAC;KACvF,CAAC;IACF,WAAW,EAAE;QACZ,KAAK,EAAE,8DAA8D;QACrE,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACQ,CAAC;AAIX,MAAM,OAAO,YAAY;IAChB,eAAe,CAAkB;IACxB,UAAU,GAAG,kBAAkB,EAAE,CAAC;IAEnD;QACC,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,CAAC;YAC1C,YAAY,EAAE,KAAK;YACnB,cAAc,EAAE,QAAQ;SACxB,CAAC,CAAC;QACH,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAGtC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;YACpC,IAAI,CAAC;gBACJ,MAAM,GAAG,GAAG,CAAE,IAAyC,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;gBACtF,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;oBACzG,OAAO,IAAI,CAAC;gBACb,CAAC;YACF,CAAC;YAAC,MAAM,CAAC;YAET,CAAC;YACD,OAAO,KAAK,CAAC;QACd,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;YAEpC,IAAI,CAAC;gBACJ,IAAI,OAAQ,IAAyC,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;oBAC7E,MAAM,GAAG,GAAG,CAAE,IAAwC,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;oBACrF,IAAI,GAAG,KAAK,KAAK,EAAE,CAAC;wBACnB,OAAO,IAAI,CAAC;oBACb,CAAC;gBACF,CAAC;YACF,CAAC;YAAC,MAAM,CAAC;YAET,CAAC;YAED,IAAI,IAAI,CAAC,QAAQ,KAAK,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,CAAC;gBACjF,OAAO,IAAI,CAAC;YACb,CAAC;YAED,MAAM,QAAQ,GAAG,CAAE,IAAyC,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;YAC3F,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;gBACxB,IAAI,CAAC;oBACJ,MAAM,GAAG,GACP,IAAsE,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC;wBAC7F,CAAE,IAAoC,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC;oBACnD,IACC,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC;wBACxB,4BAA4B,CAAC,IAAI,CAAC,GAAG,CAAC;wBACtC,GAAG,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC,EAC1C,CAAC;wBACF,OAAO,IAAI,CAAC;oBACb,CAAC;gBACF,CAAC;gBAAC,MAAM,CAAC;gBAET,CAAC;YACF,CAAC;YACD,OAAO,KAAK,CAAC;QACd,CAAC,CAAC,CAAC;QAGH,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,oBAAoB,EAAE;YAClD,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;gBAChB,IAAI,CAAC;oBACJ,MAAM,CAAC,GAAG,IAKT,CAAC;oBACF,IAAI,CAAC,CAAC,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,KAAK,GAAG;wBAAE,OAAO,KAAK,CAAC;oBAC3D,MAAM,IAAI,GAAG,CAAC,CAAC,YAAY,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;oBAC5C,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;wBAAE,OAAO,KAAK,CAAC;oBACjD,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;oBAC1C,MAAM,QAAQ,GAAI,CAA8D,CAAC,UAAU,IAAI,EAAE,CAAC;oBAClG,MAAM,SAAS,GACd,QAAQ,CAAC,MAAM,GAAG,CAAC;wBACnB,QAAQ,CAAC,KAAK,CACb,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,KAAK,KAAK,CAC/F,CAAC;oBACH,MAAM,mBAAmB,GAAG,wCAAwC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAChF,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAChE,OAAO,SAAS,IAAI,mBAAmB,IAAI,WAAW,CAAC;gBACxD,CAAC;gBAAC,MAAM,CAAC;oBACR,OAAO,KAAK,CAAC;gBACd,CAAC;YACF,CAAC;YACD,WAAW,EAAE,GAAG,EAAE,CAAC,EAAE;SACrB,CAAC,CAAC;IACJ,CAAC;IAKD,WAAW,CAAC,KAAa;QACxB,IAAI,CAAC;YACJ,mBAAmB,CAAC,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAC7C,CAAC;QAAC,MAAM,CAAC;YACR,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC3D,CAAC;IACF,CAAC;IAKD,KAAK,CAAC,KAAK,CAAC,MAAsB;QACjC,IAAI,CAAC;YACJ,MAAM,aAAa,GAAG,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACtD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,gBAAgB,CAAC,aAAa,EAAE,sBAAsB,CAAC,MAAM,EAAE,EAAE;gBAC3F,WAAW,EAAE;oBACZ,OAAO,EAAE,EAAE,MAAM,EAAE,eAAe,EAAE;iBACpC;aACD,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBAClB,MAAM,IAAI,KAAK,CAAC,6BAA6B,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACxF,CAAC;YACD,IAAI,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACpC,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;YAC/D,MAAM,iBAAiB,GAAG,WAAW,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,WAAW,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;YACtG,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAExB,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAKjD,OAAO,GAAG,OAAO;qBACf,OAAO,CAAC,6EAA6E,EAAE,EAAE,CAAC;qBAC1F,OAAO,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;gBAGzC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,sEAAsE,EAAE,EAAE,CAAC,CAAC;YACvG,CAAC;YAGD,OAAO,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC;QACxD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,6BAA6B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;QAC1G,CAAC;IACF,CAAC;IAKO,aAAa,CAAC,eAAuB,EAAE,MAAc;QAC5D,MAAM,WAAW,GAAG,cAAc,CAAC,eAAe,CAAC,CAAC;QACpD,MAAM,iBAAiB,GAAG,IAAI,CAAC;QAG/B,MAAM,UAAU,GAAG,eAAe,CAAC,MAAM,CAAC;QAC1C,MAAM,aAAa,GAAG,UAAU,GAAG,WAAW,CAAC;QAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,aAAa,CAAC,CAAC;QAGrD,IAAI,SAAS,IAAI,UAAU,EAAE,CAAC;YAC7B,OAAO,iBAAiB,MAAM,mCAAmC,WAAW,iBAAiB,CAAC;QAC/F,CAAC;QAGD,IAAI,WAAW,IAAI,iBAAiB,IAAI,MAAM,KAAK,CAAC,EAAE,CAAC;YACtD,OAAO,eAAe,CAAC;QACxB,CAAC;QAED,MAAM,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAiB,GAAG,aAAa,CAAC,CAAC;QACvE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,GAAG,gBAAgB,EAAE,UAAU,CAAC,CAAC;QACnE,MAAM,KAAK,GAAG,eAAe,CAAC,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAGxD,MAAM,UAAU,GAAG,MAAM,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,UAAU,GAAG,WAAW,CAAC;QAEzC,IAAI,MAAM,GAAG,KAAK,CAAC;QAGnB,IAAI,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,oCAAoC,gBAAgB,CAAC,IAAI,sBAAsB,UAAU,yBAAyB,CAAC;QAC9H,CAAC;QAED,OAAO,MAAM,CAAC;IACf,CAAC;CACD;AAMD,MAAM,UAAU,eAAe,CAAC,KAAa;IAC5C,IAAI,CAAC;QACJ,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QAC7B,IAAI,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YACjC,OAAO,yBAAyB,OAAO,EAAE,CAAC;QAC3C,CAAC;QACD,IAAI,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAClC,OAAO,0BAA0B,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QACrD,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;QAC7B,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;QACxC,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;YAC3B,GAAG,CAAC,QAAQ,GAAG,gBAAgB,CAAC;YAChC,OAAO,GAAG,CAAC,QAAQ,EAAE,CAAC;QACvB,CAAC;QACD,OAAO,OAAO,CAAC;IAChB,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,KAAK,CAAC;IACd,CAAC;AACF,CAAC"}
|
|
@@ -53,9 +53,11 @@ describe('DocFetchTool', () => {
|
|
|
53
53
|
contentType: 'text/markdown',
|
|
54
54
|
}));
|
|
55
55
|
const result = await tool.fetch({ doc_url: 'https://huggingface.co/docs/test' });
|
|
56
|
-
expect(fetchMock).
|
|
57
|
-
|
|
58
|
-
|
|
56
|
+
expect(fetchMock).toHaveBeenCalledTimes(1);
|
|
57
|
+
const [calledUrl, calledInit] = fetchMock.mock.calls[0] ?? [];
|
|
58
|
+
expect(calledUrl).toBe('https://huggingface.co/docs/test');
|
|
59
|
+
expect(calledInit?.redirect).toBe('manual');
|
|
60
|
+
expect(new Headers(calledInit?.headers).get('accept')).toBe('text/markdown');
|
|
59
61
|
expect(result).toBe(markdown);
|
|
60
62
|
});
|
|
61
63
|
it('should return small documents without chunking', async () => {
|
|
@@ -68,7 +70,8 @@ describe('DocFetchTool', () => {
|
|
|
68
70
|
expect(result).not.toContain('DOCUMENT TRUNCATED');
|
|
69
71
|
});
|
|
70
72
|
it('should chunk large documents and show truncation message', async () => {
|
|
71
|
-
const longHtml = '<h1>Long Document</h1>' +
|
|
73
|
+
const longHtml = '<h1>Long Document</h1>' +
|
|
74
|
+
'<p>This is a very long sentence that will be repeated many times to create a document that exceeds the 7500 token limit for testing chunking functionality.</p>'.repeat(200);
|
|
72
75
|
stubFetch(() => createMockResponse({
|
|
73
76
|
content: longHtml,
|
|
74
77
|
}));
|
|
@@ -95,9 +98,11 @@ describe('DocFetchTool', () => {
|
|
|
95
98
|
content: '<h1>Title</h1><p>Body</p>',
|
|
96
99
|
}));
|
|
97
100
|
const result = await tool.fetch({ doc_url: '/docs/test' });
|
|
98
|
-
expect(fetchMock).
|
|
99
|
-
|
|
100
|
-
|
|
101
|
+
expect(fetchMock).toHaveBeenCalledTimes(1);
|
|
102
|
+
const [calledUrl, calledInit] = fetchMock.mock.calls[0] ?? [];
|
|
103
|
+
expect(calledUrl).toBe('https://huggingface.co/docs/test');
|
|
104
|
+
expect(calledInit?.redirect).toBe('manual');
|
|
105
|
+
expect(new Headers(calledInit?.headers).get('accept')).toBe('text/markdown');
|
|
101
106
|
expect(result).toContain('# Title');
|
|
102
107
|
});
|
|
103
108
|
it('normalizes ./docs paths to the huggingface docs host', async () => {
|
|
@@ -105,12 +110,15 @@ describe('DocFetchTool', () => {
|
|
|
105
110
|
content: '<h1>Another Title</h1><p>Body</p>',
|
|
106
111
|
}));
|
|
107
112
|
await tool.fetch({ doc_url: './docs/another' });
|
|
108
|
-
expect(fetchMock).
|
|
109
|
-
|
|
110
|
-
|
|
113
|
+
expect(fetchMock).toHaveBeenCalledTimes(1);
|
|
114
|
+
const [calledUrl, calledInit] = fetchMock.mock.calls[0] ?? [];
|
|
115
|
+
expect(calledUrl).toBe('https://huggingface.co/docs/another');
|
|
116
|
+
expect(calledInit?.redirect).toBe('manual');
|
|
117
|
+
expect(new Headers(calledInit?.headers).get('accept')).toBe('text/markdown');
|
|
111
118
|
});
|
|
112
119
|
it('should return subsequent chunks with offset', async () => {
|
|
113
|
-
const longHtml = '<h1>Long Document</h1>' +
|
|
120
|
+
const longHtml = '<h1>Long Document</h1>' +
|
|
121
|
+
'<p>This is a very long sentence that will be repeated many times to create a document that exceeds the 7500 token limit for testing chunking functionality.</p>'.repeat(200);
|
|
114
122
|
stubFetch(() => createMockResponse({
|
|
115
123
|
content: longHtml,
|
|
116
124
|
}));
|
|
@@ -130,5 +138,42 @@ describe('DocFetchTool', () => {
|
|
|
130
138
|
expect(result).toContain('Error: Offset 10000 is beyond');
|
|
131
139
|
});
|
|
132
140
|
});
|
|
141
|
+
describe('security hardening', () => {
|
|
142
|
+
it('rejects traversal payload variants', async () => {
|
|
143
|
+
const traversalUrls = [
|
|
144
|
+
'https://huggingface.co/docs/../x',
|
|
145
|
+
'https://huggingface.co/docs/%2e%2e/x',
|
|
146
|
+
'https://huggingface.co/docs/%2e%2e%2fx',
|
|
147
|
+
'https://huggingface.co/docs/..%2fx',
|
|
148
|
+
'https://huggingface.co/docs/%2e%2e%5cx',
|
|
149
|
+
'https://huggingface.co/docs/%252e%252e%252fx',
|
|
150
|
+
];
|
|
151
|
+
const fetchMock = stubFetch(() => createMockResponse({
|
|
152
|
+
content: 'should never be fetched',
|
|
153
|
+
}));
|
|
154
|
+
for (const docUrl of traversalUrls) {
|
|
155
|
+
await expect(tool.fetch({ doc_url: docUrl })).rejects.toThrow('Failed to fetch document');
|
|
156
|
+
}
|
|
157
|
+
expect(fetchMock).not.toHaveBeenCalled();
|
|
158
|
+
});
|
|
159
|
+
it('rejects redirect to non-allowlisted host', async () => {
|
|
160
|
+
const fetchMock = vi.fn().mockResolvedValueOnce(new Response('', {
|
|
161
|
+
status: 302,
|
|
162
|
+
headers: { location: 'https://example.com/evil' },
|
|
163
|
+
}));
|
|
164
|
+
vi.stubGlobal('fetch', fetchMock);
|
|
165
|
+
await expect(tool.fetch({ doc_url: 'https://huggingface.co/docs/transformers' })).rejects.toThrow('Failed to fetch document');
|
|
166
|
+
expect(fetchMock).toHaveBeenCalledTimes(1);
|
|
167
|
+
});
|
|
168
|
+
it('rejects redirect to http downgrade', async () => {
|
|
169
|
+
const fetchMock = vi.fn().mockResolvedValueOnce(new Response('', {
|
|
170
|
+
status: 302,
|
|
171
|
+
headers: { location: 'http://huggingface.co/docs/transformers' },
|
|
172
|
+
}));
|
|
173
|
+
vi.stubGlobal('fetch', fetchMock);
|
|
174
|
+
await expect(tool.fetch({ doc_url: 'https://huggingface.co/docs/transformers' })).rejects.toThrow('Failed to fetch document');
|
|
175
|
+
expect(fetchMock).toHaveBeenCalledTimes(1);
|
|
176
|
+
});
|
|
177
|
+
});
|
|
133
178
|
});
|
|
134
179
|
//# sourceMappingURL=doc-fetch.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"doc-fetch.test.js","sourceRoot":"","sources":["../../src/docs-search/doc-fetch.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAC7D,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAE/D,MAAM,kBAAkB,GAAG,CAAC,EAC3B,OAAO,EACP,WAAW,GAAG,WAAW,EACzB,MAAM,GAAG,GAAG,EACZ,UAAU,GAAG,IAAI,GAMjB,EAAE,EAAE,CACJ,IAAI,QAAQ,CAAC,OAAO,EAAE;IACrB,MAAM;IACN,UAAU;IACV,OAAO,EAAE,EAAE,cAAc,EAAE,WAAW,EAAE;CACxC,CAAC,CAAC;AAEJ,MAAM,SAAS,GAAG,CAAC,OAAuB,EAAE,EAAE;IAC7C,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,
|
|
1
|
+
{"version":3,"file":"doc-fetch.test.js","sourceRoot":"","sources":["../../src/docs-search/doc-fetch.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAC7D,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAE/D,MAAM,kBAAkB,GAAG,CAAC,EAC3B,OAAO,EACP,WAAW,GAAG,WAAW,EACzB,MAAM,GAAG,GAAG,EACZ,UAAU,GAAG,IAAI,GAMjB,EAAE,EAAE,CACJ,IAAI,QAAQ,CAAC,OAAO,EAAE;IACrB,MAAM;IACN,UAAU;IACV,OAAO,EAAE,EAAE,cAAc,EAAE,WAAW,EAAE;CACxC,CAAC,CAAC;AAEJ,MAAM,SAAS,GAAG,CAAC,OAAuB,EAAE,EAAE;IAC7C,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAgB,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAC7F,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAClC,OAAO,SAAS,CAAC;AAClB,CAAC,CAAC;AAEF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC7B,MAAM,IAAI,GAAG,IAAI,YAAY,EAAE,CAAC;IAEhC,SAAS,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,gBAAgB,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACtD,MAAM,SAAS,GAAG;gBACjB,kDAAkD;gBAClD,0EAA0E;gBAC1E,yDAAyD;gBACzD,sEAAsE;gBACtE,yCAAyC;gBACzC,0CAA0C;gBAC1C,oBAAoB;gBACpB,+BAA+B;aAC/B,CAAC;YAEF,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;gBAC7B,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;YACnD,CAAC;QACF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACvE,MAAM,WAAW,GAAG;gBACnB,oCAAoC;gBACpC,6CAA6C;gBAC7C,yCAAyC;gBACzC,kCAAkC;gBAClC,iDAAiD;aACjD,CAAC;YAEF,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;gBAC/B,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,wCAAwC,CAAC,CAAC;YACvF,CAAC;QACF,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,QAAQ,GAAG,yBAAyB,CAAC;YAC3C,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,EAAE,CAChC,kBAAkB,CAAC;gBAClB,OAAO,EAAE,QAAQ;gBACjB,WAAW,EAAE,eAAe;aAC5B,CAAC,CACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,kCAAkC,EAAE,CAAC,CAAC;YACjF,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3C,MAAM,CAAC,SAAS,EAAE,UAAU,CAAC,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAC9D,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;YAC3D,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC5C,MAAM,CAAC,IAAI,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAC7E,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAE/D,SAAS,CAAC,GAAG,EAAE,CACd,kBAAkB,CAAC;gBAClB,OAAO,EAAE,yDAAyD;aAClE,CAAC,CACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,kCAAkC,EAAE,CAAC,CAAC;YAEjF,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YAEzE,MAAM,QAAQ,GACb,wBAAwB;gBACxB,iKAAiK,CAAC,MAAM,CACvK,GAAG,CACH,CAAC;YAEH,SAAS,CAAC,GAAG,EAAE,CACd,kBAAkB,CAAC;gBAClB,OAAO,EAAE,QAAQ;aACjB,CAAC,CACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,kCAAkC,EAAE,CAAC,CAAC;YAEjF,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qCAAqC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YAClE,MAAM,KAAK,GAAuC;gBACjD,EAAE,EAAE,EAAE,6BAA6B,EAAE,GAAG,EAAE,iCAAiC,EAAE;gBAC7E,EAAE,EAAE,EAAE,iCAAiC,EAAE,GAAG,EAAE,iCAAiC,EAAE;gBACjF,EAAE,EAAE,EAAE,0CAA0C,EAAE,GAAG,EAAE,0CAA0C,EAAE;gBACnG,EAAE,EAAE,EAAE,uBAAuB,EAAE,GAAG,EAAE,6CAA6C,EAAE;gBACnF,EAAE,EAAE,EAAE,wBAAwB,EAAE,GAAG,EAAE,6CAA6C,EAAE;gBACpF,EAAE,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,WAAW,EAAE;aACrC,CAAC;YACF,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;gBACvB,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YAC3C,CAAC;QACF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,EAAE,CAChC,kBAAkB,CAAC;gBAClB,OAAO,EAAE,2BAA2B;aACpC,CAAC,CACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;YAC3D,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3C,MAAM,CAAC,SAAS,EAAE,UAAU,CAAC,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAC9D,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;YAC3D,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC5C,MAAM,CAAC,IAAI,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAC7E,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,EAAE,CAChC,kBAAkB,CAAC;gBAClB,OAAO,EAAE,mCAAmC;aAC5C,CAAC,CACF,CAAC;YAEF,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC;YAChD,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3C,MAAM,CAAC,SAAS,EAAE,UAAU,CAAC,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAC9D,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;YAC9D,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC5C,MAAM,CAAC,IAAI,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC9E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAE5D,MAAM,QAAQ,GACb,wBAAwB;gBACxB,iKAAiK,CAAC,MAAM,CACvK,GAAG,CACH,CAAC;YAEH,SAAS,CAAC,GAAG,EAAE,CACd,kBAAkB,CAAC;gBAClB,OAAO,EAAE,QAAQ;aACjB,CAAC,CACF,CAAC;YAGF,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,kCAAkC,EAAE,CAAC,CAAC;YAGrF,MAAM,WAAW,GAAG,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;YACxD,MAAM,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;YACjC,MAAM,MAAM,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,EAAE,EAAE,CAAC,CAAC;YAGrD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,kCAAkC,EAAE,MAAM,EAAE,CAAC,CAAC;YAE9F,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC5D,SAAS,CAAC,GAAG,EAAE,CACd,kBAAkB,CAAC;gBAClB,OAAO,EAAE,8CAA8C;aACvD,CAAC,CACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,kCAAkC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YAEhG,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,+BAA+B,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,aAAa,GAAG;gBACrB,kCAAkC;gBAClC,sCAAsC;gBACtC,wCAAwC;gBACxC,oCAAoC;gBACpC,wCAAwC;gBACxC,8CAA8C;aAC9C,CAAC;YAEF,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,EAAE,CAChC,kBAAkB,CAAC;gBAClB,OAAO,EAAE,yBAAyB;aAClC,CAAC,CACF,CAAC;YAEF,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;gBACpC,MAAM,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAC;YAC3F,CAAC;YAED,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,qBAAqB,CAC9C,IAAI,QAAQ,CAAC,EAAE,EAAE;gBAChB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,EAAE,QAAQ,EAAE,0BAA0B,EAAE;aACjD,CAAC,CACF,CAAC;YACF,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,0CAA0C,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAChG,0BAA0B,CAC1B,CAAC;YAEF,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,qBAAqB,CAC9C,IAAI,QAAQ,CAAC,EAAE,EAAE;gBAChB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,EAAE,QAAQ,EAAE,yCAAyC,EAAE;aAChE,CAAC,CACF,CAAC;YACF,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAElC,MAAM,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,0CAA0C,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAChG,0BAA0B,CAC1B,CAAC;YAEF,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"file-icons.d.ts","sourceRoot":"","sources":["../src/file-icons.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,sBAAsB,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CA6BnE,CAAC;AAEF,wBAAgB,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAOpD"}
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
export const FILE_ICON_BY_EXTENSION = {
|
|
2
|
+
py: '🐍',
|
|
3
|
+
js: '📜',
|
|
4
|
+
ts: '📘',
|
|
5
|
+
md: '📝',
|
|
6
|
+
txt: '📄',
|
|
7
|
+
json: '📊',
|
|
8
|
+
yaml: '⚙️',
|
|
9
|
+
yml: '⚙️',
|
|
10
|
+
png: '🖼️',
|
|
11
|
+
jpg: '🖼️',
|
|
12
|
+
jpeg: '🖼️',
|
|
13
|
+
gif: '🖼️',
|
|
14
|
+
svg: '🎨',
|
|
15
|
+
mp4: '🎬',
|
|
16
|
+
mp3: '🎵',
|
|
17
|
+
pdf: '📕',
|
|
18
|
+
zip: '📦',
|
|
19
|
+
tar: '📦',
|
|
20
|
+
gz: '📦',
|
|
21
|
+
html: '🌐',
|
|
22
|
+
css: '🎨',
|
|
23
|
+
ipynb: '📓',
|
|
24
|
+
csv: '📊',
|
|
25
|
+
parquet: '🗄️',
|
|
26
|
+
safetensors: '🤖',
|
|
27
|
+
bin: '💾',
|
|
28
|
+
pkl: '🥒',
|
|
29
|
+
h5: '🗃️',
|
|
30
|
+
};
|
|
31
|
+
export function getFileIcon(filename) {
|
|
32
|
+
const extension = filename.split('.').pop()?.toLowerCase();
|
|
33
|
+
if (!extension) {
|
|
34
|
+
return '📄';
|
|
35
|
+
}
|
|
36
|
+
return FILE_ICON_BY_EXTENSION[extension] ?? '📄';
|
|
37
|
+
}
|
|
38
|
+
//# sourceMappingURL=file-icons.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"file-icons.js","sourceRoot":"","sources":["../src/file-icons.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,sBAAsB,GAAqC;IACvE,EAAE,EAAE,IAAI;IACR,EAAE,EAAE,IAAI;IACR,EAAE,EAAE,IAAI;IACR,EAAE,EAAE,IAAI;IACR,GAAG,EAAE,IAAI;IACT,IAAI,EAAE,IAAI;IACV,IAAI,EAAE,IAAI;IACV,GAAG,EAAE,IAAI;IACT,GAAG,EAAE,KAAK;IACV,GAAG,EAAE,KAAK;IACV,IAAI,EAAE,KAAK;IACX,GAAG,EAAE,KAAK;IACV,GAAG,EAAE,IAAI;IACT,GAAG,EAAE,IAAI;IACT,GAAG,EAAE,IAAI;IACT,GAAG,EAAE,IAAI;IACT,GAAG,EAAE,IAAI;IACT,GAAG,EAAE,IAAI;IACT,EAAE,EAAE,IAAI;IACR,IAAI,EAAE,IAAI;IACV,GAAG,EAAE,IAAI;IACT,KAAK,EAAE,IAAI;IACX,GAAG,EAAE,IAAI;IACT,OAAO,EAAE,KAAK;IACd,WAAW,EAAE,IAAI;IACjB,GAAG,EAAE,IAAI;IACT,GAAG,EAAE,IAAI;IACT,EAAE,EAAE,KAAK;CACT,CAAC;AAEF,MAAM,UAAU,WAAW,CAAC,QAAgB;IAC3C,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,WAAW,EAAE,CAAC;IAC3D,IAAI,CAAC,SAAS,EAAE,CAAC;QAChB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,OAAO,sBAAsB,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC;AAClD,CAAC"}
|
package/dist/gradio-files.d.ts
CHANGED
|
@@ -38,7 +38,6 @@ export declare class GradioFilesTool {
|
|
|
38
38
|
getGradioFiles(): Promise<FileWithUrl[]>;
|
|
39
39
|
private constructFileUrl;
|
|
40
40
|
generateDetailedMarkdown(fileType?: 'all' | 'image' | 'audio' | 'text'): Promise<string>;
|
|
41
|
-
private getFileIcon;
|
|
42
41
|
}
|
|
43
42
|
export {};
|
|
44
43
|
//# sourceMappingURL=gradio-files.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"gradio-files.d.ts","sourceRoot":"","sources":["../src/gradio-files.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;
|
|
1
|
+
{"version":3,"file":"gradio-files.d.ts","sourceRoot":"","sources":["../src/gradio-files.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAQxB,UAAU,WAAW;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS,CAAC;IACvC,GAAG,EAAE,MAAM,CAAC;IACZ,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,GAAG,EAAE,OAAO,CAAC;CACb;AAgFD,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;;CAa3B,CAAC;AAEX,eAAO,MAAM,0BAA0B;;;;CAItC,CAAC;AAGF,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,wBAAwB,CAAC,MAAM,CAAC,CAAC;AAKhF,qBAAa,eAAe;IAC3B,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;gBAEzB,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM;IAQvC,cAAc,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;IA0C9C,OAAO,CAAC,gBAAgB;IAOlB,wBAAwB,CAAC,QAAQ,GAAE,KAAK,GAAG,OAAO,GAAG,OAAO,GAAG,MAAc,GAAG,OAAO,CAAC,MAAM,CAAC;CAkCrG"}
|
package/dist/gradio-files.js
CHANGED
|
@@ -3,6 +3,7 @@ import { listFiles } from '@huggingface/hub';
|
|
|
3
3
|
import { formatBytes, escapeMarkdown } from './utilities.js';
|
|
4
4
|
import { HfApiError } from './hf-api-call.js';
|
|
5
5
|
import { explain } from './error-messages.js';
|
|
6
|
+
import { getFileIcon } from './file-icons.js';
|
|
6
7
|
const IMAGE_EXTENSIONS = new Set([
|
|
7
8
|
'.jpg',
|
|
8
9
|
'.jpeg',
|
|
@@ -154,45 +155,11 @@ export class GradioFilesTool {
|
|
|
154
155
|
markdown += `|------|------|------|---------------|-----|\n`;
|
|
155
156
|
for (const file of files) {
|
|
156
157
|
const fileName = file.path.split('/').pop() || file.path;
|
|
157
|
-
const icon =
|
|
158
|
+
const icon = getFileIcon(fileName);
|
|
158
159
|
const lastMod = file.lastModified ? new Date(file.lastModified).toLocaleDateString() : '-';
|
|
159
160
|
markdown += `| ${escapeMarkdown(fileName)} | ${file.sizeFormatted} | ${icon} ${file.type} | ${lastMod} | ${file.url} |\n`;
|
|
160
161
|
}
|
|
161
162
|
return markdown;
|
|
162
163
|
}
|
|
163
|
-
getFileIcon(filename) {
|
|
164
|
-
const ext = filename.split('.').pop()?.toLowerCase();
|
|
165
|
-
const iconMap = {
|
|
166
|
-
py: '🐍',
|
|
167
|
-
js: '📜',
|
|
168
|
-
ts: '📘',
|
|
169
|
-
md: '📝',
|
|
170
|
-
txt: '📄',
|
|
171
|
-
json: '📊',
|
|
172
|
-
yaml: '⚙️',
|
|
173
|
-
yml: '⚙️',
|
|
174
|
-
png: '🖼️',
|
|
175
|
-
jpg: '🖼️',
|
|
176
|
-
jpeg: '🖼️',
|
|
177
|
-
gif: '🖼️',
|
|
178
|
-
svg: '🎨',
|
|
179
|
-
mp4: '🎬',
|
|
180
|
-
mp3: '🎵',
|
|
181
|
-
pdf: '📕',
|
|
182
|
-
zip: '📦',
|
|
183
|
-
tar: '📦',
|
|
184
|
-
gz: '📦',
|
|
185
|
-
html: '🌐',
|
|
186
|
-
css: '🎨',
|
|
187
|
-
ipynb: '📓',
|
|
188
|
-
csv: '📊',
|
|
189
|
-
parquet: '🗄️',
|
|
190
|
-
safetensors: '🤖',
|
|
191
|
-
bin: '💾',
|
|
192
|
-
pkl: '🥒',
|
|
193
|
-
h5: '🗃️',
|
|
194
|
-
};
|
|
195
|
-
return iconMap[ext || ''] || '📄';
|
|
196
|
-
}
|
|
197
164
|
}
|
|
198
165
|
//# sourceMappingURL=gradio-files.js.map
|
package/dist/gradio-files.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"gradio-files.js","sourceRoot":"","sources":["../src/gradio-files.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAc9C,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC;IAChC,MAAM;IACN,OAAO;IACP,MAAM;IACN,MAAM;IACN,MAAM;IACN,OAAO;IACP,MAAM;IACN,OAAO;IACP,MAAM;IACN,MAAM;IACN,OAAO;IACP,OAAO;CACP,CAAC,CAAC;AAEH,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC;IAChC,MAAM;IACN,MAAM;IACN,OAAO;IACP,MAAM;IACN,MAAM;IACN,MAAM;IACN,MAAM;IACN,OAAO;IACP,OAAO;IACP,KAAK;IACL,KAAK;CACL,CAAC,CAAC;AAEH,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC;IAC/B,MAAM;IACN,KAAK;IACL,MAAM;IACN,KAAK;IACL,MAAM;IACN,MAAM;IACN,MAAM;IACN,MAAM;IACN,OAAO;IACP,MAAM;IACN,KAAK;IACL,KAAK;CACL,CAAC,CAAC;AAEH,SAAS,gBAAgB,CAAC,IAAY;IACrC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IACtC,OAAO,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;AACpE,CAAC;AAED,SAAS,WAAW,CAAC,IAAY;IAChC,OAAO,gBAAgB,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC;AACrD,CAAC;AAED,SAAS,WAAW,CAAC,IAAY;IAChC,OAAO,gBAAgB,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC;AACrD,CAAC;AAED,SAAS,UAAU,CAAC,IAAY;IAC/B,OAAO,eAAe,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC;AACpD,CAAC;AAED,SAAS,eAAe,CAAC,IAAiB,EAAE,QAA4C;IACvF,QAAQ,QAAQ,EAAE,CAAC;QAClB,KAAK,KAAK;YACT,OAAO,IAAI,CAAC;QACb,KAAK,OAAO;YACX,OAAO,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,KAAK,OAAO;YACX,OAAO,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,KAAK,MAAM;YACV,OAAO,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9B;YACC,OAAO,IAAI,CAAC;IACd,CAAC;AACF,CAAC;AAGD,MAAM,CAAC,MAAM,wBAAwB,GAAG;IACvC,IAAI,EAAE,cAAc;IACpB,WAAW,EACV,wIAAwI;IACzI,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,gBAAgB,CAAC;KACxG,CAAC;IACF,WAAW,EAAE;QACZ,KAAK,EAAE,mBAAmB;QAC1B,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACQ,CAAC;AAEX,MAAM,CAAC,MAAM,0BAA0B,GAAG;IACzC,IAAI,EAAE,8BAA8B;IACpC,WAAW,EAAE,sEAAsE;IACnF,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;CACpB,CAAC;AAQF,MAAM,OAAO,eAAe;IACV,WAAW,CAAS;IACpB,WAAW,CAAS;IAErC,YAAY,OAAe,EAAE,QAAgB;QAC5C,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC;QAC3B,IAAI,CAAC,WAAW,GAAG,GAAG,QAAQ,eAAe,CAAC;IAC/C,CAAC;IAKD,KAAK,CAAC,cAAc;QACnB,IAAI,CAAC;YACJ,MAAM,KAAK,GAAkB,EAAE,CAAC;YAGhC,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,SAAS,CAAC;gBAClC,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE;gBACjD,SAAS,EAAE,KAAK;gBAChB,MAAM,EAAE,IAAI;gBACZ,GAAG,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,WAAW,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;aAC3E,CAAC,EAAE,CAAC;gBACJ,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAE1B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC;oBACzD,IAAI,QAAQ,KAAK,gBAAgB,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;wBAChE,SAAS;oBACV,CAAC;oBAED,KAAK,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,GAAG,EAAE,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC;wBACrC,aAAa,EAAE,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;wBACrC,YAAY,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI;wBACnC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG;qBACf,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;YAED,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,KAAK,YAAY,UAAU,EAAE,CAAC;gBACjC,MAAM,OAAO,CAAC,KAAK,EAAE,qCAAqC,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC;YAChF,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAKO,gBAAgB,CAAC,QAAgB;QACxC,OAAO,mCAAmC,IAAI,CAAC,WAAW,iBAAiB,QAAQ,EAAE,CAAC;IACvF,CAAC;IAKD,KAAK,CAAC,wBAAwB,CAAC,WAA+C,KAAK;QAClF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC7C,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;QAEzE,IAAI,QAAQ,GAAG,uCAAuC,IAAI,CAAC,WAAW,MAAM,CAAC;QAC7E,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;YACxB,QAAQ,IAAI,eAAe,QAAQ,eAAe,CAAC;QACpD,CAAC;QAGD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;gBACxB,QAAQ,IAAI,MAAM,QAAQ,+BAA+B,CAAC;YAC3D,CAAC;iBAAM,CAAC;gBACP,QAAQ,IAAI,iCAAiC,CAAC;YAC/C,CAAC;YACD,OAAO,QAAQ,CAAC;QACjB,CAAC;QAGD,QAAQ,IAAI,kBAAkB,CAAC;QAC/B,QAAQ,IAAI,8DAA8D,CAAC;QAC3E,QAAQ,IAAI,gDAAgD,CAAC;QAE7D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC;YACzD,MAAM,IAAI,GAAG,
|
|
1
|
+
{"version":3,"file":"gradio-files.js","sourceRoot":"","sources":["../src/gradio-files.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAc9C,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC;IAChC,MAAM;IACN,OAAO;IACP,MAAM;IACN,MAAM;IACN,MAAM;IACN,OAAO;IACP,MAAM;IACN,OAAO;IACP,MAAM;IACN,MAAM;IACN,OAAO;IACP,OAAO;CACP,CAAC,CAAC;AAEH,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC;IAChC,MAAM;IACN,MAAM;IACN,OAAO;IACP,MAAM;IACN,MAAM;IACN,MAAM;IACN,MAAM;IACN,OAAO;IACP,OAAO;IACP,KAAK;IACL,KAAK;CACL,CAAC,CAAC;AAEH,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC;IAC/B,MAAM;IACN,KAAK;IACL,MAAM;IACN,KAAK;IACL,MAAM;IACN,MAAM;IACN,MAAM;IACN,MAAM;IACN,OAAO;IACP,MAAM;IACN,KAAK;IACL,KAAK;CACL,CAAC,CAAC;AAEH,SAAS,gBAAgB,CAAC,IAAY;IACrC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IACtC,OAAO,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;AACpE,CAAC;AAED,SAAS,WAAW,CAAC,IAAY;IAChC,OAAO,gBAAgB,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC;AACrD,CAAC;AAED,SAAS,WAAW,CAAC,IAAY;IAChC,OAAO,gBAAgB,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC;AACrD,CAAC;AAED,SAAS,UAAU,CAAC,IAAY;IAC/B,OAAO,eAAe,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC;AACpD,CAAC;AAED,SAAS,eAAe,CAAC,IAAiB,EAAE,QAA4C;IACvF,QAAQ,QAAQ,EAAE,CAAC;QAClB,KAAK,KAAK;YACT,OAAO,IAAI,CAAC;QACb,KAAK,OAAO;YACX,OAAO,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,KAAK,OAAO;YACX,OAAO,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,KAAK,MAAM;YACV,OAAO,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9B;YACC,OAAO,IAAI,CAAC;IACd,CAAC;AACF,CAAC;AAGD,MAAM,CAAC,MAAM,wBAAwB,GAAG;IACvC,IAAI,EAAE,cAAc;IACpB,WAAW,EACV,wIAAwI;IACzI,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,gBAAgB,CAAC;KACxG,CAAC;IACF,WAAW,EAAE;QACZ,KAAK,EAAE,mBAAmB;QAC1B,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACQ,CAAC;AAEX,MAAM,CAAC,MAAM,0BAA0B,GAAG;IACzC,IAAI,EAAE,8BAA8B;IACpC,WAAW,EAAE,sEAAsE;IACnF,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;CACpB,CAAC;AAQF,MAAM,OAAO,eAAe;IACV,WAAW,CAAS;IACpB,WAAW,CAAS;IAErC,YAAY,OAAe,EAAE,QAAgB;QAC5C,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC;QAC3B,IAAI,CAAC,WAAW,GAAG,GAAG,QAAQ,eAAe,CAAC;IAC/C,CAAC;IAKD,KAAK,CAAC,cAAc;QACnB,IAAI,CAAC;YACJ,MAAM,KAAK,GAAkB,EAAE,CAAC;YAGhC,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,SAAS,CAAC;gBAClC,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE;gBACjD,SAAS,EAAE,KAAK;gBAChB,MAAM,EAAE,IAAI;gBACZ,GAAG,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,WAAW,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;aAC3E,CAAC,EAAE,CAAC;gBACJ,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAE1B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC;oBACzD,IAAI,QAAQ,KAAK,gBAAgB,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;wBAChE,SAAS;oBACV,CAAC;oBAED,KAAK,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,GAAG,EAAE,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC;wBACrC,aAAa,EAAE,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;wBACrC,YAAY,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI;wBACnC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG;qBACf,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;YAED,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,KAAK,YAAY,UAAU,EAAE,CAAC;gBACjC,MAAM,OAAO,CAAC,KAAK,EAAE,qCAAqC,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC;YAChF,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAKO,gBAAgB,CAAC,QAAgB;QACxC,OAAO,mCAAmC,IAAI,CAAC,WAAW,iBAAiB,QAAQ,EAAE,CAAC;IACvF,CAAC;IAKD,KAAK,CAAC,wBAAwB,CAAC,WAA+C,KAAK;QAClF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC7C,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC;QAEzE,IAAI,QAAQ,GAAG,uCAAuC,IAAI,CAAC,WAAW,MAAM,CAAC;QAC7E,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;YACxB,QAAQ,IAAI,eAAe,QAAQ,eAAe,CAAC;QACpD,CAAC;QAGD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;gBACxB,QAAQ,IAAI,MAAM,QAAQ,+BAA+B,CAAC;YAC3D,CAAC;iBAAM,CAAC;gBACP,QAAQ,IAAI,iCAAiC,CAAC;YAC/C,CAAC;YACD,OAAO,QAAQ,CAAC;QACjB,CAAC;QAGD,QAAQ,IAAI,kBAAkB,CAAC;QAC/B,QAAQ,IAAI,8DAA8D,CAAC;QAC3E,QAAQ,IAAI,gDAAgD,CAAC;QAE7D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC;YACzD,MAAM,IAAI,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;YACnC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;YAE3F,QAAQ,IAAI,KAAK,cAAc,CAAC,QAAQ,CAAC,MAAM,IAAI,CAAC,aAAa,MAAM,IAAI,IAAI,IAAI,CAAC,IAAI,MAAM,OAAO,MAAM,IAAI,CAAC,GAAG,MAAM,CAAC;QAC3H,CAAC;QAED,OAAO,QAAQ,CAAC;IACjB,CAAC;CACD"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"hf-api-call.d.ts","sourceRoot":"","sources":["../src/hf-api-call.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"hf-api-call.d.ts","sourceRoot":"","sources":["../src/hf-api-call.ts"],"names":[],"mappings":"AAKA,qBAAa,UAAW,SAAQ,KAAK;aAGnB,MAAM,EAAE,MAAM;aACd,UAAU,EAAE,MAAM;aAClB,YAAY,CAAC,EAAE,MAAM;gBAHrC,OAAO,EAAE,MAAM,EACC,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,MAAM,EAClB,YAAY,CAAC,EAAE,MAAM,YAAA;IAYtC,qBAAqB,CAAC,mBAAmB,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,MAAM;IAuC5E,mBAAmB,CAAC,mBAAmB,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,UAAU;CAI9E;AAQD,qBAAa,SAAS,CAAC,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,EAAE,SAAS,GAAG,OAAO;IACvF,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IAClC,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,CAAC;IAC/C,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;gBAE1B,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM;cAe5B,YAAY,CAAC,CAAC,GAAG,SAAS,EAAE,GAAG,EAAE,GAAG,GAAG,MAAM,EAAE,OAAO,CAAC,EAAE,UAAU,CAAC,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC;IAwF5G,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,OAAO,GAAG,GAAG;cAsBxB,OAAO,CAAC,CAAC,GAAG,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,UAAU,CAAC,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC;CAIrG"}
|
package/dist/hf-api-call.js
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
1
|
+
import { fetchWithProfile, NETWORK_FETCH_PROFILES } from './network/fetch-profile.js';
|
|
1
2
|
export class HfApiError extends Error {
|
|
2
3
|
status;
|
|
3
4
|
statusText;
|
|
@@ -55,14 +56,13 @@ export class HfApiCall {
|
|
|
55
56
|
if (this.hfToken) {
|
|
56
57
|
headers['Authorization'] = `Bearer ${this.hfToken}`;
|
|
57
58
|
}
|
|
58
|
-
const
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
59
|
+
const { response } = await fetchWithProfile(url.toString(), NETWORK_FETCH_PROFILES.externalHttps(), {
|
|
60
|
+
timeoutMs: this.apiTimeout,
|
|
61
|
+
requestInit: {
|
|
62
|
+
...options,
|
|
63
|
+
headers,
|
|
64
|
+
},
|
|
64
65
|
});
|
|
65
|
-
clearTimeout(timeoutId);
|
|
66
66
|
const responseBodyText = await response.text();
|
|
67
67
|
if (!response.ok) {
|
|
68
68
|
const responseBody = responseBodyText || undefined;
|
package/dist/hf-api-call.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"hf-api-call.js","sourceRoot":"","sources":["../src/hf-api-call.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"hf-api-call.js","sourceRoot":"","sources":["../src/hf-api-call.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AAKtF,MAAM,OAAO,UAAW,SAAQ,KAAK;IAGnB;IACA;IACA;IAJjB,YACC,OAAe,EACC,MAAc,EACd,UAAkB,EAClB,YAAqB;QAErC,KAAK,CAAC,OAAO,CAAC,CAAC;QAJC,WAAM,GAAN,MAAM,CAAQ;QACd,eAAU,GAAV,UAAU,CAAQ;QAClB,iBAAY,GAAZ,YAAY,CAAS;QAGrC,IAAI,CAAC,IAAI,GAAG,YAAY,CAAC;IAC1B,CAAC;IAQD,qBAAqB,CAAC,mBAA2B,EAAE,OAAgB;QAClE,IAAI,SAAS,GAAG,EAAE,CAAC;QAGnB,IAAI,OAAO,EAAE,CAAC;YACb,SAAS,GAAG,GAAG,OAAO,IAAI,CAAC;QAC5B,CAAC;QAGD,SAAS,IAAI,mBAAmB,CAAC;QAGjC,SAAS,IAAI,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;QAGnC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,IAAI,CAAC;gBACJ,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAA0D,CAAC;gBACtG,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC;gBACpE,IAAI,WAAW,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;oBACxD,SAAS,IAAI,KAAK,WAAW,EAAE,CAAC;gBACjC,CAAC;YACF,CAAC;YAAC,MAAM,CAAC;gBAER,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;oBACjF,SAAS,IAAI,KAAK,IAAI,CAAC,YAAY,EAAE,CAAC;gBACvC,CAAC;YACF,CAAC;QACF,CAAC;QAED,OAAO,SAAS,CAAC;IAClB,CAAC;IAQD,mBAAmB,CAAC,mBAA2B,EAAE,OAAgB;QAChE,MAAM,eAAe,GAAG,IAAI,CAAC,qBAAqB,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;QACjF,OAAO,IAAI,UAAU,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;IACzF,CAAC;CACD;AAQD,MAAM,OAAO,SAAS;IACF,MAAM,CAAS;IACf,OAAO,CAAqB;IAC5B,UAAU,CAAS;IAEtC,YAAY,MAAc,EAAE,OAAgB;QAC3C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IACjG,CAAC;IAUS,KAAK,CAAC,YAAY,CAAgB,GAAiB,EAAE,OAAgC;QAC9F,IAAI,CAAC;YACJ,MAAM,OAAO,GAA2B;gBACvC,MAAM,EAAE,kBAAkB;gBAC1B,GAAG,CAAE,OAAO,EAAE,OAAkC,IAAI,EAAE,CAAC;aACvD,CAAC;YACF,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBAClB,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,IAAI,CAAC,OAAO,EAAE,CAAC;YACrD,CAAC;YAED,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,gBAAgB,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,sBAAsB,CAAC,aAAa,EAAE,EAAE;gBACnG,SAAS,EAAE,IAAI,CAAC,UAAU;gBAC1B,WAAW,EAAE;oBACZ,GAAG,OAAO;oBACV,OAAO;iBACP;aACD,CAAC,CAAC;YAEH,MAAM,gBAAgB,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAE/C,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBAElB,MAAM,YAAY,GAAuB,gBAAgB,IAAI,SAAS,CAAC;gBAGvE,OAAO,CAAC,KAAK,CAAC,eAAe,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;gBACvE,IAAI,YAAY,EAAE,CAAC;oBAClB,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,YAAY,CAAC,CAAC;gBACtD,CAAC;gBAED,MAAM,IAAI,UAAU,CACnB,uBAAuB,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,QAAQ,CAAC,UAAU,EAAE,EAC1E,QAAQ,CAAC,MAAM,EACf,QAAQ,CAAC,UAAU,EACnB,YAAY,CACZ,CAAC;YACH,CAAC;YAED,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;YAC/D,MAAM,WAAW,GAAG,gBAAgB,CAAC,IAAI,EAAE,CAAC;YAC5C,MAAM,aAAa,GAClB,WAAW,CAAC,QAAQ,CAAC,kBAAkB,CAAC,IAAI,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YAExG,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC9B,OAAO,SAAc,CAAC;YACvB,CAAC;YAED,IAAI,aAAa,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACJ,OAAO,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAM,CAAC;gBAC1C,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBAChB,MAAM,IAAI,KAAK,CACd,sDAAsD,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,GAAG,CACjH,CAAC;gBACH,CAAC;YACF,CAAC;YAED,IAAI,WAAW,CAAC,WAAW,EAAE,KAAK,IAAI,EAAE,CAAC;gBACxC,OAAO,SAAc,CAAC;YACvB,CAAC;YAED,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,WAAW,CAAC;YAC/F,MAAM,IAAI,KAAK,CACd,mDAAmD,WAAW,CAAC,CAAC,CAAC,mBAAmB,WAAW,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,aAAa,EAAE,CAC3H,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEhB,IAAI,KAAK,YAAY,UAAU,EAAE,CAAC;gBACjC,MAAM,KAAK,CAAC;YACb,CAAC;YAED,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC3D,MAAM,IAAI,KAAK,CAAC,+BAA+B,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;YACrE,CAAC;YAED,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC5B,MAAM,IAAI,KAAK,CAAC,uBAAuB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACzD,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAQS,QAAQ,CAAC,MAAe;QACjC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAGjC,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YAC1B,MAAM,KAAK,GAAG,MAAM,CAAC,GAAoB,CAAC,CAAC;YAC3C,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACzB,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAC7C,CAAC;QACF,CAAC;QAED,OAAO,GAAG,CAAC;IACZ,CAAC;IAUS,KAAK,CAAC,OAAO,CAAgB,MAAe,EAAE,OAAgC;QACvF,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAClC,OAAO,IAAI,CAAC,YAAY,CAAI,GAAG,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;CACD"}
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
interface BrowserToolConfig {
|
|
2
|
+
name: string;
|
|
3
|
+
description: string;
|
|
4
|
+
annotations: {
|
|
5
|
+
title: string;
|
|
6
|
+
destructiveHint: boolean;
|
|
7
|
+
readOnlyHint: boolean;
|
|
8
|
+
openWorldHint: boolean;
|
|
9
|
+
};
|
|
10
|
+
}
|
|
11
|
+
export declare const SEMANTIC_SEARCH_TOOL_CONFIG: BrowserToolConfig;
|
|
12
|
+
export declare const REPO_SEARCH_TOOL_CONFIG: BrowserToolConfig;
|
|
13
|
+
export declare const PAPER_SEARCH_TOOL_CONFIG: BrowserToolConfig;
|
|
14
|
+
export declare const HUB_REPO_DETAILS_TOOL_CONFIG: BrowserToolConfig;
|
|
15
|
+
export declare const DUPLICATE_SPACE_TOOL_CONFIG: BrowserToolConfig;
|
|
16
|
+
export declare const SPACE_FILES_TOOL_CONFIG: BrowserToolConfig;
|
|
17
|
+
export declare const DOCS_SEMANTIC_SEARCH_CONFIG: BrowserToolConfig;
|
|
18
|
+
export declare const DOC_FETCH_CONFIG: BrowserToolConfig;
|
|
19
|
+
export declare const SPACE_SEARCH_TOOL_ID: string;
|
|
20
|
+
export declare const MODEL_SEARCH_TOOL_ID = "model_search";
|
|
21
|
+
export declare const REPO_SEARCH_TOOL_ID: string;
|
|
22
|
+
export declare const MODEL_DETAIL_TOOL_ID = "model_details";
|
|
23
|
+
export declare const PAPER_SEARCH_TOOL_ID: string;
|
|
24
|
+
export declare const DATASET_SEARCH_TOOL_ID = "dataset_search";
|
|
25
|
+
export declare const DATASET_DETAIL_TOOL_ID = "dataset_details";
|
|
26
|
+
export declare const HUB_REPO_DETAILS_TOOL_ID: string;
|
|
27
|
+
export declare const DUPLICATE_SPACE_TOOL_ID: string;
|
|
28
|
+
export declare const SPACE_INFO_TOOL_ID = "space_info";
|
|
29
|
+
export declare const SPACE_FILES_TOOL_ID: string;
|
|
30
|
+
export declare const USE_SPACE_TOOL_ID = "use_space";
|
|
31
|
+
export declare const DOCS_SEMANTIC_SEARCH_TOOL_ID: string;
|
|
32
|
+
export declare const DOC_FETCH_TOOL_ID: string;
|
|
33
|
+
export declare const HF_JOBS_TOOL_ID = "hf_jobs";
|
|
34
|
+
export declare const DYNAMIC_SPACE_TOOL_ID = "dynamic_space";
|
|
35
|
+
export declare const ALL_BUILTIN_TOOL_IDS: readonly [string, "model_search", string, "model_details", string, "dataset_search", "dataset_details", string, string, "space_info", string, string, string, "use_space", "hf_jobs", "dynamic_space"];
|
|
36
|
+
export declare const TOOL_ID_GROUPS: {
|
|
37
|
+
readonly search: readonly [string, string, string, string];
|
|
38
|
+
readonly spaces: readonly [string, string, "space_info", string, "use_space"];
|
|
39
|
+
readonly detail: readonly ["model_details", "dataset_details", string];
|
|
40
|
+
readonly docs: readonly [string, string];
|
|
41
|
+
readonly hf_api: readonly [string, string, string, string, string];
|
|
42
|
+
readonly dynamic_space: readonly ["dynamic_space"];
|
|
43
|
+
readonly all: readonly [string, "model_search", string, "model_details", string, "dataset_search", "dataset_details", string, string, "space_info", string, string, string, "use_space", "hf_jobs", "dynamic_space"];
|
|
44
|
+
};
|
|
45
|
+
export type BuiltinToolId = (typeof ALL_BUILTIN_TOOL_IDS)[number];
|
|
46
|
+
export declare function isValidBuiltinToolId(toolId: string): toolId is BuiltinToolId;
|
|
47
|
+
export {};
|
|
48
|
+
//# sourceMappingURL=index.browser.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.browser.d.ts","sourceRoot":"","sources":["../src/index.browser.ts"],"names":[],"mappings":"AAOA,UAAU,iBAAiB;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE;QACZ,KAAK,EAAE,MAAM,CAAC;QACd,eAAe,EAAE,OAAO,CAAC;QACzB,YAAY,EAAE,OAAO,CAAC;QACtB,aAAa,EAAE,OAAO,CAAC;KACvB,CAAC;CACF;AAED,eAAO,MAAM,2BAA2B,EAAE,iBAWzC,CAAC;AAEF,eAAO,MAAM,uBAAuB,EAAE,iBAarC,CAAC;AAEF,eAAO,MAAM,wBAAwB,EAAE,iBAYtC,CAAC;AAEF,eAAO,MAAM,4BAA4B,EAAE,iBAW1C,CAAC;AAEF,eAAO,MAAM,2BAA2B,EAAE,iBASzC,CAAC;AAEF,eAAO,MAAM,uBAAuB,EAAE,iBASrC,CAAC;AAEF,eAAO,MAAM,2BAA2B,EAAE,iBAWzC,CAAC;AAEF,eAAO,MAAM,gBAAgB,EAAE,iBAU9B,CAAC;AAEF,eAAO,MAAM,oBAAoB,QAAmC,CAAC;AACrE,eAAO,MAAM,oBAAoB,iBAAiB,CAAC;AACnD,eAAO,MAAM,mBAAmB,QAA+B,CAAC;AAChE,eAAO,MAAM,oBAAoB,kBAAkB,CAAC;AACpD,eAAO,MAAM,oBAAoB,QAAgC,CAAC;AAClE,eAAO,MAAM,sBAAsB,mBAAmB,CAAC;AACvD,eAAO,MAAM,sBAAsB,oBAAoB,CAAC;AACxD,eAAO,MAAM,wBAAwB,QAAoC,CAAC;AAC1E,eAAO,MAAM,uBAAuB,QAAmC,CAAC;AACxE,eAAO,MAAM,kBAAkB,eAAe,CAAC;AAC/C,eAAO,MAAM,mBAAmB,QAA+B,CAAC;AAChE,eAAO,MAAM,iBAAiB,cAAc,CAAC;AAC7C,eAAO,MAAM,4BAA4B,QAAmC,CAAC;AAC7E,eAAO,MAAM,iBAAiB,QAAwB,CAAC;AACvD,eAAO,MAAM,eAAe,YAAY,CAAC;AACzC,eAAO,MAAM,qBAAqB,kBAAkB,CAAC;AAErD,eAAO,MAAM,oBAAoB,wMAiBvB,CAAC;AAEX,eAAO,MAAM,cAAc;;;;;;;;CAoBjB,CAAC;AAEX,MAAM,MAAM,aAAa,GAAG,CAAC,OAAO,oBAAoB,CAAC,CAAC,MAAM,CAAC,CAAC;AAElE,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,IAAI,aAAa,CAE5E"}
|