feat: 增加对第三方api的支持

Author: ashinnotfound

pom.xml

@@ -52,6 +52,11 @@
             <artifactId>service</artifactId>
             <version>0.18.2</version>
         </dependency>
+        <dependency>
+            <groupId>com.squareup.retrofit2</groupId>
+            <artifactId>converter-jackson</artifactId>
+            <version>2.9.0</version>
+        </dependency>
         <dependency>
             <groupId>com.knuddels</groupId>
             <artifactId>jtokkit</artifactId>

src/main/java/com/ashin/client/GptClient.java

@@ -2,10 +2,16 @@
 
 import com.ashin.config.GptConfig;
 import com.ashin.config.ProxyConfig;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.theokanning.openai.client.OpenAiApi;
 import com.theokanning.openai.service.OpenAiService;
 import lombok.Getter;
 import lombok.extern.slf4j.Slf4j;
+import okhttp3.OkHttpClient;
 import org.springframework.stereotype.Component;
+import retrofit2.Retrofit;
+import retrofit2.adapter.rxjava2.RxJava2CallAdapterFactory;
+import retrofit2.converter.jackson.JacksonConverterFactory;
 
 import javax.annotation.PostConstruct;
 import javax.annotation.Resource;
@@ -39,9 +45,21 @@ public void init() {
         for (String apiKey : gptConfig.getApiKey()) {
             apiKey = apiKey.trim();
             if (!apiKey.isEmpty()) {
-                openAiServiceList.add(new OpenAiService(apiKey, Duration.ofSeconds(gptConfig.getOfSeconds())));
+                openAiServiceList.add(customOpenAiService(apiKey, gptConfig.getBaseUrl()));
                 log.info("apiKey为 {} 的账号初始化成功", apiKey);
             }
         }
     }
+
+    private OpenAiService customOpenAiService(String token, String baseUrl) {
+        ObjectMapper mapper = OpenAiService.defaultObjectMapper();
+        OkHttpClient client = OpenAiService.defaultClient(token, Duration.ofSeconds(gptConfig.getOfSeconds()));
+        Retrofit retrofit = new Retrofit.Builder()
+                .baseUrl(baseUrl)
+                .client(client)
+                .addConverterFactory(JacksonConverterFactory.create(mapper))
+                .addCallAdapterFactory(RxJava2CallAdapterFactory.create())
+                .build();
+        return new OpenAiService(retrofit.create(OpenAiApi.class), client.dispatcher().executorService());
+    }
 }

src/main/java/com/ashin/config/GptConfig.java

@@ -16,6 +16,7 @@
 @Component
 @ConfigurationProperties("gpt")
 public class GptConfig {
+    private String baseUrl;
     private String model;
     private Integer maxToken;
     private Double temperature;

src/main/resources/application.yml

@@ -8,6 +8,9 @@ proxy:
   port:
 
 gpt:
+  # 如果你使用了第三方GPT 请修改baseUrl
+  # (OPENAI官方地址 https://api.openai.com/)
+  baseUrl: https://api.openai.com/
   # 使用的 chat 模型 如gpt-3.5-turbo/gpt-4 (https://platform.openai.com/docs/models/models)
   model: gpt-3.5-turbo
   # 最大token限制 越多的token意味着越多的花费(gpt-3.5-turbo上限为4096token, gpt-4则为8192)