fork: support custom model in agent command

mpazik · mpazik · commit 1a28d1b5da1c · 2025-07-09T18:27:14.000+08:00
diff --git a/packages/opencode/src/cli/cmd/agent.ts b/packages/opencode/src/cli/cmd/agent.ts
@@ -56,6 +56,11 @@ export const AgentCommand = cmd({
         type: "string",
         describe: "additional context to pass to the agent (JSON format)",
       })
+      .option("model", {
+        type: "string",
+        alias: ["m"],
+        describe: "model to use in the format of provider/model",
+      })
       .check((argv) => {
         if (!argv.list && !argv.name) {
           throw new Error("Agent name is required unless using --list")
@@ -113,12 +118,14 @@ export const AgentCommand = cmd({
         }
       }
 
-      const session = await Session.createForAgent(agentId, initContext)
+      const session = await Session.createForAgent(agentId, initContext, args.model)
       const agentConfig = await AgentServices.loadConfig(
         agentId,
         session.agentContext,
       )
-      const model = agentConfig.model
+      const model = session.model
+        ? Provider.parseModel(session.model)
+        : agentConfig.model
         ? Provider.parseModel(agentConfig.model)
         : await Provider.defaultModel()
       const modelName = `${model.providerID}/${model.modelID}`
diff --git a/packages/opencode/src/session/index.ts b/packages/opencode/src/session/index.ts
@@ -67,6 +67,7 @@ export namespace Session {
       }),
       agent: z.string().optional(),
       agentContext: Agent.Context,
+      model: z.string().optional(),
       revert: z
         .object({
           messageID: z.string(),
@@ -167,6 +168,7 @@ export namespace Session {
   export async function createForAgent(
     agentId: string,
     initContext: Agent.Context,
+    model?: string,
   ): Promise<Info> {
     const session = await create(undefined, agentId)
     const fullContext = await AgentServices.loadContext(
@@ -178,6 +180,7 @@ export namespace Session {
     return (await update(session.id, (draft) => {
       draft.agent = agentId
       draft.agentContext = fullContext
+      if (model) draft.model = model
     }))!
   }
 
@@ -328,12 +331,13 @@ export namespace Session {
       ? await AgentServices.loadConfig(session.agent, session.agentContext)
       : undefined
 
-    const agentModel = agentConfig?.model
-      ? Provider.parseModel(agentConfig.model)
-      : undefined
+    const resolvedModel = (() => {
+      if (session.model) return Provider.parseModel(session.model)
+      if (agentConfig?.model) return Provider.parseModel(agentConfig.model)
+      return { providerID: input.providerID, modelID: input.modelID }
+    })()
 
-    const modelID = agentModel ? agentModel?.modelID : input.modelID
-    const providerID = agentModel ? agentModel?.providerID : input.providerID
+    const { providerID, modelID } = resolvedModel
     const model = await Provider.getModel(providerID, modelID)
 
     if (session.revert) {
