from fastapi import APIRouter
from pydantic import BaseModel

from services.ollama_client import ask_ollama

router = APIRouter(prefix="/generate")

class PromptRequest(BaseModel):
    prompt: str

@router.post("/")
async def generate(req: PromptRequest):
    response = ask_ollama(req.prompt)

    return {
        "response": response
    }