[Router] Consolidate ConnectionMode enum to core module (#11937)

This commit is contained in:
Arthur Cheng
2025-10-23 05:15:49 -07:00
committed by GitHub
parent e321c97113
commit 53c2934dce
11 changed files with 43 additions and 48 deletions

View File

@@ -194,7 +194,7 @@ struct Router {
queue_size: usize,
queue_timeout_secs: u64,
rate_limit_tokens_per_second: Option<i32>,
connection_mode: config::ConnectionMode,
connection_mode: core::ConnectionMode,
model_path: Option<String>,
tokenizer_path: Option<String>,
chat_template: Option<String>,
@@ -211,13 +211,13 @@ struct Router {
impl Router {
/// Determine connection mode from worker URLs
fn determine_connection_mode(worker_urls: &[String]) -> config::ConnectionMode {
fn determine_connection_mode(worker_urls: &[String]) -> core::ConnectionMode {
for url in worker_urls {
if url.starts_with("grpc://") || url.starts_with("grpcs://") {
return config::ConnectionMode::Grpc;
return core::ConnectionMode::Grpc { port: None };
}
}
config::ConnectionMode::Http
core::ConnectionMode::Http
}
pub fn to_router_config(&self) -> config::ConfigResult<config::RouterConfig> {