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]: ) -> dict[str, DocProcessingStatus]:
"""Get all documents with a specific status""" """Get all documents with a specific status"""
return { return {
k: DocProcessingStatus(**v) k: DocProcessingStatus(**v)
for k, v in self._data.items() for k, v in self._data.items()
if v["status"] == status.value if v["status"] == status.value
} }
async def index_done_callback(self) -> None: async def index_done_callback(self) -> None:
write_json(self._data, self._file_name) 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 = configparser.ConfigParser()
config.read("config.ini", "utf-8") config.read("config.ini", "utf-8")
@final @final
@dataclass @dataclass
class RedisKVStorage(BaseKVStorage): class RedisKVStorage(BaseKVStorage):

8
lightrag/lightrag.py
View File

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