Refactor code structure for improved readability and maintainability

app/schemas/doc.py

@@ -1,16 +1,21 @@
+"""文档相关数据模型"""
+
 from pydantic import BaseModel
-from typing import Optional
+from typing import Optional, Any
 from datetime import datetime
 
 
 class DocumentUploadResponse(BaseModel):
+    """文档上传响应"""
     doc_id: str
     filename: str
     size: int
     status: str = "uploaded"
+    parse_task_id: Optional[str] = None  # 解析任务ID
 
 
 class DocumentInfo(BaseModel):
+    """文档信息"""
     id: str
     name: str
     chunks: int
@@ -19,10 +24,12 @@ class DocumentInfo(BaseModel):
 
 
 class DocumentListResponse(BaseModel):
+    """文档列表响应"""
     docs: list[DocumentInfo]
 
 
 class ChunkInfo(BaseModel):
+    """文本块信息"""
     chunk_id: str
     doc_name: str
     clause_id: Optional[str] = None
@@ -33,12 +40,25 @@ class ChunkInfo(BaseModel):
 
 
 class ParseResponse(BaseModel):
+    """解析响应"""
     doc_id: str
-    chunks: int
-    status: str = "parsed"
+    task_id: Optional[str] = None
+    chunks: int = 0
+    status: str = "parsing"
 
 
 class EmbedResponse(BaseModel):
+    """嵌入响应"""
     doc_id: str
-    vectors: int
-    status: str = "embedded"
+    task_id: Optional[str] = None
+    vectors: int = 0
+    status: str = "embedding"
+
+
+class TaskStatusResponse(BaseModel):
+    """任务状态响应"""
+    task_id: str
+    status: str
+    progress: int
+    message: Optional[str] = None
+    result: Optional[Any] = None