From 8e58ff4003baa19feea9f75bda3567eb057a778f Mon Sep 17 00:00:00 2001
From: Meng Zhang <meng@tabbyml.com>
Date: Sat, 11 Nov 2023 14:35:19 -0800
Subject: [PATCH] refactor: move chat.rs to services/ rename search.rs ->
 code.rs

---
 crates/tabby/src/main.rs                         |  1 -
 crates/tabby/src/serve/chat.rs                   |  2 +-
 crates/tabby/src/serve/mod.rs                    | 10 +++++-----
 crates/tabby/src/{ => services}/chat.rs          |  0
 crates/tabby/src/{ => services}/chat/prompt.rs   |  0
 crates/tabby/src/services/{search.rs => code.rs} |  0
 crates/tabby/src/services/mod.rs                 |  5 ++---
 7 files changed, 8 insertions(+), 10 deletions(-)
 rename crates/tabby/src/{ => services}/chat.rs (100%)
 rename crates/tabby/src/{ => services}/chat/prompt.rs (100%)
 rename crates/tabby/src/services/{search.rs => code.rs} (100%)

diff --git a/crates/tabby/src/main.rs b/crates/tabby/src/main.rs
index 978aed9..f7645cf 100644
--- a/crates/tabby/src/main.rs
+++ b/crates/tabby/src/main.rs
@@ -1,4 +1,3 @@
-mod chat;
 mod download;
 mod serve;
 
diff --git a/crates/tabby/src/serve/chat.rs b/crates/tabby/src/serve/chat.rs
index f24aa67..4b51683 100644
--- a/crates/tabby/src/serve/chat.rs
+++ b/crates/tabby/src/serve/chat.rs
@@ -9,7 +9,7 @@ use axum::{
 use axum_streams::StreamBodyAs;
 use tracing::instrument;
 
-use crate::chat::{ChatCompletionRequest, ChatService};
+use crate::services::chat::{ChatCompletionRequest, ChatService};
 
 #[utoipa::path(
     post,
diff --git a/crates/tabby/src/serve/mod.rs b/crates/tabby/src/serve/mod.rs
index b58141d..2aafac7 100644
--- a/crates/tabby/src/serve/mod.rs
+++ b/crates/tabby/src/serve/mod.rs
@@ -32,7 +32,7 @@ use self::{
     engine::{create_engine, EngineInfo},
     health::HealthState,
 };
-use crate::{chat::ChatService, fatal};
+use crate::{fatal, services::chat::ChatService};
 
 #[derive(OpenApi)]
 #[openapi(
@@ -61,9 +61,9 @@ Install following IDE / Editor extensions to get started with [Tabby](https://gi
         completions::Snippet,
         completions::DebugOptions,
         completions::DebugData,
-        crate::chat::ChatCompletionRequest,
-        crate::chat::Message,
-        crate::chat::ChatCompletionChunk,
+        crate::services::chat::ChatCompletionRequest,
+        crate::services::chat::Message,
+        crate::services::chat::ChatCompletionChunk,
         health::HealthState,
         health::Version,
         SearchResponse,
@@ -174,7 +174,7 @@ async fn load_model(args: &ServeArgs) {
 }
 
 async fn api_router(args: &ServeArgs, config: &Config) -> Router {
-    let code = Arc::new(crate::services::create_code_search());
+    let code = Arc::new(crate::services::code::create_code_search());
     let completion_state = {
         let (
             engine,
diff --git a/crates/tabby/src/chat.rs b/crates/tabby/src/services/chat.rs
similarity index 100%
rename from crates/tabby/src/chat.rs
rename to crates/tabby/src/services/chat.rs
diff --git a/crates/tabby/src/chat/prompt.rs b/crates/tabby/src/services/chat/prompt.rs
similarity index 100%
rename from crates/tabby/src/chat/prompt.rs
rename to crates/tabby/src/services/chat/prompt.rs
diff --git a/crates/tabby/src/services/search.rs b/crates/tabby/src/services/code.rs
similarity index 100%
rename from crates/tabby/src/services/search.rs
rename to crates/tabby/src/services/code.rs
diff --git a/crates/tabby/src/services/mod.rs b/crates/tabby/src/services/mod.rs
index 9675e1f..74476ac 100644
--- a/crates/tabby/src/services/mod.rs
+++ b/crates/tabby/src/services/mod.rs
@@ -1,3 +1,2 @@
-mod search;
-
-pub use search::*;
+pub mod chat;
+pub mod code;