diff options
| author | sigoden <sigoden@gmail.com> | 2024-05-23 19:28:56 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-23 19:28:56 +0800 |
| commit | 5458150ed3203cf13b0371efa2c791ac696cee93 (patch) | |
| tree | 702d17039d9677246a3b91deb8243e09cd768402 /src/client/cloudflare.rs | |
| parent | 2ccbb0f06a4558e15642feb53ba7b2bd72804820 (diff) | |
| download | aichat-5458150ed3203cf13b0371efa2c791ac696cee93.tar.gz | |
fix: json stream parser and refine client modules (#538)
Diffstat (limited to 'src/client/cloudflare.rs')
| -rw-r--r-- | src/client/cloudflare.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/client/cloudflare.rs b/src/client/cloudflare.rs index 3369266..659c7c0 100644 --- a/src/client/cloudflare.rs +++ b/src/client/cloudflare.rs @@ -1,6 +1,6 @@ use super::{ catch_error, sse_stream, Client, CloudflareClient, CompletionOutput, ExtraConfig, Model, - ModelData, ModelPatches, PromptAction, PromptKind, SendData, SsMmessage, SseHandler, + ModelData, ModelPatches, PromptAction, PromptKind, SendData, SseHandler, SseMmessage, }; use anyhow::{anyhow, Result}; @@ -65,7 +65,7 @@ async fn send_message(builder: RequestBuilder) -> Result<CompletionOutput> { } async fn send_message_streaming(builder: RequestBuilder, handler: &mut SseHandler) -> Result<()> { - let handle = |message: SsMmessage| -> Result<bool> { + let handle = |message: SseMmessage| -> Result<bool> { if message.data == "[DONE]" { return Ok(true); } |
