gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat (metadata): added IN clause management

Browse source
This commit is contained in:
GGrassia 2025-10-07 11:24:59 +02:00
parent e04cd3c7d8
commit e38387964f
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
lightrag/kg/postgres_impl.py
View file

@ -2070,7 +2070,16 @@ class PGVectorStorage(BaseVectorStorage):
return str(val).replace("'", "''") # escape single quotes return str(val).replace("'", "''") # escape single quotes
def build_single_condition(key, value): def build_single_condition(key, value):
if isinstance(value, (dict, list)): if isinstance(value, list):
# If list contains only scalars, use IN clause
if all(isinstance(v, (str, int, float, bool)) for v in value):
escaped_values = ", ".join(f"'{escape_str(str(v))}'" for v in value)
return f"metadata->>'{key}' IN ({escaped_values})"
else:
# fallback to JSON matching for complex types
json_value = json.dumps(value).replace("'", "''")
return f"metadata->'{key}' = '{json_value}'::jsonb"
elif isinstance(value, dict):
json_value = json.dumps(value).replace("'", "''") json_value = json.dumps(value).replace("'", "''")
return f"metadata->'{key}' = '{json_value}'::jsonb" return f"metadata->'{key}' = '{json_value}'::jsonb"
elif isinstance(value, (int, float, bool)): elif isinstance(value, (int, float, bool)):