refactor: remove crate re-export aliases from lib.rs (#18737)

This commit is contained in:
Simo Lin
2026-02-12 09:17:24 -08:00
committed by GitHub
parent e730c728d3
commit 0236522214
74 changed files with 240 additions and 284 deletions
+6 -6
View File
@@ -18,6 +18,9 @@ use std::{
sync::{Arc, Mutex, OnceLock},
};
use data_connector::{
MemoryConversationItemStorage, MemoryConversationStorage, MemoryResponseStorage,
};
use mock_worker::{MockWorker, MockWorkerConfig};
use serde_json::json;
use smg::{
@@ -27,9 +30,6 @@ use smg::{
BasicWorkerBuilder, Job, LoadMonitor, ModelCard, RuntimeType, Worker, WorkerRegistry,
WorkerType,
},
data_connector::{
MemoryConversationItemStorage, MemoryConversationStorage, MemoryResponseStorage,
},
middleware::TokenBucket,
policies::PolicyRegistry,
protocols::common::{Function, Tool},
@@ -386,7 +386,7 @@ pub async fn create_test_context(config: RouterConfig) -> Arc<AppContext> {
}
// Initialize MCP manager with empty config
use smg::mcp::{McpConfig, McpManager};
use smg_mcp::{McpConfig, McpManager};
let empty_config = McpConfig {
servers: vec![],
pool: Default::default(),
@@ -510,7 +510,7 @@ pub async fn create_test_context_with_parsers(config: RouterConfig) -> Arc<AppCo
}
// Initialize MCP manager with empty config
use smg::mcp::{McpConfig, McpManager};
use smg_mcp::{McpConfig, McpManager};
let empty_config = McpConfig {
servers: vec![],
pool: Default::default(),
@@ -535,7 +535,7 @@ pub async fn create_test_context_with_mcp_config(
config: RouterConfig,
mcp_config_path: &str,
) -> Arc<AppContext> {
use smg::mcp::{McpConfig, McpManager};
use smg_mcp::{McpConfig, McpManager};
let client = reqwest::Client::new();
+4 -4
View File
@@ -1,6 +1,9 @@
use std::sync::{Arc, OnceLock};
use axum::Router;
use data_connector::{
MemoryConversationItemStorage, MemoryConversationStorage, MemoryResponseStorage,
};
use reqwest::Client;
use smg::{
app_context::AppContext,
@@ -8,16 +11,13 @@ use smg::{
core::{
BasicWorkerBuilder, LoadMonitor, ModelCard, RuntimeType, Worker, WorkerRegistry, WorkerType,
},
data_connector::{
MemoryConversationItemStorage, MemoryConversationStorage, MemoryResponseStorage,
},
mcp::{McpConfig, McpManager},
middleware::{AuthConfig, TokenBucket},
policies::PolicyRegistry,
routers::RouterTrait,
server::{build_app, AppState},
tokenizer::registry::TokenizerRegistry,
};
use smg_mcp::{McpConfig, McpManager};
/// Create a test Axum application using the actual server's build_app function
#[allow(dead_code)]