diff --git a/api/magic.py b/api/magic.py new file mode 100644 index 0000000..7dbbdba --- /dev/null +++ b/api/magic.py @@ -0,0 +1,23 @@ +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)} + + +