Răsfoiți Sursa

feat(core): Set variant in assistant messages too (#12531)

Co-authored-by: Aiden Cline <[email protected]>
Co-authored-by: Aiden Cline <[email protected]>
Shantur Rathore 1 săptămână în urmă
părinte
comite
a486b74b14

+ 1 - 0
packages/opencode/src/session/compaction.ts

@@ -108,6 +108,7 @@ export namespace SessionCompaction {
       sessionID: input.sessionID,
       mode: "compaction",
       agent: "compaction",
+      variant: userMessage.variant,
       summary: true,
       path: {
         cwd: Instance.directory,

+ 1 - 0
packages/opencode/src/session/message-v2.ts

@@ -387,6 +387,7 @@ export namespace MessageV2 {
         write: z.number(),
       }),
     }),
+    variant: z.string().optional(),
     finish: z.string().optional(),
   }).meta({
     ref: "AssistantMessage",

+ 2 - 0
packages/opencode/src/session/prompt.ts

@@ -336,6 +336,7 @@ export namespace SessionPrompt {
           sessionID,
           mode: task.agent,
           agent: task.agent,
+          variant: lastUser.variant,
           path: {
             cwd: Instance.directory,
             root: Instance.worktree,
@@ -539,6 +540,7 @@ export namespace SessionPrompt {
           role: "assistant",
           mode: agent.name,
           agent: agent.name,
+          variant: lastUser.variant,
           path: {
             cwd: Instance.directory,
             root: Instance.worktree,

+ 1 - 0
packages/sdk/js/src/v2/gen/types.gen.ts

@@ -197,6 +197,7 @@ export type AssistantMessage = {
       write: number
     }
   }
+  variant?: string
   finish?: string
 }