Spaces:
Runtime error
Runtime error
| """Chat API endpoints.""" | |
| from fastapi import APIRouter, Depends, HTTPException, status | |
| from pydantic import BaseModel | |
| from sqlalchemy.orm import Session | |
| from app.database import get_db | |
| from app.auth import get_current_user | |
| from app.models import User | |
| from app.application.chat_service import ChatService | |
| from app.dependencies import get_chat_service | |
| from typing import Optional, List | |
| import logging | |
| logger = logging.getLogger(__name__) | |
| router = APIRouter(prefix="/chat", tags=["chat"]) | |
| class ChatRequest(BaseModel): | |
| message: str | |
| chat_id: Optional[str] = None | |
| # Scope the search to specific documents or a folder | |
| document_ids: Optional[List[str]] = None # Ask about specific files | |
| folder_id: Optional[str] = None # Ask about all files in a folder | |
| async def send_message( | |
| request: ChatRequest, | |
| chat_service: ChatService = Depends(get_chat_service), | |
| current_user: User = Depends(get_current_user) | |
| ): | |
| """Send a chat message, optionally scoped to specific documents or a folder.""" | |
| try: | |
| result = await chat_service.send_message( | |
| message=request.message, | |
| user=current_user, | |
| chat_id=request.chat_id, | |
| document_ids=request.document_ids, | |
| folder_id=request.folder_id | |
| ) | |
| return result | |
| except ValueError as e: | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=str(e)) | |
| except Exception as e: | |
| logger.error(f"Chat error: {e}") | |
| raise HTTPException( | |
| status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, | |
| detail=f"Failed to generate response: {str(e)}" | |
| ) | |
| async def get_chat_history( | |
| chat_service: ChatService = Depends(get_chat_service), | |
| current_user: User = Depends(get_current_user) | |
| ): | |
| """Get all chat sessions.""" | |
| return await chat_service.get_chat_history(current_user) | |
| async def get_chat_messages( | |
| chat_id: str, | |
| chat_service: ChatService = Depends(get_chat_service), | |
| current_user: User = Depends(get_current_user) | |
| ): | |
| """Get all messages in a chat.""" | |
| try: | |
| return await chat_service.get_chat_messages(chat_id, current_user) | |
| except ValueError as e: | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=str(e)) | |
| async def delete_chat( | |
| chat_id: str, | |
| chat_service: ChatService = Depends(get_chat_service), | |
| current_user: User = Depends(get_current_user) | |
| ): | |
| """Delete a chat.""" | |
| try: | |
| await chat_service.delete_chat(chat_id, current_user) | |
| return {"message": "Chat deleted successfully"} | |
| except ValueError as e: | |
| raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=str(e)) | |