zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-23 17:06:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: verify model api (#17183)

Browse Source
This commit is contained in:
Lynn
2026-07-21 19:09:46 +08:00
committed by GitHub
parent 3f9d9529b3
commit 340d30eb12
2 changed files with 310 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/handler/providers.go
View File

@@ -301,7 +301,7 @@ func (h *ProviderHandler) CheckConnection(c *gin.Context) {
}
userID := c.GetString("user_id")
errCode, err := h.modelProviderService.CheckConnection(providerName, req.APIKey, req.Region, req.BaseURL, userID)
errCode, err := h.modelProviderService.CheckConnection(providerName, req.APIKey, req.Region, req.BaseURL, req.InstanceID, userID, service.ListModelNames(req.ModelInfo))
if err != nil {
common.ErrorWithCode(c, errCode, err.Error())
return
@@ -334,9 +334,9 @@ func (h *ProviderHandler) CheckInstanceConnection(c *gin.Context) {
apikey, _ := instanceInfo["api_key"].(string)
region, _ := instanceInfo["region"].(string)
baseURL, _ := instanceInfo["base_url"].(string)
instanceID, _ := instanceInfo["id"].(string)
// Get tenant ID from user
errorCode, err := h.modelProviderService.CheckConnection(providerName, apikey, region, baseURL, userID)
errorCode, err := h.modelProviderService.CheckConnection(providerName, apikey, region, baseURL, instanceID, userID, nil)
if err != nil {
common.ErrorWithCode(c, errorCode, err.Error())
return