Allow additional headers in the OpenAiClient

Author: singularity-sg

src/main/java/dev/ai4j/openai4j/DefaultOpenAiClient.java

@@ -2,6 +2,8 @@
 
 import static dev.ai4j.openai4j.Json.GSON;
 
+import java.util.HashMap;
+import java.util.Map;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -44,6 +46,8 @@ private DefaultOpenAiClient(Builder serviceBuilder) {
  this.apiVersion = serviceBuilder.apiVersion;
  this.userAgent = serviceBuilder.userAgent;
 
+ Map<String, String> headers = new HashMap<>();
+
  OkHttpClient.Builder okHttpClientBuilder = new OkHttpClient.Builder()
  .callTimeout(serviceBuilder.callTimeout)
  .connectTimeout(serviceBuilder.connectTimeout)
@@ -63,15 +67,15 @@ private DefaultOpenAiClient(Builder serviceBuilder) {
  }
 
  if (serviceBuilder.organizationId != null) {
- okHttpClientBuilder.addInterceptor(new GenericHeaderInjector(Collections.singletonMap("OpenAI-Organization", serviceBuilder.organizationId)));
+ headers.put("OpenAI-Organization", serviceBuilder.organizationId);
  }
 
  if (serviceBuilder.proxy != null) {
  okHttpClientBuilder.proxy(serviceBuilder.proxy);
  }
 
  if (serviceBuilder.userAgent != null) {
- okHttpClientBuilder.addInterceptor(new GenericHeaderInjector(Collections.singletonMap("User-Agent", serviceBuilder.userAgent)));
+ headers.put("User-Agent", serviceBuilder.userAgent);
  }
 
  if (serviceBuilder.logRequests) {
@@ -83,6 +87,11 @@ private DefaultOpenAiClient(Builder serviceBuilder) {
  }
  this.logStreamingResponses = serviceBuilder.logStreamingResponses;
 
+ if (serviceBuilder.additionalHeaders != null) {
+ headers.putAll(serviceBuilder.additionalHeaders);
+ okHttpClientBuilder.addInterceptor(new GenericHeaderInjector(headers));
+ }
+
  this.okHttpClient = okHttpClientBuilder.build();
 
  Retrofit.Builder retrofitBuilder = new Retrofit.Builder().baseUrl(serviceBuilder.baseUrl).client(okHttpClient);

src/main/java/dev/ai4j/openai4j/OpenAiClient.java

@@ -20,6 +20,7 @@
 import dev.ai4j.openai4j.moderation.ModerationResult;
 import dev.ai4j.openai4j.spi.OpenAiClientBuilderFactory;
 import dev.ai4j.openai4j.spi.ServiceHelper;
+import java.util.Map;
 
 import static dev.ai4j.openai4j.LogLevel.DEBUG;
 
@@ -73,6 +74,7 @@ public abstract static class Builder<T extends OpenAiClient, B extends Builder<T
  public LogLevel logLevel = DEBUG;
  public boolean logStreamingResponses;
  public Path persistTo;
+ public Map<String, String> additionalHeaders;
 
  public abstract T build();
 
@@ -250,5 +252,16 @@ public B persistTo(Path persistTo) {
  this.persistTo = persistTo;
  return (B) this;
  }
+
+ /**
+ * Additional headers to be added to the HTTP request.
+ *
+ * @param additionalHeaders map of headers
+ * @return builder
+ */
+ public B additionalHeaders(Map<String, String> additionalHeaders) {
+ this.additionalHeaders = additionalHeaders;
+ return (B) this;
+ }
  }
 }