openai-rusty 0.3.0

A Rust wrapper for the OpenAI API.
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

use reqwest::Client;
use serde_json::json;
use crate::client::get_api_key;

#[derive(Debug)]
pub enum OpenAIError {
    MissingApiKey,
    RequestError(reqwest::Error),
    InvalidResponse,
}

impl From<reqwest::Error> for OpenAIError {
    fn from(err: reqwest::Error) -> Self {
        OpenAIError::RequestError(err)
    }
}

/// Sends a prompt using a specified model (e.g., "gpt-3.5-turbo").
pub async fn openai_chat_with_model(prompt: &str, model: &str) -> Result<String, OpenAIError> {
    let api_key = get_api_key().ok_or(OpenAIError::MissingApiKey)?;

    let client = Client::new();
    let response = client
        .post("https://api.openai.com/v1/chat/completions")
        .bearer_auth(api_key)
        .json(&json!({
            "model": model,
            "messages": [
                {
                    "role": "user",
                    "content": prompt
                }
            ]
        }))
        .send()
        .await?;

    let json: serde_json::Value = response.json().await?;
    if let Some(content) = json["choices"][0]["message"]["content"].as_str() {
        Ok(content.to_string())
    } else {
        Err(OpenAIError::InvalidResponse)
    }
}

/// Sends a prompt using the default model "gpt-3.5-turbo".
pub async fn openai_chat(prompt: &str) -> Result<String, OpenAIError> {
    openai_chat_with_model(prompt, "gpt-3.5-turbo").await
}