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

feat(go-agent): Ported retrieval node, added Keenable web search tool (#16396)

Browse Source 
Ported retrieval node, added Keenable web search tool
- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
Zhichang Yu
2026-06-26 22:55:49 +08:00
committed by GitHub
parent 9d18f33296
commit 70546ea406
91 changed files with 5920 additions and 3817 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
internal/engine/elasticsearch/common.go
View File

@@ -95,4 +95,4 @@ func (e *elasticsearchEngine) indexExists(ctx context.Context, indexName string)
// buildMetadataIndexName returns the metadata index name for a tenant
func buildMetadataIndexName(tenantID string) string {
return fmt.Sprintf("ragflow_doc_meta_%s", tenantID)
}
}

50
internal/engine/elasticsearch/meta_filter.go
View File

@@ -35,33 +35,33 @@ var dateRegex = regexp.MustCompile(`^\d{4}-\d{2}-\d{2}$`)
// Supported operators
var supportedOperators = map[string]bool{
"=": true,
"≠": true,
">": true,
"<": true,
"≥": true,
"≤": true,
"in": true,
"not in": true,
"contains": true,
"=": true,
"≠": true,
">": true,
"<": true,
"≥": true,
"≤": true,
"in": true,
"not in": true,
"contains": true,
"not contains": true,
"start with": true,
"end with": true,
"empty": true,
"not empty": true,
"start with": true,
"end with": true,
"empty": true,
"not empty": true,
}
// Range operators mapping
var rangeOps = map[string]string{
">": "gt",
"<": "lt",
"≥": "gte",
"≤": "lte",
">": "gt",
"<": "lt",
"≥": "gte",
"≤": "lte",
}
// Negative operators unsafe for multi-valued fields
var multivalueUnsafeNegativeOps = map[string]bool{
"≠": true,
"≠": true,
"not in": true,
}
@@ -77,8 +77,8 @@ func (e *UnsupportedMetaFilterError) Error() string {
// TranslatedFilter represents a single filter rendered as ES bool clauses
type TranslatedFilter struct {
Must []map[string]interface{}
MustNot []map[string]interface{}
Must []map[string]interface{}
MustNot []map[string]interface{}
}
// ToClauses converts to ES clauses
@@ -97,7 +97,7 @@ func (f *TranslatedFilter) ToClauses() []map[string]interface{} {
// MetaFilterPushdownPlan represents composed ES bool query body
type MetaFilterPushdownPlan struct {
Logic string
Logic string
translated []*TranslatedFilter
}
@@ -522,7 +522,7 @@ func termOrMatch(fieldPath string, value interface{}) map[string]interface{} {
return map[string]interface{}{
"term": map[string]interface{}{
keywordPath(fieldPath): map[string]interface{}{
"value": s,
"value": s,
"case_insensitive": true,
},
},
@@ -556,7 +556,7 @@ func termsStringOrNumeric(fieldPath string, members []interface{}) map[string]in
return map[string]interface{}{
"bool": map[string]interface{}{
"should": shouldClauses,
"should": shouldClauses,
"minimum_should_match": 1,
},
}
@@ -567,7 +567,7 @@ func wildcard(fieldPath string, pattern string) map[string]interface{} {
return map[string]interface{}{
"wildcard": map[string]interface{}{
keywordPath(fieldPath): map[string]interface{}{
"value": pattern,
"value": pattern,
"case_insensitive": true,
},
},
@@ -679,4 +679,4 @@ func splitJSONParts(s string) []string {
}
return parts
}
}

38
internal/engine/infinity/meta_filter.go
View File

@@ -30,28 +30,28 @@ var keyPattern = regexp.MustCompile(`^[a-zA-Z_][a-zA-Z0-9_]*$`)
// Supported operators
var supportedOperators = map[string]bool{
"=": true,
"≠": true,
">": true,
"<": true,
"≥": true,
"≤": true,
"in": true,
"not in": true,
"contains": true,
"=": true,
"≠": true,
">": true,
"<": true,
"≥": true,
"≤": true,
"in": true,
"not in": true,
"contains": true,
"not contains": true,
"start with": true,
"end with": true,
"empty": true,
"not empty": true,
"start with": true,
"end with": true,
"empty": true,
"not empty": true,
}
// Range operators mapping
var rangeOps = map[string]string{
">": ">",
"<": "<",
"≥": ">=",
"≤": "<=",
">": ">",
"<": "<",
"≥": ">=",
"≤": "<=",
}
// MetaFilterTranslator translates filter clauses to Infinity SQL
@@ -150,7 +150,7 @@ func (t *MetaFilterTranslator) translateIn(key string, value interface{}, flt ma
coerced := coerceRangeValue(m, flt)
if num, ok := coerceToFloat(coerced); ok {
numParts = append(numParts, fmt.Sprintf("JSON_CONTAINS(meta_fields, '$.%s', %v)", key, num))
} else if s, ok := coerced.(string); ok {
} else if s, ok := coerced.(string); ok {
escaped := escapeSQLString(s)
stringParts = append(stringParts, fmt.Sprintf("JSON_CONTAINS(meta_fields, '$.%s', '\"%s\"')", key, escaped))
}
@@ -579,4 +579,4 @@ func splitJSONParts(s string) []string {
}
return parts
}
}