Changes async response to return body only when de-referenced

wkok · Jan 31, 2024 · efcf882 · efcf882
1 parent cfd715b
commit efcf882
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/wkok/openai_clojure/sse.clj b/src/wkok/openai_clojure/sse.clj
@@ -170,7 +170,13 @@
 
               (if async?
 
-                ((:leave martian.hato/perform-request-async) ctx')
+                (-> ((:leave martian.hato/perform-request-async) ctx')
+                    (update :response
+                            #(.thenApply
+                               %
+                               (reify java.util.function.Function
+                                 (apply [_ response']
+                                   (:body response'))))))
 
                 (assoc ctx :response (if (:stream params)
                                        (sse-request ctx')