fix(autofix): Fix coding step rethinking (#1896)

We were using the wrong memory key when rethinking in the coding step, causing it to be missing the memory and error out.
getsentry · Feb 10, 2025 · 556ee9a · 556ee9a
1 parent 2747f20
commit 556ee9a
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/src/seer/automation/autofix/components/coding/component.py b/src/seer/automation/autofix/components/coding/component.py
@@ -177,12 +177,22 @@ def invoke(self, request: CodingRequest) -> CodingOutput | None:
 
             custom_solution = request.solution if isinstance(request.solution, str) else None
 
+            if not request.initial_memory:
+                agent.memory.insert(
+                    0,
+                    Message(
+                        role="user",
+                        content=CodingPrompts.format_fix_msg(
+                            has_tools=not is_obvious,
+                            custom_solution=custom_solution,
+                            mode=request.mode,
+                        ),
+                    ),
+                )
+
             response = agent.run(
                 RunConfig(
                     system_prompt=CodingPrompts.format_system_msg(has_tools=not is_obvious),
-                    prompt=CodingPrompts.format_fix_msg(
-                        has_tools=not is_obvious, custom_solution=custom_solution, mode=request.mode
-                    ),
                     model=AnthropicProvider.model("claude-3-5-sonnet-v2@20241022"),
                     memory_storage_key="code",
                     run_name="Code",

diff --git a/src/seer/automation/autofix/tasks.py b/src/seer/automation/autofix/tasks.py
@@ -337,7 +337,7 @@ def receive_user_message(request: AutofixUpdateRequest):
         is_solution_step = step_to_restart.key == "solution_processing"
         is_root_cause_step = step_to_restart.key == "root_cause_analysis_processing"
         memory = (
-            context.get_memory("plan_and_code")
+            context.get_memory("code")
             if is_coding_step
             else (
                 context.get_memory("root_cause_analysis")
@@ -461,7 +461,7 @@ def restart_from_point_with_feedback(
     is_solution_step = step_to_restart.key == "solution_processing"
     is_root_cause_step = step_to_restart.key == "root_cause_analysis_processing"
     memory = (
-        context.get_memory("plan_and_code")
+        context.get_memory("code")
         if is_coding_step
         else (
             context.get_memory("root_cause_analysis")