242 lines
9.1 KiB
Python
242 lines
9.1 KiB
Python
import traceback
|
||
from flask import Blueprint, request
|
||
from services.knowledgebases.service import KnowledgebaseService
|
||
from utils import success_response, error_response
|
||
from .. import knowledgebase_bp
|
||
|
||
@knowledgebase_bp.route('', methods=['GET'])
|
||
def get_knowledgebase_list():
|
||
"""获取知识库列表"""
|
||
try:
|
||
params = {
|
||
'page': int(request.args.get('currentPage', 1)),
|
||
'size': int(request.args.get('size', 10)),
|
||
'name': request.args.get('name', '')
|
||
}
|
||
result = KnowledgebaseService.get_knowledgebase_list(**params)
|
||
return success_response(result)
|
||
except ValueError as e:
|
||
return error_response("参数类型错误", code=400)
|
||
except Exception as e:
|
||
return error_response(str(e))
|
||
|
||
@knowledgebase_bp.route('/<string:kb_id>', methods=['GET'])
|
||
def get_knowledgebase_detail(kb_id):
|
||
"""获取知识库详情"""
|
||
try:
|
||
knowledgebase = KnowledgebaseService.get_knowledgebase_detail(
|
||
kb_id=kb_id
|
||
)
|
||
if not knowledgebase:
|
||
return error_response('知识库不存在', code=404)
|
||
return success_response(knowledgebase)
|
||
except Exception as e:
|
||
return error_response(str(e))
|
||
|
||
@knowledgebase_bp.route('', methods=['POST'])
|
||
def create_knowledgebase():
|
||
"""创建知识库"""
|
||
try:
|
||
data = request.json
|
||
if not data.get('name'):
|
||
return error_response('知识库名称不能为空', code=400)
|
||
|
||
# 移除 created_by 参数
|
||
kb = KnowledgebaseService.create_knowledgebase(**data)
|
||
return success_response(kb, "创建成功", code=0)
|
||
except Exception as e:
|
||
return error_response(str(e))
|
||
|
||
@knowledgebase_bp.route('/<string:kb_id>', methods=['PUT'])
|
||
def update_knowledgebase(kb_id):
|
||
"""更新知识库"""
|
||
try:
|
||
data = request.json
|
||
kb = KnowledgebaseService.update_knowledgebase(
|
||
kb_id=kb_id,
|
||
**data
|
||
)
|
||
if not kb:
|
||
return error_response('知识库不存在', code=404)
|
||
return success_response(kb)
|
||
except Exception as e:
|
||
return error_response(str(e))
|
||
|
||
@knowledgebase_bp.route('/<string:kb_id>', methods=['DELETE'])
|
||
def delete_knowledgebase(kb_id):
|
||
"""删除知识库"""
|
||
try:
|
||
result = KnowledgebaseService.delete_knowledgebase(
|
||
kb_id=kb_id
|
||
)
|
||
if not result:
|
||
return error_response('知识库不存在', code=404)
|
||
return success_response(message='删除成功')
|
||
except Exception as e:
|
||
return error_response(str(e))
|
||
|
||
@knowledgebase_bp.route('/batch', methods=['DELETE'])
|
||
def batch_delete_knowledgebase():
|
||
"""批量删除知识库"""
|
||
try:
|
||
data = request.json
|
||
if not data or not data.get('ids'):
|
||
return error_response('请选择要删除的知识库', code=400)
|
||
|
||
result = KnowledgebaseService.batch_delete_knowledgebase(
|
||
kb_ids=data['ids']
|
||
)
|
||
return success_response(message=f'成功删除 {result} 个知识库')
|
||
except Exception as e:
|
||
return error_response(str(e))
|
||
|
||
@knowledgebase_bp.route('/<string:kb_id>/documents', methods=['GET'])
|
||
def get_knowledgebase_documents(kb_id):
|
||
"""获取知识库下的文档列表"""
|
||
try:
|
||
params = {
|
||
'kb_id': kb_id,
|
||
'page': int(request.args.get('currentPage', 1)),
|
||
'size': int(request.args.get('size', 10)),
|
||
'name': request.args.get('name', '')
|
||
}
|
||
result = KnowledgebaseService.get_knowledgebase_documents(**params)
|
||
return success_response(result)
|
||
except ValueError as e:
|
||
return error_response("参数类型错误", code=400)
|
||
except Exception as e:
|
||
return error_response(str(e))
|
||
|
||
@knowledgebase_bp.route('/<string:kb_id>/documents', methods=['POST'])
|
||
def add_documents_to_knowledgebase(kb_id):
|
||
"""添加文档到知识库"""
|
||
try:
|
||
print(f"[DEBUG] 接收到添加文档请求，kb_id: {kb_id}")
|
||
data = request.json
|
||
if not data:
|
||
print("[ERROR] 请求数据为空")
|
||
return error_response('请求数据不能为空', code=400)
|
||
|
||
file_ids = data.get('file_ids', [])
|
||
print(f"[DEBUG] 接收到的file_ids: {file_ids}, 类型: {type(file_ids)}")
|
||
|
||
try:
|
||
result = KnowledgebaseService.add_documents_to_knowledgebase(
|
||
kb_id=kb_id,
|
||
file_ids=file_ids
|
||
)
|
||
print(f"[DEBUG] 服务层处理成功，结果: {result}")
|
||
return success_response(
|
||
data=result,
|
||
message="添加成功",
|
||
code=201
|
||
)
|
||
except Exception as service_error:
|
||
print(f"[ERROR] 服务层错误详情: {str(service_error)}")
|
||
|
||
traceback.print_exc()
|
||
return error_response(str(service_error), code=500)
|
||
|
||
except Exception as e:
|
||
print(f"[ERROR] 路由层错误详情: {str(e)}")
|
||
traceback.print_exc()
|
||
return error_response(str(e), code=500)
|
||
|
||
@knowledgebase_bp.route('/documents/<string:doc_id>', methods=['DELETE', 'OPTIONS'])
|
||
def delete_document(doc_id):
|
||
"""删除文档"""
|
||
# 处理 OPTIONS 预检请求
|
||
if request.method == 'OPTIONS':
|
||
response = success_response({})
|
||
# 添加 CORS 相关头
|
||
response.headers.add('Access-Control-Allow-Methods', 'DELETE')
|
||
response.headers.add('Access-Control-Allow-Headers', 'Content-Type,Authorization')
|
||
return response
|
||
|
||
try:
|
||
KnowledgebaseService.delete_document(doc_id)
|
||
return success_response(message="删除成功")
|
||
except Exception as e:
|
||
return error_response(str(e))
|
||
|
||
@knowledgebase_bp.route('/documents/<doc_id>/parse', methods=['POST'])
|
||
def parse_document(doc_id):
|
||
"""开始解析文档"""
|
||
# 处理 OPTIONS 预检请求
|
||
if request.method == 'OPTIONS':
|
||
response = success_response({})
|
||
# 添加 CORS 相关头
|
||
response.headers.add('Access-Control-Allow-Methods', 'POST')
|
||
response.headers.add('Access-Control-Allow-Headers', 'Content-Type,Authorization')
|
||
return response
|
||
|
||
try:
|
||
result = KnowledgebaseService.async_parse_document(doc_id)
|
||
return success_response(data=result)
|
||
except Exception as e:
|
||
return error_response(str(e), code=500)
|
||
|
||
@knowledgebase_bp.route('/documents/<doc_id>/parse/progress', methods=['GET'])
|
||
def get_parse_progress(doc_id):
|
||
"""获取文档解析进度"""
|
||
# 处理 OPTIONS 预检请求
|
||
if request.method == 'OPTIONS':
|
||
response = success_response({})
|
||
# 添加 CORS 相关头
|
||
response.headers.add('Access-Control-Allow-Methods', 'GET')
|
||
response.headers.add('Access-Control-Allow-Headers', 'Content-Type,Authorization')
|
||
return response
|
||
|
||
try:
|
||
result = KnowledgebaseService.get_document_parse_progress(doc_id)
|
||
if isinstance(result, dict) and 'error' in result:
|
||
return error_response(result['error'], code=404)
|
||
return success_response(data=result)
|
||
except Exception as e:
|
||
print(f"获取解析进度失败: {str(e)}")
|
||
return error_response("解析进行中，请稍后重试", code=202)
|
||
|
||
# 获取系统 Embedding 配置路由
|
||
@knowledgebase_bp.route('/system_embedding_config', methods=['GET'])
|
||
def get_system_embedding_config_route():
|
||
"""获取系统级 Embedding 配置的API端点"""
|
||
try:
|
||
config_data = KnowledgebaseService.get_system_embedding_config()
|
||
return success_response(data=config_data)
|
||
except Exception as e:
|
||
print(f"获取系统 Embedding 配置失败: {str(e)}")
|
||
return error_response(message=f"获取配置失败: {str(e)}", code=500) # 返回通用错误信息
|
||
|
||
# 设置系统 Embedding 配置路由
|
||
@knowledgebase_bp.route('/system_embedding_config', methods=['POST'])
|
||
def set_system_embedding_config_route():
|
||
"""设置系统级 Embedding 配置的API端点"""
|
||
try:
|
||
data = request.json
|
||
if not data:
|
||
return error_response('请求数据不能为空', code=400)
|
||
|
||
llm_name = data.get('llm_name', '').strip()
|
||
api_base = data.get('api_base', '').strip()
|
||
api_key = data.get('api_key', '').strip() # 允许空
|
||
|
||
if not llm_name or not api_base:
|
||
return error_response('模型名称和 API 地址不能为空', code=400)
|
||
|
||
# 调用服务层进行处理（包括连接测试和数据库操作）
|
||
success, message = KnowledgebaseService.set_system_embedding_config(
|
||
llm_name=llm_name,
|
||
api_base=api_base,
|
||
api_key=api_key
|
||
)
|
||
|
||
if success:
|
||
return success_response(message=message)
|
||
else:
|
||
# 如果服务层返回失败（例如连接测试失败或数据库错误），将消息返回给前端
|
||
return error_response(message=message, code=400) # 使用 400 表示操作失败
|
||
|
||
except Exception as e:
|
||
# 捕获路由层或未预料的服务层异常
|
||
print(f"设置系统 Embedding 配置失败: {str(e)}")
|
||
return error_response(message=f"设置配置时发生内部错误: {str(e)}", code=500) |