wolfpack-mcp 1.0.88 → 1.0.90

package/dist/agentBuilderTools.js

@@ -10,6 +10,26 @@ const ORG_SLUG_PROP = {
         description: 'Organisation slug. Required when you belong to multiple organisations. Use list_organisations to find slugs.',
     },
 };
+/**
+ * The per-task model pin, shared by create_agent_task and update_agent_task so
+ * both describe it the same way. Null on update clears the pin and the task
+ * falls back to the agent's own model; omitting a field leaves it unchanged.
+ */
+const TASK_MODEL_PROPS = {
+    llm_provider: {
+        type: 'string',
+        description: 'Pin this task to an LLM provider ID (e.g. "anthropic") instead of the agent\'s own model. ' +
+            'Use list_llm_models for IDs. Set to null to unpin.',
+    },
+    llm_model: {
+        type: 'string',
+        description: 'Pin this task to a model ID (e.g. "claude-sonnet-4-6"). Set to null for family auto-select.',
+    },
+    llm_family: {
+        type: 'string',
+        description: 'Pin this task to a model family (e.g. "sonnet"); the task auto-upgrades to the latest in the family.',
+    },
+};
 export const AGENT_BUILDER_TOOLS = [
     // ─── Group 0: Organisations ─────────────────────────────────────────────
     {
@@ -338,6 +358,7 @@ export const AGENT_BUILDER_TOOLS = [
                     type: 'number',
                     description: 'Order within scheduled batch (lower = first)',
                 },
+                ...TASK_MODEL_PROPS,
                 disabled_tools: {
                     type: 'array',
                     items: { type: 'string' },
@@ -363,7 +384,7 @@ export const AGENT_BUILDER_TOOLS = [
     },
     {
         name: 'update_agent_task',
-        description: "Update a task's prompt, schedule settings, or disabled tools/MCP servers.",
+        description: "Update a task's prompt, schedule settings, model pin, or disabled tools/MCP servers.",
         inputSchema: {
             type: 'object',
             properties: {
@@ -374,6 +395,7 @@ export const AGENT_BUILDER_TOOLS = [
                 scheduled_enabled: { type: 'boolean', description: 'Include in scheduled runs' },
                 sort_order: { type: 'number', description: 'Order within scheduled batch' },
                 is_active: { type: 'boolean', description: 'Whether this task is active' },
+                ...TASK_MODEL_PROPS,
                 disabled_tools: {
                     type: 'array',
                     items: { type: 'string' },
@@ -406,6 +428,12 @@ export const AGENT_BUILDER_TOOLS = [
             properties: {
                 agent_id: { type: 'string', description: 'Agent profile ID' },
                 task_id: { type: 'string', description: 'Task ID to run' },
+                prompt: {
+                    type: 'string',
+                    description: "Extra instructions for this run only, appended to the task's saved prompt. " +
+                        'The task still runs as written — this is added after it, and the task is not changed. ' +
+                        'Omit it to run the task exactly as saved.',
+                },
                 project_slug: { type: 'string', description: 'Project to scope the session to (optional)' },
                 git_ref: {
                     type: 'string',
@@ -747,7 +775,8 @@ export const AGENT_BUILDER_TOOLS = [
     {
         name: 'list_llm_models',
         description: 'List LLM providers and models available to the organisation. ' +
-            'Use provider and model IDs when calling create_agent or update_agent.',
+            'Use provider, model and family IDs when calling create_agent, update_agent, ' +
+            'create_agent_task or update_agent_task.',
         inputSchema: { type: 'object', properties: { ...ORG_SLUG_PROP } },
     },
 ];
@@ -1014,6 +1043,9 @@ export async function handleAgentBuilderTool(name, args, client) {
                 prompt: z.string(),
                 scheduled_enabled: z.boolean().optional(),
                 sort_order: z.number().optional(),
+                llm_provider: z.string().nullable().optional(),
+                llm_model: z.string().nullable().optional(),
+                llm_family: z.string().nullable().optional(),
                 disabled_tools: z.array(z.string()).optional(),
                 disabled_mcp_servers: z.array(z.string()).optional(),
                 clone_repository: z.boolean().optional(),
@@ -1025,6 +1057,9 @@ export async function handleAgentBuilderTool(name, args, client) {
                 prompt: parsed.prompt,
                 scheduledEnabled: parsed.scheduled_enabled,
                 sortOrder: parsed.sort_order,
+                llmProvider: parsed.llm_provider,
+                llmModel: parsed.llm_model,
+                llmFamily: parsed.llm_family,
                 disabledTools: parsed.disabled_tools,
                 disabledMcpServers: parsed.disabled_mcp_servers,
                 cloneRepository: parsed.clone_repository,
@@ -1041,6 +1076,9 @@ export async function handleAgentBuilderTool(name, args, client) {
                 scheduled_enabled: z.boolean().optional(),
                 sort_order: z.number().optional(),
                 is_active: z.boolean().optional(),
+                llm_provider: z.string().nullable().optional(),
+                llm_model: z.string().nullable().optional(),
+                llm_family: z.string().nullable().optional(),
                 disabled_tools: z.array(z.string()).nullable().optional(),
                 disabled_mcp_servers: z.array(z.string()).nullable().optional(),
                 clone_repository: z.boolean().optional(),
@@ -1058,6 +1096,12 @@ export async function handleAgentBuilderTool(name, args, client) {
                 fields.sortOrder = parsed.sort_order;
             if (parsed.is_active !== undefined)
                 fields.isActive = parsed.is_active;
+            if (parsed.llm_provider !== undefined)
+                fields.llmProvider = parsed.llm_provider;
+            if (parsed.llm_model !== undefined)
+                fields.llmModel = parsed.llm_model;
+            if (parsed.llm_family !== undefined)
+                fields.llmFamily = parsed.llm_family;
             if (parsed.disabled_tools !== undefined)
                 fields.disabledTools = parsed.disabled_tools;
             if (parsed.disabled_mcp_servers !== undefined)
@@ -1072,13 +1116,14 @@ export async function handleAgentBuilderTool(name, args, client) {
                 .object({
                 agent_id: z.string(),
                 task_id: z.string(),
+                prompt: z.string().optional(),
                 project_slug: z.string().optional(),
                 git_ref: z.string().optional(),
                 pr_number: z.number().int().positive().optional(),
                 org_slug: orgSlugField,
             })
                 .parse(args);
-            const result = await client.runAgentTask(parsed.agent_id, parsed.task_id, parsed.project_slug, resolveOrg(parsed), parsed.git_ref, parsed.pr_number);
+            const result = await client.runAgentTask(parsed.agent_id, parsed.task_id, parsed.project_slug, resolveOrg(parsed), parsed.git_ref, parsed.pr_number, parsed.prompt);
             return {
                 content: [
                     {

package/dist/agentBuilderTools.test.js

@@ -0,0 +1,100 @@
+import { describe, it, expect, vi } from 'vitest';
+import { AGENT_BUILDER_TOOLS, handleAgentBuilderTool } from './agentBuilderTools.js';
+const toolNamed = (name) => {
+    const tool = AGENT_BUILDER_TOOLS.find((t) => t.name === name);
+    if (!tool)
+        throw new Error(`No tool named ${name}`);
+    return tool;
+};
+const MODEL_FIELDS = ['llm_provider', 'llm_model', 'llm_family'];
+/** A client that records the one call each test makes and returns a bare task. */
+function fakeClient() {
+    const createAgentTask = vi.fn().mockResolvedValue({ id: 'task-1', name: 'Analyse' });
+    const updateAgentTask = vi.fn().mockResolvedValue({ id: 'task-1', name: 'Analyse' });
+    return {
+        client: { createAgentTask, updateAgentTask },
+        createAgentTask,
+        updateAgentTask,
+    };
+}
+describe('per-task model pin over MCP (#2034)', () => {
+    it.each(['create_agent_task', 'update_agent_task'])('%s advertises the model fields', (name) => {
+        const properties = toolNamed(name).inputSchema.properties;
+        expect(Object.keys(properties)).toEqual(expect.arrayContaining(MODEL_FIELDS));
+    });
+    it('sends the model pin when creating a task', async () => {
+        const { client, createAgentTask } = fakeClient();
+        await handleAgentBuilderTool('create_agent_task', {
+            agent_id: 'agent-1',
+            name: 'Analyse',
+            prompt: 'Analyse the transcript.',
+            llm_provider: 'anthropic',
+            llm_model: 'claude-sonnet-4-6',
+            llm_family: 'sonnet',
+        }, client);
+        expect(createAgentTask.mock.calls[0][0]).toBe('agent-1');
+        expect(createAgentTask.mock.calls[0][1]).toMatchObject({
+            llmProvider: 'anthropic',
+            llmModel: 'claude-sonnet-4-6',
+            llmFamily: 'sonnet',
+        });
+    });
+    it('sends the model pin when updating a task', async () => {
+        const { client, updateAgentTask } = fakeClient();
+        await handleAgentBuilderTool('update_agent_task', {
+            agent_id: 'agent-1',
+            task_id: 'task-1',
+            llm_provider: 'anthropic',
+            llm_model: 'claude-opus-4-1',
+        }, client);
+        expect(updateAgentTask.mock.calls[0].slice(0, 3)).toEqual([
+            'agent-1',
+            'task-1',
+            { llmProvider: 'anthropic', llmModel: 'claude-opus-4-1' },
+        ]);
+    });
+    it('unpins a task when the model fields are sent as null', async () => {
+        const { client, updateAgentTask } = fakeClient();
+        await handleAgentBuilderTool('update_agent_task', {
+            agent_id: 'agent-1',
+            task_id: 'task-1',
+            llm_provider: null,
+            llm_model: null,
+            llm_family: null,
+        }, client);
+        expect(updateAgentTask.mock.calls[0][2]).toEqual({
+            llmProvider: null,
+            llmModel: null,
+            llmFamily: null,
+        });
+    });
+    it('leaves an existing pin alone when updating something else', async () => {
+        const { client, updateAgentTask } = fakeClient();
+        await handleAgentBuilderTool('update_agent_task', { agent_id: 'agent-1', task_id: 'task-1', prompt: 'Analyse the transcript twice.' }, client);
+        expect(updateAgentTask.mock.calls[0][2]).toEqual({ prompt: 'Analyse the transcript twice.' });
+    });
+});
+// #2048 — run_agent_task can add instructions for one run without editing the
+// task. The prompt is the 7th argument of client.runAgentTask, so these assert
+// on the position as well as the value.
+describe('extra instructions on a task run over MCP (#2048)', () => {
+    const runClient = () => {
+        const runAgentTask = vi.fn().mockResolvedValue({ entryId: 'e1', position: 0, executing: true });
+        return { client: { runAgentTask }, runAgentTask };
+    };
+    it('advertises the prompt argument as an addition, not a replacement', () => {
+        const properties = toolNamed('run_agent_task').inputSchema.properties;
+        expect(properties.prompt.description).toContain("appended to the task's saved prompt");
+    });
+    it('sends the extra instructions with the run', async () => {
+        const { client, runAgentTask } = runClient();
+        await handleAgentBuilderTool('run_agent_task', { agent_id: 'agent-1', task_id: 'task-1', prompt: 'Only cover last week.' }, client);
+        expect(runAgentTask.mock.calls[0][1]).toBe('task-1');
+        expect(runAgentTask.mock.calls[0][6]).toBe('Only cover last week.');
+    });
+    it('sends none when the caller only wants the task as saved', async () => {
+        const { client, runAgentTask } = runClient();
+        await handleAgentBuilderTool('run_agent_task', { agent_id: 'agent-1', task_id: 'task-1' }, client);
+        expect(runAgentTask.mock.calls[0][6]).toBeUndefined();
+    });
+});

package/dist/client.js

@@ -748,8 +748,9 @@ export class WolfpackClient {
     async updateAgentTask(agentId, taskId, body, orgSlug) {
         return this.api.patch(this.withOrgSlug(`/agents/${agentId}/tasks/${taskId}`, orgSlug), body);
     }
-    async runAgentTask(agentId, taskId, teamSlug, orgSlug, gitRef, prNumber) {
-        return this.api.post(this.withOrgSlug(`/agents/${agentId}/tasks/${taskId}/run`, orgSlug), { teamSlug, gitRef, prNumber });
+    /** `prompt` is appended to the task's saved prompt for this run only (#2048). */
+    async runAgentTask(agentId, taskId, teamSlug, orgSlug, gitRef, prNumber, prompt) {
+        return this.api.post(this.withOrgSlug(`/agents/${agentId}/tasks/${taskId}/run`, orgSlug), { teamSlug, gitRef, prNumber, prompt });
     }
     // ─── Agent Builder: Queue ──────────────────────────────────────────────────
     async listAgentQueue(agentId, status, orgSlug) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wolfpack-mcp",
-  "version": "1.0.88",
+  "version": "1.0.90",
   "description": "MCP server for Wolfpack AI-enhanced software delivery tools",
   "main": "dist/index.js",
   "type": "module",