feat(agent): add experimental option: strip auto-closing chars in pro…

…mpt suffix.
TabbyML · Oct 27, 2023 · 58372bd · 58372bd
1 parent d6c1324
commit 58372bd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/clients/tabby-agent/src/AgentConfig.ts b/clients/tabby-agent/src/AgentConfig.ts
@@ -8,6 +8,7 @@ export type AgentConfig = {
   };
   completion: {
     prompt: {
+      stripAutoClosingCharacters: boolean;
       maxPrefixLines: number;
       maxSuffixLines: number;
     };
@@ -46,6 +47,7 @@ export const defaultAgentConfig: AgentConfig = {
   },
   completion: {
     prompt: {
+      stripAutoClosingCharacters: false,
       maxPrefixLines: 20,
       maxSuffixLines: 20,
     },

diff --git a/clients/tabby-agent/src/TabbyAgent.ts b/clients/tabby-agent/src/TabbyAgent.ts
@@ -264,10 +264,14 @@ export class TabbyAgent extends EventEmitter implements Agent {
     const maxPrefixLines = this.config.completion.prompt.maxPrefixLines;
     const maxSuffixLines = this.config.completion.prompt.maxSuffixLines;
     const { prefixLines, suffixLines } = context;
-    return {
-      prefix: prefixLines.slice(Math.max(prefixLines.length - maxPrefixLines, 0)).join(""),
-      suffix: suffixLines.slice(0, maxSuffixLines).join(""),
-    };
+    const prefix = prefixLines.slice(Math.max(prefixLines.length - maxPrefixLines, 0)).join("");
+    let suffix;
+    if (this.config.completion.prompt.stripAutoClosingCharacters && context.mode !== "fill-in-line") {
+      suffix = "\n" + suffixLines.slice(1, maxSuffixLines).join("");
+    } else {
+      suffix = suffixLines.slice(0, maxSuffixLines).join("");
+    }
+    return { prefix, suffix };
   }
 
   private calculateReplaceRange(response: CompletionResponse, context: CompletionContext): CompletionResponse {