Skip to content

Commit 2f0a1f2

Browse files
committed
fix: forward model info to SDK in serve mode launch path
1 parent 8d9e444 commit 2f0a1f2

2 files changed

Lines changed: 5 additions & 2 deletions

File tree

src/tools/launch.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ import { detectOMO } from '../lib/omo';
1818
import { copyPlansToWorktree } from '../lib/plan-copier';
1919
import { resolvePostCreateHook } from '../lib/worktree-setup';
2020
import { writePromptFile, cleanupPromptFile, writeLauncherScript, cleanupLauncherScript, writeServeLauncherScript } from '../lib/prompt-file';
21-
import { getCurrentModel } from '../lib/model-tracker';
21+
import { getCurrentModel, getCurrentModelInfo } from '../lib/model-tracker';
2222
import { allocatePort, releasePort } from '../lib/port-allocator';
2323
import { waitForServer, createSessionAndPrompt } from '../lib/sdk-client';
2424

@@ -301,7 +301,8 @@ export const mc_launch: ToolDefinition = tool({
301301
planFile: args.planFile,
302302
autoCommit: config.autoCommit,
303303
});
304-
await createSessionAndPrompt(client, fullPrompt);
304+
const modelInfo = getCurrentModelInfo(context?.sessionID);
305+
await createSessionAndPrompt(client, fullPrompt, undefined, modelInfo);
305306
} catch (error) {
306307
await releasePort(allocatedPort).catch(() => {});
307308
try {

tests/tools/launch.test.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -644,6 +644,8 @@ describe('mc_launch', () => {
644644
expect(mockCreateSessionAndPrompt).toHaveBeenCalledWith(
645645
expect.anything(),
646646
expect.stringContaining('Do task'),
647+
undefined,
648+
undefined,
647649
);
648650
});
649651

0 commit comments

Comments
 (0)