Merge pull request #225 from cau1k/fix/variant-support

fix: propagate variant to notification messages

Author: Tarquinen

index.ts

@@ -68,6 +68,21 @@ const plugin: Plugin = (async (ctx) => {
             logger,
             config,
         ),
+        "chat.message": async (
+            input: {
+                sessionID: string
+                agent?: string
+                model?: { providerID: string; modelID: string }
+                messageID?: string
+                variant?: string
+            },
+            _output: any,
+        ) => {
+            // Cache variant from real user messages (not synthetic)
+            // This avoids scanning all messages to find variant
+            state.variant = input.variant
+            logger.debug("Cached variant from chat.message hook", { variant: input.variant })
+        },
         tool: {
             ...(config.tools.discard.enabled && {
                 discard: createDiscardTool({

lib/messages/inject.ts

@@ -1,6 +1,7 @@
 import type { SessionState, WithParts } from "../state"
 import type { Logger } from "../logger"
 import type { PluginConfig } from "../config"
+import type { UserMessage } from "@opencode-ai/sdk/v2"
 import { loadPrompt } from "../prompts"
 import { extractParameterKey, buildToolIdList, createSyntheticUserMessage } from "./utils"
 import { getLastUserMessage } from "../shared-utils"
@@ -125,5 +126,6 @@ export const insertPruneToolContext = (
     if (!lastUserMessage) {
         return
     }
-    messages.push(createSyntheticUserMessage(lastUserMessage, prunableToolsContent))
+    const variant = state.variant ?? (lastUserMessage.info as UserMessage).variant
+    messages.push(createSyntheticUserMessage(lastUserMessage, prunableToolsContent, variant))
 }

lib/messages/utils.ts

@@ -1,14 +1,17 @@
 import { Logger } from "../logger"
 import { isMessageCompacted } from "../shared-utils"
 import type { SessionState, WithParts } from "../state"
-import type { UserMessage } from "@opencode-ai/sdk"
+import type { UserMessage } from "@opencode-ai/sdk/v2"
 
 const SYNTHETIC_MESSAGE_ID = "msg_01234567890123456789012345"
 const SYNTHETIC_PART_ID = "prt_01234567890123456789012345"
 
-export const createSyntheticUserMessage = (baseMessage: WithParts, content: string): WithParts => {
+export const createSyntheticUserMessage = (
+    baseMessage: WithParts,
+    content: string,
+    variant?: string,
+): WithParts => {
     const userInfo = baseMessage.info as UserMessage
-    const variant = (userInfo as any).variant
     return {
         info: {
             id: SYNTHETIC_MESSAGE_ID,

lib/state/state.ts

@@ -56,6 +56,7 @@ export function createSessionState(): SessionState {
         lastToolPrune: false,
         lastCompaction: 0,
         currentTurn: 0,
+        variant: undefined,
     }
 }
 
@@ -75,6 +76,7 @@ export function resetSessionState(state: SessionState): void {
     state.lastToolPrune = false
     state.lastCompaction = 0
     state.currentTurn = 0
+    state.variant = undefined
 }
 
 export async function ensureSessionInitialized(

lib/state/types.ts

@@ -1,4 +1,4 @@
-import { Message, Part } from "@opencode-ai/sdk"
+import { Message, Part } from "@opencode-ai/sdk/v2"
 
 export interface WithParts {
     info: Message
@@ -35,4 +35,5 @@ export interface SessionState {
     lastToolPrune: boolean
     lastCompaction: number
     currentTurn: number
+    variant: string | undefined
 }

lib/strategies/tools.ts

@@ -58,7 +58,7 @@ async function executePruneOperation(
 
     await ensureSessionInitialized(ctx.client, state, sessionId, logger, messages)
 
-    const currentParams = getCurrentParams(messages, logger)
+    const currentParams = getCurrentParams(state, messages, logger)
     const toolIdList: string[] = buildToolIdList(state, messages, logger)
 
     // Validate that all numeric IDs are within bounds

lib/strategies/utils.ts

@@ -1,27 +1,36 @@
 import { SessionState, WithParts } from "../state"
-import { UserMessage } from "@opencode-ai/sdk"
+import { UserMessage } from "@opencode-ai/sdk/v2"
 import { Logger } from "../logger"
 import { encode } from "gpt-tokenizer"
 import { getLastUserMessage, isMessageCompacted } from "../shared-utils"
 
 export function getCurrentParams(
+    state: SessionState,
     messages: WithParts[],
     logger: Logger,
 ): {
     providerId: string | undefined
     modelId: string | undefined
     agent: string | undefined
+    variant: string | undefined
 } {
     const userMsg = getLastUserMessage(messages)
     if (!userMsg) {
         logger.debug("No user message found when determining current params")
-        return { providerId: undefined, modelId: undefined, agent: undefined }
+        return {
+            providerId: undefined,
+            modelId: undefined,
+            agent: undefined,
+            variant: state.variant,
+        }
     }
-    const agent: string = (userMsg.info as UserMessage).agent
-    const providerId: string | undefined = (userMsg.info as UserMessage).model.providerID
-    const modelId: string | undefined = (userMsg.info as UserMessage).model.modelID
+    const userInfo = userMsg.info as UserMessage
+    const agent: string = userInfo.agent
+    const providerId: string | undefined = userInfo.model.providerID
+    const modelId: string | undefined = userInfo.model.modelID
+    const variant: string | undefined = state.variant ?? userInfo.variant
 
-    return { providerId, modelId, agent }
+    return { providerId, modelId, agent, variant }
 }
 
 /**

lib/ui/notification.ts

@@ -99,6 +99,7 @@ export async function sendIgnoredMessage(
     logger: Logger,
 ): Promise<void> {
     const agent = params.agent || undefined
+    const variant = params.variant || undefined
     const model =
         params.providerId && params.modelId
             ? {
@@ -116,6 +117,7 @@ export async function sendIgnoredMessage(
                 noReply: true,
                 agent: agent,
                 model: model,
+                variant: variant,
                 parts: [
                     {
                         type: "text",

package-lock.json

@@ -40,7 +40,7 @@
         "@opencode-ai/plugin": ">=0.13.7"
     },
     "dependencies": {
-        "@opencode-ai/sdk": "latest",
+        "@opencode-ai/sdk": "^1.1.3",
         "gpt-tokenizer": "^3.4.0",
         "jsonc-parser": "^3.3.1",
         "zod": "^4.1.13"