From b385fdafc09492bdf269491fcdcfba07580d0255 Mon Sep 17 00:00:00 2001
From: Maohao <codercat@163.com>
Date: Sun, 23 Nov 2025 15:17:58 +0800
Subject: [PATCH] feat: add Gemini 3 Pro thought signature support

---
 src/tau2/utils/llm_utils.py | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

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(
                 {