@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.
Files changed (91) hide show
  1. package/dist/docs-search/docs-semantic-search.d.ts +2 -2
  2. package/dist/docs-search/docs-semantic-search.d.ts.map +1 -1
  3. package/dist/docs-search/docs-semantic-search.js +56 -21
  4. package/dist/docs-search/docs-semantic-search.js.map +1 -1
  5. package/dist/hf-api-call.d.ts.map +1 -1
  6. package/dist/hf-api-call.js +4 -0
  7. package/dist/hf-api-call.js.map +1 -1
  8. package/dist/index.d.ts +1 -0
  9. package/dist/index.d.ts.map +1 -1
  10. package/dist/index.js +1 -0
  11. package/dist/index.js.map +1 -1
  12. package/dist/jobs/api-client.d.ts +19 -0
  13. package/dist/jobs/api-client.d.ts.map +1 -0
  14. package/dist/jobs/api-client.js +104 -0
  15. package/dist/jobs/api-client.js.map +1 -0
  16. package/dist/jobs/commands/inspect.d.ts +5 -0
  17. package/dist/jobs/commands/inspect.d.ts.map +1 -0
  18. package/dist/jobs/commands/inspect.js +21 -0
  19. package/dist/jobs/commands/inspect.js.map +1 -0
  20. package/dist/jobs/commands/logs.d.ts +4 -0
  21. package/dist/jobs/commands/logs.d.ts.map +1 -0
  22. package/dist/jobs/commands/logs.js +24 -0
  23. package/dist/jobs/commands/logs.js.map +1 -0
  24. package/dist/jobs/commands/ps.d.ts +4 -0
  25. package/dist/jobs/commands/ps.d.ts.map +1 -0
  26. package/dist/jobs/commands/ps.js +23 -0
  27. package/dist/jobs/commands/ps.js.map +1 -0
  28. package/dist/jobs/commands/run.d.ts +5 -0
  29. package/dist/jobs/commands/run.d.ts.map +1 -0
  30. package/dist/jobs/commands/run.js +89 -0
  31. package/dist/jobs/commands/run.js.map +1 -0
  32. package/dist/jobs/commands/scheduled.d.ts +10 -0
  33. package/dist/jobs/commands/scheduled.d.ts.map +1 -0
  34. package/dist/jobs/commands/scheduled.js +111 -0
  35. package/dist/jobs/commands/scheduled.js.map +1 -0
  36. package/dist/jobs/commands/utils.d.ts +19 -0
  37. package/dist/jobs/commands/utils.d.ts.map +1 -0
  38. package/dist/jobs/commands/utils.js +99 -0
  39. package/dist/jobs/commands/utils.js.map +1 -0
  40. package/dist/jobs/formatters.d.ts +6 -0
  41. package/dist/jobs/formatters.d.ts.map +1 -0
  42. package/dist/jobs/formatters.js +98 -0
  43. package/dist/jobs/formatters.js.map +1 -0
  44. package/dist/jobs/sse-handler.d.ts +12 -0
  45. package/dist/jobs/sse-handler.d.ts.map +1 -0
  46. package/dist/jobs/sse-handler.js +80 -0
  47. package/dist/jobs/sse-handler.js.map +1 -0
  48. package/dist/jobs/tool.d.ts +35 -0
  49. package/dist/jobs/tool.d.ts.map +1 -0
  50. package/dist/jobs/tool.js +333 -0
  51. package/dist/jobs/tool.js.map +1 -0
  52. package/dist/jobs/types.d.ts +295 -0
  53. package/dist/jobs/types.d.ts.map +1 -0
  54. package/dist/jobs/types.js +95 -0
  55. package/dist/jobs/types.js.map +1 -0
  56. package/dist/tool-ids.d.ts +3 -2
  57. package/dist/tool-ids.d.ts.map +1 -1
  58. package/dist/tool-ids.js +10 -2
  59. package/dist/tool-ids.js.map +1 -1
  60. package/dist/types/tool-result.d.ts +1 -0
  61. package/dist/types/tool-result.d.ts.map +1 -1
  62. package/dist/use-space.d.ts +0 -1
  63. package/dist/use-space.d.ts.map +1 -1
  64. package/dist/use-space.js +0 -1
  65. package/dist/use-space.js.map +1 -1
  66. package/package.json +4 -2
  67. package/src/docs-search/docs-semantic-search.ts +71 -20
  68. package/src/hf-api-call.ts +6 -0
  69. package/src/index.ts +1 -0
  70. package/src/jobs/api-client.ts +195 -0
  71. package/src/jobs/commands/inspect.ts +38 -0
  72. package/src/jobs/commands/logs.ts +36 -0
  73. package/src/jobs/commands/ps.ts +40 -0
  74. package/src/jobs/commands/run.ts +134 -0
  75. package/src/jobs/commands/scheduled.ts +189 -0
  76. package/src/jobs/commands/utils.ts +158 -0
  77. package/src/jobs/formatters.ts +149 -0
  78. package/src/jobs/sse-handler.ts +144 -0
  79. package/src/jobs/tool.ts +435 -0
  80. package/src/jobs/types.ts +237 -0
  81. package/src/tool-ids.ts +11 -1
  82. package/src/types/tool-result.ts +6 -0
  83. package/src/use-space.ts +0 -1
  84. package/test/jobs/command-translation.spec.ts +277 -0
  85. package/test/jobs/formatters.spec.ts +267 -0
  86. package/test/jobs/uv-command.spec.ts +81 -0
  87. package/dist/types/mcp-ui-server-shim.d.ts +0 -37
  88. package/dist/types/mcp-ui-server-shim.d.ts.map +0 -1
  89. package/dist/types/mcp-ui-server-shim.js +0 -2
  90. package/dist/types/mcp-ui-server-shim.js.map +0 -1
  91. 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"}