Skip to content

Commit

Permalink
Support OpenAI reasoning models (#850)
Browse files Browse the repository at this point in the history
  • Loading branch information
abatilo authored Sep 17, 2024
1 parent 194a03e commit a5fb553
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions completion.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,10 @@ var (
// GPT3 Models are designed for text-based tasks. For code-specific
// tasks, please refer to the Codex series of models.
const (
O1Mini = "o1-mini"
O1Mini20240912 = "o1-mini-2024-09-12"
O1Preview = "o1-preview"
O1Preview20240912 = "o1-preview-2024-09-12"
GPT432K0613 = "gpt-4-32k-0613"
GPT432K0314 = "gpt-4-32k-0314"
GPT432K = "gpt-4-32k"
Expand Down Expand Up @@ -83,6 +87,10 @@ const (

var disabledModelsForEndpoints = map[string]map[string]bool{
"/completions": {
O1Mini: true,
O1Mini20240912: true,
O1Preview: true,
O1Preview20240912: true,
GPT3Dot5Turbo: true,
GPT3Dot5Turbo0301: true,
GPT3Dot5Turbo0613: true,
Expand Down

0 comments on commit a5fb553

Please sign in to comment.