@@ -34,8 +34,8 @@ pub enum Model {
34
34
Gpt4 ,
35
35
#[ serde( alias = "gpt-3.5-turbo" , rename = "gpt-3.5-turbo" ) ]
36
36
Gpt3_5Turbo ,
37
- #[ serde( alias = "o1-preview " , rename = "o1" ) ]
38
- O1Preview ,
37
+ #[ serde( alias = "o1" , rename = "o1" ) ]
38
+ O1 ,
39
39
#[ serde( alias = "o1-mini" , rename = "o1-mini" ) ]
40
40
O1Mini ,
41
41
#[ serde( alias = "claude-3-5-sonnet" , rename = "claude-3.5-sonnet" ) ]
@@ -46,7 +46,7 @@ impl Model {
46
46
pub fn uses_streaming ( & self ) -> bool {
47
47
match self {
48
48
Self :: Gpt4o | Self :: Gpt4 | Self :: Gpt3_5Turbo | Self :: Claude3_5Sonnet => true ,
49
- Self :: O1Mini | Self :: O1Preview => false ,
49
+ Self :: O1Mini | Self :: O1 => false ,
50
50
}
51
51
}
52
52
@@ -55,7 +55,7 @@ impl Model {
55
55
"gpt-4o" => Ok ( Self :: Gpt4o ) ,
56
56
"gpt-4" => Ok ( Self :: Gpt4 ) ,
57
57
"gpt-3.5-turbo" => Ok ( Self :: Gpt3_5Turbo ) ,
58
- "o1-preview " => Ok ( Self :: O1Preview ) ,
58
+ "o1" => Ok ( Self :: O1 ) ,
59
59
"o1-mini" => Ok ( Self :: O1Mini ) ,
60
60
"claude-3-5-sonnet" => Ok ( Self :: Claude3_5Sonnet ) ,
61
61
_ => Err ( anyhow ! ( "Invalid model id: {}" , id) ) ,
@@ -68,7 +68,7 @@ impl Model {
68
68
Self :: Gpt4 => "gpt-4" ,
69
69
Self :: Gpt4o => "gpt-4o" ,
70
70
Self :: O1Mini => "o1-mini" ,
71
- Self :: O1Preview => "o1-preview " ,
71
+ Self :: O1 => "o1" ,
72
72
Self :: Claude3_5Sonnet => "claude-3-5-sonnet" ,
73
73
}
74
74
}
@@ -79,7 +79,7 @@ impl Model {
79
79
Self :: Gpt4 => "GPT-4" ,
80
80
Self :: Gpt4o => "GPT-4o" ,
81
81
Self :: O1Mini => "o1-mini" ,
82
- Self :: O1Preview => "o1-preview " ,
82
+ Self :: O1 => "o1" ,
83
83
Self :: Claude3_5Sonnet => "Claude 3.5 Sonnet" ,
84
84
}
85
85
}
@@ -90,7 +90,7 @@ impl Model {
90
90
Self :: Gpt4 => 32768 ,
91
91
Self :: Gpt3_5Turbo => 12288 ,
92
92
Self :: O1Mini => 20000 ,
93
- Self :: O1Preview => 20000 ,
93
+ Self :: O1 => 20000 ,
94
94
Self :: Claude3_5Sonnet => 200_000 ,
95
95
}
96
96
}
0 commit comments