main
HuangHai 4 months ago
parent faae3ff8b2
commit 153d3d56b9

@ -26,6 +26,7 @@ milvus_pool = MilvusConnectionPool(host=MS_HOST, port=MS_PORT, max_connections=M
collection_name = MS_COLLECTION_NAME collection_name = MS_COLLECTION_NAME
collection_manager = MilvusCollectionManager(collection_name) collection_manager = MilvusCollectionManager(collection_name)
# 使用 Lifespan Events 处理应用启动和关闭逻辑 # 使用 Lifespan Events 处理应用启动和关闭逻辑
@asynccontextmanager @asynccontextmanager
async def lifespan(app: FastAPI): async def lifespan(app: FastAPI):
@ -42,6 +43,7 @@ async def lifespan(app: FastAPI):
await app.state.mysql_pool.wait_closed() await app.state.mysql_pool.wait_closed()
logger.info("Milvus 和 MySQL 连接池已关闭。") logger.info("Milvus 和 MySQL 连接池已关闭。")
# 初始化 FastAPI 应用 # 初始化 FastAPI 应用
app = FastAPI(lifespan=lifespan) app = FastAPI(lifespan=lifespan)
@ -51,6 +53,7 @@ client = AsyncOpenAI(
base_url="https://dashscope.aliyuncs.com/compatible-mode/v1", base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
) )
@app.post("/reply") @app.post("/reply")
async def reply(session_id: str = Form(...), prompt: str = Form(...)): async def reply(session_id: str = Form(...), prompt: str = Form(...)):
""" """
@ -140,8 +143,6 @@ async def reply(session_id: str = Form(...), prompt: str = Form(...)):
await asyncio.to_thread(collection_manager.insert_data, entities) await asyncio.to_thread(collection_manager.insert_data, entities)
logger.info("用户输入和大模型反馈已记录到向量数据库。") logger.info("用户输入和大模型反馈已记录到向量数据库。")
# 调用 TTS 生成 MP3 # 调用 TTS 生成 MP3
uuid_str = str(uuid.uuid4()) uuid_str = str(uuid.uuid4())
timestamp = int(time.time()) timestamp = int(time.time())
@ -149,7 +150,8 @@ async def reply(session_id: str = Form(...), prompt: str = Form(...)):
# 生成 TTS 音频数据(不落盘) # 生成 TTS 音频数据(不落盘)
t = TTS(None) # 传入 None 表示不保存到本地文件 t = TTS(None) # 传入 None 表示不保存到本地文件
audio_data, duration = await asyncio.to_thread(t.generate_audio, result) # 假设 TTS 类有一个 generate_audio 方法返回音频数据 audio_data, duration = await asyncio.to_thread(t.generate_audio,
result) # 假设 TTS 类有一个 generate_audio 方法返回音频数据
print(f"音频时长: {duration}") print(f"音频时长: {duration}")
# 将音频数据直接上传到 OSS # 将音频数据直接上传到 OSS
@ -160,13 +162,14 @@ async def reply(session_id: str = Form(...), prompt: str = Form(...)):
url = 'https://ylt.oss-cn-hangzhou.aliyuncs.com/' + tts_file url = 'https://ylt.oss-cn-hangzhou.aliyuncs.com/' + tts_file
# 记录聊天数据到 MySQL # 记录聊天数据到 MySQL
await save_chat_to_mysql(app.state.mysql_pool, session_id, prompt, result,url,duration) await save_chat_to_mysql(app.state.mysql_pool, session_id, prompt, result, url, duration)
logger.info("用户输入和大模型反馈已记录到 MySQL 数据库。") logger.info("用户输入和大模型反馈已记录到 MySQL 数据库。")
return { return {
"success": True, "success": True,
"url": url, "url": url,
"search_time": end_time - start_time, # 返回查询耗时 "search_time": end_time - start_time, # 返回查询耗时
"duration": duration, # 返回大模型的回复时长
"response": result # 返回大模型的回复 "response": result # 返回大模型的回复
} }
else: else:
@ -182,9 +185,9 @@ async def reply(session_id: str = Form(...), prompt: str = Form(...)):
# 获取聊天记录 # 获取聊天记录
@app.get("/get_chat_log") @app.get("/get_chat_log")
async def get_chat_log( async def get_chat_log(
session_id: str, session_id: str,
page: int = Query(default=1, ge=1, description="当前页码"), page: int = Query(default=1, ge=1, description="当前页码"),
page_size: int = Query(default=10, ge=1, le=100, description="每页记录数") page_size: int = Query(default=10, ge=1, le=100, description="每页记录数")
): ):
""" """
根据 session_id 查询聊天记录并按 id 降序分页 根据 session_id 查询聊天记录并按 id 降序分页
@ -194,14 +197,15 @@ async def get_chat_log(
:return: 分页数据 :return: 分页数据
""" """
try: try:
result = await get_chat_log_by_session(app.state.mysql_pool,session_id, page, page_size) result = await get_chat_log_by_session(app.state.mysql_pool, session_id, page, page_size)
return result return result
except Exception as e: except Exception as e:
logger.error(f"查询聊天记录失败: {str(e)}") logger.error(f"查询聊天记录失败: {str(e)}")
raise HTTPException(status_code=500, detail=f"查询聊天记录失败: {str(e)}") raise HTTPException(status_code=500, detail=f"查询聊天记录失败: {str(e)}")
# 运行 FastAPI 应用 # 运行 FastAPI 应用
if __name__ == "__main__": if __name__ == "__main__":
import uvicorn import uvicorn
uvicorn.run("Start:app", host="0.0.0.0", port=5600, workers=1) uvicorn.run("Start:app", host="0.0.0.0", port=5600, workers=1)

Loading…
Cancel
Save