kova/youchat_proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix model selection

Browse Source
This commit is contained in:
Archeb
2024-06-22 15:08:35 +08:00
parent 5d73bf8b4e
commit cb8414d646
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
index.mjs
+4 -2
View File
@@ -64,10 +64,12 @@ app.post("/v1/messages", apiKeyAuth, (req, res) => {
console.log("Using session " + randomSession); console.log("Using session " + randomSession);
// decide which model to use // 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]; var proxyModel = modelMappping[jsonBody.model];
}else{ }else{
var proxyModel = process.env.AI_MODEL || "claude_3_opus"; var proxyModel = "claude_3_opus";
} }
console.log("Using model " + proxyModel); console.log("Using model " + proxyModel);