fix: restore original behavior

epam · Dec 11, 2024 · 3528a53 · 3528a53
1 parent a65d17a
commit 3528a53
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/server/src/main/java/com/epam/aidial/core/server/log/GfLogStore.java b/server/src/main/java/com/epam/aidial/core/server/log/GfLogStore.java
@@ -131,9 +131,13 @@ private void append(ProxyContext context, LogEntry entry, String assembledStream
             append(entry, ProxyUtil.MAPPER.writeValueAsString(executionPath), false);
         }
 
-        if (!context.isSecuredApiKey() && assembledStreamingResponse != null) {
+        if (!context.isSecuredApiKey()) {
             append(entry, ",\"assembled_response\":\"", false);
-            append(entry, assembledStreamingResponse, true);
+            if (assembledStreamingResponse != null) {
+                append(entry, assembledStreamingResponse, true);
+            } else {
+                append(entry, context.getResponseBody());
+            }
             append(entry, "\"", false);
         }