yangdx
commited on
Commit
·
0516d51
1
Parent(s):
186aa61
fix: Add content column to doc status and fix SQL parameter indexing
Browse files• Add content column to doc status table
• Fix SQL param index in get_by_status query
• Update insert SQL to include content field
lightrag/kg/postgres_impl.py
CHANGED
@@ -471,7 +471,7 @@ class PGDocStatusStorage(DocStatusStorage):
|
|
471 |
self, status: DocStatus
|
472 |
) -> Dict[str, DocProcessingStatus]:
|
473 |
"""Get all documents by status"""
|
474 |
-
sql = "select * from LIGHTRAG_DOC_STATUS where workspace=$1 and status=$
|
475 |
params = {"workspace": self.db.workspace, "status": status}
|
476 |
result = await self.db.query(sql, params, True)
|
477 |
return {
|
@@ -505,8 +505,8 @@ class PGDocStatusStorage(DocStatusStorage):
|
|
505 |
Args:
|
506 |
data: Dictionary of document IDs and their status data
|
507 |
"""
|
508 |
-
sql = """insert into LIGHTRAG_DOC_STATUS(workspace,id,content_summary,content_length,chunks_count,status)
|
509 |
-
values($1,$2,$3,$4,$5,$6)
|
510 |
on conflict(id,workspace) do update set
|
511 |
content = EXCLUDED.content,
|
512 |
content_summary = EXCLUDED.content_summary,
|
@@ -1103,6 +1103,7 @@ TABLES = {
|
|
1103 |
"ddl": """CREATE TABLE LIGHTRAG_DOC_STATUS (
|
1104 |
workspace varchar(255) NOT NULL,
|
1105 |
id varchar(255) NOT NULL,
|
|
|
1106 |
content_summary varchar(255) NULL,
|
1107 |
content_length int4 NULL,
|
1108 |
chunks_count int4 NULL,
|
|
|
471 |
self, status: DocStatus
|
472 |
) -> Dict[str, DocProcessingStatus]:
|
473 |
"""Get all documents by status"""
|
474 |
+
sql = "select * from LIGHTRAG_DOC_STATUS where workspace=$1 and status=$2"
|
475 |
params = {"workspace": self.db.workspace, "status": status}
|
476 |
result = await self.db.query(sql, params, True)
|
477 |
return {
|
|
|
505 |
Args:
|
506 |
data: Dictionary of document IDs and their status data
|
507 |
"""
|
508 |
+
sql = """insert into LIGHTRAG_DOC_STATUS(workspace,id,content,content_summary,content_length,chunks_count,status)
|
509 |
+
values($1,$2,$3,$4,$5,$6,$7)
|
510 |
on conflict(id,workspace) do update set
|
511 |
content = EXCLUDED.content,
|
512 |
content_summary = EXCLUDED.content_summary,
|
|
|
1103 |
"ddl": """CREATE TABLE LIGHTRAG_DOC_STATUS (
|
1104 |
workspace varchar(255) NOT NULL,
|
1105 |
id varchar(255) NOT NULL,
|
1106 |
+
content TEXT NULL,
|
1107 |
content_summary varchar(255) NULL,
|
1108 |
content_length int4 NULL,
|
1109 |
chunks_count int4 NULL,
|