fix: ollama host overriding

carlrobertoh · Sep 11, 2024 · 6b7e264 · 6b7e264
1 parent 03648e9
commit 6b7e264
Showing 1 changed file with 2 additions and 1 deletion.
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,7 +244,8 @@ private void processStreamRequest(
   private HttpRequest buildPostHttpRequest(
       Object request,
       String path) throws JsonProcessingException {
-    var requestBuilder = HttpRequest.newBuilder(URI.create(BASE_URL + path))
+    var baseHost = port == null ? BASE_URL : format("http://localhost:%d", port);
+    var requestBuilder = HttpRequest.newBuilder(URI.create((host == null ? baseHost : host) + path))
         .POST(HttpRequest.BodyPublishers.ofString(new ObjectMapper().writeValueAsString(request)))
         .header("Content-Type", "application/x-ndjson");