Add support for azure ai

Author: wengchaoxi

controller/channel-billing.go

@@ -295,7 +295,9 @@ func updateChannelBalance(channel *model.Channel) (float64, error) {
 		if channel.GetBaseURL() != "" {
 			baseURL = channel.GetBaseURL()
 		}
-	case channeltype.Azure:
+	case channeltype.AzureAI:
+		return 0, errors.New("尚未实现")
+	case channeltype.AzureOpenAI:
 		return 0, errors.New("尚未实现")
 	case channeltype.Custom:
 		baseURL = channel.GetBaseURL()

controller/model.go

@@ -84,7 +84,7 @@ func init() {
 		}
 	}
 	for _, channelType := range openai.CompatibleChannels {
-		if channelType == channeltype.Azure {
+		if channelType == channeltype.AzureAI || channelType == channeltype.AzureOpenAI {
 			continue
 		}
 		channelName, channelModelList := openai.GetCompatibleChannelMeta(channelType)

go.mod

@@ -27,6 +27,7 @@ require (
 	github.com/stretchr/testify v1.9.0
 	golang.org/x/crypto v0.31.0
 	golang.org/x/image v0.18.0
+	golang.org/x/sync v0.10.0
 	google.golang.org/api v0.187.0
 	gorm.io/driver/mysql v1.5.6
 	gorm.io/driver/postgres v1.5.7
@@ -99,7 +100,6 @@ require (
 	golang.org/x/arch v0.8.0 // indirect
 	golang.org/x/net v0.26.0 // indirect
 	golang.org/x/oauth2 v0.21.0 // indirect
-	golang.org/x/sync v0.10.0 // indirect
 	golang.org/x/sys v0.28.0 // indirect
 	golang.org/x/text v0.21.0 // indirect
 	golang.org/x/time v0.5.0 // indirect

middleware/distributor.go

@@ -73,7 +73,11 @@ func SetupContextForSelectedChannel(c *gin.Context, channel *model.Channel, mode
 	// this is for backward compatibility
 	if channel.Other != nil {
 		switch channel.Type {
-		case channeltype.Azure:
+		case channeltype.AzureAI:
+			if cfg.APIVersion == "" {
+				cfg.APIVersion = *channel.Other
+			}
+		case channeltype.AzureOpenAI:
 			if cfg.APIVersion == "" {
 				cfg.APIVersion = *channel.Other
 			}