fix metadata

This commit is contained in:
jyong 2025-03-06 13:44:56 +08:00
parent 7990219aa4
commit 7f6c6b7791

View File

@ -390,21 +390,21 @@ class KnowledgeRetrievalNode(LLMNode):
query = query.filter(Document.doc_metadata[metadata_name].like(f"{value}%")) query = query.filter(Document.doc_metadata[metadata_name].like(f"{value}%"))
case "end with": case "end with":
query = query.filter(Document.doc_metadata[metadata_name].like(f"%{value}")) query = query.filter(Document.doc_metadata[metadata_name].like(f"%{value}"))
case "is", "=": case "=" | "is":
query = query.filter(Document.doc_metadata[metadata_name] == value) query = query.filter(Document.doc_metadata[metadata_name] == value)
case "is not", "": case "is not" | "":
query = query.filter(Document.doc_metadata[metadata_name] != value) query = query.filter(Document.doc_metadata[metadata_name] != value)
case "is empty": case "is empty":
query = query.filter(Document.doc_metadata[metadata_name].is_(None)) query = query.filter(Document.doc_metadata[metadata_name].is_(None))
case "is not empty": case "is not empty":
query = query.filter(Document.doc_metadata[metadata_name].isnot(None)) query = query.filter(Document.doc_metadata[metadata_name].isnot(None))
case "before", "<": case "before" | "<":
query = query.filter(Document.doc_metadata[metadata_name] < value) query = query.filter(Document.doc_metadata[metadata_name] < value)
case "after", ">": case "after" | ">":
query = query.filter(Document.doc_metadata[metadata_name] > value) query = query.filter(Document.doc_metadata[metadata_name] > value)
case "", ">=": case "" | ">=":
query = query.filter(Document.doc_metadata[metadata_name] <= value) query = query.filter(Document.doc_metadata[metadata_name] <= value)
case "", ">=": case "" | ">=":
query = query.filter(Document.doc_metadata[metadata_name] >= value) query = query.filter(Document.doc_metadata[metadata_name] >= value)
case _: case _:
raise InvalidConditionError(f"Invalid condition: {condition}") raise InvalidConditionError(f"Invalid condition: {condition}")