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

Files changed (1) hide show
  1. lightrag/kg/postgres_impl.py +4 -3
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=$1"
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,