from fastapi import APIRouter, Depends, FastAPI
from pydantic import BaseModel
from sqlalchemy.orm import Session

from core.database import get_db
from magic_llm.llm_func import llm_func



magic_router = APIRouter(tags=["智能数据库接口"])


class MagicRequest(BaseModel):
    user_request: str


@magic_router.post("/magic")
def Fastapi_magic(req: MagicRequest, db: Session = Depends(get_db)):
    executor, code = llm_func(req.user_request)
    return {"generated_code": code, "result": executor(db)}