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/claude.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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

diff --git a/src/client/claude.rs b/src/client/claude.rs
index be68af1..c76a589 100644
--- a/src/client/claude.rs
+++ b/src/client/claude.rs
@@ -1,7 +1,7 @@
 use super::{
     catch_error, extract_system_message, sse_stream, ClaudeClient, CompletionDetails, ExtraConfig,
     ImageUrl, MessageContent, MessageContentPart, Model, ModelConfig, PromptType, SendData,
-    SseHandler,
+    SsMmessage, SseHandler,
 };
 
 use crate::utils::PromptKind;
@@ -67,8 +67,8 @@ pub async fn claude_send_message_streaming(
     builder: RequestBuilder,
     handler: &mut SseHandler,
 ) -> 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)?;
         if let Some(typ) = data["type"].as_str() {
             if typ == "content_block_delta" {
                 if let Some(text) = data["delta"]["text"].as_str() {
-- 
cgit v1.2.3