Ling0925
/
buster
mirror of https://github.com/buster-so/buster.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix ambiguous exports in metrics module and start fixing unused variables in post_chat_handler

This commit is contained in:
dal 2025-03-21 11:23:17 -06:00
parent 38c95180c1
commit 2ea69275a6
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
2 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
api/libs/handlers/src/chats/post_chat_handler.rs
View File

@ -133,7 +133,7 @@ impl ChunkTracker {
pub async fn post_chat_handler( pub async fn post_chat_handler(
request: ChatCreateNewChat, request: ChatCreateNewChat,
user: AuthenticatedUser, user: AuthenticatedUser,
tx: Option<mpsc::Sender<Result<(BusterContainer, ThreadEvent)>>>, _tx: Option<mpsc::Sender<Result<(BusterContainer, ThreadEvent)>>>,
) -> Result<ChatWithMessages> { ) -> Result<ChatWithMessages> {
// Create a request-local chunk tracker instance instead of using global static // Create a request-local chunk tracker instance instead of using global static
let chunk_tracker = ChunkTracker::new(); let chunk_tracker = ChunkTracker::new();
@ -158,7 +158,7 @@ pub async fn post_chat_handler(
); );
// Send initial chat state to client // Send initial chat state to client
if let Some(tx) = tx.clone() { if let Some(tx) = _tx.clone() {
tx.send(Ok(( tx.send(Ok((
BusterContainer::Chat(chat_with_messages.clone()), BusterContainer::Chat(chat_with_messages.clone()),
ThreadEvent::InitializeChat, ThreadEvent::InitializeChat,
@ -198,13 +198,13 @@ pub async fn post_chat_handler(
// Initialize raw_llm_messages with initial_messages // Initialize raw_llm_messages with initial_messages
let mut raw_llm_messages = initial_messages.clone(); let mut raw_llm_messages = initial_messages.clone();
let raw_response_message = String::new(); let _raw_response_message = String::new();
// Initialize the agent thread // Initialize the agent thread
let mut chat = AgentThread::new(Some(chat_id), user.id, initial_messages); let mut chat = AgentThread::new(Some(chat_id), user.id, initial_messages);
let title_handle = { let title_handle = {
let tx = tx.clone(); let tx = _tx.clone();
let chat_id = chat_id.clone(); let chat_id = chat_id.clone();
let message_id = message_id.clone(); let message_id = message_id.clone();
let user_id = user.id.clone(); let user_id = user.id.clone();
@ -514,8 +514,8 @@ async fn process_completed_files(
conn: &mut diesel_async::AsyncPgConnection, conn: &mut diesel_async::AsyncPgConnection,
message: &Message, message: &Message,
messages: &[AgentMessage], messages: &[AgentMessage],
organization_id: &Uuid, _organization_id: &Uuid,
user_id: &Uuid, _user_id: &Uuid,
chunk_tracker: &ChunkTracker, chunk_tracker: &ChunkTracker,
) -> Result<()> { ) -> Result<()> {
let mut transformed_messages = Vec::new(); let mut transformed_messages = Vec::new();
@ -540,7 +540,7 @@ async fn process_completed_files(
continue; continue;
} }
if let Some(file_content) = file.files.get(file_id) { if let Some(_file_content) = file.files.get(file_id) {
// Only process files that have completed reasoning // Only process files that have completed reasoning
if file.status == "completed" { if file.status == "completed" {
// Create message-to-file association // Create message-to-file association

8
api/libs/handlers/src/metrics/mod.rs
View File

@ -9,13 +9,19 @@ pub mod update_metric_handler;
pub mod types; pub mod types;
pub mod sharing; pub mod sharing;
// Re-export specific items from handlers
pub use add_metric_to_collections_handler::*; pub use add_metric_to_collections_handler::*;
pub use delete_metric_handler::*; pub use delete_metric_handler::*;
pub use get_metric_data_handler::*;
pub use get_metric_handler::*; pub use get_metric_handler::*;
pub use list_metrics_handler::*; pub use list_metrics_handler::*;
pub use post_metric_dashboard_handler::*; pub use post_metric_dashboard_handler::*;
pub use remove_metrics_from_collection_handler::*; pub use remove_metrics_from_collection_handler::*;
pub use update_metric_handler::*; pub use update_metric_handler::*;
// For get_metric_data_handler, only export the handler functions and request types
// but not the types that conflict with types.rs
pub use get_metric_data_handler::{get_metric_data_handler, GetMetricDataRequest, MetricDataResponse};
// Re-export types and sharing
pub use types::*; pub use types::*;
pub use sharing::*; pub use sharing::*;