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

Merge branch 'fix--psql-return-type'

Browse Source
This commit is contained in:
yangdx
2025-03-18 17:15:29 +08:00
parent 10d2bbfcd7 54e4a31aa6
commit b46f2ff871
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