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/replicate.rs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/client/replicate.rs') diff --git a/src/client/replicate.rs b/src/client/replicate.rs index c549ae8..c39cb07 100644 --- a/src/client/replicate.rs +++ b/src/client/replicate.rs @@ -1,7 +1,7 @@ use super::{ - catch_error, prompt_format::*, sse_stream, Client, CompletionOutput, ExtraConfig, - Model, ModelData, ModelPatches, PromptAction, PromptKind, ReplicateClient, SendData, - SsMmessage, SseHandler, + catch_error, prompt_format::*, sse_stream, Client, CompletionOutput, ExtraConfig, Model, + ModelData, ModelPatches, PromptAction, PromptKind, ReplicateClient, SendData, SseHandler, + SseMmessage, }; use anyhow::{anyhow, Result}; @@ -125,7 +125,7 @@ async fn send_message_streaming( let sse_builder = client.get(stream_url).header("accept", "text/event-stream"); - let handle = |message: SsMmessage| -> Result { + let handle = |message: SseMmessage| -> Result { if message.event == "done" { return Ok(true); } -- cgit v1.2.3