from fastapi import Depends, APIRouter
|
from sqlalchemy.orm import Session
|
from starlette.responses import StreamingResponse
|
from app.api import get_current_user
|
from app.models import UserModel
|
from app.models.base_model import get_db
|
from app.models.v2.session_model import ChatDialogData
|
from app.service.v2.chat import service_chat_dialog
|
|
chat1_router = APIRouter()
|
|
|
@chat1_router.post("/chat_dialog")
|
async def api_chat_dialog(dialog: ChatDialogData, db: Session = Depends(get_db)): # current_user: UserModel = Depends(get_current_user)
|
return StreamingResponse(service_chat_dialog(db, dialog.chatId ,dialog.question, dialog.sessionId, 1), media_type="text/event-stream")
|