Просмотр исходного кода

feat(tasks): Add model info as part of metadata (#10307)

Shantur Rathore 3 недель назад
Родитель
Сommit
4173adf5e2
1 измененных файлов с 8 добавлено и 5 удалено
  1. 8 5
      packages/opencode/src/tool/task.ts

+ 8 - 5
packages/opencode/src/tool/task.ts

@@ -99,10 +99,16 @@ export const TaskTool = Tool.define("task", async (ctx) => {
       const msg = await MessageV2.get({ sessionID: ctx.sessionID, messageID: ctx.messageID })
       if (msg.info.role !== "assistant") throw new Error("Not an assistant message")
 
+      const model = agent.model ?? {
+        modelID: msg.info.modelID,
+        providerID: msg.info.providerID,
+      }
+
       ctx.metadata({
         title: params.description,
         metadata: {
           sessionId: session.id,
+          model,
         },
       })
 
@@ -126,15 +132,11 @@ export const TaskTool = Tool.define("task", async (ctx) => {
           metadata: {
             summary: Object.values(parts).sort((a, b) => a.id.localeCompare(b.id)),
             sessionId: session.id,
+            model,
           },
         })
       })
 
-      const model = agent.model ?? {
-        modelID: msg.info.modelID,
-        providerID: msg.info.providerID,
-      }
-
       function cancel() {
         SessionPrompt.cancel(session.id)
       }
@@ -180,6 +182,7 @@ export const TaskTool = Tool.define("task", async (ctx) => {
         metadata: {
           summary,
           sessionId: session.id,
+          model,
         },
         output,
       }