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

move to traces

This commit is contained in:
dal 2025-03-10 08:55:06 -06:00
parent 7b22e06640
commit 87b5489e10
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
4 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/libs/agents/src/agent.rs
View File

@ -416,7 +416,7 @@ impl Agent {
generation_name: "agent".to_string(), generation_name: "agent".to_string(),
user_id: thread.user_id.to_string(), user_id: thread.user_id.to_string(),
session_id: thread.id.to_string(), session_id: thread.id.to_string(),
trace_id: thread.id.to_string(), trace_id: None,
}), }),
..Default::default() ..Default::default()
}; };

2
api/libs/agents/src/tools/categories/file_tools/search_data_catalog.rs
View File

@ -141,7 +141,7 @@ impl SearchDataCatalogTool {
generation_name: "search_data_catalog".to_string(), generation_name: "search_data_catalog".to_string(),
user_id: user_id.to_string(), user_id: user_id.to_string(),
session_id: session_id.to_string(), session_id: session_id.to_string(),
trace_id: session_id.to_string(), trace_id: None,
}), }),
reasoning_effort: Some("low".to_string()), reasoning_effort: Some("low".to_string()),
max_completion_tokens: Some(8092), max_completion_tokens: Some(8092),

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

@ -1572,7 +1572,7 @@ pub async fn generate_conversation_title(
generation_name: "conversation_title".to_string(), generation_name: "conversation_title".to_string(),
user_id: user_id.to_string(), user_id: user_id.to_string(),
session_id: session_id.to_string(), session_id: session_id.to_string(),
trace_id: session_id.to_string(), trace_id: None,
}), }),
..Default::default() ..Default::default()
}; };

4
api/libs/litellm/src/types.rs
View File

@ -59,7 +59,7 @@ pub struct Metadata {
pub generation_name: String, pub generation_name: String,
pub user_id: String, pub user_id: String,
pub session_id: String, pub session_id: String,
pub trace_id: String, pub trace_id: Option<String>,
} }
impl Default for ChatCompletionRequest { impl Default for ChatCompletionRequest {
@ -149,7 +149,7 @@ pub enum AgentMessage {
#[serde(skip)] #[serde(skip)]
progress: MessageProgress, progress: MessageProgress,
}, },
Done Done,
} }
// Helper methods for Message // Helper methods for Message