technofab/pay-respects
1
0
Fork 0
mirror of https://github.com/TECHNOFAB11/pay-respects.git synced 2025-12-12 14:30:10 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

deps: use libcurl

Browse source
This commit is contained in:
iff 2024-11-22 10:49:24 +01:00
parent 98a00674f3
commit c97d6440f3
4 changed files with 152 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/main.rs
View file

@ -63,7 +63,7 @@ fn main() {
}
};
let mut last_command = shell::last_command(&shell);
let mut last_command = shell::last_command(&shell).trim().to_string();
last_command = shell::expand_alias(&shell, &last_command);
let mut error_msg = command_output(&shell, &last_command);
error_msg = error_msg

65
src/requests.rs
View file

@ -1,8 +1,12 @@
use std::collections::HashMap;
use std::io::Read;
use serde::{Deserialize, Serialize};
use serde_json::Value;
use curl::easy::Easy as Curl;
use curl::easy::List;
#[derive(Serialize, Deserialize)]
struct Input {
role: String,
@ -97,32 +101,43 @@ The command `{last_command}` returns the following error message: `{error_msg}`.
model,
};
let res = std::process::Command::new("curl")
.arg("-X")
.arg("POST")
.arg("-H")
.arg(format!("Authorization: Bearer {}", api_key))
.arg("-H")
.arg("Content-Type: application/json")
.arg("-d")
.arg(serde_json::to_string(&messages).unwrap())
.arg(request_url)
.output();
let str_json = serde_json::to_string(&messages).unwrap();
let mut data = str_json.as_bytes();
let res = match res {
Ok(res) => res.stdout,
Err(_) => {
return None;
}
};
let mut dst = Vec::new();
let mut handle = Curl::new();
let json: Value = {
let json = serde_json::from_str(std::str::from_utf8(&res).unwrap());
if json.is_err() {
return None;
}
json.unwrap()
};
handle.url(&request_url).unwrap();
handle.post(true).unwrap();
handle.post_field_size(data.len() as u64).unwrap();
let mut headers = List::new();
headers
.append(&format!("Authorization: Bearer {}", api_key))
.unwrap();
headers.append("Content-Type: application/json").unwrap();
handle.http_headers(headers).unwrap();
{
let mut transfer = handle.transfer();
transfer
.read_function(|buf| Ok(data.read(buf).unwrap_or(0)))
.unwrap();
transfer
.write_function(|buf| {
dst.extend_from_slice(buf);
Ok(buf.len())
})
.unwrap();
transfer.perform().expect("Failed to perform request");
}
let res = String::from_utf8(dst).unwrap();
let json: Value = serde_json::from_str(&res).unwrap();
let content = &json["choices"][0]["message"]["content"];
@ -130,7 +145,7 @@ The command `{last_command}` returns the following error message: `{error_msg}`.
let str = {
let str = content.as_str();
str?;
str.unwrap()
str.expect("Failed to get content from response")
.trim_start_matches("```")
.trim_end_matches("```")
};