diff --git a/src/main/kotlin/com/github/simiacryptus/aicoder/actions/dev/CodeChatServer.kt b/src/main/kotlin/com/github/simiacryptus/aicoder/actions/dev/CodeChatServer.kt
index b3bac850..dc3a5226 100644
--- a/src/main/kotlin/com/github/simiacryptus/aicoder/actions/dev/CodeChatServer.kt
+++ b/src/main/kotlin/com/github/simiacryptus/aicoder/actions/dev/CodeChatServer.kt
@@ -5,55 +5,42 @@ import com.intellij.openapi.project.Project
import com.simiacryptus.openai.OpenAIClient
import com.simiacryptus.skyenet.sessions.*
import com.simiacryptus.skyenet.util.ClasspathResource
-import com.simiacryptus.skyenet.util.MarkdownUtil
import org.eclipse.jetty.util.resource.Resource
-import org.intellij.lang.annotations.Language
class CodeChatServer(
val project: Project,
val language: String,
val codeSelection: String,
val api: OpenAIClient,
+ resourceBase: String = "codeChat",
) : ChatApplicationBase(
applicationName = "Code Chat",
+ resourceBase = resourceBase,
) {
- private val rootOperationID = (0..5).map { ('a'..'z').random() }.joinToString("")
- private var rootMessageTrail: String = ""
-
- override fun newSession(sessionId: String): ChatSession {
- val newSession = ChatSession(
- sessionId = sessionId,
- parent = this@CodeChatServer,
- model = AppSettingsState.instance.defaultChatModel(),
- api = api,
- visiblePrompt = """
- Code:
-
${htmlEscape(codeSelection)}
-
Code:
-
${htmlEscape(codeSelection)}
-
Code:
+ |
${htmlEscape(codeSelection)}
+ |
+
+
diff --git a/src/main/resources/codeChat/index.html b/src/main/resources/codeChat/index.html
new file mode 100644
index 00000000..79979eb8
--- /dev/null
+++ b/src/main/resources/codeChat/index.html
@@ -0,0 +1,37 @@
+
+
+
+ WebSocket Client
+
+
Disconnected. Attempting to reconnect...
+
" + text + "
";
+ Prism.highlightAll();
+ } catch (error) {
+ console.error('Error fetching data:', error);
+ }
+}
+
+let messageVersions = {};
+
+function onWebSocketText(event) {
+ console.log('WebSocket message:', event);
+ const messagesDiv = document.getElementById('messages');
+
+ // Parse message e.g. "id,version,content"
+ const firstCommaIndex = event.data.indexOf(',');
+ const secondCommaIndex = event.data.indexOf(',', firstCommaIndex + 1);
+ const messageId = event.data.substring(0, firstCommaIndex);
+ const messageVersion = event.data.substring(firstCommaIndex + 1, secondCommaIndex);
+ const messageContent = event.data.substring(secondCommaIndex + 1);
+ // If messageVersion isn't more than the version for the messageId using the version map, then ignore the message
+ if (messageVersion <= (messageVersions[messageId] || 0)) {
+ console.log("Ignoring message with id " + messageId + " and version " + messageVersion);
+ return;
+ } else {
+ messageVersions[messageId] = messageVersion;
+ }
+
+ let messageDiv = document.getElementById(messageId);
+
+ if (messageDiv) {
+ messageDiv.innerHTML = messageContent;
+ } else {
+ messageDiv = document.createElement('div');
+ messageDiv.className = 'message message-container'; // Add the message-container class
+ messageDiv.id = messageId;
+ messageDiv.innerHTML = messageContent;
+ messagesDiv.appendChild(messageDiv);
+ }
+
+ messagesDiv.scrollTop = messagesDiv.scrollHeight;
+ Prism.highlightAll();
+}
+
+document.addEventListener('DOMContentLoaded', () => {
+
+ document.querySelector('.close').addEventListener('click', closeModal);
+
+ window.addEventListener('click', (event) => {
+ if (event.target === document.getElementById('modal')) {
+ closeModal();
+ }
+ });
+
+ const form = document.getElementById('form');
+ const messageInput = document.getElementById('message');
+
+ form.addEventListener('submit', (event) => {
+ event.preventDefault();
+ send(messageInput.value);
+ messageInput.value = '';
+ });
+
+ messageInput.addEventListener('keydown', (event) => {
+ if (event.key === 'Enter' && !event.shiftKey) {
+ event.preventDefault();
+ form.dispatchEvent(new Event('submit'));
+ }
+ });
+
+ connect(undefined, onWebSocketText);
+
+ document.body.addEventListener('click', (event) => {
+ const target = event.target;
+ if (target.classList.contains('play-button')) {
+ const messageId = target.getAttribute('data-id');
+ send('!' + messageId + ',run');
+ } else if (target.classList.contains('regen-button')) {
+ const messageId = target.getAttribute('data-id');
+ send('!' + messageId + ',regen');
+ } else if (target.classList.contains('cancel-button')) {
+ const messageId = target.getAttribute('data-id');
+ send('!' + messageId + ',stop');
+ } else if (target.classList.contains('href-link')) {
+ const messageId = target.getAttribute('data-id');
+ send('!' + messageId + ',link');
+ } else if (target.classList.contains('text-submit-button')) {
+ const messageId = target.getAttribute('data-id');
+ const text = document.querySelector('.reply-input[data-id="' + messageId + '"]').value;
+ send('!' + messageId + ',userTxt,' + text);
+ }
+ });
+
+ fetch('appInfo')
+ .then(response => {
+ if (!response.ok) {
+ throw new Error('Network response was not ok');
+ }
+ return response.json();
+ })
+ .then(data => {
+ if (data.applicationName) {
+ document.title = data.applicationName;
+ }
+ })
+ .catch(error => {
+ console.error('There was a problem with the fetch operation:', error);
+ });
+
+});
+