更新后台管理消息列表;

This commit is contained in:
karlcarlo
2025-12-15 17:44:35 +08:00
parent 41d2d893d3
commit a7b3454f61
22 changed files with 1620 additions and 356 deletions

View File

@@ -1,3 +1,6 @@
import os
import logging
from contextlib import asynccontextmanager
from pymongo import AsyncMongoClient
@@ -6,10 +9,16 @@ from app.models import models
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
import uvicorn
import os
from app.routes import auth_routes, wework_routes, glm_routes, message_routes
# 配置日志
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
datefmt="%Y-%m-%d %H:%M:%S",
)
logger = logging.getLogger(__name__)
@asynccontextmanager
async def lifespan(app: FastAPI):
@@ -24,27 +33,8 @@ async def lifespan(app: FastAPI):
) # 批量注册模型
print(f"🗄️ 数据库连接成功: {database_name}")
# 初始化GLM服务,确保默认角色存在
try:
from app.services.glm_service import get_glm_service
glm_service = get_glm_service()
await glm_service.ensure_default_role()
print(f"✅ GLM角色配置初始化完成")
except Exception as e:
print(f"⚠️ GLM角色配置初始化失败: {e}")
yield # 应用在此处进入正常运行状态,接收请求
# 关闭GLM服务
try:
from app.services.glm_service import close_glm_service
close_glm_service()
print(f"GLM服务已关闭")
except Exception as e:
print(f"关闭GLM服务时出错: {e}")
# 关闭数据库连接
await client.close()
print(f"数据库连接关闭")
@@ -85,50 +75,4 @@ async def root():
"service": "企业微信认证与会话存档服务",
"version": "1.0.0",
"description": "获取和管理企业微信access_token,会话存档功能,以及DeepSeek AI聊天服务",
"endpoints": {
"docs": "/api/docs",
"redoc": "/api/redoc",
"auth": {"token": "/api/auth/token", "token_info": "/api/auth/token/info"},
"wework": {
"health": "/api/wework/health",
"chat": "/api/wework/chat",
"media": "/api/wework/media",
"decrypt": "/api/wework/decrypt",
"config": "/api/wework/config",
},
"glm": {
"root": "/api/glm/",
"role": "/api/glm/role",
"chat": "/api/glm/chat",
"conversations": "/api/glm/conversations/{conversation_id}",
},
"messages": {
"load": "/api/messages/load",
"list": "/api/messages/",
"stats": "/api/messages/stats",
"detail": "/api/messages/{msgid}",
},
},
}
def main():
"""
主函数,启动FastAPI服务器
"""
# 从环境变量获取主机和端口,使用默认值
host = os.getenv("APP_HOST", "0.0.0.0")
port = int(os.getenv("APP_PORT", "8000"))
print("启动企业微信认证与会话存档服务...")
print(f"API文档地址: http://{host}:{port}/docs")
print(f"API根地址: http://{host}:{port}")
print("企业微信会话存档API: /api/wework/*")
print("DeepSeek AI聊天API: /api/glm/*")
print("消息管理API: /api/messages/*")
uvicorn.run("app.main:app", host=host, port=port, reload=True, log_level="info")
if __name__ == "__main__":
main()