diff --git a/service/generalModel/openai.go b/service/generalModel/openai.go index 2e89040..568c1cb 100644 --- a/service/generalModel/openai.go +++ b/service/generalModel/openai.go @@ -165,8 +165,8 @@ func SingleTurnConversationText(prompt string, param proto.ModelParam) (string, var answer string var err error apiKey := param.APIKey - if apiKey == "" { - log.Printf("openai apiKey is empty\n") + if apiKey == "" || param.Url == "" { + log.Printf("openai apiKey or url is empty\n") return "", nil } openaiClient := openai.NewClient( diff --git a/service/imService.go b/service/imService.go index 68dd4fe..042d580 100644 --- a/service/imService.go +++ b/service/imService.go @@ -39,6 +39,7 @@ func UpdateSessionNameByAIAndID(id, userID int, originSessionName string) error log.Println("UpdateSessionNameByAIAndID model param error:", err) return errors.New("session name ai model param error") } + modelParam.Url = model.Url newSessionName, err2 := generalModel.SingleTurnConversationText(prompt, modelParam) if err2 != nil { return err