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

updated some names and removed some bad fields

This commit is contained in:
dal 2025-03-04 15:08:07 -07:00
parent e6aafa1d30
commit 0e95dfd4cb
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
2 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -329,7 +329,7 @@ impl Agent {
model: self.model.clone(), model: self.model.clone(),
messages: thread.messages.clone(), messages: thread.messages.clone(),
tools: if tools.is_empty() { None } else { Some(tools) }, tools: if tools.is_empty() { None } else { Some(tools) },
tool_choice: Some(ToolChoice::Required), tool_choice: Some(ToolChoice::Auto),
stream: Some(true), // Enable streaming stream: Some(true), // Enable streaming
metadata: Some(Metadata { metadata: Some(Metadata {
generation_name: "agent".to_string(), generation_name: "agent".to_string(),
@ -337,7 +337,6 @@ impl Agent {
session_id: thread.id.to_string(), session_id: thread.id.to_string(),
trace_id: thread.id.to_string(), trace_id: thread.id.to_string(),
}), }),
store: Some(true),
..Default::default() ..Default::default()
}; };

8
api/libs/agents/src/agents/buster_super_agent.rs
View File

@ -104,7 +104,7 @@ impl BusterSuperAgent {
HashMap::new(), HashMap::new(),
user_id, user_id,
session_id, session_id,
"manager_agent".to_string(), "buster_super_agent".to_string(),
)); ));
let manager = Self { agent }; let manager = Self { agent };
@ -116,7 +116,7 @@ impl BusterSuperAgent {
// Create a new agent with the same core properties and shared state/stream // Create a new agent with the same core properties and shared state/stream
let agent = Arc::new(Agent::from_existing( let agent = Arc::new(Agent::from_existing(
existing_agent, existing_agent,
"manager_agent".to_string(), "buster_super_agent".to_string(),
)); ));
let manager = Self { agent }; let manager = Self { agent };
manager.load_tools().await?; manager.load_tools().await?;
@ -127,7 +127,7 @@ impl BusterSuperAgent {
&self, &self,
thread: &mut AgentThread, thread: &mut AgentThread,
) -> Result<broadcast::Receiver<Result<AgentMessage, AgentError>>> { ) -> Result<broadcast::Receiver<Result<AgentMessage, AgentError>>> {
thread.set_developer_message(MANAGER_AGENT_PROMPT.to_string()); thread.set_developer_message(BUSTER_SUPER_AGENT_PROMPT.to_string());
// Get shutdown receiver // Get shutdown receiver
let rx = self.stream_process_thread(thread).await?; let rx = self.stream_process_thread(thread).await?;
@ -141,7 +141,7 @@ impl BusterSuperAgent {
} }
} }
const MANAGER_AGENT_PROMPT: &str = r##"### Role & Task const BUSTER_SUPER_AGENT_PROMPT: &str = r##"### Role & Task
You are Buster, an expert analytics and data engineer. Your job is to assess what data is available and then provide fast, accurate answers to analytics questions from non-technical users. You do this by analyzing user requests, searching across a data catalog, and building metrics or dashboards. You are Buster, an expert analytics and data engineer. Your job is to assess what data is available and then provide fast, accurate answers to analytics questions from non-technical users. You do this by analyzing user requests, searching across a data catalog, and building metrics or dashboards.
--- ---
### Actions Available (Tools) ### Actions Available (Tools)