mirror of
https://github.com/TECHNOFAB11/pay-respects.git
synced 2025-12-12 06:20:09 +01:00
fix: always write inside thinking
This commit is contained in:
iff
parent
5d210c38e6
commit
005a3b6661
1 changed files with 127 additions and 77 deletions
|
|
@ -26,7 +26,8 @@ use colored::Colorize;
|
||||||
#[derive(PartialEq)]
|
#[derive(PartialEq)]
|
||||||
enum State {
|
enum State {
|
||||||
Write,
|
Write,
|
||||||
Store,
|
Think,
|
||||||
|
Buf,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub struct Buffer {
|
pub struct Buffer {
|
||||||
|
|
@ -42,88 +43,17 @@ impl Buffer {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
pub fn proc(&mut self, data: &str) {
|
pub fn proc(&mut self, data: &str) {
|
||||||
if self.state == State::Write {
|
match self.state {
|
||||||
if !data.contains("\n") {
|
State::Write => self.proc_write(data),
|
||||||
self.buf.push(data.to_string());
|
State::Think => self.proc_think(data),
|
||||||
let buffered = self.buf.join("").trim().to_string();
|
State::Buf => self.buf.push(data.to_string()),
|
||||||
let filled = fill(&buffered);
|
|
||||||
if let Some(filled) = filled {
|
|
||||||
self.buf.clear();
|
|
||||||
let formatted = format!("\r{}", filled);
|
|
||||||
eprint!("{}", formatted);
|
|
||||||
self.buf
|
|
||||||
.push(formatted.split_once("\n").unwrap().1.to_string());
|
|
||||||
std::io::stdout().flush().unwrap();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
eprint!("{}", data);
|
|
||||||
std::io::stdout().flush().unwrap();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut data = data.to_string();
|
|
||||||
while data.contains("\n") {
|
|
||||||
let lines = data.split_once("\n").unwrap();
|
|
||||||
let first = lines.0;
|
|
||||||
let last = lines.1;
|
|
||||||
self.buf.push(first.to_string());
|
|
||||||
let buffered = self.buf.join("").trim().to_string();
|
|
||||||
self.buf.clear();
|
|
||||||
if buffered.ends_with("<note>") {
|
|
||||||
let warn = format!("\r{}:", t!("ai-suggestion"))
|
|
||||||
.bold()
|
|
||||||
.blue()
|
|
||||||
.to_string();
|
|
||||||
let first = buffered.replace("<note>", &warn);
|
|
||||||
eprintln!("{}", first);
|
|
||||||
std::io::stdout().flush().unwrap();
|
|
||||||
} else if buffered.ends_with("</note>") {
|
|
||||||
let tag = "</note>";
|
|
||||||
let whitespace = " ".repeat(tag.len());
|
|
||||||
let formatted = format!("\r{}", whitespace);
|
|
||||||
let first = buffered.replace("</note>", &formatted);
|
|
||||||
eprintln!("{}", first);
|
|
||||||
self.state = State::Store;
|
|
||||||
std::io::stdout().flush().unwrap();
|
|
||||||
} else if buffered.ends_with("<think>") {
|
|
||||||
let tag = "<think>";
|
|
||||||
let warn = format!("\r{}:", t!("ai-thinking"))
|
|
||||||
.bold()
|
|
||||||
.blue()
|
|
||||||
.to_string();
|
|
||||||
let first = buffered.replace(tag, &warn);
|
|
||||||
eprintln!("{}", first);
|
|
||||||
std::io::stdout().flush().unwrap();
|
|
||||||
} else if buffered.ends_with("</think>") {
|
|
||||||
let tag = "</think>";
|
|
||||||
let whitespace = " ".repeat(tag.len());
|
|
||||||
let formatted = format!("\r{}", whitespace);
|
|
||||||
let first = buffered.replace(tag, &formatted);
|
|
||||||
eprintln!("{}", first);
|
|
||||||
std::io::stdout().flush().unwrap();
|
|
||||||
} else if buffered.ends_with("```") {
|
|
||||||
let tag = "```";
|
|
||||||
let whitespace = " ".repeat(tag.len());
|
|
||||||
let formatted = format!("\r{}", whitespace);
|
|
||||||
let first = buffered.replace(tag, &formatted);
|
|
||||||
eprintln!("{}", first);
|
|
||||||
std::io::stdout().flush().unwrap();
|
|
||||||
} else {
|
|
||||||
eprintln!("{}", first);
|
|
||||||
std::io::stdout().flush().unwrap();
|
|
||||||
}
|
|
||||||
data = last.to_string();
|
|
||||||
}
|
|
||||||
eprint!("{}", data);
|
|
||||||
return;
|
|
||||||
}
|
}
|
||||||
self.buf.push(data.to_string());
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn print_return_remain(&mut self) -> String {
|
pub fn print_return_remain(&mut self) -> String {
|
||||||
let buffered = self.buf.join("").trim().to_string();
|
let buffered = self.buf.join("").trim().to_string();
|
||||||
self.buf.clear();
|
self.buf.clear();
|
||||||
if self.state == State::Store {
|
if self.state == State::Buf {
|
||||||
return buffered;
|
return buffered;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -135,4 +65,124 @@ impl Buffer {
|
||||||
}
|
}
|
||||||
"".to_string()
|
"".to_string()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn proc_write(&mut self, data: &str) {
|
||||||
|
if !data.contains("\n") {
|
||||||
|
self.buf.push(data.to_string());
|
||||||
|
let buffered = self.buf.join("").trim().to_string();
|
||||||
|
let filled = fill(&buffered);
|
||||||
|
if let Some(filled) = filled {
|
||||||
|
self.buf.clear();
|
||||||
|
let formatted = format!("\r{}", filled);
|
||||||
|
eprint!("{}", formatted);
|
||||||
|
self.buf
|
||||||
|
.push(formatted.split_once("\n").unwrap().1.to_string());
|
||||||
|
std::io::stdout().flush().unwrap();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
eprint!("{}", data);
|
||||||
|
std::io::stdout().flush().unwrap();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut data = data.to_string();
|
||||||
|
while data.contains("\n") {
|
||||||
|
let lines = data.split_once("\n").unwrap();
|
||||||
|
let first = lines.0;
|
||||||
|
let last = lines.1;
|
||||||
|
self.buf.push(first.to_string());
|
||||||
|
let buffered = self.buf.join("").trim().to_string();
|
||||||
|
self.buf.clear();
|
||||||
|
if buffered.ends_with("<note>") {
|
||||||
|
let warn = format!("\r{}:", t!("ai-suggestion"))
|
||||||
|
.bold()
|
||||||
|
.blue()
|
||||||
|
.to_string();
|
||||||
|
let first = buffered.replace("<note>", &warn);
|
||||||
|
eprintln!("{}", first);
|
||||||
|
std::io::stdout().flush().unwrap();
|
||||||
|
} else if buffered.ends_with("</note>") {
|
||||||
|
let tag = "</note>";
|
||||||
|
let whitespace = " ".repeat(tag.len());
|
||||||
|
let formatted = format!("\r{}", whitespace);
|
||||||
|
let first = buffered.replace("</note>", &formatted);
|
||||||
|
eprintln!("{}", first);
|
||||||
|
self.state = State::Buf;
|
||||||
|
std::io::stdout().flush().unwrap();
|
||||||
|
} else if buffered.ends_with("<think>") {
|
||||||
|
let tag = "<think>";
|
||||||
|
let warn = format!("\r{}:", t!("ai-thinking"))
|
||||||
|
.bold()
|
||||||
|
.blue()
|
||||||
|
.to_string();
|
||||||
|
let first = buffered.replace(tag, &warn);
|
||||||
|
self.state = State::Think;
|
||||||
|
eprintln!("{}", first);
|
||||||
|
std::io::stdout().flush().unwrap();
|
||||||
|
} else if buffered.ends_with("</think>") {
|
||||||
|
let tag = "</think>";
|
||||||
|
let whitespace = " ".repeat(tag.len());
|
||||||
|
let formatted = format!("\r{}", whitespace);
|
||||||
|
let first = buffered.replace(tag, &formatted);
|
||||||
|
eprintln!("{}", first);
|
||||||
|
std::io::stdout().flush().unwrap();
|
||||||
|
} else if buffered.ends_with("```") {
|
||||||
|
let tag = "```";
|
||||||
|
let whitespace = " ".repeat(tag.len());
|
||||||
|
let formatted = format!("\r{}", whitespace);
|
||||||
|
let first = buffered.replace(tag, &formatted);
|
||||||
|
eprintln!("{}", first);
|
||||||
|
std::io::stdout().flush().unwrap();
|
||||||
|
} else {
|
||||||
|
eprintln!("{}", first);
|
||||||
|
std::io::stdout().flush().unwrap();
|
||||||
|
}
|
||||||
|
data = last.to_string();
|
||||||
|
}
|
||||||
|
eprint!("{}", data);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn proc_think(&mut self, data: &str) {
|
||||||
|
if !data.contains("\n") {
|
||||||
|
self.buf.push(data.to_string());
|
||||||
|
let buffered = self.buf.join("").trim().to_string();
|
||||||
|
let filled = fill(&buffered);
|
||||||
|
if let Some(filled) = filled {
|
||||||
|
self.buf.clear();
|
||||||
|
let formatted = format!("\r{}", filled);
|
||||||
|
eprint!("{}", formatted);
|
||||||
|
self.buf
|
||||||
|
.push(formatted.split_once("\n").unwrap().1.to_string());
|
||||||
|
std::io::stdout().flush().unwrap();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
eprint!("{}", data);
|
||||||
|
std::io::stdout().flush().unwrap();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut data = data.to_string();
|
||||||
|
while data.contains("\n") {
|
||||||
|
let lines = data.split_once("\n").unwrap();
|
||||||
|
let first = lines.0;
|
||||||
|
let last = lines.1;
|
||||||
|
self.buf.push(first.to_string());
|
||||||
|
let buffered = self.buf.join("").trim().to_string();
|
||||||
|
self.buf.clear();
|
||||||
|
if buffered.ends_with("</think>") {
|
||||||
|
let tag = "</think>";
|
||||||
|
let whitespace = " ".repeat(tag.len());
|
||||||
|
let formatted = format!("\r{}", whitespace);
|
||||||
|
let first = buffered.replace(tag, &formatted);
|
||||||
|
self.state = State::Write;
|
||||||
|
eprintln!("{}", first);
|
||||||
|
std::io::stdout().flush().unwrap();
|
||||||
|
} else {
|
||||||
|
eprintln!("{}", first);
|
||||||
|
std::io::stdout().flush().unwrap();
|
||||||
|
}
|
||||||
|
data = last.to_string();
|
||||||
|
}
|
||||||
|
eprint!("{}", data);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue