imp-core 0.1.1

Agent engine for imp: loop, tools, sessions, hooks, context, and SDK
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

use imp_core::sdk::{AgentEvent, ImpSession, Result, SessionOptions};

#[tokio::main]
async fn main() -> Result<()> {
    let prompt = std::env::args()
        .nth(1)
        .unwrap_or_else(|| "Summarize the current project.".to_string());

    let mut session = ImpSession::create(SessionOptions {
        cwd: std::env::current_dir()?,
        ..Default::default()
    })
    .await?;

    session.prompt(&prompt).await?;

    while let Some(event) = session.recv_event().await {
        match event {
            AgentEvent::MessageDelta { delta } => {
                if let imp_core::imp_llm::StreamEvent::TextDelta { text } = delta {
                    print!("{text}");
                }
            }
            AgentEvent::ToolExecutionStart { tool_name, .. } => {
                eprintln!("\n[tool:{tool_name}]");
            }
            AgentEvent::AgentEnd { .. } => break,
            AgentEvent::Error { error } => eprintln!("\n[error] {error}"),
            _ => {}
        }
    }

    session.wait().await
}