from fastapi import FastAPI
from dotenv import load_dotenv # 导入 load_dotenv

# 加载 .env 文件中的环境变量
load_dotenv()

from apps.app01 import App01
from apps.app02 import App02
from apps.airs_tasks import Tasks

# 初始化应用
app01_instance = App01()
app02_instance = App02()
tasks_instance = Tasks()

# 创建 FastAPI 应用
app = FastAPI()

# 挂载 MCP 应用的 SSE 端点
# 使用 Mount 来指定路径前缀
app.mount("/app01", app=app01_instance.mcp.sse_app())
app.mount("/app02", app=app02_instance.mcp.sse_app())
app.mount("/tasks", app=tasks_instance.mcp.sse_app())

# 根路径，用于测试 FastAPI 是否正常工作
@app.get("/")
async def read_root():
    return {"message": "FastAPI is running"}