forked from yaroslavyaroslav/OpenAI-sublime-text
-
Notifications
You must be signed in to change notification settings - Fork 0
/
assistant_settings.py
41 lines (37 loc) · 922 Bytes
/
assistant_settings.py
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
from dataclasses import dataclass
from enum import Enum
from typing import Optional
class PromptMode(Enum):
panel = "panel"
append = "append"
insert = "insert"
replace = "replace"
@dataclass
class AssistantSettings():
name: str
prompt_mode: PromptMode
url: Optional[str]
token: Optional[str]
chat_model: str
assistant_role: str
temperature: int
max_tokens: int
top_p: int
frequency_penalty: int
presence_penalty: int
placeholder: Optional[str]
DEFAULT_ASSISTANT_SETTINGS = {
"placeholder": None,
"url": None,
"token": None,
"temperature": 1,
"max_tokens": 2048,
"top_p": 1,
"frequency_penalty": 0,
"presence_penalty": 0,
}
class CommandMode(Enum):
refresh_output_panel = "refresh_output_panel"
create_new_tab = "create_new_tab"
reset_chat_history = "reset_chat_history"
chat_completion = "chat_completion"