groq-rust 0.1.0

SDK for Groq in Rust
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

pub mod groq {
    use reqwest::blocking::Client;

    pub struct Groq {
        api_key: String,
        client: Client,
    }

    impl Groq {
        pub fn new(api_key: String) -> Self {
            Self {
                api_key,
                client: Client::new(),
            }
        }

        pub fn test_request(&self) -> String {
            let response = self.client
                .post("https://api.groq.com/openai/v1/chat/completions")
                .header("Authorization", format!("Bearer {}", self.api_key))
                .header("Content-Type", "application/json")
                .body("{\"messages\": [{\"role\": \"user\", \"content\": \"Explain the importance of fast language models\"}], \"model\": \"mixtral-8x7b-32768\"}")
                .send().unwrap();

            response.text().unwrap()
        }
    }
}

#[cfg(test)]
mod tests {
    use std::env;

    use crate::groq::Groq;

    #[test]
    fn basic_test() {
        let api_key = env::var("GROQ_API_KEY").unwrap();
        let groq = Groq::new(api_key);

        println!("{}", groq.test_request())
    }
}