aaditkumar commited on
Commit
0c5751a
·
verified ·
1 Parent(s): ca994c1

Update app/models.py

Browse files
Files changed (1) hide show
  1. app/models.py +21 -21
app/models.py CHANGED
@@ -1,22 +1,22 @@
1
- from pydantic import BaseModel, Field
2
- from typing import List, Optional
3
-
4
- class ChatMessage(BaseModel):
5
- role: str
6
- content: str
7
-
8
- class ChatRequest(BaseModel):
9
- message: str = Field(..., min_length=1, max_length=32_000)
10
- session_id: Optional[str] = None
11
- tts: bool = False
12
-
13
- class ChatResponse(BaseModel):
14
- response: str
15
- session_id: str
16
-
17
- class ChatHistory(BaseModel):
18
- session_id: str
19
- messages: List[ChatMessage]
20
-
21
- class TTSRequest(BaseModel):
22
  text: str = Field(..., min_length=1, max_length=5000)
 
1
+ from pydantic import BaseModel, Field
2
+ from typing import List, Optional
3
+
4
+ class ChatMessage(BaseModel):
5
+ role: str
6
+ content: str
7
+
8
+ class ChatRequest(BaseModel):
9
+ message: str = Field(..., min_length=1, max_length=32_000)
10
+ session_id: Optional[str] = None
11
+ tts: bool = False
12
+
13
+ class ChatResponse(BaseModel):
14
+ response: str
15
+ session_id: str
16
+
17
+ class ChatHistory(BaseModel):
18
+ session_id: str
19
+ messages: List[ChatMessage]
20
+
21
+ class TTSRequest(BaseModel):
22
  text: str = Field(..., min_length=1, max_length=5000)