refactor: move tabby/src/search.rs -> tabby/src/services/search.rs

extract-routes
Meng Zhang 2023-11-11 14:20:17 -08:00
parent fce94f622b
commit 7695a667bb
6 changed files with 20 additions and 3 deletions

12
Cargo.lock generated
View File

@ -1842,6 +1842,17 @@ dependencies = [
"want",
]
[[package]]
name = "hyper-reverse-proxy"
version = "0.5.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "cc1af9b1b483fb9f33bd1cda26b35eacf902f0d116fcf0d56075ea5e5923b935"
dependencies = [
"hyper",
"lazy_static",
"unicase",
]
[[package]]
name = "hyper-timeout"
version = "0.4.1"
@ -4076,6 +4087,7 @@ dependencies = [
"futures",
"http-api-bindings",
"hyper",
"hyper-reverse-proxy",
"lazy_static",
"llama-cpp-bindings",
"mime_guess",

View File

@ -46,6 +46,7 @@ regex.workspace = true
llama-cpp-bindings = { path = "../llama-cpp-bindings" }
futures.workspace = true
async-trait.workspace = true
hyper-reverse-proxy = "0.5.1"
[dependencies.uuid]
version = "1.3.3"

View File

@ -1,8 +1,9 @@
mod chat;
mod download;
mod search;
mod serve;
mod services;
use clap::{Parser, Subcommand};
use opentelemetry::{
global,

View File

@ -32,7 +32,7 @@ use self::{
engine::{create_engine, EngineInfo},
health::HealthState,
};
use crate::{chat::ChatService, fatal, search::create_code_search};
use crate::{chat::ChatService, fatal};
#[derive(OpenApi)]
#[openapi(
@ -181,7 +181,7 @@ async fn load_model(args: &ServeArgs) {
}
async fn api_router(args: &ServeArgs, config: &Config) -> Router {
let code = Arc::new(create_code_search());
let code = Arc::new(crate::services::create_code_search());
let completion_state = {
let (
engine,

View File

@ -0,0 +1,3 @@
mod search;
pub use search::*;