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

Merge pull request #1107 from JoramMillenaar/fix--psql-return-type

Browse Source 
Fixed get_by_id type error in PSQL impl
This commit is contained in:
Daniel.y
2025-03-18 20:40:28 +08:00
committed by GitHub
parent a3de421b14 54e4a31aa6
commit 251f624273
2 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lightrag/kg/postgres_impl.py
View File

@@ -735,7 +735,7 @@ class PGDocStatusStorage(DocStatusStorage):
if result is None or result == []: if result is None or result == []:
return None return None
else: else:
return DocProcessingStatus( return dict(
content=result[0]["content"], content=result[0]["content"],
content_length=result[0]["content_length"], content_length=result[0]["content_length"],
content_summary=result[0]["content_summary"], content_summary=result[0]["content_summary"],
@@ -1065,7 +1065,6 @@ class PGGraphStorage(BaseGraphStorage):
Args: Args:
query (str): a cypher query to be executed query (str): a cypher query to be executed
params (dict): parameters for the query
Returns: Returns:
list[dict[str, Any]]: a list of dictionaries containing the result set list[dict[str, Any]]: a list of dictionaries containing the result set

3
lightrag/lightrag.py
View File

@@ -1544,8 +1544,7 @@ class LightRAG:
""" """
try: try:
# 1. Get the document status and related data # 1. Get the document status and related data
doc_status = await self.doc_status.get_by_id(doc_id) if not await self.doc_status.get_by_id(doc_id):
if not doc_status:
logger.warning(f"Document {doc_id} not found") logger.warning(f"Document {doc_id} not found")
return return