@llmindset/hf-mcp 0.2.32 → 0.2.34
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/docs-search/docs-semantic-search.d.ts +2 -2
- package/dist/docs-search/docs-semantic-search.d.ts.map +1 -1
- package/dist/docs-search/docs-semantic-search.js +56 -21
- package/dist/docs-search/docs-semantic-search.js.map +1 -1
- package/dist/hf-api-call.d.ts.map +1 -1
- package/dist/hf-api-call.js +4 -0
- package/dist/hf-api-call.js.map +1 -1
- package/dist/index.d.ts +1 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +1 -0
- package/dist/index.js.map +1 -1
- package/dist/jobs/api-client.d.ts +19 -0
- package/dist/jobs/api-client.d.ts.map +1 -0
- package/dist/jobs/api-client.js +104 -0
- package/dist/jobs/api-client.js.map +1 -0
- package/dist/jobs/commands/inspect.d.ts +5 -0
- package/dist/jobs/commands/inspect.d.ts.map +1 -0
- package/dist/jobs/commands/inspect.js +21 -0
- package/dist/jobs/commands/inspect.js.map +1 -0
- package/dist/jobs/commands/logs.d.ts +4 -0
- package/dist/jobs/commands/logs.d.ts.map +1 -0
- package/dist/jobs/commands/logs.js +24 -0
- package/dist/jobs/commands/logs.js.map +1 -0
- package/dist/jobs/commands/ps.d.ts +4 -0
- package/dist/jobs/commands/ps.d.ts.map +1 -0
- package/dist/jobs/commands/ps.js +23 -0
- package/dist/jobs/commands/ps.js.map +1 -0
- package/dist/jobs/commands/run.d.ts +5 -0
- package/dist/jobs/commands/run.d.ts.map +1 -0
- package/dist/jobs/commands/run.js +89 -0
- package/dist/jobs/commands/run.js.map +1 -0
- package/dist/jobs/commands/scheduled.d.ts +10 -0
- package/dist/jobs/commands/scheduled.d.ts.map +1 -0
- package/dist/jobs/commands/scheduled.js +111 -0
- package/dist/jobs/commands/scheduled.js.map +1 -0
- package/dist/jobs/commands/utils.d.ts +19 -0
- package/dist/jobs/commands/utils.d.ts.map +1 -0
- package/dist/jobs/commands/utils.js +99 -0
- package/dist/jobs/commands/utils.js.map +1 -0
- package/dist/jobs/formatters.d.ts +6 -0
- package/dist/jobs/formatters.d.ts.map +1 -0
- package/dist/jobs/formatters.js +98 -0
- package/dist/jobs/formatters.js.map +1 -0
- package/dist/jobs/sse-handler.d.ts +12 -0
- package/dist/jobs/sse-handler.d.ts.map +1 -0
- package/dist/jobs/sse-handler.js +80 -0
- package/dist/jobs/sse-handler.js.map +1 -0
- package/dist/jobs/tool.d.ts +35 -0
- package/dist/jobs/tool.d.ts.map +1 -0
- package/dist/jobs/tool.js +333 -0
- package/dist/jobs/tool.js.map +1 -0
- package/dist/jobs/types.d.ts +295 -0
- package/dist/jobs/types.d.ts.map +1 -0
- package/dist/jobs/types.js +95 -0
- package/dist/jobs/types.js.map +1 -0
- package/dist/tool-ids.d.ts +3 -2
- package/dist/tool-ids.d.ts.map +1 -1
- package/dist/tool-ids.js +10 -2
- package/dist/tool-ids.js.map +1 -1
- package/dist/types/tool-result.d.ts +1 -0
- package/dist/types/tool-result.d.ts.map +1 -1
- package/dist/use-space.d.ts +0 -1
- package/dist/use-space.d.ts.map +1 -1
- package/dist/use-space.js +0 -1
- package/dist/use-space.js.map +1 -1
- package/package.json +4 -2
- package/src/docs-search/docs-semantic-search.ts +71 -20
- package/src/hf-api-call.ts +6 -0
- package/src/index.ts +1 -0
- package/src/jobs/api-client.ts +195 -0
- package/src/jobs/commands/inspect.ts +38 -0
- package/src/jobs/commands/logs.ts +36 -0
- package/src/jobs/commands/ps.ts +40 -0
- package/src/jobs/commands/run.ts +134 -0
- package/src/jobs/commands/scheduled.ts +189 -0
- package/src/jobs/commands/utils.ts +158 -0
- package/src/jobs/formatters.ts +149 -0
- package/src/jobs/sse-handler.ts +144 -0
- package/src/jobs/tool.ts +435 -0
- package/src/jobs/types.ts +237 -0
- package/src/tool-ids.ts +11 -1
- package/src/types/tool-result.ts +6 -0
- package/src/use-space.ts +0 -1
- package/test/jobs/command-translation.spec.ts +277 -0
- package/test/jobs/formatters.spec.ts +267 -0
- package/test/jobs/uv-command.spec.ts +81 -0
- package/dist/types/mcp-ui-server-shim.d.ts +0 -37
- package/dist/types/mcp-ui-server-shim.d.ts.map +0 -1
- package/dist/types/mcp-ui-server-shim.js +0 -2
- package/dist/types/mcp-ui-server-shim.js.map +0 -1
- package/src/types/mcp-ui-server-shim.ts +0 -35
|
@@ -0,0 +1,333 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { JobsApiClient } from './api-client.js';
|
|
3
|
+
import { HfApiError } from '../hf-api-call.js';
|
|
4
|
+
import { runCommand, uvCommand } from './commands/run.js';
|
|
5
|
+
import { psCommand } from './commands/ps.js';
|
|
6
|
+
import { logsCommand } from './commands/logs.js';
|
|
7
|
+
import { inspectCommand, cancelCommand } from './commands/inspect.js';
|
|
8
|
+
import { scheduledRunCommand, scheduledUvCommand, scheduledPsCommand, scheduledInspectCommand, scheduledDeleteCommand, scheduledSuspendCommand, scheduledResumeCommand, } from './commands/scheduled.js';
|
|
9
|
+
export * from './types.js';
|
|
10
|
+
export { JobsApiClient } from './api-client.js';
|
|
11
|
+
import { runArgsSchema, uvArgsSchema, psArgsSchema, logsArgsSchema, inspectArgsSchema, cancelArgsSchema, scheduledRunArgsSchema, scheduledUvArgsSchema, scheduledPsArgsSchema, scheduledJobArgsSchema, } from './types.js';
|
|
12
|
+
const COMMAND_SCHEMAS = {
|
|
13
|
+
run: runArgsSchema,
|
|
14
|
+
uv: uvArgsSchema,
|
|
15
|
+
ps: psArgsSchema,
|
|
16
|
+
logs: logsArgsSchema,
|
|
17
|
+
inspect: inspectArgsSchema,
|
|
18
|
+
cancel: cancelArgsSchema,
|
|
19
|
+
'scheduled run': scheduledRunArgsSchema,
|
|
20
|
+
'scheduled uv': scheduledUvArgsSchema,
|
|
21
|
+
'scheduled ps': scheduledPsArgsSchema,
|
|
22
|
+
'scheduled inspect': scheduledJobArgsSchema,
|
|
23
|
+
'scheduled delete': scheduledJobArgsSchema,
|
|
24
|
+
'scheduled suspend': scheduledJobArgsSchema,
|
|
25
|
+
'scheduled resume': scheduledJobArgsSchema,
|
|
26
|
+
};
|
|
27
|
+
function validateArgs(schema, args, commandName) {
|
|
28
|
+
const result = schema.safeParse(args);
|
|
29
|
+
if (result.success) {
|
|
30
|
+
return { success: true };
|
|
31
|
+
}
|
|
32
|
+
const errors = result.error.errors;
|
|
33
|
+
const missingFields = [];
|
|
34
|
+
const invalidFields = [];
|
|
35
|
+
for (const err of errors) {
|
|
36
|
+
const field = err.path.join('.');
|
|
37
|
+
if (err.code === 'invalid_type' && err.received === 'undefined') {
|
|
38
|
+
missingFields.push(` • ${field}: ${err.message}`);
|
|
39
|
+
}
|
|
40
|
+
else {
|
|
41
|
+
invalidFields.push(` • ${field}: ${err.message}`);
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
let errorMessage = `Error: Invalid parameters for '${commandName}'\n\n`;
|
|
45
|
+
if (missingFields.length > 0) {
|
|
46
|
+
errorMessage += `Missing required parameters:\n${missingFields.join('\n')}\n\n`;
|
|
47
|
+
}
|
|
48
|
+
if (invalidFields.length > 0) {
|
|
49
|
+
errorMessage += `Invalid parameters:\n${invalidFields.join('\n')}\n\n`;
|
|
50
|
+
}
|
|
51
|
+
const providedKeys = args && typeof args === 'object' ? Object.keys(args) : [];
|
|
52
|
+
if (providedKeys.length > 0) {
|
|
53
|
+
errorMessage += `You provided: ${JSON.stringify(args, null, 2)}`;
|
|
54
|
+
}
|
|
55
|
+
else {
|
|
56
|
+
errorMessage += `You provided: {} (no parameters)`;
|
|
57
|
+
}
|
|
58
|
+
return {
|
|
59
|
+
success: false,
|
|
60
|
+
errorResult: {
|
|
61
|
+
formatted: errorMessage,
|
|
62
|
+
totalResults: 0,
|
|
63
|
+
resultsShared: 0,
|
|
64
|
+
isError: true,
|
|
65
|
+
},
|
|
66
|
+
};
|
|
67
|
+
}
|
|
68
|
+
const USAGE_INSTRUCTIONS = `# HuggingFace Jobs API
|
|
69
|
+
|
|
70
|
+
Manage compute jobs on Hugging Face infrastructure.
|
|
71
|
+
|
|
72
|
+
## Available Commands
|
|
73
|
+
|
|
74
|
+
### Job Management
|
|
75
|
+
- **run** - Run a job with a Docker image
|
|
76
|
+
- **uv** - Run a Python script with UV (inline dependencies)
|
|
77
|
+
- **ps** - List jobs
|
|
78
|
+
- **logs** - Fetch job logs
|
|
79
|
+
- **inspect** - Get detailed job information
|
|
80
|
+
- **cancel** - Cancel a running job
|
|
81
|
+
|
|
82
|
+
### Scheduled Jobs
|
|
83
|
+
- **scheduled run** - Create a scheduled job
|
|
84
|
+
- **scheduled uv** - Create a scheduled UV job
|
|
85
|
+
- **scheduled ps** - List scheduled jobs
|
|
86
|
+
- **scheduled inspect** - Get scheduled job details
|
|
87
|
+
- **scheduled delete** - Delete a scheduled job
|
|
88
|
+
- **scheduled suspend** - Pause a scheduled job
|
|
89
|
+
- **scheduled resume** - Resume a suspended job
|
|
90
|
+
|
|
91
|
+
## Examples
|
|
92
|
+
|
|
93
|
+
### Run a simple job
|
|
94
|
+
\`\`\`
|
|
95
|
+
# Command as array (recommended, especially for complex commands)
|
|
96
|
+
hf_jobs("run", {
|
|
97
|
+
"image": "python:3.12",
|
|
98
|
+
"command": ["python", "-c", "print('Hello from HF Jobs!')"],
|
|
99
|
+
"flavor": "cpu-basic"
|
|
100
|
+
})
|
|
101
|
+
|
|
102
|
+
# Command as string (parsed with POSIX shell semantics)
|
|
103
|
+
hf_jobs("run", {
|
|
104
|
+
"image": "python:3.12",
|
|
105
|
+
"command": "python -c \\"print('Hello world!')\\"",
|
|
106
|
+
"flavor": "cpu-basic"
|
|
107
|
+
})
|
|
108
|
+
|
|
109
|
+
# Use a Hugging Face Space as the image
|
|
110
|
+
hf_jobs("run", {
|
|
111
|
+
"image": "hf.co/spaces/username/spacename",
|
|
112
|
+
"command": ["python", "app.py"],
|
|
113
|
+
"flavor": "cpu-basic"
|
|
114
|
+
})
|
|
115
|
+
\`\`\`
|
|
116
|
+
|
|
117
|
+
### Run multiline Python scripts
|
|
118
|
+
\`\`\`
|
|
119
|
+
# Use array format with newlines in the -c argument
|
|
120
|
+
hf_jobs("run", {
|
|
121
|
+
"image": "python:3.12",
|
|
122
|
+
"command": ["python", "-c", "import sys\\nprint('Line 1')\\nprint('Line 2')"],
|
|
123
|
+
"flavor": "cpu-basic"
|
|
124
|
+
})
|
|
125
|
+
\`\`\`
|
|
126
|
+
|
|
127
|
+
### Run bash/shell commands
|
|
128
|
+
\`\`\`
|
|
129
|
+
hf_jobs("run", {
|
|
130
|
+
"image": "ubuntu:22.04",
|
|
131
|
+
"command": ["bash", "-c", "apt-get update && apt-get install -y curl"],
|
|
132
|
+
"flavor": "cpu-basic"
|
|
133
|
+
})
|
|
134
|
+
\`\`\`
|
|
135
|
+
|
|
136
|
+
### List running jobs
|
|
137
|
+
\`\`\`
|
|
138
|
+
hf_jobs("ps")
|
|
139
|
+
\`\`\`
|
|
140
|
+
|
|
141
|
+
### Get job logs
|
|
142
|
+
\`\`\`
|
|
143
|
+
hf_jobs("logs", {"job_id": "your-job-id"})
|
|
144
|
+
\`\`\`
|
|
145
|
+
|
|
146
|
+
### Run with GPU
|
|
147
|
+
\`\`\`
|
|
148
|
+
hf_jobs("run", {
|
|
149
|
+
"image": "pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel",
|
|
150
|
+
"command": ["python", "train.py"],
|
|
151
|
+
"flavor": "a10g-small"
|
|
152
|
+
})
|
|
153
|
+
\`\`\`
|
|
154
|
+
|
|
155
|
+
### Schedule a job
|
|
156
|
+
\`\`\`
|
|
157
|
+
hf_jobs("scheduled run", {
|
|
158
|
+
"schedule": "@hourly",
|
|
159
|
+
"image": "python:3.12",
|
|
160
|
+
"command": ["python", "backup.py"]
|
|
161
|
+
})
|
|
162
|
+
\`\`\`
|
|
163
|
+
|
|
164
|
+
## Hardware Flavors
|
|
165
|
+
|
|
166
|
+
**CPU:** cpu-basic, cpu-upgrade, cpu-performance, cpu-xl
|
|
167
|
+
**GPU:** t4-small, t4-medium, l4x1, a10g-small, a10g-large, a100-large, h100
|
|
168
|
+
**Specialized:** inf2x6 (AWS Inferentia)
|
|
169
|
+
|
|
170
|
+
## Command Format Guidelines
|
|
171
|
+
|
|
172
|
+
**Array format (recommended):**
|
|
173
|
+
- Use for complex commands, multiline scripts, or commands with special characters
|
|
174
|
+
- No quoting/escaping needed: \`["python", "-c", "print('hello')"]\`
|
|
175
|
+
- Works with any language: Python, bash, npm, etc.
|
|
176
|
+
|
|
177
|
+
**String format:**
|
|
178
|
+
- Parsed with POSIX shell semantics (quotes, escaping)
|
|
179
|
+
- Good for simple commands: \`"python script.py"\`
|
|
180
|
+
- Shell operators (|, &&, >, etc.) are NOT supported - use array with \`bash -c\` instead
|
|
181
|
+
- Variable tokens like \`$HF_TOKEN\` are kept literal—the remote container expands them at runtime
|
|
182
|
+
|
|
183
|
+
**Multiline inline scripts:**
|
|
184
|
+
- Automatically wrapped in \`["/bin/sh", "-lc", "..."]\` for shell execution
|
|
185
|
+
- Paste entire Python/bash snippets—they execute as if typed in a shell
|
|
186
|
+
|
|
187
|
+
## Tips
|
|
188
|
+
|
|
189
|
+
- Jobs default to detached mode (return immediately with job ID)
|
|
190
|
+
- Use Hub resources directly: \`load_dataset('squad')\`, \`AutoModel.from_pretrained('bert-base')\`
|
|
191
|
+
- Pass HF_TOKEN via secrets for private resources
|
|
192
|
+
- Logs are time-limited (10s max) - check job page for full logs
|
|
193
|
+
- For shell pipes/operators, use: \`["bash", "-c", "cmd1 | cmd2"]\`
|
|
194
|
+
`;
|
|
195
|
+
export const HF_JOBS_TOOL_CONFIG = {
|
|
196
|
+
name: 'hf_jobs',
|
|
197
|
+
description: 'Manage HuggingFace compute jobs. Run any command in Docker containers (Python, Node.js, bash, etc.), ' +
|
|
198
|
+
'execute Python scripts with UV package manager, manage scheduled jobs, and monitor job status and logs. ' +
|
|
199
|
+
'Supports CPU and GPU hardware. Call with no arguments for full usage instructions and examples.',
|
|
200
|
+
schema: z.object({
|
|
201
|
+
command: z
|
|
202
|
+
.string()
|
|
203
|
+
.optional()
|
|
204
|
+
.describe('Command to execute: "run", "uv", "ps", "logs", "inspect", "cancel", ' +
|
|
205
|
+
'"scheduled run", "scheduled uv", "scheduled ps", "scheduled inspect", ' +
|
|
206
|
+
'"scheduled delete", "scheduled suspend", "scheduled resume"'),
|
|
207
|
+
args: z.record(z.any()).optional().describe('Command-specific arguments as a JSON object'),
|
|
208
|
+
}),
|
|
209
|
+
annotations: {
|
|
210
|
+
title: 'HuggingFace Jobs',
|
|
211
|
+
destructiveHint: false,
|
|
212
|
+
readOnlyHint: false,
|
|
213
|
+
openWorldHint: true,
|
|
214
|
+
},
|
|
215
|
+
};
|
|
216
|
+
export class HfJobsTool {
|
|
217
|
+
client;
|
|
218
|
+
hfToken;
|
|
219
|
+
isAuthenticated;
|
|
220
|
+
constructor(hfToken, isAuthenticated, namespace) {
|
|
221
|
+
this.hfToken = hfToken;
|
|
222
|
+
this.isAuthenticated = isAuthenticated ?? !!hfToken;
|
|
223
|
+
this.client = new JobsApiClient(hfToken, namespace);
|
|
224
|
+
}
|
|
225
|
+
async execute(params) {
|
|
226
|
+
if (!this.isAuthenticated) {
|
|
227
|
+
return {
|
|
228
|
+
formatted: 'Jobs are available for Pro, Team and Enterprise users. Go to https://huggingface.co/pricing to get started.',
|
|
229
|
+
totalResults: 0,
|
|
230
|
+
resultsShared: 0,
|
|
231
|
+
};
|
|
232
|
+
}
|
|
233
|
+
if (!params.command) {
|
|
234
|
+
return {
|
|
235
|
+
formatted: USAGE_INSTRUCTIONS,
|
|
236
|
+
totalResults: 1,
|
|
237
|
+
resultsShared: 1,
|
|
238
|
+
};
|
|
239
|
+
}
|
|
240
|
+
const command = params.command.toLowerCase();
|
|
241
|
+
const args = params.args || {};
|
|
242
|
+
const schema = COMMAND_SCHEMAS[command];
|
|
243
|
+
if (schema) {
|
|
244
|
+
const validation = validateArgs(schema, args, command);
|
|
245
|
+
if (!validation.success) {
|
|
246
|
+
return validation.errorResult;
|
|
247
|
+
}
|
|
248
|
+
}
|
|
249
|
+
try {
|
|
250
|
+
let result;
|
|
251
|
+
switch (command) {
|
|
252
|
+
case 'run':
|
|
253
|
+
result = await runCommand(args, this.client, this.hfToken);
|
|
254
|
+
break;
|
|
255
|
+
case 'uv':
|
|
256
|
+
result = await uvCommand(args, this.client, this.hfToken);
|
|
257
|
+
break;
|
|
258
|
+
case 'ps':
|
|
259
|
+
result = await psCommand(args, this.client);
|
|
260
|
+
break;
|
|
261
|
+
case 'logs':
|
|
262
|
+
result = await logsCommand(args, this.client, this.hfToken);
|
|
263
|
+
break;
|
|
264
|
+
case 'inspect':
|
|
265
|
+
result = await inspectCommand(args, this.client);
|
|
266
|
+
break;
|
|
267
|
+
case 'cancel':
|
|
268
|
+
result = await cancelCommand(args, this.client);
|
|
269
|
+
break;
|
|
270
|
+
case 'scheduled run':
|
|
271
|
+
result = await scheduledRunCommand(args, this.client);
|
|
272
|
+
break;
|
|
273
|
+
case 'scheduled uv':
|
|
274
|
+
result = await scheduledUvCommand(args, this.client);
|
|
275
|
+
break;
|
|
276
|
+
case 'scheduled ps':
|
|
277
|
+
result = await scheduledPsCommand(args, this.client);
|
|
278
|
+
break;
|
|
279
|
+
case 'scheduled inspect':
|
|
280
|
+
result = await scheduledInspectCommand(args, this.client);
|
|
281
|
+
break;
|
|
282
|
+
case 'scheduled delete':
|
|
283
|
+
result = await scheduledDeleteCommand(args, this.client);
|
|
284
|
+
break;
|
|
285
|
+
case 'scheduled suspend':
|
|
286
|
+
result = await scheduledSuspendCommand(args, this.client);
|
|
287
|
+
break;
|
|
288
|
+
case 'scheduled resume':
|
|
289
|
+
result = await scheduledResumeCommand(args, this.client);
|
|
290
|
+
break;
|
|
291
|
+
default:
|
|
292
|
+
return {
|
|
293
|
+
formatted: `Unknown command: "${params.command}"
|
|
294
|
+
|
|
295
|
+
Available commands:
|
|
296
|
+
- run, uv, ps, logs, inspect, cancel
|
|
297
|
+
- scheduled run, scheduled uv, scheduled ps, scheduled inspect, scheduled delete, scheduled suspend, scheduled resume
|
|
298
|
+
|
|
299
|
+
Call hf_jobs() with no arguments for full usage instructions.`,
|
|
300
|
+
totalResults: 0,
|
|
301
|
+
resultsShared: 0,
|
|
302
|
+
};
|
|
303
|
+
}
|
|
304
|
+
return {
|
|
305
|
+
formatted: result,
|
|
306
|
+
totalResults: 1,
|
|
307
|
+
resultsShared: 1,
|
|
308
|
+
};
|
|
309
|
+
}
|
|
310
|
+
catch (error) {
|
|
311
|
+
let errorMessage = error instanceof Error ? error.message : String(error);
|
|
312
|
+
if (error instanceof HfApiError && error.responseBody) {
|
|
313
|
+
try {
|
|
314
|
+
const parsed = JSON.parse(error.responseBody);
|
|
315
|
+
const formattedBody = JSON.stringify(parsed, null, 2);
|
|
316
|
+
errorMessage += `\n\nServer response:\n${formattedBody}`;
|
|
317
|
+
}
|
|
318
|
+
catch {
|
|
319
|
+
if (error.responseBody.length < 500) {
|
|
320
|
+
errorMessage += `\n\nServer response: ${error.responseBody}`;
|
|
321
|
+
}
|
|
322
|
+
}
|
|
323
|
+
}
|
|
324
|
+
return {
|
|
325
|
+
formatted: `Error executing ${params.command}: ${errorMessage}`,
|
|
326
|
+
totalResults: 0,
|
|
327
|
+
resultsShared: 0,
|
|
328
|
+
isError: true,
|
|
329
|
+
};
|
|
330
|
+
}
|
|
331
|
+
}
|
|
332
|
+
}
|
|
333
|
+
//# sourceMappingURL=tool.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool.js","sourceRoot":"","sources":["../../src/jobs/tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC1D,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtE,OAAO,EACN,mBAAmB,EACnB,kBAAkB,EAClB,kBAAkB,EAClB,uBAAuB,EACvB,sBAAsB,EACtB,uBAAuB,EACvB,sBAAsB,GACtB,MAAM,yBAAyB,CAAC;AAgBjC,cAAc,YAAY,CAAC;AAC3B,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAGhD,OAAO,EACN,aAAa,EACb,YAAY,EACZ,YAAY,EACZ,cAAc,EACd,iBAAiB,EACjB,gBAAgB,EAChB,sBAAsB,EACtB,qBAAqB,EACrB,qBAAqB,EACrB,sBAAsB,GACtB,MAAM,YAAY,CAAC;AAKpB,MAAM,eAAe,GAAG;IACvB,GAAG,EAAE,aAAa;IAClB,EAAE,EAAE,YAAY;IAChB,EAAE,EAAE,YAAY;IAChB,IAAI,EAAE,cAAc;IACpB,OAAO,EAAE,iBAAiB;IAC1B,MAAM,EAAE,gBAAgB;IACxB,eAAe,EAAE,sBAAsB;IACvC,cAAc,EAAE,qBAAqB;IACrC,cAAc,EAAE,qBAAqB;IACrC,mBAAmB,EAAE,sBAAsB;IAC3C,kBAAkB,EAAE,sBAAsB;IAC1C,mBAAmB,EAAE,sBAAsB;IAC3C,kBAAkB,EAAE,sBAAsB;CACjC,CAAC;AAMX,SAAS,YAAY,CACpB,MAAmB,EACnB,IAAa,EACb,WAAmB;IAEnB,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAEtC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC1B,CAAC;IAGD,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC;IACnC,MAAM,aAAa,GAAa,EAAE,CAAC;IACnC,MAAM,aAAa,GAAa,EAAE,CAAC;IAEnC,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC1B,MAAM,KAAK,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,GAAG,CAAC,IAAI,KAAK,cAAc,IAAI,GAAG,CAAC,QAAQ,KAAK,WAAW,EAAE,CAAC;YACjE,aAAa,CAAC,IAAI,CAAC,OAAO,KAAK,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QACpD,CAAC;aAAM,CAAC;YACP,aAAa,CAAC,IAAI,CAAC,OAAO,KAAK,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QACpD,CAAC;IACF,CAAC;IAED,IAAI,YAAY,GAAG,kCAAkC,WAAW,OAAO,CAAC;IAExE,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9B,YAAY,IAAI,iCAAiC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;IACjF,CAAC;IAED,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9B,YAAY,IAAI,wBAAwB,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;IACxE,CAAC;IAGD,MAAM,YAAY,GAAG,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC/E,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,YAAY,IAAI,iBAAiB,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;IAClE,CAAC;SAAM,CAAC;QACP,YAAY,IAAI,kCAAkC,CAAC;IACpD,CAAC;IAED,OAAO;QACN,OAAO,EAAE,KAAK;QACd,WAAW,EAAE;YACZ,SAAS,EAAE,YAAY;YACvB,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,OAAO,EAAE,IAAI;SACb;KACD,CAAC;AACH,CAAC;AAKD,MAAM,kBAAkB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8H1B,CAAC;AAKF,MAAM,CAAC,MAAM,mBAAmB,GAAG;IAClC,IAAI,EAAE,SAAS;IACf,WAAW,EACV,uGAAuG;QACvG,0GAA0G;QAC1G,iGAAiG;IAClG,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,OAAO,EAAE,CAAC;aACR,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CACR,sEAAsE;YACrE,wEAAwE;YACxE,6DAA6D,CAC9D;QACF,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6CAA6C,CAAC;KAC1F,CAAC;IACF,WAAW,EAAE;QACZ,KAAK,EAAE,kBAAkB;QACzB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,KAAK;QACnB,aAAa,EAAE,IAAI;KACnB;CACQ,CAAC;AAKX,MAAM,OAAO,UAAU;IACd,MAAM,CAAgB;IACtB,OAAO,CAAU;IACjB,eAAe,CAAU;IAEjC,YAAY,OAAgB,EAAE,eAAyB,EAAE,SAAkB;QAC1E,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,eAAe,GAAG,eAAe,IAAI,CAAC,CAAC,OAAO,CAAC;QACpD,IAAI,CAAC,MAAM,GAAG,IAAI,aAAa,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IACrD,CAAC;IAKD,KAAK,CAAC,OAAO,CAAC,MAA4D;QAEzE,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC3B,OAAO;gBACN,SAAS,EACR,6GAA6G;gBAC9G,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;aAChB,CAAC;QACH,CAAC;QAGD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACrB,OAAO;gBACN,SAAS,EAAE,kBAAkB;gBAC7B,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;aAChB,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QAC7C,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;QAG/B,MAAM,MAAM,GAAG,eAAe,CAAC,OAAuC,CAAC,CAAC;QACxE,IAAI,MAAM,EAAE,CAAC;YACZ,MAAM,UAAU,GAAG,YAAY,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;YACvD,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;gBACzB,OAAO,UAAU,CAAC,WAAW,CAAC;YAC/B,CAAC;QACF,CAAC;QAED,IAAI,CAAC;YACJ,IAAI,MAAc,CAAC;YAEnB,QAAQ,OAAO,EAAE,CAAC;gBACjB,KAAK,KAAK;oBACT,MAAM,GAAG,MAAM,UAAU,CAAC,IAAe,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;oBACtE,MAAM;gBAEP,KAAK,IAAI;oBACR,MAAM,GAAG,MAAM,SAAS,CAAC,IAAc,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;oBACpE,MAAM;gBAEP,KAAK,IAAI;oBACR,MAAM,GAAG,MAAM,SAAS,CAAC,IAAc,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBACtD,MAAM;gBAEP,KAAK,MAAM;oBACV,MAAM,GAAG,MAAM,WAAW,CAAC,IAAgB,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;oBACxE,MAAM;gBAEP,KAAK,SAAS;oBACb,MAAM,GAAG,MAAM,cAAc,CAAC,IAAmB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBAChE,MAAM;gBAEP,KAAK,QAAQ;oBACZ,MAAM,GAAG,MAAM,aAAa,CAAC,IAAkB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC9D,MAAM;gBAEP,KAAK,eAAe;oBACnB,MAAM,GAAG,MAAM,mBAAmB,CAAC,IAAwB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC1E,MAAM;gBAEP,KAAK,cAAc;oBAClB,MAAM,GAAG,MAAM,kBAAkB,CAAC,IAAuB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBACxE,MAAM;gBAEP,KAAK,cAAc;oBAClB,MAAM,GAAG,MAAM,kBAAkB,CAAC,IAAuB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBACxE,MAAM;gBAEP,KAAK,mBAAmB;oBACvB,MAAM,GAAG,MAAM,uBAAuB,CAAC,IAAwB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC9E,MAAM;gBAEP,KAAK,kBAAkB;oBACtB,MAAM,GAAG,MAAM,sBAAsB,CAAC,IAAwB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC7E,MAAM;gBAEP,KAAK,mBAAmB;oBACvB,MAAM,GAAG,MAAM,uBAAuB,CAAC,IAAwB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC9E,MAAM;gBAEP,KAAK,kBAAkB;oBACtB,MAAM,GAAG,MAAM,sBAAsB,CAAC,IAAwB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC7E,MAAM;gBAEP;oBACC,OAAO;wBACN,SAAS,EAAE,qBAAqB,MAAM,CAAC,OAAO;;;;;;8DAMU;wBACxD,YAAY,EAAE,CAAC;wBACf,aAAa,EAAE,CAAC;qBAChB,CAAC;YACJ,CAAC;YAED,OAAO;gBACN,SAAS,EAAE,MAAM;gBACjB,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;aAChB,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAG1E,IAAI,KAAK,YAAY,UAAU,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;gBACvD,IAAI,CAAC;oBAEJ,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;oBACvD,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;oBACtD,YAAY,IAAI,yBAAyB,aAAa,EAAE,CAAC;gBAC1D,CAAC;gBAAC,MAAM,CAAC;oBAER,IAAI,KAAK,CAAC,YAAY,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;wBACrC,YAAY,IAAI,wBAAwB,KAAK,CAAC,YAAY,EAAE,CAAC;oBAC9D,CAAC;gBACF,CAAC;YACF,CAAC;YAED,OAAO;gBACN,SAAS,EAAE,mBAAmB,MAAM,CAAC,OAAO,KAAK,YAAY,EAAE;gBAC/D,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;gBAChB,OAAO,EAAE,IAAI;aACb,CAAC;QACH,CAAC;IACF,CAAC;CACD"}
|
|
@@ -0,0 +1,295 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
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"];
|
|
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"];
|
|
6
|
+
export type JobFlavor = (typeof ALL_FLAVORS)[number];
|
|
7
|
+
export type JobStage = 'RUNNING' | 'COMPLETED' | 'CANCELED' | 'ERROR' | 'DELETED';
|
|
8
|
+
export interface JobStatus {
|
|
9
|
+
stage: JobStage;
|
|
10
|
+
message?: string | null;
|
|
11
|
+
}
|
|
12
|
+
export interface JobOwner {
|
|
13
|
+
id: string;
|
|
14
|
+
name: string;
|
|
15
|
+
type: 'user' | 'org';
|
|
16
|
+
}
|
|
17
|
+
export interface JobInfo {
|
|
18
|
+
id: string;
|
|
19
|
+
createdAt: string;
|
|
20
|
+
dockerImage?: string;
|
|
21
|
+
spaceId?: string;
|
|
22
|
+
command?: string[];
|
|
23
|
+
arguments?: string[];
|
|
24
|
+
environment: Record<string, string>;
|
|
25
|
+
secrets?: Record<string, string | null>;
|
|
26
|
+
flavor: string;
|
|
27
|
+
status: JobStatus;
|
|
28
|
+
owner: JobOwner;
|
|
29
|
+
createdBy?: JobOwner;
|
|
30
|
+
tags?: string[];
|
|
31
|
+
timeout?: number;
|
|
32
|
+
url?: string;
|
|
33
|
+
endpoint?: string;
|
|
34
|
+
finishedAt?: string;
|
|
35
|
+
}
|
|
36
|
+
export interface JobSpec {
|
|
37
|
+
dockerImage?: string;
|
|
38
|
+
spaceId?: string;
|
|
39
|
+
command: string[];
|
|
40
|
+
arguments?: string[];
|
|
41
|
+
environment?: Record<string, string>;
|
|
42
|
+
secrets?: Record<string, string>;
|
|
43
|
+
flavor: string;
|
|
44
|
+
timeoutSeconds?: number;
|
|
45
|
+
}
|
|
46
|
+
export interface ScheduledJobSpec {
|
|
47
|
+
schedule: string;
|
|
48
|
+
suspend?: boolean;
|
|
49
|
+
jobSpec: JobSpec;
|
|
50
|
+
}
|
|
51
|
+
export interface ScheduledJobInfo {
|
|
52
|
+
id: string;
|
|
53
|
+
schedule: string;
|
|
54
|
+
suspend: boolean;
|
|
55
|
+
jobSpec: JobSpec;
|
|
56
|
+
lastRun?: string;
|
|
57
|
+
nextRun?: string;
|
|
58
|
+
owner: JobOwner;
|
|
59
|
+
createdAt: string;
|
|
60
|
+
}
|
|
61
|
+
export interface LogEvent {
|
|
62
|
+
timestamp: string;
|
|
63
|
+
data: string;
|
|
64
|
+
}
|
|
65
|
+
export declare const runArgsSchema: z.ZodObject<{
|
|
66
|
+
namespace: z.ZodOptional<z.ZodString>;
|
|
67
|
+
} & {
|
|
68
|
+
image: z.ZodString;
|
|
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"]>>>;
|
|
71
|
+
env: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
|
|
72
|
+
secrets: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
|
|
73
|
+
timeout: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
74
|
+
detach: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
|
|
75
|
+
}, "strip", z.ZodTypeAny, {
|
|
76
|
+
image: string;
|
|
77
|
+
command: string | string[];
|
|
78
|
+
flavor: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "cpu-performance" | "cpu-xl" | "sprx8" | "l40sx1" | "l40sx4" | "l40sx8" | "h100" | "h100x8" | "inf2x6";
|
|
79
|
+
timeout: string;
|
|
80
|
+
detach: boolean;
|
|
81
|
+
namespace?: string | undefined;
|
|
82
|
+
env?: Record<string, string> | undefined;
|
|
83
|
+
secrets?: Record<string, string> | undefined;
|
|
84
|
+
}, {
|
|
85
|
+
image: string;
|
|
86
|
+
command: string | string[];
|
|
87
|
+
namespace?: string | undefined;
|
|
88
|
+
flavor?: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "cpu-performance" | "cpu-xl" | "sprx8" | "l40sx1" | "l40sx4" | "l40sx8" | "h100" | "h100x8" | "inf2x6" | undefined;
|
|
89
|
+
env?: Record<string, string> | undefined;
|
|
90
|
+
secrets?: Record<string, string> | undefined;
|
|
91
|
+
timeout?: string | undefined;
|
|
92
|
+
detach?: boolean | undefined;
|
|
93
|
+
}>;
|
|
94
|
+
export declare const uvArgsSchema: z.ZodObject<{
|
|
95
|
+
namespace: z.ZodOptional<z.ZodString>;
|
|
96
|
+
} & {
|
|
97
|
+
script: z.ZodString;
|
|
98
|
+
repo: z.ZodOptional<z.ZodString>;
|
|
99
|
+
with_deps: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
|
|
100
|
+
script_args: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
|
|
101
|
+
python: z.ZodOptional<z.ZodString>;
|
|
102
|
+
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"]>>>;
|
|
103
|
+
env: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
|
|
104
|
+
secrets: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
|
|
105
|
+
timeout: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
106
|
+
detach: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
|
|
107
|
+
}, "strip", z.ZodTypeAny, {
|
|
108
|
+
script: string;
|
|
109
|
+
flavor: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "cpu-performance" | "cpu-xl" | "sprx8" | "l40sx1" | "l40sx4" | "l40sx8" | "h100" | "h100x8" | "inf2x6";
|
|
110
|
+
timeout: string;
|
|
111
|
+
detach: boolean;
|
|
112
|
+
namespace?: string | undefined;
|
|
113
|
+
env?: Record<string, string> | undefined;
|
|
114
|
+
secrets?: Record<string, string> | undefined;
|
|
115
|
+
repo?: string | undefined;
|
|
116
|
+
with_deps?: string[] | undefined;
|
|
117
|
+
script_args?: string[] | undefined;
|
|
118
|
+
python?: string | undefined;
|
|
119
|
+
}, {
|
|
120
|
+
script: string;
|
|
121
|
+
namespace?: string | undefined;
|
|
122
|
+
flavor?: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "cpu-performance" | "cpu-xl" | "sprx8" | "l40sx1" | "l40sx4" | "l40sx8" | "h100" | "h100x8" | "inf2x6" | undefined;
|
|
123
|
+
env?: Record<string, string> | undefined;
|
|
124
|
+
secrets?: Record<string, string> | undefined;
|
|
125
|
+
timeout?: string | undefined;
|
|
126
|
+
detach?: boolean | undefined;
|
|
127
|
+
repo?: string | undefined;
|
|
128
|
+
with_deps?: string[] | undefined;
|
|
129
|
+
script_args?: string[] | undefined;
|
|
130
|
+
python?: string | undefined;
|
|
131
|
+
}>;
|
|
132
|
+
export declare const psArgsSchema: z.ZodObject<{
|
|
133
|
+
namespace: z.ZodOptional<z.ZodString>;
|
|
134
|
+
} & {
|
|
135
|
+
all: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
|
|
136
|
+
status: z.ZodOptional<z.ZodString>;
|
|
137
|
+
}, "strip", z.ZodTypeAny, {
|
|
138
|
+
all: boolean;
|
|
139
|
+
status?: string | undefined;
|
|
140
|
+
namespace?: string | undefined;
|
|
141
|
+
}, {
|
|
142
|
+
status?: string | undefined;
|
|
143
|
+
all?: boolean | undefined;
|
|
144
|
+
namespace?: string | undefined;
|
|
145
|
+
}>;
|
|
146
|
+
export declare const logsArgsSchema: z.ZodObject<{
|
|
147
|
+
namespace: z.ZodOptional<z.ZodString>;
|
|
148
|
+
} & {
|
|
149
|
+
job_id: z.ZodString;
|
|
150
|
+
tail: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
|
|
151
|
+
}, "strip", z.ZodTypeAny, {
|
|
152
|
+
job_id: string;
|
|
153
|
+
tail: number;
|
|
154
|
+
namespace?: string | undefined;
|
|
155
|
+
}, {
|
|
156
|
+
job_id: string;
|
|
157
|
+
namespace?: string | undefined;
|
|
158
|
+
tail?: number | undefined;
|
|
159
|
+
}>;
|
|
160
|
+
export declare const inspectArgsSchema: z.ZodObject<{
|
|
161
|
+
namespace: z.ZodOptional<z.ZodString>;
|
|
162
|
+
} & {
|
|
163
|
+
job_id: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
|
|
164
|
+
}, "strip", z.ZodTypeAny, {
|
|
165
|
+
job_id: string | string[];
|
|
166
|
+
namespace?: string | undefined;
|
|
167
|
+
}, {
|
|
168
|
+
job_id: string | string[];
|
|
169
|
+
namespace?: string | undefined;
|
|
170
|
+
}>;
|
|
171
|
+
export declare const cancelArgsSchema: z.ZodObject<{
|
|
172
|
+
namespace: z.ZodOptional<z.ZodString>;
|
|
173
|
+
} & {
|
|
174
|
+
job_id: z.ZodString;
|
|
175
|
+
}, "strip", z.ZodTypeAny, {
|
|
176
|
+
job_id: string;
|
|
177
|
+
namespace?: string | undefined;
|
|
178
|
+
}, {
|
|
179
|
+
job_id: string;
|
|
180
|
+
namespace?: string | undefined;
|
|
181
|
+
}>;
|
|
182
|
+
export declare const scheduledRunArgsSchema: z.ZodObject<{
|
|
183
|
+
namespace: z.ZodOptional<z.ZodString>;
|
|
184
|
+
} & {
|
|
185
|
+
image: z.ZodString;
|
|
186
|
+
command: z.ZodUnion<[z.ZodString, z.ZodArray<z.ZodString, "many">]>;
|
|
187
|
+
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"]>>>;
|
|
188
|
+
env: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
|
|
189
|
+
secrets: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
|
|
190
|
+
timeout: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
191
|
+
detach: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
|
|
192
|
+
} & {
|
|
193
|
+
schedule: z.ZodString;
|
|
194
|
+
suspend: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
|
|
195
|
+
}, "strip", z.ZodTypeAny, {
|
|
196
|
+
image: string;
|
|
197
|
+
command: string | string[];
|
|
198
|
+
flavor: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "cpu-performance" | "cpu-xl" | "sprx8" | "l40sx1" | "l40sx4" | "l40sx8" | "h100" | "h100x8" | "inf2x6";
|
|
199
|
+
timeout: string;
|
|
200
|
+
detach: boolean;
|
|
201
|
+
schedule: string;
|
|
202
|
+
suspend: boolean;
|
|
203
|
+
namespace?: string | undefined;
|
|
204
|
+
env?: Record<string, string> | undefined;
|
|
205
|
+
secrets?: Record<string, string> | undefined;
|
|
206
|
+
}, {
|
|
207
|
+
image: string;
|
|
208
|
+
command: string | string[];
|
|
209
|
+
schedule: string;
|
|
210
|
+
namespace?: string | undefined;
|
|
211
|
+
flavor?: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "cpu-performance" | "cpu-xl" | "sprx8" | "l40sx1" | "l40sx4" | "l40sx8" | "h100" | "h100x8" | "inf2x6" | undefined;
|
|
212
|
+
env?: Record<string, string> | undefined;
|
|
213
|
+
secrets?: Record<string, string> | undefined;
|
|
214
|
+
timeout?: string | undefined;
|
|
215
|
+
detach?: boolean | undefined;
|
|
216
|
+
suspend?: boolean | undefined;
|
|
217
|
+
}>;
|
|
218
|
+
export declare const scheduledUvArgsSchema: z.ZodObject<{
|
|
219
|
+
namespace: z.ZodOptional<z.ZodString>;
|
|
220
|
+
} & {
|
|
221
|
+
script: z.ZodString;
|
|
222
|
+
repo: z.ZodOptional<z.ZodString>;
|
|
223
|
+
with_deps: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
|
|
224
|
+
script_args: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
|
|
225
|
+
python: z.ZodOptional<z.ZodString>;
|
|
226
|
+
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"]>>>;
|
|
227
|
+
env: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
|
|
228
|
+
secrets: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
|
|
229
|
+
timeout: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
230
|
+
detach: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
|
|
231
|
+
} & {
|
|
232
|
+
schedule: z.ZodString;
|
|
233
|
+
suspend: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
|
|
234
|
+
}, "strip", z.ZodTypeAny, {
|
|
235
|
+
script: string;
|
|
236
|
+
flavor: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "cpu-performance" | "cpu-xl" | "sprx8" | "l40sx1" | "l40sx4" | "l40sx8" | "h100" | "h100x8" | "inf2x6";
|
|
237
|
+
timeout: string;
|
|
238
|
+
detach: boolean;
|
|
239
|
+
schedule: string;
|
|
240
|
+
suspend: boolean;
|
|
241
|
+
namespace?: string | undefined;
|
|
242
|
+
env?: Record<string, string> | undefined;
|
|
243
|
+
secrets?: Record<string, string> | undefined;
|
|
244
|
+
repo?: string | undefined;
|
|
245
|
+
with_deps?: string[] | undefined;
|
|
246
|
+
script_args?: string[] | undefined;
|
|
247
|
+
python?: string | undefined;
|
|
248
|
+
}, {
|
|
249
|
+
script: string;
|
|
250
|
+
schedule: string;
|
|
251
|
+
namespace?: string | undefined;
|
|
252
|
+
flavor?: "cpu-basic" | "zero-a10g" | "cpu-upgrade" | "t4-small" | "t4-medium" | "l4x1" | "l4x4" | "a10g-small" | "a10g-large" | "a10g-largex2" | "a10g-largex4" | "a100-large" | "cpu-performance" | "cpu-xl" | "sprx8" | "l40sx1" | "l40sx4" | "l40sx8" | "h100" | "h100x8" | "inf2x6" | undefined;
|
|
253
|
+
env?: Record<string, string> | undefined;
|
|
254
|
+
secrets?: Record<string, string> | undefined;
|
|
255
|
+
timeout?: string | undefined;
|
|
256
|
+
detach?: boolean | undefined;
|
|
257
|
+
repo?: string | undefined;
|
|
258
|
+
with_deps?: string[] | undefined;
|
|
259
|
+
script_args?: string[] | undefined;
|
|
260
|
+
python?: string | undefined;
|
|
261
|
+
suspend?: boolean | undefined;
|
|
262
|
+
}>;
|
|
263
|
+
export declare const scheduledPsArgsSchema: z.ZodObject<{
|
|
264
|
+
namespace: z.ZodOptional<z.ZodString>;
|
|
265
|
+
} & {
|
|
266
|
+
all: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
|
|
267
|
+
}, "strip", z.ZodTypeAny, {
|
|
268
|
+
all: boolean;
|
|
269
|
+
namespace?: string | undefined;
|
|
270
|
+
}, {
|
|
271
|
+
all?: boolean | undefined;
|
|
272
|
+
namespace?: string | undefined;
|
|
273
|
+
}>;
|
|
274
|
+
export declare const scheduledJobArgsSchema: z.ZodObject<{
|
|
275
|
+
namespace: z.ZodOptional<z.ZodString>;
|
|
276
|
+
} & {
|
|
277
|
+
scheduled_job_id: z.ZodString;
|
|
278
|
+
}, "strip", z.ZodTypeAny, {
|
|
279
|
+
scheduled_job_id: string;
|
|
280
|
+
namespace?: string | undefined;
|
|
281
|
+
}, {
|
|
282
|
+
scheduled_job_id: string;
|
|
283
|
+
namespace?: string | undefined;
|
|
284
|
+
}>;
|
|
285
|
+
export type RunArgs = z.infer<typeof runArgsSchema>;
|
|
286
|
+
export type UvArgs = z.infer<typeof uvArgsSchema>;
|
|
287
|
+
export type PsArgs = z.infer<typeof psArgsSchema>;
|
|
288
|
+
export type LogsArgs = z.infer<typeof logsArgsSchema>;
|
|
289
|
+
export type InspectArgs = z.infer<typeof inspectArgsSchema>;
|
|
290
|
+
export type CancelArgs = z.infer<typeof cancelArgsSchema>;
|
|
291
|
+
export type ScheduledRunArgs = z.infer<typeof scheduledRunArgsSchema>;
|
|
292
|
+
export type ScheduledUvArgs = z.infer<typeof scheduledUvArgsSchema>;
|
|
293
|
+
export type ScheduledPsArgs = z.infer<typeof scheduledPsArgsSchema>;
|
|
294
|
+
export type ScheduledJobArgs = z.infer<typeof scheduledJobArgsSchema>;
|
|
295
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +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;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA0BxB,CAAC;AAGH,eAAO,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAavB,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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAKjC,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"}
|