File size: 845 Bytes
3d00f61
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9c94aa7
3d00f61
 
 
e1ce9ca
9c94aa7
 
 
 
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
from pydantic import BaseModel, Field
from typing import List, Optional, Dict


'''
input_info:
{
    query:"",
    field_inputs:
    {
        "Reddit":["Pain point analysis", "Competitor analysis"],
        "Twitter":["Competitor analysis"],
    }
}
'''

'''
session_info:
{
    "Reddit":[{"Pain point analysis":-1}, {"Competitor analysis":-1}],
    "Twitter":[{"Competitor analysis":-1}],
}
'''
class InputInfoModel(BaseModel) :
    query: str = ""
    field_inputs: dict = {}

class UserSessionModel(BaseModel):
    user_id: Optional[int] = None
    input_info: Optional[dict] = Field(InputInfoModel)
    session_info: dict
    process_info: Optional[dict] = None
    session_completed: Optional[bool] = False
    is_deleted: Optional[bool] = False

class UserAllSessionResponseModel(BaseModel):
    msg: str
    data: List[UserSessionModel]