[Obs AI Assistant] Revert changes to the stateful test (elastic#202926)

viduni94 · Jan 13, 2025 · 36b628e · 36b628e
1 parent dc4a603
commit 36b628e
Showing 1 changed file with 26 additions and 29 deletions.
diff --git a/...t_suites/observability/ai_assistant/tests/knowledge_base/knowledge_base_migration.spec.ts b/...t_suites/observability/ai_assistant/tests/knowledge_base/knowledge_base_migration.spec.ts
@@ -22,7 +22,6 @@ import { FtrProviderContext } from '../../common/ftr_provider_context';
 export default function ApiTest({ getService }: FtrProviderContext) {
   const ml = getService('ml');
   const es = getService('es');
-  const retry = getService('retry');
   const esArchiver = getService('esArchiver');
   const observabilityAIAssistantAPIClient = getService('observabilityAIAssistantAPIClient');
 
@@ -93,34 +92,32 @@ export default function ApiTest({ getService }: FtrProviderContext) {
       });
 
       it('the docs have semantic_text embeddings', async () => {
-        await retry.try(async () => {
-          const hits = await getKnowledgeBaseEntries();
-          const hasSemanticTextEmbeddings = hits.every((hit) => hit._source?.semantic_text);
-          expect(hasSemanticTextEmbeddings).to.be(true);
-
-          expect(
-            orderBy(hits, '_source.title').map(({ _source }) => {
-              const { text, inference } = _source?.semantic_text!;
-
-              return {
-                text,
-                inferenceId: inference.inference_id,
-                chunkCount: inference.chunks.length,
-              };
-            })
-          ).to.eql([
-            {
-              text: 'To infinity and beyond!',
-              inferenceId: AI_ASSISTANT_KB_INFERENCE_ID,
-              chunkCount: 1,
-            },
-            {
-              text: "The user's favourite color is blue.",
-              inferenceId: AI_ASSISTANT_KB_INFERENCE_ID,
-              chunkCount: 1,
-            },
-          ]);
-        });
+        const hits = await getKnowledgeBaseEntries();
+        const hasSemanticTextEmbeddings = hits.every((hit) => hit._source?.semantic_text);
+        expect(hasSemanticTextEmbeddings).to.be(true);
+
+        expect(
+          orderBy(hits, '_source.title').map(({ _source }) => {
+            const { text, inference } = _source?.semantic_text!;
+
+            return {
+              text,
+              inferenceId: inference.inference_id,
+              chunkCount: inference.chunks.length,
+            };
+          })
+        ).to.eql([
+          {
+            text: 'To infinity and beyond!',
+            inferenceId: AI_ASSISTANT_KB_INFERENCE_ID,
+            chunkCount: 1,
+          },
+          {
+            text: "The user's favourite color is blue.",
+            inferenceId: AI_ASSISTANT_KB_INFERENCE_ID,
+            chunkCount: 1,
+          },
+        ]);
       });
 
       it('returns entries correctly via API', async () => {