From c5ab158f3d42f91b95329e169acfa6aa94d27dc0 Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Tue, 1 Aug 2023 07:29:15 +0900
Subject: [PATCH] [feat] claude request

---
 src/ts/process/request.ts | 34 ++++++++++++++++++++++++++++++----
 1 file changed, 30 insertions(+), 4 deletions(-)

diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 933161ea..7ec6c509 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -89,7 +89,14 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
     let bias = arg.bias
     let currentChar = arg.currentChar
     const aiModel = (model === 'model' || (!db.advancedBotSettings)) ? db.aiModel : db.subModel
-    switch(aiModel){
+
+    let raiModel = aiModel
+    if(aiModel === 'reverse_proxy'){
+        if(db.proxyRequestModel.startsWith('claude')){
+            raiModel = 'claude'
+        }
+    }
+    switch(raiModel){
         case 'gpt35':
         case 'gpt35_0613':
         case 'gpt35_16k':
@@ -655,7 +662,26 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
             }
         }
         default:{     
-            if(aiModel.startsWith('claude')){
+            if(raiModel.startsWith('claude')){
+
+                let replacerURL = (aiModel === 'reverse_proxy') ? (db.forceReplaceUrl) : ('https://api.anthropic.com/v1/complete')
+                if(aiModel === 'reverse_proxy'){
+                    if(replacerURL.endsWith('v1')){
+                        replacerURL += '/complete'
+                    }
+                    else if(replacerURL.endsWith('v1/')){
+                        replacerURL += 'complete'
+                    }
+                    else if(!(replacerURL.endsWith('complete') || replacerURL.endsWith('complete/'))){
+                        if(replacerURL.endsWith('/')){
+                            replacerURL += 'v1/complete'
+                        }
+                        else{
+                            replacerURL += '/v1/complete'
+                        }
+                    }
+                }
+
                 for(let i=0;i<formated.length;i++){
                     if(arg.isGroupChat && formated[i].name){
                         formated[i].content = formated[i].name + ": " + formated[i].content
@@ -681,11 +707,11 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
                     return prefix + v.content
                 }).join('') + '\n\nAssistant: '
 
-                const da = await globalFetch('https://api.anthropic.com/v1/complete', {
+                const da = await globalFetch(replacerURL, {
                     method: "POST",
                     body: {
                         prompt : "\n\nHuman: " + requestPrompt,
-                        model: aiModel,
+                        model: raiModel,
                         max_tokens_to_sample: maxTokens,
                         stop_sequences: ["\n\nHuman:", "\n\nSystem:", "\n\nAssistant:"],
                         temperature: temperature,