@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.
Files changed (270) hide show
  1. package/README.md +173 -16
  2. package/dist/client.d.ts +10 -0
  3. package/dist/client.d.ts.map +1 -1
  4. package/dist/client.js +39 -20
  5. package/dist/client.js.map +1 -1
  6. package/dist/index.js +4 -2
  7. package/dist/index.js.map +1 -1
  8. package/dist/instructions.d.ts +2 -0
  9. package/dist/instructions.d.ts.map +1 -0
  10. package/dist/instructions.js +25 -0
  11. package/dist/instructions.js.map +1 -0
  12. package/dist/playbooks.d.ts +8 -0
  13. package/dist/playbooks.d.ts.map +1 -0
  14. package/dist/playbooks.js +665 -0
  15. package/dist/playbooks.js.map +1 -0
  16. package/dist/prompts.d.ts +4 -0
  17. package/dist/prompts.d.ts.map +1 -0
  18. package/dist/prompts.js +133 -0
  19. package/dist/prompts.js.map +1 -0
  20. package/dist/server.d.ts +9 -2
  21. package/dist/server.d.ts.map +1 -1
  22. package/dist/server.js +9 -3
  23. package/dist/server.js.map +1 -1
  24. package/dist/tools/apply-agent-skills.d.ts +4 -0
  25. package/dist/tools/apply-agent-skills.d.ts.map +1 -0
  26. package/dist/tools/apply-agent-skills.js +545 -0
  27. package/dist/tools/apply-agent-skills.js.map +1 -0
  28. package/dist/tools/cancel-agent-message.d.ts +4 -0
  29. package/dist/tools/cancel-agent-message.d.ts.map +1 -0
  30. package/dist/tools/cancel-agent-message.js +126 -0
  31. package/dist/tools/cancel-agent-message.js.map +1 -0
  32. package/dist/tools/cancel-task.js +3 -3
  33. package/dist/tools/cancel-task.js.map +1 -1
  34. package/dist/tools/clone-workflow.d.ts +4 -0
  35. package/dist/tools/clone-workflow.d.ts.map +1 -0
  36. package/dist/tools/clone-workflow.js +83 -0
  37. package/dist/tools/clone-workflow.js.map +1 -0
  38. package/dist/tools/connect-agent-oauth.d.ts +4 -0
  39. package/dist/tools/connect-agent-oauth.d.ts.map +1 -0
  40. package/dist/tools/connect-agent-oauth.js +508 -0
  41. package/dist/tools/connect-agent-oauth.js.map +1 -0
  42. package/dist/tools/control-agent.d.ts +4 -0
  43. package/dist/tools/control-agent.d.ts.map +1 -0
  44. package/dist/tools/control-agent.js +274 -0
  45. package/dist/tools/control-agent.js.map +1 -0
  46. package/dist/tools/create-workflow.d.ts +100 -0
  47. package/dist/tools/create-workflow.d.ts.map +1 -0
  48. package/dist/tools/create-workflow.js +103 -0
  49. package/dist/tools/create-workflow.js.map +1 -0
  50. package/dist/tools/delete-agent-session.d.ts +6 -0
  51. package/dist/tools/delete-agent-session.d.ts.map +1 -0
  52. package/dist/tools/delete-agent-session.js +134 -0
  53. package/dist/tools/delete-agent-session.js.map +1 -0
  54. package/dist/tools/delete-agent.d.ts +4 -0
  55. package/dist/tools/delete-agent.d.ts.map +1 -0
  56. package/dist/tools/delete-agent.js +144 -0
  57. package/dist/tools/delete-agent.js.map +1 -0
  58. package/dist/tools/delete-workflow.d.ts +4 -0
  59. package/dist/tools/delete-workflow.d.ts.map +1 -0
  60. package/dist/tools/delete-workflow.js +77 -0
  61. package/dist/tools/delete-workflow.js.map +1 -0
  62. package/dist/tools/deploy-agent.d.ts +4 -0
  63. package/dist/tools/deploy-agent.d.ts.map +1 -0
  64. package/dist/tools/deploy-agent.js +455 -0
  65. package/dist/tools/deploy-agent.js.map +1 -0
  66. package/dist/tools/disconnect-agent-oauth.d.ts +4 -0
  67. package/dist/tools/disconnect-agent-oauth.d.ts.map +1 -0
  68. package/dist/tools/disconnect-agent-oauth.js +118 -0
  69. package/dist/tools/disconnect-agent-oauth.js.map +1 -0
  70. package/dist/tools/get-account.d.ts +4 -0
  71. package/dist/tools/get-account.d.ts.map +1 -0
  72. package/dist/tools/get-account.js +110 -0
  73. package/dist/tools/get-account.js.map +1 -0
  74. package/dist/tools/get-agent-activity.d.ts +4 -0
  75. package/dist/tools/get-agent-activity.d.ts.map +1 -0
  76. package/dist/tools/get-agent-activity.js +383 -0
  77. package/dist/tools/get-agent-activity.js.map +1 -0
  78. package/dist/tools/get-agent-conversation.d.ts +4 -0
  79. package/dist/tools/get-agent-conversation.d.ts.map +1 -0
  80. package/dist/tools/get-agent-conversation.js +224 -0
  81. package/dist/tools/get-agent-conversation.js.map +1 -0
  82. package/dist/tools/get-agent-message.d.ts +4 -0
  83. package/dist/tools/get-agent-message.d.ts.map +1 -0
  84. package/dist/tools/get-agent-message.js +68 -0
  85. package/dist/tools/get-agent-message.js.map +1 -0
  86. package/dist/tools/get-agent-template.d.ts +4 -0
  87. package/dist/tools/get-agent-template.d.ts.map +1 -0
  88. package/dist/tools/get-agent-template.js +107 -0
  89. package/dist/tools/get-agent-template.js.map +1 -0
  90. package/dist/tools/get-model-schema.d.ts +5 -1
  91. package/dist/tools/get-model-schema.d.ts.map +1 -1
  92. package/dist/tools/get-model-schema.js +136 -8
  93. package/dist/tools/get-model-schema.js.map +1 -1
  94. package/dist/tools/get-my-agent.d.ts +4 -0
  95. package/dist/tools/get-my-agent.d.ts.map +1 -0
  96. package/dist/tools/get-my-agent.js +78 -0
  97. package/dist/tools/get-my-agent.js.map +1 -0
  98. package/dist/tools/get-playbook.d.ts +3 -0
  99. package/dist/tools/get-playbook.d.ts.map +1 -0
  100. package/dist/tools/get-playbook.js +50 -0
  101. package/dist/tools/get-playbook.js.map +1 -0
  102. package/dist/tools/get-task-price.d.ts.map +1 -1
  103. package/dist/tools/get-task-price.js +42 -15
  104. package/dist/tools/get-task-price.js.map +1 -1
  105. package/dist/tools/get-task.d.ts.map +1 -1
  106. package/dist/tools/get-task.js +4 -3
  107. package/dist/tools/get-task.js.map +1 -1
  108. package/dist/tools/get-workflow.d.ts +9 -0
  109. package/dist/tools/get-workflow.d.ts.map +1 -0
  110. package/dist/tools/get-workflow.js +99 -0
  111. package/dist/tools/get-workflow.js.map +1 -0
  112. package/dist/tools/index.d.ts +62 -2
  113. package/dist/tools/index.d.ts.map +1 -1
  114. package/dist/tools/index.js +127 -14
  115. package/dist/tools/index.js.map +1 -1
  116. package/dist/tools/kill-task.d.ts.map +1 -1
  117. package/dist/tools/kill-task.js +10 -6
  118. package/dist/tools/kill-task.js.map +1 -1
  119. package/dist/tools/list-agent-skills.d.ts +4 -0
  120. package/dist/tools/list-agent-skills.d.ts.map +1 -0
  121. package/dist/tools/list-agent-skills.js +242 -0
  122. package/dist/tools/list-agent-skills.js.map +1 -0
  123. package/dist/tools/list-agent-templates.d.ts +4 -0
  124. package/dist/tools/list-agent-templates.d.ts.map +1 -0
  125. package/dist/tools/list-agent-templates.js +110 -0
  126. package/dist/tools/list-agent-templates.js.map +1 -0
  127. package/dist/tools/list-my-agents.d.ts +4 -0
  128. package/dist/tools/list-my-agents.d.ts.map +1 -0
  129. package/dist/tools/list-my-agents.js +90 -0
  130. package/dist/tools/list-my-agents.js.map +1 -0
  131. package/dist/tools/list-studio-results.d.ts +4 -0
  132. package/dist/tools/list-studio-results.d.ts.map +1 -0
  133. package/dist/tools/list-studio-results.js +95 -0
  134. package/dist/tools/list-studio-results.js.map +1 -0
  135. package/dist/tools/list-tasks.d.ts.map +1 -1
  136. package/dist/tools/list-tasks.js +7 -5
  137. package/dist/tools/list-tasks.js.map +1 -1
  138. package/dist/tools/list-workflow-runs.d.ts +4 -0
  139. package/dist/tools/list-workflow-runs.d.ts.map +1 -0
  140. package/dist/tools/list-workflow-runs.js +110 -0
  141. package/dist/tools/list-workflow-runs.js.map +1 -0
  142. package/dist/tools/list-workflows.d.ts +4 -0
  143. package/dist/tools/list-workflows.d.ts.map +1 -0
  144. package/dist/tools/list-workflows.js +143 -0
  145. package/dist/tools/list-workflows.js.map +1 -0
  146. package/dist/tools/manage-agent-billing.d.ts +4 -0
  147. package/dist/tools/manage-agent-billing.d.ts.map +1 -0
  148. package/dist/tools/manage-agent-billing.js +764 -0
  149. package/dist/tools/manage-agent-billing.js.map +1 -0
  150. package/dist/tools/manage-agent-custom-skill.d.ts +4 -0
  151. package/dist/tools/manage-agent-custom-skill.d.ts.map +1 -0
  152. package/dist/tools/manage-agent-custom-skill.js +537 -0
  153. package/dist/tools/manage-agent-custom-skill.js.map +1 -0
  154. package/dist/tools/recommend-model.d.ts.map +1 -1
  155. package/dist/tools/recommend-model.js +6 -5
  156. package/dist/tools/recommend-model.js.map +1 -1
  157. package/dist/tools/resume-workflow-run.d.ts +4 -0
  158. package/dist/tools/resume-workflow-run.d.ts.map +1 -0
  159. package/dist/tools/resume-workflow-run.js +69 -0
  160. package/dist/tools/resume-workflow-run.js.map +1 -0
  161. package/dist/tools/run-model.d.ts.map +1 -1
  162. package/dist/tools/run-model.js +39 -3
  163. package/dist/tools/run-model.js.map +1 -1
  164. package/dist/tools/run-workflow.d.ts +6 -0
  165. package/dist/tools/run-workflow.d.ts.map +1 -0
  166. package/dist/tools/run-workflow.js +132 -0
  167. package/dist/tools/run-workflow.js.map +1 -0
  168. package/dist/tools/save-workflow-from-agent.d.ts +4 -0
  169. package/dist/tools/save-workflow-from-agent.d.ts.map +1 -0
  170. package/dist/tools/save-workflow-from-agent.js +283 -0
  171. package/dist/tools/save-workflow-from-agent.js.map +1 -0
  172. package/dist/tools/schemas.d.ts +62 -12
  173. package/dist/tools/schemas.d.ts.map +1 -1
  174. package/dist/tools/schemas.js +17 -3
  175. package/dist/tools/schemas.js.map +1 -1
  176. package/dist/tools/search-models.js +6 -6
  177. package/dist/tools/search-models.js.map +1 -1
  178. package/dist/tools/send-agent-message.d.ts +4 -0
  179. package/dist/tools/send-agent-message.d.ts.map +1 -0
  180. package/dist/tools/send-agent-message.js +156 -0
  181. package/dist/tools/send-agent-message.js.map +1 -0
  182. package/dist/tools/set-agent-credentials.d.ts +4 -0
  183. package/dist/tools/set-agent-credentials.d.ts.map +1 -0
  184. package/dist/tools/set-agent-credentials.js +529 -0
  185. package/dist/tools/set-agent-credentials.js.map +1 -0
  186. package/dist/tools/update-agent.d.ts +4 -0
  187. package/dist/tools/update-agent.d.ts.map +1 -0
  188. package/dist/tools/update-agent.js +178 -0
  189. package/dist/tools/update-agent.js.map +1 -0
  190. package/dist/tools/update-workflow.d.ts +201 -0
  191. package/dist/tools/update-workflow.d.ts.map +1 -0
  192. package/dist/tools/update-workflow.js +168 -0
  193. package/dist/tools/update-workflow.js.map +1 -0
  194. package/dist/tools/upload-file.d.ts.map +1 -1
  195. package/dist/tools/upload-file.js +4 -3
  196. package/dist/tools/upload-file.js.map +1 -1
  197. package/dist/tools/wait-for-workflow-run.d.ts +259 -0
  198. package/dist/tools/wait-for-workflow-run.d.ts.map +1 -0
  199. package/dist/tools/wait-for-workflow-run.js +190 -0
  200. package/dist/tools/wait-for-workflow-run.js.map +1 -0
  201. package/dist/types.d.ts +12 -0
  202. package/dist/types.d.ts.map +1 -1
  203. package/dist/types.js.map +1 -1
  204. package/dist/utils/agent-catalog.d.ts +1256 -0
  205. package/dist/utils/agent-catalog.d.ts.map +1 -0
  206. package/dist/utils/agent-catalog.js +388 -0
  207. package/dist/utils/agent-catalog.js.map +1 -0
  208. package/dist/utils/agent-config.d.ts +441 -0
  209. package/dist/utils/agent-config.d.ts.map +1 -0
  210. package/dist/utils/agent-config.js +1667 -0
  211. package/dist/utils/agent-config.js.map +1 -0
  212. package/dist/utils/agent-message.d.ts +115 -0
  213. package/dist/utils/agent-message.d.ts.map +1 -0
  214. package/dist/utils/agent-message.js +246 -0
  215. package/dist/utils/agent-message.js.map +1 -0
  216. package/dist/utils/agent-targets.d.ts +76 -0
  217. package/dist/utils/agent-targets.d.ts.map +1 -0
  218. package/dist/utils/agent-targets.js +408 -0
  219. package/dist/utils/agent-targets.js.map +1 -0
  220. package/dist/utils/agent-turns.d.ts +354 -0
  221. package/dist/utils/agent-turns.d.ts.map +1 -0
  222. package/dist/utils/agent-turns.js +623 -0
  223. package/dist/utils/agent-turns.js.map +1 -0
  224. package/dist/utils/api.d.ts +40 -0
  225. package/dist/utils/api.d.ts.map +1 -0
  226. package/dist/utils/api.js +111 -0
  227. package/dist/utils/api.js.map +1 -0
  228. package/dist/utils/format.d.ts +38 -1
  229. package/dist/utils/format.d.ts.map +1 -1
  230. package/dist/utils/format.js +162 -13
  231. package/dist/utils/format.js.map +1 -1
  232. package/dist/utils/media.d.ts +28 -0
  233. package/dist/utils/media.d.ts.map +1 -0
  234. package/dist/utils/media.js +65 -0
  235. package/dist/utils/media.js.map +1 -0
  236. package/dist/utils/poll.d.ts +39 -0
  237. package/dist/utils/poll.d.ts.map +1 -0
  238. package/dist/utils/poll.js +111 -0
  239. package/dist/utils/poll.js.map +1 -0
  240. package/dist/utils/remote-file.js +2 -2
  241. package/dist/utils/remote-file.js.map +1 -1
  242. package/dist/utils/structured.d.ts +2 -0
  243. package/dist/utils/structured.d.ts.map +1 -1
  244. package/dist/utils/structured.js +9 -6
  245. package/dist/utils/structured.js.map +1 -1
  246. package/dist/utils/studio-runs.d.ts +87 -0
  247. package/dist/utils/studio-runs.d.ts.map +1 -0
  248. package/dist/utils/studio-runs.js +167 -0
  249. package/dist/utils/studio-runs.js.map +1 -0
  250. package/dist/utils/user-agent.d.ts +1140 -0
  251. package/dist/utils/user-agent.d.ts.map +1 -0
  252. package/dist/utils/user-agent.js +802 -0
  253. package/dist/utils/user-agent.js.map +1 -0
  254. package/dist/utils/workflow-agent.d.ts +151 -0
  255. package/dist/utils/workflow-agent.d.ts.map +1 -0
  256. package/dist/utils/workflow-agent.js +292 -0
  257. package/dist/utils/workflow-agent.js.map +1 -0
  258. package/dist/utils/workflow-artifacts.d.ts +160 -0
  259. package/dist/utils/workflow-artifacts.d.ts.map +1 -0
  260. package/dist/utils/workflow-artifacts.js +801 -0
  261. package/dist/utils/workflow-artifacts.js.map +1 -0
  262. package/dist/utils/workflow-save.d.ts +123 -0
  263. package/dist/utils/workflow-save.d.ts.map +1 -0
  264. package/dist/utils/workflow-save.js +734 -0
  265. package/dist/utils/workflow-save.js.map +1 -0
  266. package/dist/utils/workflow.d.ts +559 -0
  267. package/dist/utils/workflow.d.ts.map +1 -0
  268. package/dist/utils/workflow.js +613 -0
  269. package/dist/utils/workflow.js.map +1 -0
  270. 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"}
@@ -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
- export declare function registerTools(server: McpServer, client: WiroClient): void;
17
- export { registerSearchModels, registerGetModelSchema, registerRecommendModel, registerExplore, registerRunModel, registerWaitForTask, registerGetTask, registerListTasks, registerGetTaskPrice, registerCancelTask, registerKillTask, registerUploadFile, registerSearchDocs, };
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;AAEtD,wBAAgB,aAAa,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,GAAG,IAAI,CAczE;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,GACnB,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"}
@@ -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
- export function registerTools(server, client) {
15
- registerSearchModels(server, client);
16
- registerGetModelSchema(server, client);
17
- registerRecommendModel(server, client);
18
- registerExplore(server, client);
19
- registerRunModel(server, client);
20
- registerWaitForTask(server, client);
21
- registerGetTask(server, client);
22
- registerListTasks(server, client);
23
- registerGetTaskPrice(server, client);
24
- registerCancelTask(server, client);
25
- registerKillTask(server, client);
26
- registerUploadFile(server, client);
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
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAEA,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;AAEtD,MAAM,UAAU,aAAa,CAAC,MAAiB,EAAE,MAAkB;IACjE,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACrC,sBAAsB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC,sBAAsB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACpC,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAClC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACrC,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnC,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnC,kBAAkB,CAAC,MAAM,CAAC,CAAC;AAC7B,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,GACnB,CAAC"}
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;AAG/C,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,GAAG,IAAI,CA4D5E"}
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"}
@@ -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: 'Kill a running Wiro task',
6
- description: 'Stop a task that is already running. This changes remote '
7
- + 'state and can discard in-progress work. Use `cancel_task` for a '
8
- + 'task that is still queued.',
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: `Task killed successfully.` }],
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: 'Task killed successfully.',
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,0BAA0B;QACjC,WAAW,EAAE,2DAA2D;cACpE,kEAAkE;cAClE,4BAA4B;QAChC,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,2BAA2B,EAAE,CAAC;gBACvE,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,2BAA2B;iBACrC;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"}
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,4 @@
1
+ import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
2
+ import type { WiroClient } from '../client.js';
3
+ export declare function registerListAgentSkills(server: McpServer, client: WiroClient): void;
4
+ //# sourceMappingURL=list-agent-skills.d.ts.map
@@ -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"}