from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session

from config.database import get_db
from schema.agent import ReportAgentRequest
from service import agent_service

router = APIRouter(prefix="/agent", tags=["Agent"])


@router.post("/tip/generate")
def generate_article(db: Session = Depends(get_db)):
    return agent_service.query_article_agent(db)


@router.get("/tip/latest")
def latest_article(db: Session = Depends(get_db)):
    return agent_service.get_latest_article(db)


@router.post("/recipe/generate")
def generate_recipe(db: Session = Depends(get_db)):
    return agent_service.query_recipe_agent(db)


@router.get("/recipe/latest")
def latest_recipe(db: Session = Depends(get_db)):
    return agent_service.get_latest_recipe(db)


@router.post("/report")
def query_report(req: ReportAgentRequest):
    return agent_service.query_report_agent(req)