outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix metadata

Browse Source
This commit is contained in:
jyong 2025-03-06 11:46:46 +08:00
parent 5b4589951d
commit 7990219aa4
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/workflow/nodes/knowledge_retrieval/entities.py
View File

@ -103,9 +103,9 @@ class Condition(BaseModel):
Conditon detail Conditon detail
""" """
metadata_name: str name: str
comparison_operator: SupportedComparisonOperator comparison_operator: SupportedComparisonOperator
value: str | Sequence[str] | None = None value: str | Sequence[str] | None | int | float = None
class MetadataFilteringCondition(BaseModel): class MetadataFilteringCondition(BaseModel):

4
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
View File

@ -301,7 +301,7 @@ class KnowledgeRetrievalNode(LLMNode):
) )
elif node_data.metadata_filtering_mode == "manual": elif node_data.metadata_filtering_mode == "manual":
for condition in node_data.metadata_filtering_conditions.conditions: for condition in node_data.metadata_filtering_conditions.conditions:
metadata_name = condition.metadata_name metadata_name = condition.name
expected_value = condition.value expected_value = condition.value
if isinstance(expected_value, str): if isinstance(expected_value, str):
expected_value = self.graph_runtime_state.variable_pool.convert_template(expected_value).text expected_value = self.graph_runtime_state.variable_pool.convert_template(expected_value).text
@ -380,7 +380,7 @@ class KnowledgeRetrievalNode(LLMNode):
return [] return []
return automatic_metadata_filters return automatic_metadata_filters
def _process_metadata_filter_func(*, condition: str, metadata_name: str, value: str, query): def _process_metadata_filter_func(self, condition: str, metadata_name: str, value: str, query):
match condition: match condition:
case "contains": case "contains":
query = query.filter(Document.doc_metadata[metadata_name].like(f"%{value}%")) query = query.filter(Document.doc_metadata[metadata_name].like(f"%{value}%"))