feat: do not use fim template when suffix is empty string

improve-workflow
Meng Zhang 2023-06-15 09:27:32 -07:00
parent 11e4a5991d
commit 8ee700089f
1 changed files with 2 additions and 3 deletions

View File

@ -87,11 +87,10 @@ pub async fn completion(
.unwrap(); .unwrap();
let prompt = if let Some(Segments { prefix, suffix }) = request.segments { let prompt = if let Some(Segments { prefix, suffix }) = request.segments {
if let Some(prompt_template) = &state.prompt_template { if let (Some(prompt_template), Some(suffix)) = (&state.prompt_template, suffix) {
if let Some(suffix) = suffix { if !suffix.is_empty() {
strfmt!(prompt_template, prefix => prefix, suffix => suffix).unwrap() strfmt!(prompt_template, prefix => prefix, suffix => suffix).unwrap()
} else { } else {
// If suffix is empty, just returns prefix.
prefix prefix
} }
} else { } else {