diff --git a/src/tau2/utils/llm_utils.py b/src/tau2/utils/llm_utils.py
index 83e31ce2..8b535c6f 100644
--- a/src/tau2/utils/llm_utils.py
+++ b/src/tau2/utils/llm_utils.py
@@ -157,13 +157,16 @@ def to_litellm_messages(messages: list[Message]) -> list[dict]:
                     }
                     for tc in message.tool_calls
                 ]
-            litellm_messages.append(
-                {
-                    "role": "assistant",
-                    "content": message.content,
-                    "tool_calls": tool_calls,
-                }
-            )
+            if message.raw_data:
+                litellm_messages.append(message.raw_data["message"])
+            else:
+                litellm_messages.append(
+                    {
+                        "role": "assistant",
+                        "content": message.content,
+                        "tool_calls": tool_calls,
+                    }
+                )
         elif isinstance(message, ToolMessage):
             litellm_messages.append(
                 {