From 4d4a100fe6fcde7ca3afcf7971ac387d355349eb Mon Sep 17 00:00:00 2001 From: sigoden Date: Mon, 29 Apr 2024 23:04:25 +0800 Subject: refactor: sse handling (#465) --- src/client/cloudflare.rs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/client/cloudflare.rs') diff --git a/src/client/cloudflare.rs b/src/client/cloudflare.rs index bb638fb..80cde81 100644 --- a/src/client/cloudflare.rs +++ b/src/client/cloudflare.rs @@ -1,6 +1,6 @@ use super::{ catch_error, sse_stream, CloudflareClient, CompletionDetails, ExtraConfig, Model, ModelConfig, - PromptType, SendData, SseHandler, + PromptType, SendData, SsMmessage, SseHandler, }; use crate::utils::PromptKind; @@ -64,11 +64,11 @@ async fn send_message(builder: RequestBuilder) -> Result<(String, CompletionDeta } async fn send_message_streaming(builder: RequestBuilder, handler: &mut SseHandler) -> Result<()> { - let handle = |data: &str| -> Result { - if data == "[DONE]" { + let handle = |message: SsMmessage| -> Result { + if message.data == "[DONE]" { return Ok(true); } - let data: Value = serde_json::from_str(data)?; + let data: Value = serde_json::from_str(&message.data)?; if let Some(text) = data["response"].as_str() { handler.text(text)?; } -- cgit v1.2.3