kwaak 0.8.2

Run a team of autonomous agents on your code, right from your terminal
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

use anyhow::Result;
use std::sync::Arc;
use swiftide::agents::{Agent, DefaultContext};
use swiftide::chat_completion::{ChatCompletion, Tool};
use swiftide::traits::AgentContext;

use crate::agent::{env_setup::AgentEnvironment, v1, RunningAgent};
use crate::commands::Responder;
use crate::repository::Repository;

pub async fn start_tool_evaluation_agent(
    repository: &Repository,
    responder: Arc<dyn Responder>,
    tools: Vec<Box<dyn Tool>>,
) -> Result<RunningAgent> {
    // Create agent with simplified tools
    let system_prompt = v1::build_system_prompt(repository)?;
    let agent_context: Arc<dyn AgentContext> = Arc::new(DefaultContext::default());
    let executor = Arc::new(swiftide::agents::tools::local_executor::LocalExecutor::default());
    let query_provider: Box<dyn ChatCompletion> =
        repository.config().query_provider().try_into()?;

    let responder_for_messages = responder.clone();
    let responder_for_tools = responder.clone();

    let agent = Agent::builder()
        .tools(tools)
        .system_prompt(system_prompt)
        .context(agent_context.clone())
        .llm(&query_provider)
        .on_new_message(move |_, message| {
            let responder = responder_for_messages.clone();
            let message = message.clone();
            Box::pin(async move {
                responder.agent_message(message);
                Ok(())
            })
        })
        .before_tool(move |_, tool| {
            let responder = responder_for_tools.clone();
            let tool = tool.clone();
            Box::pin(async move {
                responder.update(&format!("running tool {}", tool.name()));
                Ok(())
            })
        })
        .build()?;

    let agent = RunningAgent::builder()
        .agent(agent)
        .executor(executor)
        .agent_context(agent_context)
        .agent_environment(Arc::new(AgentEnvironment::default()))
        .build()?;

    Ok(agent)
}