From 5510e6da31e3798f0262333666bfb91a10fcd4ee Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Mon, 14 Apr 2025 22:36:56 +0300
Subject: [PATCH] Enable multi-swipe for xAI

---
 public/index.html        | 2 +-
 public/scripts/openai.js | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/public/index.html b/public/index.html
index 5623db848..736e7300d 100644
--- a/public/index.html
+++ b/public/index.html
@@ -646,7 +646,7 @@
                                         <input type="number" id="openai_max_tokens" name="openai_max_tokens" class="text_pole" min="1" max="65536">
                                     </div>
                                 </div>
-                                <div class="range-block" data-source="openai,custom">
+                                <div class="range-block" data-source="openai,custom,xai">
                                     <div class="range-block-title" data-i18n="Multiple swipes per generation">
                                         Multiple swipes per generation
                                     </div>
diff --git a/public/scripts/openai.js b/public/scripts/openai.js
index 9aac63be1..56d2e6ee0 100644
--- a/public/scripts/openai.js
+++ b/public/scripts/openai.js
@@ -1980,7 +1980,7 @@ async function sendOpenAIRequest(type, messages, signal) {
     const isContinue = type === 'continue';
     const stream = oai_settings.stream_openai && !isQuiet && !isScale && !(isOAI && ['o1-2024-12-17', 'o1'].includes(oai_settings.openai_model));
     const useLogprobs = !!power_user.request_token_probabilities;
-    const canMultiSwipe = oai_settings.n > 1 && !isContinue && !isImpersonate && !isQuiet && (isOAI || isCustom);
+    const canMultiSwipe = oai_settings.n > 1 && !isContinue && !isImpersonate && !isQuiet && (isOAI || isCustom || isXAI);
 
     // If we're using the window.ai extension, use that instead
     // Doesn't support logit bias yet