auto-rust 0.1.5

A tool to automatically generate Rust code using LLMs at compilation time.
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

use reqwest::header;
use serde_json::json;
use std::env;

use crate::data::ChatCompletionResponse;

pub fn generic_chat_completion(
    system_message: String,
    user_message: String,
) -> Result<ChatCompletionResponse, Box<dyn std::error::Error>> {
    let open_ai_key = env::var("OPENAI_API_KEY").unwrap();
    let model_name = env::var("OPENAI_MODEL_NAME").unwrap_or("gpt-3.5-turbo".to_string());

    let mut headers = header::HeaderMap::new();

    headers.insert("Content-Type", "application/json".parse().unwrap());
    headers.insert(
        "Authorization",
        format!("Bearer {}", open_ai_key).parse().unwrap(),
    );

    let client = reqwest::blocking::Client::builder()
        .redirect(reqwest::redirect::Policy::none())
        .build()
        .unwrap();

    let body = json!({
        "model": model_name,
        "messages": [
            {"role": "system", "content": system_message},
            {"role": "user", "content": user_message}
        ]
    });

    let res = client
        .post("https://openrouter.ai/api/v1/chat/completions")
        .headers(headers)
        .body(body.to_string())
        .send()?
        .json::<ChatCompletionResponse>()?;

    Ok(res)
}