feat: set conversation title

Author: entelecheia

src/apis/chatgpt.ts

@@ -33,6 +33,22 @@ export async function setConversationProperty(
   await request(token, 'PATCH', `/conversation/${conversationId}`, propertyObject)
 }
 
+export async function setTitle(accessToken: string, conversationId: string) {
+  const title = 'ἐντελέχεια.άι'
+  if (conversationId) {
+    const ret = await request(accessToken, 'PATCH', `/conversation/${conversationId}`, {
+      title: title,
+    })
+    const data = JSON.parse(ret.responseText)
+    console.debug('set_title', data)
+    if (data.sucssess === true) {
+      return title
+    } else {
+      return null
+    }
+  }
+}
+
 export async function sendModerations(
   token: string,
   question: string,

src/configs/userConfig.ts

@@ -147,6 +147,7 @@ type UserConfigType = {
   tokenSavedOn: number
   messageId: string | null
   conversationId: string | null
+  conversationTitle: string | null
 }
 
 export const defaultConfig: UserConfigType = {
@@ -166,6 +167,7 @@ export const defaultConfig: UserConfigType = {
   tokenSavedOn: 0,
   messageId: uuidv4(),
   conversationId: null,
+  conversationTitle: null,
 }
 
 export async function getUserConfig(): Promise<UserConfigType> {

src/service/index.ts

@@ -7,13 +7,15 @@ import {
   generateAnswersWithChatgptWebApi,
   generateAnswersWithGptCompletionApi,
   sendMessageFeedback,
+  setTitle,
 } from '../apis'
 import {
   chatgptApiModelKeys,
   chatgptWebModelKeys,
   getUserConfig,
   gptApiModelKeys,
   isUsingApiKey,
+  updateUserConfig,
 } from '../configs'
 import { cache, getAccessToken, initSession, KEY_ACCESS_TOKEN } from '../utils'
 
@@ -47,6 +49,16 @@ Browser.runtime.onConnect.addListener((port) => {
         }
         console.debug('session', session)
         await generateAnswersWithChatgptWebApi(port, session.question, session, accessToken)
+
+        // set conversation title
+        if (session.conversationId && session.conversationTitle == null) {
+          console.debug('set title')
+          const title = await setTitle(accessToken, session.conversationId)
+          if (title) {
+            session.conversationTitle = title
+            await updateUserConfig({ conversationTitle: title })
+          }
+        }
       } else if (gptApiModelKeys.includes(config.modelName)) {
         await generateAnswersWithGptCompletionApi(
           port,
@@ -118,6 +130,19 @@ Browser.runtime.onMessage.addListener(async (message) => {
   } else if (message.type === 'SET_SESSION') {
     console.debug('setting session', message.session)
     sessionDataMap.set(sessionId, message.session)
+  } else if (message.type === 'SET_TITLE') {
+    // set conversation title
+    const accessToken = await getAccessToken()
+    const config = await getUserConfig()
+    const session = sessionDataMap.get(sessionId)
+
+    if (session.conversationId && session.conversationId !== config.conversationId) {
+      const title = await setTitle(accessToken, session.conversationId)
+      if (title) {
+        session.conversationTitle = title
+        await updateUserConfig({ conversationTitle: title })
+      }
+    }
   }
 })
 

src/utils/initSession.ts

@@ -9,6 +9,7 @@ export interface Session {
   conversationId: string | null
   messageId: string | null
   parentMessageId: string | null
+  conversationTitle: string | null
   conversationRecords: ConversationRecord[] | null
   useApiKey: boolean | null
 }
@@ -18,6 +19,7 @@ export function initSession({
   conversationId = null,
   messageId = null,
   parentMessageId = null,
+  conversationTitle = null,
   conversationRecords = [],
   useApiKey = null,
 }: Partial<Session> = {}): Session {
@@ -26,6 +28,7 @@ export function initSession({
     conversationId,
     messageId,
     parentMessageId,
+    conversationTitle,
     conversationRecords,
     useApiKey,
   }