OpenAI o1-preview and o1-mini support (#17796)
Release Notes: - Added support for OpenAI o1-mini and o1-preview models. --------- Co-authored-by: Jason Mancuso <7891333+jvmncs@users.noreply.github.com> Co-authored-by: Bennet <bennet@zed.dev>
This commit is contained in:
parent
e145c13f73
commit
d245f5e75c
5 changed files with 39 additions and 14 deletions
|
@ -241,6 +241,7 @@ pub struct LanguageModelRequest {
|
|||
|
||||
impl LanguageModelRequest {
|
||||
pub fn into_open_ai(self, model: String, max_output_tokens: Option<u32>) -> open_ai::Request {
|
||||
let stream = !model.starts_with("o1-");
|
||||
open_ai::Request {
|
||||
model,
|
||||
messages: self
|
||||
|
@ -259,7 +260,7 @@ impl LanguageModelRequest {
|
|||
},
|
||||
})
|
||||
.collect(),
|
||||
stream: true,
|
||||
stream,
|
||||
stop: self.stop,
|
||||
temperature: self.temperature,
|
||||
max_tokens: max_output_tokens,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue