[Obs AI Assistant] Fix failing tests in serverless due to not emittin…

…g the tokenCount event (#203407)
elastic · Jan 2, 2025 · 7aa27b1 · 7aa27b1
1 parent 34e1401
commit 7aa27b1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...tion/test_suites/observability/ai_assistant/tests/public_complete/public_complete.spec.ts b/...tion/test_suites/observability/ai_assistant/tests/public_complete/public_complete.spec.ts
@@ -260,6 +260,7 @@ export default function ApiTest({ getService }: FtrProviderContext) {
                 },
               ],
             });
+            await conversationSimulator.tokenCount({ completion: 0, prompt: 0, total: 0 });
             await conversationSimulator.complete();
           }
         );
@@ -276,6 +277,7 @@ export default function ApiTest({ getService }: FtrProviderContext) {
       before(async () => {
         responseBody = await getOpenAIResponse(async (conversationSimulator) => {
           await conversationSimulator.next('Hello');
+          await conversationSimulator.tokenCount({ completion: 1, prompt: 1, total: 2 });
           await conversationSimulator.complete();
         });
       });