87 lines
7.0 KiB
TypeScript
87 lines
7.0 KiB
TypeScript
import type { CloudAgentOperations } from '../../shared/cloud-agents';
|
|
|
|
type OperationSpec = {
|
|
method: string; path: string; body?: readonly string[]; output: readonly string[];
|
|
query?: Record<string, string>; cloud?: 'ws';
|
|
};
|
|
const prompt = ['request_id', 'thread_id', 'query', 'expected_revision', 'attachment_file_ids'];
|
|
const request = ['request_id', 'thread_id', 'run_id', 'status', 'version'];
|
|
const schedule = ['name', 'prompt', 'cron_expression', 'timezone', 'enabled'];
|
|
const job = ['id', 'agent_slug', ...schedule, 'next_run_at', 'runs'];
|
|
const entry = ['slug', 'name', 'purpose', 'published_version', 'is_creator', 'payer'];
|
|
/** Only these product operations can cross the Main boundary; no arbitrary upstream URL or credentials. */
|
|
const operations: Record<keyof CloudAgentOperations, OperationSpec> = {
|
|
knowledge: { method: 'GET', path: '/agents/:slug/knowledge', output: ['databases', 'models'] },
|
|
createKnowledge: { method: 'POST', path: '/agents/:slug/knowledge', body: ['operation_id', 'name', 'embedding_model'], output: ['kb_id', 'name', 'description', 'embedding_model'] },
|
|
knowledgeFiles: { method: 'GET', path: '/agents/:slug/knowledge/:kb_id/files', query: { offset: 'offset' }, output: ['files', 'next_offset'] },
|
|
processKnowledge: { method: 'POST', path: '/agents/:slug/knowledge/:kb_id/files/:file_id/process', body: ['operation_id'], output: ['task_id'] },
|
|
catalog: { method: 'GET', path: '/catalog', output: ['models', 'resources', 'pricing'] },
|
|
received: { method: 'GET', path: '/received', output: ['agents'] },
|
|
entry: { method: 'GET', path: '/entry/:slug', output: entry },
|
|
publish: { method: 'POST', path: '/agents/:slug/publish', body: ['operation_id', 'expected_revision'], output: ['version', 'draft_revision'] },
|
|
access: { method: 'GET', path: '/agents/:slug/access', output: ['published_version', 'enabled', 'share_url', 'versions', 'grants', 'applications'] },
|
|
setEnabled: { method: 'PATCH', path: '/agents/:slug/enabled', body: ['enabled'], output: ['enabled'] },
|
|
users: { method: 'GET', cloud: 'ws', path: '/users', query: { query: 'query' }, output: ['users'] },
|
|
share: { method: 'PUT', path: '/agents/:slug/shares/:account_id', body: ['enabled'], output: ['account_id', 'enabled'] },
|
|
createApplication: { method: 'POST', path: '/agents/:slug/applications', body: ['operation_id', 'name'], output: ['id', 'name', 'enabled'] },
|
|
setApplicationEnabled: { method: 'PATCH', path: '/applications/:application_id', body: ['enabled'], output: ['enabled'] },
|
|
keys: { method: 'GET', path: '/applications/:application_id/keys', output: ['keys'] },
|
|
createKey: { method: 'POST', path: '/applications/:application_id/keys', body: ['operation_id'], output: ['id', 'prefix', 'secret'] },
|
|
revokeKey: { method: 'DELETE', path: '/applications/:application_id/keys/:key_id', output: ['revoked'] },
|
|
costs: { method: 'GET', cloud: 'ws', path: '/costs', query: { slug: 'agent_slug' }, output: ['unit', 'items'] },
|
|
threads: { method: 'GET', path: '/threads', query: { slug: 'slug', offset: 'offset' }, output: ['threads', 'next_offset'] },
|
|
createThread: { method: 'POST', path: '/agents/:slug/threads', body: ['thread_id', 'preview', 'expected_revision'], output: ['thread_id', 'client_thread_id'] },
|
|
history: { method: 'GET', path: '/threads/:thread_id', query: { offset: 'offset' }, output: ['thread_id', 'messages', 'run', 'next_offset'] },
|
|
viewed: { method: 'POST', path: '/threads/:thread_id/viewed', body: ['run_id'], output: ['viewed'] },
|
|
submit: { method: 'POST', path: '/agents/:slug/requests', body: prompt, output: request },
|
|
preview: { method: 'POST', path: '/agents/:slug/preview', body: prompt, output: request },
|
|
request: { method: 'GET', path: '/requests/:request_id', output: request },
|
|
cancelRequest: { method: 'POST', path: '/requests/:request_id/cancel', output: ['status'] },
|
|
run: { method: 'GET', path: '/runs/:run_id', output: ['agent_run_id', 'request_id', 'thread_id', 'agent_slug', 'status', 'output', 'version', 'error', 'interrupt'] },
|
|
cancelRun: { method: 'POST', path: '/runs/:run_id/cancel', output: ['status'] },
|
|
resume: { method: 'POST', path: '/runs/:run_id/resume', body: ['operation_id', 'decision'], output: ['run_id', 'status'] },
|
|
schedules: { method: 'GET', path: '/agents/:slug/schedules', output: ['jobs'] },
|
|
createSchedule: { method: 'POST', path: '/agents/:slug/schedules', body: ['operation_id', ...schedule], output: job },
|
|
updateSchedule: { method: 'PUT', path: '/agents/:slug/schedules/:job_id', body: schedule, output: job },
|
|
deleteSchedule: { method: 'DELETE', path: '/agents/:slug/schedules/:job_id', output: ['deleted'] },
|
|
runSchedule: { method: 'POST', path: '/agents/:slug/schedules/:job_id/run-now', body: ['operation_id'], output: ['thread_id', 'status'] },
|
|
attachments: { method: 'GET', path: '/threads/:thread_id/attachments', output: ['attachments'] },
|
|
parseAttachment: { method: 'POST', path: '/attachments/tmp/parse', body: ['object_name', 'parse_method'], output: ['parsed_object_name'] },
|
|
confirmAttachment: { method: 'POST', path: '/threads/:thread_id/attachments/confirm', body: ['attachments'], output: ['attachments'] },
|
|
deleteAttachment: { method: 'DELETE', path: '/threads/:thread_id/attachments/:file_id', output: ['message'] },
|
|
files: { method: 'GET', path: '/threads/:thread_id/files', query: { path: 'path' }, output: ['files'] },
|
|
};
|
|
|
|
export function operationPlan(value: unknown) {
|
|
if (!value || typeof value !== 'object') throw new Error('invalid_operation');
|
|
const { operation, input } = value as { operation?: unknown; input?: unknown };
|
|
if (typeof operation !== 'string' || !Object.hasOwn(operations, operation)
|
|
|| !input || typeof input !== 'object' || Array.isArray(input)) throw new Error('invalid_operation');
|
|
const spec = operations[operation as keyof CloudAgentOperations];
|
|
const args = input as Record<string, unknown>;
|
|
const path = spec.path.replace(/:([a-z_]+)/g, (_, key: string) => {
|
|
const id = args[key];
|
|
if (typeof id !== 'string' || !/^[a-zA-Z0-9_-]{1,128}$/.test(id)
|
|
|| (key === 'slug' && !/^ml-[a-f0-9]{32}$/.test(id))) throw new Error('invalid_id');
|
|
return encodeURIComponent(id);
|
|
});
|
|
const query = new URLSearchParams();
|
|
for (const [key, name] of Object.entries(spec.query ?? {})) {
|
|
const v = args[key];
|
|
if (v === undefined) continue;
|
|
if (key === 'offset' ? !Number.isSafeInteger(v) || Number(v) < 0 : typeof v !== 'string' || v.length > (key === 'path' ? 2048 : 100)) {
|
|
throw new Error('invalid_query');
|
|
}
|
|
query.set(name, String(v));
|
|
}
|
|
return {
|
|
...spec, path: (spec.cloud === 'ws' ? '/api/cloud-agents' : '/api/makelore') + path + (query.size ? '?' + query : ''),
|
|
body: spec.body ? Object.fromEntries(spec.body.filter(key => args[key] !== undefined).map(key => [key, args[key]])) : undefined,
|
|
project: (result: unknown) => {
|
|
if (!result || typeof result !== 'object' || Array.isArray(result)) throw new Error('invalid_response');
|
|
const record = result as Record<string, unknown>;
|
|
return Object.fromEntries(spec.output.filter(key => record[key] !== undefined).map(key => [key, record[key]]));
|
|
},
|
|
};
|
|
}
|