完善后端日志
This commit is contained in:
@@ -196,18 +196,22 @@ async def process_document(
|
||||
meta={"progress": 50, "message": "正在存储到MySQL并生成字段描述"}
|
||||
)
|
||||
|
||||
# 使用 TableRAG 服务完成建表和RAG索引
|
||||
rag_result = await table_rag_service.build_table_rag_index(
|
||||
file_path=file_path,
|
||||
filename=original_filename,
|
||||
sheet_name=parse_options.get("sheet_name"),
|
||||
header_row=parse_options.get("header_row", 0)
|
||||
)
|
||||
try:
|
||||
# 使用 TableRAG 服务完成建表和RAG索引
|
||||
logger.info(f"开始存储Excel到MySQL: {original_filename}, file_path: {file_path}")
|
||||
rag_result = await table_rag_service.build_table_rag_index(
|
||||
file_path=file_path,
|
||||
filename=original_filename,
|
||||
sheet_name=parse_options.get("sheet_name"),
|
||||
header_row=parse_options.get("header_row", 0)
|
||||
)
|
||||
|
||||
if rag_result.get("success"):
|
||||
logger.info(f"RAG索引构建成功: {original_filename}")
|
||||
else:
|
||||
logger.warning(f"RAG索引构建失败: {rag_result.get('error')}")
|
||||
if rag_result.get("success"):
|
||||
logger.info(f"Excel存储到MySQL成功: {original_filename}, table: {rag_result.get('table_name')}")
|
||||
else:
|
||||
logger.error(f"RAG索引构建失败: {rag_result.get('error')}")
|
||||
except Exception as e:
|
||||
logger.error(f"Excel存储到MySQL异常: {str(e)}", exc_info=True)
|
||||
|
||||
else:
|
||||
# 非结构化文档
|
||||
|
||||
@@ -10,6 +10,7 @@ import io
|
||||
|
||||
from app.services.file_service import file_service
|
||||
from app.core.document_parser import XlsxParser
|
||||
from app.services.table_rag_service import table_rag_service
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
@@ -27,7 +28,7 @@ async def upload_excel(
|
||||
header_row: int = Query(0, description="表头所在的行索引")
|
||||
):
|
||||
"""
|
||||
上传并解析 Excel 文件
|
||||
上传并解析 Excel 文件,同时存储到 MySQL 数据库
|
||||
|
||||
Args:
|
||||
file: 上传的 Excel 文件
|
||||
@@ -77,6 +78,23 @@ async def upload_excel(
|
||||
result.metadata['saved_path'] = saved_path
|
||||
result.metadata['original_filename'] = file.filename
|
||||
|
||||
# 存储到 MySQL 数据库
|
||||
try:
|
||||
store_result = await table_rag_service.build_table_rag_index(
|
||||
file_path=saved_path,
|
||||
filename=file.filename,
|
||||
sheet_name=sheet_name if sheet_name else None,
|
||||
header_row=header_row
|
||||
)
|
||||
if store_result.get("success"):
|
||||
result.metadata['mysql_table'] = store_result.get('table_name')
|
||||
result.metadata['row_count'] = store_result.get('row_count')
|
||||
logger.info(f"Excel已存储到MySQL: {file.filename}, 表: {store_result.get('table_name')}")
|
||||
else:
|
||||
logger.warning(f"Excel存储到MySQL失败: {store_result.get('error')}")
|
||||
except Exception as e:
|
||||
logger.error(f"Excel存储到MySQL异常: {str(e)}", exc_info=True)
|
||||
|
||||
return result.to_dict()
|
||||
|
||||
except HTTPException:
|
||||
|
||||
Reference in New Issue
Block a user