Support the 'Temporary' chat option in ChatGPT, replacing 'Reason' and 'Search'

Author: robertpiosik

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

@@ -11,6 +11,9 @@ import { Message } from '@/types/messages'
 
 export const chatgpt: Chatbot = {
   wait_until_ready: async () => {
+    const max_wait_time = 2000
+    const start_time = Date.now()
+
     await new Promise((resolve) => {
       const check_for_element = () => {
         if (
@@ -19,6 +22,8 @@ export const chatgpt: Chatbot = {
           )
         ) {
           resolve(null)
+        } else if (Date.now() - start_time >= max_wait_time) {
+          resolve(null)
         } else {
           setTimeout(check_for_element, 100)
         }
@@ -29,16 +34,27 @@ export const chatgpt: Chatbot = {
   set_options: async (options: string[]) => {
     const supported_options = CHATBOTS['ChatGPT'].supported_options
     for (const option of options) {
-      if (option == 'reason' && supported_options['reason']) {
-        const reason_button = document.querySelector(
-          'button[data-testid="composer-button-reason"]'
-        ) as HTMLButtonElement
-        reason_button.click()
-      } else if (option == 'search' && supported_options['search']) {
-        const search_button = document.querySelector(
-          'button[data-testid="composer-button-search"]'
-        ) as HTMLButtonElement
-        search_button.click()
+      if (option == 'temporary' && supported_options['temporary']) {
+        const buttons = document.querySelectorAll('button')
+        for (const item of Array.from(buttons)) {
+          const path_element = item.querySelector(
+            'path[d="M6.319 1.334a.667.667 0 0 1-.512.792 5.43 5.43 0 0 0-2.602 1.362.667.667 0 1 1-.918-.967A6.76 6.76 0 0 1 5.527.822a.667.667 0 0 1 .792.512m1.363 0a.667.667 0 0 1 .791-.512 6.76 6.76 0 0 1 3.24 1.699.667.667 0 1 1-.917.967 5.43 5.43 0 0 0-2.602-1.362.667.667 0 0 1-.512-.792M1.51 4.614c.348.12.533.5.413.848a4.7 4.7 0 0 0 0 3.076.667.667 0 0 1-1.26.435 6.04 6.04 0 0 1 0-3.945.666.666 0 0 1 .847-.413m10.979 0a.667.667 0 0 1 .847.414A6 6 0 0 1 13.667 7a6 6 0 0 1-.33 1.973.667.667 0 1 1-1.26-.435 4.7 4.7 0 0 0 0-3.076.667.667 0 0 1 .413-.847M2.27 10.352a.667.667 0 0 1 .479.812q-.052.2-.111.397.629-.097 1.228-.267a.67.67 0 0 1 .496.054c.445.238.93.417 1.445.528a.667.667 0 1 1-.28 1.303 7 7 0 0 1-1.553-.533c-.73.189-1.479.305-2.266.354a.667.667 0 0 1-.664-.905c.164-.425.305-.844.414-1.264a.667.667 0 0 1 .812-.48m9.468.186a.666.666 0 0 1-.024.942 6.76 6.76 0 0 1-3.24 1.7.667.667 0 0 1-.28-1.304 5.43 5.43 0 0 0 2.601-1.362.667.667 0 0 1 .943.024"]'
+          )
+          if (path_element) {
+            ;(item as HTMLElement).click()
+            break
+          }
+        }
+        await new Promise((resolve) => {
+          const check_for_param = () => {
+            if (window.location.search.includes('temporary-chat=true')) {
+              resolve(null)
+            } else {
+              setTimeout(check_for_param, 100)
+            }
+          }
+          check_for_param()
+        })
       }
     }
   },
@@ -58,7 +74,10 @@ export const chatgpt: Chatbot = {
       let has_eligible_block = false
       for (const code_block of Array.from(code_blocks)) {
         const first_line_text = code_block?.textContent?.split('\n')[0]
-        if (first_line_text && extract_path_from_line_of_code(first_line_text)) {
+        if (
+          first_line_text &&
+          extract_path_from_line_of_code(first_line_text)
+        ) {
           has_eligible_block = true
           break
         }

packages/shared/src/constants/chatbots.ts

@@ -77,9 +77,7 @@ export const CHATBOTS = {
     default_system_instructions: "You're a helpful coding assistant.",
     supported_options: {},
     default_top_p: 1,
-    models: {
-      // Populated dynamically
-    }
+    models: {}
   },
   ChatGPT: {
     url: 'https://chatgpt.com/',
@@ -90,8 +88,7 @@ export const CHATBOTS = {
     supports_user_provided_port: false,
     default_system_instructions: '',
     supported_options: {
-      reason: 'Reason',
-      search: 'Search'
+      temporary: 'Temporary'
     },
     default_top_p: 0,
     models: {}