@wiro-ai/wiro-mcp 1.3.4 → 1.4.0
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/README.md +173 -16
- package/dist/client.d.ts +10 -0
- package/dist/client.d.ts.map +1 -1
- package/dist/client.js +39 -20
- package/dist/client.js.map +1 -1
- package/dist/index.js +4 -2
- package/dist/index.js.map +1 -1
- package/dist/instructions.d.ts +2 -0
- package/dist/instructions.d.ts.map +1 -0
- package/dist/instructions.js +25 -0
- package/dist/instructions.js.map +1 -0
- package/dist/playbooks.d.ts +8 -0
- package/dist/playbooks.d.ts.map +1 -0
- package/dist/playbooks.js +665 -0
- package/dist/playbooks.js.map +1 -0
- package/dist/prompts.d.ts +4 -0
- package/dist/prompts.d.ts.map +1 -0
- package/dist/prompts.js +133 -0
- package/dist/prompts.js.map +1 -0
- package/dist/server.d.ts +9 -2
- package/dist/server.d.ts.map +1 -1
- package/dist/server.js +9 -3
- package/dist/server.js.map +1 -1
- package/dist/tools/apply-agent-skills.d.ts +4 -0
- package/dist/tools/apply-agent-skills.d.ts.map +1 -0
- package/dist/tools/apply-agent-skills.js +545 -0
- package/dist/tools/apply-agent-skills.js.map +1 -0
- package/dist/tools/cancel-agent-message.d.ts +4 -0
- package/dist/tools/cancel-agent-message.d.ts.map +1 -0
- package/dist/tools/cancel-agent-message.js +126 -0
- package/dist/tools/cancel-agent-message.js.map +1 -0
- package/dist/tools/cancel-task.js +3 -3
- package/dist/tools/cancel-task.js.map +1 -1
- package/dist/tools/clone-workflow.d.ts +4 -0
- package/dist/tools/clone-workflow.d.ts.map +1 -0
- package/dist/tools/clone-workflow.js +83 -0
- package/dist/tools/clone-workflow.js.map +1 -0
- package/dist/tools/connect-agent-oauth.d.ts +4 -0
- package/dist/tools/connect-agent-oauth.d.ts.map +1 -0
- package/dist/tools/connect-agent-oauth.js +508 -0
- package/dist/tools/connect-agent-oauth.js.map +1 -0
- package/dist/tools/control-agent.d.ts +4 -0
- package/dist/tools/control-agent.d.ts.map +1 -0
- package/dist/tools/control-agent.js +274 -0
- package/dist/tools/control-agent.js.map +1 -0
- package/dist/tools/create-workflow.d.ts +100 -0
- package/dist/tools/create-workflow.d.ts.map +1 -0
- package/dist/tools/create-workflow.js +103 -0
- package/dist/tools/create-workflow.js.map +1 -0
- package/dist/tools/delete-agent-session.d.ts +6 -0
- package/dist/tools/delete-agent-session.d.ts.map +1 -0
- package/dist/tools/delete-agent-session.js +134 -0
- package/dist/tools/delete-agent-session.js.map +1 -0
- package/dist/tools/delete-agent.d.ts +4 -0
- package/dist/tools/delete-agent.d.ts.map +1 -0
- package/dist/tools/delete-agent.js +144 -0
- package/dist/tools/delete-agent.js.map +1 -0
- package/dist/tools/delete-workflow.d.ts +4 -0
- package/dist/tools/delete-workflow.d.ts.map +1 -0
- package/dist/tools/delete-workflow.js +77 -0
- package/dist/tools/delete-workflow.js.map +1 -0
- package/dist/tools/deploy-agent.d.ts +4 -0
- package/dist/tools/deploy-agent.d.ts.map +1 -0
- package/dist/tools/deploy-agent.js +455 -0
- package/dist/tools/deploy-agent.js.map +1 -0
- package/dist/tools/disconnect-agent-oauth.d.ts +4 -0
- package/dist/tools/disconnect-agent-oauth.d.ts.map +1 -0
- package/dist/tools/disconnect-agent-oauth.js +118 -0
- package/dist/tools/disconnect-agent-oauth.js.map +1 -0
- package/dist/tools/get-account.d.ts +4 -0
- package/dist/tools/get-account.d.ts.map +1 -0
- package/dist/tools/get-account.js +110 -0
- package/dist/tools/get-account.js.map +1 -0
- package/dist/tools/get-agent-activity.d.ts +4 -0
- package/dist/tools/get-agent-activity.d.ts.map +1 -0
- package/dist/tools/get-agent-activity.js +383 -0
- package/dist/tools/get-agent-activity.js.map +1 -0
- package/dist/tools/get-agent-conversation.d.ts +4 -0
- package/dist/tools/get-agent-conversation.d.ts.map +1 -0
- package/dist/tools/get-agent-conversation.js +224 -0
- package/dist/tools/get-agent-conversation.js.map +1 -0
- package/dist/tools/get-agent-message.d.ts +4 -0
- package/dist/tools/get-agent-message.d.ts.map +1 -0
- package/dist/tools/get-agent-message.js +68 -0
- package/dist/tools/get-agent-message.js.map +1 -0
- package/dist/tools/get-agent-template.d.ts +4 -0
- package/dist/tools/get-agent-template.d.ts.map +1 -0
- package/dist/tools/get-agent-template.js +107 -0
- package/dist/tools/get-agent-template.js.map +1 -0
- package/dist/tools/get-model-schema.d.ts +5 -1
- package/dist/tools/get-model-schema.d.ts.map +1 -1
- package/dist/tools/get-model-schema.js +136 -8
- package/dist/tools/get-model-schema.js.map +1 -1
- package/dist/tools/get-my-agent.d.ts +4 -0
- package/dist/tools/get-my-agent.d.ts.map +1 -0
- package/dist/tools/get-my-agent.js +78 -0
- package/dist/tools/get-my-agent.js.map +1 -0
- package/dist/tools/get-playbook.d.ts +3 -0
- package/dist/tools/get-playbook.d.ts.map +1 -0
- package/dist/tools/get-playbook.js +50 -0
- package/dist/tools/get-playbook.js.map +1 -0
- package/dist/tools/get-task-price.d.ts.map +1 -1
- package/dist/tools/get-task-price.js +42 -15
- package/dist/tools/get-task-price.js.map +1 -1
- package/dist/tools/get-task.d.ts.map +1 -1
- package/dist/tools/get-task.js +4 -3
- package/dist/tools/get-task.js.map +1 -1
- package/dist/tools/get-workflow.d.ts +9 -0
- package/dist/tools/get-workflow.d.ts.map +1 -0
- package/dist/tools/get-workflow.js +99 -0
- package/dist/tools/get-workflow.js.map +1 -0
- package/dist/tools/index.d.ts +62 -2
- package/dist/tools/index.d.ts.map +1 -1
- package/dist/tools/index.js +127 -14
- package/dist/tools/index.js.map +1 -1
- package/dist/tools/kill-task.d.ts.map +1 -1
- package/dist/tools/kill-task.js +10 -6
- package/dist/tools/kill-task.js.map +1 -1
- package/dist/tools/list-agent-skills.d.ts +4 -0
- package/dist/tools/list-agent-skills.d.ts.map +1 -0
- package/dist/tools/list-agent-skills.js +242 -0
- package/dist/tools/list-agent-skills.js.map +1 -0
- package/dist/tools/list-agent-templates.d.ts +4 -0
- package/dist/tools/list-agent-templates.d.ts.map +1 -0
- package/dist/tools/list-agent-templates.js +110 -0
- package/dist/tools/list-agent-templates.js.map +1 -0
- package/dist/tools/list-my-agents.d.ts +4 -0
- package/dist/tools/list-my-agents.d.ts.map +1 -0
- package/dist/tools/list-my-agents.js +90 -0
- package/dist/tools/list-my-agents.js.map +1 -0
- package/dist/tools/list-studio-results.d.ts +4 -0
- package/dist/tools/list-studio-results.d.ts.map +1 -0
- package/dist/tools/list-studio-results.js +95 -0
- package/dist/tools/list-studio-results.js.map +1 -0
- package/dist/tools/list-tasks.d.ts.map +1 -1
- package/dist/tools/list-tasks.js +7 -5
- package/dist/tools/list-tasks.js.map +1 -1
- package/dist/tools/list-workflow-runs.d.ts +4 -0
- package/dist/tools/list-workflow-runs.d.ts.map +1 -0
- package/dist/tools/list-workflow-runs.js +110 -0
- package/dist/tools/list-workflow-runs.js.map +1 -0
- package/dist/tools/list-workflows.d.ts +4 -0
- package/dist/tools/list-workflows.d.ts.map +1 -0
- package/dist/tools/list-workflows.js +143 -0
- package/dist/tools/list-workflows.js.map +1 -0
- package/dist/tools/manage-agent-billing.d.ts +4 -0
- package/dist/tools/manage-agent-billing.d.ts.map +1 -0
- package/dist/tools/manage-agent-billing.js +764 -0
- package/dist/tools/manage-agent-billing.js.map +1 -0
- package/dist/tools/manage-agent-custom-skill.d.ts +4 -0
- package/dist/tools/manage-agent-custom-skill.d.ts.map +1 -0
- package/dist/tools/manage-agent-custom-skill.js +537 -0
- package/dist/tools/manage-agent-custom-skill.js.map +1 -0
- package/dist/tools/recommend-model.d.ts.map +1 -1
- package/dist/tools/recommend-model.js +6 -5
- package/dist/tools/recommend-model.js.map +1 -1
- package/dist/tools/resume-workflow-run.d.ts +4 -0
- package/dist/tools/resume-workflow-run.d.ts.map +1 -0
- package/dist/tools/resume-workflow-run.js +69 -0
- package/dist/tools/resume-workflow-run.js.map +1 -0
- package/dist/tools/run-model.d.ts.map +1 -1
- package/dist/tools/run-model.js +39 -3
- package/dist/tools/run-model.js.map +1 -1
- package/dist/tools/run-workflow.d.ts +6 -0
- package/dist/tools/run-workflow.d.ts.map +1 -0
- package/dist/tools/run-workflow.js +132 -0
- package/dist/tools/run-workflow.js.map +1 -0
- package/dist/tools/save-workflow-from-agent.d.ts +4 -0
- package/dist/tools/save-workflow-from-agent.d.ts.map +1 -0
- package/dist/tools/save-workflow-from-agent.js +283 -0
- package/dist/tools/save-workflow-from-agent.js.map +1 -0
- package/dist/tools/schemas.d.ts +62 -12
- package/dist/tools/schemas.d.ts.map +1 -1
- package/dist/tools/schemas.js +17 -3
- package/dist/tools/schemas.js.map +1 -1
- package/dist/tools/search-models.js +6 -6
- package/dist/tools/search-models.js.map +1 -1
- package/dist/tools/send-agent-message.d.ts +4 -0
- package/dist/tools/send-agent-message.d.ts.map +1 -0
- package/dist/tools/send-agent-message.js +156 -0
- package/dist/tools/send-agent-message.js.map +1 -0
- package/dist/tools/set-agent-credentials.d.ts +4 -0
- package/dist/tools/set-agent-credentials.d.ts.map +1 -0
- package/dist/tools/set-agent-credentials.js +529 -0
- package/dist/tools/set-agent-credentials.js.map +1 -0
- package/dist/tools/update-agent.d.ts +4 -0
- package/dist/tools/update-agent.d.ts.map +1 -0
- package/dist/tools/update-agent.js +178 -0
- package/dist/tools/update-agent.js.map +1 -0
- package/dist/tools/update-workflow.d.ts +201 -0
- package/dist/tools/update-workflow.d.ts.map +1 -0
- package/dist/tools/update-workflow.js +168 -0
- package/dist/tools/update-workflow.js.map +1 -0
- package/dist/tools/upload-file.d.ts.map +1 -1
- package/dist/tools/upload-file.js +4 -3
- package/dist/tools/upload-file.js.map +1 -1
- package/dist/tools/wait-for-workflow-run.d.ts +259 -0
- package/dist/tools/wait-for-workflow-run.d.ts.map +1 -0
- package/dist/tools/wait-for-workflow-run.js +190 -0
- package/dist/tools/wait-for-workflow-run.js.map +1 -0
- package/dist/types.d.ts +12 -0
- package/dist/types.d.ts.map +1 -1
- package/dist/types.js.map +1 -1
- package/dist/utils/agent-catalog.d.ts +1256 -0
- package/dist/utils/agent-catalog.d.ts.map +1 -0
- package/dist/utils/agent-catalog.js +388 -0
- package/dist/utils/agent-catalog.js.map +1 -0
- package/dist/utils/agent-config.d.ts +441 -0
- package/dist/utils/agent-config.d.ts.map +1 -0
- package/dist/utils/agent-config.js +1667 -0
- package/dist/utils/agent-config.js.map +1 -0
- package/dist/utils/agent-message.d.ts +115 -0
- package/dist/utils/agent-message.d.ts.map +1 -0
- package/dist/utils/agent-message.js +246 -0
- package/dist/utils/agent-message.js.map +1 -0
- package/dist/utils/agent-targets.d.ts +76 -0
- package/dist/utils/agent-targets.d.ts.map +1 -0
- package/dist/utils/agent-targets.js +408 -0
- package/dist/utils/agent-targets.js.map +1 -0
- package/dist/utils/agent-turns.d.ts +354 -0
- package/dist/utils/agent-turns.d.ts.map +1 -0
- package/dist/utils/agent-turns.js +623 -0
- package/dist/utils/agent-turns.js.map +1 -0
- package/dist/utils/api.d.ts +40 -0
- package/dist/utils/api.d.ts.map +1 -0
- package/dist/utils/api.js +111 -0
- package/dist/utils/api.js.map +1 -0
- package/dist/utils/format.d.ts +38 -1
- package/dist/utils/format.d.ts.map +1 -1
- package/dist/utils/format.js +162 -13
- package/dist/utils/format.js.map +1 -1
- package/dist/utils/media.d.ts +28 -0
- package/dist/utils/media.d.ts.map +1 -0
- package/dist/utils/media.js +65 -0
- package/dist/utils/media.js.map +1 -0
- package/dist/utils/poll.d.ts +39 -0
- package/dist/utils/poll.d.ts.map +1 -0
- package/dist/utils/poll.js +111 -0
- package/dist/utils/poll.js.map +1 -0
- package/dist/utils/remote-file.js +2 -2
- package/dist/utils/remote-file.js.map +1 -1
- package/dist/utils/structured.d.ts +2 -0
- package/dist/utils/structured.d.ts.map +1 -1
- package/dist/utils/structured.js +9 -6
- package/dist/utils/structured.js.map +1 -1
- package/dist/utils/studio-runs.d.ts +87 -0
- package/dist/utils/studio-runs.d.ts.map +1 -0
- package/dist/utils/studio-runs.js +167 -0
- package/dist/utils/studio-runs.js.map +1 -0
- package/dist/utils/user-agent.d.ts +1140 -0
- package/dist/utils/user-agent.d.ts.map +1 -0
- package/dist/utils/user-agent.js +802 -0
- package/dist/utils/user-agent.js.map +1 -0
- package/dist/utils/workflow-agent.d.ts +151 -0
- package/dist/utils/workflow-agent.d.ts.map +1 -0
- package/dist/utils/workflow-agent.js +292 -0
- package/dist/utils/workflow-agent.js.map +1 -0
- package/dist/utils/workflow-artifacts.d.ts +160 -0
- package/dist/utils/workflow-artifacts.d.ts.map +1 -0
- package/dist/utils/workflow-artifacts.js +801 -0
- package/dist/utils/workflow-artifacts.js.map +1 -0
- package/dist/utils/workflow-save.d.ts +123 -0
- package/dist/utils/workflow-save.d.ts.map +1 -0
- package/dist/utils/workflow-save.js +734 -0
- package/dist/utils/workflow-save.js.map +1 -0
- package/dist/utils/workflow.d.ts +559 -0
- package/dist/utils/workflow.d.ts.map +1 -0
- package/dist/utils/workflow.js +613 -0
- package/dist/utils/workflow.js.map +1 -0
- package/package.json +1 -1
|
@@ -0,0 +1,99 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { errorResult, requireApiSuccess } from '../utils/api.js';
|
|
3
|
+
import { formatWorkflowInputs, parseWorkflowSlug, toWorkflowDetail, workflowDetailSchema, } from '../utils/workflow.js';
|
|
4
|
+
import { nextActionSchema } from './schemas.js';
|
|
5
|
+
export async function fetchWorkflowRow(client, fullSlug, signal) {
|
|
6
|
+
const result = requireApiSuccess(await client.post('/Workflow/Detail', { slug: fullSlug }, signal), 'Workflow not found.');
|
|
7
|
+
const row = result.workflow?.[0];
|
|
8
|
+
if (!row)
|
|
9
|
+
throw new Error('Workflow not found.');
|
|
10
|
+
return row;
|
|
11
|
+
}
|
|
12
|
+
export function registerGetWorkflow(server, client) {
|
|
13
|
+
server.registerTool('get_workflow', {
|
|
14
|
+
title: 'Get a Wiro workflow',
|
|
15
|
+
description: 'Read one workflow: its inputs (the keys run_workflow takes), outputs, '
|
|
16
|
+
+ 'state, and the full step graph when you are allowed to see it. Read this before '
|
|
17
|
+
+ 'running a workflow, and before editing one with update_workflow (pass the returned '
|
|
18
|
+
+ '`updatetime` back as `loaded_updatetime`). Protected workflows from other creators '
|
|
19
|
+
+ 'show only their inputs, but they can still be run. Input lists and saved values hide '
|
|
20
|
+
+ 'the values of fields named like api_key, token, secret or password, and values that '
|
|
21
|
+
+ 'look like a provider key. Other saved values come back as saved: never repeat a key '
|
|
22
|
+
+ 'you see there.',
|
|
23
|
+
inputSchema: {
|
|
24
|
+
slug: z.string().min(1)
|
|
25
|
+
.describe('The full workflow slug ("<name>-<5 character id>") or the workflow page URL.'),
|
|
26
|
+
include_graph: z.boolean().default(true)
|
|
27
|
+
.describe('Include the step graph (contents). Set false for a short answer.'),
|
|
28
|
+
},
|
|
29
|
+
outputSchema: {
|
|
30
|
+
workflow: workflowDetailSchema,
|
|
31
|
+
nextAction: nextActionSchema.optional(),
|
|
32
|
+
},
|
|
33
|
+
annotations: {
|
|
34
|
+
readOnlyHint: true,
|
|
35
|
+
destructiveHint: false,
|
|
36
|
+
idempotentHint: true,
|
|
37
|
+
openWorldHint: true,
|
|
38
|
+
},
|
|
39
|
+
}, async ({ slug, include_graph }, extra) => {
|
|
40
|
+
try {
|
|
41
|
+
const fullSlug = parseWorkflowSlug(slug);
|
|
42
|
+
const row = await fetchWorkflowRow(client, fullSlug, extra.signal);
|
|
43
|
+
const workflow = toWorkflowDetail(row, { includeGraph: include_graph });
|
|
44
|
+
// Wiro shows a protected workflow's graph only inside its own workspace. A
|
|
45
|
+
// run keeps the saved values for the owner and the team's members, whatever
|
|
46
|
+
// workspace their key is in, and leaves them out for anyone else. A caller
|
|
47
|
+
// outside the workflow's workspace can still own it or be in its team, so
|
|
48
|
+
// the text names both cases.
|
|
49
|
+
const leftOut = workflow.graphHidden && !workflow.insider
|
|
50
|
+
? 'This protected workflow is outside this key\'s workspace. Unless you own it or are in its team, a key you leave out uses its default, not its saved value, so pass the saved values you want.'
|
|
51
|
+
: 'A key you leave out uses its saved value or default.';
|
|
52
|
+
const lines = [`## ${workflow.title}`, ''];
|
|
53
|
+
lines.push(`**Slug:** \`${workflow.fullSlug}\``);
|
|
54
|
+
lines.push(`**Page:** ${workflow.url}`);
|
|
55
|
+
if (workflow.state)
|
|
56
|
+
lines.push(`**State:** ${workflow.state}`);
|
|
57
|
+
if (workflow.updatetime && workflow.insider)
|
|
58
|
+
lines.push(`**updatetime:** \`${workflow.updatetime}\` (pass it as loaded_updatetime when editing)`);
|
|
59
|
+
const count = workflow.stepCount !== undefined ? `${workflow.stepCount}, ` : '';
|
|
60
|
+
lines.push(`**Steps:** ${!workflow.graphHidden
|
|
61
|
+
? workflow.stepCount
|
|
62
|
+
: workflow.isClone
|
|
63
|
+
? `${count}hidden. This is a copy of a protected workflow, so its steps stay hidden, even from you. You can run it and change its title, description and tags.`
|
|
64
|
+
: `${count}hidden by the creator`}`);
|
|
65
|
+
if (workflow.description)
|
|
66
|
+
lines.push('', workflow.description);
|
|
67
|
+
lines.push('', '### Inputs', '', ...formatWorkflowInputs(workflow.inputs));
|
|
68
|
+
if (workflow.graphHidden && !workflow.insider && workflow.inputs.length > 0)
|
|
69
|
+
lines.push('', leftOut);
|
|
70
|
+
const outputs = Object.keys(workflow.outputs);
|
|
71
|
+
if (outputs.length > 0)
|
|
72
|
+
lines.push('', `### Outputs`, '', ...outputs.map(key => `- \`${key}\``));
|
|
73
|
+
if (workflow.estimatedCost?.valueUsd !== undefined && workflow.estimatedCost.valueUsd !== null) {
|
|
74
|
+
lines.push('', `**Estimated cost per run:** about $${workflow.estimatedCost.valueUsd}`);
|
|
75
|
+
}
|
|
76
|
+
if (workflow.contents) {
|
|
77
|
+
lines.push('', 'The full step graph is in the structured result under `workflow.contents`.');
|
|
78
|
+
}
|
|
79
|
+
return {
|
|
80
|
+
content: [{ type: 'text', text: lines.join('\n') }],
|
|
81
|
+
structuredContent: {
|
|
82
|
+
workflow,
|
|
83
|
+
nextAction: {
|
|
84
|
+
tool: 'run_workflow',
|
|
85
|
+
arguments: { slug: workflow.fullSlug },
|
|
86
|
+
reason: 'Run it only if the user asked for a run, after telling them the estimate; otherwise ask the user first (each model step is charged). Pass input values'
|
|
87
|
+
+ (workflow.inputs.length > 0
|
|
88
|
+
? ` for: ${workflow.inputs.map(input => input.key).join(', ')}. ${leftOut}`
|
|
89
|
+
: '. It has no inputs.'),
|
|
90
|
+
},
|
|
91
|
+
},
|
|
92
|
+
};
|
|
93
|
+
}
|
|
94
|
+
catch (error) {
|
|
95
|
+
return errorResult(error);
|
|
96
|
+
}
|
|
97
|
+
});
|
|
98
|
+
}
|
|
99
|
+
//# sourceMappingURL=get-workflow.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"get-workflow.js","sourceRoot":"","sources":["../../src/tools/get-workflow.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAoB,MAAM,iBAAiB,CAAC;AACnF,OAAO,EACL,oBAAoB,EACpB,iBAAiB,EACjB,gBAAgB,EAChB,oBAAoB,GACrB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAMhD,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,MAAkB,EAClB,QAAgB,EAChB,MAAoB;IAEpB,MAAM,MAAM,GAAG,iBAAiB,CAC9B,MAAM,MAAM,CAAC,IAAI,CAAyB,kBAAkB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,MAAM,CAAC,EACzF,qBAAqB,CACtB,CAAC;IACF,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;IACjC,IAAI,CAAC,GAAG;QAAE,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACjD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAiB,EAAE,MAAkB;IACvE,MAAM,CAAC,YAAY,CACjB,cAAc,EACd;QACE,KAAK,EAAE,qBAAqB;QAC5B,WAAW,EAAE,wEAAwE;cACjF,kFAAkF;cAClF,qFAAqF;cACrF,qFAAqF;cACrF,uFAAuF;cACvF,sFAAsF;cACtF,sFAAsF;cACtF,gBAAgB;QACpB,WAAW,EAAE;YACX,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;iBACpB,QAAQ,CAAC,8EAA8E,CAAC;YAC3F,aAAa,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;iBACrC,QAAQ,CAAC,kEAAkE,CAAC;SAChF;QACD,YAAY,EAAE;YACZ,QAAQ,EAAE,oBAAoB;YAC9B,UAAU,EAAE,gBAAgB,CAAC,QAAQ,EAAE;SACxC;QACD,WAAW,EAAE;YACX,YAAY,EAAE,IAAI;YAClB,eAAe,EAAE,KAAK;YACtB,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;SACpB;KACF,EACD,KAAK,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,KAAK,EAAE,EAAE;QACvC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,MAAM,EAAE,QAAQ,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;YACnE,MAAM,QAAQ,GAAG,gBAAgB,CAAC,GAAG,EAAE,EAAE,YAAY,EAAE,aAAa,EAAE,CAAC,CAAC;YAExE,2EAA2E;YAC3E,4EAA4E;YAC5E,2EAA2E;YAC3E,0EAA0E;YAC1E,6BAA6B;YAC7B,MAAM,OAAO,GAAG,QAAQ,CAAC,WAAW,IAAI,CAAC,QAAQ,CAAC,OAAO;gBACvD,CAAC,CAAC,+LAA+L;gBACjM,CAAC,CAAC,sDAAsD,CAAC;YAC3D,MAAM,KAAK,GAAG,CAAC,MAAM,QAAQ,CAAC,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YAC3C,KAAK,CAAC,IAAI,CAAC,eAAe,QAAQ,CAAC,QAAQ,IAAI,CAAC,CAAC;YACjD,KAAK,CAAC,IAAI,CAAC,aAAa,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC;YACxC,IAAI,QAAQ,CAAC,KAAK;gBAAE,KAAK,CAAC,IAAI,CAAC,cAAc,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;YAC/D,IAAI,QAAQ,CAAC,UAAU,IAAI,QAAQ,CAAC,OAAO;gBAAE,KAAK,CAAC,IAAI,CAAC,qBAAqB,QAAQ,CAAC,UAAU,gDAAgD,CAAC,CAAC;YAClJ,MAAM,KAAK,GAAG,QAAQ,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;YAChF,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,WAAW;gBAC5C,CAAC,CAAC,QAAQ,CAAC,SAAS;gBACpB,CAAC,CAAC,QAAQ,CAAC,OAAO;oBAChB,CAAC,CAAC,GAAG,KAAK,qJAAqJ;oBAC/J,CAAC,CAAC,GAAG,KAAK,uBAAuB,EAAE,CAAC,CAAC;YACzC,IAAI,QAAQ,CAAC,WAAW;gBAAE,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,QAAQ,CAAC,WAAW,CAAC,CAAC;YAC/D,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,GAAG,oBAAoB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;YAC3E,IAAI,QAAQ,CAAC,WAAW,IAAI,CAAC,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC;gBAAE,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACrG,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAC9C,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;gBAAE,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC;YACjG,IAAI,QAAQ,CAAC,aAAa,EAAE,QAAQ,KAAK,SAAS,IAAI,QAAQ,CAAC,aAAa,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;gBAC/F,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,sCAAsC,QAAQ,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC,CAAC;YAC1F,CAAC;YACD,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACtB,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,4EAA4E,CAAC,CAAC;YAC/F,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5D,iBAAiB,EAAE;oBACjB,QAAQ;oBACR,UAAU,EAAE;wBACV,IAAI,EAAE,cAAc;wBACpB,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,QAAQ,EAAE;wBACtC,MAAM,EAAE,wJAAwJ;8BAC5J,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC;gCAC3B,CAAC,CAAC,SAAS,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,OAAO,EAAE;gCAC3E,CAAC,CAAC,qBAAqB,CAAC;qBAC7B;iBACF;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,WAAW,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}
|
package/dist/tools/index.d.ts
CHANGED
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
2
|
+
import type { ToolAnnotations } from '@modelcontextprotocol/sdk/types.js';
|
|
2
3
|
import type { WiroClient } from '../client.js';
|
|
3
4
|
import { registerSearchModels } from './search-models.js';
|
|
4
5
|
import { registerGetModelSchema } from './get-model-schema.js';
|
|
@@ -13,6 +14,65 @@ import { registerSearchDocs } from './search-docs.js';
|
|
|
13
14
|
import { registerRecommendModel } from './recommend-model.js';
|
|
14
15
|
import { registerExplore } from './explore.js';
|
|
15
16
|
import { registerUploadFile } from './upload-file.js';
|
|
16
|
-
|
|
17
|
-
|
|
17
|
+
import { registerGetAccount } from './get-account.js';
|
|
18
|
+
import { registerListWorkflows } from './list-workflows.js';
|
|
19
|
+
import { registerGetWorkflow } from './get-workflow.js';
|
|
20
|
+
import { registerCreateWorkflow } from './create-workflow.js';
|
|
21
|
+
import { registerUpdateWorkflow } from './update-workflow.js';
|
|
22
|
+
import { registerCloneWorkflow } from './clone-workflow.js';
|
|
23
|
+
import { registerDeleteWorkflow } from './delete-workflow.js';
|
|
24
|
+
import { registerRunWorkflow } from './run-workflow.js';
|
|
25
|
+
import { registerWaitForWorkflowRun } from './wait-for-workflow-run.js';
|
|
26
|
+
import { registerResumeWorkflowRun } from './resume-workflow-run.js';
|
|
27
|
+
import { registerListWorkflowRuns } from './list-workflow-runs.js';
|
|
28
|
+
import { registerGetPlaybook } from './get-playbook.js';
|
|
29
|
+
import { registerSendAgentMessage } from './send-agent-message.js';
|
|
30
|
+
import { registerGetAgentMessage } from './get-agent-message.js';
|
|
31
|
+
import { registerGetAgentConversation } from './get-agent-conversation.js';
|
|
32
|
+
import { registerCancelAgentMessage } from './cancel-agent-message.js';
|
|
33
|
+
import { registerDeleteAgentSession } from './delete-agent-session.js';
|
|
34
|
+
import { registerListStudioResults } from './list-studio-results.js';
|
|
35
|
+
import { registerSaveWorkflowFromAgent } from './save-workflow-from-agent.js';
|
|
36
|
+
import { registerListAgentTemplates } from './list-agent-templates.js';
|
|
37
|
+
import { registerGetAgentTemplate } from './get-agent-template.js';
|
|
38
|
+
import { registerListAgentSkills } from './list-agent-skills.js';
|
|
39
|
+
import { registerDeployAgent } from './deploy-agent.js';
|
|
40
|
+
import { registerListMyAgents } from './list-my-agents.js';
|
|
41
|
+
import { registerGetMyAgent } from './get-my-agent.js';
|
|
42
|
+
import { registerControlAgent } from './control-agent.js';
|
|
43
|
+
import { registerUpdateAgent } from './update-agent.js';
|
|
44
|
+
import { registerManageAgentBilling } from './manage-agent-billing.js';
|
|
45
|
+
import { registerGetAgentActivity } from './get-agent-activity.js';
|
|
46
|
+
import { registerDeleteAgent } from './delete-agent.js';
|
|
47
|
+
import { registerApplyAgentSkills } from './apply-agent-skills.js';
|
|
48
|
+
import { registerSetAgentCredentials } from './set-agent-credentials.js';
|
|
49
|
+
import { registerConnectAgentOauth } from './connect-agent-oauth.js';
|
|
50
|
+
import { registerDisconnectAgentOauth } from './disconnect-agent-oauth.js';
|
|
51
|
+
import { registerManageAgentCustomSkill } from './manage-agent-custom-skill.js';
|
|
52
|
+
export declare const TOOLSETS: readonly ["models", "workflows", "agents"];
|
|
53
|
+
export type Toolset = typeof TOOLSETS[number];
|
|
54
|
+
/** What a tool wrapper is told about the tool it wraps. */
|
|
55
|
+
export interface ToolInfo {
|
|
56
|
+
name: string;
|
|
57
|
+
title?: string;
|
|
58
|
+
annotations?: ToolAnnotations;
|
|
59
|
+
}
|
|
60
|
+
/** A tool's handler as the MCP server calls it: the arguments, then the request context. */
|
|
61
|
+
export type ToolHandler = (...args: never[]) => unknown;
|
|
62
|
+
/**
|
|
63
|
+
* Lets a server that embeds these tools put its own checks around every call,
|
|
64
|
+
* such as logging or an extra permission check. It gets each tool and its handler
|
|
65
|
+
* and returns the handler to register.
|
|
66
|
+
*/
|
|
67
|
+
export type ToolWrapper = (tool: ToolInfo, handler: ToolHandler) => ToolHandler;
|
|
68
|
+
export interface RegisterToolsOptions {
|
|
69
|
+
/** Which groups of tools to register. Defaults to all of them. */
|
|
70
|
+
toolsets?: readonly Toolset[];
|
|
71
|
+
/** Wraps every tool's handler before it is registered. */
|
|
72
|
+
wrapTool?: ToolWrapper;
|
|
73
|
+
}
|
|
74
|
+
/** Parses a comma-separated list such as "models,workflows"; unknown names are ignored. */
|
|
75
|
+
export declare function parseToolsets(value: string | undefined): Toolset[] | undefined;
|
|
76
|
+
export declare function registerTools(mcpServer: McpServer, client: WiroClient, options?: RegisterToolsOptions): void;
|
|
77
|
+
export { registerSearchModels, registerGetModelSchema, registerRecommendModel, registerExplore, registerRunModel, registerWaitForTask, registerGetTask, registerListTasks, registerGetTaskPrice, registerCancelTask, registerKillTask, registerUploadFile, registerSearchDocs, registerGetAccount, registerListWorkflows, registerGetWorkflow, registerCreateWorkflow, registerUpdateWorkflow, registerCloneWorkflow, registerDeleteWorkflow, registerRunWorkflow, registerWaitForWorkflowRun, registerResumeWorkflowRun, registerListWorkflowRuns, registerGetPlaybook, registerSendAgentMessage, registerGetAgentMessage, registerGetAgentConversation, registerCancelAgentMessage, registerDeleteAgentSession, registerListStudioResults, registerSaveWorkflowFromAgent, registerListAgentTemplates, registerGetAgentTemplate, registerListAgentSkills, registerDeployAgent, registerListMyAgents, registerGetMyAgent, registerControlAgent, registerUpdateAgent, registerManageAgentBilling, registerGetAgentActivity, registerDeleteAgent, registerApplyAgentSkills, registerSetAgentCredentials, registerConnectAgentOauth, registerDisconnectAgentOauth, registerManageAgentCustomSkill, };
|
|
18
78
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,oCAAoC,CAAC;AAC1E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,0BAA0B,EAAE,MAAM,4BAA4B,CAAC;AACxE,OAAO,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,4BAA4B,EAAE,MAAM,6BAA6B,CAAC;AAC3E,OAAO,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,6BAA6B,EAAE,MAAM,+BAA+B,CAAC;AAC9E,OAAO,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,2BAA2B,EAAE,MAAM,4BAA4B,CAAC;AACzE,OAAO,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,4BAA4B,EAAE,MAAM,6BAA6B,CAAC;AAC3E,OAAO,EAAE,8BAA8B,EAAE,MAAM,gCAAgC,CAAC;AAEhF,eAAO,MAAM,QAAQ,4CAA6C,CAAC;AACnE,MAAM,MAAM,OAAO,GAAG,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC;AAE9C,2DAA2D;AAC3D,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,eAAe,CAAC;CAC/B;AAED,4FAA4F;AAC5F,MAAM,MAAM,WAAW,GAAG,CAAC,GAAG,IAAI,EAAE,KAAK,EAAE,KAAK,OAAO,CAAC;AAExD;;;;GAIG;AACH,MAAM,MAAM,WAAW,GAAG,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,WAAW,KAAK,WAAW,CAAC;AAEhF,MAAM,WAAW,oBAAoB;IACnC,kEAAkE;IAClE,QAAQ,CAAC,EAAE,SAAS,OAAO,EAAE,CAAC;IAC9B,0DAA0D;IAC1D,QAAQ,CAAC,EAAE,WAAW,CAAC;CACxB;AAkBD,2FAA2F;AAC3F,wBAAgB,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,SAAS,GAAG,OAAO,EAAE,GAAG,SAAS,CAK9E;AAED,wBAAgB,aAAa,CAC3B,SAAS,EAAE,SAAS,EACpB,MAAM,EAAE,UAAU,EAClB,OAAO,GAAE,oBAAyB,GACjC,IAAI,CA6EN;AAED,OAAO,EACL,oBAAoB,EACpB,sBAAsB,EACtB,sBAAsB,EACtB,eAAe,EACf,gBAAgB,EAChB,mBAAmB,EACnB,eAAe,EACf,iBAAiB,EACjB,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,EAChB,kBAAkB,EAClB,kBAAkB,EAClB,kBAAkB,EAClB,qBAAqB,EACrB,mBAAmB,EACnB,sBAAsB,EACtB,sBAAsB,EACtB,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EACnB,0BAA0B,EAC1B,yBAAyB,EACzB,wBAAwB,EACxB,mBAAmB,EACnB,wBAAwB,EACxB,uBAAuB,EACvB,4BAA4B,EAC5B,0BAA0B,EAC1B,0BAA0B,EAC1B,yBAAyB,EACzB,6BAA6B,EAC7B,0BAA0B,EAC1B,wBAAwB,EACxB,uBAAuB,EACvB,mBAAmB,EACnB,oBAAoB,EACpB,kBAAkB,EAClB,oBAAoB,EACpB,mBAAmB,EACnB,0BAA0B,EAC1B,wBAAwB,EACxB,mBAAmB,EACnB,wBAAwB,EACxB,2BAA2B,EAC3B,yBAAyB,EACzB,4BAA4B,EAC5B,8BAA8B,GAC/B,CAAC"}
|
package/dist/tools/index.js
CHANGED
|
@@ -11,20 +11,133 @@ import { registerSearchDocs } from './search-docs.js';
|
|
|
11
11
|
import { registerRecommendModel } from './recommend-model.js';
|
|
12
12
|
import { registerExplore } from './explore.js';
|
|
13
13
|
import { registerUploadFile } from './upload-file.js';
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
14
|
+
import { registerGetAccount } from './get-account.js';
|
|
15
|
+
import { registerListWorkflows } from './list-workflows.js';
|
|
16
|
+
import { registerGetWorkflow } from './get-workflow.js';
|
|
17
|
+
import { registerCreateWorkflow } from './create-workflow.js';
|
|
18
|
+
import { registerUpdateWorkflow } from './update-workflow.js';
|
|
19
|
+
import { registerCloneWorkflow } from './clone-workflow.js';
|
|
20
|
+
import { registerDeleteWorkflow } from './delete-workflow.js';
|
|
21
|
+
import { registerRunWorkflow } from './run-workflow.js';
|
|
22
|
+
import { registerWaitForWorkflowRun } from './wait-for-workflow-run.js';
|
|
23
|
+
import { registerResumeWorkflowRun } from './resume-workflow-run.js';
|
|
24
|
+
import { registerListWorkflowRuns } from './list-workflow-runs.js';
|
|
25
|
+
import { registerGetPlaybook } from './get-playbook.js';
|
|
26
|
+
import { registerSendAgentMessage } from './send-agent-message.js';
|
|
27
|
+
import { registerGetAgentMessage } from './get-agent-message.js';
|
|
28
|
+
import { registerGetAgentConversation } from './get-agent-conversation.js';
|
|
29
|
+
import { registerCancelAgentMessage } from './cancel-agent-message.js';
|
|
30
|
+
import { registerDeleteAgentSession } from './delete-agent-session.js';
|
|
31
|
+
import { registerListStudioResults } from './list-studio-results.js';
|
|
32
|
+
import { registerSaveWorkflowFromAgent } from './save-workflow-from-agent.js';
|
|
33
|
+
import { registerListAgentTemplates } from './list-agent-templates.js';
|
|
34
|
+
import { registerGetAgentTemplate } from './get-agent-template.js';
|
|
35
|
+
import { registerListAgentSkills } from './list-agent-skills.js';
|
|
36
|
+
import { registerDeployAgent } from './deploy-agent.js';
|
|
37
|
+
import { registerListMyAgents } from './list-my-agents.js';
|
|
38
|
+
import { registerGetMyAgent } from './get-my-agent.js';
|
|
39
|
+
import { registerControlAgent } from './control-agent.js';
|
|
40
|
+
import { registerUpdateAgent } from './update-agent.js';
|
|
41
|
+
import { registerManageAgentBilling } from './manage-agent-billing.js';
|
|
42
|
+
import { registerGetAgentActivity } from './get-agent-activity.js';
|
|
43
|
+
import { registerDeleteAgent } from './delete-agent.js';
|
|
44
|
+
import { registerApplyAgentSkills } from './apply-agent-skills.js';
|
|
45
|
+
import { registerSetAgentCredentials } from './set-agent-credentials.js';
|
|
46
|
+
import { registerConnectAgentOauth } from './connect-agent-oauth.js';
|
|
47
|
+
import { registerDisconnectAgentOauth } from './disconnect-agent-oauth.js';
|
|
48
|
+
import { registerManageAgentCustomSkill } from './manage-agent-custom-skill.js';
|
|
49
|
+
export const TOOLSETS = ['models', 'workflows', 'agents'];
|
|
50
|
+
/** The server as the register functions see it, with every handler passed through `wrap`. */
|
|
51
|
+
function withToolWrapper(server, wrap) {
|
|
52
|
+
const registerTool = server.registerTool.bind(server);
|
|
53
|
+
return new Proxy(server, {
|
|
54
|
+
get(target, property, receiver) {
|
|
55
|
+
if (property !== 'registerTool')
|
|
56
|
+
return Reflect.get(target, property, receiver);
|
|
57
|
+
return (name, config, handler) => registerTool(name, config, wrap({ name, title: config.title, annotations: config.annotations }, handler));
|
|
58
|
+
},
|
|
59
|
+
});
|
|
60
|
+
}
|
|
61
|
+
/** Parses a comma-separated list such as "models,workflows"; unknown names are ignored. */
|
|
62
|
+
export function parseToolsets(value) {
|
|
63
|
+
if (!value || !value.trim())
|
|
64
|
+
return undefined;
|
|
65
|
+
const requested = value.split(',').map(name => name.trim().toLowerCase());
|
|
66
|
+
const toolsets = TOOLSETS.filter(toolset => requested.includes(toolset));
|
|
67
|
+
return toolsets.length > 0 ? toolsets : undefined;
|
|
68
|
+
}
|
|
69
|
+
export function registerTools(mcpServer, client, options = {}) {
|
|
70
|
+
const server = options.wrapTool ? withToolWrapper(mcpServer, options.wrapTool) : mcpServer;
|
|
71
|
+
const enabled = new Set(options.toolsets ?? TOOLSETS);
|
|
72
|
+
// Always available: account context, guides and documentation.
|
|
73
|
+
registerGetAccount(server, client);
|
|
74
|
+
registerGetPlaybook(server);
|
|
27
75
|
registerSearchDocs(server);
|
|
76
|
+
if (enabled.has('models')) {
|
|
77
|
+
registerSearchModels(server, client);
|
|
78
|
+
registerGetModelSchema(server, client);
|
|
79
|
+
registerRecommendModel(server, client);
|
|
80
|
+
registerExplore(server, client);
|
|
81
|
+
registerRunModel(server, client);
|
|
82
|
+
registerWaitForTask(server, client);
|
|
83
|
+
registerGetTask(server, client);
|
|
84
|
+
registerListTasks(server, client);
|
|
85
|
+
registerGetTaskPrice(server, client);
|
|
86
|
+
registerCancelTask(server, client);
|
|
87
|
+
registerKillTask(server, client);
|
|
88
|
+
registerUploadFile(server, client);
|
|
89
|
+
}
|
|
90
|
+
if (enabled.has('workflows')) {
|
|
91
|
+
if (!enabled.has('models')) {
|
|
92
|
+
// Authoring a workflow needs the model catalog. Without run_model, the
|
|
93
|
+
// schema points at workflow steps instead.
|
|
94
|
+
registerSearchModels(server, client);
|
|
95
|
+
registerGetModelSchema(server, client, { runModel: false });
|
|
96
|
+
registerRecommendModel(server, client);
|
|
97
|
+
registerUploadFile(server, client);
|
|
98
|
+
}
|
|
99
|
+
registerListWorkflows(server, client);
|
|
100
|
+
registerGetWorkflow(server, client);
|
|
101
|
+
registerCreateWorkflow(server, client);
|
|
102
|
+
registerUpdateWorkflow(server, client);
|
|
103
|
+
registerCloneWorkflow(server, client);
|
|
104
|
+
registerDeleteWorkflow(server, client);
|
|
105
|
+
registerRunWorkflow(server, client);
|
|
106
|
+
registerWaitForWorkflowRun(server, client);
|
|
107
|
+
registerResumeWorkflowRun(server, client);
|
|
108
|
+
registerListWorkflowRuns(server, client);
|
|
109
|
+
}
|
|
110
|
+
// One conversation API serves Studio, the Workflow Agent and deployed agents,
|
|
111
|
+
// so Studio and the Workflow Agent are reachable from either toolset. The Studio
|
|
112
|
+
// guidance collects late results with list_studio_results, and a Workflow Agent
|
|
113
|
+
// reply points at save_workflow_from_agent, so both come along.
|
|
114
|
+
if (enabled.has('agents') || enabled.has('workflows')) {
|
|
115
|
+
registerSendAgentMessage(server, client);
|
|
116
|
+
registerGetAgentMessage(server, client);
|
|
117
|
+
registerGetAgentConversation(server, client);
|
|
118
|
+
registerCancelAgentMessage(server, client);
|
|
119
|
+
registerDeleteAgentSession(server, client);
|
|
120
|
+
registerListStudioResults(server, client);
|
|
121
|
+
registerSaveWorkflowFromAgent(server, client);
|
|
122
|
+
}
|
|
123
|
+
if (enabled.has('agents')) {
|
|
124
|
+
registerListAgentTemplates(server, client);
|
|
125
|
+
registerGetAgentTemplate(server, client);
|
|
126
|
+
registerListAgentSkills(server, client);
|
|
127
|
+
registerDeployAgent(server, client);
|
|
128
|
+
registerListMyAgents(server, client);
|
|
129
|
+
registerGetMyAgent(server, client);
|
|
130
|
+
registerControlAgent(server, client);
|
|
131
|
+
registerUpdateAgent(server, client);
|
|
132
|
+
registerManageAgentBilling(server, client);
|
|
133
|
+
registerGetAgentActivity(server, client);
|
|
134
|
+
registerDeleteAgent(server, client);
|
|
135
|
+
registerApplyAgentSkills(server, client);
|
|
136
|
+
registerSetAgentCredentials(server, client);
|
|
137
|
+
registerConnectAgentOauth(server, client);
|
|
138
|
+
registerDisconnectAgentOauth(server, client);
|
|
139
|
+
registerManageAgentCustomSkill(server, client);
|
|
140
|
+
}
|
|
28
141
|
}
|
|
29
|
-
export { registerSearchModels, registerGetModelSchema, registerRecommendModel, registerExplore, registerRunModel, registerWaitForTask, registerGetTask, registerListTasks, registerGetTaskPrice, registerCancelTask, registerKillTask, registerUploadFile, registerSearchDocs, };
|
|
142
|
+
export { registerSearchModels, registerGetModelSchema, registerRecommendModel, registerExplore, registerRunModel, registerWaitForTask, registerGetTask, registerListTasks, registerGetTaskPrice, registerCancelTask, registerKillTask, registerUploadFile, registerSearchDocs, registerGetAccount, registerListWorkflows, registerGetWorkflow, registerCreateWorkflow, registerUpdateWorkflow, registerCloneWorkflow, registerDeleteWorkflow, registerRunWorkflow, registerWaitForWorkflowRun, registerResumeWorkflowRun, registerListWorkflowRuns, registerGetPlaybook, registerSendAgentMessage, registerGetAgentMessage, registerGetAgentConversation, registerCancelAgentMessage, registerDeleteAgentSession, registerListStudioResults, registerSaveWorkflowFromAgent, registerListAgentTemplates, registerGetAgentTemplate, registerListAgentSkills, registerDeployAgent, registerListMyAgents, registerGetMyAgent, registerControlAgent, registerUpdateAgent, registerManageAgentBilling, registerGetAgentActivity, registerDeleteAgent, registerApplyAgentSkills, registerSetAgentCredentials, registerConnectAgentOauth, registerDisconnectAgentOauth, registerManageAgentCustomSkill, };
|
|
30
143
|
//# sourceMappingURL=index.js.map
|
package/dist/tools/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,0BAA0B,EAAE,MAAM,4BAA4B,CAAC;AACxE,OAAO,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,4BAA4B,EAAE,MAAM,6BAA6B,CAAC;AAC3E,OAAO,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,6BAA6B,EAAE,MAAM,+BAA+B,CAAC;AAC9E,OAAO,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,2BAA2B,EAAE,MAAM,4BAA4B,CAAC;AACzE,OAAO,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,4BAA4B,EAAE,MAAM,6BAA6B,CAAC;AAC3E,OAAO,EAAE,8BAA8B,EAAE,MAAM,gCAAgC,CAAC;AAEhF,MAAM,CAAC,MAAM,QAAQ,GAAG,CAAC,QAAQ,EAAE,WAAW,EAAE,QAAQ,CAAU,CAAC;AA2BnE,6FAA6F;AAC7F,SAAS,eAAe,CAAC,MAAiB,EAAE,IAAiB;IAC3D,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAIxC,CAAC;IACb,OAAO,IAAI,KAAK,CAAC,MAAM,EAAE;QACvB,GAAG,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ;YAC5B,IAAI,QAAQ,KAAK,cAAc;gBAAE,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAChF,OAAO,CAAC,IAAY,EAAE,MAAyD,EAAE,OAAoB,EAAE,EAAE,CACvG,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC;QAC9G,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,2FAA2F;AAC3F,MAAM,UAAU,aAAa,CAAC,KAAyB;IACrD,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE;QAAE,OAAO,SAAS,CAAC;IAC9C,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC;IAC1E,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;IACzE,OAAO,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,aAAa,CAC3B,SAAoB,EACpB,MAAkB,EAClB,UAAgC,EAAE;IAElC,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,SAAS,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3F,MAAM,OAAO,GAAG,IAAI,GAAG,CAAU,OAAO,CAAC,QAAQ,IAAI,QAAQ,CAAC,CAAC;IAE/D,+DAA+D;IAC/D,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnC,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAC5B,kBAAkB,CAAC,MAAM,CAAC,CAAC;IAE3B,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1B,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,sBAAsB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACvC,sBAAsB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACvC,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAChC,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACjC,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpC,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAChC,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAClC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACnC,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACjC,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACrC,CAAC;IAED,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3B,uEAAuE;YACvE,2CAA2C;YAC3C,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACrC,sBAAsB,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;YAC5D,sBAAsB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACvC,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,CAAC;QACD,qBAAqB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACtC,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpC,sBAAsB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACvC,sBAAsB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACvC,qBAAqB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACtC,sBAAsB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACvC,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpC,0BAA0B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC3C,yBAAyB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC1C,wBAAwB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3C,CAAC;IAED,8EAA8E;IAC9E,iFAAiF;IACjF,gFAAgF;IAChF,gEAAgE;IAChE,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;QACtD,wBAAwB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACzC,uBAAuB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACxC,4BAA4B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC7C,0BAA0B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC3C,0BAA0B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC3C,yBAAyB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC1C,6BAA6B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChD,CAAC;IAED,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1B,0BAA0B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC3C,wBAAwB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACzC,uBAAuB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACxC,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACnC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpC,0BAA0B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC3C,wBAAwB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACzC,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpC,wBAAwB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACzC,2BAA2B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC5C,yBAAyB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC1C,4BAA4B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC7C,8BAA8B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjD,CAAC;AACH,CAAC;AAED,OAAO,EACL,oBAAoB,EACpB,sBAAsB,EACtB,sBAAsB,EACtB,eAAe,EACf,gBAAgB,EAChB,mBAAmB,EACnB,eAAe,EACf,iBAAiB,EACjB,oBAAoB,EACpB,kBAAkB,EAClB,gBAAgB,EAChB,kBAAkB,EAClB,kBAAkB,EAClB,kBAAkB,EAClB,qBAAqB,EACrB,mBAAmB,EACnB,sBAAsB,EACtB,sBAAsB,EACtB,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EACnB,0BAA0B,EAC1B,yBAAyB,EACzB,wBAAwB,EACxB,mBAAmB,EACnB,wBAAwB,EACxB,uBAAuB,EACvB,4BAA4B,EAC5B,0BAA0B,EAC1B,0BAA0B,EAC1B,yBAAyB,EACzB,6BAA6B,EAC7B,0BAA0B,EAC1B,wBAAwB,EACxB,uBAAuB,EACvB,mBAAmB,EACnB,oBAAoB,EACpB,kBAAkB,EAClB,oBAAoB,EACpB,mBAAmB,EACnB,0BAA0B,EAC1B,wBAAwB,EACxB,mBAAmB,EACnB,wBAAwB,EACxB,2BAA2B,EAC3B,yBAAyB,EACzB,4BAA4B,EAC5B,8BAA8B,GAC/B,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"kill-task.d.ts","sourceRoot":"","sources":["../../src/tools/kill-task.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEzE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;
|
|
1
|
+
{"version":3,"file":"kill-task.d.ts","sourceRoot":"","sources":["../../src/tools/kill-task.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEzE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAM/C,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,GAAG,IAAI,CA8D5E"}
|
package/dist/tools/kill-task.js
CHANGED
|
@@ -1,11 +1,15 @@
|
|
|
1
1
|
import { z } from 'zod';
|
|
2
2
|
import { taskActionOutputSchema } from './schemas.js';
|
|
3
|
+
const STOP_REQUESTED = 'Stop requested; the task now shows as cancelled. A task that was still queued is not '
|
|
4
|
+
+ 'charged; one that was already running can still be charged for its run.';
|
|
3
5
|
export function registerKillTask(server, client) {
|
|
4
6
|
server.registerTool('kill_task', {
|
|
5
|
-
title: '
|
|
6
|
-
description: 'Stop a task that
|
|
7
|
-
+ '
|
|
8
|
-
+ '
|
|
7
|
+
title: 'Stop a Wiro task',
|
|
8
|
+
description: 'Stop a model task that has not finished. A queued task is cancelled and '
|
|
9
|
+
+ 'never charged. A task that was already running can still be charged for its run, '
|
|
10
|
+
+ 'and get_task_price shows any charge once the run ends. If the task finished first, '
|
|
11
|
+
+ 'the call is refused with "Process ended": read it with get_task and show the result. '
|
|
12
|
+
+ 'Not for the steps of a workflow run or the runs Studio starts.',
|
|
9
13
|
inputSchema: {
|
|
10
14
|
tasktoken: z.string().optional().describe('The task token returned from run_model.'),
|
|
11
15
|
taskid: z.string().optional().describe('The task ID (alternative to tasktoken).'),
|
|
@@ -34,7 +38,7 @@ export function registerKillTask(server, client) {
|
|
|
34
38
|
};
|
|
35
39
|
}
|
|
36
40
|
return {
|
|
37
|
-
content: [{ type: 'text', text:
|
|
41
|
+
content: [{ type: 'text', text: STOP_REQUESTED }],
|
|
38
42
|
structuredContent: {
|
|
39
43
|
action: 'kill',
|
|
40
44
|
success: true,
|
|
@@ -42,7 +46,7 @@ export function registerKillTask(server, client) {
|
|
|
42
46
|
...(taskid ? { id: taskid } : {}),
|
|
43
47
|
...(tasktoken ? { token: tasktoken } : {}),
|
|
44
48
|
},
|
|
45
|
-
message:
|
|
49
|
+
message: STOP_REQUESTED,
|
|
46
50
|
},
|
|
47
51
|
};
|
|
48
52
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"kill-task.js","sourceRoot":"","sources":["../../src/tools/kill-task.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC;AAEtD,MAAM,UAAU,gBAAgB,CAAC,MAAiB,EAAE,MAAkB;IACpE,MAAM,CAAC,YAAY,CACjB,WAAW,EACX;QACE,KAAK,EAAE,
|
|
1
|
+
{"version":3,"file":"kill-task.js","sourceRoot":"","sources":["../../src/tools/kill-task.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC;AAEtD,MAAM,cAAc,GAAG,uFAAuF;MAC1G,yEAAyE,CAAC;AAE9E,MAAM,UAAU,gBAAgB,CAAC,MAAiB,EAAE,MAAkB;IACpE,MAAM,CAAC,YAAY,CACjB,WAAW,EACX;QACE,KAAK,EAAE,kBAAkB;QACzB,WAAW,EAAE,0EAA0E;cACnF,mFAAmF;cACnF,qFAAqF;cACrF,uFAAuF;cACvF,gEAAgE;QACpE,WAAW,EAAE;YACX,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;YACpF,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;SAClF;QACD,YAAY,EAAE,sBAAsB;QACpC,WAAW,EAAE;YACX,YAAY,EAAE,KAAK;YACnB,eAAe,EAAE,IAAI;YACrB,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;SACpB;KACF,EACD,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE;QACrC,IAAI,CAAC;YACH,IAAI,CAAC,SAAS,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC1B,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,yDAAyD,EAAE,CAAC;oBACrG,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAClC,EAAE,SAAS,EAAE,MAAM,EAAE,EACrB,KAAK,CAAC,MAAM,CACb,CAAC;YACF,IAAI,MAAM,CAAC,MAAM,KAAK,KAAK,EAAE,CAAC;gBAC5B,MAAM,GAAG,GAAI,MAAM,CAAC,MAAqC,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,aAAa,CAAC;gBAC3G,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,eAAe,GAAG,EAAE,EAAE,CAAC;oBAChE,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;gBAC1D,iBAAiB,EAAE;oBACjB,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE,IAAI;oBACb,IAAI,EAAE;wBACJ,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;wBACjC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC3C;oBACD,OAAO,EAAE,cAAc;iBACxB;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,eAAe,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;gBACnH,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"list-agent-skills.d.ts","sourceRoot":"","sources":["../../src/tools/list-agent-skills.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEzE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAgI/C,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,GAAG,IAAI,CAgKnF"}
|