summaryrefslogtreecommitdiffstats
path: root/src/client/replicate.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/client/replicate.rs')
-rw-r--r--src/client/replicate.rs8
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);
}