diff options
Diffstat (limited to 'continuedev')
| -rw-r--r-- | continuedev/src/continuedev/core/context.py | 5 | ||||
| -rw-r--r-- | continuedev/src/continuedev/plugins/steps/main.py | 4 |
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), |