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

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-05-01 10:04:31 +08:00
parent 950d42c527
commit 36f8787bc7
4 changed files with 21 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lightrag/api/routers/document_routes.py
View File

@@ -20,6 +20,7 @@ from lightrag.base import DocProcessingStatus, DocStatus
from lightrag.api.utils_api import get_combined_auth_dependency from lightrag.api.utils_api import get_combined_auth_dependency
from ..config import global_args from ..config import global_args
# Function to format datetime to ISO format string with timezone information # Function to format datetime to ISO format string with timezone information
def format_datetime(dt: Any) -> Optional[str]: def format_datetime(dt: Any) -> Optional[str]:
"""Format datetime to ISO format string with timezone information """Format datetime to ISO format string with timezone information
@@ -44,6 +45,7 @@ def format_datetime(dt: Any) -> Optional[str]:
# Return ISO format string with timezone information # Return ISO format string with timezone information
return dt.isoformat() return dt.isoformat()
router = APIRouter( router = APIRouter(
prefix="/documents", prefix="/documents",
tags=["documents"], tags=["documents"],
@@ -338,7 +340,7 @@ class PipelineStatusResponse(BaseModel):
history_messages: Optional[List[str]] = None history_messages: Optional[List[str]] = None
update_status: Optional[dict] = None update_status: Optional[dict] = None
@field_validator('job_start', mode='before') @field_validator("job_start", mode="before")
@classmethod @classmethod
def parse_job_start(cls, value): def parse_job_start(cls, value):
"""Process datetime and return as ISO format string with timezone""" """Process datetime and return as ISO format string with timezone"""
@@ -1263,12 +1265,8 @@ def create_document_routes(
content_summary=doc_status.content_summary, content_summary=doc_status.content_summary,
content_length=doc_status.content_length, content_length=doc_status.content_length,
status=doc_status.status, status=doc_status.status,
created_at=format_datetime( created_at=format_datetime(doc_status.created_at),
doc_status.created_at updated_at=format_datetime(doc_status.updated_at),
),
updated_at=format_datetime(
doc_status.updated_at
),
chunks_count=doc_status.chunks_count, chunks_count=doc_status.chunks_count,
error=doc_status.error, error=doc_status.error,
metadata=doc_status.metadata, metadata=doc_status.metadata,

12
lightrag/lightrag.py
View File

@@ -958,7 +958,9 @@ class LightRAG:
"content_summary": status_doc.content_summary, "content_summary": status_doc.content_summary,
"content_length": status_doc.content_length, "content_length": status_doc.content_length,
"created_at": status_doc.created_at, "created_at": status_doc.created_at,
"updated_at": datetime.now(timezone.utc).isoformat(), "updated_at": datetime.now(
timezone.utc
).isoformat(),
"file_path": file_path, "file_path": file_path,
} }
} }
@@ -1018,7 +1020,9 @@ class LightRAG:
"content_summary": status_doc.content_summary, "content_summary": status_doc.content_summary,
"content_length": status_doc.content_length, "content_length": status_doc.content_length,
"created_at": status_doc.created_at, "created_at": status_doc.created_at,
"updated_at": datetime.now(timezone.utc).isoformat(), "updated_at": datetime.now(
timezone.utc
).isoformat(),
"file_path": file_path, "file_path": file_path,
} }
} }
@@ -1053,7 +1057,9 @@ class LightRAG:
"content_summary": status_doc.content_summary, "content_summary": status_doc.content_summary,
"content_length": status_doc.content_length, "content_length": status_doc.content_length,
"created_at": status_doc.created_at, "created_at": status_doc.created_at,
"updated_at": datetime.now(timezone.utc).isoformat(), "updated_at": datetime.now(
timezone.utc
).isoformat(),
"file_path": file_path, "file_path": file_path,
} }
} }