fix: update AI provider configuration to include Azure OpenAI and adjust request handling

Author: hgaol

internal/migrations/init_data.go

@@ -353,7 +353,7 @@ var (
 		{ID: 128, Key: "rank.answer.undeleted", Value: `-1`},
 		{ID: 129, Key: "rank.question.undeleted", Value: `-1`},
 		{ID: 130, Key: "rank.tag.undeleted", Value: `-1`},
-		{ID: 131, Key: "ai_config.provider", Value: `[{"default_api_host":"https://api.openai.com","display_name":"OpenAI","name":"openai"},{"default_api_host":"https://generativelanguage.googleapis.com","display_name":"Gemini","name":"gemini"},{"default_api_host":"https://api.anthropic.com","display_name":"Anthropic","name":"anthropic"}]`},
+		{ID: 131, Key: "ai_config.provider", Value: `[{"default_api_host":"https://api.openai.com","display_name":"OpenAI","name":"openai"},{"default_api_host":"https://generativelanguage.googleapis.com","display_name":"Gemini","name":"gemini"},{"default_api_host":"https://api.anthropic.com","display_name":"Anthropic","name":"anthropic"},{"default_api_host":"https://{your-resource}.openai.azure.com","display_name":"Azure OpenAI","name":"azure_openai"}]`},
 	}
 
 	defaultBadgeGroupTable = []*entity.BadgeGroup{

internal/migrations/v31.go

@@ -61,7 +61,7 @@ func addAPIKey(ctx context.Context, x *xorm.Engine) error {
 	}
 
 	defaultConfigTable := []*entity.Config{
-		{ID: 131, Key: "ai_config.provider", Value: `[{"default_api_host":"https://api.openai.com","display_name":"OpenAI","name":"openai"},{"default_api_host":"https://generativelanguage.googleapis.com","display_name":"Gemini","name":"gemini"},{"default_api_host":"https://api.anthropic.com","display_name":"Anthropic","name":"anthropic"}]`},
+		{ID: 131, Key: "ai_config.provider", Value: `[{"default_api_host":"https://api.openai.com","display_name":"OpenAI","name":"openai"},{"default_api_host":"https://generativelanguage.googleapis.com","display_name":"Gemini","name":"gemini"},{"default_api_host":"https://api.anthropic.com","display_name":"Anthropic","name":"anthropic"},{"default_api_host":"https://{your-resource}.openai.azure.com","display_name":"Azure OpenAI","name":"azure_openai"}]`},
 	}
 	for _, c := range defaultConfigTable {
 		exist, err := x.Context(ctx).Get(&entity.Config{Key: c.Key})

internal/schema/ai_config_schema.go

@@ -38,8 +38,9 @@ type GetAIModelsResp struct {
 }
 
 type GetAIModelsReq struct {
-	APIHost string `json:"api_host"`
-	APIKey  string `json:"api_key"`
+	Provider string `json:"provider"`
+	APIHost  string `json:"api_host"`
+	APIKey   string `json:"api_key"`
 }
 
 // GetAIModelResp get AI model response
@@ -49,3 +50,18 @@ type GetAIModelResp struct {
 	Created int    `json:"created"`
 	OwnedBy string `json:"owned_by"`
 }
+
+// GetAzureDeploymentsResp get Azure OpenAI deployments response
+type GetAzureDeploymentsResp struct {
+	Data []struct {
+		Id         string `json:"id"`
+		Model      string `json:"model"`
+		Owner      string `json:"owner"`
+		Object     string `json:"object"`
+		Status     string `json:"status"`
+		CreatedAt  int    `json:"created_at"`
+		UpdatedAt  int    `json:"updated_at"`
+	} `json:"data"`
+}
+
+

internal/service/siteinfo/siteinfo_service.go

@@ -724,9 +724,20 @@ func (s *SiteInfoService) GetAIModels(ctx context.Context, req *schema.GetAIMode
 	}
 
 	r := resty.New()
-	r.SetHeader("Authorization", fmt.Sprintf("Bearer %s", req.APIKey))
 	r.SetHeader("Content-Type", "application/json")
-	respBody, err := r.R().Get(req.APIHost + "/v1/models")
+
+	var respBody *resty.Response
+	apiHost := strings.TrimRight(req.APIHost, "/")
+	if req.Provider == "azure_openai" {
+		// Azure OpenAI uses api-key header and lists deployments
+		r.SetHeader("api-key", req.APIKey)
+		url := fmt.Sprintf("%s/openai/deployments?api-version=2022-12-01", apiHost)
+		respBody, err = r.R().Get(url)
+	} else {
+		// Standard OpenAI-compatible providers
+		r.SetHeader("Authorization", fmt.Sprintf("Bearer %s", req.APIKey))
+		respBody, err = r.R().Get(apiHost + "/v1/models")
+	}
 	if err != nil {
 		log.Error(err)
 		return resp, errors.BadRequest(fmt.Sprintf("failed to get AI models %s", err.Error()))
@@ -736,6 +747,20 @@ func (s *SiteInfoService) GetAIModels(ctx context.Context, req *schema.GetAIMode
 		return resp, errors.BadRequest(fmt.Sprintf("failed to get AI models, response: %s", respBody.String()))
 	}
 
+	if req.Provider == "azure_openai" {
+		data := schema.GetAzureDeploymentsResp{}
+		_ = json.Unmarshal(respBody.Body(), &data)
+		for _, d := range data.Data {
+			resp = append(resp, &schema.GetAIModelResp{
+				Id:      d.Id,
+				Object:  d.Object,
+				Created: d.CreatedAt,
+				OwnedBy: d.Model,
+			})
+		}
+		return resp, nil
+	}
+
 	data := schema.GetAIModelsResp{}
 	_ = json.Unmarshal(respBody.Body(), &data)
 

ui/src/pages/Admin/AiSettings/index.tsx

@@ -84,6 +84,7 @@ const Index = () => {
 
   const checkAiConfigData = (data) => {
     const params = data || {
+      provider: formData.provider.value,
       api_host: formData.api_host.value || apiHostPlaceholder,
       api_key: formData.api_key.value,
     };
@@ -151,6 +152,7 @@ const Index = () => {
     const host = findHistoryProvider?.api_host || provider?.default_api_host;
     if (findHistoryProvider?.model) {
       checkAiConfigData({
+        provider: value,
         api_host: host,
         api_key: findHistoryProvider.api_key,
       });
@@ -263,6 +265,7 @@ const Index = () => {
       );
       const host = currentAiConfig.api_host || provider?.default_api_host;
       checkAiConfigData({
+        provider: currentAiConfig.provider,
         api_host: host,
         api_key: currentAiConfig.api_key,
       });