gpto 0.2.2

A tiny unofficial OpenAI client
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
50
51
52
53
54
55
56
57

use std::io;

use serde_json::json;

use crate::config;
use crate::request;
use crate::Cli;

/// Get completions from input prompt
pub fn completions(cli: Cli, text: &Option<String>, stdin: &bool) -> Result<String, String> {
    let text = string_or_stdin(text, stdin)?;
    let config = config::get_or_create(cli.config)?;

    // Config or CLI
    let model = cli.model.unwrap_or(config.model);
    let timeout = cli.timeout.unwrap_or(config.timeout);
    let endpoint = cli.endpoint.unwrap_or(config.endpoint);

    // Config only
    let token = config.token.clone();

    // CLI only
    let temperature = cli.temperature;
    let max_tokens = cli.max_tokens;
    let top_p = cli.top_p;
    let disable_spinner = cli.disable_spinner;
    let number = cli.number;
    let suffix = cli.suffix;

    let body = json!({
        "model": model, 
        "max_tokens": max_tokens,
        "messages": [{"role": "user", "content": text}],
        "n": number, 
        "temperature": temperature, 
        "top_p": top_p});

    request::completions(endpoint, body, token, disable_spinner, timeout, suffix)
}

fn string_or_stdin(text: &Option<String>, stdin: &bool) -> Result<String, String> {
    let text = text.clone().unwrap_or_default();

    let prompt = if *stdin {
        let mut buffer = String::new();
        io::stdin().read_line(&mut buffer).unwrap();
        format!("{text}\n{buffer}")
    } else {
        text
    };

    if prompt.is_empty() {
        Err(String::from("No prompt provided"))
    } else {
        Ok(prompt)
    }
}