From 694d8e3c86dda17c1df6820a33ff6856e674beed Mon Sep 17 00:00:00 2001
From: Daniel Smolsky <dannysmo@gmail.com>
Date: Mon, 5 Jan 2026 22:37:28 -0500
Subject: [PATCH] fix: use variant from baseMessage in synthetic user message

Instead of creating a separate findUserVariant function that iterates
through messages, access the variant directly from the baseMessage
that is already passed to createSyntheticUserMessage.
---
 lib/messages/utils.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lib/messages/utils.ts b/lib/messages/utils.ts
index 756c1de..88dd75a 100644
--- a/lib/messages/utils.ts
+++ b/lib/messages/utils.ts
@@ -8,6 +8,7 @@ const SYNTHETIC_PART_ID = "prt_01234567890123456789012345"
 
 export const createSyntheticUserMessage = (baseMessage: WithParts, content: string): WithParts => {
     const userInfo = baseMessage.info as UserMessage
+    const variant = (userInfo as any).variant
     return {
         info: {
             id: SYNTHETIC_MESSAGE_ID,
@@ -19,6 +20,7 @@ export const createSyntheticUserMessage = (baseMessage: WithParts, content: stri
                 providerID: userInfo.model.providerID,
                 modelID: userInfo.model.modelID,
             },
+            ...(variant !== undefined && { variant }),
         },
         parts: [
             {