summaryrefslogtreecommitdiff
path: root/continuedev/src/continuedev/libs/steps/main.py
diff options
context:
space:
mode:
authorNate Sesti <sestinj@gmail.com>2023-06-02 13:44:33 -0400
committerNate Sesti <sestinj@gmail.com>2023-06-02 13:44:33 -0400
commita6a66f8a8b42eca861b05d6e7ccc0d0c2cfc8706 (patch)
tree4a1223bed2d52dbff9ae5e9fb110155e87778a50 /continuedev/src/continuedev/libs/steps/main.py
parent2c141f3ac1436be9ed5b8cc15ba882c3d91279ac (diff)
downloadsncontinue-a6a66f8a8b42eca861b05d6e7ccc0d0c2cfc8706.tar.gz
sncontinue-a6a66f8a8b42eca861b05d6e7ccc0d0c2cfc8706.tar.bz2
sncontinue-a6a66f8a8b42eca861b05d6e7ccc0d0c2cfc8706.zip
policy + starcoder description fixes
Diffstat (limited to 'continuedev/src/continuedev/libs/steps/main.py')
-rw-r--r--continuedev/src/continuedev/libs/steps/main.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/continuedev/src/continuedev/libs/steps/main.py b/continuedev/src/continuedev/libs/steps/main.py
index aefbe084..73ad3352 100644
--- a/continuedev/src/continuedev/libs/steps/main.py
+++ b/continuedev/src/continuedev/libs/steps/main.py
@@ -253,6 +253,8 @@ class StarCoderEditHighlightedCodeStep(Step):
if completion.endswith(eot_token):
completion = completion[:completion.rindex(eot_token)]
+ self._prompt_and_completion += prompt + completion
+
await sdk.ide.applyFileSystemEdit(
FileEdit(filepath=rif.filepath, range=rif.range, replacement=completion))
await sdk.ide.saveFile(rif.filepath)