Reopen plugin's source code (1.10.8 → 2.0.5)

src/main/java/ee/carlrobert/codegpt/conversations/Conversation.java

@@ -0,0 +1,87 @@
+package ee.carlrobert.codegpt.conversations;
+
+import static java.util.stream.Collectors.toList;
+
+import ee.carlrobert.codegpt.conversations.message.Message;
+import ee.carlrobert.openai.client.ClientCode;
+import java.time.LocalDateTime;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.UUID;
+
+public class Conversation {
+
+  private UUID id;
+  private List<Message> messages = new ArrayList<>();
+  private ClientCode clientCode;
+  private String model;
+  private LocalDateTime createdOn;
+  private LocalDateTime updatedOn;
+  private boolean discardTokenLimit;
+
+  public UUID getId() {
+    return id;
+  }
+
+  public void setId(UUID id) {
+    this.id = id;
+  }
+
+  public List<Message> getMessages() {
+    return messages;
+  }
+
+  public void setMessages(List<Message> messages) {
+    this.messages = messages;
+  }
+
+  public ClientCode getClientCode() {
+    return clientCode;
+  }
+
+  public void setClientCode(ClientCode clientCode) {
+    this.clientCode = clientCode;
+  }
+
+  public void addMessage(Message message) {
+    messages.add(message);
+  }
+
+  public String getModel() {
+    return model;
+  }
+
+  public void setModel(String model) {
+    this.model = model;
+  }
+
+  public LocalDateTime getCreatedOn() {
+    return createdOn;
+  }
+
+  public void setCreatedOn(LocalDateTime createdOn) {
+    this.createdOn = createdOn;
+  }
+
+  public LocalDateTime getUpdatedOn() {
+    return updatedOn;
+  }
+
+  public void setUpdatedOn(LocalDateTime updatedOn) {
+    this.updatedOn = updatedOn;
+  }
+
+  public void discardTokenLimits() {
+    this.discardTokenLimit = true;
+  }
+
+  public boolean isDiscardTokenLimit() {
+    return discardTokenLimit;
+  }
+
+  public void removeMessage(UUID messageId) {
+    setMessages(messages.stream()
+        .filter(message -> !message.getId().equals(messageId))
+        .collect(toList()));
+  }
+}