From cb8414d646beb639386248a3a13c06e5e3b66329 Mon Sep 17 00:00:00 2001
From: Archeb <archebasic@hotmail.com>
Date: Sat, 22 Jun 2024 15:08:35 +0800
Subject: [PATCH] fix model selection

---
 index.mjs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/index.mjs b/index.mjs
index c9d86f1..ec75fb6 100644
--- a/index.mjs
+++ b/index.mjs
@@ -64,10 +64,12 @@ app.post("/v1/messages", apiKeyAuth, (req, res) => {
 			console.log("Using session " + randomSession);
 
 			// decide which model to use
-			if (jsonBody.model && modelMappping[jsonBody.model]) {
+			if(process.env.AI_MODEL){
+				var proxyModel = process.env.AI_MODEL;
+			}else if (jsonBody.model && modelMappping[jsonBody.model]) {
 				var proxyModel = modelMappping[jsonBody.model];
 			}else{
-				var proxyModel = process.env.AI_MODEL || "claude_3_opus";
+				var proxyModel = "claude_3_opus";
 			}
 			console.log("Using model " + proxyModel);