From f1224eed4ab9676fbfb50392cda7f14669586dcf Mon Sep 17 00:00:00 2001
From: Arun Isaac
Date: Sun, 12 Apr 2026 03:51:54 +0100
Subject: Strip message fields based on role.

---
 kaakaa/tea.scm | 23 +++++++++++++++--------
 1 file changed, 15 insertions(+), 8 deletions(-)

diff --git a/kaakaa/tea.scm b/kaakaa/tea.scm
index 05881f3..3986340 100644
--- a/kaakaa/tea.scm
+++ b/kaakaa/tea.scm
@@ -141,14 +141,21 @@ in @var{state}."
       ;; progress; dispatch to LLM.
       (list (llm-request session-id
                          (map (lambda (message)
-                                ;; Strip out all fields (such as reasoning
-                                ;; fields) other than role, content and
-                                ;; tool_calls.
-                                (filter (match-lambda
-                                          ((key . _)
-                                           (member key (list "role" "content"
-                                                             "tool_calls"))))
-                                        message))
+                                ;; Strip unnecessary fields (such as reasoning
+                                ;; fields) based on role.
+                                (let* ((role (focus (key-ref "role") message))
+                                       (allowed-fields
+                                        (cond
+                                         ((string=? role "user")
+                                          '("role" "content"))
+                                         ((string=? role "assistant")
+                                          '("role" "content" "tool_calls"))
+                                         ((string=? role "tool")
+                                          '("role" "content" "tool_call_id")))))
+                                  (filter (match-lambda
+                                            ((key . _)
+                                             (member key allowed-fields)))
+                                          message)))
                               ;; Reverse because we have been prepending new
                               ;; messages onto the list.
                               (reverse (focus (state-messages session-id)
-- 
cgit 1.4.1