diff options
Diffstat (limited to 'src/client/replicate.rs')
| -rw-r--r-- | src/client/replicate.rs | 8 |
1 files changed, 4 insertions, 4 deletions
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<bool> { + let handle = |message: SseMmessage| -> Result<bool> { if message.event == "done" { return Ok(true); } |
