Browse Source

use or not use proxy config

fengchang 1 year ago
parent
commit
a96d39ecfb

+ 11 - 2
hichina-admin-backend/src/main/java/com/hichina/admin/hichinaadminbackend/service/HichinaOpenAiService.java

@@ -9,6 +9,8 @@ import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 
+import java.net.InetSocketAddress;
+import java.net.Proxy;
 import java.util.List;
 
 import static java.time.Duration.ofSeconds;
@@ -28,10 +30,17 @@ public class HichinaOpenAiService {
     public void setNameStatic(String tokenAndTimeout){
         String[] args = tokenAndTimeout.split(",");
         // key trick: the timeout here is very important
-        LOG.info(String.format("===init openai args: %s, %s",args[0],args[1]));
+        LOG.info(String.format("===init openai args: api key: %s, timeout in second: %s, local proxy port:%s",args[0],args[1],args[2]));
         service = new OpenAiService(args[0], ofSeconds(Integer.parseInt(args[1])));
 
-        chatGPT = new ChatGPT(args[0]);
+
+        if(!"-1".equals(args[2])){
+            // use proxy
+            Proxy proxy = new Proxy(Proxy.Type.HTTP, new InetSocketAddress("127.0.0.1", Integer.parseInt(args[2])));
+            chatGPT = new ChatGPT(args[0],proxy);
+        }else{
+            chatGPT = new ChatGPT(args[0]);
+        }
     }
 
     private String generateDescriptionRequest(String destinationName){

+ 1 - 1
hichina-admin-backend/src/main/resources/application-dev.properties

@@ -44,4 +44,4 @@ logging.level.org.springframework.web=DEBUG
 logging.level.org.hibernate=ERROR
 
 chatgpt.model=text-davinci-003
-openai.token=**,20
+openai.token=**,20,-1

+ 1 - 1
hichina-admin-backend/src/main/resources/application-qa.properties

@@ -43,4 +43,4 @@ logging.level.org.springframework.web=DEBUG
 logging.level.org.hibernate=ERROR
 
 chatgpt.model=text-davinci-003
-openai.token=**,20
+openai.token=**,20,-1