From 4e4118d406e644e4e46830eec3f7e9f54f4d533e Mon Sep 17 00:00:00 2001
From: liweixin <liweixin@cvte.com>
Date: Tue, 7 Apr 2026 20:14:20 +0800
Subject: [PATCH] fix: pass provider to connection test so Azure OpenAI uses
 correct endpoint

---
 frontend/src/api/initialization/index.ts      | 1 +
 frontend/src/components/ModelEditorDialog.vue | 6 ++++--
 internal/handler/initialization.go            | 7 +++++--
 3 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/frontend/src/api/initialization/index.ts b/frontend/src/api/initialization/index.ts
index 56303d17..e079816b 100644
--- a/frontend/src/api/initialization/index.ts
+++ b/frontend/src/api/initialization/index.ts
@@ -255,6 +255,7 @@ export function checkRemoteModel(modelConfig: {
     modelName: string;
     baseUrl: string;
     apiKey?: string;
+    provider?: string;
 }): Promise<{
     available: boolean;
     message?: string;
diff --git a/frontend/src/components/ModelEditorDialog.vue b/frontend/src/components/ModelEditorDialog.vue
index df1f2ed7..269aadef 100644
--- a/frontend/src/components/ModelEditorDialog.vue
+++ b/frontend/src/components/ModelEditorDialog.vue
@@ -822,7 +822,8 @@ const checkRemoteAPI = async () => {
         result = await checkRemoteModel({
           modelName: formData.value.modelName,
           baseUrl: formData.value.baseUrl,
-          apiKey: formData.value.apiKey || ''
+          apiKey: formData.value.apiKey || '',
+          provider: formData.value.provider
         })
         break
         
@@ -858,7 +859,8 @@ const checkRemoteAPI = async () => {
         result = await checkRemoteModel({
           modelName: formData.value.modelName,
           baseUrl: formData.value.baseUrl,
-          apiKey: formData.value.apiKey || ''
+          apiKey: formData.value.apiKey || '',
+          provider: formData.value.provider
         })
         break
 
diff --git a/internal/handler/initialization.go b/internal/handler/initialization.go
index 5bb7f26f..2c49ffc8 100644
--- a/internal/handler/initialization.go
+++ b/internal/handler/initialization.go
@@ -1468,6 +1468,7 @@ type RemoteModelCheckRequest struct {
 	ModelName string `json:"modelName" binding:"required"`
 	BaseURL   string `json:"baseUrl"   binding:"required"`
 	APIKey    string `json:"apiKey"`
+	Provider  string `json:"provider"`
 }
 
 // CheckRemoteModel godoc
@@ -1513,8 +1514,9 @@ func (h *InitializationHandler) CheckRemoteModel(c *gin.Context) {
 		Name:   req.ModelName,
 		Source: "remote",
 		Parameters: types.ModelParameters{
-			BaseURL: req.BaseURL,
-			APIKey:  req.APIKey,
+			BaseURL:  req.BaseURL,
+			APIKey:   req.APIKey,
+			Provider: req.Provider,
 		},
 		Type: "llm", // 默认类型，实际检查时不区分具体类型
 	}
@@ -1644,6 +1646,7 @@ func (h *InitializationHandler) checkRemoteModelConnection(ctx context.Context,
 		ModelName: model.Name,
 		APIKey:    model.Parameters.APIKey,
 		ModelID:   model.Name,
+		Provider:  model.Parameters.Provider,
 	}
 
 	// 创建聊天实例