diff options
| author | sigoden <sigoden@gmail.com> | 2024-10-29 14:15:02 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-10-29 14:15:02 +0800 |
| commit | bf07cdf256e1e3008578972f5916a4d6763554f5 (patch) | |
| tree | 3f48ead051403a7f270198578e86a92672bb4208 /src/client/stream.rs | |
| parent | bb542f6e92fd782b70c44fddb21d013c289d0a3c (diff) | |
| download | aichat-bf07cdf256e1e3008578972f5916a4d6763554f5.tar.gz | |
refactor: improve code quality (#956)
Diffstat (limited to 'src/client/stream.rs')
| -rw-r--r-- | src/client/stream.rs | 12 |
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] { |
