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

Refine handling of POST /api/v1/datasets/search in GO (#15583)

Browse Source 
### What problem does this PR solve?

Refine handling of POST /api/v1/datasets/search in GO

### Type of change

- [x] Refactoring
This commit is contained in:
qinling0210
2026-06-08 11:49:37 +08:00
committed by GitHub
parent 074c331cdf
commit c960dc2a4c
70 changed files with 8580 additions and 1915 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
internal/cli/user_command.go
View File

@@ -498,8 +498,8 @@ func (c *RAGFlowClient) getDatasetID(datasetName string) (string, error) {
return "", fmt.Errorf("dataset '%s' not found", datasetName)
}
// ListMetadata lists metadata for datasets
func (c *RAGFlowClient) ListMetadata(cmd *Command) (ResponseIf, error) {
// GetMetadata gets metadata for one or more datasets
func (c *RAGFlowClient) GetMetadata(cmd *Command) (ResponseIf, error) {
if c.ServerType != "user" {
return nil, fmt.Errorf("this command is only allowed in USER mode")
}
@@ -608,6 +608,68 @@ func (c *RAGFlowClient) SearchOnDatasets(cmd *Command) (ResponseIf, error) {
"vector_similarity_weight": 0.3,
}
// Add optional parameters from command
if val, ok := cmd.Params["top_k"]; ok {
payload["top_k"] = val
}
if val, ok := cmd.Params["similarity_threshold"]; ok {
payload["similarity_threshold"] = val
}
if val, ok := cmd.Params["vector_similarity_weight"]; ok {
payload["vector_similarity_weight"] = val
}
if val, ok := cmd.Params["keyword"]; ok {
payload["keyword"] = val
}
if val, ok := cmd.Params["use_kg"]; ok {
payload["use_kg"] = val
}
if val, ok := cmd.Params["rerank_id"]; ok {
payload["rerank_id"] = val
}
if val, ok := cmd.Params["tenant_rerank_id"]; ok {
payload["tenant_rerank_id"] = val
}
if val, ok := cmd.Params["page_size"]; ok {
payload["page_size"] = val
}
if val, ok := cmd.Params["page"]; ok {
payload["page"] = val
}
if val, ok := cmd.Params["search_id"]; ok {
if s, ok := val.(string); ok {
payload["search_id"] = s
}
}
if val, ok := cmd.Params["cross_languages"]; ok {
if list, ok := val.([]string); ok {
payload["cross_languages"] = list
}
}
if val, ok := cmd.Params["doc_ids"]; ok {
if list, ok := val.([]string); ok {
payload["doc_ids"] = list
}
}
if val, ok := cmd.Params["meta_data_filter"]; ok {
// Accept either a raw JSON string from the CLI or a pre-decoded
// map[string]interface{} (future-proofing for callers that
// construct the command programmatically). The string form is
// the public CLI surface; the map form is for unit tests.
switch v := val.(type) {
case string:
var decoded map[string]interface{}
if err := json.Unmarshal([]byte(v), &decoded); err != nil {
return nil, fmt.Errorf("invalid meta_data_filter JSON: %w", err)
}
payload["meta_data_filter"] = decoded
case map[string]interface{}:
payload["meta_data_filter"] = v
default:
return nil, fmt.Errorf("meta_data_filter must be JSON string or object")
}
}
if iterations > 1 {
// Benchmark mode - return raw result for benchmark stats
return c.HTTPClient.RequestWithIterations("POST", "/datasets/search", "web", nil, payload, iterations)