Spaces:
Running
Running
""" | |
/************************************************************************* | |
* | |
* CONFIDENTIAL | |
* __________________ | |
* | |
* Copyright (2023-2025) AI Labs, IronOne Technologies, LLC | |
* All Rights Reserved | |
* | |
* Author : Theekshana Samaradiwakara | |
* Description :Python Backend API to chat with private data | |
* CreatedDate : 14/11/2023 | |
* LastModifiedDate : 15/10/2024 | |
*************************************************************************/ | |
""" | |
from typing import Optional, List, Any, Dict | |
from pydantic import BaseModel | |
class LoginRequest(BaseModel): | |
username: str | |
password: str | |
# model for front end session | |
# Output model (UserModel response) | |
class UserModel(BaseModel): | |
userId: int | |
firstName: str | |
lastName: str | |
userName: str | |
token: str | |
class UserQuery(BaseModel): | |
content: str | |
userId: int | |
aiModel: int = "default" | |
class Document(BaseModel): | |
name: Optional[str] | |
page_content: str | |
metadata: Dict[str, Any] | |
class ResponseModel(BaseModel): | |
question: str | |
answer: str | |
source_documents: List[Document] = None | |
# class Feedback(BaseModel): | |
# """ | |
# Schema for collecting feedback from the user. | |
# It includes the question, bot response, and user feedback. | |
# """ | |
# question: str | |
# botResponse: str | |
# userFeedback: str | |
# feedback: str | |