feat: use specific model name for openai protocol

JingYiJun · Feb 9, 2024 · 2f0d71e · 2f0d71e
1 parent 9c0c975
commit 2f0d71e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/apis/record/infer.go b/apis/record/infer.go
@@ -63,7 +63,7 @@ func InferOpenAI(
 	client := openai.NewClientWithConfig(openaiConfig)
 
 	request := openai.ChatCompletionRequest{
-		Model:    model.Description,
+		Model:    model.OpenAIModelName,
 		Messages: postRecord.ToOpenAIMessages(),
 	}
 

diff --git a/models/config.go b/models/config.go
@@ -23,6 +23,7 @@ type ModelConfig struct {
 	Url                      string          `json:"url"`
 	CallbackUrl              string          `json:"callback_url"`
 	APIType                  APIType         `json:"api_type"`
+	OpenAIModelName          string          `json:"openai_model_name"`
 }
 
 type ModelConfigs = []*ModelConfig