chat-ai-api/main.py

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from app.routers import users, chat, chat_tools, ai_config, moderation

# 创建FastAPI应用实例
app = FastAPI(title="聊天机器人", version="1.0.0", description="基于fastapi+VUE的聊天机器人")

# 添加CORS中间件，允许前端跨域访问
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # 生产环境建议指定具体域名
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 包含用户路由模块
# prefix: 为该路由模块添加URL前缀，所有用户相关路由都会以/users开头
# tags: 在API文档中用于分组显示，便于组织和查看
app.include_router(users.router, prefix="/main/users", tags=["用户管理"])
app.include_router(chat.router, prefix="/main/chat", tags=["聊天管理"])
app.include_router(chat_tools.router, prefix="/main/chatTools", tags=["AI工具管理"])
app.include_router(ai_config.router, prefix="/main/aiConfig", tags=["AI配置管理"])
app.include_router(moderation.router, prefix="/main/moderation", tags=["内容审核"])



# ==================== 应用启动配置 ====================

if __name__ == "__main__":
    import uvicorn

    # 打印启动信息
    print("=" * 50)
    print("聊天机器人服务启动中...")
    print("=" * 50)
    print(f"Web界面: http://localhost:8000")
    print(f"API文档: http://localhost:8000/docs")
    print("=" * 50)

    # 启动服务器
    uvicorn.run(
        "main:app",  # 应用模块路径
        host="0.0.0.0",  # 监听所有网络接口
        port=8000,  # 端口号
        # reload=True,  # 开发模式热重载
        workers=2,
        log_level="info"  # 日志级别
    )