creamlike1024 9 месяцев назад
Родитель
Сommit
108b67be6c
2 измененных файлов с 3 добавлено и 1 удалено
  1. 2 0
      middleware/distributor.go
  2. 1 1
      relay/channel/coze/relay-coze.go

+ 2 - 0
middleware/distributor.go

@@ -240,5 +240,7 @@ func SetupContextForSelectedChannel(c *gin.Context, channel *model.Channel, mode
 		c.Set("api_version", channel.Other)
 	case common.ChannelTypeMokaAI:
 		c.Set("api_version", channel.Other)
+	case common.ChannelTypeCoze:
+		c.Set("bot_id", channel.Other)
 	}
 }

+ 1 - 1
relay/channel/coze/relay-coze.go

@@ -30,7 +30,7 @@ func convertCozeChatRequest(c *gin.Context, request dto.GeneralOpenAIRequest) *C
 	}
 	cozeRequest := &CozeChatRequest{
 		// TODO: model to botid
-		BotId:              "1",
+		BotId:              c.GetString("bot_id"),
 		UserId:             c.GetString("id"),
 		AdditionalMessages: messages,
 		Stream:             request.Stream,