rig-openapi-tools 0.1.7

Turn any OpenAPI spec into LLM-callable tools for rig
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

use rig::client::{CompletionClient, ProviderClient};
use rig::completion::Prompt;
use rig_openapi_tools::OpenApiToolset;

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let openai = rig::providers::openai::Client::from_env();

    let toolset = OpenApiToolset::from_file("examples/openapi.yaml")?;
    println!("Loaded {} tools from OpenAPI spec", toolset.len());

    let agent = openai
        .agent("gpt-4o")
        .preamble("You have access to API tools. Use them when asked.")
        .tools(toolset.into_tools())
        .build();

    let response: String = agent
        .prompt("Use the API tool to get user 1 and summarize the result.")
        .await?;

    println!("{response}");

    Ok(())
}