@llmindset/hf-mcp 0.3.10 → 0.3.11

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.
@@ -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,2 @@
1
+ export {};
2
+ //# sourceMappingURL=create-repo.test.d.ts.map
@@ -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"}
@@ -10,10 +10,10 @@ export declare const HUB_REPO_DETAILS_TOOL_CONFIG: {
10
10
  }, "strip", z.ZodTypeAny, {
11
11
  repo_ids: string[];
12
12
  include_readme: boolean;
13
- repo_type?: "dataset" | "space" | "model" | undefined;
13
+ repo_type?: "space" | "dataset" | "model" | undefined;
14
14
  }, {
15
15
  repo_ids: string[];
16
- repo_type?: "dataset" | "space" | "model" | undefined;
16
+ repo_type?: "space" | "dataset" | "model" | undefined;
17
17
  include_readme?: boolean | undefined;
18
18
  }>;
19
19
  readonly annotations: {
@@ -10,6 +10,7 @@ interface BrowserToolConfig {
10
10
  }
11
11
  export declare const SEMANTIC_SEARCH_TOOL_CONFIG: BrowserToolConfig;
12
12
  export declare const REPO_SEARCH_TOOL_CONFIG: BrowserToolConfig;
13
+ export declare const CREATE_REPO_TOOL_CONFIG: BrowserToolConfig;
13
14
  export declare const PAPER_SEARCH_TOOL_CONFIG: BrowserToolConfig;
14
15
  export declare const HUB_REPO_DETAILS_TOOL_CONFIG: BrowserToolConfig;
15
16
  export declare const DUPLICATE_SPACE_TOOL_CONFIG: BrowserToolConfig;
@@ -19,6 +20,7 @@ export declare const DOC_FETCH_CONFIG: BrowserToolConfig;
19
20
  export declare const SPACE_SEARCH_TOOL_ID: string;
20
21
  export declare const MODEL_SEARCH_TOOL_ID = "model_search";
21
22
  export declare const REPO_SEARCH_TOOL_ID: string;
23
+ export declare const CREATE_REPO_TOOL_ID: string;
22
24
  export declare const MODEL_DETAIL_TOOL_ID = "model_details";
23
25
  export declare const PAPER_SEARCH_TOOL_ID: string;
24
26
  export declare const DATASET_SEARCH_TOOL_ID = "dataset_search";
@@ -32,15 +34,15 @@ export declare const DOCS_SEMANTIC_SEARCH_TOOL_ID: string;
32
34
  export declare const DOC_FETCH_TOOL_ID: string;
33
35
  export declare const HF_JOBS_TOOL_ID = "hf_jobs";
34
36
  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"];
37
+ export declare const ALL_BUILTIN_TOOL_IDS: readonly [string, "model_search", string, string, "model_details", string, "dataset_search", "dataset_details", string, string, "space_info", string, string, string, "use_space", "hf_jobs", "dynamic_space"];
36
38
  export declare const TOOL_ID_GROUPS: {
37
39
  readonly search: readonly [string, string, string, string];
38
40
  readonly spaces: readonly [string, string, "space_info", string, "use_space"];
39
41
  readonly detail: readonly ["model_details", "dataset_details", string];
40
42
  readonly docs: readonly [string, string];
41
- readonly hf_api: readonly [string, string, string, string, string];
43
+ readonly hf_api: readonly [string, string, string, string, string, string];
42
44
  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"];
45
+ readonly all: readonly [string, "model_search", string, string, "model_details", string, "dataset_search", "dataset_details", string, string, "space_info", string, string, string, "use_space", "hf_jobs", "dynamic_space"];
44
46
  };
45
47
  export type BuiltinToolId = (typeof ALL_BUILTIN_TOOL_IDS)[number];
46
48
  export declare function isValidBuiltinToolId(toolId: string): toolId is BuiltinToolId;
@@ -1 +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"}
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,uBAAuB,EAAE,iBAWrC,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,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,gNAkBvB,CAAC;AAEX,eAAO,MAAM,cAAc;;;;;;;;CAqBjB,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"}
@@ -28,6 +28,17 @@ export const REPO_SEARCH_TOOL_CONFIG = {
28
28
  openWorldHint: true,
29
29
  },
30
30
  };
31
+ export const CREATE_REPO_TOOL_CONFIG = {
32
+ name: 'create_repo',
33
+ description: 'Create a Hugging Face model, dataset, Space, or bucket repository. ' +
34
+ "name must be fully qualified, for example 'username/repo-name'.",
35
+ annotations: {
36
+ title: 'Create Hugging Face Repository',
37
+ destructiveHint: false,
38
+ readOnlyHint: false,
39
+ openWorldHint: true,
40
+ },
41
+ };
31
42
  export const PAPER_SEARCH_TOOL_CONFIG = {
32
43
  name: 'paper_search',
33
44
  description: 'Find Machine Learning research papers on the Hugging Face hub. ' +
@@ -95,6 +106,7 @@ export const DOC_FETCH_CONFIG = {
95
106
  export const SPACE_SEARCH_TOOL_ID = SEMANTIC_SEARCH_TOOL_CONFIG.name;
96
107
  export const MODEL_SEARCH_TOOL_ID = 'model_search';
97
108
  export const REPO_SEARCH_TOOL_ID = REPO_SEARCH_TOOL_CONFIG.name;
109
+ export const CREATE_REPO_TOOL_ID = CREATE_REPO_TOOL_CONFIG.name;
98
110
  export const MODEL_DETAIL_TOOL_ID = 'model_details';
99
111
  export const PAPER_SEARCH_TOOL_ID = PAPER_SEARCH_TOOL_CONFIG.name;
100
112
  export const DATASET_SEARCH_TOOL_ID = 'dataset_search';
@@ -112,6 +124,7 @@ export const ALL_BUILTIN_TOOL_IDS = [
112
124
  SPACE_SEARCH_TOOL_ID,
113
125
  MODEL_SEARCH_TOOL_ID,
114
126
  REPO_SEARCH_TOOL_ID,
127
+ CREATE_REPO_TOOL_ID,
115
128
  MODEL_DETAIL_TOOL_ID,
116
129
  PAPER_SEARCH_TOOL_ID,
117
130
  DATASET_SEARCH_TOOL_ID,
@@ -140,6 +153,7 @@ export const TOOL_ID_GROUPS = {
140
153
  hf_api: [
141
154
  SPACE_SEARCH_TOOL_ID,
142
155
  REPO_SEARCH_TOOL_ID,
156
+ CREATE_REPO_TOOL_ID,
143
157
  PAPER_SEARCH_TOOL_ID,
144
158
  HUB_REPO_DETAILS_TOOL_ID,
145
159
  DOCS_SEMANTIC_SEARCH_TOOL_ID,
@@ -1 +1 @@
1
- {"version":3,"file":"index.browser.js","sourceRoot":"","sources":["../src/index.browser.ts"],"names":[],"mappings":"AAAA,MAAM,cAAc,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE;IACvD,GAAG,EAAE,SAAS;IACd,KAAK,EAAE,MAAM;IACb,IAAI,EAAE,SAAS;IACf,QAAQ,EAAE,KAAK;CACf,CAAC,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;AAatB,MAAM,CAAC,MAAM,2BAA2B,GAAsB;IAC7D,IAAI,EAAE,cAAc;IACpB,WAAW,EACV,oHAAoH;QACpH,yDAAyD;IAC1D,WAAW,EAAE;QACZ,KAAK,EAAE,2BAA2B;QAClC,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,uBAAuB,GAAsB;IACzD,IAAI,EAAE,iBAAiB;IACvB,WAAW,EACV,kEAAkE;QAClE,gGAAgG;QAChG,2DAA2D;QAC3D,iDAAiD;IAClD,WAAW,EAAE;QACZ,KAAK,EAAE,aAAa;QACpB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,wBAAwB,GAAsB;IAC1D,IAAI,EAAE,cAAc;IACpB,WAAW,EACV,iEAAiE;QACjE,uDAAuD;QACvD,0DAA0D;IAC3D,WAAW,EAAE;QACZ,KAAK,EAAE,cAAc;QACrB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,4BAA4B,GAAsB;IAC9D,IAAI,EAAE,kBAAkB;IACxB,WAAW,EACV,6EAA6E;QAC7E,qCAAqC;IACtC,WAAW,EAAE;QACZ,KAAK,EAAE,kBAAkB;QACzB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,KAAK;KACpB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAsB;IAC7D,IAAI,EAAE,iBAAiB;IACvB,WAAW,EAAE,EAAE;IACf,WAAW,EAAE;QACZ,KAAK,EAAE,8BAA8B;QACrC,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,KAAK;QACnB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,uBAAuB,GAAsB;IACzD,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,EAAE;IACf,WAAW,EAAE;QACZ,KAAK,EAAE,kBAAkB;QACzB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAsB;IAC7D,IAAI,EAAE,eAAe;IACrB,WAAW,EACV,6IAA6I;QAC7I,8BAA8B,cAAc,qDAAqD;IAClG,WAAW,EAAE;QACZ,KAAK,EAAE,mCAAmC;QAC1C,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAsB;IAClD,IAAI,EAAE,cAAc;IACpB,WAAW,EACV,mIAAmI;IACpI,WAAW,EAAE;QACZ,KAAK,EAAE,8DAA8D;QACrE,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,oBAAoB,GAAG,2BAA2B,CAAC,IAAI,CAAC;AACrE,MAAM,CAAC,MAAM,oBAAoB,GAAG,cAAc,CAAC;AACnD,MAAM,CAAC,MAAM,mBAAmB,GAAG,uBAAuB,CAAC,IAAI,CAAC;AAChE,MAAM,CAAC,MAAM,oBAAoB,GAAG,eAAe,CAAC;AACpD,MAAM,CAAC,MAAM,oBAAoB,GAAG,wBAAwB,CAAC,IAAI,CAAC;AAClE,MAAM,CAAC,MAAM,sBAAsB,GAAG,gBAAgB,CAAC;AACvD,MAAM,CAAC,MAAM,sBAAsB,GAAG,iBAAiB,CAAC;AACxD,MAAM,CAAC,MAAM,wBAAwB,GAAG,4BAA4B,CAAC,IAAI,CAAC;AAC1E,MAAM,CAAC,MAAM,uBAAuB,GAAG,2BAA2B,CAAC,IAAI,CAAC;AACxE,MAAM,CAAC,MAAM,kBAAkB,GAAG,YAAY,CAAC;AAC/C,MAAM,CAAC,MAAM,mBAAmB,GAAG,uBAAuB,CAAC,IAAI,CAAC;AAChE,MAAM,CAAC,MAAM,iBAAiB,GAAG,WAAW,CAAC;AAC7C,MAAM,CAAC,MAAM,4BAA4B,GAAG,2BAA2B,CAAC,IAAI,CAAC;AAC7E,MAAM,CAAC,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,IAAI,CAAC;AACvD,MAAM,CAAC,MAAM,eAAe,GAAG,SAAS,CAAC;AACzC,MAAM,CAAC,MAAM,qBAAqB,GAAG,eAAe,CAAC;AAErD,MAAM,CAAC,MAAM,oBAAoB,GAAG;IACnC,oBAAoB;IACpB,oBAAoB;IACpB,mBAAmB;IACnB,oBAAoB;IACpB,oBAAoB;IACpB,sBAAsB;IACtB,sBAAsB;IACtB,wBAAwB;IACxB,uBAAuB;IACvB,kBAAkB;IAClB,mBAAmB;IACnB,4BAA4B;IAC5B,iBAAiB;IACjB,iBAAiB;IACjB,eAAe;IACf,qBAAqB;CACZ,CAAC;AAEX,MAAM,CAAC,MAAM,cAAc,GAAG;IAC7B,MAAM,EAAE,CAAC,oBAAoB,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,4BAA4B,CAAU;IAChH,MAAM,EAAE;QACP,oBAAoB;QACpB,uBAAuB;QACvB,kBAAkB;QAClB,mBAAmB;QACnB,iBAAiB;KACR;IACV,MAAM,EAAE,CAAC,oBAAoB,EAAE,sBAAsB,EAAE,wBAAwB,CAAU;IACzF,IAAI,EAAE,CAAC,4BAA4B,EAAE,iBAAiB,CAAU;IAChE,MAAM,EAAE;QACP,oBAAoB;QACpB,mBAAmB;QACnB,oBAAoB;QACpB,wBAAwB;QACxB,4BAA4B;KACnB;IACV,aAAa,EAAE,CAAC,qBAAqB,CAAU;IAC/C,GAAG,EAAE,CAAC,GAAG,oBAAoB,CAAU;CAC9B,CAAC;AAIX,MAAM,UAAU,oBAAoB,CAAC,MAAc;IAClD,OAAQ,oBAA0C,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AACrE,CAAC"}
1
+ {"version":3,"file":"index.browser.js","sourceRoot":"","sources":["../src/index.browser.ts"],"names":[],"mappings":"AAAA,MAAM,cAAc,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE;IACvD,GAAG,EAAE,SAAS;IACd,KAAK,EAAE,MAAM;IACb,IAAI,EAAE,SAAS;IACf,QAAQ,EAAE,KAAK;CACf,CAAC,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;AAatB,MAAM,CAAC,MAAM,2BAA2B,GAAsB;IAC7D,IAAI,EAAE,cAAc;IACpB,WAAW,EACV,oHAAoH;QACpH,yDAAyD;IAC1D,WAAW,EAAE;QACZ,KAAK,EAAE,2BAA2B;QAClC,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,uBAAuB,GAAsB;IACzD,IAAI,EAAE,iBAAiB;IACvB,WAAW,EACV,kEAAkE;QAClE,gGAAgG;QAChG,2DAA2D;QAC3D,iDAAiD;IAClD,WAAW,EAAE;QACZ,KAAK,EAAE,aAAa;QACpB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,uBAAuB,GAAsB;IACzD,IAAI,EAAE,aAAa;IACnB,WAAW,EACV,qEAAqE;QACrE,iEAAiE;IAClE,WAAW,EAAE;QACZ,KAAK,EAAE,gCAAgC;QACvC,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,KAAK;QACnB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,wBAAwB,GAAsB;IAC1D,IAAI,EAAE,cAAc;IACpB,WAAW,EACV,iEAAiE;QACjE,uDAAuD;QACvD,0DAA0D;IAC3D,WAAW,EAAE;QACZ,KAAK,EAAE,cAAc;QACrB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,4BAA4B,GAAsB;IAC9D,IAAI,EAAE,kBAAkB;IACxB,WAAW,EACV,6EAA6E;QAC7E,qCAAqC;IACtC,WAAW,EAAE;QACZ,KAAK,EAAE,kBAAkB;QACzB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,KAAK;KACpB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAsB;IAC7D,IAAI,EAAE,iBAAiB;IACvB,WAAW,EAAE,EAAE;IACf,WAAW,EAAE;QACZ,KAAK,EAAE,8BAA8B;QACrC,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,KAAK;QACnB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,uBAAuB,GAAsB;IACzD,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,EAAE;IACf,WAAW,EAAE;QACZ,KAAK,EAAE,kBAAkB;QACzB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAAsB;IAC7D,IAAI,EAAE,eAAe;IACrB,WAAW,EACV,6IAA6I;QAC7I,8BAA8B,cAAc,qDAAqD;IAClG,WAAW,EAAE;QACZ,KAAK,EAAE,mCAAmC;QAC1C,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAsB;IAClD,IAAI,EAAE,cAAc;IACpB,WAAW,EACV,mIAAmI;IACpI,WAAW,EAAE;QACZ,KAAK,EAAE,8DAA8D;QACrE,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,oBAAoB,GAAG,2BAA2B,CAAC,IAAI,CAAC;AACrE,MAAM,CAAC,MAAM,oBAAoB,GAAG,cAAc,CAAC;AACnD,MAAM,CAAC,MAAM,mBAAmB,GAAG,uBAAuB,CAAC,IAAI,CAAC;AAChE,MAAM,CAAC,MAAM,mBAAmB,GAAG,uBAAuB,CAAC,IAAI,CAAC;AAChE,MAAM,CAAC,MAAM,oBAAoB,GAAG,eAAe,CAAC;AACpD,MAAM,CAAC,MAAM,oBAAoB,GAAG,wBAAwB,CAAC,IAAI,CAAC;AAClE,MAAM,CAAC,MAAM,sBAAsB,GAAG,gBAAgB,CAAC;AACvD,MAAM,CAAC,MAAM,sBAAsB,GAAG,iBAAiB,CAAC;AACxD,MAAM,CAAC,MAAM,wBAAwB,GAAG,4BAA4B,CAAC,IAAI,CAAC;AAC1E,MAAM,CAAC,MAAM,uBAAuB,GAAG,2BAA2B,CAAC,IAAI,CAAC;AACxE,MAAM,CAAC,MAAM,kBAAkB,GAAG,YAAY,CAAC;AAC/C,MAAM,CAAC,MAAM,mBAAmB,GAAG,uBAAuB,CAAC,IAAI,CAAC;AAChE,MAAM,CAAC,MAAM,iBAAiB,GAAG,WAAW,CAAC;AAC7C,MAAM,CAAC,MAAM,4BAA4B,GAAG,2BAA2B,CAAC,IAAI,CAAC;AAC7E,MAAM,CAAC,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,IAAI,CAAC;AACvD,MAAM,CAAC,MAAM,eAAe,GAAG,SAAS,CAAC;AACzC,MAAM,CAAC,MAAM,qBAAqB,GAAG,eAAe,CAAC;AAErD,MAAM,CAAC,MAAM,oBAAoB,GAAG;IACnC,oBAAoB;IACpB,oBAAoB;IACpB,mBAAmB;IACnB,mBAAmB;IACnB,oBAAoB;IACpB,oBAAoB;IACpB,sBAAsB;IACtB,sBAAsB;IACtB,wBAAwB;IACxB,uBAAuB;IACvB,kBAAkB;IAClB,mBAAmB;IACnB,4BAA4B;IAC5B,iBAAiB;IACjB,iBAAiB;IACjB,eAAe;IACf,qBAAqB;CACZ,CAAC;AAEX,MAAM,CAAC,MAAM,cAAc,GAAG;IAC7B,MAAM,EAAE,CAAC,oBAAoB,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,4BAA4B,CAAU;IAChH,MAAM,EAAE;QACP,oBAAoB;QACpB,uBAAuB;QACvB,kBAAkB;QAClB,mBAAmB;QACnB,iBAAiB;KACR;IACV,MAAM,EAAE,CAAC,oBAAoB,EAAE,sBAAsB,EAAE,wBAAwB,CAAU;IACzF,IAAI,EAAE,CAAC,4BAA4B,EAAE,iBAAiB,CAAU;IAChE,MAAM,EAAE;QACP,oBAAoB;QACpB,mBAAmB;QACnB,mBAAmB;QACnB,oBAAoB;QACpB,wBAAwB;QACxB,4BAA4B;KACnB;IACV,aAAa,EAAE,CAAC,qBAAqB,CAAU;IAC/C,GAAG,EAAE,CAAC,GAAG,oBAAoB,CAAU;CAC9B,CAAC;AAIX,MAAM,UAAU,oBAAoB,CAAC,MAAc;IAClD,OAAQ,oBAA0C,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AACrE,CAAC"}
package/dist/index.d.ts CHANGED
@@ -7,6 +7,7 @@ export * from './utilities.js';
7
7
  export * from './paper-search.js';
8
8
  export * from './dataset-search.js';
9
9
  export * from './repo-search.js';
10
+ export * from './create-repo.js';
10
11
  export * from './dataset-detail.js';
11
12
  export * from './hub-inspect.js';
12
13
  export * from './duplicate-space.js';
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,cAAc,kBAAkB,CAAC;AACjC,cAAc,qBAAqB,CAAC;AACpC,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,kBAAkB,CAAC;AACjC,cAAc,qBAAqB,CAAC;AACpC,cAAc,kBAAkB,CAAC;AACjC,cAAc,sBAAsB,CAAC;AACrC,cAAc,iBAAiB,CAAC;AAChC,cAAc,kBAAkB,CAAC;AACjC,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,oBAAoB,CAAC;AACnC,cAAc,uCAAuC,CAAC;AACtD,cAAc,4BAA4B,CAAC;AAC3C,cAAc,mBAAmB,CAAC;AAClC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,qBAAqB,CAAC;AACpC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,yBAAyB,CAAC;AAGxC,cAAc,wBAAwB,CAAC;AAGvC,cAAc,eAAe,CAAC"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,cAAc,kBAAkB,CAAC;AACjC,cAAc,qBAAqB,CAAC;AACpC,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,kBAAkB,CAAC;AACjC,cAAc,kBAAkB,CAAC;AACjC,cAAc,qBAAqB,CAAC;AACpC,cAAc,kBAAkB,CAAC;AACjC,cAAc,sBAAsB,CAAC;AACrC,cAAc,iBAAiB,CAAC;AAChC,cAAc,kBAAkB,CAAC;AACjC,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,oBAAoB,CAAC;AACnC,cAAc,uCAAuC,CAAC;AACtD,cAAc,4BAA4B,CAAC;AAC3C,cAAc,mBAAmB,CAAC;AAClC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,qBAAqB,CAAC;AACpC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,yBAAyB,CAAC;AAGxC,cAAc,wBAAwB,CAAC;AAGvC,cAAc,eAAe,CAAC"}
package/dist/index.js CHANGED
@@ -7,6 +7,7 @@ export * from './utilities.js';
7
7
  export * from './paper-search.js';
8
8
  export * from './dataset-search.js';
9
9
  export * from './repo-search.js';
10
+ export * from './create-repo.js';
10
11
  export * from './dataset-detail.js';
11
12
  export * from './hub-inspect.js';
12
13
  export * from './duplicate-space.js';
package/dist/index.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,cAAc,kBAAkB,CAAC;AACjC,cAAc,qBAAqB,CAAC;AACpC,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,kBAAkB,CAAC;AACjC,cAAc,qBAAqB,CAAC;AACpC,cAAc,kBAAkB,CAAC;AACjC,cAAc,sBAAsB,CAAC;AACrC,cAAc,iBAAiB,CAAC;AAChC,cAAc,kBAAkB,CAAC;AACjC,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,oBAAoB,CAAC;AACnC,cAAc,uCAAuC,CAAC;AACtD,cAAc,4BAA4B,CAAC;AAC3C,cAAc,mBAAmB,CAAC;AAClC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,qBAAqB,CAAC;AACpC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,yBAAyB,CAAC;AAGxC,cAAc,wBAAwB,CAAC;AAGvC,cAAc,eAAe,CAAC"}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,cAAc,kBAAkB,CAAC;AACjC,cAAc,qBAAqB,CAAC;AACpC,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,mBAAmB,CAAC;AAClC,cAAc,qBAAqB,CAAC;AACpC,cAAc,kBAAkB,CAAC;AACjC,cAAc,kBAAkB,CAAC;AACjC,cAAc,qBAAqB,CAAC;AACpC,cAAc,kBAAkB,CAAC;AACjC,cAAc,sBAAsB,CAAC;AACrC,cAAc,iBAAiB,CAAC;AAChC,cAAc,kBAAkB,CAAC;AACjC,cAAc,mBAAmB,CAAC;AAClC,cAAc,mBAAmB,CAAC;AAClC,cAAc,oBAAoB,CAAC;AACnC,cAAc,uCAAuC,CAAC;AACtD,cAAc,4BAA4B,CAAC;AAC3C,cAAc,mBAAmB,CAAC;AAClC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,qBAAqB,CAAC;AACpC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,yBAAyB,CAAC;AAGxC,cAAc,wBAAwB,CAAC;AAGvC,cAAc,eAAe,CAAC"}
@@ -1,8 +1,8 @@
1
1
  import { z } from 'zod';
2
2
  export declare const CPU_FLAVORS: readonly ["cpu-basic", "cpu-upgrade", "cpu-performance", "cpu-xl"];
3
- export declare const GPU_FLAVORS: readonly ["sprx8", "zero-a10g", "t4-small", "t4-medium", "l4x1", "l4x4", "l40sx1", "l40sx4", "l40sx8", "a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4", "a100-large", "h100", "h100x8"];
3
+ export declare const GPU_FLAVORS: readonly ["sprx8", "zero-a10g", "t4-small", "t4-medium", "l4x1", "l4x4", "l40sx1", "l40sx4", "l40sx8", "a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4", "a100-large", "a100x4", "a100x8"];
4
4
  export declare const SPECIALIZED_FLAVORS: readonly ["inf2x6"];
5
- export declare const ALL_FLAVORS: readonly ["cpu-basic", "cpu-upgrade", "cpu-performance", "cpu-xl", "sprx8", "zero-a10g", "t4-small", "t4-medium", "l4x1", "l4x4", "l40sx1", "l40sx4", "l40sx8", "a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4", "a100-large", "h100", "h100x8", "inf2x6"];
5
+ export declare const ALL_FLAVORS: readonly ["cpu-basic", "cpu-upgrade", "cpu-performance", "cpu-xl", "sprx8", "zero-a10g", "t4-small", "t4-medium", "l4x1", "l4x4", "l40sx1", "l40sx4", "l40sx8", "a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4", "a100-large", "a100x4", "a100x8", "inf2x6"];
6
6
  export type JobFlavor = (typeof ALL_FLAVORS)[number];
7
7
  export type JobStage = 'RUNNING' | 'COMPLETED' | 'CANCELED' | 'ERROR' | 'DELETED';
8
8
  export interface JobStatus {
@@ -67,28 +67,28 @@ export declare const runArgsSchema: z.ZodObject<{
67
67
  } & {
68
68
  image: z.ZodDefault<z.ZodOptional<z.ZodString>>;
69
69
  command: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
70
- flavor: z.ZodDefault<z.ZodOptional<z.ZodEnum<["cpu-basic", "cpu-upgrade", "cpu-performance", "cpu-xl", "sprx8", "zero-a10g", "t4-small", "t4-medium", "l4x1", "l4x4", "l40sx1", "l40sx4", "l40sx8", "a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4", "a100-large", "h100", "h100x8", "inf2x6"]>>>;
70
+ flavor: z.ZodDefault<z.ZodOptional<z.ZodEnum<["cpu-basic", "cpu-upgrade", "cpu-performance", "cpu-xl", "sprx8", "zero-a10g", "t4-small", "t4-medium", "l4x1", "l4x4", "l40sx1", "l40sx4", "l40sx8", "a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4", "a100-large", "a100x4", "a100x8", "inf2x6"]>>>;
71
71
  env: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
72
72
  secrets: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
73
73
  timeout: z.ZodDefault<z.ZodOptional<z.ZodString>>;
74
74
  detach: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
75
75
  }, "strip", z.ZodTypeAny, {
76
+ timeout: string;
76
77
  image: string;
77
78
  command: string | string[];
78
- flavor: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "h100" | "h100x8";
79
- timeout: string;
79
+ flavor: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "a100x4" | "a100x8";
80
80
  detach: boolean;
81
81
  namespace?: string | undefined;
82
82
  env?: Record<string, string> | undefined;
83
83
  secrets?: Record<string, string> | undefined;
84
84
  }, {
85
85
  command: string | string[];
86
+ timeout?: string | undefined;
86
87
  image?: string | undefined;
87
88
  namespace?: string | undefined;
88
- flavor?: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "h100" | "h100x8" | undefined;
89
+ flavor?: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "a100x4" | "a100x8" | undefined;
89
90
  env?: Record<string, string> | undefined;
90
91
  secrets?: Record<string, string> | undefined;
91
- timeout?: string | undefined;
92
92
  detach?: boolean | undefined;
93
93
  }>;
94
94
  export declare const uvArgsSchema: z.ZodObject<{
@@ -98,15 +98,15 @@ export declare const uvArgsSchema: z.ZodObject<{
98
98
  with_deps: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
99
99
  script_args: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
100
100
  python: z.ZodOptional<z.ZodString>;
101
- flavor: z.ZodDefault<z.ZodOptional<z.ZodEnum<["cpu-basic", "cpu-upgrade", "cpu-performance", "cpu-xl", "sprx8", "zero-a10g", "t4-small", "t4-medium", "l4x1", "l4x4", "l40sx1", "l40sx4", "l40sx8", "a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4", "a100-large", "h100", "h100x8", "inf2x6"]>>>;
101
+ flavor: z.ZodDefault<z.ZodOptional<z.ZodEnum<["cpu-basic", "cpu-upgrade", "cpu-performance", "cpu-xl", "sprx8", "zero-a10g", "t4-small", "t4-medium", "l4x1", "l4x4", "l40sx1", "l40sx4", "l40sx8", "a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4", "a100-large", "a100x4", "a100x8", "inf2x6"]>>>;
102
102
  env: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
103
103
  secrets: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
104
104
  timeout: z.ZodDefault<z.ZodOptional<z.ZodString>>;
105
105
  detach: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
106
106
  }, "strip", z.ZodTypeAny, {
107
- script: string;
108
- flavor: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "h100" | "h100x8";
109
107
  timeout: string;
108
+ script: string;
109
+ flavor: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "a100x4" | "a100x8";
110
110
  detach: boolean;
111
111
  namespace?: string | undefined;
112
112
  env?: Record<string, string> | undefined;
@@ -116,11 +116,11 @@ export declare const uvArgsSchema: z.ZodObject<{
116
116
  python?: string | undefined;
117
117
  }, {
118
118
  script: string;
119
+ timeout?: string | undefined;
119
120
  namespace?: string | undefined;
120
- flavor?: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "h100" | "h100x8" | undefined;
121
+ flavor?: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "a100x4" | "a100x8" | undefined;
121
122
  env?: Record<string, string> | undefined;
122
123
  secrets?: Record<string, string> | undefined;
123
- timeout?: string | undefined;
124
124
  detach?: boolean | undefined;
125
125
  with_deps?: string[] | undefined;
126
126
  script_args?: string[] | undefined;
@@ -136,8 +136,8 @@ export declare const psArgsSchema: z.ZodObject<{
136
136
  status?: string | undefined;
137
137
  namespace?: string | undefined;
138
138
  }, {
139
- all?: boolean | undefined;
140
139
  status?: string | undefined;
140
+ all?: boolean | undefined;
141
141
  namespace?: string | undefined;
142
142
  }>;
143
143
  export declare const logsArgsSchema: z.ZodObject<{
@@ -181,7 +181,7 @@ export declare const scheduledRunArgsSchema: z.ZodObject<{
181
181
  } & {
182
182
  image: z.ZodDefault<z.ZodOptional<z.ZodString>>;
183
183
  command: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
184
- flavor: z.ZodDefault<z.ZodOptional<z.ZodEnum<["cpu-basic", "cpu-upgrade", "cpu-performance", "cpu-xl", "sprx8", "zero-a10g", "t4-small", "t4-medium", "l4x1", "l4x4", "l40sx1", "l40sx4", "l40sx8", "a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4", "a100-large", "h100", "h100x8", "inf2x6"]>>>;
184
+ flavor: z.ZodDefault<z.ZodOptional<z.ZodEnum<["cpu-basic", "cpu-upgrade", "cpu-performance", "cpu-xl", "sprx8", "zero-a10g", "t4-small", "t4-medium", "l4x1", "l4x4", "l40sx1", "l40sx4", "l40sx8", "a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4", "a100-large", "a100x4", "a100x8", "inf2x6"]>>>;
185
185
  env: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
186
186
  secrets: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
187
187
  timeout: z.ZodDefault<z.ZodOptional<z.ZodString>>;
@@ -190,10 +190,10 @@ export declare const scheduledRunArgsSchema: z.ZodObject<{
190
190
  schedule: z.ZodString;
191
191
  suspend: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
192
192
  }, "strip", z.ZodTypeAny, {
193
+ timeout: string;
193
194
  image: string;
194
195
  command: string | string[];
195
- flavor: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "h100" | "h100x8";
196
- timeout: string;
196
+ flavor: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "a100x4" | "a100x8";
197
197
  detach: boolean;
198
198
  schedule: string;
199
199
  suspend: boolean;
@@ -203,12 +203,12 @@ export declare const scheduledRunArgsSchema: z.ZodObject<{
203
203
  }, {
204
204
  command: string | string[];
205
205
  schedule: string;
206
+ timeout?: string | undefined;
206
207
  image?: string | undefined;
207
208
  namespace?: string | undefined;
208
- flavor?: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "h100" | "h100x8" | undefined;
209
+ flavor?: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "a100x4" | "a100x8" | undefined;
209
210
  env?: Record<string, string> | undefined;
210
211
  secrets?: Record<string, string> | undefined;
211
- timeout?: string | undefined;
212
212
  detach?: boolean | undefined;
213
213
  suspend?: boolean | undefined;
214
214
  }>;
@@ -219,7 +219,7 @@ export declare const scheduledUvArgsSchema: z.ZodObject<{
219
219
  with_deps: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
220
220
  script_args: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
221
221
  python: z.ZodOptional<z.ZodString>;
222
- flavor: z.ZodDefault<z.ZodOptional<z.ZodEnum<["cpu-basic", "cpu-upgrade", "cpu-performance", "cpu-xl", "sprx8", "zero-a10g", "t4-small", "t4-medium", "l4x1", "l4x4", "l40sx1", "l40sx4", "l40sx8", "a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4", "a100-large", "h100", "h100x8", "inf2x6"]>>>;
222
+ flavor: z.ZodDefault<z.ZodOptional<z.ZodEnum<["cpu-basic", "cpu-upgrade", "cpu-performance", "cpu-xl", "sprx8", "zero-a10g", "t4-small", "t4-medium", "l4x1", "l4x4", "l40sx1", "l40sx4", "l40sx8", "a10g-small", "a10g-large", "a10g-largex2", "a10g-largex4", "a100-large", "a100x4", "a100x8", "inf2x6"]>>>;
223
223
  env: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
224
224
  secrets: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
225
225
  timeout: z.ZodDefault<z.ZodOptional<z.ZodString>>;
@@ -228,9 +228,9 @@ export declare const scheduledUvArgsSchema: z.ZodObject<{
228
228
  schedule: z.ZodString;
229
229
  suspend: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
230
230
  }, "strip", z.ZodTypeAny, {
231
- script: string;
232
- flavor: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "h100" | "h100x8";
233
231
  timeout: string;
232
+ script: string;
233
+ flavor: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "a100x4" | "a100x8";
234
234
  detach: boolean;
235
235
  schedule: string;
236
236
  suspend: boolean;
@@ -243,11 +243,11 @@ export declare const scheduledUvArgsSchema: z.ZodObject<{
243
243
  }, {
244
244
  script: string;
245
245
  schedule: string;
246
+ timeout?: string | undefined;
246
247
  namespace?: string | undefined;
247
- flavor?: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "h100" | "h100x8" | undefined;
248
+ flavor?: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "cpu-performance" | "cpu-xl" | "sprx8" | "inf2x6" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "l40sx1" | "l40sx4" | "l40sx8" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "a100x4" | "a100x8" | undefined;
248
249
  env?: Record<string, string> | undefined;
249
250
  secrets?: Record<string, string> | undefined;
250
- timeout?: string | undefined;
251
251
  detach?: boolean | undefined;
252
252
  with_deps?: string[] | undefined;
253
253
  script_args?: string[] | undefined;
@@ -1 +1 @@
1
- {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/jobs/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,eAAO,MAAM,WAAW,oEAAqE,CAAC;AAE9F,eAAO,MAAM,WAAW,oMAiBd,CAAC;AAEX,eAAO,MAAM,mBAAmB,qBAAsB,CAAC;AAEvD,eAAO,MAAM,WAAW,uQAAoE,CAAC;AAE7F,MAAM,MAAM,SAAS,GAAG,CAAC,OAAO,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC;AAKrD,MAAM,MAAM,QAAQ,GAAG,SAAS,GAAG,WAAW,GAAG,UAAU,GAAG,OAAO,GAAG,SAAS,CAAC;AAKlF,MAAM,WAAW,SAAS;IACzB,KAAK,EAAE,QAAQ,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACxB;AAKD,MAAM,WAAW,QAAQ;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,GAAG,KAAK,CAAC;CACrB;AAMD,MAAM,WAAW,OAAO;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACpC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC,CAAC;IACxC,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,SAAS,CAAC;IAClB,KAAK,EAAE,QAAQ,CAAC;IAChB,SAAS,CAAC,EAAE,QAAQ,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;CACpB;AAKD,MAAM,WAAW,OAAO;IACvB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACrC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,MAAM,CAAC;CACxB;AAKD,MAAM,WAAW,gBAAgB;IAChC,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,OAAO,EAAE,OAAO,CAAC;CACjB;AAKD,MAAM,WAAW,gBAAgB;IAChC,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,QAAQ,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CAClB;AAKD,MAAM,WAAW,QAAQ;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;CACb;AAYD,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA6BxB,CAAC;AAGH,eAAO,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAgBvB,CAAC;AAGH,eAAO,MAAM,YAAY;;;;;;;;;;;;;EAGvB,CAAC;AAGH,eAAO,MAAM,cAAc;;;;;;;;;;;;;EAGzB,CAAC;AAGH,eAAO,MAAM,iBAAiB;;;;;;;;;;EAE5B,CAAC;AAGH,eAAO,MAAM,gBAAgB;;;;;;;;;;EAE3B,CAAC;AAGH,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAGjC,CAAC;AAGH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAGhC,CAAC;AAGH,eAAO,MAAM,qBAAqB;;;;;;;;;;EAEhC,CAAC;AAGH,eAAO,MAAM,sBAAsB;;;;;;;;;;EAEjC,CAAC;AAKH,MAAM,MAAM,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AACpD,MAAM,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,YAAY,CAAC,CAAC;AAClD,MAAM,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,YAAY,CAAC,CAAC;AAClD,MAAM,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAC;AACtD,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAC5D,MAAM,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAC1D,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AACtE,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AACpE,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AACpE,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC"}
1
+ {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/jobs/types.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,eAAO,MAAM,WAAW,oEAAqE,CAAC;AAE9F,eAAO,MAAM,WAAW,sMAiBd,CAAC;AAEX,eAAO,MAAM,mBAAmB,qBAAsB,CAAC;AAEvD,eAAO,MAAM,WAAW,yQAAoE,CAAC;AAE7F,MAAM,MAAM,SAAS,GAAG,CAAC,OAAO,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC;AAYrD,MAAM,MAAM,QAAQ,GAAG,SAAS,GAAG,WAAW,GAAG,UAAU,GAAG,OAAO,GAAG,SAAS,CAAC;AAKlF,MAAM,WAAW,SAAS;IACzB,KAAK,EAAE,QAAQ,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACxB;AAKD,MAAM,WAAW,QAAQ;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,GAAG,KAAK,CAAC;CACrB;AAMD,MAAM,WAAW,OAAO;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACpC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC,CAAC;IACxC,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,SAAS,CAAC;IAClB,KAAK,EAAE,QAAQ,CAAC;IAChB,SAAS,CAAC,EAAE,QAAQ,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;CACpB;AAKD,MAAM,WAAW,OAAO;IACvB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACrC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,MAAM,CAAC;CACxB;AAKD,MAAM,WAAW,gBAAgB;IAChC,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,OAAO,EAAE,OAAO,CAAC;CACjB;AAKD,MAAM,WAAW,gBAAgB;IAChC,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,QAAQ,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CAClB;AAKD,MAAM,WAAW,QAAQ;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;CACb;AAYD,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA6BxB,CAAC;AAGH,eAAO,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAgBvB,CAAC;AAGH,eAAO,MAAM,YAAY;;;;;;;;;;;;;EAGvB,CAAC;AAGH,eAAO,MAAM,cAAc;;;;;;;;;;;;;EAGzB,CAAC;AAGH,eAAO,MAAM,iBAAiB;;;;;;;;;;EAE5B,CAAC;AAGH,eAAO,MAAM,gBAAgB;;;;;;;;;;EAE3B,CAAC;AAGH,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAGjC,CAAC;AAGH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAGhC,CAAC;AAGH,eAAO,MAAM,qBAAqB;;;;;;;;;;EAEhC,CAAC;AAGH,eAAO,MAAM,sBAAsB;;;;;;;;;;EAEjC,CAAC;AAKH,MAAM,MAAM,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AACpD,MAAM,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,YAAY,CAAC,CAAC;AAClD,MAAM,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,YAAY,CAAC,CAAC;AAClD,MAAM,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAC;AACtD,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAC5D,MAAM,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAC1D,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AACtE,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AACpE,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AACpE,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC"}
@@ -15,11 +15,16 @@ export const GPU_FLAVORS = [
15
15
  'a10g-largex2',
16
16
  'a10g-largex4',
17
17
  'a100-large',
18
- 'h100',
19
- 'h100x8',
18
+ 'a100x4',
19
+ 'a100x8',
20
20
  ];
21
21
  export const SPECIALIZED_FLAVORS = ['inf2x6'];
22
22
  export const ALL_FLAVORS = [...CPU_FLAVORS, ...GPU_FLAVORS, ...SPECIALIZED_FLAVORS];
23
+ function assertExhaustiveHardwareUnion(_value) {
24
+ void _value;
25
+ }
26
+ assertExhaustiveHardwareUnion();
27
+ assertExhaustiveHardwareUnion();
23
28
  const commonArgsSchema = z.object({
24
29
  namespace: z.string().optional().describe('Target namespace (username or organization). Defaults to current user.'),
25
30
  });
@@ -1 +1 @@
1
- {"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/jobs/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,WAAW,EAAE,aAAa,EAAE,iBAAiB,EAAE,QAAQ,CAAU,CAAC;AAE9F,MAAM,CAAC,MAAM,WAAW,GAAG;IAC1B,OAAO;IACP,WAAW;IACX,UAAU;IACV,WAAW;IACX,MAAM;IACN,MAAM;IACN,QAAQ;IACR,QAAQ;IACR,QAAQ;IACR,YAAY;IACZ,YAAY;IACZ,cAAc;IACd,cAAc;IACd,YAAY;IACZ,MAAM;IACN,QAAQ;CACC,CAAC;AAEX,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,QAAQ,CAAU,CAAC;AAEvD,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,GAAG,WAAW,EAAE,GAAG,WAAW,EAAE,GAAG,mBAAmB,CAAU,CAAC;AAqG7F,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IACjC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wEAAwE,CAAC;CACnH,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,aAAa,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACpD,KAAK,EAAE,CAAC;SACN,MAAM,EAAE;SACR,QAAQ,CAAC,iFAAiF,CAAC;SAC3F,QAAQ,EAAE;SACV,OAAO,CAAC,aAAa,CAAC;IACxB,OAAO,EAAE,CAAC;SACR,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;SACxC,QAAQ,CACR,gFAAgF;QAC/E,yEAAyE;QACzE,8DAA8D,CAC/D;IACF,MAAM,EAAE,CAAC;SACP,IAAI,CAAC,WAAW,CAAC;SACjB,QAAQ,EAAE;SACV,OAAO,CAAC,WAAW,CAAC;SACpB,QAAQ,CAAC,6BAA6B,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IACjE,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,0CAA0C,CAAC;IACzF,OAAO,EAAE,CAAC;SACR,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SAClB,QAAQ,EAAE;SACV,QAAQ,CAAC,0EAA0E,CAAC;IACtF,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sDAAsD,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC;IAC9G,MAAM,EAAE,CAAC;SACP,OAAO,EAAE;SACT,QAAQ,EAAE;SACV,OAAO,CAAC,KAAK,CAAC;SACd,QAAQ,CAAC,8FAA8F,CAAC;CAC1G,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,YAAY,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACnD,MAAM,EAAE,CAAC;SACP,MAAM,EAAE;SACR,QAAQ,CAAC,iGAAiG,CAAC;IAE7G,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;IACrF,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;IACvF,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2CAA2C,CAAC;IACnF,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,QAAQ,CAAC,iBAAiB,CAAC;IACvF,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,0CAA0C,CAAC;IACzF,OAAO,EAAE,CAAC;SACR,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SAClB,QAAQ,EAAE;SACV,QAAQ,CAAC,0EAA0E,CAAC;IACtF,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC;IACtE,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,8FAA8F,CAAC;CACtJ,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,YAAY,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACnD,GAAG,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,uCAAuC,CAAC;IAC5F,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2EAA2E,CAAC;CACnH,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,cAAc,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACrD,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;IACxD,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,yCAAyC,CAAC;CAC3F,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACxD,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,sBAAsB,CAAC;CACnF,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,gBAAgB,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACvD,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;CAC/C,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,sBAAsB,GAAG,aAAa,CAAC,MAAM,CAAC;IAC1D,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,sFAAsF,CAAC;IACrH,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;CACpF,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,qBAAqB,GAAG,YAAY,CAAC,MAAM,CAAC;IACxD,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wCAAwC,CAAC;IACvE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;CACpF,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,qBAAqB,GAAG,gBAAgB,CAAC,MAAM,CAAC;IAC5D,GAAG,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,mDAAmD,CAAC;CACxG,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,sBAAsB,GAAG,gBAAgB,CAAC,MAAM,CAAC;IAC7D,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;CACzD,CAAC,CAAC"}
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/jobs/types.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,WAAW,EAAE,aAAa,EAAE,iBAAiB,EAAE,QAAQ,CAAU,CAAC;AAE9F,MAAM,CAAC,MAAM,WAAW,GAAG;IAC1B,OAAO;IACP,WAAW;IACX,UAAU;IACV,WAAW;IACX,MAAM;IACN,MAAM;IACN,QAAQ;IACR,QAAQ;IACR,QAAQ;IACR,YAAY;IACZ,YAAY;IACZ,cAAc;IACd,cAAc;IACd,YAAY;IACZ,QAAQ;IACR,QAAQ;CACC,CAAC;AAEX,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,QAAQ,CAAU,CAAC;AAEvD,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,GAAG,WAAW,EAAE,GAAG,WAAW,EAAE,GAAG,mBAAmB,CAAU,CAAC;AAI7F,SAAS,6BAA6B,CAAkB,MAAU;IACjE,KAAK,MAAM,CAAC;AACb,CAAC;AAED,6BAA6B,EAA2C,CAAC;AACzE,6BAA6B,EAA2C,CAAC;AAmGzE,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IACjC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wEAAwE,CAAC;CACnH,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,aAAa,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACpD,KAAK,EAAE,CAAC;SACN,MAAM,EAAE;SACR,QAAQ,CAAC,iFAAiF,CAAC;SAC3F,QAAQ,EAAE;SACV,OAAO,CAAC,aAAa,CAAC;IACxB,OAAO,EAAE,CAAC;SACR,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;SACxC,QAAQ,CACR,gFAAgF;QAC/E,yEAAyE;QACzE,8DAA8D,CAC/D;IACF,MAAM,EAAE,CAAC;SACP,IAAI,CAAC,WAAW,CAAC;SACjB,QAAQ,EAAE;SACV,OAAO,CAAC,WAAW,CAAC;SACpB,QAAQ,CAAC,6BAA6B,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IACjE,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,0CAA0C,CAAC;IACzF,OAAO,EAAE,CAAC;SACR,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SAClB,QAAQ,EAAE;SACV,QAAQ,CAAC,0EAA0E,CAAC;IACtF,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sDAAsD,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC;IAC9G,MAAM,EAAE,CAAC;SACP,OAAO,EAAE;SACT,QAAQ,EAAE;SACV,OAAO,CAAC,KAAK,CAAC;SACd,QAAQ,CAAC,8FAA8F,CAAC;CAC1G,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,YAAY,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACnD,MAAM,EAAE,CAAC;SACP,MAAM,EAAE;SACR,QAAQ,CAAC,iGAAiG,CAAC;IAE7G,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;IACrF,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;IACvF,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2CAA2C,CAAC;IACnF,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,QAAQ,CAAC,iBAAiB,CAAC;IACvF,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,0CAA0C,CAAC;IACzF,OAAO,EAAE,CAAC;SACR,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SAClB,QAAQ,EAAE;SACV,QAAQ,CAAC,0EAA0E,CAAC;IACtF,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC;IACtE,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,8FAA8F,CAAC;CACtJ,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,YAAY,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACnD,GAAG,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,uCAAuC,CAAC;IAC5F,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2EAA2E,CAAC;CACnH,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,cAAc,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACrD,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;IACxD,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,yCAAyC,CAAC;CAC3F,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACxD,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,sBAAsB,CAAC;CACnF,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,gBAAgB,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACvD,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;CAC/C,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,sBAAsB,GAAG,aAAa,CAAC,MAAM,CAAC;IAC1D,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,sFAAsF,CAAC;IACrH,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;CACpF,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,qBAAqB,GAAG,YAAY,CAAC,MAAM,CAAC;IACxD,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wCAAwC,CAAC;IACvE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;CACpF,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,qBAAqB,GAAG,gBAAgB,CAAC,MAAM,CAAC;IAC5D,GAAG,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,mDAAmD,CAAC;CACxG,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,sBAAsB,GAAG,gBAAgB,CAAC,MAAM,CAAC;IAC7D,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;CACzD,CAAC,CAAC"}
@@ -15,7 +15,7 @@ export declare const REPO_SEARCH_TOOL_CONFIG: {
15
15
  limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
16
16
  }, "strip", z.ZodTypeAny, {
17
17
  limit: number;
18
- repo_types: ("dataset" | "space" | "model")[];
18
+ repo_types: ("space" | "dataset" | "model")[];
19
19
  sort?: "createdAt" | "downloads" | "lastModified" | "likes" | "trendingScore" | undefined;
20
20
  author?: string | undefined;
21
21
  query?: string | undefined;
@@ -25,7 +25,7 @@ export declare const REPO_SEARCH_TOOL_CONFIG: {
25
25
  author?: string | undefined;
26
26
  query?: string | undefined;
27
27
  limit?: number | undefined;
28
- repo_types?: ("dataset" | "space" | "model")[] | undefined;
28
+ repo_types?: ("space" | "dataset" | "model")[] | undefined;
29
29
  filters?: string[] | undefined;
30
30
  }>;
31
31
  readonly annotations: {
@@ -1,6 +1,7 @@
1
1
  export declare const SPACE_SEARCH_TOOL_ID: "space_search";
2
2
  export declare const MODEL_SEARCH_TOOL_ID: "model_search";
3
3
  export declare const REPO_SEARCH_TOOL_ID: "hub_repo_search";
4
+ export declare const CREATE_REPO_TOOL_ID: "create_repo";
4
5
  export declare const MODEL_DETAIL_TOOL_ID: "model_details";
5
6
  export declare const PAPER_SEARCH_TOOL_ID: "paper_search";
6
7
  export declare const DATASET_SEARCH_TOOL_ID: "dataset_search";
@@ -18,15 +19,15 @@ export declare const MODEL_DETAIL_PROMPT_ID: string;
18
19
  export declare const DATASET_DETAIL_PROMPT_ID: string;
19
20
  export declare const HF_JOBS_TOOL_ID: "hf_jobs";
20
21
  export declare const DYNAMIC_SPACE_TOOL_ID: "dynamic_space";
21
- export declare const ALL_BUILTIN_TOOL_IDS: readonly ["space_search", "model_search", "hub_repo_search", "model_details", "paper_search", "dataset_search", "dataset_details", "hub_repo_details", "duplicate_space", "space_info", "space_files", "hf_doc_search", "hf_doc_fetch", "use_space", "hf_jobs", "dynamic_space"];
22
+ export declare const ALL_BUILTIN_TOOL_IDS: readonly ["space_search", "model_search", "hub_repo_search", "create_repo", "model_details", "paper_search", "dataset_search", "dataset_details", "hub_repo_details", "duplicate_space", "space_info", "space_files", "hf_doc_search", "hf_doc_fetch", "use_space", "hf_jobs", "dynamic_space"];
22
23
  export declare const TOOL_ID_GROUPS: {
23
24
  readonly search: readonly ["space_search", "hub_repo_search", "paper_search", "hf_doc_search"];
24
25
  readonly spaces: readonly ["space_search", "duplicate_space", "space_info", "space_files", "use_space"];
25
26
  readonly detail: readonly ["model_details", "dataset_details", "hub_repo_details"];
26
27
  readonly docs: readonly ["hf_doc_search", "hf_doc_fetch"];
27
- readonly hf_api: readonly ["space_search", "hub_repo_search", "paper_search", "hub_repo_details", "hf_doc_search"];
28
+ readonly hf_api: readonly ["space_search", "hub_repo_search", "create_repo", "paper_search", "hub_repo_details", "hf_doc_search"];
28
29
  readonly dynamic_space: readonly ["dynamic_space"];
29
- readonly all: readonly ["space_search", "model_search", "hub_repo_search", "model_details", "paper_search", "dataset_search", "dataset_details", "hub_repo_details", "duplicate_space", "space_info", "space_files", "hf_doc_search", "hf_doc_fetch", "use_space", "hf_jobs", "dynamic_space"];
30
+ readonly all: readonly ["space_search", "model_search", "hub_repo_search", "create_repo", "model_details", "paper_search", "dataset_search", "dataset_details", "hub_repo_details", "duplicate_space", "space_info", "space_files", "hf_doc_search", "hf_doc_fetch", "use_space", "hf_jobs", "dynamic_space"];
30
31
  };
31
32
  export type BuiltinToolId = (typeof ALL_BUILTIN_TOOL_IDS)[number];
32
33
  export declare function isValidBuiltinToolId(toolId: string): toolId is BuiltinToolId;
@@ -1 +1 @@
1
- {"version":3,"file":"tool-ids.d.ts","sourceRoot":"","sources":["../src/tool-ids.ts"],"names":[],"mappings":"AA6BA,eAAO,MAAM,oBAAoB,gBAAmC,CAAC;AACrE,eAAO,MAAM,oBAAoB,gBAAgC,CAAC;AAClE,eAAO,MAAM,mBAAmB,mBAA+B,CAAC;AAChE,eAAO,MAAM,oBAAoB,iBAAgC,CAAC;AAClE,eAAO,MAAM,oBAAoB,gBAAgC,CAAC;AAClE,eAAO,MAAM,sBAAsB,kBAAkC,CAAC;AACtE,eAAO,MAAM,sBAAsB,mBAAkC,CAAC;AACtE,eAAO,MAAM,wBAAwB,oBAAoC,CAAC;AAC1E,eAAO,MAAM,uBAAuB,mBAAmC,CAAC;AACxE,eAAO,MAAM,kBAAkB,cAA8B,CAAC;AAC9D,eAAO,MAAM,mBAAmB,eAA+B,CAAC;AAChE,eAAO,MAAM,iBAAiB,aAA6B,CAAC;AAC5D,eAAO,MAAM,4BAA4B,iBAAmC,CAAC;AAC7E,eAAO,MAAM,iBAAiB,gBAAwB,CAAC;AACvD,eAAO,MAAM,sBAAsB,gBAAkC,CAAC;AACtE,eAAO,MAAM,uBAAuB,iBAAmC,CAAC;AACxE,eAAO,MAAM,sBAAsB,QAAkC,CAAC;AACtE,eAAO,MAAM,wBAAwB,QAAoC,CAAC;AAC1E,eAAO,MAAM,eAAe,WAA2B,CAAC;AACxD,eAAO,MAAM,qBAAqB,iBAAiC,CAAC;AAGpE,eAAO,MAAM,oBAAoB,kRAiBvB,CAAC;AAEX,eAAO,MAAM,cAAc;;;;;;;;CAqBjB,CAAC;AAGX,MAAM,MAAM,aAAa,GAAG,CAAC,OAAO,oBAAoB,CAAC,CAAC,MAAM,CAAC,CAAC;AAGlE,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,IAAI,aAAa,CAE5E"}
1
+ {"version":3,"file":"tool-ids.d.ts","sourceRoot":"","sources":["../src/tool-ids.ts"],"names":[],"mappings":"AA8BA,eAAO,MAAM,oBAAoB,gBAAmC,CAAC;AACrE,eAAO,MAAM,oBAAoB,gBAAgC,CAAC;AAClE,eAAO,MAAM,mBAAmB,mBAA+B,CAAC;AAChE,eAAO,MAAM,mBAAmB,eAA+B,CAAC;AAChE,eAAO,MAAM,oBAAoB,iBAAgC,CAAC;AAClE,eAAO,MAAM,oBAAoB,gBAAgC,CAAC;AAClE,eAAO,MAAM,sBAAsB,kBAAkC,CAAC;AACtE,eAAO,MAAM,sBAAsB,mBAAkC,CAAC;AACtE,eAAO,MAAM,wBAAwB,oBAAoC,CAAC;AAC1E,eAAO,MAAM,uBAAuB,mBAAmC,CAAC;AACxE,eAAO,MAAM,kBAAkB,cAA8B,CAAC;AAC9D,eAAO,MAAM,mBAAmB,eAA+B,CAAC;AAChE,eAAO,MAAM,iBAAiB,aAA6B,CAAC;AAC5D,eAAO,MAAM,4BAA4B,iBAAmC,CAAC;AAC7E,eAAO,MAAM,iBAAiB,gBAAwB,CAAC;AACvD,eAAO,MAAM,sBAAsB,gBAAkC,CAAC;AACtE,eAAO,MAAM,uBAAuB,iBAAmC,CAAC;AACxE,eAAO,MAAM,sBAAsB,QAAkC,CAAC;AACtE,eAAO,MAAM,wBAAwB,QAAoC,CAAC;AAC1E,eAAO,MAAM,eAAe,WAA2B,CAAC;AACxD,eAAO,MAAM,qBAAqB,iBAAiC,CAAC;AAGpE,eAAO,MAAM,oBAAoB,iSAkBvB,CAAC;AAEX,eAAO,MAAM,cAAc;;;;;;;;CAsBjB,CAAC;AAGX,MAAM,MAAM,aAAa,GAAG,CAAC,OAAO,oBAAoB,CAAC,CAAC,MAAM,CAAC,CAAC;AAGlE,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,IAAI,aAAa,CAE5E"}
package/dist/tool-ids.js CHANGED
@@ -1,7 +1,8 @@
1
- import { SEMANTIC_SEARCH_TOOL_CONFIG, MODEL_SEARCH_TOOL_CONFIG, MODEL_DETAIL_TOOL_CONFIG, MODEL_DETAIL_PROMPT_CONFIG, PAPER_SEARCH_TOOL_CONFIG, REPO_SEARCH_TOOL_CONFIG, DATASET_SEARCH_TOOL_CONFIG, DATASET_DETAIL_TOOL_CONFIG, DATASET_DETAIL_PROMPT_CONFIG, HUB_REPO_DETAILS_TOOL_CONFIG, DUPLICATE_SPACE_TOOL_CONFIG, SPACE_INFO_TOOL_CONFIG, SPACE_FILES_TOOL_CONFIG, USER_SUMMARY_PROMPT_CONFIG, PAPER_SUMMARY_PROMPT_CONFIG, DOCS_SEMANTIC_SEARCH_CONFIG, DOC_FETCH_CONFIG, USE_SPACE_TOOL_CONFIG, HF_JOBS_TOOL_CONFIG, DYNAMIC_SPACE_TOOL_CONFIG, } from './index.js';
1
+ import { SEMANTIC_SEARCH_TOOL_CONFIG, MODEL_SEARCH_TOOL_CONFIG, MODEL_DETAIL_TOOL_CONFIG, MODEL_DETAIL_PROMPT_CONFIG, PAPER_SEARCH_TOOL_CONFIG, REPO_SEARCH_TOOL_CONFIG, CREATE_REPO_TOOL_CONFIG, DATASET_SEARCH_TOOL_CONFIG, DATASET_DETAIL_TOOL_CONFIG, DATASET_DETAIL_PROMPT_CONFIG, HUB_REPO_DETAILS_TOOL_CONFIG, DUPLICATE_SPACE_TOOL_CONFIG, SPACE_INFO_TOOL_CONFIG, SPACE_FILES_TOOL_CONFIG, USER_SUMMARY_PROMPT_CONFIG, PAPER_SUMMARY_PROMPT_CONFIG, DOCS_SEMANTIC_SEARCH_CONFIG, DOC_FETCH_CONFIG, USE_SPACE_TOOL_CONFIG, HF_JOBS_TOOL_CONFIG, DYNAMIC_SPACE_TOOL_CONFIG, } from './index.js';
2
2
  export const SPACE_SEARCH_TOOL_ID = SEMANTIC_SEARCH_TOOL_CONFIG.name;
3
3
  export const MODEL_SEARCH_TOOL_ID = MODEL_SEARCH_TOOL_CONFIG.name;
4
4
  export const REPO_SEARCH_TOOL_ID = REPO_SEARCH_TOOL_CONFIG.name;
5
+ export const CREATE_REPO_TOOL_ID = CREATE_REPO_TOOL_CONFIG.name;
5
6
  export const MODEL_DETAIL_TOOL_ID = MODEL_DETAIL_TOOL_CONFIG.name;
6
7
  export const PAPER_SEARCH_TOOL_ID = PAPER_SEARCH_TOOL_CONFIG.name;
7
8
  export const DATASET_SEARCH_TOOL_ID = DATASET_SEARCH_TOOL_CONFIG.name;
@@ -23,6 +24,7 @@ export const ALL_BUILTIN_TOOL_IDS = [
23
24
  SPACE_SEARCH_TOOL_ID,
24
25
  MODEL_SEARCH_TOOL_ID,
25
26
  REPO_SEARCH_TOOL_ID,
27
+ CREATE_REPO_TOOL_ID,
26
28
  MODEL_DETAIL_TOOL_ID,
27
29
  PAPER_SEARCH_TOOL_ID,
28
30
  DATASET_SEARCH_TOOL_ID,
@@ -51,6 +53,7 @@ export const TOOL_ID_GROUPS = {
51
53
  hf_api: [
52
54
  SPACE_SEARCH_TOOL_ID,
53
55
  REPO_SEARCH_TOOL_ID,
56
+ CREATE_REPO_TOOL_ID,
54
57
  PAPER_SEARCH_TOOL_ID,
55
58
  HUB_REPO_DETAILS_TOOL_ID,
56
59
  DOCS_SEMANTIC_SEARCH_TOOL_ID,
@@ -1 +1 @@
1
- {"version":3,"file":"tool-ids.js","sourceRoot":"","sources":["../src/tool-ids.ts"],"names":[],"mappings":"AAKA,OAAO,EACN,2BAA2B,EAC3B,wBAAwB,EACxB,wBAAwB,EACxB,0BAA0B,EAC1B,wBAAwB,EACxB,uBAAuB,EACvB,0BAA0B,EAC1B,0BAA0B,EAC1B,4BAA4B,EAC5B,4BAA4B,EAC5B,2BAA2B,EAC3B,sBAAsB,EACtB,uBAAuB,EACvB,0BAA0B,EAC1B,2BAA2B,EAC3B,2BAA2B,EAC3B,gBAAgB,EAChB,qBAAqB,EACrB,mBAAmB,EACnB,yBAAyB,GACzB,MAAM,YAAY,CAAC;AAGpB,MAAM,CAAC,MAAM,oBAAoB,GAAG,2BAA2B,CAAC,IAAI,CAAC;AACrE,MAAM,CAAC,MAAM,oBAAoB,GAAG,wBAAwB,CAAC,IAAI,CAAC;AAClE,MAAM,CAAC,MAAM,mBAAmB,GAAG,uBAAuB,CAAC,IAAI,CAAC;AAChE,MAAM,CAAC,MAAM,oBAAoB,GAAG,wBAAwB,CAAC,IAAI,CAAC;AAClE,MAAM,CAAC,MAAM,oBAAoB,GAAG,wBAAwB,CAAC,IAAI,CAAC;AAClE,MAAM,CAAC,MAAM,sBAAsB,GAAG,0BAA0B,CAAC,IAAI,CAAC;AACtE,MAAM,CAAC,MAAM,sBAAsB,GAAG,0BAA0B,CAAC,IAAI,CAAC;AACtE,MAAM,CAAC,MAAM,wBAAwB,GAAG,4BAA4B,CAAC,IAAI,CAAC;AAC1E,MAAM,CAAC,MAAM,uBAAuB,GAAG,2BAA2B,CAAC,IAAI,CAAC;AACxE,MAAM,CAAC,MAAM,kBAAkB,GAAG,sBAAsB,CAAC,IAAI,CAAC;AAC9D,MAAM,CAAC,MAAM,mBAAmB,GAAG,uBAAuB,CAAC,IAAI,CAAC;AAChE,MAAM,CAAC,MAAM,iBAAiB,GAAG,qBAAqB,CAAC,IAAI,CAAC;AAC5D,MAAM,CAAC,MAAM,4BAA4B,GAAG,2BAA2B,CAAC,IAAI,CAAC;AAC7E,MAAM,CAAC,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,IAAI,CAAC;AACvD,MAAM,CAAC,MAAM,sBAAsB,GAAG,0BAA0B,CAAC,IAAI,CAAC;AACtE,MAAM,CAAC,MAAM,uBAAuB,GAAG,2BAA2B,CAAC,IAAI,CAAC;AACxE,MAAM,CAAC,MAAM,sBAAsB,GAAG,0BAA0B,CAAC,IAAI,CAAC;AACtE,MAAM,CAAC,MAAM,wBAAwB,GAAG,4BAA4B,CAAC,IAAI,CAAC;AAC1E,MAAM,CAAC,MAAM,eAAe,GAAG,mBAAmB,CAAC,IAAI,CAAC;AACxD,MAAM,CAAC,MAAM,qBAAqB,GAAG,yBAAyB,CAAC,IAAI,CAAC;AAGpE,MAAM,CAAC,MAAM,oBAAoB,GAAG;IACnC,oBAAoB;IACpB,oBAAoB;IACpB,mBAAmB;IACnB,oBAAoB;IACpB,oBAAoB;IACpB,sBAAsB;IACtB,sBAAsB;IACtB,wBAAwB;IACxB,uBAAuB;IACvB,kBAAkB;IAClB,mBAAmB;IACnB,4BAA4B;IAC5B,iBAAiB;IACjB,iBAAiB;IACjB,eAAe;IACf,qBAAqB;CACZ,CAAC;AAEX,MAAM,CAAC,MAAM,cAAc,GAAG;IAC7B,MAAM,EAAE,CAAC,oBAAoB,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,4BAA4B,CAAU;IAChH,MAAM,EAAE;QACP,oBAAoB;QACpB,uBAAuB;QACvB,kBAAkB;QAClB,mBAAmB;QACnB,iBAAiB;KACR;IACV,MAAM,EAAE,CAAC,oBAAoB,EAAE,sBAAsB,EAAE,wBAAwB,CAAU;IACzF,IAAI,EAAE,CAAC,4BAA4B,EAAE,iBAAiB,CAAU;IAChE,MAAM,EAAE;QACP,oBAAoB;QACpB,mBAAmB;QACnB,oBAAoB;QACpB,wBAAwB;QACxB,4BAA4B;KAEnB;IACV,aAAa,EAAE,CAAC,qBAAqB,CAAU;IAC/C,GAAG,EAAE,CAAC,GAAG,oBAAoB,CAAU;CAC9B,CAAC;AAMX,MAAM,UAAU,oBAAoB,CAAC,MAAc;IAClD,OAAQ,oBAA0C,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AACrE,CAAC"}
1
+ {"version":3,"file":"tool-ids.js","sourceRoot":"","sources":["../src/tool-ids.ts"],"names":[],"mappings":"AAKA,OAAO,EACN,2BAA2B,EAC3B,wBAAwB,EACxB,wBAAwB,EACxB,0BAA0B,EAC1B,wBAAwB,EACxB,uBAAuB,EACvB,uBAAuB,EACvB,0BAA0B,EAC1B,0BAA0B,EAC1B,4BAA4B,EAC5B,4BAA4B,EAC5B,2BAA2B,EAC3B,sBAAsB,EACtB,uBAAuB,EACvB,0BAA0B,EAC1B,2BAA2B,EAC3B,2BAA2B,EAC3B,gBAAgB,EAChB,qBAAqB,EACrB,mBAAmB,EACnB,yBAAyB,GACzB,MAAM,YAAY,CAAC;AAGpB,MAAM,CAAC,MAAM,oBAAoB,GAAG,2BAA2B,CAAC,IAAI,CAAC;AACrE,MAAM,CAAC,MAAM,oBAAoB,GAAG,wBAAwB,CAAC,IAAI,CAAC;AAClE,MAAM,CAAC,MAAM,mBAAmB,GAAG,uBAAuB,CAAC,IAAI,CAAC;AAChE,MAAM,CAAC,MAAM,mBAAmB,GAAG,uBAAuB,CAAC,IAAI,CAAC;AAChE,MAAM,CAAC,MAAM,oBAAoB,GAAG,wBAAwB,CAAC,IAAI,CAAC;AAClE,MAAM,CAAC,MAAM,oBAAoB,GAAG,wBAAwB,CAAC,IAAI,CAAC;AAClE,MAAM,CAAC,MAAM,sBAAsB,GAAG,0BAA0B,CAAC,IAAI,CAAC;AACtE,MAAM,CAAC,MAAM,sBAAsB,GAAG,0BAA0B,CAAC,IAAI,CAAC;AACtE,MAAM,CAAC,MAAM,wBAAwB,GAAG,4BAA4B,CAAC,IAAI,CAAC;AAC1E,MAAM,CAAC,MAAM,uBAAuB,GAAG,2BAA2B,CAAC,IAAI,CAAC;AACxE,MAAM,CAAC,MAAM,kBAAkB,GAAG,sBAAsB,CAAC,IAAI,CAAC;AAC9D,MAAM,CAAC,MAAM,mBAAmB,GAAG,uBAAuB,CAAC,IAAI,CAAC;AAChE,MAAM,CAAC,MAAM,iBAAiB,GAAG,qBAAqB,CAAC,IAAI,CAAC;AAC5D,MAAM,CAAC,MAAM,4BAA4B,GAAG,2BAA2B,CAAC,IAAI,CAAC;AAC7E,MAAM,CAAC,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,IAAI,CAAC;AACvD,MAAM,CAAC,MAAM,sBAAsB,GAAG,0BAA0B,CAAC,IAAI,CAAC;AACtE,MAAM,CAAC,MAAM,uBAAuB,GAAG,2BAA2B,CAAC,IAAI,CAAC;AACxE,MAAM,CAAC,MAAM,sBAAsB,GAAG,0BAA0B,CAAC,IAAI,CAAC;AACtE,MAAM,CAAC,MAAM,wBAAwB,GAAG,4BAA4B,CAAC,IAAI,CAAC;AAC1E,MAAM,CAAC,MAAM,eAAe,GAAG,mBAAmB,CAAC,IAAI,CAAC;AACxD,MAAM,CAAC,MAAM,qBAAqB,GAAG,yBAAyB,CAAC,IAAI,CAAC;AAGpE,MAAM,CAAC,MAAM,oBAAoB,GAAG;IACnC,oBAAoB;IACpB,oBAAoB;IACpB,mBAAmB;IACnB,mBAAmB;IACnB,oBAAoB;IACpB,oBAAoB;IACpB,sBAAsB;IACtB,sBAAsB;IACtB,wBAAwB;IACxB,uBAAuB;IACvB,kBAAkB;IAClB,mBAAmB;IACnB,4BAA4B;IAC5B,iBAAiB;IACjB,iBAAiB;IACjB,eAAe;IACf,qBAAqB;CACZ,CAAC;AAEX,MAAM,CAAC,MAAM,cAAc,GAAG;IAC7B,MAAM,EAAE,CAAC,oBAAoB,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,4BAA4B,CAAU;IAChH,MAAM,EAAE;QACP,oBAAoB;QACpB,uBAAuB;QACvB,kBAAkB;QAClB,mBAAmB;QACnB,iBAAiB;KACR;IACV,MAAM,EAAE,CAAC,oBAAoB,EAAE,sBAAsB,EAAE,wBAAwB,CAAU;IACzF,IAAI,EAAE,CAAC,4BAA4B,EAAE,iBAAiB,CAAU;IAChE,MAAM,EAAE;QACP,oBAAoB;QACpB,mBAAmB;QACnB,mBAAmB;QACnB,oBAAoB;QACpB,wBAAwB;QACxB,4BAA4B;KAEnB;IACV,aAAa,EAAE,CAAC,qBAAqB,CAAU;IAC/C,GAAG,EAAE,CAAC,GAAG,oBAAoB,CAAU;CAC9B,CAAC;AAMX,MAAM,UAAU,oBAAoB,CAAC,MAAc;IAClD,OAAQ,oBAA0C,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AACrE,CAAC"}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@llmindset/hf-mcp",
3
- "version": "0.3.10",
3
+ "version": "0.3.11",
4
4
  "type": "module",
5
5
  "main": "dist/index.js",
6
6
  "types": "dist/index.d.ts",
@@ -0,0 +1,182 @@
1
+ import { createRepo } from '@huggingface/hub';
2
+ import { beforeEach, describe, expect, it, vi } from 'vitest';
3
+ import { CreateRepoTool, formatCreateRepoResult } from './create-repo.js';
4
+
5
+ vi.mock('@huggingface/hub', () => ({
6
+ createRepo: vi.fn(),
7
+ }));
8
+
9
+ describe('CreateRepoTool', () => {
10
+ beforeEach(() => {
11
+ vi.mocked(createRepo).mockReset();
12
+ vi.mocked(createRepo).mockResolvedValue({
13
+ repoUrl: 'https://huggingface.co/alice/example-model',
14
+ id: '0123456789abcdef01234567',
15
+ });
16
+ });
17
+
18
+ it('creates a model repository with client-native repo params', async () => {
19
+ const tool = new CreateRepoTool('token');
20
+ const result = await tool.create({
21
+ name: 'alice/example-model',
22
+ repo_type: 'model',
23
+ });
24
+
25
+ expect(createRepo).toHaveBeenCalledWith({
26
+ accessToken: 'token',
27
+ repo: { name: 'alice/example-model', type: 'model' },
28
+ private: undefined,
29
+ });
30
+ expect(result).toEqual({
31
+ url: 'https://huggingface.co/alice/example-model',
32
+ name: 'alice/example-model',
33
+ repoType: 'model',
34
+ id: '0123456789abcdef01234567',
35
+ });
36
+ });
37
+
38
+ it('defaults to model repositories', async () => {
39
+ const tool = new CreateRepoTool('token');
40
+ await tool.create({
41
+ name: 'alice/default-model',
42
+ });
43
+
44
+ expect(createRepo).toHaveBeenCalledWith({
45
+ accessToken: 'token',
46
+ repo: { name: 'alice/default-model', type: 'model' },
47
+ private: undefined,
48
+ });
49
+ });
50
+
51
+ it('creates a dataset repository', async () => {
52
+ const tool = new CreateRepoTool('token');
53
+ await tool.create({
54
+ name: 'alice/example-dataset',
55
+ repo_type: 'dataset',
56
+ private: true,
57
+ });
58
+
59
+ expect(createRepo).toHaveBeenCalledWith({
60
+ accessToken: 'token',
61
+ repo: { name: 'alice/example-dataset', type: 'dataset' },
62
+ private: true,
63
+ });
64
+ });
65
+
66
+ it('creates a bucket repository', async () => {
67
+ vi.mocked(createRepo).mockResolvedValue({
68
+ repoUrl: 'https://huggingface.co/buckets/alice/example-bucket',
69
+ id: 'bucket-id',
70
+ });
71
+ const tool = new CreateRepoTool('token');
72
+ const result = await tool.create({
73
+ name: 'alice/example-bucket',
74
+ repo_type: 'bucket',
75
+ private: true,
76
+ });
77
+
78
+ expect(createRepo).toHaveBeenCalledWith({
79
+ accessToken: 'token',
80
+ repo: { name: 'alice/example-bucket', type: 'bucket' },
81
+ private: true,
82
+ });
83
+ expect(result).toEqual({
84
+ url: 'https://huggingface.co/buckets/alice/example-bucket',
85
+ name: 'alice/example-bucket',
86
+ repoType: 'bucket',
87
+ id: 'bucket-id',
88
+ });
89
+ });
90
+
91
+ it('creates a Space repository with sdk', async () => {
92
+ vi.mocked(createRepo).mockResolvedValue({
93
+ repoUrl: 'https://huggingface.co/spaces/alice/demo',
94
+ id: 'abcdefabcdefabcdefabcdef',
95
+ });
96
+ const tool = new CreateRepoTool('token');
97
+ await tool.create({
98
+ name: 'alice/demo',
99
+ repo_type: 'space',
100
+ sdk: 'gradio',
101
+ });
102
+
103
+ expect(createRepo).toHaveBeenCalledWith({
104
+ accessToken: 'token',
105
+ repo: { name: 'alice/demo', type: 'space' },
106
+ private: undefined,
107
+ sdk: 'gradio',
108
+ });
109
+ });
110
+
111
+ it('creates a Streamlit Space repository', async () => {
112
+ vi.mocked(createRepo).mockResolvedValue({
113
+ repoUrl: 'https://huggingface.co/spaces/alice/streamlit-demo',
114
+ id: 'streamlit-space-id',
115
+ });
116
+ const tool = new CreateRepoTool('token');
117
+ await tool.create({
118
+ name: 'alice/streamlit-demo',
119
+ repo_type: 'space',
120
+ sdk: 'streamlit',
121
+ });
122
+
123
+ expect(createRepo).toHaveBeenCalledWith({
124
+ accessToken: 'token',
125
+ repo: { name: 'alice/streamlit-demo', type: 'space' },
126
+ private: undefined,
127
+ sdk: 'streamlit',
128
+ });
129
+ });
130
+
131
+ it('requires sdk for Space repositories', async () => {
132
+ const tool = new CreateRepoTool('token');
133
+ await expect(
134
+ tool.create({
135
+ name: 'alice/demo',
136
+ repo_type: 'space',
137
+ })
138
+ ).rejects.toThrow('sdk is required when repo_type is space');
139
+ expect(createRepo).not.toHaveBeenCalled();
140
+ });
141
+
142
+ it('requires fully-qualified repo names', async () => {
143
+ const tool = new CreateRepoTool('token');
144
+ await expect(
145
+ tool.create({
146
+ name: 'example-model',
147
+ repo_type: 'model',
148
+ })
149
+ ).rejects.toThrow("name must be fully qualified in 'namespace/repo-name' format");
150
+ expect(createRepo).not.toHaveBeenCalled();
151
+ });
152
+
153
+ it('requires an auth token', async () => {
154
+ const tool = new CreateRepoTool(undefined);
155
+ await expect(
156
+ tool.create({
157
+ name: 'alice/example-model',
158
+ repo_type: 'model',
159
+ })
160
+ ).rejects.toThrow('Requires Authentication');
161
+ expect(createRepo).not.toHaveBeenCalled();
162
+ });
163
+
164
+ it('formats the created repository result', () => {
165
+ expect(
166
+ formatCreateRepoResult({
167
+ url: 'https://huggingface.co/alice/example-model',
168
+ name: 'alice/example-model',
169
+ repoType: 'model',
170
+ id: '0123456789abcdef01234567',
171
+ })
172
+ ).toBe(
173
+ [
174
+ 'Repository created.',
175
+ 'Name: alice/example-model',
176
+ 'Type: model',
177
+ 'URL: https://huggingface.co/alice/example-model',
178
+ 'ID: 0123456789abcdef01234567',
179
+ ].join('\n')
180
+ );
181
+ });
182
+ });
@@ -0,0 +1,103 @@
1
+ import { createRepo, type RepoType as HubRepoType, type SpaceSdk } from '@huggingface/hub';
2
+ import { z } from 'zod';
3
+ import { NO_TOKEN_INSTRUCTIONS } from './utilities.js';
4
+
5
+ const REPO_TYPES = ['model', 'dataset', 'space', 'bucket'] as const satisfies readonly HubRepoType[];
6
+ const SPACE_SDKS = ['streamlit', 'gradio', 'docker', 'static'] as const satisfies readonly SpaceSdk[];
7
+
8
+ export const CREATE_REPO_TOOL_CONFIG = {
9
+ name: 'create_repo',
10
+ description: '',
11
+ schema: z.object({
12
+ name: z.string().min(1).describe("Fully-qualified repository name in 'namespace/repo-name' format."),
13
+ repo_type: z.enum(REPO_TYPES).optional().default('bucket').describe('Repository type. Defaults to bucket.'),
14
+ private: z.boolean().optional().describe('Whether to create the repository as private.'),
15
+ sdk: z.enum(SPACE_SDKS).optional().default('static').describe("SDK type - only required for repo_type='space'."),
16
+ }),
17
+ annotations: {
18
+ title: 'Create Hugging Face Repository',
19
+ destructiveHint: false,
20
+ readOnlyHint: false,
21
+ openWorldHint: true,
22
+ },
23
+ } as const;
24
+
25
+ export type CreateRepoParams = z.input<typeof CREATE_REPO_TOOL_CONFIG.schema>;
26
+ type RepoType = (typeof REPO_TYPES)[number];
27
+ type SupportedSpaceSdk = (typeof SPACE_SDKS)[number];
28
+
29
+ function assertExhaustiveUnion<T extends never>(_value?: T): void {
30
+ void _value;
31
+ }
32
+
33
+ assertExhaustiveUnion<Exclude<HubRepoType, RepoType>>();
34
+ assertExhaustiveUnion<Exclude<RepoType, HubRepoType>>();
35
+ assertExhaustiveUnion<Exclude<SpaceSdk, SupportedSpaceSdk>>();
36
+ assertExhaustiveUnion<Exclude<SupportedSpaceSdk, SpaceSdk>>();
37
+
38
+ export interface CreateRepoResult {
39
+ url: string;
40
+ name: string;
41
+ repoType: RepoType;
42
+ id: string;
43
+ }
44
+
45
+ export class CreateRepoTool {
46
+ constructor(private readonly hfToken?: string) {}
47
+
48
+ static createToolConfig(): Omit<typeof CREATE_REPO_TOOL_CONFIG, 'description'> & { description: string } {
49
+ return {
50
+ ...CREATE_REPO_TOOL_CONFIG,
51
+ description: 'Create a Hugging Face model, dataset, space, or bucket repository.',
52
+ };
53
+ }
54
+
55
+ async create(params: CreateRepoParams): Promise<CreateRepoResult> {
56
+ if (!this.hfToken) throw new Error(NO_TOKEN_INSTRUCTIONS);
57
+
58
+ const repoType = params.repo_type ?? 'model';
59
+ validateParams(params, repoType);
60
+
61
+ const result = await createRepo({
62
+ accessToken: this.hfToken,
63
+ repo: {
64
+ name: params.name,
65
+ type: repoType,
66
+ },
67
+ private: params.private,
68
+ ...(repoType === 'space' ? { sdk: params.sdk } : {}),
69
+ });
70
+
71
+ return {
72
+ url: result.repoUrl,
73
+ name: params.name,
74
+ repoType,
75
+ id: result.id,
76
+ };
77
+ }
78
+ }
79
+
80
+ function validateParams(params: CreateRepoParams, repoType: RepoType): void {
81
+ if (!isFullyQualifiedRepoName(params.name)) {
82
+ throw new Error("name must be fully qualified in 'namespace/repo-name' format.");
83
+ }
84
+
85
+ if (repoType === 'space' && !params.sdk) {
86
+ throw new Error('sdk is required when repo_type is space.');
87
+ }
88
+ }
89
+
90
+ function isFullyQualifiedRepoName(name: string): boolean {
91
+ const parts = name.split('/');
92
+ return parts.length === 2 && parts.every((part) => part.length > 0);
93
+ }
94
+
95
+ export const formatCreateRepoResult = (result: CreateRepoResult): string => {
96
+ return [
97
+ 'Repository created.',
98
+ `Name: ${result.name}`,
99
+ `Type: ${result.repoType}`,
100
+ `URL: ${result.url}`,
101
+ `ID: ${result.id}`,
102
+ ].join('\n');
103
+ };
@@ -44,6 +44,19 @@ export const REPO_SEARCH_TOOL_CONFIG: BrowserToolConfig = {
44
44
  },
45
45
  };
46
46
 
47
+ export const CREATE_REPO_TOOL_CONFIG: BrowserToolConfig = {
48
+ name: 'create_repo',
49
+ description:
50
+ 'Create a Hugging Face model, dataset, Space, or bucket repository. ' +
51
+ "name must be fully qualified, for example 'username/repo-name'.",
52
+ annotations: {
53
+ title: 'Create Hugging Face Repository',
54
+ destructiveHint: false,
55
+ readOnlyHint: false,
56
+ openWorldHint: true,
57
+ },
58
+ };
59
+
47
60
  export const PAPER_SEARCH_TOOL_CONFIG: BrowserToolConfig = {
48
61
  name: 'paper_search',
49
62
  description:
@@ -121,6 +134,7 @@ export const DOC_FETCH_CONFIG: BrowserToolConfig = {
121
134
  export const SPACE_SEARCH_TOOL_ID = SEMANTIC_SEARCH_TOOL_CONFIG.name;
122
135
  export const MODEL_SEARCH_TOOL_ID = 'model_search';
123
136
  export const REPO_SEARCH_TOOL_ID = REPO_SEARCH_TOOL_CONFIG.name;
137
+ export const CREATE_REPO_TOOL_ID = CREATE_REPO_TOOL_CONFIG.name;
124
138
  export const MODEL_DETAIL_TOOL_ID = 'model_details';
125
139
  export const PAPER_SEARCH_TOOL_ID = PAPER_SEARCH_TOOL_CONFIG.name;
126
140
  export const DATASET_SEARCH_TOOL_ID = 'dataset_search';
@@ -139,6 +153,7 @@ export const ALL_BUILTIN_TOOL_IDS = [
139
153
  SPACE_SEARCH_TOOL_ID,
140
154
  MODEL_SEARCH_TOOL_ID,
141
155
  REPO_SEARCH_TOOL_ID,
156
+ CREATE_REPO_TOOL_ID,
142
157
  MODEL_DETAIL_TOOL_ID,
143
158
  PAPER_SEARCH_TOOL_ID,
144
159
  DATASET_SEARCH_TOOL_ID,
@@ -168,6 +183,7 @@ export const TOOL_ID_GROUPS = {
168
183
  hf_api: [
169
184
  SPACE_SEARCH_TOOL_ID,
170
185
  REPO_SEARCH_TOOL_ID,
186
+ CREATE_REPO_TOOL_ID,
171
187
  PAPER_SEARCH_TOOL_ID,
172
188
  HUB_REPO_DETAILS_TOOL_ID,
173
189
  DOCS_SEMANTIC_SEARCH_TOOL_ID,
package/src/index.ts CHANGED
@@ -8,6 +8,7 @@ export * from './utilities.js';
8
8
  export * from './paper-search.js';
9
9
  export * from './dataset-search.js';
10
10
  export * from './repo-search.js';
11
+ export * from './create-repo.js';
11
12
  export * from './dataset-detail.js';
12
13
  export * from './hub-inspect.js';
13
14
  export * from './duplicate-space.js';
package/src/jobs/types.ts CHANGED
@@ -1,3 +1,4 @@
1
+ import type { SpaceHardwareFlavor } from '@huggingface/hub';
1
2
  import { z } from 'zod';
2
3
 
3
4
  /**
@@ -20,8 +21,8 @@ export const GPU_FLAVORS = [
20
21
  'a10g-largex2',
21
22
  'a10g-largex4',
22
23
  'a100-large',
23
- 'h100',
24
- 'h100x8',
24
+ 'a100x4',
25
+ 'a100x8',
25
26
  ] as const;
26
27
 
27
28
  export const SPECIALIZED_FLAVORS = ['inf2x6'] as const;
@@ -30,6 +31,13 @@ export const ALL_FLAVORS = [...CPU_FLAVORS, ...GPU_FLAVORS, ...SPECIALIZED_FLAVO
30
31
 
31
32
  export type JobFlavor = (typeof ALL_FLAVORS)[number];
32
33
 
34
+ function assertExhaustiveHardwareUnion<T extends never>(_value?: T): void {
35
+ void _value;
36
+ }
37
+
38
+ assertExhaustiveHardwareUnion<Exclude<SpaceHardwareFlavor, JobFlavor>>();
39
+ assertExhaustiveHardwareUnion<Exclude<JobFlavor, SpaceHardwareFlavor>>();
40
+
33
41
  /**
34
42
  * Job status stages (from OpenAPI spec)
35
43
  */
package/src/tool-ids.ts CHANGED
@@ -10,6 +10,7 @@ import {
10
10
  MODEL_DETAIL_PROMPT_CONFIG,
11
11
  PAPER_SEARCH_TOOL_CONFIG,
12
12
  REPO_SEARCH_TOOL_CONFIG,
13
+ CREATE_REPO_TOOL_CONFIG,
13
14
  DATASET_SEARCH_TOOL_CONFIG,
14
15
  DATASET_DETAIL_TOOL_CONFIG,
15
16
  DATASET_DETAIL_PROMPT_CONFIG,
@@ -30,6 +31,7 @@ import {
30
31
  export const SPACE_SEARCH_TOOL_ID = SEMANTIC_SEARCH_TOOL_CONFIG.name;
31
32
  export const MODEL_SEARCH_TOOL_ID = MODEL_SEARCH_TOOL_CONFIG.name;
32
33
  export const REPO_SEARCH_TOOL_ID = REPO_SEARCH_TOOL_CONFIG.name;
34
+ export const CREATE_REPO_TOOL_ID = CREATE_REPO_TOOL_CONFIG.name;
33
35
  export const MODEL_DETAIL_TOOL_ID = MODEL_DETAIL_TOOL_CONFIG.name;
34
36
  export const PAPER_SEARCH_TOOL_ID = PAPER_SEARCH_TOOL_CONFIG.name;
35
37
  export const DATASET_SEARCH_TOOL_ID = DATASET_SEARCH_TOOL_CONFIG.name;
@@ -53,6 +55,7 @@ export const ALL_BUILTIN_TOOL_IDS = [
53
55
  SPACE_SEARCH_TOOL_ID,
54
56
  MODEL_SEARCH_TOOL_ID,
55
57
  REPO_SEARCH_TOOL_ID,
58
+ CREATE_REPO_TOOL_ID,
56
59
  MODEL_DETAIL_TOOL_ID,
57
60
  PAPER_SEARCH_TOOL_ID,
58
61
  DATASET_SEARCH_TOOL_ID,
@@ -82,6 +85,7 @@ export const TOOL_ID_GROUPS = {
82
85
  hf_api: [
83
86
  SPACE_SEARCH_TOOL_ID,
84
87
  REPO_SEARCH_TOOL_ID,
88
+ CREATE_REPO_TOOL_ID,
85
89
  PAPER_SEARCH_TOOL_ID,
86
90
  HUB_REPO_DETAILS_TOOL_ID,
87
91
  DOCS_SEMANTIC_SEARCH_TOOL_ID,