small patches to streaming

2 files changed, 6 insertions, 1 deletions

diff --git a/continuedev/src/continuedev/models/main.py b/continuedev/src/continuedev/models/main.py
index c9011b29..d5f6e650 100644
--- a/continuedev/src/continuedev/models/main.py
+++ b/continuedev/src/continuedev/models/main.py
@@ -107,7 +107,7 @@ class Range(BaseModel):
         lines = content.splitlines()
         if len(lines) == 0:
             return Range.from_shorthand(0, 0, 0, 0)
-        return Range.from_shorthand(0, 0, len(lines) - 1, len(lines[-1]) - 1)
+        return Range.from_shorthand(0, 0, len(lines), 0)
 
     @staticmethod
     def from_snippet_in_file(content: str, snippet: str) -> "Range":
diff --git a/extension/src/continueIdeClient.ts b/extension/src/continueIdeClient.ts
index d983d93a..2e641054 100644
--- a/extension/src/continueIdeClient.ts
+++ b/extension/src/continueIdeClient.ts
@@ -173,6 +173,11 @@ class IdeProtocolClient {
         }
       );
 
+      setTimeout(() => {
+        cursorDisposable.dispose();
+        editor.setDecorations(decorationType, []);
+      }, 2500);
+
       if (this._lastDecorationType) {
         editor.setDecorations(this._lastDecorationType, []);
       }