From 5458150ed3203cf13b0371efa2c791ac696cee93 Mon Sep 17 00:00:00 2001 From: sigoden Date: Thu, 23 May 2024 19:28:56 +0800 Subject: fix: json stream parser and refine client modules (#538) --- src/client/ernie.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/client/ernie.rs') diff --git a/src/client/ernie.rs b/src/client/ernie.rs index 751eb62..68c4224 100644 --- a/src/client/ernie.rs +++ b/src/client/ernie.rs @@ -1,8 +1,8 @@ use super::access_token::*; use super::{ maybe_catch_error, patch_system_message, sse_stream, Client, CompletionOutput, ErnieClient, - ExtraConfig, Model, ModelData, ModelPatches, PromptAction, PromptKind, SendData, SsMmessage, - SseHandler, + ExtraConfig, Model, ModelData, ModelPatches, PromptAction, PromptKind, SendData, SseHandler, + SseMmessage, }; use anyhow::{anyhow, Context, Result}; @@ -108,7 +108,7 @@ async fn send_message(builder: RequestBuilder) -> Result { } async fn send_message_streaming(builder: RequestBuilder, handler: &mut SseHandler) -> Result<()> { - let handle = |message: SsMmessage| -> Result { + let handle = |message: SseMmessage| -> Result { let data: Value = serde_json::from_str(&message.data)?; debug!("stream-data: {data}"); if let Some(text) = data["result"].as_str() { -- cgit v1.2.3