Skip to content

Commit a5fb553

Browse files
authored
Support OpenAI reasoning models (#850)
These model strings are now available for use. More info: https://openai.com/index/introducing-openai-o1-preview/ https://platform.openai.com/docs/guides/reasoning
1 parent 194a03e commit a5fb553

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

completion.go

+8
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,10 @@ var (
1717
// GPT3 Models are designed for text-based tasks. For code-specific
1818
// tasks, please refer to the Codex series of models.
1919
const (
20+
O1Mini = "o1-mini"
21+
O1Mini20240912 = "o1-mini-2024-09-12"
22+
O1Preview = "o1-preview"
23+
O1Preview20240912 = "o1-preview-2024-09-12"
2024
GPT432K0613 = "gpt-4-32k-0613"
2125
GPT432K0314 = "gpt-4-32k-0314"
2226
GPT432K = "gpt-4-32k"
@@ -83,6 +87,10 @@ const (
8387

8488
var disabledModelsForEndpoints = map[string]map[string]bool{
8589
"/completions": {
90+
O1Mini: true,
91+
O1Mini20240912: true,
92+
O1Preview: true,
93+
O1Preview20240912: true,
8694
GPT3Dot5Turbo: true,
8795
GPT3Dot5Turbo0301: true,
8896
GPT3Dot5Turbo0613: true,

0 commit comments

Comments
 (0)