diff --git a/x-pack/test_serverless/api_integration/test_suites/observability/ai_assistant/tests/complete/complete.spec.ts b/x-pack/test_serverless/api_integration/test_suites/observability/ai_assistant/tests/complete/complete.spec.ts
index 5081d6b12879b..4d3e1baed3f9b 100644
--- a/x-pack/test_serverless/api_integration/test_suites/observability/ai_assistant/tests/complete/complete.spec.ts
+++ b/x-pack/test_serverless/api_integration/test_suites/observability/ai_assistant/tests/complete/complete.spec.ts
@@ -67,8 +67,7 @@ export default function ApiTest({ getService }: FtrProviderContext) {
     },
   ];
 
-  // Failing: See https://github.com/elastic/kibana/issues/203407
-  describe.skip('/internal/observability_ai_assistant/chat/complete', function () {
+  describe('/internal/observability_ai_assistant/chat/complete', function () {
     // TODO: https://github.com/elastic/kibana/issues/192751
     this.tags(['skipMKI']);
     let proxy: LlmProxy;
@@ -107,6 +106,7 @@ export default function ApiTest({ getService }: FtrProviderContext) {
 
       await titleSimulator.status(200);
       await titleSimulator.next('My generated title');
+      await titleSimulator.tokenCount({ completion: 1, prompt: 1, total: 2 });
       await titleSimulator.complete();
 
       await conversationSimulator.status(200);
diff --git a/x-pack/test_serverless/api_integration/test_suites/observability/ai_assistant/tests/public_complete/public_complete.spec.ts b/x-pack/test_serverless/api_integration/test_suites/observability/ai_assistant/tests/public_complete/public_complete.spec.ts
index 3f4a31e92de91..8c4cf35c0bf2c 100644
--- a/x-pack/test_serverless/api_integration/test_suites/observability/ai_assistant/tests/public_complete/public_complete.spec.ts
+++ b/x-pack/test_serverless/api_integration/test_suites/observability/ai_assistant/tests/public_complete/public_complete.spec.ts
@@ -49,8 +49,7 @@ export default function ApiTest({ getService }: FtrProviderContext) {
     },
   ];
 
-  // Failing: See https://github.com/elastic/kibana/issues/203408
-  describe.skip('/api/observability_ai_assistant/chat/complete', function () {
+  describe('/api/observability_ai_assistant/chat/complete', function () {
     // TODO: https://github.com/elastic/kibana/issues/192751
     this.tags(['skipMKI']);
 
@@ -107,6 +106,7 @@ export default function ApiTest({ getService }: FtrProviderContext) {
 
       await titleSimulator.status(200);
       await titleSimulator.next('My generated title');
+      await titleSimulator.tokenCount({ completion: 1, prompt: 1, total: 2 });
       await titleSimulator.complete();
 
       await conversationSimulator.status(200);
@@ -202,6 +202,7 @@ export default function ApiTest({ getService }: FtrProviderContext) {
                 },
               ],
             });
+            await conversationSimulator.tokenCount({ completion: 0, prompt: 0, total: 0 });
             await conversationSimulator.complete();
           }
         );
@@ -259,6 +260,7 @@ export default function ApiTest({ getService }: FtrProviderContext) {
                 },
               ],
             });
+            await conversationSimulator.tokenCount({ completion: 0, prompt: 0, total: 0 });
             await conversationSimulator.complete();
           }
         );
@@ -275,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();
         });
       });