From 4d4a100fe6fcde7ca3afcf7971ac387d355349eb Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 29 Apr 2024 23:04:25 +0800
Subject: refactor: sse handling (#465)

---
 src/client/qianwen.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'src/client/qianwen.rs')

diff --git a/src/client/qianwen.rs b/src/client/qianwen.rs
index c16aeaa..1f42fd9 100644
--- a/src/client/qianwen.rs
+++ b/src/client/qianwen.rs
@@ -1,6 +1,6 @@
 use super::{
     maybe_catch_error, message::*, sse_stream, Client, CompletionDetails, ExtraConfig, Model,
-    ModelConfig, PromptType, QianwenClient, SendData, SseHandler,
+    ModelConfig, PromptType, QianwenClient, SendData, SsMmessage, SseHandler,
 };
 
 use crate::utils::{sha256sum, PromptKind};
@@ -107,8 +107,8 @@ async fn send_message_streaming(
     handler: &mut SseHandler,
     is_vl: bool,
 ) -> Result<()> {
-    let handle = |data: &str| -> Result<bool> {
-        let data: Value = serde_json::from_str(data)?;
+    let handle = |message: SsMmessage| -> Result<bool> {
+        let data: Value = serde_json::from_str(&message.data)?;
         maybe_catch_error(&data)?;
         if is_vl {
             if let Some(text) =
-- 
cgit v1.2.3