diff --git a/x-pack/test/observability_ai_assistant_api_integration/tests/knowledge_base/knowledge_base_user_instructions.spec.ts b/x-pack/test/observability_ai_assistant_api_integration/tests/knowledge_base/knowledge_base_user_instructions.spec.ts index d5022a052d781..e6f7d728c6e0b 100644 --- a/x-pack/test/observability_ai_assistant_api_integration/tests/knowledge_base/knowledge_base_user_instructions.spec.ts +++ b/x-pack/test/observability_ai_assistant_api_integration/tests/knowledge_base/knowledge_base_user_instructions.spec.ts @@ -233,12 +233,9 @@ export default function ApiTest({ getService }: FtrProviderContext) { }) .expect(200); - const interceptPromises = [ - proxy.interceptConversationTitle('LLM-generated title').completeAfterIntercept(), - proxy - .interceptConversation({ name: 'conversation', response: 'I, the LLM, hear you!' }) - .completeAfterIntercept(), - ]; + const interceptPromise = proxy + .interceptConversation({ name: 'conversation', response: 'I, the LLM, hear you!' }) + .completeAfterIntercept(); const messages: Message[] = [ { @@ -283,8 +280,7 @@ export default function ApiTest({ getService }: FtrProviderContext) { }, }); - // wait for all interceptors to be settled - await Promise.all(interceptPromises); + await interceptPromise; const conversation = res.body; return conversation; @@ -297,6 +293,8 @@ export default function ApiTest({ getService }: FtrProviderContext) { after(async () => { proxy.close(); + await clearKnowledgeBase(es); + await clearConversations(es); await deleteActionConnector({ supertest, connectorId, log }); }); diff --git a/x-pack/test_serverless/api_integration/test_suites/observability/ai_assistant/tests/knowledge_base/knowledge_base_user_instructions.spec.ts b/x-pack/test_serverless/api_integration/test_suites/observability/ai_assistant/tests/knowledge_base/knowledge_base_user_instructions.spec.ts index d2b8f96060527..ddd5c6957c4e2 100644 --- a/x-pack/test_serverless/api_integration/test_suites/observability/ai_assistant/tests/knowledge_base/knowledge_base_user_instructions.spec.ts +++ b/x-pack/test_serverless/api_integration/test_suites/observability/ai_assistant/tests/knowledge_base/knowledge_base_user_instructions.spec.ts @@ -247,12 +247,9 @@ export default function ApiTest({ getService }: FtrProviderContext) { }) .expect(200); - const interceptPromises = [ - proxy.interceptConversationTitle('LLM-generated title').completeAfterIntercept(), - proxy - .interceptConversation({ name: 'conversation', response: 'I, the LLM, hear you!' }) - .completeAfterIntercept(), - ]; + const interceptPromise = proxy + .interceptConversation({ name: 'conversation', response: 'I, the LLM, hear you!' }) + .completeAfterIntercept(); const messages: Message[] = [ { @@ -297,8 +294,7 @@ export default function ApiTest({ getService }: FtrProviderContext) { }, }); - // wait for all interceptors to be settled - await Promise.all(interceptPromises); + await interceptPromise; const conversation = res.body; return conversation; @@ -317,6 +313,8 @@ export default function ApiTest({ getService }: FtrProviderContext) { after(async () => { proxy.close(); + await clearKnowledgeBase(es); + await clearConversations(es); await deleteActionConnector({ supertest: supertestWithoutAuth, connectorId,