assistant test fix

build/build_number.txt

@@ -1 +1 @@
-166
+167

src/services/assistant.ts

@@ -180,13 +180,10 @@ export default class {
       if (error.name !== 'AbortError') {
         if (error.status === 401 || error.message.includes('401') || error.message.toLowerCase().includes('apikey')) {
           message.setText('You need to enter your API key in the Models tab of <a href="#settings">Settings</a> in order to chat.')
-          callback?.call(null, { done: true })
         } else if (error.status === 400 && (error.message.includes('credit') || error.message.includes('balance'))) {
           message.setText('Sorry, it seems you have run out of credits. Check the balance of your LLM provider account.')
-          callback?.call(null, { done: true })
         } else if (message.content === '') {
           message.setText('Sorry, I could not generate text for that prompt.')
-          callback?.call(null, { done: true })
         } else {
           message.appendText({ text: '\n\nSorry, I am not able to continue here.', done: true })
         }

tests/unit/assistant.test.ts

@@ -27,16 +27,14 @@ let assistant: Assistant = null
 const prompt = async (prompt: string, opts: LlmCompletionOpts = {}): Promise<string> => {
 
   // callback
-  let done = false
   let content = ''
   const callback = (chunk: LlmStream) => {
-    done = chunk?.done
     content += chunk?.text || ''
   }
 
   // call and wait
   await assistant.prompt(prompt, { ...opts, save: false }, callback)
-  await vi.waitUntil(async () => done)
+  await vi.waitUntil(async () => !assistant.chat.lastMessage().transient)
 
   // return
   return content