File size: 1,146 Bytes
13ba451
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
from pydantic import BaseModel
from typing import List, Union, Optional

class DataAnswer(BaseModel):
    answer: str

class DataAnswer1(BaseModel):
    id: int
    answer: str
    data_relevant: List[str]
    sources: List[str]

class FileMetadata(BaseModel):
    source: str

class FileResponse(BaseModel):
    page_content: str
    metadata: FileMetadata
    type: str

class DataExtractFile(BaseModel):
   text_all: Union[List[FileResponse], None, str]

class Message(BaseModel):
    message: str
   
class CheckModel(BaseModel):
    check: bool

class ResponseQuery2Upgrade(BaseModel):
    status: int
    data: DataAnswer

class GenerateQuestion(BaseModel):
    question: Union[List[str], bool]

class ResponseGenerateQuestion(BaseModel):
    status: int
    data: GenerateQuestion

class ResponseQuery2UpgradeOld(BaseModel):
    status: int
    data: DataAnswer1

class ResponseExtractFile(BaseModel):
    status: int
    data: DataExtractFile

class ResponseDeleteChat(BaseModel):
    status: int
    data: Message
    
class ReponseError(BaseModel):
    status: int
    data: Message