Merge pull request #785 from quarkiverse/ollama-pretty-print

Pretty print Ollama responses
quarkiverse · Jul 26, 2024 · f2efe18 · f2efe18
2 parents e8ae2d2 + b887b92
commit f2efe18
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...oviders/ollama/runtime/src/main/java/io/quarkiverse/langchain4j/ollama/OllamaRestApi.java b/...oviders/ollama/runtime/src/main/java/io/quarkiverse/langchain4j/ollama/OllamaRestApi.java
@@ -28,6 +28,7 @@
 import org.jboss.resteasy.reactive.client.api.ClientLogger;
 
 import com.fasterxml.jackson.core.JsonParseException;
+import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
 
 import io.quarkiverse.langchain4j.QuarkusJsonCodecFactory;
@@ -210,7 +211,12 @@ private String bodyToString(Buffer body) {
             if (body == null) {
                 return "";
             }
-            return body.toString();
+            String rawBody = body.toString();
+            try {
+                return QuarkusJsonCodecFactory.SnakeCaseObjectMapperHolder.MAPPER.readTree(rawBody).toPrettyString();
+            } catch (JsonProcessingException ignored) {
+                return rawBody;
+            }
         }
 
         private String inOneLine(MultiMap headers) {