响应格式修改，原始数据查询修改

app/schemas/comprehensive_data.py

@@ -70,7 +70,7 @@ class SectionDataImportRequest(BaseModel):
 
 # 原始数据查询请求
 class OriginalDataQueryRequest(BaseModel):
-    account_id: int  # 账号ID,必填
+    account_id: Optional[int] = None  # 账号ID，可选。不填则查询所有分表
     linecode: Optional[str] = None
     id: Optional[int] = None
     bfpcode: Optional[str] = None
@@ -190,19 +190,40 @@ class BatchOriginalDataImportRequest(BaseModel):
 
 # 新增响应模型
 class DataImportResponse(BaseModel):
-    success: bool
+    code: int = 0  # 响应状态码，0表示成功
     message: str
-    total_count: int
-    success_count: int
-    failed_count: int
-    failed_items: List[Dict[str, Any]] = []
+    data: Optional[Dict[str, Any]] = None  # 将详细信息放入data中
+
+    class Config:
+        schema_extra = {
+            "example": {
+                "code": 0,
+                "message": "批量导入完成",
+                "data": {
+                    "total_count": 100,
+                    "success_count": 100,
+                    "failed_count": 0,
+                    "failed_items": []
+                }
+            }
+        }
 
 # 查询响应模型
 class DataResponse(BaseModel):
-    success: bool
+    code: int = 0  # 响应状态码，0表示成功
     message: str
-    count: int
     data: List[Dict[str, Any]] = []
+    total: Optional[int] = None  # 总数
+
+    class Config:
+        schema_extra = {
+            "example": {
+                "code": 0,
+                "message": "查询成功",
+                "total": 10,
+                "data": []
+            }
+        }
 
 # 兼容旧接口的模型
 class ComprehensiveDataImportRequest(BaseModel):