Update postgres_impl.py
Browse files
lightrag/kg/postgres_impl.py
CHANGED
|
@@ -264,8 +264,6 @@ class PGKVStorage(BaseKVStorage):
|
|
| 264 |
else:
|
| 265 |
exist_keys = []
|
| 266 |
new_keys = set([s for s in keys if s not in exist_keys])
|
| 267 |
-
print(f"keys: {keys}")
|
| 268 |
-
print(f"new_keys: {new_keys}")
|
| 269 |
return new_keys
|
| 270 |
except Exception as e:
|
| 271 |
logger.error(f"PostgreSQL database error: {e}")
|
|
@@ -500,7 +498,6 @@ class PGDocStatusStorage(DocStatusStorage):
|
|
| 500 |
sql = "select * from LIGHTRAG_DOC_STATUS where workspace=$1 and status=$2"
|
| 501 |
params = {"workspace": self.db.workspace, "status": status.value}
|
| 502 |
result = await self.db.query(sql, params, True)
|
| 503 |
-
print("")
|
| 504 |
docs_by_status = {
|
| 505 |
element["id"]: DocProcessingStatus(
|
| 506 |
content=result[0]["content"],
|
|
@@ -513,7 +510,6 @@ class PGDocStatusStorage(DocStatusStorage):
|
|
| 513 |
)
|
| 514 |
for element in result
|
| 515 |
}
|
| 516 |
-
print(f"Docs by status: {docs_by_status}")
|
| 517 |
return docs_by_status
|
| 518 |
|
| 519 |
async def index_done_callback(self) -> None:
|
|
|
|
| 264 |
else:
|
| 265 |
exist_keys = []
|
| 266 |
new_keys = set([s for s in keys if s not in exist_keys])
|
|
|
|
|
|
|
| 267 |
return new_keys
|
| 268 |
except Exception as e:
|
| 269 |
logger.error(f"PostgreSQL database error: {e}")
|
|
|
|
| 498 |
sql = "select * from LIGHTRAG_DOC_STATUS where workspace=$1 and status=$2"
|
| 499 |
params = {"workspace": self.db.workspace, "status": status.value}
|
| 500 |
result = await self.db.query(sql, params, True)
|
|
|
|
| 501 |
docs_by_status = {
|
| 502 |
element["id"]: DocProcessingStatus(
|
| 503 |
content=result[0]["content"],
|
|
|
|
| 510 |
)
|
| 511 |
for element in result
|
| 512 |
}
|
|
|
|
| 513 |
return docs_by_status
|
| 514 |
|
| 515 |
async def index_done_callback(self) -> None:
|