diff options
Diffstat (limited to 'src/client/ernie.rs')
| -rw-r--r-- | src/client/ernie.rs | 6 |
1 files changed, 3 insertions, 3 deletions
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<CompletionOutput> { } async fn send_message_streaming(builder: RequestBuilder, handler: &mut SseHandler) -> Result<()> { - let handle = |message: SsMmessage| -> Result<bool> { + let handle = |message: SseMmessage| -> Result<bool> { let data: Value = serde_json::from_str(&message.data)?; debug!("stream-data: {data}"); if let Some(text) = data["result"].as_str() { |