From d3756d6477655884de8144784db0d9c4ad20c3fd Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Sun, 18 Jun 2023 21:29:36 -0700
Subject: moving gradient loader

---
 continuedev/src/continuedev/core/autopilot.py | 2 ++
 continuedev/src/continuedev/core/main.py      | 1 +
 2 files changed, 3 insertions(+)

(limited to 'continuedev')

diff --git a/continuedev/src/continuedev/core/autopilot.py b/continuedev/src/continuedev/core/autopilot.py
index ee249c0b..8b0eaca3 100644
--- a/continuedev/src/continuedev/core/autopilot.py
+++ b/continuedev/src/continuedev/core/autopilot.py
@@ -181,6 +181,7 @@ class Autopilot(ContinueBaseModel):
 
             # i is now the index of the step that we want to show/rerun
             self.history.timeline[i].observation = observation
+            self.history.timeline[i].active = False
 
             await self.update_subscribers()
 
@@ -205,6 +206,7 @@ class Autopilot(ContinueBaseModel):
         # Add observation to history, unless already attached error observation
         if not caught_error:
             self.history.timeline[index_of_history_node].observation = observation
+            self.history.timeline[index_of_history_node].active = False
             await self.update_subscribers()
 
         # Update its description
diff --git a/continuedev/src/continuedev/core/main.py b/continuedev/src/continuedev/core/main.py
index 0c7ec67f..efb91488 100644
--- a/continuedev/src/continuedev/core/main.py
+++ b/continuedev/src/continuedev/core/main.py
@@ -21,6 +21,7 @@ class HistoryNode(ContinueBaseModel):
     observation: Union[Observation, None]
     depth: int
     deleted: bool = False
+    active: bool = True
 
     def to_chat_messages(self) -> List[ChatMessage]:
         if self.step.description is None:
-- 
cgit v1.2.3-70-g09d2