fix: :bug: fix when multiple cursor ranges are selected

2 files changed, 8 insertions, 1 deletions

diff --git a/continuedev/src/continuedev/core/context.py b/continuedev/src/continuedev/core/context.py
index e6642404..d51a32e2 100644
--- a/continuedev/src/continuedev/core/context.py
+++ b/continuedev/src/continuedev/core/context.py
@@ -253,7 +253,10 @@ class ContextManager:
                 index = await search_client.get_index(SEARCH_INDEX_NAME)
                 await index.add_documents(documents or [])
 
-            await asyncio.wait_for(add_docs(), timeout=5)
+            try:
+                await asyncio.wait_for(add_docs(), timeout=5)
+            except asyncio.TimeoutError:
+                logger.warning("Failed to add document to meilisearch in 5 seconds")
 
     @staticmethod
     async def delete_documents(ids):
diff --git a/continuedev/src/continuedev/plugins/steps/main.py b/continuedev/src/continuedev/plugins/steps/main.py
index da9cf5b2..ab5775c6 100644
--- a/continuedev/src/continuedev/plugins/steps/main.py
+++ b/continuedev/src/continuedev/plugins/steps/main.py
@@ -273,6 +273,10 @@ class EditHighlightedCodeStep(Step):
                 title="No Code Selected",
             )
 
+        # If all of the ranges are point ranges, only edit the last one
+        if all([rif.range.start == rif.range.end for rif in range_in_files]):
+            range_in_files = [range_in_files[-1]]
+
         range_in_files = list(
             map(
                 lambda x: RangeInFile(filepath=x.filepath, range=x.range),