Merge pull request #20 from otiai10/develop

add new model

Author: otiai10

models.go

@@ -43,14 +43,19 @@ const (
 	// }}}
 
 	// {{{ 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          = "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"
-	GPT4_0314     = "gpt-3-0314"
+	GPT4_0314     = "gpt-4-0314"
+	GPT4_0613     = "gpt-4-0613"
 	GPT4_32K      = "gpt-4-32k"
 	GPT4_32K_0314 = "gpt-4-32k-0314"
+	GPT4_32K_0613 = "gpt-4-32k-0613"
 	// }}}
 )

testapp/main.go

@@ -29,7 +29,7 @@ var (
 			Run: func() (any, error) {
 				client := openaigo.NewClient(OPENAI_API_KEY)
 				request := openaigo.CompletionRequestBody{
-					Model:  "text-davinci-003",
+					Model:  openaigo.TextDavinci003,
 					Prompt: []string{"Say this is a test"},
 				}
 				return client.Completion(nil, request)
@@ -112,7 +112,7 @@ var (
 					}
 				}
 				request := openaigo.ChatCompletionRequestBody{
-					Model:          openaigo.GPT3_5Turbo,
+					Model:          openaigo.GPT3_5Turbo_0613,
 					StreamCallback: calback,
 					Messages: []openaigo.ChatMessage{
 						{