From 7e29f642a99682ea7bb074eb4e7aa7d34d523515 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sat, 19 Oct 2024 18:25:40 +0800
Subject: feat: support openai o1 models (#935)

---
 src/config/input.rs | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'src/config/input.rs')

diff --git a/src/config/input.rs b/src/config/input.rs
index 7662d2c..0325e1c 100644
--- a/src/config/input.rs
+++ b/src/config/input.rs
@@ -135,6 +135,10 @@ impl Input {
         self.text = text;
     }
 
+    pub fn stream(&self) -> bool {
+        self.config.read().stream && !self.role().model().no_stream()
+    }
+
     pub fn continue_output(&self) -> Option<&str> {
         self.continue_output.as_deref()
     }
-- 
cgit v1.2.3