diff --git a/src/prediction.rs b/src/prediction.rs
index 3985725..db9f1b3 100644
--- a/src/prediction.rs
+++ b/src/prediction.rs
@@ -219,7 +219,7 @@ impl Session {
             ];
             let request = CreateChatCompletionRequestArgs::default()
                 .messages(full_conversation.clone())
-                .model("gpt-5.4")
+                .model("gpt-5.4-mini")
                 .tools(tools)
                 .max_completion_tokens(1024u32)
                 .response_format(ResponseFormat::JsonSchema {