use crate::{FlowFuture, call_tools, invoke, Agent, Message};
pub fn call_tools_flow<'a>(agent: &'a mut Agent, prompt: String) -> FlowFuture<'a> {
Box::pin(async move {
agent.history.push(Message::user(prompt));
let response = invoke(agent).await?;
if let Some(tc) = response.message.tool_calls.clone() {
for tool_msg in call_tools(agent, &tc).await {
agent.history.push(tool_msg);
}
}
agent.notify(crate::NotificationContent::Done(true, response.message.content.clone())).await;
Ok(response.message)
})
}