admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-08 10:08:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

enhance: add recalls response for restful api (#39408)

Browse Source 
issue: #37899

Signed-off-by: chasingegg <chao.gao@zilliz.com>
This commit is contained in:
Gao 2025-01-20 10:55:04 +08:00 committed by GitHub
parent 228bff5e7b
commit 9c4a81c80c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
internal/distributed/proxy/httpserver/constant.go
View File

@ -120,6 +120,7 @@ const (
HTTPReturnMessage = "message" HTTPReturnMessage = "message"
HTTPReturnData = "data" HTTPReturnData = "data"
HTTPReturnCost = "cost" HTTPReturnCost = "cost"
HTTPReturnRecalls = "recalls"
HTTPReturnLoadState = "loadState" HTTPReturnLoadState = "loadState"
HTTPReturnLoadProgress = "loadProgress" HTTPReturnLoadProgress = "loadProgress"

4
internal/distributed/proxy/httpserver/handler_v2.go
View File

@ -1141,11 +1141,15 @@ func (h *HandlersV2) search(ctx context.Context, c *gin.Context, anyReq any, dbN
HTTPReturnCode: merr.Code(merr.ErrInvalidSearchResult), HTTPReturnCode: merr.Code(merr.ErrInvalidSearchResult),
HTTPReturnMessage: merr.ErrInvalidSearchResult.Error() + ", error: " + err.Error(), HTTPReturnMessage: merr.ErrInvalidSearchResult.Error() + ", error: " + err.Error(),
}) })
} else {
if len(searchResp.Results.Recalls) > 0 {
HTTPReturnStream(c, http.StatusOK, gin.H{HTTPReturnCode: merr.Code(nil), HTTPReturnData: outputData, HTTPReturnCost: cost, HTTPReturnRecalls: searchResp.Results.Recalls})
} else { } else {
HTTPReturnStream(c, http.StatusOK, gin.H{HTTPReturnCode: merr.Code(nil), HTTPReturnData: outputData, HTTPReturnCost: cost}) HTTPReturnStream(c, http.StatusOK, gin.H{HTTPReturnCode: merr.Code(nil), HTTPReturnData: outputData, HTTPReturnCost: cost})
} }
} }
} }
}
return resp, err return resp, err
} }