Merge pull request #369 from twinnydotdev/development

development > main
twinnydotdev · Nov 2, 2024 · 57cc2ae · 57cc2ae
2 parents 5049404 + c9d523c
commit 57cc2ae
Show file tree

Hide file tree

Showing 3 changed files with 21 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.18.2",
+  "version": "3.18.3",
   "icon": "assets/icon.png",
   "keywords": [
     "code-inference",

diff --git a/src/extension/fim-templates.ts b/src/extension/fim-templates.ts
@@ -81,9 +81,25 @@ const getFimPromptTemplateCodestral = ({
   return `${fileContext}\n\n[SUFFIX]${suffix}[PREFIX]${heading}${prefix}`
 }
 
-const getFimPromptTemplateQwen = ({ prefixSuffix }: FimPromptTemplate) => {
+const getFimPromptTemplateQwen = ({
+  context,
+  header,
+  fileContextEnabled,
+  prefixSuffix,
+  language
+}: FimPromptTemplate) => {
   const { prefix, suffix } = prefixSuffix
-  return `<|fim_prefix|>${prefix}<|fim_suffix|>${suffix}<|fim_middle|>`
+  const { fileContext, heading } = getFileContext(
+    fileContextEnabled,
+    context,
+    language,
+    header
+  )
+  if (fileContextEnabled) {
+    return `<|file_sep|>${fileContext}\n\n<|file_sep|>${heading}<|fim_prefix|>${prefix}<|fim_suffix|>${suffix}<|fim_middle|>`
+  } else {
+    return `<|fim_prefix|>${prefix}<|fim_suffix|>${suffix}<|fim_middle|>`
+  }
 }
 
 const getFimPromptTemplateOther = ({