fix: add api key auth (ollama)

carlrobertoh · Sep 5, 2024 · 6d71485 · 6d71485
1 parent 2722590
commit 6d71485
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/main/java/ee/carlrobert/llm/client/ollama/OllamaClient.java b/src/main/java/ee/carlrobert/llm/client/ollama/OllamaClient.java
@@ -244,10 +244,15 @@ private void processStreamRequest(
   private HttpRequest buildPostHttpRequest(
       Object request,
       String path) throws JsonProcessingException {
-    return HttpRequest.newBuilder(URI.create(BASE_URL + path))
+    var requestBuilder = HttpRequest.newBuilder(URI.create(BASE_URL + path))
         .POST(HttpRequest.BodyPublishers.ofString(new ObjectMapper().writeValueAsString(request)))
-        .header("Content-Type", "application/json")
-        .timeout(Duration.ofSeconds(30))
+        .header("Content-Type", "application/x-ndjson");
+
+    if (apiKey != null) {
+      requestBuilder.header("Authorization", "Bearer " + apiKey);
+    }
+
+    return requestBuilder.timeout(Duration.ofSeconds(30))
         .build();
   }