mii/openai-api-rs
1
0
Fork 0
mirror of https://github.com/mii443/openai-api-rs.git synced 2025-08-23 07:35:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

* use reqwest

Browse Source 
* support multipart form
* fix audio apits
This commit is contained in:
Dongri Jin
2024-07-09 18:47:51 +09:00
parent be660448f4
commit 3e5286dad6
18 changed files with 489 additions and 675 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
examples/assistant.rs
View File

@ -1,4 +1,4 @@
use openai_api_rs::v1::api::Client;
use openai_api_rs::v1::api::OpenAIClient;
use openai_api_rs::v1::assistant::AssistantRequest;
use openai_api_rs::v1::common::GPT4_O;
use openai_api_rs::v1::message::{CreateMessageRequest, MessageRole};
@ -7,8 +7,9 @@ use openai_api_rs::v1::thread::CreateThreadRequest;
use std::collections::HashMap;
use std::env;
fn main() -> Result<(), Box<dyn std::error::Error>> {
let client = Client::new(env::var("OPENAI_API_KEY").unwrap().to_string());
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
let client = OpenAIClient::new(env::var("OPENAI_API_KEY").unwrap().to_string());
let mut tools = HashMap::new();
tools.insert("type".to_string(), "code_interpreter".to_string());
@ -21,11 +22,11 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
let req = req.clone().tools(vec![tools]);
println!("AssistantRequest: {:?}", req);
let result = client.create_assistant(req)?;
let result = client.create_assistant(req).await?;
println!("Create Assistant Result ID: {:?}", result.id);
let thread_req = CreateThreadRequest::new();
let thread_result = client.create_thread(thread_req)?;
let thread_result = client.create_thread(thread_req).await?;
println!("Create Thread Result ID: {:?}", thread_result.id.clone());
let message_req = CreateMessageRequest::new(
@ -33,16 +34,19 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
"`I need to solve the equation 3x + 11 = 14. Can you help me?".to_string(),
);
let message_result = client.create_message(thread_result.id.clone(), message_req)?;
let message_result = client
.create_message(thread_result.id.clone(), message_req)
.await?;
println!("Create Message Result ID: {:?}", message_result.id.clone());
let run_req = CreateRunRequest::new(result.id);
let run_result = client.create_run(thread_result.id.clone(), run_req)?;
let run_result = client.create_run(thread_result.id.clone(), run_req).await?;
println!("Create Run Result ID: {:?}", run_result.id.clone());
loop {
let run_result = client
.retrieve_run(thread_result.id.clone(), run_result.id.clone())
.await
.unwrap();
if run_result.status == "completed" {
break;
@ -52,7 +56,10 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
}
}
let list_message_result = client.list_messages(thread_result.id.clone()).unwrap();
let list_message_result = client
.list_messages(thread_result.id.clone())
.await
.unwrap();
for data in list_message_result.data {
for content in data.content {
println!(