oris-runtime 0.15.0

An agentic workflow runtime and programmable AI execution system in Rust: stateful graphs, agents, tools, and multi-step execution.
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

use std::sync::Arc;

use oris_runtime::{
    agent::{create_agent, HandoffAgentBuilder},
    schemas::messages::Message,
};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Initialize logging
    env_logger::init();

    // Create specialized agents
    let customer_service_agent = Arc::new(create_agent(
        "gpt-4o-mini",
        &[],
        Some("You are a customer service agent. Help customers with their inquiries."),
        None,
    )?);

    let technical_support_agent = Arc::new(create_agent(
        "gpt-4o-mini",
        &[],
        Some("You are a technical support agent. Help with technical issues and troubleshooting."),
        None,
    )?);

    // Create handoff system
    let handoff_system = HandoffAgentBuilder::new()
        .with_base_agent(customer_service_agent.clone())
        .with_handoff_agent(
            "technical_support".to_string(),
            technical_support_agent.clone(),
        )
        .build()?;

    println!("Testing Handoffs pattern...\n");

    // Test: Customer service can hand off to technical support
    println!("Question: I'm having trouble connecting to the database");
    let response = handoff_system
        .invoke_messages(vec![Message::new_human_message(
            "I'm having trouble connecting to the database",
        )])
        .await?;
    println!("Response: {}\n", response);

    Ok(())
}