فهرست منبع

fix: 修复升级go-openai版本之后azure不可用的问题 (#275)

二丫讲梵 1 سال پیش
والد
کامیت
08f46f35a3
2فایلهای تغییر یافته به همراه7 افزوده شده و 4 حذف شده
  1. 5 2
      pkg/chatgpt/chatgpt.go
  2. 2 2
      pkg/dingbot/dingbot.go

+ 5 - 2
pkg/chatgpt/chatgpt.go

@@ -39,9 +39,12 @@ func New(userId string) *ChatGPT {
 	if public.Config.AzureOn {
 		config = openai.DefaultAzureConfig(
 			public.Config.AzureOpenAIToken,
-			"https://"+public.Config.AzureResourceName+".openai."+
-				"azure.com/",
+			"https://"+public.Config.AzureResourceName+".openai.azure.com",
 		)
+		config.APIVersion = public.Config.AzureApiVersion
+		config.AzureModelMapperFunc = func(model string) string {
+			return public.Config.AzureDeploymentName
+		}
 	} else {
 		if public.Config.HttpProxy != "" {
 			config.HTTPClient.Transport = &http.Transport{

+ 2 - 2
pkg/dingbot/dingbot.go

@@ -97,10 +97,10 @@ func (r ReceiveMsg) ReplyToDingtalk(msgType, msg string) (statuscode int, err er
 	case string(TEXT):
 		msgtmp = &TextMessage{Text: &Text{Content: msg}, MsgType: TEXT, At: &At{AtUserIds: []string{atUser}}}
 	case string(MARKDOWN):
-		if atUser != "" {
+		if atUser != "" && r.ConversationType != "1" {
 			msg = fmt.Sprintf("%s\n\n@%s", msg, atUser)
 		}
-		msgtmp = &MarkDownMessage{MsgType: MARKDOWN, At: &At{AtUserIds: []string{atUser}}, MarkDown: &MarkDown{Title: "Markdown Type", Text: msg}}
+		msgtmp = &MarkDownMessage{MsgType: MARKDOWN, At: &At{AtUserIds: []string{atUser}}, MarkDown: &MarkDown{Title: "Markdown Msg", Text: msg}}
 	default:
 		msgtmp = &TextMessage{Text: &Text{Content: msg}, MsgType: TEXT, At: &At{AtUserIds: []string{atUser}}}
 	}