1.4.4 - Replace model on conversation change, start new conversation on model change

This commit is contained in:
Carl-Robert Linnupuu 2023-03-13 00:55:10 +00:00
parent 483abe146b
commit 63020ba49d
13 changed files with 89 additions and 101 deletions

View file

@ -6,6 +6,7 @@ import com.intellij.openapi.components.State;
import com.intellij.openapi.components.Storage;
import com.intellij.util.xmlb.XmlSerializerUtil;
import com.intellij.util.xmlb.annotations.OptionTag;
import ee.carlrobert.codegpt.client.BaseModel;
import ee.carlrobert.codegpt.client.ClientCode;
import ee.carlrobert.codegpt.client.ClientFactory;
import ee.carlrobert.codegpt.ide.conversations.converter.ConversationConverter;
@ -57,12 +58,13 @@ public class ConversationsState implements PersistentStateComponent<Conversation
var conversation = new Conversation();
conversation.setId(UUID.randomUUID());
conversation.setClientCode(clientCode);
if (!settings.isChatGPTOptionSelected) {
if (settings.isChatCompletionOptionSelected) {
conversation.setModel(settings.chatCompletionBaseModel);
} else {
conversation.setModel(settings.textCompletionBaseModel);
}
if (settings.isChatGPTOptionSelected) {
conversation.setModel(BaseModel.UNOFFICIAL_CHATGPT);
} else {
conversation.setModel(
settings.isChatCompletionOptionSelected ?
settings.chatCompletionBaseModel :
settings.textCompletionBaseModel);
}
conversation.setCreatedOn(LocalDateTime.now());
conversation.setUpdatedOn(LocalDateTime.now());