dirty fix langchain4j/langchain4j#670

ai-for-java · Mar 8, 2024 · e09682d · e09682d
1 parent 98cb30c
commit e09682d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/dev/ai4j/openai4j/StreamingRequestExecutor.java b/src/main/java/dev/ai4j/openai4j/StreamingRequestExecutor.java
@@ -200,6 +200,12 @@ public void onFailure(EventSource eventSource, Throwable t, okhttp3.Response res
                     return;
                 }
 
+                // TODO remove this when migrating from okhttp
+                if (t instanceof IllegalArgumentException && "byteCount < 0: -1".equals(t.getMessage())) {
+                    streamingCompletionCallback.run();
+                    return;
+                }
+
                 if (logStreamingResponses) {
                     log.debug("onFailure()", t);
                     responseLogger.log(response);