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

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

@ -104,7 +104,7 @@ impl BusterSuperAgent {
HashMap::new(),
user_id,
session_id,
"manager_agent".to_string(),
"buster_super_agent".to_string(),
));
let manager = Self { agent };
@ -116,7 +116,7 @@ impl BusterSuperAgent {
// Create a new agent with the same core properties and shared state/stream
let agent = Arc::new(Agent::from_existing(
existing_agent,
"manager_agent".to_string(),
"buster_super_agent".to_string(),
));
let manager = Self { agent };
manager.load_tools().await?;
@ -127,7 +127,7 @@ impl BusterSuperAgent {
&self,
thread: &mut AgentThread,
) -> 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
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.
---
### Actions Available (Tools)