1 files changed, 8 insertions, 3 deletions
diff --git a/src/client/openai.rs b/src/client/openai.rs
index a82ee9a..98de5a8 100644
--- a/src/client/openai.rs
+++ b/src/client/openai.rs
@@ -1,6 +1,11 @@
 use super::{
-    ExtraConfig, ModelInfo, OpenAIClient, PromptKind, PromptType, ReplyStreamHandler, SendData,
-    TokensCountFactors,
+    ExtraConfig, OpenAIClient, PromptType, SendData,
+};
+
+use crate::{
+    config::{ModelInfo, TokensCountFactors},
+    render::ReplyHandler,
+    utils::PromptKind,
 };
 
 use anyhow::{anyhow, bail, Result};
@@ -88,7 +93,7 @@ pub async fn openai_send_message(builder: RequestBuilder) -> Result<String> {
 
 pub async fn openai_send_message_streaming(
     builder: RequestBuilder,
-    handler: &mut ReplyStreamHandler,
+    handler: &mut ReplyHandler,
 ) -> Result<()> {
     let mut es = builder.eventsource()?;
     while let Some(event) = es.next().await {