Fix Q&A for gemini

getsentry · Feb 8, 2025 · 2c848af · 2c848af
1 parent 7a94f30
commit 2c848af
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/seer/automation/autofix/tasks.py b/src/seer/automation/autofix/tasks.py
@@ -301,8 +301,14 @@ def restart_step_with_user_response(
     cur_state = state.get()
     if memory:
         tool_call_id = memory[-1].tool_call_id
+        tool_call_function = memory[-1].tool_call_function
         if tool_call_id:
-            user_response = Message(role="tool", content=text, tool_call_id=tool_call_id)
+            user_response = Message(
+                role="tool",
+                content=text,
+                tool_call_id=tool_call_id,
+                tool_call_function=tool_call_function,
+            )
             if memory[-1].role == "tool":
                 memory[-1] = user_response
             else: