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

Fix create_time conflict in MongoKVStorage updates

Browse source
This commit is contained in:
yangdx 2025-07-03 22:58:08 +08:00
parent 04d793abbd
commit 3355a0ce95
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
lightrag/kg/mongo_impl.py
View file

@ -153,14 +153,19 @@ class MongoKVStorage(BaseKVStorage):
if "llm_cache_list" not in v: if "llm_cache_list" not in v:
v["llm_cache_list"] = [] v["llm_cache_list"] = []
v["_id"] = k # Use flattened key as _id # Create a copy of v for $set operation, excluding create_time to avoid conflicts
v["update_time"] = current_time # Always update update_time v_for_set = v.copy()
v_for_set["_id"] = k # Use flattened key as _id
v_for_set["update_time"] = current_time # Always update update_time
# Remove create_time from $set to avoid conflict with $setOnInsert
v_for_set.pop("create_time", None)
operations.append( operations.append(
UpdateOne( UpdateOne(
{"_id": k}, {"_id": k},
{ {
"$set": v, # Update all fields including update_time "$set": v_for_set, # Update all fields except create_time
"$setOnInsert": { "$setOnInsert": {
"create_time": current_time "create_time": current_time
}, # Set create_time only on insert }, # Set create_time only on insert