Fix JSON parsing error

Author: TiesdeKok

src/background/chatgpt.ts

@@ -1,3 +1,7 @@
+import { v4 as uuidv4 } from 'uuid'
+import { Event } from './event'
+import { fetchSSE } from './fetch-sse'
+
 async function request(token: string, method: string, path: string, data: unknown) {
   return fetch(`https://chat.openai.com/backend-api${path}`, {
     method,
@@ -20,3 +24,59 @@ export async function setConversationProperty(
 ) {
   await request(token, 'PATCH', `/conversation/${conversationId}`, propertyObject)
 }
+
+export async function sendMessage(params: {
+  token: string
+  prompt: string
+  onEvent: (event: Event) => void
+  signal?: AbortSignal
+}) {
+  await fetchSSE('https://chat.openai.com/backend-api/conversation', {
+    method: 'POST',
+    signal: params.signal,
+    headers: {
+      'Content-Type': 'application/json',
+      Authorization: `Bearer ${params.token}`,
+    },
+    body: JSON.stringify({
+      action: 'next',
+      messages: [
+        {
+          id: uuidv4(),
+          role: 'user',
+          content: {
+            content_type: 'text',
+            parts: [params.prompt],
+          },
+        },
+      ],
+      model: 'text-davinci-002-render',
+      parent_message_id: uuidv4(),
+    }),
+    onMessage(message: string) {
+      console.debug('sse message', message)
+      if (message === '[DONE]') {
+        params.onEvent({ type: 'done' })
+        return
+      }
+      let data
+      try {
+        data = JSON.parse(message)
+      } catch (err) {
+        console.error(err)
+        return
+      }
+      const text = data.message?.content?.parts?.[0]
+      if (text) {
+        params.onEvent({
+          type: 'answer',
+          data: {
+            text,
+            messageId: data.message.id,
+            conversationId: data.conversation_id,
+          },
+        })
+      }
+    },
+  })
+}

src/background/event.ts

@@ -0,0 +1,10 @@
+import { Answer } from '../messaging'
+
+export type Event =
+  | {
+      type: 'answer'
+      data: Answer
+    }
+  | {
+      type: 'done'
+    }

src/background/fetch-sse.ts

@@ -1,4 +1,5 @@
 import { createParser } from 'eventsource-parser'
+import { isEmpty } from 'lodash-es'
 import { streamAsyncIterable } from './stream-async-iterable.js'
 
 export async function fetchSSE(
@@ -8,8 +9,8 @@ export async function fetchSSE(
   const { onMessage, ...fetchOptions } = options
   const resp = await fetch(resource, fetchOptions)
   if (!resp.ok) {
-    const detail = (await resp.json().catch(() => ({}))).detail
-    throw new Error(detail || `${resp.status} ${resp.statusText}`)
+    const error = await resp.json().catch(() => ({}))
+    throw new Error(!isEmpty(error) ? JSON.stringify(error) : `${resp.status} ${resp.statusText}`)
   }
   const parser = createParser((event) => {
     if (event.type === 'event') {

src/background/index.ts

@@ -1,9 +1,6 @@
 import ExpiryMap from 'expiry-map'
-import { v4 as uuidv4 } from 'uuid'
 import Browser from 'webextension-polyfill'
-import { Answer } from '../messaging.js'
-import { sendMessageFeedback, setConversationProperty } from './chatgpt.js'
-import { fetchSSE } from './fetch-sse.js'
+import { sendMessage, sendMessageFeedback, setConversationProperty } from './chatgpt.js'
 
 const KEY_ACCESS_TOKEN = 'accessToken'
 
@@ -41,45 +38,18 @@ async function generateAnswers(port: Browser.Runtime.Port, question: string) {
     deleteConversation()
   })
 
-  await fetchSSE('https://chat.openai.com/backend-api/conversation', {
-    method: 'POST',
+  await sendMessage({
+    token: accessToken,
+    prompt: question,
     signal: controller.signal,
-    headers: {
-      'Content-Type': 'application/json',
-      Authorization: `Bearer ${accessToken}`,
-    },
-    body: JSON.stringify({
-      action: 'next',
-      messages: [
-        {
-          id: uuidv4(),
-          role: 'user',
-          content: {
-            content_type: 'text',
-            parts: [question],
-          },
-        },
-      ],
-      model: 'text-davinci-002-render',
-      parent_message_id: uuidv4(),
-    }),
-    onMessage(message: string) {
-      console.debug('sse message', message)
-      if (message === '[DONE]') {
+    onEvent(event) {
+      if (event.type === 'done') {
         port.postMessage({ event: 'DONE' })
         deleteConversation()
         return
       }
-      const data = JSON.parse(message)
-      const text = data.message?.content?.parts?.[0]
-      conversationId = data.conversation_id
-      if (text) {
-        port.postMessage({
-          text,
-          messageId: data.message.id,
-          conversationId: data.conversation_id,
-        } as Answer)
-      }
+      conversationId = event.data.conversationId
+      port.postMessage(event.data)
     },
   })
 }
@@ -101,5 +71,15 @@ Browser.runtime.onMessage.addListener(async (message) => {
   if (message.type === 'FEEDBACK') {
     const token = await getAccessToken()
     await sendMessageFeedback(token, message.data)
+  } else if (message.type === 'OPEN_OPTIONS_PAGE') {
+    Browser.runtime.openOptionsPage()
+  } else if (message.type === 'GET_ACCESS_TOKEN') {
+    return getAccessToken()
+  }
+})
+
+Browser.runtime.onInstalled.addListener((details) => {
+  if (details.reason === 'install') {
+    Browser.runtime.openOptionsPage()
   }
 })

src/manifest.json

@@ -1,7 +1,7 @@
 {
   "name": "ChatGPT - Jupyter - AI Assistant",
   "description": "ChatGPT-powered AI assistant for Jupyter Notebooks",
-  "version": "0.2.3",
+  "version": "0.3.0",
   "manifest_version": 3,
   "icons": {
     "16": "logo.png",

src/manifest.v2.json

@@ -1,7 +1,7 @@
 {
   "name": "ChatGPT - Jupyter - AI Assistant",
   "description": "ChatGPT-powered AI assistant for Jupyter Notebooks",
-  "version": "0.2.3",
+  "version": "0.3.0",
   "manifest_version": 2,
   "icons": {
     "16": "logo.png",