Bläddra i källkod

Add embeddings initialization to evaluation tests

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
Colby McHenry 5 månader sedan
förälder
incheckning
2fffcf4736
1 ändrade filer med 10 tillägg och 2 borttagningar
  1. 10 2
      __tests__/evaluation/evaluation.test.ts

+ 10 - 2
__tests__/evaluation/evaluation.test.ts

@@ -225,7 +225,11 @@ describe('CodeGraph Evaluation', () => {
 
       // Initialize and index
       cg = await CodeGraph.init(fixturePath, { index: true });
-    }, 60000);
+
+      // Initialize embeddings for semantic search
+      await cg.initializeEmbeddings();
+      await cg.generateEmbeddings();
+    }, 120000);
 
     afterAll(() => {
       // Print summary table after all tests
@@ -266,7 +270,11 @@ describe('CodeGraph Evaluation', () => {
 
       // Initialize and index
       cg = await CodeGraph.init(fixturePath, { index: true });
-    }, 60000);
+
+      // Initialize embeddings for semantic search
+      await cg.initializeEmbeddings();
+      await cg.generateEmbeddings();
+    }, 120000);
 
     afterAll(() => {
       // Print summary table after all tests