Files
fastapi-sqlalchemy-asyncpg/app/api/ml.py
2025-11-14 10:31:17 +01:00

17 lines
443 B
Python

from typing import Annotated
from fastapi import APIRouter, Depends, Form
from fastapi.responses import StreamingResponse
from rotoger import get_logger
from app.services.llm import get_llm_service
logger = get_logger()
router = APIRouter()
@router.post("/chat/")
async def chat(prompt: Annotated[str, Form()], llm_service=Depends(get_llm_service)):
return StreamingResponse(llm_service.stream_chat(prompt), media_type="text/plain")