8000 complete the parameters management of "chat" model · ag-python-qt/pyqt-openai@4a6e886 · GitHub
[go: up one dir, main page]
Skip to content

Commit 4a6e886

Browse files
yjg30737yjg30737
committed
complete the parameters management of "chat" model
1 parent 19f5624 commit 4a6e886

File tree

2 files changed

+26
-8
lines changed

2 files changed

+26
-8
lines changed

‎pyqt_openai/openAiChatBotWidget.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,13 +180,16 @@ def __chat(self):
180180
{"role": "user", "content": self.__prompt.getContent()},
181181
],
182182
'temperature': temperature,
183-
'max_tokens': max_tokens,
184183
'top_p': top_p,
185184
'frequency_penalty': frequency_penalty,
186185
'presence_penalty': presence_penalty,
187186
'stream': stream,
188187
}
189188

189+
use_max_token = self.__settings_ini.value('use_max_token', type=bool)
190+
if use_max_token:
191+
openai_arg['max_tokens'] = max_tokens
192+
190193
if self.__leftSideBarWidget.isCurrentConvExists():
191194
pass
192195
else:

‎pyqt_openai/right_sidebar/chatPage.py

Lines changed: 22 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,10 @@ def __initVal(self):
3636
self.__settings_ini.setValue('presence_penalty', 0)
3737

3838
10000 # etc
39+
if not self.__settings_ini.contains('use_max_tokens'):
40+
self.__settings_ini.setValue('use_max_tokens', False)
3941
if not self.__settings_ini.contains('finish_reason'):
40-
self.__settings_ini.setValue('finish_reason', True)
42+
self.__settings_ini.setValue('finish_reason', False)
4143

4244
self.__stream = self.__settings_ini.value('stream', type=bool)
4345
self.__model = self.__settings_ini.value('model', type=str)
@@ -48,6 +50,7 @@ def __initVal(self):
4850
self.__frequency_penalty = self.__settings_ini.value('frequency_penalty', type=float)
4951
self.__presence_penalty = self.__settings_ini.value('presence_penalty', type=float)
5052

53+
self.__use_max_tokens = self.__settings_ini.value('use_max_tokens', type=bool)
5154
self.__finish_reason = self.__settings_ini.value('finish_reason', type=bool)
5255

5356
def __initUi(self):
@@ -71,11 +74,11 @@ def __initUi(self):
7174
temperatureSpinBox.setValue(self.__temperature)
7275
temperatureSpinBox.valueChanged.connect(self.__temperatureChanged)
7376

74-
maxTokensSpinBox = QSpinBox()
75-
maxTokensSpinBox.setRange(1, 2048)
76-
maxTokensSpinBox.setAccelerated(True)
77-
maxTokensSpinBox.setValue(self.__max_tokens)
78-
maxTokensSpinBox.valueChanged.connect(self.__maxTokensChanged)
77+
self.__maxTokensSpinBox = QSpinBox()
78+
self.__maxTokensSpinBox.setRange(1, 2048)
79+
self.__maxTokensSpinBox.setAccelerated(True)
80+
self.__maxTokensSpinBox.setValue(self.__max_tokens)
81+
self.__maxTokensSpinBox.valueChanged.connect(self.__maxTokensChanged)
7982

8083
toppSpinBox = QDoubleSpinBox()
8184
toppSpinBox.setRange(0, 1)
@@ -103,6 +106,12 @@ def __initUi(self):
103106
streamChkBox.toggled.connect(self.__streamChecked)
104107
streamChkBox.setText('Stream')
105108

109+
useMaxTokenChkBox = QCheckBox()
110+
useMaxTokenChkBox.toggled.connect(self.__useMaxChecked)
111+
useMaxTokenChkBox.setChecked(self.__use_max_tokens)
112+
useMaxTokenChkBox.setText('Use Max Tokens')
113+
self.__maxTokensSpinBox.setEnabled(self.__use_max_tokens)
114+
106115
finishReasonChkBox = QCheckBox('Show Finish Reason (working)')
107116
finishReasonChkBox.setChecked(self.__finish_reason)
108117
finishReasonChkBox.toggled.connect(self.__finishReasonChecked)
@@ -111,7 +120,7 @@ def __initUi(self):
111120
lay = QFormLayout()
112121

113122
lay.addRow('temperature', temperatureSpinBox)
114-
lay.addRow('maxTokens', maxTokensSpinBox)
123+
lay.addRow('maxTokens', self.__maxTokensSpinBox)
115124
lay.addRow('topp', toppSpinBox)
116125
lay.addRow('frequencyPenalty', frequencyPenaltySpinBox)
117126
lay.addRow('presencePenalty', presencePenaltySpinBox)
@@ -126,6 +135,7 @@ def __initUi(self):
126135
lay.addWidget(saveSystemBtn)
127136
lay.addWidget(modelCmbBox)
128137
lay.addWidget(streamChkBox)
138+
lay.addWidget(useMaxTokenChkBox)
129139
lay.addWidget(finishReasonChkBox)
130140
lay.addWidget(paramWidget)
131141
lay.setAlignment(Qt.AlignTop)
@@ -144,6 +154,11 @@ def __streamChecked(self, f):
144154
self.__stream = f
145155
self.__settings_ini.setValue('stream', f)
146156

157+
def __useMaxChecked(self, f):
158+
self.__use_max_tokens = f
159+
self.__settings_ini.setValue('use_max_tokens', f)
160+
self.__maxTokensSpinBox.setEnabled(f)
161+
147162
def __finishReasonChecked(self, f):
148163
self.__finish_reason = f
149164
self.__settings_ini.setValue('finish_reason', f)

0 commit comments

Comments
 (0)
0