KeyStrategyFunction的apply方法增加request参数

src/main/java/com/unfbx/chatgpt/function/KeyRandomStrategy.java

@@ -1,6 +1,7 @@
 package com.unfbx.chatgpt.function;
 
 import cn.hutool.core.util.RandomUtil;
+import okhttp3.Request;
 
 import java.util.List;
 
@@ -13,7 +14,7 @@
 public class KeyRandomStrategy implements KeyStrategyFunction<List<String>, String> {
 
     @Override
-    public String apply(List<String> apiKeys) {
+    public String apply(List<String> apiKeys, Request request) {
         return RandomUtil.randomEle(apiKeys);
     }
 }

src/main/java/com/unfbx/chatgpt/function/KeyStrategyFunction.java

@@ -1,5 +1,7 @@
 package com.unfbx.chatgpt.function;
 
+import okhttp3.Request;
+
 import java.util.function.Function;
 
 /**
@@ -19,6 +21,6 @@ public interface KeyStrategyFunction<T, R> {
      * @param t the function argument
      * @return the function result
      */
-    R apply(T t);
+    R apply(T t, Request request);
 
 }

src/main/java/com/unfbx/chatgpt/interceptor/DefaultOpenAiAuthInterceptor.java

@@ -42,7 +42,7 @@ public DefaultOpenAiAuthInterceptor(Map warringConfig) {
     @Override
     public Response intercept(Chain chain) throws IOException {
         Request original = chain.request();
-        return chain.proceed(auth(super.getKey(), original));
+        return chain.proceed(auth(super.getKey(original), original));
     }
 
     /**

src/main/java/com/unfbx/chatgpt/interceptor/DynamicKeyOpenAiAuthInterceptor.java

@@ -52,8 +52,8 @@ public DynamicKeyOpenAiAuthInterceptor(Map warringConfig) {
 
     @Override
     public Response intercept(Chain chain) throws IOException {
-        String key = getKey();
         Request original = chain.request();
+        String key = getKey(original);
         Request request = this.auth(key, original);
         Response response = chain.proceed(request);
         if (!response.isSuccessful() && response.body() != null) {

src/main/java/com/unfbx/chatgpt/interceptor/OpenAiAuthInterceptor.java

@@ -58,12 +58,12 @@ public abstract class OpenAiAuthInterceptor implements Interceptor {
      *
      * @return key
      */
-    public final String getKey() {
+    public final String getKey(Request request) {
         if (CollectionUtil.isEmpty(apiKey)) {
             this.noHaveActiveKeyWarring();
             throw new BaseException(CommonError.NO_ACTIVE_API_KEYS);
         }
-        return keyStrategy.apply(apiKey);
+        return keyStrategy.apply(apiKey, request);
     }
 
     /**

src/test/java/com/unfbx/chatgpt/FirstKeyStrategy.java

@@ -1,6 +1,7 @@
 package com.unfbx.chatgpt;
 
 import com.unfbx.chatgpt.function.KeyStrategyFunction;
+import okhttp3.Request;
 
 import java.util.List;
 
@@ -19,7 +20,7 @@ public class FirstKeyStrategy implements KeyStrategyFunction<List<String>, Strin
      * @return
      */
     @Override
-    public String apply(List<String> keys) {
+    public String apply(List<String> keys, Request request) {
         return keys.get(0);
     }
 }