summaryrefslogtreecommitdiffstats
path: root/src/client/stream.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-10-29 14:15:02 +0800
committerGitHub <noreply@github.com>2024-10-29 14:15:02 +0800
commitbf07cdf256e1e3008578972f5916a4d6763554f5 (patch)
tree3f48ead051403a7f270198578e86a92672bb4208 /src/client/stream.rs
parentbb542f6e92fd782b70c44fddb21d013c289d0a3c (diff)
downloadaichat-bf07cdf256e1e3008578972f5916a4d6763554f5.tar.gz
refactor: improve code quality (#956)
Diffstat (limited to 'src/client/stream.rs')
-rw-r--r--src/client/stream.rs12
1 files changed, 6 insertions, 6 deletions
diff --git a/src/client/stream.rs b/src/client/stream.rs
index 735ec93..e89e7b9 100644
--- a/src/client/stream.rs
+++ b/src/client/stream.rs
@@ -10,16 +10,16 @@ use tokio::sync::mpsc::UnboundedSender;
pub struct SseHandler {
sender: UnboundedSender<SseEvent>,
- abort: AbortSignal,
+ abort_signal: AbortSignal,
buffer: String,
tool_calls: Vec<ToolCall>,
}
impl SseHandler {
- pub fn new(sender: UnboundedSender<SseEvent>, abort: AbortSignal) -> Self {
+ pub fn new(sender: UnboundedSender<SseEvent>, abort_signal: AbortSignal) -> Self {
Self {
sender,
- abort,
+ abort_signal,
buffer: String::new(),
tool_calls: Vec::new(),
}
@@ -36,7 +36,7 @@ impl SseHandler {
.send(SseEvent::Text(text.to_string()))
.with_context(|| "Failed to send SseEvent:Text");
if let Err(err) = ret {
- if self.abort.aborted() {
+ if self.abort_signal.aborted() {
return Ok(());
}
return Err(err);
@@ -48,7 +48,7 @@ impl SseHandler {
// debug!("HandleDone");
let ret = self.sender.send(SseEvent::Done);
if ret.is_err() {
- if self.abort.aborted() {
+ if self.abort_signal.aborted() {
return;
}
warn!("Failed to send SseEvent:Done");
@@ -62,7 +62,7 @@ impl SseHandler {
}
pub fn abort(&self) -> AbortSignal {
- self.abort.clone()
+ self.abort_signal.clone()
}
pub fn tool_calls(&self) -> &[ToolCall] {