feat: :sparkles: FileTreeContextProvider

also pass workspace_dir: str to provide_context_items

2 files changed, 4 insertions, 4 deletions

diff --git a/continuedev/src/continuedev/core/autopilot.py b/continuedev/src/continuedev/core/autopilot.py
index 3f25e64e..57e39d5c 100644
--- a/continuedev/src/continuedev/core/autopilot.py
+++ b/continuedev/src/continuedev/core/autopilot.py
@@ -75,7 +75,7 @@ class Autopilot(ContinueBaseModel):
                 HighlightedCodeContextProvider(ide=ide),
                 FileContextProvider(workspace_dir=ide.workspace_directory)
             ])
-        await autopilot.context_manager.load_index()
+        await autopilot.context_manager.load_index(ide.workspace_directory)
 
         return autopilot
 
diff --git a/continuedev/src/continuedev/core/context.py b/continuedev/src/continuedev/core/context.py
index 78a747b2..86522ce1 100644
--- a/continuedev/src/continuedev/core/context.py
+++ b/continuedev/src/continuedev/core/context.py
@@ -35,7 +35,7 @@ class ContextProvider(BaseModel):
         return self.selected_items
 
     @abstractmethod
-    async def provide_context_items(self) -> List[ContextItem]:
+    async def provide_context_items(self, workspace_dir: str) -> List[ContextItem]:
         """
         Provide documents for search index. This is run on startup.
 
@@ -159,9 +159,9 @@ class ContextManager:
 
         return cls(context_providers)
 
-    async def load_index(self):
+    async def load_index(self, workspace_dir: str):
         for _, provider in self.context_providers.items():
-            context_items = await provider.provide_context_items()
+            context_items = await provider.provide_context_items(workspace_dir)
             documents = [
                 {
                     "id": item.description.id.to_string(),