Merge pull request #162 from rjmacarthy/development

development > main
twinnydotdev · Mar 4, 2024 · 0e56487 · 0e56487
2 parents 7bf3858 + 4693136
commit 0e56487
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -2,7 +2,7 @@
   "name": "twinny",
   "displayName": "twinny - AI Code Completion and Chat",
   "description": "Locally hosted AI code completion plugin for vscode",
-  "version": "3.7.9",
+  "version": "3.7.10",
   "icon": "assets/icon.png",
   "keywords": [
     "code-inference",

diff --git a/src/extension/utils.ts b/src/extension/utils.ts
@@ -58,14 +58,18 @@ export const getPrefixSuffix = (
   numLines: number,
   document: TextDocument,
   position: Position,
-  contextRatio = [0.15, 0.85]
+  contextRatio = [0.85, 0.15]
 ): PrefixSuffix => {
   const currentLine = position.line
   const numLinesToEnd = document.lineCount - currentLine
-
   let numLinesPrefix = Math.floor(Math.abs(numLines * contextRatio[0]))
   let numLinesSuffix = Math.ceil(Math.abs(numLines * contextRatio[1]))
 
+  if (numLinesPrefix > currentLine) {
+    numLinesSuffix += numLinesPrefix - currentLine
+    numLinesPrefix = currentLine
+  }
+
   if (numLinesSuffix > numLinesToEnd) {
     numLinesPrefix += numLinesSuffix - numLinesToEnd
     numLinesSuffix = numLinesToEnd