assistant/
assistant.rs

1use openai_api_rs::v1::api::OpenAIClient;
2use openai_api_rs::v1::assistant::AssistantRequest;
3use openai_api_rs::v1::common::GPT4_O;
4use openai_api_rs::v1::message::{CreateMessageRequest, MessageRole};
5use openai_api_rs::v1::run::CreateRunRequest;
6use openai_api_rs::v1::thread::CreateThreadRequest;
7use std::collections::HashMap;
8use std::env;
9
10#[tokio::main]
11async fn main() -> Result<(), Box<dyn std::error::Error>> {
12    let api_key = env::var("OPENAI_API_KEY").unwrap().to_string();
13    let mut client = OpenAIClient::builder().with_api_key(api_key).build()?;
14
15    let mut tools = HashMap::new();
16    tools.insert("type".to_string(), "code_interpreter".to_string());
17
18    let req = AssistantRequest::new(GPT4_O.to_string());
19    let req = req
20        .clone()
21        .description("this is a test assistant".to_string());
22    let req = req.clone().instructions("You are a personal math tutor. When asked a question, write and run Python code to answer the question.".to_string());
23    let req = req.clone().tools(vec![tools]);
24    println!("AssistantRequest: {:?}", req);
25
26    let result = client.create_assistant(req).await?;
27    println!("Create Assistant Result ID: {:?}", result.id);
28
29    let thread_req = CreateThreadRequest::new();
30    let thread_result = client.create_thread(thread_req).await?;
31    println!("Create Thread Result ID: {:?}", thread_result.id.clone());
32
33    let message_req = CreateMessageRequest::new(
34        MessageRole::user,
35        "`I need to solve the equation 3x + 11 = 14. Can you help me?".to_string(),
36    );
37
38    let message_result = client
39        .create_message(thread_result.id.clone(), message_req)
40        .await?;
41    println!("Create Message Result ID: {:?}", message_result.id.clone());
42
43    let run_req = CreateRunRequest::new(result.id);
44    let run_result = client.create_run(thread_result.id.clone(), run_req).await?;
45    println!("Create Run Result ID: {:?}", run_result.id.clone());
46
47    loop {
48        let run_result = client
49            .retrieve_run(thread_result.id.clone(), run_result.id.clone())
50            .await
51            .unwrap();
52        if run_result.status == "completed" {
53            break;
54        } else {
55            println!("waiting...");
56            std::thread::sleep(std::time::Duration::from_secs(1));
57        }
58    }
59
60    let list_message_result = client
61        .list_messages(thread_result.id.clone())
62        .await
63        .unwrap();
64    for data in list_message_result.data {
65        for content in data.content {
66            println!(
67                "{:?}: {:?} {:?}",
68                data.role, content.text.value, content.text.annotations
69            );
70        }
71    }
72
73    Ok(())
74}
75
76// OPENAI_API_KEY=xxxx cargo run --package openai-api-rs --example assistant
assistant/assistant.rs

assistant/
assistant.rs
