root
/
tabby
mirror of https://gitee.com/zhang_1334717033/tabby.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(agent): add experimental option: strip auto-closing chars in prompt suffix. (#651) 

* feat(agent): add experimental option: strip auto-closing chars in prompt suffix.

* fix: rename settings adding experimental prefix.
release-notes-05
Zhiming Ma 2023-10-30 10:58:25 +08:00 committed by GitHub
parent b47bdd5d77
commit 238d81ad4f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clients/tabby-agent/src/AgentConfig.ts vendored
View File

@ -8,6 +8,7 @@ export type AgentConfig = {
};
completion: {
prompt: {
experimentalStripAutoClosingCharacters: boolean;
maxPrefixLines: number;
maxSuffixLines: number;
};
@ -46,6 +47,7 @@ export const defaultAgentConfig: AgentConfig = {
},
completion: {
prompt: {
experimentalStripAutoClosingCharacters: false,
maxPrefixLines: 20,
maxSuffixLines: 20,
},

12
clients/tabby-agent/src/TabbyAgent.ts vendored
View File

@ -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.experimentalStripAutoClosingCharacters && 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 {