vrr/ProxyAI
2
0
Fork 0
mirror of https://github.com/carlrobertoh/ProxyAI.git synced 2026-05-11 04:50:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use string type model state

Browse source
This commit is contained in:
Carl-Robert Linnupuu 2023-03-21 23:46:57 +00:00
parent 1b312da9ab
commit 525c316739
9 changed files with 24 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/main/java/ee/carlrobert/codegpt/settings/SettingsComponent.java
View file

@ -36,7 +36,7 @@ public class SettingsComponent {
ChatCompletionModel.GPT_3_5_SNAPSHOT,
ChatCompletionModel.GPT_4
},
settings.textCompletionBaseModel);
ChatCompletionModel.findByCode(settings.chatCompletionBaseModel));
textCompletionBaseModelComboBox = new BaseModelComboBox(
new TextCompletionModel[] {
TextCompletionModel.DAVINCI,
@ -44,7 +44,7 @@ public class SettingsComponent {
TextCompletionModel.BABBAGE,
TextCompletionModel.ADA,
},
settings.textCompletionBaseModel);
TextCompletionModel.findByCode(settings.textCompletionBaseModel));
useChatCompletionRadioButton = new JBRadioButton("Use chat completion", settings.isChatCompletionOptionSelected);
useTextCompletionRadioButton = new JBRadioButton("Use text completion", settings.isTextCompletionOptionSelected);
mainPanel = FormBuilder.createFormBuilder()
@ -95,16 +95,16 @@ public class SettingsComponent {
return (TextCompletionModel) textCompletionBaseModelComboBox.getSelectedItem();
}
public void setTextCompletionBaseModel(CompletionModel model) {
textCompletionBaseModelComboBox.setSelectedItem(model);
public void setTextCompletionBaseModel(String model) {
textCompletionBaseModelComboBox.setSelectedItem(TextCompletionModel.valueOf(model));
}
public ChatCompletionModel getChatCompletionBaseModel() {
return (ChatCompletionModel) chatCompletionBaseModelComboBox.getSelectedItem();
}
public void setChatCompletionBaseModel(CompletionModel model) {
chatCompletionBaseModelComboBox.setSelectedItem(model);
public void setChatCompletionBaseModel(String model) {
chatCompletionBaseModelComboBox.setSelectedItem(ChatCompletionModel.valueOf(model));
}
private JPanel createMainSelectionForm() {