瀏覽代碼

Add embeddings initialization to evaluation tests

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
Colby McHenry 5 月之前
父節點
當前提交
2fffcf4736
共有 1 個文件被更改,包括 10 次插入2 次删除
  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