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

make value on str enum

Browse Source
This commit is contained in:
Yannick Stephan
2025-02-16 15:54:54 +01:00
parent 2bf238396e
commit abad9f235c
3 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lightrag/kg/json_doc_status_impl.py
View File

@@ -49,10 +49,10 @@ class JsonDocStatusStorage(DocStatusStorage):
) -> dict[str, DocProcessingStatus]:
"""Get all documents with a specific status"""
return {
k: DocProcessingStatus(**v)
for k, v in self._data.items()
if v["status"] == status.value
}
k: DocProcessingStatus(**v)
for k, v in self._data.items()
if v["status"] == status.value
}
async def index_done_callback(self) -> None:
write_json(self._data, self._file_name)

1
lightrag/kg/redis_impl.py
View File

@@ -18,6 +18,7 @@ import json
config = configparser.ConfigParser()
config.read("config.ini", "utf-8")
@final
@dataclass
class RedisKVStorage(BaseKVStorage):

8
lightrag/lightrag.py
View File

@@ -674,7 +674,7 @@ class LightRAG:
"content": content,
"content_summary": self._get_content_summary(content),
"content_length": len(content),
"status": DocStatus.PENDING,
"status": DocStatus.PENDING.value,
"created_at": datetime.now().isoformat(),
"updated_at": datetime.now().isoformat(),
}
@@ -745,7 +745,7 @@ class LightRAG:
await self.doc_status.upsert(
{
doc_status_id: {
"status": DocStatus.PROCESSING,
"status": DocStatus.PROCESSING.value,
"updated_at": datetime.now().isoformat(),
"content": status_doc.content,
"content_summary": status_doc.content_summary,
@@ -782,7 +782,7 @@ class LightRAG:
await self.doc_status.upsert(
{
doc_status_id: {
"status": DocStatus.PROCESSED,
"status": DocStatus.PROCESSED.value,
"chunks_count": len(chunks),
"content": status_doc.content,
"content_summary": status_doc.content_summary,
@@ -799,7 +799,7 @@ class LightRAG:
await self.doc_status.upsert(
{
doc_status_id: {
"status": DocStatus.FAILED,
"status": DocStatus.FAILED.value,
"error": str(e),
"content": status_doc.content,
"content_summary": status_doc.content_summary,