Introduce a new initialize-chat WebSocket message type for single chat initialization, ensuring backward compatibility with older browser extension versions

Author: robertpiosik

packages/browser/src/background/message-handler.ts

@@ -1,6 +1,7 @@
 import {
   WebSocketMessage,
   InitializeChatsMessage,
+  InitializeChatMessage,
   ApplyChatResponseMessage
 } from '@shared/types/websocket-message'
 import browser from 'webextension-polyfill'
@@ -9,7 +10,6 @@ import { is_message } from '@/utils/is-message'
 import { GetTabDataResponse } from '@/types/responses'
 import { image_url_to_base64 } from '@/utils/image-url-to-base64'
 
-// Queue to manage multiple chat initialization
 interface ChatQueueItem {
   message: InitializeChatsMessage
   remaining_chats: number
@@ -26,6 +26,8 @@ const CHAT_INITIALIZATION_TIMEOUT = 5000
 export const handle_messages = (message: WebSocketMessage) => {
   if (message.action == 'initialize-chats') {
     handle_initialize_chats_message(message as InitializeChatsMessage)
+  } else if (message.action == 'initialize-chat') {
+    handle_initialize_chat_message(message as InitializeChatMessage)
   }
 }
 
@@ -140,6 +142,7 @@ const start_processing = async () => {
   }
 }
 
+// Will be deprecated
 const handle_initialize_chats_message = async (
   message: InitializeChatsMessage
 ) => {
@@ -156,6 +159,28 @@ const handle_initialize_chats_message = async (
   }
 }
 
+const handle_initialize_chat_message = async (
+  message: InitializeChatMessage
+) => {
+  const chat_config = {
+    url: message.url,
+    model: message.model,
+    temperature: message.temperature,
+    top_p: message.top_p,
+    system_instructions: message.system_instructions,
+    options: message.options
+  }
+
+  const chats_message: InitializeChatsMessage = {
+    action: 'initialize-chats',
+    text: message.text,
+    chats: [chat_config],
+    client_id: message.client_id
+  }
+
+  handle_initialize_chats_message(chats_message)
+}
+
 const handle_chat_initialized = async () => {
   // Process the next chat in the queue if one exists
   if (chat_queue.length > 0) {

packages/browser/src/content-scripts/send-prompt-content-script/chatbots/huggingchat.ts

@@ -9,6 +9,9 @@ import {
 import { Message } from '@/types/messages'
 
 export const huggingchat: Chatbot = {
+  wait_until_ready: async () => {
+    await new Promise((resolve) => setTimeout(resolve, 500))
+  },
   inject_apply_response_button: (client_id: number) => {
     const debounced_add_buttons = debounce((params: { footer: Element }) => {
       const apply_response_button_text = 'Apply response with CWC'

packages/browser/src/manifest.json

@@ -3,8 +3,8 @@
   "author": "Robert Piosik",
   "name": "Code Web Chat Connector",
   "short_name": "CWC",
-  "description": "Initialize any web chat with your code",
-  "version": "1.1.0",
+  "description": "Initialize chatbots and apply responses.",
+  "version": "1.2.0",
   "homepage_url": "https://github.com/robertpiosik/CodeWebChat",
   "icons": {
     "16": "icons/icon-16.png",

packages/shared/src/types/websocket-message.ts

@@ -7,13 +7,26 @@ export type Chat = {
   options?: string[]
 }
 
+// Deprecated, used by browser clients older than 1.2.0
 export type InitializeChatsMessage = {
   action: 'initialize-chats'
   text: string
   chats: Chat[]
   client_id: number // Client ID to identify which editor sent this message
 }
 
+export type InitializeChatMessage = {
+  action: 'initialize-chat'
+  text: string
+  url: string
+  client_id: number // Client ID to identify which editor sent this message
+  model?: string
+  temperature?: number
+  top_p?: number
+  system_instructions?: string
+  options?: string[]
+}
+
 export type Website = {
   url: string
   title: string
@@ -43,6 +56,7 @@ export type ClientIdAssignmentMessage = {
 
 export type WebSocketMessage =
   | InitializeChatsMessage
+  | InitializeChatMessage
   | UpdateSavedWebsitesMessage
   | BrowserConnectionStatusMessage
   | ClientIdAssignmentMessage

packages/vscode/package.json

@@ -1,8 +1,8 @@
 {
   "name": "gemini-coder",
   "displayName": "Code Web Chat (CWC)",
-  "description": "Initialize any web chat with your code",
-  "version": "1.53.0",
+  "description": "Initialize chatbots and apply responses",
+  "version": "1.54.0",
   "scripts": {
     "build": "npx vsce package --no-dependencies",
     "vscode:prepublish": "npm run compile",

packages/vscode/src/services/websocket-manager.ts

@@ -4,7 +4,7 @@ import * as child_process from 'child_process'
 import * as path from 'path'
 import * as net from 'net'
 import {
-  InitializeChatsMessage,
+  InitializeChatMessage,
   UpdateSavedWebsitesMessage
 } from '@shared/types/websocket-message'
 import { CHATBOTS } from '@shared/constants/chatbots'
@@ -296,19 +296,15 @@ export class WebSocketManager {
         url = chatbot.url
       }
 
-      const message: InitializeChatsMessage = {
-        action: 'initialize-chats',
+      const message: InitializeChatMessage = {
+        action: 'initialize-chat',
         text: chat.text,
-        chats: [
-          {
-            url,
-            model: preset.model,
-            temperature: preset.temperature,
-            top_p: preset.top_p,
-            system_instructions: preset.systemInstructions,
-            options: preset.options
-          }
-        ],
+        url,
+        model: preset.model,
+        temperature: preset.temperature,
+        top_p: preset.top_p,
+        system_instructions: preset.systemInstructions,
+        options: preset.options,
         client_id: this.client_id || 0 // 0 is a temporary fallback and should be removed few weeks from 28.03.25
       }
 
@@ -342,19 +338,15 @@ export class WebSocketManager {
       url = chatbot.url
     }
 
-    const message: InitializeChatsMessage = {
-      action: 'initialize-chats',
+    const message: InitializeChatMessage = {
+      action: 'initialize-chat',
       text: instruction,
-      chats: [
-        {
-          url,
-          model: preset.model,
-          temperature: preset.temperature,
-          top_p: preset.top_p,
-          system_instructions: preset.system_instructions,
-          options: preset.options
-        }
-      ],
+      url,
+      model: preset.model,
+      temperature: preset.temperature,
+      top_p: preset.top_p,
+      system_instructions: preset.system_instructions,
+      options: preset.options,
       client_id: this.client_id || 0 // 0 is a temporary fallback and should be removed few weeks from 28.03.25
     }
 

packages/vscode/src/services/websocket-server-process.ts

@@ -178,14 +178,39 @@ class WebSocketServer {
     const msg_string = message.toString()
     const msg_data = JSON.parse(msg_string)
 
-    // Handle different message types
-    if (msg_data.action == 'initialize-chats') {
+    if (msg_data.action == 'initialize-chat') {
       if (
         this.current_browser_client &&
         this.current_browser_client.ws.readyState === WebSocket.OPEN
       ) {
-        // Forward the message with client ID to browser client
-        this.current_browser_client.ws.send(msg_string)
+        const browser_version = this.current_browser_client.version
+        const needs_legacy_format = this._is_version_lower_than(
+          browser_version,
+          '1.2.0'
+        )
+
+        if (needs_legacy_format) {
+          // Convert InitializeChatMessage to InitializeChatsMessage for older clients
+          const legacy_message = {
+            action: 'initialize-chats',
+            text: msg_data.text,
+            chats: [
+              {
+                url: msg_data.url,
+                model: msg_data.model,
+                temperature: msg_data.temperature,
+                top_p: msg_data.top_p,
+                system_instructions: msg_data.system_instructions,
+                options: msg_data.options
+              }
+            ],
+            client_id: msg_data.client_id
+          }
+          this.current_browser_client.ws.send(JSON.stringify(legacy_message))
+        } else {
+          // Forward the message as-is for newer browser clients
+          this.current_browser_client.ws.send(msg_string)
+        }
       }
     } else if (msg_data.action == 'update-saved-websites') {
       // Store the updated websites
@@ -335,6 +360,24 @@ class WebSocketServer {
     }
     return false // Versions are equal or v1 is not newer
   }
+
+  private _is_version_lower_than(
+    version: string,
+    target_version: string
+  ): boolean {
+    if (version === 'unknown') return true // Assume older version if unknown
+
+    const parts1 = version.split('.').map(Number)
+    const parts2 = target_version.split('.').map(Number)
+
+    for (let i = 0; i < Math.max(parts1.length, parts2.length); i++) {
+      const p1 = parts1[i] || 0
+      const p2 = parts2[i] || 0
+      if (p1 < p2) return true
+      if (p1 > p2) return false
+    }
+    return false // Versions are equal
+  }
 }
 
 const server = new WebSocketServer()