1 files changed, 7 insertions, 3 deletions
diff --git a/src/config/input.rs b/src/config/input.rs
index 44bbb4b..ae94799 100644
--- a/src/config/input.rs
+++ b/src/config/input.rs
@@ -1,7 +1,7 @@
 use super::{role::Role, session::Session, GlobalConfig};
 
 use crate::client::{
-    init_client, list_models, Client, CompletionData, ImageUrl, Message, MessageContent,
+    init_client, list_models, ChatCompletionsData, Client, ImageUrl, Message, MessageContent,
     MessageContentPart, MessageRole, Model,
 };
 use crate::function::{ToolCallResult, ToolResults};
@@ -149,7 +149,11 @@ impl Input {
         init_client(&self.config, Some(self.model()))
     }
 
-    pub fn prepare_completion_data(&self, model: &Model, stream: bool) -> Result<CompletionData> {
+    pub fn prepare_completion_data(
+        &self,
+        model: &Model,
+        stream: bool,
+    ) -> Result<ChatCompletionsData> {
         if !self.medias.is_empty() && !model.supports_vision() {
             bail!("The current model does not support vision.");
         }
@@ -176,7 +180,7 @@ impl Input {
             };
             functions = config.function.select(function_matcher);
         };
-        Ok(CompletionData {
+        Ok(ChatCompletionsData {
             messages,
             temperature,
             top_p,