feat: :loud_sound: add context to dev data loggign

1 files changed, 19 insertions, 2 deletions

diff --git a/continuedev/src/continuedev/core/autopilot.py b/continuedev/src/continuedev/core/autopilot.py
index 9ebf288b..9f2338ff 100644
--- a/continuedev/src/continuedev/core/autopilot.py
+++ b/continuedev/src/continuedev/core/autopilot.py
@@ -2,6 +2,7 @@ import json
 import os
 import time
 import traceback
+import uuid
 from functools import cached_property
 from typing import Callable, Coroutine, Dict, List, Optional
 
@@ -380,11 +381,27 @@ class Autopilot(ContinueBaseModel):
         #     last_depth = self.history.timeline[i].depth
         #     i -= 1
 
+        # Log the context and step to dev data
+        context_used = await self.context_manager.get_selected_items()
         posthog_logger.capture_event(
             "step run", {"step_name": step.name, "params": step.dict()}
         )
+        step_id = uuid.uuid4().hex
         dev_data_logger.capture(
-            "step_run", {"step_name": step.name, "params": step.dict()}
+            "step_run",
+            {"step_name": step.name, "params": step.dict(), "step_id": step_id},
+        )
+        dev_data_logger.capture(
+            "context_used",
+            {
+                "context": list(
+                    map(
+                        lambda item: item.dict(),
+                        context_used,
+                    )
+                ),
+                "step_id": step_id,
+            },
         )
 
         if not is_future_step:
@@ -402,7 +419,7 @@ class Autopilot(ContinueBaseModel):
                 step=step,
                 observation=None,
                 depth=self._step_depth,
-                context_used=await self.context_manager.get_selected_items(),
+                context_used=context_used,
             )
         )