ais: fix context dump

getkyo · Dec 13, 2023 · 6ed1f01 · 6ed1f01
1 parent f898bd2
commit 6ed1f01
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/kyo-llm/shared/src/main/scala/kyo/llm/contexts.scala b/kyo-llm/shared/src/main/scala/kyo/llm/contexts.scala
@@ -96,8 +96,8 @@ object contexts {
         if (s.contains('\n') || s.contains('"') || s.contains('\\')) s"p\"\"\"$s\"\"\""
         else s""""$s""""
       }
-      val seedStr     = seed.map(s => s".seed(${stringify(s)})").getOrElse("")
-      val reminderStr = reminder.map(s => s".reminder(${stringify(s)})").getOrElse("")
+      val seedStr     = seed.map(s => s"\n  .seed(${stringify(s)})").getOrElse("")
+      val reminderStr = reminder.map(s => s"\n  .reminder(${stringify(s)})").getOrElse("")
       val messagesStr = messages.reverse.map {
         case Message.SystemMessage(content, _) =>
           s"\n  .systemMessage(${stringify(content)})"
@@ -112,9 +112,9 @@ object contexts {
           s"\n  .assistantMessage(${stringify(content)}${if (calls.isEmpty) ""
             else s", List($callsStr)"})"
         case Message.AgentMessage(callId, content, _) =>
-          s"\n  .toolMessage(CallId(\"${callId.id}\"), ${stringify(content)})"
+          s"\n  .agentMessage(CallId(\"${callId.id}\"), ${stringify(content)})"
       }.mkString
-      s"Contexts.init$seedStr$reminderStr$messagesStr"
+      s"Context.empty$seedStr$reminderStr$messagesStr"
     }
   }