From 3b13239328e538a3b3806bde80a00cdbb48a4cb9 Mon Sep 17 00:00:00 2001
From: Concedo <Concedo@noreply.com>
Date: Sat, 30 Dec 2023 23:49:36 +0800
Subject: [PATCH] compatibility with mistral AI working

---
 index.html | 24 ++++++++++++++++--------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/index.html b/index.html
index aff900d..52d25c1 100644
--- a/index.html
+++ b/index.html
@@ -6155,12 +6155,16 @@
 			}
 		}
 
+		let oaiheaders = {
+			'Authorization': 'Bearer '+desired_oai_key,
+		};
+		if (!desired_oai_ep.toLowerCase().includes("api.mistral.ai")) {
+			oaiheaders["x-api-key"] = desired_oai_key;
+		}
+
 		fetch((desired_oai_ep + oai_models_endpoint), {
 			method: 'GET',
-			headers: {
-				'Authorization': 'Bearer '+desired_oai_key,
-				'x-api-key': desired_oai_key,
-			},
+			headers: oaiheaders,
 			referrerPolicy: 'no-referrer',
 		})
 		.then((response) => response.json())
@@ -8903,13 +8907,17 @@
 				// 	//use cors proxy for mistral ai
 				// 	targetep = cors_proxy + "?" + targetep;
 				// }
-				if(targetep.toLowerCase().includes("openrouter.ai")
-				|| targetep.toLowerCase().includes("api.mistral.ai"))
+				if(!targetep.toLowerCase().includes("openrouter.ai") &&
+				!targetep.toLowerCase().includes("api.mistral.ai"))
 				{
-					oaiheaders["HTTP-Referer"] = "https://lite.koboldai.net";
-				}else{
 					oaiheaders["x-api-key"] = custom_oai_key;
 				}
+
+				if(targetep.toLowerCase().includes("openrouter.ai"))
+				{
+					oaiheaders["HTTP-Referer"] = "https://lite.koboldai.net";
+				}
+
 				fetch(targetep, {
 					method: 'POST',
 					headers: oaiheaders,