diff options
Diffstat (limited to 'continuedev')
| -rw-r--r-- | continuedev/src/continuedev/core/sdk.py | 2 | ||||
| -rw-r--r-- | continuedev/src/continuedev/steps/chat.py | 10 |
2 files changed, 5 insertions, 7 deletions
diff --git a/continuedev/src/continuedev/core/sdk.py b/continuedev/src/continuedev/core/sdk.py index cfe2e436..efebe9e3 100644 --- a/continuedev/src/continuedev/core/sdk.py +++ b/continuedev/src/continuedev/core/sdk.py @@ -136,6 +136,8 @@ class ContinueSDK(AbstractContinueSDK): async def add_file(self, filename: str, content: Union[str, None]): filepath = await self._ensure_absolute_path(filename) + dir_name = os.path.dirname(filepath) + os.makedirs(dir_name, exist_ok=True) return await self.run_step(FileSystemEditStep(edit=AddFile(filepath=filepath, content=content))) async def delete_file(self, filename: str): diff --git a/continuedev/src/continuedev/steps/chat.py b/continuedev/src/continuedev/steps/chat.py index 34a97a17..9d3384dd 100644 --- a/continuedev/src/continuedev/steps/chat.py +++ b/continuedev/src/continuedev/steps/chat.py @@ -62,7 +62,7 @@ class SimpleChatStep(Step): class AddFileStep(Step): name: str = "Add File" - description = "Add a file to the workspace." + description = "Add a file to the workspace. Should always view the directory tree before this." filename: str file_contents: str @@ -70,11 +70,7 @@ class AddFileStep(Step): return f"Added a file named `{self.filename}` to the workspace." async def run(self, sdk: ContinueSDK): - try: - await sdk.add_file(self.filename, self.file_contents) - except FileNotFoundError: - self.description = f"File {self.filename} does not exist." - return + await sdk.add_file(self.filename, self.file_contents) await sdk.ide.setFileOpen(os.path.join(sdk.ide.workspace_directory, self.filename)) @@ -132,7 +128,7 @@ class ViewDirectoryTreeStep(Step): return f"Viewed the directory tree." async def run(self, sdk: ContinueSDK): - self.description = f"```\n{display_tree(sdk.ide.workspace_directory, True)}\n```" + self.description = f"```\n{display_tree(sdk.ide.workspace_directory, True, max_depth=2)}\n```" class EditFileStep(Step): |