wolfpack-mcp 1.0.88 → 1.0.89

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' },
@@ -747,7 +769,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 +1037,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 +1051,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 +1070,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 +1090,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)

package/dist/agentBuilderTools.test.js

@@ -0,0 +1,76 @@
+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.' });
+    });
+});

package/package.json

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