feat: do not use fim template when suffix is empty string
parent
11e4a5991d
commit
8ee700089f
|
|
@ -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 {
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue