0xMRTT/baichat-rs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: add parent message id

Browse Source
This commit is contained in:
0xMRTT 2023-06-17 17:08:59 +02:00
parent 80eabb9f9f
commit bf5fb5df49
Signed by: 0xMRTT
GPG Key ID: 910B287304120902
1 changed files with 67 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
src/lib.rs
View File

@ -34,43 +34,75 @@ pub struct DeltaChoice {
pub content: String, pub content: String,
} }
pub async fn ask(prompt: &str) -> Result<Vec<Delta>, Box<dyn std::error::Error>> { struct ThebAI {
pub parent_message_id: Option<String>,
}
let mut body = String::new(); impl ThebAI {
body.push_str(r#"{ pub fn new(parent_message_id: Option<&str>) -> ThebAI {
"prompt": ""#); if let Some(parent_message_id) = parent_message_id {
body.push_str(&prompt); return ThebAI {
body.push_str(r#"", parent_message_id: Some(parent_message_id.to_string()),
"options": { }
"parentMessageId": "8c00bd29-75b0-42c7-9d4f-05a94ac8b2de" } else {
return ThebAI {
parent_message_id: String::from("8c00bd29-75b0-42c7-9d4f-05a94ac8b2de"),
}
} }
}"#);
let mut request = Request::builder()
.method("POST")
.uri("https://chatbot.theb.ai/api/chat-process")
.header("User-Agent", "Mozilla/5.0 (X11; Linux x86_64; rv:109.0) Gecko/20100101 Firefox/112.0")
.header("Accept-Language", "en-US,en;q=0.5")
.header("Content-Type", "application/json")
.header("Host", "chatbot.theb.ai")
.header("Referer", "https://chatbot.theb.ai")
.header("Origin", "https://chatbot.theb.ai")
.body(body)?
.send()?;
let result = request.text()?;
let mut deltas: Vec<Delta> = Vec::new();
for line in result.lines() {
if line == "" {
continue;
}
println!("{}", line);
println!("------------------");
let delta: Delta = serde_json::from_str(line).unwrap();
deltas.push(delta);
} }
println!("{:?}", deltas); pub async fn ask(&self, prompt: &str, parent_message_id: Option<String>) -> Result<Vec<Delta>, Box<dyn std::error::Error>> {
return Ok(deltas); let mut body = String::new();
body.push_str(r#"{
"prompt": ""#);
body.push_str(&prompt);
if let Some(parent_message_id) = parent_message_id {
body.push_str(r#"",
"options": {
"parentMessageId": ""#);
body.push_str(parent_message_id.as_str());
body.push_str(r#""
}
}"#);
} else {
body.push_str(r#"",
"options": {
"parentMessageId": ""#);
body.push_str(self.parent_message_id.as_ref().unwrap().as_str());
body.push_str(r#""
}
}"#);
}
let mut request = Request::builder()
.method("POST")
.uri("https://chatbot.theb.ai/api/chat-process")
.header("User-Agent", "Mozilla/5.0 (X11; Linux x86_64; rv:109.0) Gecko/20100101 Firefox/112.0")
.header("Accept-Language", "en-US,en;q=0.5")
.header("Content-Type", "application/json")
.header("Host", "chatbot.theb.ai")
.header("Referer", "https://chatbot.theb.ai")
.header("Origin", "https://chatbot.theb.ai")
.body(body)?
.send()?;
let result = request.text()?;
let mut deltas: Vec<Delta> = Vec::new();
for line in result.lines() {
if line == "" {
continue;
}
let delta: Delta = serde_json::from_str(line).unwrap();
deltas.push(delta);
}
self.parent_message_id = Some(deltas.last().unwrap().id.clone());
println!("{:?}", deltas);
return Ok(deltas);
}
} }