areev 0.1.4

Rust SDK for the Areev knowledge database — gRPC and HTTP transports
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

//! End-to-end example: Harness chat with a client-executed tool (HPL).
//!
//! Demonstrates the `chat_interactive` sugar that abstracts the Flow-A
//! pause/resume round-trip.
//!
//! Run (with a live Areev + harness bound to a `client://get_weather` tool):
//!
//! ```bash
//! AREEV_URL=http://localhost:4009 \
//! AREEV_MEMORY_ID=default \
//! AREEV_API_KEY=ar_... \
//!   cargo run --example harness_client_tool
//! ```

use std::pin::Pin;

use areev::{Areev, ChatExecutors, Result};
use serde_json::json;

#[tokio::main]
async fn main() -> Result<()> {
    let areev = Areev::from_env();

    // Register a trivial in-process executor for `get_weather(city: String)`.
    let mut executors: ChatExecutors = ChatExecutors::new();
    executors.insert("get_weather", Box::new(|_name, args| {
        Pin::from(Box::new(async move {
            let city = args
                .get("city")
                .and_then(|v| v.as_str())
                .unwrap_or("unknown");
            Ok(json!({
                "city": city,
                "temp_c": 22,
                "conditions": "sunny",
            }))
        }))
    }));

    let response = areev
        .chat_interactive(
            "weather-harness",
            "conv-example-1",
            "What's the weather in Paris?",
            &executors,
        )
        .await?;

    println!("assistant: {}", response.text);
    println!(
        "iterations={} tool_calls={} provider={}",
        response.iterations,
        response.tool_calls.len(),
        response.provider
    );
    Ok(())
}