feat: implement Azure OpenAI client support and update provider configuration

Author: hgaol

internal/controller/ai_controller.go

@@ -292,6 +292,10 @@ func (c *AIController) createOpenAIClient() *openai.Client {
 
 	aiProvider := aiConfig.GetProvider()
 
+	if aiProvider.Provider == "azure_ai" {
+		return c.createAzureAIClient(aiProvider)
+	}
+
 	config = openai.DefaultConfig(aiProvider.APIKey)
 	config.BaseURL = aiProvider.APIHost
 	if !strings.HasSuffix(config.BaseURL, "/v1") {
@@ -300,6 +304,36 @@ func (c *AIController) createOpenAIClient() *openai.Client {
 	return openai.NewClientWithConfig(config)
 }
 
+// createAzureAIClient creates an OpenAI client configured for Azure AI.
+// Uses the Azure OpenAI compatibility endpoint: https://{resource}.openai.azure.com/openai/v1
+func (c *AIController) createAzureAIClient(aiProvider *schema.SiteAIProvider) *openai.Client {
+	azureBaseURL := strings.TrimRight(aiProvider.APIHost, "/") + "/openai/v1"
+
+	config := openai.DefaultConfig(aiProvider.APIKey)
+	config.BaseURL = azureBaseURL
+	config.HTTPClient = &http.Client{
+		Transport: &azureAPIKeyTransport{
+			apiKey:    aiProvider.APIKey,
+			transport: http.DefaultTransport,
+		},
+	}
+	return openai.NewClientWithConfig(config)
+}
+
+// azureAPIKeyTransport is an http.RoundTripper that replaces the Authorization
+// header with the Azure-style api-key header for Azure OpenAI requests.
+type azureAPIKeyTransport struct {
+	apiKey    string
+	transport http.RoundTripper
+}
+
+func (t *azureAPIKeyTransport) RoundTrip(req *http.Request) (*http.Response, error) {
+	req = req.Clone(req.Context())
+	req.Header.Del("Authorization")
+	req.Header.Set("api-key", t.apiKey)
+	return t.transport.RoundTrip(req)
+}
+
 // getPromptByLanguage
 func (c *AIController) getPromptByLanguage(language i18n.Language, question string) string {
 	aiConfig, err := c.siteInfoService.GetSiteAI(context.Background())
@@ -497,6 +531,10 @@ func (c *AIController) processAIStream(
 			break
 		}
 
+		if len(response.Choices) == 0 {
+			continue
+		}
+
 		choice := response.Choices[0]
 
 		if len(choice.Delta.ToolCalls) > 0 {

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"},{"default_api_host":"https://{your-project}.services.ai.azure.com","display_name":"Azure AI Foundry","name":"azure_openai"}]`},
+		{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 AI","name":"azure_ai"}]`},
 	}
 
 	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"},{"default_api_host":"https://{your-project}.services.ai.azure.com","display_name":"Azure AI Foundry","name":"azure_openai"}]`},
+		{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 AI","name":"azure_ai"}]`},
 	}
 	for _, c := range defaultConfigTable {
 		exist, err := x.Context(ctx).Get(&entity.Config{Key: c.Key})

internal/service/siteinfo/siteinfo_service.go

@@ -24,7 +24,6 @@ import (
 	"encoding/json"
 	errpkg "errors"
 	"fmt"
-	"net/url"
 	"strings"
 
 	"github.com/apache/answer/internal/base/constant"
@@ -730,14 +729,9 @@ func (s *SiteInfoService) GetAIModels(ctx context.Context, req *schema.GetAIMode
 	var respBody *resty.Response
 	apiHost := strings.TrimRight(req.APIHost, "/")
 	if req.Provider == "azure_ai" {
-		// Azure AI: parse resource name from apiHost and list deployments via *.openai.azure.com
+		// Azure AI: list deployments via the Azure OpenAI endpoint
 		r.SetHeader("api-key", req.APIKey)
-		parsedURL, parseErr := url.Parse(apiHost)
-		if parseErr != nil || parsedURL.Host == "" {
-			return resp, errors.BadRequest("invalid api_host URL")
-		}
-		resourceName := strings.Split(parsedURL.Hostname(), ".")[0]
-		deploymentsURL := fmt.Sprintf("https://%s.openai.azure.com/openai/deployments?api-version=2022-12-01", resourceName)
+		deploymentsURL := apiHost + "/openai/deployments?api-version=2022-12-01"
 		respBody, err = r.R().Get(deploymentsURL)
 	} else {
 		// Standard OpenAI-compatible providers