Fixing Add memories with functions (#280)

xebia-functional · Jul 27, 2023 · e6f900a · e6f900a
1 parent 6c58cb2
commit e6f900a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/Chat.kt b/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/Chat.kt
@@ -236,7 +236,12 @@ interface Chat : LLM {
         Memory(
           conversationId = conversationId,
           content =
-            Message(role = role, content = firstChoice.message?.content ?: "", name = role.name),
+            Message(
+              role = role,
+              content = firstChoice.message?.content
+                  ?: firstChoice.message?.functionCall?.arguments ?: "",
+              name = role.name
+            ), //
           timestamp = getTimeMillis()
         )
       context.addMemories(listOf(requestMemory, firstChoiceMemory))