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

feat: turn on index server if avaiable (#536)

r0.3
Meng Zhang 2023-10-11 16:27:41 -07:00 committed by GitHub
parent 1f9e248dfa
commit 92c1f5a8c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
crates/tabby/src/serve/completions/prompt.rs
View File

@ -1,7 +1,7 @@
use std::{env, sync::Arc}; use std::sync::Arc;
use strfmt::strfmt; use strfmt::strfmt;
use tracing::{info, warn}; use tracing::warn;
use super::Segments; use super::Segments;
use crate::serve::{completions::languages::get_language, search::IndexServer}; use crate::serve::{completions::languages::get_language, search::IndexServer};
@ -17,13 +17,6 @@ pub struct PromptBuilder {
impl PromptBuilder { impl PromptBuilder {
pub fn new(prompt_template: Option<String>, index_server: Option<Arc<IndexServer>>) -> Self { pub fn new(prompt_template: Option<String>, index_server: Option<Arc<IndexServer>>) -> Self {
let index_server = if env::var("TABBY_ENABLE_PROMPT_REWRITE").is_ok() {
info!("Prompt rewriting is enabled...");
index_server
} else {
None
};
PromptBuilder { PromptBuilder {
prompt_template, prompt_template,
index_server, index_server,