run-cli 0.0.53

A CLI to help you run any command
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

use crate::executor::Processor;
use async_trait::async_trait;
use serde::Deserialize;
use serde::Serialize;
use serde_json::json;

pub struct Openai {
    lines: Vec<String>,
    api_key: String,
    api_base_url: String,
}

impl Openai {
    pub fn new(api_base_url: String, api_key: String) -> Self {
        Self {
            lines: vec![],
            api_base_url,
            api_key,
        }
    }
}

#[async_trait]
impl Processor for Openai {
    fn process(&mut self, input: String) -> anyhow::Result<String> {
        self.lines.push(input.clone());
        Ok(input)
    }

    async fn flush(&mut self) -> anyhow::Result<()> {
        eprintln!(
            "\n+=============================[ ChatGPT Feedback ]=============================+"
        );

        // TODO: include more information in the prompt (cmd, args, os, arch, current directory, $PATH, etc)
        let body = json!({
          "model": "gpt-3.5-turbo",
          "messages": [
              {
                  "role": "user",
                  "content": format!("I am a developer working in a terminal. The command I run fails with the following error message. In a first paragraph explain what the issue is and why it is happening. In a second paragraph explain how to fix the issue. You can suggest at most one command that might help fixing the issue. Use the triple backtick notation if you need to print code.\n\n{}", self.lines.join("\n")),
              },
          ]
        });

        let url = format!("{}/v1/chat/completions", self.api_base_url);
        let resp = reqwest::Client::new()
            .post(&url)
            .bearer_auth(&self.api_key)
            .json(&body)
            .send()
            .await?;
        let json: Response = resp.json().await?;

        let Some(choice) = json.choices.get(0) else {
            anyhow::bail!("No choice returned by OpenAI");
        };

        eprintln!(
            "|                                                                              |"
        );
        for line in textwrap::wrap(choice.message.content.trim(), 76) {
            eprintln!("| {line:<76} |");
        }
        eprintln!(
            "|                                                                              |"
        );
        eprintln!(
            "+==============================================================================+\n"
        );

        Ok(())
    }
}

#[derive(Default, Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
struct Response {
    pub id: String,
    pub object: String,
    pub created: i64,
    pub model: String,
    pub usage: Usage,
    pub choices: Vec<Choice>,
}

#[derive(Default, Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
struct Usage {
    #[serde(rename = "prompt_tokens")]
    pub prompt_tokens: i64,
    #[serde(rename = "completion_tokens")]
    pub completion_tokens: i64,
    #[serde(rename = "total_tokens")]
    pub total_tokens: i64,
}

#[derive(Default, Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
struct Choice {
    pub message: Message,
    #[serde(rename = "finish_reason")]
    pub finish_reason: String,
    pub index: i64,
}

#[derive(Default, Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
struct Message {
    pub role: String,
    pub content: String,
}