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, } // 创建聊天实例