Skip to content

Commit 10d0366

Browse files
committed
Add a default interceptor
1 parent a608fa5 commit 10d0366

2 files changed

Lines changed: 7 additions & 21 deletions

File tree

src/main/java/org/devlive/sdk/openai/OpenAiClient.java

Lines changed: 6 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
import okhttp3.OkHttpClient;
99
import org.apache.commons.lang3.ObjectUtils;
1010
import org.apache.commons.lang3.StringUtils;
11-
import org.devlive.sdk.openai.exception.ParamException;
1211
import org.devlive.sdk.openai.interceptor.DefaultInterceptor;
1312
import retrofit2.Retrofit;
1413
import retrofit2.adapter.rxjava2.RxJava2CallAdapterFactory;
@@ -102,30 +101,21 @@ public OpenAiClientBuilder unit(TimeUnit unit)
102101

103102
public OpenAiClientBuilder client(OkHttpClient client)
104103
{
105-
DefaultInterceptor interceptor = new DefaultInterceptor();
106-
interceptor.setApiKey(this.apiKey);
107104
if (ObjectUtils.isEmpty(client)) {
108105
log.warn("No client, creating default client");
109106
client = new OkHttpClient.Builder()
110-
.addInterceptor(interceptor)
111107
.connectTimeout(this.timeout, this.unit)
112108
.writeTimeout(this.timeout, this.unit)
113109
.readTimeout(this.timeout, this.unit)
114110
.callTimeout(this.timeout, this.unit)
115111
.build();
116112
}
117-
118-
if (client.interceptors().size() <= 0) {
119-
throw new ParamException("No interceptors available");
120-
}
121-
122-
long count = client.interceptors()
123-
.stream()
124-
.filter(inter -> inter instanceof DefaultInterceptor)
125-
.count();
126-
if (count <= 0) {
127-
throw new ParamException("Must inject DefaultInterceptor");
128-
}
113+
// Add default interceptor
114+
DefaultInterceptor interceptor = new DefaultInterceptor();
115+
interceptor.setApiKey(this.apiKey);
116+
client = client.newBuilder()
117+
.addInterceptor(interceptor)
118+
.build();
129119
this.client = client;
130120
return this;
131121
}

src/test/java/org/devlive/sdk/openai/OpenAiClientTest.java

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import org.devlive.sdk.openai.entity.CompletionEntity;
77
import org.devlive.sdk.openai.entity.CompletionMessageEntity;
88
import org.devlive.sdk.openai.exception.AuthorizedException;
9-
import org.devlive.sdk.openai.interceptor.DefaultInterceptor;
109
import org.devlive.sdk.openai.model.CompletionModel;
1110
import org.junit.Assert;
1211
import org.junit.Before;
@@ -39,17 +38,14 @@ public void testBuilder()
3938
@Test
4039
public void testClient()
4140
{
42-
DefaultInterceptor interceptor = new DefaultInterceptor();
43-
interceptor.setApiKey(System.getProperty("proxy.token"));
4441
OkHttpClient okHttpClient = new OkHttpClient.Builder()
45-
.addInterceptor(interceptor)
4642
.connectTimeout(openApiTimeout, TimeUnit.SECONDS)
4743
.writeTimeout(openApiTimeout, TimeUnit.SECONDS)
4844
.readTimeout(openApiTimeout, TimeUnit.SECONDS)
4945
.build();
5046
client = OpenAiClient.builder()
5147
.apiHost(System.getProperty("proxy.host"))
52-
.apiKey(invalidApiKey)
48+
.apiKey(System.getProperty("proxy.token"))
5349
.client(okHttpClient)
5450
.build();
5551
Assert.assertTrue(client.getModels().getModels().size() > 0);

0 commit comments

Comments
 (0)