j-js commited on
Commit
664a146
·
verified ·
1 Parent(s): 2ea6ddc

Create models.py

Browse files
Files changed (1) hide show
  1. models.py +42 -0
models.py ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from __future__ import annotations
2
+
3
+ from dataclasses import dataclass
4
+ from typing import Any, Dict, List, Optional
5
+
6
+ from pydantic import BaseModel
7
+
8
+
9
+ class ChatRequest(BaseModel):
10
+ message: Optional[str] = None
11
+ prompt: Optional[str] = None
12
+ query: Optional[str] = None
13
+ text: Optional[str] = None
14
+ user_message: Optional[str] = None
15
+
16
+ tone: Optional[float] = 0.5
17
+ verbosity: Optional[float] = 0.5
18
+ transparency: Optional[float] = 0.5
19
+
20
+ help_mode: Optional[str] = "answer"
21
+ history: Optional[List[Dict[str, Any]]] = None
22
+
23
+
24
+ @dataclass
25
+ class SolverResult:
26
+ reply: str
27
+ domain: str
28
+ solved: bool
29
+ help_mode: str
30
+ answer_letter: Optional[str] = None
31
+ answer_value: Optional[str] = None
32
+
33
+
34
+ @dataclass
35
+ class ParsedContext:
36
+ raw_message: str
37
+ visible_user_text: str
38
+ full_context_text: str
39
+ question_text: str
40
+ options_text: str
41
+ combined_question_block: str
42
+ recent_conversation: str