24 lines
514 B
Python
24 lines
514 B
Python
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)}
|
|
|
|
|
|
|