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: { completion: {
prompt: { prompt: {
experimentalStripAutoClosingCharacters: boolean;
maxPrefixLines: number; maxPrefixLines: number;
maxSuffixLines: number; maxSuffixLines: number;
}; };
@ -46,6 +47,7 @@ export const defaultAgentConfig: AgentConfig = {
}, },
completion: { completion: {
prompt: { prompt: {
experimentalStripAutoClosingCharacters: false,
maxPrefixLines: 20, maxPrefixLines: 20,
maxSuffixLines: 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 maxPrefixLines = this.config.completion.prompt.maxPrefixLines;
const maxSuffixLines = this.config.completion.prompt.maxSuffixLines; const maxSuffixLines = this.config.completion.prompt.maxSuffixLines;
const { prefixLines, suffixLines } = context; const { prefixLines, suffixLines } = context;
return { const prefix = prefixLines.slice(Math.max(prefixLines.length - maxPrefixLines, 0)).join("");
prefix: prefixLines.slice(Math.max(prefixLines.length - maxPrefixLines, 0)).join(""), let suffix;
suffix: suffixLines.slice(0, maxSuffixLines).join(""), 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 { private calculateReplaceRange(response: CompletionResponse, context: CompletionContext): CompletionResponse {