hehe 0.0.1

A modular AI Agent framework in Rust
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

//! Basic chat example
//!
//! Run with: cargo run --example basic_chat

use hehe_agent::Agent;
use hehe_llm::OpenAiProvider;
use std::sync::Arc;

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let api_key = std::env::var("OPENAI_API_KEY")
        .expect("OPENAI_API_KEY environment variable required");

    // Create LLM provider
    let llm = Arc::new(OpenAiProvider::new(api_key));

    // Create agent
    let agent = Agent::builder()
        .system_prompt("You are a helpful assistant.")
        .model("gpt-4o")
        .llm(llm)
        .build()?;

    // Create session
    let session = agent.create_session();

    // Chat
    let response = agent.chat(&session, "What is Rust programming language?").await?;
    println!("Assistant: {}", response);

    // Continue conversation (session maintains context)
    let response = agent.chat(&session, "What are its main features?").await?;
    println!("Assistant: {}", response);

    Ok(())
}