gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(postgres): update document status with partial update instead of full upsert

Browse Source
This commit is contained in:
Brenon
2025-02-10 15:05:44 +03:00
parent d0779209d9
commit 4723e9b535
3 changed files with 35 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lightrag/lightrag.py
View File

@@ -542,6 +542,7 @@ class LightRAG:
doc_status_id: {
"status": DocStatus.PROCESSING,
"updated_at": datetime.now().isoformat(),
"content": status_doc.content,
"content_summary": status_doc.content_summary,
"content_length": status_doc.content_length,
"created_at": status_doc.created_at,
@@ -573,7 +574,7 @@ class LightRAG:
]
try:
await asyncio.gather(*tasks)
await self.doc_status.upsert(
await self.doc_status.update_doc_status(
{
doc_status_id: {
"status": DocStatus.PROCESSED,
@@ -586,7 +587,7 @@ class LightRAG:
except Exception as e:
logger.error(f"Failed to process document {doc_id}: {str(e)}")
await self.doc_status.upsert(
await self.doc_status.update_doc_status(
{
doc_status_id: {
"status": DocStatus.FAILED,