chore: minor fixes

src/chatgpt-api.ts

@@ -231,7 +231,6 @@ export class ChatGPTAPI {
                     }
 
                     result.detail = response
-
                     onProgress?.(result)
                   }
                 } catch (err) {
@@ -299,15 +298,21 @@ export class ChatGPTAPI {
       }
     ).then(async (message) => {
       if (message.detail && !message.detail.usage) {
-        const promptTokens = numTokens
+        try {
-        const completionTokens = await this._getTokenCount(message.text)
+          const promptTokens = numTokens
-        message.detail.usage = {
+          const completionTokens = await this._getTokenCount(message.text)
-          prompt_tokens: promptTokens,
+          message.detail.usage = {
-          completion_tokens: completionTokens,
+            prompt_tokens: promptTokens,
-          total_tokens: promptTokens + completionTokens,
+            completion_tokens: completionTokens,
-          estimated: true
+            total_tokens: promptTokens + completionTokens,
+            estimated: true
+          }
+        } catch (err) {
+          // TODO: this should really never happen, but if it does,
+          // we should handle notify the user gracefully
         }
       }
+
       return this._upsertMessage(message).then(() => message)
     })
 

src/types.ts

@@ -61,16 +61,6 @@ export type SendMessageBrowserOptions = {
   abortSignal?: AbortSignal
 }
 
-interface CreateChatCompletionStreamResponse
-  extends openai.CreateChatCompletionDeltaResponse {
-  usage: CreateCompletionStreamResponseUsage
-}
-
-interface CreateCompletionStreamResponseUsage
-  extends openai.CreateCompletionResponseUsage {
-  estimated: true
-}
-
 export interface ChatMessage {
   id: string
   text: string
@@ -100,6 +90,16 @@ export type GetMessageByIdFunction = (id: string) => Promise<ChatMessage>
 /** Upserts a chat message to a store. */
 export type UpsertMessageFunction = (message: ChatMessage) => Promise<void>
 
+export interface CreateChatCompletionStreamResponse
+  extends openai.CreateChatCompletionDeltaResponse {
+  usage: CreateCompletionStreamResponseUsage
+}
+
+export interface CreateCompletionStreamResponseUsage
+  extends openai.CreateCompletionResponseUsage {
+  estimated: true
+}
+
 /**
  * https://chat.openapi.com/backend-api/conversation
  */