@llmindset/hf-mcp 0.3.10 → 0.3.12
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/create-repo.d.ts +47 -0
- package/dist/create-repo.d.ts.map +1 -0
- package/dist/create-repo.js +83 -0
- package/dist/create-repo.js.map +1 -0
- package/dist/create-repo.test.d.ts +2 -0
- package/dist/create-repo.test.d.ts.map +1 -0
- package/dist/create-repo.test.js +155 -0
- package/dist/create-repo.test.js.map +1 -0
- package/dist/dataset-detail.js +1 -1
- package/dist/dataset-detail.js.map +1 -1
- package/dist/dataset-viewer-inspect.d.ts +48 -0
- package/dist/dataset-viewer-inspect.d.ts.map +1 -0
- package/dist/dataset-viewer-inspect.js +660 -0
- package/dist/dataset-viewer-inspect.js.map +1 -0
- package/dist/dataset-viewer-inspect.test.d.ts +2 -0
- package/dist/dataset-viewer-inspect.test.d.ts.map +1 -0
- package/dist/dataset-viewer-inspect.test.js +218 -0
- package/dist/dataset-viewer-inspect.test.js.map +1 -0
- package/dist/gradio-files.d.ts +2 -2
- package/dist/hub-inspect.d.ts +19 -2
- package/dist/hub-inspect.d.ts.map +1 -1
- package/dist/hub-inspect.js +68 -4
- package/dist/hub-inspect.js.map +1 -1
- package/dist/hub-inspect.test.d.ts +2 -0
- package/dist/hub-inspect.test.d.ts.map +1 -0
- package/dist/hub-inspect.test.js +24 -0
- package/dist/hub-inspect.test.js.map +1 -0
- package/dist/index.browser.d.ts +5 -3
- package/dist/index.browser.d.ts.map +1 -1
- package/dist/index.browser.js +16 -1
- package/dist/index.browser.js.map +1 -1
- 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/types.d.ts +23 -23
- package/dist/jobs/types.d.ts.map +1 -1
- package/dist/jobs/types.js +7 -2
- package/dist/jobs/types.js.map +1 -1
- package/dist/model-detail.js +1 -1
- package/dist/model-detail.js.map +1 -1
- package/dist/readme-utils.d.ts +1 -1
- package/dist/readme-utils.d.ts.map +1 -1
- package/dist/readme-utils.js +2 -13
- package/dist/readme-utils.js.map +1 -1
- package/dist/repo-search.d.ts +2 -2
- package/dist/tool-ids.d.ts +4 -3
- package/dist/tool-ids.d.ts.map +1 -1
- package/dist/tool-ids.js +4 -1
- package/dist/tool-ids.js.map +1 -1
- package/package.json +1 -1
- package/src/create-repo.test.ts +182 -0
- package/src/create-repo.ts +103 -0
- package/src/dataset-detail.ts +1 -1
- package/src/dataset-viewer-inspect.test.ts +234 -0
- package/src/dataset-viewer-inspect.ts +809 -0
- package/src/hub-inspect.test.ts +28 -0
- package/src/hub-inspect.ts +88 -4
- package/src/index.browser.ts +18 -1
- package/src/index.ts +2 -0
- package/src/jobs/types.ts +10 -2
- package/src/model-detail.ts +1 -1
- package/src/readme-utils.ts +2 -32
- package/src/tool-ids.ts +4 -0
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
declare const REPO_TYPES: readonly ["model", "dataset", "space", "bucket"];
|
|
3
|
+
export declare const CREATE_REPO_TOOL_CONFIG: {
|
|
4
|
+
readonly name: "create_repo";
|
|
5
|
+
readonly description: "";
|
|
6
|
+
readonly schema: z.ZodObject<{
|
|
7
|
+
name: z.ZodString;
|
|
8
|
+
repo_type: z.ZodDefault<z.ZodOptional<z.ZodEnum<["model", "dataset", "space", "bucket"]>>>;
|
|
9
|
+
private: z.ZodOptional<z.ZodBoolean>;
|
|
10
|
+
sdk: z.ZodDefault<z.ZodOptional<z.ZodEnum<["streamlit", "gradio", "docker", "static"]>>>;
|
|
11
|
+
}, "strip", z.ZodTypeAny, {
|
|
12
|
+
name: string;
|
|
13
|
+
repo_type: "space" | "dataset" | "model" | "bucket";
|
|
14
|
+
sdk: "streamlit" | "gradio" | "docker" | "static";
|
|
15
|
+
private?: boolean | undefined;
|
|
16
|
+
}, {
|
|
17
|
+
name: string;
|
|
18
|
+
repo_type?: "space" | "dataset" | "model" | "bucket" | undefined;
|
|
19
|
+
private?: boolean | undefined;
|
|
20
|
+
sdk?: "streamlit" | "gradio" | "docker" | "static" | undefined;
|
|
21
|
+
}>;
|
|
22
|
+
readonly annotations: {
|
|
23
|
+
readonly title: "Create Hugging Face Repository";
|
|
24
|
+
readonly destructiveHint: false;
|
|
25
|
+
readonly readOnlyHint: false;
|
|
26
|
+
readonly openWorldHint: true;
|
|
27
|
+
};
|
|
28
|
+
};
|
|
29
|
+
export type CreateRepoParams = z.input<typeof CREATE_REPO_TOOL_CONFIG.schema>;
|
|
30
|
+
type RepoType = (typeof REPO_TYPES)[number];
|
|
31
|
+
export interface CreateRepoResult {
|
|
32
|
+
url: string;
|
|
33
|
+
name: string;
|
|
34
|
+
repoType: RepoType;
|
|
35
|
+
id: string;
|
|
36
|
+
}
|
|
37
|
+
export declare class CreateRepoTool {
|
|
38
|
+
private readonly hfToken?;
|
|
39
|
+
constructor(hfToken?: string | undefined);
|
|
40
|
+
static createToolConfig(): Omit<typeof CREATE_REPO_TOOL_CONFIG, 'description'> & {
|
|
41
|
+
description: string;
|
|
42
|
+
};
|
|
43
|
+
create(params: CreateRepoParams): Promise<CreateRepoResult>;
|
|
44
|
+
}
|
|
45
|
+
export declare const formatCreateRepoResult: (result: CreateRepoResult) => string;
|
|
46
|
+
export {};
|
|
47
|
+
//# sourceMappingURL=create-repo.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"create-repo.d.ts","sourceRoot":"","sources":["../src/create-repo.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,QAAA,MAAM,UAAU,kDAAoF,CAAC;AAGrG,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;;CAe1B,CAAC;AAEX,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,MAAM,CAAC,CAAC;AAC9E,KAAK,QAAQ,GAAG,CAAC,OAAO,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC;AAY5C,MAAM,WAAW,gBAAgB;IAChC,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,QAAQ,CAAC;IACnB,EAAE,EAAE,MAAM,CAAC;CACX;AAED,qBAAa,cAAc;IACd,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAAR,OAAO,CAAC,EAAE,MAAM,YAAA;IAE7C,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,OAAO,uBAAuB,EAAE,aAAa,CAAC,GAAG;QAAE,WAAW,EAAE,MAAM,CAAA;KAAE;IAOlG,MAAM,CAAC,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAuBjE;AAiBD,eAAO,MAAM,sBAAsB,GAAI,QAAQ,gBAAgB,KAAG,MAQjE,CAAC"}
|
|
@@ -0,0 +1,83 @@
|
|
|
1
|
+
import { createRepo } from '@huggingface/hub';
|
|
2
|
+
import { z } from 'zod';
|
|
3
|
+
import { NO_TOKEN_INSTRUCTIONS } from './utilities.js';
|
|
4
|
+
const REPO_TYPES = ['model', 'dataset', 'space', 'bucket'];
|
|
5
|
+
const SPACE_SDKS = ['streamlit', 'gradio', 'docker', 'static'];
|
|
6
|
+
export const CREATE_REPO_TOOL_CONFIG = {
|
|
7
|
+
name: 'create_repo',
|
|
8
|
+
description: '',
|
|
9
|
+
schema: z.object({
|
|
10
|
+
name: z.string().min(1).describe("Fully-qualified repository name in 'namespace/repo-name' format."),
|
|
11
|
+
repo_type: z.enum(REPO_TYPES).optional().default('bucket').describe('Repository type. Defaults to bucket.'),
|
|
12
|
+
private: z.boolean().optional().describe('Whether to create the repository as private.'),
|
|
13
|
+
sdk: z.enum(SPACE_SDKS).optional().default('static').describe("SDK type - only required for repo_type='space'."),
|
|
14
|
+
}),
|
|
15
|
+
annotations: {
|
|
16
|
+
title: 'Create Hugging Face Repository',
|
|
17
|
+
destructiveHint: false,
|
|
18
|
+
readOnlyHint: false,
|
|
19
|
+
openWorldHint: true,
|
|
20
|
+
},
|
|
21
|
+
};
|
|
22
|
+
function assertExhaustiveUnion(_value) {
|
|
23
|
+
void _value;
|
|
24
|
+
}
|
|
25
|
+
assertExhaustiveUnion();
|
|
26
|
+
assertExhaustiveUnion();
|
|
27
|
+
assertExhaustiveUnion();
|
|
28
|
+
assertExhaustiveUnion();
|
|
29
|
+
export class CreateRepoTool {
|
|
30
|
+
hfToken;
|
|
31
|
+
constructor(hfToken) {
|
|
32
|
+
this.hfToken = hfToken;
|
|
33
|
+
}
|
|
34
|
+
static createToolConfig() {
|
|
35
|
+
return {
|
|
36
|
+
...CREATE_REPO_TOOL_CONFIG,
|
|
37
|
+
description: 'Create a Hugging Face model, dataset, space, or bucket repository.',
|
|
38
|
+
};
|
|
39
|
+
}
|
|
40
|
+
async create(params) {
|
|
41
|
+
if (!this.hfToken)
|
|
42
|
+
throw new Error(NO_TOKEN_INSTRUCTIONS);
|
|
43
|
+
const repoType = params.repo_type ?? 'model';
|
|
44
|
+
validateParams(params, repoType);
|
|
45
|
+
const result = await createRepo({
|
|
46
|
+
accessToken: this.hfToken,
|
|
47
|
+
repo: {
|
|
48
|
+
name: params.name,
|
|
49
|
+
type: repoType,
|
|
50
|
+
},
|
|
51
|
+
private: params.private,
|
|
52
|
+
...(repoType === 'space' ? { sdk: params.sdk } : {}),
|
|
53
|
+
});
|
|
54
|
+
return {
|
|
55
|
+
url: result.repoUrl,
|
|
56
|
+
name: params.name,
|
|
57
|
+
repoType,
|
|
58
|
+
id: result.id,
|
|
59
|
+
};
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
function validateParams(params, repoType) {
|
|
63
|
+
if (!isFullyQualifiedRepoName(params.name)) {
|
|
64
|
+
throw new Error("name must be fully qualified in 'namespace/repo-name' format.");
|
|
65
|
+
}
|
|
66
|
+
if (repoType === 'space' && !params.sdk) {
|
|
67
|
+
throw new Error('sdk is required when repo_type is space.');
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
function isFullyQualifiedRepoName(name) {
|
|
71
|
+
const parts = name.split('/');
|
|
72
|
+
return parts.length === 2 && parts.every((part) => part.length > 0);
|
|
73
|
+
}
|
|
74
|
+
export const formatCreateRepoResult = (result) => {
|
|
75
|
+
return [
|
|
76
|
+
'Repository created.',
|
|
77
|
+
`Name: ${result.name}`,
|
|
78
|
+
`Type: ${result.repoType}`,
|
|
79
|
+
`URL: ${result.url}`,
|
|
80
|
+
`ID: ${result.id}`,
|
|
81
|
+
].join('\n');
|
|
82
|
+
};
|
|
83
|
+
//# sourceMappingURL=create-repo.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"create-repo.js","sourceRoot":"","sources":["../src/create-repo.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAA+C,MAAM,kBAAkB,CAAC;AAC3F,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AAEvD,MAAM,UAAU,GAAG,CAAC,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,CAA2C,CAAC;AACrG,MAAM,UAAU,GAAG,CAAC,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAwC,CAAC;AAEtG,MAAM,CAAC,MAAM,uBAAuB,GAAG;IACtC,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,EAAE;IACf,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,kEAAkE,CAAC;QACpG,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,sCAAsC,CAAC;QAC3G,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8CAA8C,CAAC;QACxF,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,iDAAiD,CAAC;KAChH,CAAC;IACF,WAAW,EAAE;QACZ,KAAK,EAAE,gCAAgC;QACvC,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,KAAK;QACnB,aAAa,EAAE,IAAI;KACnB;CACQ,CAAC;AAMX,SAAS,qBAAqB,CAAkB,MAAU;IACzD,KAAK,MAAM,CAAC;AACb,CAAC;AAED,qBAAqB,EAAkC,CAAC;AACxD,qBAAqB,EAAkC,CAAC;AACxD,qBAAqB,EAAwC,CAAC;AAC9D,qBAAqB,EAAwC,CAAC;AAS9D,MAAM,OAAO,cAAc;IACG;IAA7B,YAA6B,OAAgB;QAAhB,YAAO,GAAP,OAAO,CAAS;IAAG,CAAC;IAEjD,MAAM,CAAC,gBAAgB;QACtB,OAAO;YACN,GAAG,uBAAuB;YAC1B,WAAW,EAAE,oEAAoE;SACjF,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAwB;QACpC,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAE1D,MAAM,QAAQ,GAAG,MAAM,CAAC,SAAS,IAAI,OAAO,CAAC;QAC7C,cAAc,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAEjC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC;YAC/B,WAAW,EAAE,IAAI,CAAC,OAAO;YACzB,IAAI,EAAE;gBACL,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,IAAI,EAAE,QAAQ;aACd;YACD,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,GAAG,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACpD,CAAC,CAAC;QAEH,OAAO;YACN,GAAG,EAAE,MAAM,CAAC,OAAO;YACnB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,QAAQ;YACR,EAAE,EAAE,MAAM,CAAC,EAAE;SACb,CAAC;IACH,CAAC;CACD;AAED,SAAS,cAAc,CAAC,MAAwB,EAAE,QAAkB;IACnE,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5C,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;IAClF,CAAC;IAED,IAAI,QAAQ,KAAK,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;QACzC,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;IAC7D,CAAC;AACF,CAAC;AAED,SAAS,wBAAwB,CAAC,IAAY;IAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC9B,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACrE,CAAC;AAED,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,MAAwB,EAAU,EAAE;IAC1E,OAAO;QACN,qBAAqB;QACrB,SAAS,MAAM,CAAC,IAAI,EAAE;QACtB,SAAS,MAAM,CAAC,QAAQ,EAAE;QAC1B,QAAQ,MAAM,CAAC,GAAG,EAAE;QACpB,OAAO,MAAM,CAAC,EAAE,EAAE;KAClB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACd,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"create-repo.test.d.ts","sourceRoot":"","sources":["../src/create-repo.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,155 @@
|
|
|
1
|
+
import { createRepo } from '@huggingface/hub';
|
|
2
|
+
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
|
3
|
+
import { CreateRepoTool, formatCreateRepoResult } from './create-repo.js';
|
|
4
|
+
vi.mock('@huggingface/hub', () => ({
|
|
5
|
+
createRepo: vi.fn(),
|
|
6
|
+
}));
|
|
7
|
+
describe('CreateRepoTool', () => {
|
|
8
|
+
beforeEach(() => {
|
|
9
|
+
vi.mocked(createRepo).mockReset();
|
|
10
|
+
vi.mocked(createRepo).mockResolvedValue({
|
|
11
|
+
repoUrl: 'https://huggingface.co/alice/example-model',
|
|
12
|
+
id: '0123456789abcdef01234567',
|
|
13
|
+
});
|
|
14
|
+
});
|
|
15
|
+
it('creates a model repository with client-native repo params', async () => {
|
|
16
|
+
const tool = new CreateRepoTool('token');
|
|
17
|
+
const result = await tool.create({
|
|
18
|
+
name: 'alice/example-model',
|
|
19
|
+
repo_type: 'model',
|
|
20
|
+
});
|
|
21
|
+
expect(createRepo).toHaveBeenCalledWith({
|
|
22
|
+
accessToken: 'token',
|
|
23
|
+
repo: { name: 'alice/example-model', type: 'model' },
|
|
24
|
+
private: undefined,
|
|
25
|
+
});
|
|
26
|
+
expect(result).toEqual({
|
|
27
|
+
url: 'https://huggingface.co/alice/example-model',
|
|
28
|
+
name: 'alice/example-model',
|
|
29
|
+
repoType: 'model',
|
|
30
|
+
id: '0123456789abcdef01234567',
|
|
31
|
+
});
|
|
32
|
+
});
|
|
33
|
+
it('defaults to model repositories', async () => {
|
|
34
|
+
const tool = new CreateRepoTool('token');
|
|
35
|
+
await tool.create({
|
|
36
|
+
name: 'alice/default-model',
|
|
37
|
+
});
|
|
38
|
+
expect(createRepo).toHaveBeenCalledWith({
|
|
39
|
+
accessToken: 'token',
|
|
40
|
+
repo: { name: 'alice/default-model', type: 'model' },
|
|
41
|
+
private: undefined,
|
|
42
|
+
});
|
|
43
|
+
});
|
|
44
|
+
it('creates a dataset repository', async () => {
|
|
45
|
+
const tool = new CreateRepoTool('token');
|
|
46
|
+
await tool.create({
|
|
47
|
+
name: 'alice/example-dataset',
|
|
48
|
+
repo_type: 'dataset',
|
|
49
|
+
private: true,
|
|
50
|
+
});
|
|
51
|
+
expect(createRepo).toHaveBeenCalledWith({
|
|
52
|
+
accessToken: 'token',
|
|
53
|
+
repo: { name: 'alice/example-dataset', type: 'dataset' },
|
|
54
|
+
private: true,
|
|
55
|
+
});
|
|
56
|
+
});
|
|
57
|
+
it('creates a bucket repository', async () => {
|
|
58
|
+
vi.mocked(createRepo).mockResolvedValue({
|
|
59
|
+
repoUrl: 'https://huggingface.co/buckets/alice/example-bucket',
|
|
60
|
+
id: 'bucket-id',
|
|
61
|
+
});
|
|
62
|
+
const tool = new CreateRepoTool('token');
|
|
63
|
+
const result = await tool.create({
|
|
64
|
+
name: 'alice/example-bucket',
|
|
65
|
+
repo_type: 'bucket',
|
|
66
|
+
private: true,
|
|
67
|
+
});
|
|
68
|
+
expect(createRepo).toHaveBeenCalledWith({
|
|
69
|
+
accessToken: 'token',
|
|
70
|
+
repo: { name: 'alice/example-bucket', type: 'bucket' },
|
|
71
|
+
private: true,
|
|
72
|
+
});
|
|
73
|
+
expect(result).toEqual({
|
|
74
|
+
url: 'https://huggingface.co/buckets/alice/example-bucket',
|
|
75
|
+
name: 'alice/example-bucket',
|
|
76
|
+
repoType: 'bucket',
|
|
77
|
+
id: 'bucket-id',
|
|
78
|
+
});
|
|
79
|
+
});
|
|
80
|
+
it('creates a Space repository with sdk', async () => {
|
|
81
|
+
vi.mocked(createRepo).mockResolvedValue({
|
|
82
|
+
repoUrl: 'https://huggingface.co/spaces/alice/demo',
|
|
83
|
+
id: 'abcdefabcdefabcdefabcdef',
|
|
84
|
+
});
|
|
85
|
+
const tool = new CreateRepoTool('token');
|
|
86
|
+
await tool.create({
|
|
87
|
+
name: 'alice/demo',
|
|
88
|
+
repo_type: 'space',
|
|
89
|
+
sdk: 'gradio',
|
|
90
|
+
});
|
|
91
|
+
expect(createRepo).toHaveBeenCalledWith({
|
|
92
|
+
accessToken: 'token',
|
|
93
|
+
repo: { name: 'alice/demo', type: 'space' },
|
|
94
|
+
private: undefined,
|
|
95
|
+
sdk: 'gradio',
|
|
96
|
+
});
|
|
97
|
+
});
|
|
98
|
+
it('creates a Streamlit Space repository', async () => {
|
|
99
|
+
vi.mocked(createRepo).mockResolvedValue({
|
|
100
|
+
repoUrl: 'https://huggingface.co/spaces/alice/streamlit-demo',
|
|
101
|
+
id: 'streamlit-space-id',
|
|
102
|
+
});
|
|
103
|
+
const tool = new CreateRepoTool('token');
|
|
104
|
+
await tool.create({
|
|
105
|
+
name: 'alice/streamlit-demo',
|
|
106
|
+
repo_type: 'space',
|
|
107
|
+
sdk: 'streamlit',
|
|
108
|
+
});
|
|
109
|
+
expect(createRepo).toHaveBeenCalledWith({
|
|
110
|
+
accessToken: 'token',
|
|
111
|
+
repo: { name: 'alice/streamlit-demo', type: 'space' },
|
|
112
|
+
private: undefined,
|
|
113
|
+
sdk: 'streamlit',
|
|
114
|
+
});
|
|
115
|
+
});
|
|
116
|
+
it('requires sdk for Space repositories', async () => {
|
|
117
|
+
const tool = new CreateRepoTool('token');
|
|
118
|
+
await expect(tool.create({
|
|
119
|
+
name: 'alice/demo',
|
|
120
|
+
repo_type: 'space',
|
|
121
|
+
})).rejects.toThrow('sdk is required when repo_type is space');
|
|
122
|
+
expect(createRepo).not.toHaveBeenCalled();
|
|
123
|
+
});
|
|
124
|
+
it('requires fully-qualified repo names', async () => {
|
|
125
|
+
const tool = new CreateRepoTool('token');
|
|
126
|
+
await expect(tool.create({
|
|
127
|
+
name: 'example-model',
|
|
128
|
+
repo_type: 'model',
|
|
129
|
+
})).rejects.toThrow("name must be fully qualified in 'namespace/repo-name' format");
|
|
130
|
+
expect(createRepo).not.toHaveBeenCalled();
|
|
131
|
+
});
|
|
132
|
+
it('requires an auth token', async () => {
|
|
133
|
+
const tool = new CreateRepoTool(undefined);
|
|
134
|
+
await expect(tool.create({
|
|
135
|
+
name: 'alice/example-model',
|
|
136
|
+
repo_type: 'model',
|
|
137
|
+
})).rejects.toThrow('Requires Authentication');
|
|
138
|
+
expect(createRepo).not.toHaveBeenCalled();
|
|
139
|
+
});
|
|
140
|
+
it('formats the created repository result', () => {
|
|
141
|
+
expect(formatCreateRepoResult({
|
|
142
|
+
url: 'https://huggingface.co/alice/example-model',
|
|
143
|
+
name: 'alice/example-model',
|
|
144
|
+
repoType: 'model',
|
|
145
|
+
id: '0123456789abcdef01234567',
|
|
146
|
+
})).toBe([
|
|
147
|
+
'Repository created.',
|
|
148
|
+
'Name: alice/example-model',
|
|
149
|
+
'Type: model',
|
|
150
|
+
'URL: https://huggingface.co/alice/example-model',
|
|
151
|
+
'ID: 0123456789abcdef01234567',
|
|
152
|
+
].join('\n'));
|
|
153
|
+
});
|
|
154
|
+
});
|
|
155
|
+
//# sourceMappingURL=create-repo.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"create-repo.test.js","sourceRoot":"","sources":["../src/create-repo.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,cAAc,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAE1E,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;CACnB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC/B,UAAU,CAAC,GAAG,EAAE;QACf,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,EAAE,CAAC;QAClC,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC;YACvC,OAAO,EAAE,4CAA4C;YACrD,EAAE,EAAE,0BAA0B;SAC9B,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC;YAChC,IAAI,EAAE,qBAAqB;YAC3B,SAAS,EAAE,OAAO;SAClB,CAAC,CAAC;QAEH,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACvC,WAAW,EAAE,OAAO;YACpB,IAAI,EAAE,EAAE,IAAI,EAAE,qBAAqB,EAAE,IAAI,EAAE,OAAO,EAAE;YACpD,OAAO,EAAE,SAAS;SAClB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACtB,GAAG,EAAE,4CAA4C;YACjD,IAAI,EAAE,qBAAqB;YAC3B,QAAQ,EAAE,OAAO;YACjB,EAAE,EAAE,0BAA0B;SAC9B,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,IAAI,CAAC,MAAM,CAAC;YACjB,IAAI,EAAE,qBAAqB;SAC3B,CAAC,CAAC;QAEH,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACvC,WAAW,EAAE,OAAO;YACpB,IAAI,EAAE,EAAE,IAAI,EAAE,qBAAqB,EAAE,IAAI,EAAE,OAAO,EAAE;YACpD,OAAO,EAAE,SAAS;SAClB,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,IAAI,CAAC,MAAM,CAAC;YACjB,IAAI,EAAE,uBAAuB;YAC7B,SAAS,EAAE,SAAS;YACpB,OAAO,EAAE,IAAI;SACb,CAAC,CAAC;QAEH,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACvC,WAAW,EAAE,OAAO;YACpB,IAAI,EAAE,EAAE,IAAI,EAAE,uBAAuB,EAAE,IAAI,EAAE,SAAS,EAAE;YACxD,OAAO,EAAE,IAAI;SACb,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC5C,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC;YACvC,OAAO,EAAE,qDAAqD;YAC9D,EAAE,EAAE,WAAW;SACf,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC;YAChC,IAAI,EAAE,sBAAsB;YAC5B,SAAS,EAAE,QAAQ;YACnB,OAAO,EAAE,IAAI;SACb,CAAC,CAAC;QAEH,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACvC,WAAW,EAAE,OAAO;YACpB,IAAI,EAAE,EAAE,IAAI,EAAE,sBAAsB,EAAE,IAAI,EAAE,QAAQ,EAAE;YACtD,OAAO,EAAE,IAAI;SACb,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACtB,GAAG,EAAE,qDAAqD;YAC1D,IAAI,EAAE,sBAAsB;YAC5B,QAAQ,EAAE,QAAQ;YAClB,EAAE,EAAE,WAAW;SACf,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACpD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC;YACvC,OAAO,EAAE,0CAA0C;YACnD,EAAE,EAAE,0BAA0B;SAC9B,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,IAAI,CAAC,MAAM,CAAC;YACjB,IAAI,EAAE,YAAY;YAClB,SAAS,EAAE,OAAO;YAClB,GAAG,EAAE,QAAQ;SACb,CAAC,CAAC;QAEH,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACvC,WAAW,EAAE,OAAO;YACpB,IAAI,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE;YAC3C,OAAO,EAAE,SAAS;YAClB,GAAG,EAAE,QAAQ;SACb,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACrD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,iBAAiB,CAAC;YACvC,OAAO,EAAE,oDAAoD;YAC7D,EAAE,EAAE,oBAAoB;SACxB,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,IAAI,CAAC,MAAM,CAAC;YACjB,IAAI,EAAE,sBAAsB;YAC5B,SAAS,EAAE,OAAO;YAClB,GAAG,EAAE,WAAW;SAChB,CAAC,CAAC;QAEH,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC;YACvC,WAAW,EAAE,OAAO;YACpB,IAAI,EAAE,EAAE,IAAI,EAAE,sBAAsB,EAAE,IAAI,EAAE,OAAO,EAAE;YACrD,OAAO,EAAE,SAAS;YAClB,GAAG,EAAE,WAAW;SAChB,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,MAAM,CACX,IAAI,CAAC,MAAM,CAAC;YACX,IAAI,EAAE,YAAY;YAClB,SAAS,EAAE,OAAO;SAClB,CAAC,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC;QAC7D,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,MAAM,CACX,IAAI,CAAC,MAAM,CAAC;YACX,IAAI,EAAE,eAAe;YACrB,SAAS,EAAE,OAAO;SAClB,CAAC,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,8DAA8D,CAAC,CAAC;QAClF,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC,SAAS,CAAC,CAAC;QAC3C,MAAM,MAAM,CACX,IAAI,CAAC,MAAM,CAAC;YACX,IAAI,EAAE,qBAAqB;YAC3B,SAAS,EAAE,OAAO;SAClB,CAAC,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAC7C,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAChD,MAAM,CACL,sBAAsB,CAAC;YACtB,GAAG,EAAE,4CAA4C;YACjD,IAAI,EAAE,qBAAqB;YAC3B,QAAQ,EAAE,OAAO;YACjB,EAAE,EAAE,0BAA0B;SAC9B,CAAC,CACF,CAAC,IAAI,CACL;YACC,qBAAqB;YACrB,2BAA2B;YAC3B,aAAa;YACb,iDAAiD;YACjD,8BAA8B;SAC9B,CAAC,IAAI,CAAC,IAAI,CAAC,CACZ,CAAC;IACH,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}
|
package/dist/dataset-detail.js
CHANGED
|
@@ -81,7 +81,7 @@ export class DatasetDetailTool {
|
|
|
81
81
|
}
|
|
82
82
|
}
|
|
83
83
|
if (includeReadme) {
|
|
84
|
-
const readmeContent = await fetchReadmeContent(datasetDetails.name, 'datasets'
|
|
84
|
+
const readmeContent = await fetchReadmeContent(datasetDetails.name, 'datasets');
|
|
85
85
|
if (readmeContent) {
|
|
86
86
|
const result = formatDatasetDetails(datasetDetails);
|
|
87
87
|
result.formatted +=
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"dataset-detail.js","sourceRoot":"","sources":["../src/dataset-detail.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAGvD,MAAM,CAAC,MAAM,0BAA0B,GAAG;IACzC,IAAI,EAAE,iBAAiB;IACvB,WAAW,EAAE,wEAAwE;IACrF,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,wBAAwB,CAAC,CAAC,QAAQ,CAAC,gDAAgD,CAAC;KAClH,CAAC;IACF,WAAW,EAAE;QACZ,KAAK,EAAE,iBAAiB;QACxB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,KAAK;KACpB;CACQ,CAAC;AAEX,MAAM,CAAC,MAAM,4BAA4B,GAAG;IAC3C,IAAI,EAAE,iBAAiB;IACvB,KAAK,EAAE,iBAAiB;IACxB,WAAW,EACV,8HAA8H;IAC/H,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,UAAU,EAAE,CAAC;aACX,MAAM,EAAE;aACR,GAAG,CAAC,CAAC,EAAE,wBAAwB,CAAC;aAChC,GAAG,CAAC,GAAG,CAAC;aACR,QAAQ,CAAC,oDAAoD,CAAC;KAChE,CAAC;CACF,CAAC;AA2CF,MAAM,OAAO,iBAAiB;IACZ,MAAM,CAAU;IAChB,WAAW,CAAU;IAOtC,YAAY,OAAgB,EAAE,MAAe;QAC5C,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IASD,KAAK,CAAC,UAAU,CAAC,SAAiB,EAAE,gBAAyB,KAAK;QACjE,IAAI,CAAC;YAEJ,MAAM,gBAAgB,GAAG,CAAC,QAAQ,EAAE,kBAAkB,EAAE,MAAM,EAAE,aAAa,EAAE,UAAU,CAAU,CAAC;YAEpG,MAAM,WAAW,GAAG,MAAM,WAAW,CAAoC;gBACxE,IAAI,EAAE,SAAS;gBACf,gBAAgB,EAAE,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBAC9C,GAAG,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,WAAW,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;gBAC3E,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;aAC3C,CAAC,CAAC;YAGH,MAAM,cAAc,GAAuB;gBAE1C,EAAE,EAAE,SAAS;gBACb,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,SAAS,EAAE,WAAW,CAAC,SAAS;gBAChC,KAAK,EAAE,WAAW,CAAC,KAAK;gBACxB,OAAO,EAAE,WAAW,CAAC,OAAO;gBAC5B,KAAK,EAAE,WAAW,CAAC,KAAK;gBACxB,SAAS,EAAE,WAAW,CAAC,SAAS;gBAGhC,QAAQ,EAAE;oBACT,MAAM,EAAE,WAAW,CAAC,MAAM;oBAC1B,gBAAgB,EAAE,WAAW,CAAC,gBAAgB;oBAC9C,IAAI,EAAE,WAAW,CAAC,IAAI;oBACtB,WAAW,EAAE,WAAW,CAAC,WAAW;iBACpC;aACD,CAAC;YAGF,IAAI,WAAW,CAAC,QAAQ,EAAE,CAAC;gBAC1B,MAAM,QAAQ,GAAoB,EAAE,CAAC;gBACrC,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAmC,CAAC;gBAEjE,IAAI,UAAU,IAAI,QAAQ,EAAE,CAAC;oBAC5B,QAAQ,CAAC,QAAQ,GAAG,QAAQ,CAAC,QAA6B,CAAC;gBAC5D,CAAC;gBAED,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;oBAC3B,QAAQ,CAAC,OAAO,GAAG,QAAQ,CAAC,OAA4B,CAAC;gBAC1D,CAAC;gBAED,IAAI,iBAAiB,IAAI,QAAQ,EAAE,CAAC;oBACnC,QAAQ,CAAC,eAAe,GAAG,QAAQ,CAAC,eAAoC,CAAC;gBAC1E,CAAC;gBAED,IAAI,iBAAiB,IAAI,QAAQ,EAAE,CAAC;oBACnC,QAAQ,CAAC,eAAe,GAAG,QAAQ,CAAC,eAAoC,CAAC;gBAC1E,CAAC;gBAED,IAAI,cAAc,IAAI,QAAQ,EAAE,CAAC;oBAChC,QAAQ,CAAC,YAAY,GAAG,QAAQ,CAAC,YAAuC,CAAC;gBAC1E,CAAC;gBAGD,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACtC,cAAc,CAAC,QAAQ,GAAG,QAAQ,CAAC;gBACpC,CAAC;YACF,CAAC;YAMD,IAAI,aAAa,EAAE,CAAC;gBACnB,MAAM,aAAa,GAAG,MAAM,kBAAkB,CAAC,cAAc,CAAC,IAAI,EAAE,UAAU,
|
|
1
|
+
{"version":3,"file":"dataset-detail.js","sourceRoot":"","sources":["../src/dataset-detail.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAGvD,MAAM,CAAC,MAAM,0BAA0B,GAAG;IACzC,IAAI,EAAE,iBAAiB;IACvB,WAAW,EAAE,wEAAwE;IACrF,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,wBAAwB,CAAC,CAAC,QAAQ,CAAC,gDAAgD,CAAC;KAClH,CAAC;IACF,WAAW,EAAE;QACZ,KAAK,EAAE,iBAAiB;QACxB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,KAAK;KACpB;CACQ,CAAC;AAEX,MAAM,CAAC,MAAM,4BAA4B,GAAG;IAC3C,IAAI,EAAE,iBAAiB;IACvB,KAAK,EAAE,iBAAiB;IACxB,WAAW,EACV,8HAA8H;IAC/H,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,UAAU,EAAE,CAAC;aACX,MAAM,EAAE;aACR,GAAG,CAAC,CAAC,EAAE,wBAAwB,CAAC;aAChC,GAAG,CAAC,GAAG,CAAC;aACR,QAAQ,CAAC,oDAAoD,CAAC;KAChE,CAAC;CACF,CAAC;AA2CF,MAAM,OAAO,iBAAiB;IACZ,MAAM,CAAU;IAChB,WAAW,CAAU;IAOtC,YAAY,OAAgB,EAAE,MAAe;QAC5C,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IASD,KAAK,CAAC,UAAU,CAAC,SAAiB,EAAE,gBAAyB,KAAK;QACjE,IAAI,CAAC;YAEJ,MAAM,gBAAgB,GAAG,CAAC,QAAQ,EAAE,kBAAkB,EAAE,MAAM,EAAE,aAAa,EAAE,UAAU,CAAU,CAAC;YAEpG,MAAM,WAAW,GAAG,MAAM,WAAW,CAAoC;gBACxE,IAAI,EAAE,SAAS;gBACf,gBAAgB,EAAE,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBAC9C,GAAG,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,WAAW,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;gBAC3E,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;aAC3C,CAAC,CAAC;YAGH,MAAM,cAAc,GAAuB;gBAE1C,EAAE,EAAE,SAAS;gBACb,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,SAAS,EAAE,WAAW,CAAC,SAAS;gBAChC,KAAK,EAAE,WAAW,CAAC,KAAK;gBACxB,OAAO,EAAE,WAAW,CAAC,OAAO;gBAC5B,KAAK,EAAE,WAAW,CAAC,KAAK;gBACxB,SAAS,EAAE,WAAW,CAAC,SAAS;gBAGhC,QAAQ,EAAE;oBACT,MAAM,EAAE,WAAW,CAAC,MAAM;oBAC1B,gBAAgB,EAAE,WAAW,CAAC,gBAAgB;oBAC9C,IAAI,EAAE,WAAW,CAAC,IAAI;oBACtB,WAAW,EAAE,WAAW,CAAC,WAAW;iBACpC;aACD,CAAC;YAGF,IAAI,WAAW,CAAC,QAAQ,EAAE,CAAC;gBAC1B,MAAM,QAAQ,GAAoB,EAAE,CAAC;gBACrC,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAmC,CAAC;gBAEjE,IAAI,UAAU,IAAI,QAAQ,EAAE,CAAC;oBAC5B,QAAQ,CAAC,QAAQ,GAAG,QAAQ,CAAC,QAA6B,CAAC;gBAC5D,CAAC;gBAED,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;oBAC3B,QAAQ,CAAC,OAAO,GAAG,QAAQ,CAAC,OAA4B,CAAC;gBAC1D,CAAC;gBAED,IAAI,iBAAiB,IAAI,QAAQ,EAAE,CAAC;oBACnC,QAAQ,CAAC,eAAe,GAAG,QAAQ,CAAC,eAAoC,CAAC;gBAC1E,CAAC;gBAED,IAAI,iBAAiB,IAAI,QAAQ,EAAE,CAAC;oBACnC,QAAQ,CAAC,eAAe,GAAG,QAAQ,CAAC,eAAoC,CAAC;gBAC1E,CAAC;gBAED,IAAI,cAAc,IAAI,QAAQ,EAAE,CAAC;oBAChC,QAAQ,CAAC,YAAY,GAAG,QAAQ,CAAC,YAAuC,CAAC;gBAC1E,CAAC;gBAGD,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACtC,cAAc,CAAC,QAAQ,GAAG,QAAQ,CAAC;gBACpC,CAAC;YACF,CAAC;YAMD,IAAI,aAAa,EAAE,CAAC;gBACnB,MAAM,aAAa,GAAG,MAAM,kBAAkB,CAAC,cAAc,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;gBAChF,IAAI,aAAa,EAAE,CAAC;oBACnB,MAAM,MAAM,GAAG,oBAAoB,CAAC,cAAc,CAAC,CAAC;oBACpD,MAAM,CAAC,SAAS;wBACf,uCAAuC,GAAG,aAAa,CAAC,IAAI,EAAE,GAAG,yBAAyB,CAAC;oBAC5F,OAAO,MAAM,CAAC;gBACf,CAAC;YACF,CAAC;YAED,OAAO,oBAAoB,CAAC,cAAc,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC5B,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC;gBAEhC,MAAM,UAAU,GACf,CAAC,KAAK,YAAY,WAAW,IAAI,KAAK,CAAC,UAAU,KAAK,GAAG,CAAC;oBAC1D,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC;oBACnB,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC;oBACzB,GAAG,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;gBAEtC,IAAI,UAAU,EAAE,CAAC;oBAChB,MAAM,IAAI,KAAK,CAAC,YAAY,SAAS,2CAA2C,CAAC,CAAC;gBACnF,CAAC;gBAED,MAAM,cAAc,GACnB,CAAC,KAAK,YAAY,WAAW,IAAI,CAAC,KAAK,CAAC,UAAU,KAAK,GAAG,IAAI,KAAK,CAAC,UAAU,KAAK,GAAG,CAAC,CAAC;oBACxF,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC;oBACnB,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC;oBACnB,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC;oBACxB,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC;oBACxB,GAAG,CAAC,QAAQ,CAAC,mCAAmC,CAAC,CAAC;gBAEnD,IAAI,cAAc,EAAE,CAAC;oBACpB,MAAM,IAAI,KAAK,CAAC,0EAA0E,SAAS,IAAI,CAAC,CAAC;gBAC1G,CAAC;gBAED,MAAM,IAAI,KAAK,CAAC,kCAAkC,GAAG,EAAE,CAAC,CAAC;YAC1D,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;CACD;AAGD,SAAS,oBAAoB,CAAC,OAA2B;IACxD,MAAM,CAAC,GAAa,EAAE,CAAC;IACvB,MAAM,CAAC,cAAc,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAEnG,CAAC,CAAC,IAAI,CAAC,KAAK,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAC5B,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAGX,IAAI,OAAO,CAAC,QAAQ,EAAE,WAAW,EAAE,CAAC;QACnC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACzB,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QACrC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACZ,CAAC;IAGD,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAGtB,IAAI,OAAO,CAAC,QAAQ,EAAE,MAAM,IAAI,cAAc,EAAE,CAAC;QAChD,CAAC,CAAC,IAAI,CAAC,iBAAiB,OAAO,CAAC,QAAQ,EAAE,MAAM,IAAI,cAAc,IAAI,EAAE,EAAE,CAAC,CAAC;IAC7E,CAAC;IAGD,MAAM,KAAK,GAAG,EAAE,CAAC;IACjB,IAAI,OAAO,CAAC,QAAQ,EAAE,gBAAgB,EAAE,CAAC;QACxC,KAAK,CAAC,IAAI,CAAC,kBAAkB,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;IACjF,CAAC;IACD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACnB,KAAK,CAAC,IAAI,CAAC,cAAc,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IACtD,CAAC;IACD,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,CAAC,CAAC,IAAI,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAClC,CAAC;IAGD,CAAC,CAAC,IAAI,CAAC,kBAAkB,UAAU,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IAG1D,MAAM,MAAM,GAAG,EAAE,CAAC;IAClB,IAAI,OAAO,CAAC,KAAK;QAAE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC3C,IAAI,OAAO,CAAC,OAAO;QAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC/C,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,CAAC,CAAC,IAAI,CAAC,iBAAiB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC/C,CAAC;IACD,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAGX,IAAI,OAAO,CAAC,QAAQ,EAAE,IAAI,IAAI,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAClB,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QACnE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACZ,CAAC;IAGD,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACtB,MAAM,QAAQ,GAAG,EAAE,CAAC;QAEpB,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YAC/B,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBACzD,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC;gBACtC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAC7B,QAAQ,CAAC,IAAI,CAAC,mBAAmB,SAAS,EAAE,CAAC,CAAC;QAC/C,CAAC;QAED,IAAI,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC9B,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;gBACtD,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;gBACrC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;YAC5B,QAAQ,CAAC,IAAI,CAAC,kBAAkB,OAAO,EAAE,CAAC,CAAC;QAC5C,CAAC;QAED,IAAI,OAAO,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC;YACtC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC;gBAC5D,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC7C,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC;YACpC,QAAQ,CAAC,IAAI,CAAC,0BAA0B,KAAK,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,IAAI,OAAO,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC;YACtC,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC;gBAC3D,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC7C,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC;YACpC,QAAQ,CAAC,IAAI,CAAC,wBAAwB,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACtB,CAAC,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;YACpB,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACZ,CAAC;IACF,CAAC;IAGD,CAAC,CAAC,IAAI,CAAC,qCAAqC,OAAO,CAAC,IAAI,4BAA4B,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC;IAErG,OAAO;QACN,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QACvB,YAAY,EAAE,CAAC;QACf,aAAa,EAAE,CAAC;KAChB,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
export interface DatasetStructureOptions {
|
|
2
|
+
config?: string;
|
|
3
|
+
split?: string;
|
|
4
|
+
}
|
|
5
|
+
export interface DatasetPreviewOptions {
|
|
6
|
+
config?: string;
|
|
7
|
+
split?: string;
|
|
8
|
+
offset?: number;
|
|
9
|
+
limit?: number;
|
|
10
|
+
}
|
|
11
|
+
interface DatasetViewerClientLike {
|
|
12
|
+
getJson<T>(path: string, params: Record<string, string | number | undefined>): Promise<T>;
|
|
13
|
+
}
|
|
14
|
+
interface DatasetMetadataProvider {
|
|
15
|
+
getMetadata(datasetId: string): Promise<DatasetStructureMetadata>;
|
|
16
|
+
}
|
|
17
|
+
export interface DatasetViewerInspectorOptions {
|
|
18
|
+
client?: DatasetViewerClientLike;
|
|
19
|
+
metadataProvider?: DatasetMetadataProvider;
|
|
20
|
+
hubUrl?: string;
|
|
21
|
+
}
|
|
22
|
+
interface DatasetStructureMetadata {
|
|
23
|
+
name: string;
|
|
24
|
+
author?: string;
|
|
25
|
+
description?: string;
|
|
26
|
+
downloadsAllTime?: number;
|
|
27
|
+
likes?: number;
|
|
28
|
+
gated?: false | 'auto' | 'manual';
|
|
29
|
+
private?: boolean;
|
|
30
|
+
updatedAt?: Date | string;
|
|
31
|
+
createdAt?: Date | string;
|
|
32
|
+
tags?: string[];
|
|
33
|
+
sha?: string;
|
|
34
|
+
cardData?: Record<string, unknown>;
|
|
35
|
+
}
|
|
36
|
+
export declare class DatasetViewerInspector {
|
|
37
|
+
private readonly client;
|
|
38
|
+
private readonly metadataProvider;
|
|
39
|
+
constructor(hfToken?: string, options?: DatasetViewerInspectorOptions);
|
|
40
|
+
getStructure(datasetId: string, options?: DatasetStructureOptions): Promise<string>;
|
|
41
|
+
getPreview(datasetId: string, options: DatasetPreviewOptions): Promise<string>;
|
|
42
|
+
private fetchSplits;
|
|
43
|
+
private fetchSize;
|
|
44
|
+
private fetchParquet;
|
|
45
|
+
private fetchRows;
|
|
46
|
+
}
|
|
47
|
+
export {};
|
|
48
|
+
//# sourceMappingURL=dataset-viewer-inspect.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"dataset-viewer-inspect.d.ts","sourceRoot":"","sources":["../src/dataset-viewer-inspect.ts"],"names":[],"mappings":"AAWA,MAAM,WAAW,uBAAuB;IACvC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,qBAAqB;IACrC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,UAAU,uBAAuB;IAChC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;CAC1F;AAED,UAAU,uBAAuB;IAChC,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,wBAAwB,CAAC,CAAC;CAClE;AAED,MAAM,WAAW,6BAA6B;IAC7C,MAAM,CAAC,EAAE,uBAAuB,CAAC;IACjC,gBAAgB,CAAC,EAAE,uBAAuB,CAAC;IAC3C,MAAM,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,UAAU,wBAAwB;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,KAAK,GAAG,MAAM,GAAG,QAAQ,CAAC;IAClC,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,SAAS,CAAC,EAAE,IAAI,GAAG,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAE,IAAI,GAAG,MAAM,CAAC;IAC1B,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AA0HD,qBAAa,sBAAsB;IAClC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA0B;IACjD,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAA0B;gBAE/C,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,GAAE,6BAAkC;IAKnE,YAAY,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,GAAE,uBAA4B,GAAG,OAAO,CAAC,MAAM,CAAC;IA8CvF,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,qBAAqB,GAAG,OAAO,CAAC,MAAM,CAAC;YAgCtE,WAAW;YASX,SAAS;YAST,YAAY;YAWZ,SAAS;CAcvB"}
|