variants

Author: spoons-and-mirrors

lib/messages/inject.ts

@@ -149,11 +149,16 @@ export const insertPruneToolContext = (
         injectionType: isGitHubCopilot ? "assistant-with-tool-part" : "user-message",
     })
 
+    const variant = state.variant ?? (lastUserMessage.info as UserMessage).variant
     if (isGitHubCopilot) {
         messages.push(
-            createSyntheticAssistantMessageWithToolPart(lastUserMessage, prunableToolsContent),
+            createSyntheticAssistantMessageWithToolPart(
+                lastUserMessage,
+                prunableToolsContent,
+                variant,
+            ),
         )
     } else {
-        messages.push(createSyntheticUserMessage(lastUserMessage, prunableToolsContent))
+        messages.push(createSyntheticUserMessage(lastUserMessage, prunableToolsContent, variant))
     }
 }

lib/messages/utils.ts

@@ -41,6 +41,7 @@ export const createSyntheticUserMessage = (
 export const createSyntheticAssistantMessageWithToolPart = (
     baseMessage: WithParts,
     content: string,
+    variant?: string,
 ): WithParts => {
     const userInfo = baseMessage.info as UserMessage
     const now = Date.now()
@@ -61,6 +62,7 @@ export const createSyntheticAssistantMessageWithToolPart = (
             time: { created: now, completed: now },
             cost: 0,
             tokens: { input: 0, output: 0, reasoning: 0, cache: { read: 0, write: 0 } },
+            ...(variant !== undefined && { variant }),
         },
         parts: [
             {