From 80627c7f6ee504ebc308ead92d1963f6210b098c Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Sun, 2 Jul 2023 18:31:54 -0700
Subject: create new file to edit if no ranges selected
---
continuedev/src/continuedev/steps/main.py | 7 +++++++
1 file changed, 7 insertions(+)
(limited to 'continuedev')
diff --git a/continuedev/src/continuedev/steps/main.py b/continuedev/src/continuedev/steps/main.py
index def1af4e..3cf78c40 100644
--- a/continuedev/src/continuedev/steps/main.py
+++ b/continuedev/src/continuedev/steps/main.py
@@ -266,6 +266,13 @@ class EditHighlightedCodeStep(Step):
range_in_files = [RangeInFile.from_entire_file(
filepath, content) for filepath, content in contents.items()]
+ # If still no highlighted code, create a new file and edit there
+ if len(range_in_files) == 0:
+ # Create a new file
+ new_file_path = "new_file.txt"
+ await sdk.add_file(new_file_path)
+ range_in_files = [RangeInFile.from_entire_file(new_file_path, "")]
+
await sdk.run_step(DefaultModelEditCodeStep(user_input=self.user_input, range_in_files=range_in_files))
--
cgit v1.2.3-70-g09d2