234 lines
7.8 KiB
Python
234 lines
7.8 KiB
Python
"""
|
||
低代码智能体平台 - FastAPI 主应用
|
||
"""
|
||
import logging
|
||
from fastapi import FastAPI, Request
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
from fastapi.exceptions import RequestValidationError
|
||
from sqlalchemy.exc import SQLAlchemyError
|
||
from app.core.config import settings
|
||
from app.core.error_handler import (
|
||
validation_exception_handler,
|
||
api_exception_handler,
|
||
sqlalchemy_exception_handler,
|
||
general_exception_handler
|
||
)
|
||
from app.core.exceptions import BaseAPIException
|
||
from app.core.database import init_db
|
||
|
||
# 配置日志
|
||
logging.basicConfig(
|
||
level=logging.INFO,
|
||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||
)
|
||
logger = logging.getLogger(__name__)
|
||
|
||
app = FastAPI(
|
||
title=settings.APP_NAME,
|
||
version=settings.APP_VERSION,
|
||
description="""
|
||
## 低代码智能体平台 API
|
||
|
||
一个支持可视化工作流设计和智能Agent配置的低代码平台。
|
||
|
||
### 主要功能
|
||
|
||
* **用户认证** - 用户注册、登录、JWT认证
|
||
* **工作流管理** - 工作流的创建、读取、更新、删除、执行
|
||
* **工作流版本管理** - 版本保存、版本列表、版本回滚
|
||
* **执行管理** - 工作流执行、执行记录查询、执行状态监控
|
||
* **执行日志** - 详细的执行日志记录和查询
|
||
* **数据源管理** - 多种数据源的连接和管理
|
||
* **WebSocket实时推送** - 执行状态实时更新
|
||
|
||
### 认证方式
|
||
|
||
大部分API需要JWT认证。请先通过 `/api/v1/auth/login` 获取token，然后在请求头中添加：
|
||
|
||
```
|
||
Authorization: Bearer <your_token>
|
||
```
|
||
|
||
### API版本
|
||
|
||
当前API版本：v1
|
||
|
||
### 文档
|
||
|
||
* **Swagger UI**: `/docs` - 交互式API文档
|
||
* **ReDoc**: `/redoc` - 可读性更好的API文档
|
||
""",
|
||
docs_url="/docs",
|
||
redoc_url="/redoc",
|
||
openapi_tags=[
|
||
{
|
||
"name": "auth",
|
||
"description": "用户认证相关API，包括注册、登录、获取用户信息等。"
|
||
},
|
||
{
|
||
"name": "workflows",
|
||
"description": "工作流管理API，包括工作流的CRUD操作、执行、版本管理等。"
|
||
},
|
||
{
|
||
"name": "executions",
|
||
"description": "执行管理API，包括执行记录的创建、查询、状态获取等。"
|
||
},
|
||
{
|
||
"name": "execution-logs",
|
||
"description": "执行日志API，包括日志查询、日志统计等。"
|
||
},
|
||
{
|
||
"name": "data-sources",
|
||
"description": "数据源管理API，包括数据源的CRUD操作、连接测试、数据查询等。"
|
||
},
|
||
{
|
||
"name": "websocket",
|
||
"description": "WebSocket API，用于实时推送执行状态。"
|
||
}
|
||
]
|
||
)
|
||
|
||
# CORS 配置
|
||
cors_origins = [origin.strip() for origin in settings.CORS_ORIGINS.split(",")]
|
||
# 添加调试日志
|
||
logger.info(f"CORS允许的源: {cors_origins}")
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=cors_origins,
|
||
allow_credentials=True,
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
expose_headers=["*"],
|
||
)
|
||
|
||
# 注册全局异常处理器
|
||
app.add_exception_handler(RequestValidationError, validation_exception_handler)
|
||
app.add_exception_handler(BaseAPIException, api_exception_handler)
|
||
app.add_exception_handler(SQLAlchemyError, sqlalchemy_exception_handler)
|
||
app.add_exception_handler(Exception, general_exception_handler)
|
||
|
||
# 请求日志中间件
|
||
@app.middleware("http")
|
||
async def log_requests(request: Request, call_next):
|
||
"""记录请求日志"""
|
||
import time
|
||
start_time = time.time()
|
||
|
||
# 记录请求
|
||
logger.info(f"{request.method} {request.url.path} - 客户端: {request.client.host if request.client else 'unknown'}")
|
||
|
||
try:
|
||
response = await call_next(request)
|
||
process_time = time.time() - start_time
|
||
|
||
# 记录响应
|
||
logger.info(
|
||
f"{request.method} {request.url.path} - "
|
||
f"状态码: {response.status_code} - "
|
||
f"耗时: {process_time:.3f}s"
|
||
)
|
||
|
||
return response
|
||
except Exception as e:
|
||
process_time = time.time() - start_time
|
||
logger.error(
|
||
f"{request.method} {request.url.path} - "
|
||
f"异常: {str(e)} - "
|
||
f"耗时: {process_time:.3f}s"
|
||
)
|
||
raise
|
||
|
||
@app.get("/")
|
||
async def root():
|
||
"""根路径"""
|
||
return {
|
||
"message": "欢迎使用低代码智能体平台 API",
|
||
"version": settings.APP_VERSION,
|
||
"docs": "/docs"
|
||
}
|
||
|
||
@app.get("/health")
|
||
async def health_check():
|
||
"""健康检查"""
|
||
return {"status": "healthy"}
|
||
|
||
# 应用启动时初始化数据库和工具
|
||
@app.on_event("startup")
|
||
async def startup_event():
|
||
"""应用启动事件"""
|
||
try:
|
||
logger.info("正在初始化数据库...")
|
||
init_db()
|
||
logger.info("数据库初始化完成")
|
||
except Exception as e:
|
||
logger.error(f"数据库初始化失败: {e}")
|
||
# 不抛出异常，允许应用继续启动
|
||
|
||
# 注册内置工具
|
||
try:
|
||
from app.services.tool_registry import tool_registry
|
||
from app.services.builtin_tools import (
|
||
http_request_tool,
|
||
file_read_tool,
|
||
file_write_tool,
|
||
text_analyze_tool,
|
||
datetime_tool,
|
||
math_calculate_tool,
|
||
system_info_tool,
|
||
json_process_tool,
|
||
database_query_tool,
|
||
adb_log_tool,
|
||
HTTP_REQUEST_SCHEMA,
|
||
FILE_READ_SCHEMA,
|
||
FILE_WRITE_SCHEMA,
|
||
TEXT_ANALYZE_SCHEMA,
|
||
DATETIME_SCHEMA,
|
||
MATH_CALCULATE_SCHEMA,
|
||
SYSTEM_INFO_SCHEMA,
|
||
JSON_PROCESS_SCHEMA,
|
||
DATABASE_QUERY_SCHEMA,
|
||
ADB_LOG_SCHEMA
|
||
)
|
||
|
||
tool_registry.register_builtin_tool("http_request", http_request_tool, HTTP_REQUEST_SCHEMA)
|
||
tool_registry.register_builtin_tool("file_read", file_read_tool, FILE_READ_SCHEMA)
|
||
tool_registry.register_builtin_tool("file_write", file_write_tool, FILE_WRITE_SCHEMA)
|
||
tool_registry.register_builtin_tool("text_analyze", text_analyze_tool, TEXT_ANALYZE_SCHEMA)
|
||
tool_registry.register_builtin_tool("datetime", datetime_tool, DATETIME_SCHEMA)
|
||
tool_registry.register_builtin_tool("math_calculate", math_calculate_tool, MATH_CALCULATE_SCHEMA)
|
||
tool_registry.register_builtin_tool("system_info", system_info_tool, SYSTEM_INFO_SCHEMA)
|
||
tool_registry.register_builtin_tool("json_process", json_process_tool, JSON_PROCESS_SCHEMA)
|
||
tool_registry.register_builtin_tool("database_query", database_query_tool, DATABASE_QUERY_SCHEMA)
|
||
tool_registry.register_builtin_tool("adb_log", adb_log_tool, ADB_LOG_SCHEMA)
|
||
|
||
logger.info("内置工具注册完成（共10个工具）")
|
||
except Exception as e:
|
||
logger.error(f"内置工具注册失败: {e}")
|
||
# 不抛出异常，允许应用继续启动
|
||
|
||
# 注册路由
|
||
from app.api import auth, workflows, executions, websocket, execution_logs, data_sources, agents, model_configs, webhooks, template_market, batch_operations, collaboration, permissions, monitoring, alert_rules, node_test, node_templates, tools
|
||
|
||
app.include_router(auth.router)
|
||
app.include_router(workflows.router)
|
||
app.include_router(executions.router)
|
||
app.include_router(websocket.router)
|
||
app.include_router(execution_logs.router)
|
||
app.include_router(data_sources.router)
|
||
app.include_router(agents.router)
|
||
app.include_router(model_configs.router)
|
||
app.include_router(webhooks.router)
|
||
app.include_router(template_market.router)
|
||
app.include_router(batch_operations.router)
|
||
app.include_router(collaboration.router)
|
||
app.include_router(permissions.router)
|
||
app.include_router(monitoring.router)
|
||
app.include_router(alert_rules.router)
|
||
app.include_router(node_test.router)
|
||
app.include_router(node_templates.router)
|
||
app.include_router(tools.router)
|
||
|
||
if __name__ == "__main__":
|
||
import uvicorn
|
||
uvicorn.run(app, host="0.0.0.0", port=8000)
|