vrr/ProxyAI
2
0
Fork 0
mirror of https://github.com/carlrobertoh/ProxyAI.git synced 2026-05-19 07:54:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Switch to openai-client, add conversation history empty label, remove unofficial reverse proxy (closes #43)

Browse source
This commit is contained in:
Carl-Robert Linnupuu 2023-03-21 00:21:07 +00:00
parent 33a5e520f2
commit 57e1095dd1
85 changed files with 619 additions and 1828 deletions
Download patch file Download diff file Expand all files Collapse all files

115
src/main/java/ee/carlrobert/codegpt/toolwindow/conversations/ConversationsToolWindow.java Normal file
View file

@ -0,0 +1,115 @@
package ee.carlrobert.codegpt.toolwindow.conversations;
import com.intellij.openapi.actionSystem.ActionManager;
import com.intellij.openapi.actionSystem.ActionToolbar;
import com.intellij.openapi.actionSystem.DefaultActionGroup;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.roots.ui.componentsList.components.ScrollablePanel;
import com.intellij.openapi.ui.SimpleToolWindowPanel;
import com.intellij.ui.components.JBScrollPane;
import com.intellij.util.ui.JBFont;
import com.intellij.util.ui.JBUI;
import ee.carlrobert.codegpt.conversations.Conversation;
import ee.carlrobert.codegpt.conversations.ConversationsState;
import ee.carlrobert.codegpt.settings.SettingsState;
import ee.carlrobert.codegpt.toolwindow.ContentManagerService;
import ee.carlrobert.codegpt.toolwindow.ToolWindowService;
import ee.carlrobert.codegpt.toolwindow.conversations.actions.ClearAllConversationsAction;
import ee.carlrobert.codegpt.toolwindow.conversations.actions.DeleteConversationAction;
import ee.carlrobert.codegpt.toolwindow.conversations.actions.MoveDownAction;
import ee.carlrobert.codegpt.toolwindow.conversations.actions.MoveUpAction;
import ee.carlrobert.openai.client.ClientCode;
import javax.swing.BoxLayout;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.ScrollPaneConstants;
import org.jetbrains.annotations.NotNull;
public class ConversationsToolWindow {
private final Project project;
private JPanel conversationsToolWindowContent;
private JScrollPane scrollPane;
private ScrollablePanel scrollablePanel;
public ConversationsToolWindow(@NotNull Project project) {
this.project = project;
refresh();
}
public JPanel getContent() {
SimpleToolWindowPanel panel = new SimpleToolWindowPanel(true);
panel.setContent(conversationsToolWindowContent);
var actionGroup = new DefaultActionGroup("TOOLBAR_ACTION_GROUP", false);
actionGroup.add(new MoveDownAction(this::refresh));
actionGroup.add(new MoveUpAction(this::refresh));
actionGroup.addSeparator();
actionGroup.add(new DeleteConversationAction(this::refresh));
actionGroup.add(new ClearAllConversationsAction(this::refresh));
ActionToolbar actionToolbar = ActionManager.getInstance()
.createActionToolbar("NAVIGATION_BAR_TOOLBAR", actionGroup, true);
panel.setToolbar(actionToolbar.getComponent());
return panel;
}
public void refresh() {
scrollablePanel.removeAll();
var sortedConversations = ConversationsState.getInstance().getSortedConversations();
if (sortedConversations.isEmpty()) {
var emptyLabel = new JLabel("No conversations exist.");
emptyLabel.setFont(JBFont.h2());
emptyLabel.setBorder(JBUI.Borders.empty(8));
scrollablePanel.add(emptyLabel);
} else {
sortedConversations.forEach(this::addContent);
}
scrollablePanel.revalidate();
scrollablePanel.repaint();
}
private void addContent(Conversation conversation) {
var mainPanel = new RootConversationPanel(() -> {
ConversationsState.getInstance().setCurrentConversation(conversation);
changeSettings(conversation);
project.getService(ContentManagerService.class).displayChatTab(project);
project.getService(ToolWindowService.class)
.getChatToolWindow()
.displayConversation(conversation);
});
mainPanel.setBackground(conversationsToolWindowContent.getBackground());
var currentConversation = ConversationsState.getCurrentConversation();
var isSelected = currentConversation != null && currentConversation.getId().equals(conversation.getId());
mainPanel.add(new ConversationPanel(conversation, isSelected));
scrollablePanel.add(mainPanel);
}
private void changeSettings(Conversation conversation) {
var settings = SettingsState.getInstance();
var isChatCompletions = ClientCode.CHAT_COMPLETION.equals(conversation.getClientCode());
if (isChatCompletions) {
settings.chatCompletionBaseModel = conversation.getModel();
} else {
settings.textCompletionBaseModel = conversation.getModel();
}
settings.isChatCompletionOptionSelected = isChatCompletions;
settings.isTextCompletionOptionSelected = !isChatCompletions;
}
private void createUIComponents() {
scrollablePanel = new ScrollablePanel();
scrollablePanel.setLayout(new BoxLayout(scrollablePanel, BoxLayout.Y_AXIS));
scrollPane = new JBScrollPane();
scrollPane.setHorizontalScrollBarPolicy(ScrollPaneConstants.HORIZONTAL_SCROLLBAR_NEVER);
scrollPane.setViewportView(scrollablePanel);
scrollPane.setBorder(null);
scrollPane.setViewportBorder(null);
}
}