Fix output for run-less prompts

* the output-handler was expecting maps but with just prompts it's a coll
docker · Aug 19, 2024 · 89e9589 · 89e9589
1 parent fa0d673
commit 89e9589
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/prompts.clj b/src/prompts.clj
@@ -296,9 +296,16 @@
                [nil "--help" "print option summary"]])
 
 (def output-handler (fn [x]
-                      (jsonrpc/notify :message {:content (json/generate-string (if (= "error" (:done x))
-                                                                                 (update x :messages last) 
-                                                                                 (select-keys x [:done])))})))
+                      (jsonrpc/notify
+                       :message
+                       {:content
+                        (json/generate-string
+                          (if (map? x)
+                            (if (= "error" (:done x))
+                              (update x :messages last)
+                              (select-keys x [:done]))
+                            x))})))
+
 (defn output-prompts [coll]
   (->> coll
        (mapcat (fn [{:keys [role content]}]