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

fixed str enum

Browse Source
This commit is contained in:
Yannick Stephan
2025-02-17 18:26:07 +01:00
parent 70ea5ec878
commit 3b30b7985b
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/base.py
View File

@@ -1,7 +1,7 @@
from __future__ import annotations
from abc import ABC, abstractmethod
from enum import StrEnum
from enum import Enum
import os
from dotenv import load_dotenv
from dataclasses import dataclass, field
@@ -205,7 +205,7 @@ class BaseGraphStorage(StorageNameSpace, ABC):
"""Retrieve a subgraph of the knowledge graph starting from a given node."""
class DocStatus(StrEnum):
class DocStatus(str, Enum):
"""Document processing status"""
PENDING = "pending"

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.value,
"status": DocStatus.PENDING,
"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.value,
"status": DocStatus.PROCESSING,
"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.value,
"status": DocStatus.PROCESSED,
"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.value,
"status": DocStatus.FAILED,
"error": str(e),
"content": status_doc.content,
"content_summary": status_doc.content_summary,