Skip to content

Commit

Permalink
Update model names to fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
otiai10 committed May 8, 2024
1 parent a2f4b64 commit ac3cf88
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 15 deletions.
25 changes: 11 additions & 14 deletions models.go
Original file line number Diff line number Diff line change
Expand Up @@ -35,20 +35,6 @@ type ModelRetrieveResponse ModelData

// https://beta.openai.com/docs/models/overview
const (
// {{{ https://beta.openai.com/docs/models/gpt-3
TextDavinci003 = "text-davinci-003"
TextCurie001 = "text-curie-001"
TextBabbage001 = "text-babbage-001"
TextAda001 = "text-ada-001"
// }}}

// {{{ https://platform.openai.com/docs/models/gpt-3-5
GPT3_5Turbo = "gpt-3.5-turbo"
GPT3_5Turbo_0301 = "gpt-3.5-turbo-0301"
GPT3_5Turbo_0613 = "gpt-3.5-turbo-0613"
GPT3_5Turbo_16K = "gpt-3.5-turbo-16k"
GPT3_5Turbo_16K_0613 = "gpt-3.5-turbo-16k-0613"
// }}}

// {{{ https://platform.openai.com/docs/models/gpt-4
GPT4 = "gpt-4"
Expand All @@ -58,4 +44,15 @@ const (
GPT4_32K_0314 = "gpt-4-32k-0314"
GPT4_32K_0613 = "gpt-4-32k-0613"
// }}}

// {{{ https://platform.openai.com/docs/models/gpt-3-5
GPT3_5Turbo_0125 = "gpt-3.5-turbo-0125"
GPT3_5Turbo = "gpt-3.5-turbo"
GPT3_5Turbo_1106 = "gpt-3.5-turbo-1106"
GPT3_5Turbo_Instruct = "gpt-3.5-turbo-instruct"
GPT3_5Turbo_16K_0613 = "gpt-3.5-turbo-16k-0613" // @deprecated
GPT3_5Turbo_0613 = "gpt-3.5-turbo-0613" // @deprecated
GPT3_5Turbo_16K = "gpt-3.5-turbo-16k" // @deprecated
GPT3_5Turbo_0301 = "gpt-3.5-turbo-0301" // @deprecated
// }}}
)
2 changes: 1 addition & 1 deletion testapp/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ var (
Run: func() (any, error) {
client := openaigo.NewClient(OPENAI_API_KEY)
request := openaigo.CompletionRequestBody{
Model: openaigo.TextDavinci003,
Model: openaigo.GPT3_5Turbo_Instruct,
Prompt: []string{"Say this is a test"},
}
return client.Completion(nil, request)
Expand Down

0 comments on commit ac3cf88

Please sign in to comment.