From 8ee700089ff996e5ec7dbc6a351daedcbb2a85d4 Mon Sep 17 00:00:00 2001 From: Meng Zhang Date: Thu, 15 Jun 2023 09:27:32 -0700 Subject: [PATCH] feat: do not use fim template when suffix is empty string --- crates/tabby/src/serve/completions.rs | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/crates/tabby/src/serve/completions.rs b/crates/tabby/src/serve/completions.rs index fca76e1..93c0541 100644 --- a/crates/tabby/src/serve/completions.rs +++ b/crates/tabby/src/serve/completions.rs @@ -87,11 +87,10 @@ pub async fn completion( .unwrap(); let prompt = if let Some(Segments { prefix, suffix }) = request.segments { - if let Some(prompt_template) = &state.prompt_template { - if let Some(suffix) = suffix { + if let (Some(prompt_template), Some(suffix)) = (&state.prompt_template, suffix) { + if !suffix.is_empty() { strfmt!(prompt_template, prefix => prefix, suffix => suffix).unwrap() } else { - // If suffix is empty, just returns prefix. prefix } } else {