llmconversation: correct to store response correctly

OpenAF · May 15, 2024 · 406ca1d · 406ca1d
1 parent 01395c1
commit 406ca1d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/include/inputFns.js b/src/include/inputFns.js
@@ -455,7 +455,11 @@ var _inputFns = new Map([
                 __res = res.promptJSON(_res) 
             }   
         }
-        if (isDef(params.llmconversation)) io.writeFileJSON( params.llmconversation, res.getGPT().getConversation(), "" )
+        if (isDef(params.llmconversation)) {
+            var _conv = res.getGPT().getConversation()
+            _conv.push({ role: "assistant", content: stringify(__res, __, "") })
+            io.writeFileJSON( params.llmconversation, _conv, "" )
+        }
 
         _$o(jsonParse(__res, __, __, isString(__res)), options)
     }],