diff --git a/AI/WxMini/Milvus/Config/MulvusConfig.py b/AI/WxMini/Milvus/Config/MulvusConfig.py index 1b89b234..15634e42 100644 --- a/AI/WxMini/Milvus/Config/MulvusConfig.py +++ b/AI/WxMini/Milvus/Config/MulvusConfig.py @@ -35,6 +35,8 @@ MYSQL_USER = "root" MYSQL_PASSWORD = "DsideaL147258369" MYSQL_DB_NAME = "ai_db" +# JWT密匙 +JWT_SECRET_KEY = "DsideaL4r5t6y7u" # ----------------下面的配置需要根据情况进行修改------------------------- ''' diff --git a/AI/WxMini/Milvus/Config/__pycache__/MulvusConfig.cpython-310.pyc b/AI/WxMini/Milvus/Config/__pycache__/MulvusConfig.cpython-310.pyc index 215d4d8f..3f2c9382 100644 Binary files a/AI/WxMini/Milvus/Config/__pycache__/MulvusConfig.cpython-310.pyc and b/AI/WxMini/Milvus/Config/__pycache__/MulvusConfig.cpython-310.pyc differ diff --git a/AI/WxMini/Start.py b/AI/WxMini/Start.py index 408726cb..740a4446 100644 --- a/AI/WxMini/Start.py +++ b/AI/WxMini/Start.py @@ -29,7 +29,7 @@ logger = logging.getLogger(__name__) # 配置 JWT SECRET_KEY = "DsideaL4r5t6y7u" # 替换为你的密钥 ALGORITHM = "HS256" -ACCESS_TOKEN_EXPIRE_MINUTES = 30 +ACCESS_TOKEN_EXPIRE_MINUTES = 24 * 60 * 30 # 一个月有效期 # 密码加密上下文 pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto") # OAuth2 密码模式 @@ -214,9 +214,13 @@ async def login( } } + # 与用户交流聊天 @app.post("/aichat/reply") -async def reply(person_id: str = Form(...), prompt: str = Form(...)): +async def reply(person_id: str = Form(...), + prompt: str = Form(...), + current_user: dict = Depends(get_current_user) + ): """ 接收用户输入的 prompt，调用大模型并返回结果 :param person_id: 用户会话 ID @@ -335,7 +339,9 @@ async def reply(person_id: str = Form(...), prompt: str = Form(...)): "url": url, "search_time": end_time - start_time, # 返回查询耗时 "duration": duration, # 返回大模型的回复时长 - "response": result # 返回大模型的回复 + "response": result, # 返回大模型的回复 + "login_name": current_user["login_name"], + "person_name": current_user["person_name"] } else: raise HTTPException(status_code=500, detail="大模型未返回有效结果") @@ -352,7 +358,8 @@ async def reply(person_id: str = Form(...), prompt: str = Form(...)): async def get_chat_log( person_id: str, page: int = Query(default=1, ge=1, description="当前页码（默认值为 1，但会动态计算为最后一页）"), - page_size: int = Query(default=10, ge=1, le=100, description="每页记录数") + page_size: int = Query(default=10, ge=1, le=100, description="每页记录数"), + current_user: dict = Depends(get_current_user) ): """ 获取指定会话的聊天记录，默认返回最新的记录（最后一页） @@ -362,7 +369,7 @@ async def get_chat_log( :return: 分页数据 """ # 调用 get_chat_log_by_session 方法 - result = await get_chat_log_by_session(app.state.mysql_pool, person_id, page, page_size) + result = await get_chat_log_by_session(app.state.mysql_pool, current_user, person_id, page, page_size) return result @@ -371,7 +378,8 @@ async def get_chat_log( async def get_risk_chat_logs( risk_flag: int = Query(..., description="风险标志（1 表示有风险，0 表示无风险）"), page: int = Query(default=1, ge=1, description="当前页码（默认值为 1）"), - page_size: int = Query(default=10, ge=1, le=100, description="每页记录数（默认值为 10，最大值为 100）") + page_size: int = Query(default=10, ge=1, le=100, description="每页记录数（默认值为 10，最大值为 100）"), + current_user: dict = Depends(get_current_user) ): """ 获取聊天记录，支持分页和风险标志过滤 @@ -396,14 +404,16 @@ async def get_risk_chat_logs( "total": total, "page": page, "page_size": page_size, - "logs": logs + "logs": logs, + "login_name": current_user["login_name"], + "person_name": current_user["person_name"] } } # 获取上传OSS的授权Token @app.get("/aichat/get_oss_upload_token") -async def get_oss_upload_token(): +async def get_oss_upload_token(current_user: dict = Depends(get_current_user)): # 获取 STS 临时凭证 sts_token = get_sts_token() return { @@ -414,10 +424,13 @@ async def get_oss_upload_token(): "access_key_secret": sts_token['AccessKeySecret'], "security_token": sts_token['SecurityToken'], "bucket_name": BUCKET_NAME, - "endpoint": ENDPOINT + "endpoint": ENDPOINT, + "login_name": current_user["login_name"], + "person_name": current_user["person_name"] } } + # 受保护的接口示例 @app.get("/aichat/protected-route") async def protected_route(current_user: dict = Depends(get_current_user)): @@ -435,6 +448,7 @@ async def protected_route(current_user: dict = Depends(get_current_user)): } } + # 运行 FastAPI 应用 if __name__ == "__main__": import uvicorn diff --git a/AI/WxMini/Utils/MySQLUtil.py b/AI/WxMini/Utils/MySQLUtil.py index 46704c25..4155b90d 100644 --- a/AI/WxMini/Utils/MySQLUtil.py +++ b/AI/WxMini/Utils/MySQLUtil.py @@ -47,8 +47,9 @@ async def truncate_chat_log(mysql_pool): from aiomysql import DictCursor + # 分页查询聊天记录 -async def get_chat_log_by_session(mysql_pool, person_id, page=1, page_size=10): +async def get_chat_log_by_session(mysql_pool, current_user, person_id, page=1, page_size=10): """ 根据 person_id 查询聊天记录，并按 id 降序分页 :param mysql_pool: MySQL 连接池 @@ -106,7 +107,9 @@ async def get_chat_log_by_session(mysql_pool, person_id, page=1, page_size=10): "total": total, "page": page, "page_size": page_size, - "total_pages": total_pages + "total_pages": total_pages, + "login_name": current_user["login_name"], + "person_name": current_user["person_name"] } @@ -167,6 +170,7 @@ async def get_user_by_login_name(mysql_pool, login_name: str) -> Optional[Dict]: columns = [column[0] for column in cursor.description] return dict(zip(columns, row)) + # 查询聊天记录 async def get_chat_logs_by_risk_flag(mysql_pool, risk_flag: int, offset: int, page_size: int) -> (List[Dict], int): """ @@ -208,4 +212,4 @@ async def get_chat_logs_by_risk_flag(mysql_pool, risk_flag: int, offset: int, pa columns = [column[0] for column in cursor.description] logs = [dict(zip(columns, row)) for row in rows] return logs, total - return [], 0 \ No newline at end of file + return [], 0 diff --git a/AI/WxMini/Utils/__pycache__/MySQLUtil.cpython-310.pyc b/AI/WxMini/Utils/__pycache__/MySQLUtil.cpython-310.pyc index eb556647..11f241b7 100644 Binary files a/AI/WxMini/Utils/__pycache__/MySQLUtil.cpython-310.pyc and b/AI/WxMini/Utils/__pycache__/MySQLUtil.cpython-310.pyc differ diff --git a/AI/WxMini/__pycache__/Start.cpython-310.pyc b/AI/WxMini/__pycache__/Start.cpython-310.pyc index b123c031..fddac2d0 100644 Binary files a/AI/WxMini/__pycache__/Start.cpython-310.pyc and b/AI/WxMini/__pycache__/Start.cpython-310.pyc differ