npm - @aiready/context-analyzer - Versions diffs - 0.21.22 → 0.21.24 - Mend

@aiready/context-analyzer 0.21.22 → 0.21.24

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (78) hide show

package/.turbo/turbo-build.log +26 -25
package/.turbo/turbo-lint.log +5 -5
package/.turbo/turbo-test.log +91 -41
package/coverage/clover.xml +2615 -1242
package/coverage/coverage-final.json +30 -13
package/coverage/dist/chunk-64U3PNO3.mjs.html +367 -0
package/coverage/dist/chunk-J3MUOWHC.mjs.html +5326 -0
package/coverage/dist/index.html +146 -0
package/coverage/{classifier.ts.html → dist/index.mjs.html} +537 -912
package/coverage/index.html +84 -189
package/coverage/src/analyzer.ts.html +88 -0
package/coverage/src/analyzers/index.html +116 -0
package/coverage/src/analyzers/python-context.ts.html +910 -0
package/coverage/{ast-utils.ts.html → src/ast-utils.ts.html} +84 -54
package/coverage/src/classifier.ts.html +892 -0
package/coverage/src/classify/classification-patterns.ts.html +307 -0
package/coverage/src/classify/file-classifiers.ts.html +973 -0
package/coverage/src/classify/index.html +131 -0
package/coverage/{cluster-detector.ts.html → src/cluster-detector.ts.html} +154 -91
package/coverage/{defaults.ts.html → src/defaults.ts.html} +74 -65
package/coverage/{graph-builder.ts.html → src/graph-builder.ts.html} +268 -229
package/coverage/src/index.html +341 -0
package/coverage/{index.ts.html → src/index.ts.html} +70 -13
package/coverage/{scoring.ts.html → src/issue-analyzer.ts.html} +201 -261
package/coverage/src/mapper.ts.html +439 -0
package/coverage/{metrics.ts.html → src/metrics.ts.html} +201 -132
package/coverage/src/orchestrator.ts.html +493 -0
package/coverage/{provider.ts.html → src/provider.ts.html} +21 -21
package/coverage/{remediation.ts.html → src/remediation.ts.html} +112 -52
package/coverage/src/report/console-report.ts.html +415 -0
package/coverage/src/report/html-report.ts.html +361 -0
package/coverage/src/report/index.html +146 -0
package/coverage/src/report/interactive-setup.ts.html +373 -0
package/coverage/src/scoring.ts.html +895 -0
package/coverage/src/semantic/co-usage.ts.html +340 -0
package/coverage/src/semantic/consolidation.ts.html +223 -0
package/coverage/src/semantic/domain-inference.ts.html +859 -0
package/coverage/src/semantic/index.html +161 -0
package/coverage/src/semantic/type-graph.ts.html +163 -0
package/coverage/{summary.ts.html → src/summary.ts.html} +155 -275
package/coverage/{types.ts.html → src/types.ts.html} +133 -31
package/coverage/src/utils/dependency-graph-utils.ts.html +463 -0
package/coverage/src/utils/index.html +131 -0
package/coverage/src/utils/string-utils.ts.html +148 -0
package/dist/chunk-AMPK6SWS.mjs +1754 -0
package/dist/chunk-BHCRDEE4.mjs +1745 -0
package/dist/chunk-IKRP7ECY.mjs +1754 -0
package/dist/chunk-J3MUOWHC.mjs +1747 -0
package/dist/chunk-TWWPY7FD.mjs +1754 -0
package/dist/chunk-Z5WY6A4P.mjs +1754 -0
package/dist/cli.js +77 -185
package/dist/cli.mjs +1 -1
package/dist/index.d.mts +1 -1
package/dist/index.d.ts +1 -1
package/dist/index.js +73 -181
package/dist/index.mjs +1 -1
package/dist/python-context-BWDC4E5Z.mjs +162 -0
package/package.json +3 -3
package/src/__tests__/analyzer.test.ts +14 -14
package/src/__tests__/auto-detection.test.ts +16 -16
package/src/__tests__/consolidation.test.ts +247 -0
package/src/__tests__/defaults.test.ts +121 -0
package/src/__tests__/domain-inference.test.ts +420 -0
package/src/__tests__/fragmentation-coupling.test.ts +4 -4
package/src/__tests__/issue-analyzer.test.ts +155 -0
package/src/__tests__/orchestrator.test.ts +143 -0
package/src/__tests__/python-context.test.ts +100 -0
package/src/__tests__/report/console-report.test.ts +292 -0
package/src/__tests__/report/html-report.test.ts +243 -0
package/src/__tests__/scoring.test.ts +17 -11
package/src/analyzers/python-context.ts +2 -2
package/src/ast-utils.ts +3 -3
package/src/graph-builder.ts +4 -4
package/src/mapper.ts +6 -0
package/src/orchestrator.ts +1 -1
package/src/report/html-report.ts +73 -181
package/coverage/analyzer.ts.html +0 -1369
package/coverage/semantic-analysis.ts.html +0 -1201

package/dist/index.js CHANGED Viewed

@@ -130,7 +130,7 @@ __export(python_context_exports, {
 });
 async function analyzePythonContext(files, rootDir) {
   const results = [];
-  const parser = (0, import_core5.getParser)("dummy.py");
+  const parser = await (0, import_core5.getParser)("dummy.py");
   if (!parser) {
     console.warn("Python parser not available");
     return results;
@@ -194,7 +194,7 @@ async function analyzePythonContext(files, rootDir) {
 }
 async function buildPythonDependencyGraph(files, rootDir) {
   const graph = /* @__PURE__ */ new Map();
-  const parser = (0, import_core5.getParser)("dummy.py");
+  const parser = await (0, import_core5.getParser)("dummy.py");
   if (!parser) return graph;
   for (const file of files) {
     try {
@@ -352,7 +352,7 @@ __export(index_exports, {
   runInteractiveSetup: () => runInteractiveSetup
 });
 module.exports = __toCommonJS(index_exports);
-var import_core11 = require("@aiready/core");
+var import_core12 = require("@aiready/core");
 // src/provider.ts
 var import_core9 = require("@aiready/core");
@@ -552,9 +552,9 @@ function inferDomain(name, filePath, domainOptions, fileImports) {
 }
 // src/ast-utils.ts
-function extractExportsWithAST(content, filePath, domainOptions, fileImports) {
+async function extractExportsWithAST(content, filePath, domainOptions, fileImports) {
   try {
-    const { exports: astExports } = (0, import_core.parseFileExports)(content, filePath);
+    const { exports: astExports } = await (0, import_core.parseFileExports)(content, filePath);
     if (astExports.length === 0 && !isTestFile(filePath)) {
       return extractExports(content, filePath, domainOptions, fileImports);
     }
@@ -952,16 +952,16 @@ function extractDomainKeywordsFromPaths(files) {
   }
   return Array.from(folderNames);
 }
-function buildDependencyGraph(files, options) {
+async function buildDependencyGraph(files, options) {
   const nodes = /* @__PURE__ */ new Map();
   const edges = /* @__PURE__ */ new Map();
   const autoDetectedKeywords = options?.domainKeywords ?? extractDomainKeywordsFromPaths(files);
   const allFilePaths = new Set(files.map((f) => f.file));
   for (const { file, content } of files) {
-    const { imports: astImports } = (0, import_core4.parseFileExports)(content, file);
+    const { imports: astImports } = await (0, import_core4.parseFileExports)(content, file);
     const resolvedImports = astImports.map((i) => resolveImport(i.source, file, allFilePaths)).filter((path) => path !== null);
     const importSources = astImports.map((i) => i.source);
-    const exports2 = extractExportsWithAST(
+    const exports2 = await extractExportsWithAST(
       content,
       file,
       { domainKeywords: autoDetectedKeywords },
@@ -1687,7 +1687,7 @@ async function analyzeContext(options) {
       content: await (0, import_core6.readFileContent)(file)
     }))
   );
-  const graph = buildDependencyGraph(
+  const graph = await buildDependencyGraph(
     fileContents.filter((f) => !f.file.toLowerCase().endsWith(".py"))
   );
   let pythonResults = [];
@@ -2204,182 +2204,74 @@ function displayConsoleReport(summary, results, maxResults = 10) {
 }
 // src/report/html-report.ts
+var import_core11 = require("@aiready/core");
 function generateHTMLReport(summary, results) {
   const totalIssues = summary.criticalIssues + summary.majorIssues + summary.minorIssues;
   void results;
-  return `<!DOCTYPE html>
-<html lang="en">
-<head>
-  <meta charset="UTF-8">
-  <meta name="viewport" content="width=device-width, initial-scale=1.0">
-  <title>aiready Context Analysis Report</title>
-  <style>
-    body {
-      font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif;
-      line-height: 1.6;
-      color: #333;
-      max-width: 1200px;
-      margin: 0 auto;
-      padding: 20px;
-      background-color: #f5f5f5;
-    }
-    h1, h2, h3 { color: #2c3e50; }
-    .header {
-      background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-      color: white;
-      padding: 30px;
-      border-radius: 8px;
-      margin-bottom: 30px;
-    }
-    .summary {
-      display: grid;
-      grid-template-columns: repeat(auto-fit, minmax(250px, 1fr));
-      gap: 20px;
-      margin-bottom: 30px;
-    }
-    .card {
-      background: white;
-      padding: 20px;
-      border-radius: 8px;
-      box-shadow: 0 2px 4px rgba(0,0,0,0.1);
-    }
-    .metric {
-      font-size: 2em;
-      font-weight: bold;
-      color: #667eea;
-    }
-    .label {
-      color: #666;
-      font-size: 0.9em;
-      margin-top: 5px;
-    }
-    .issue-critical { color: #e74c3c; }
-    .issue-major { color: #f39c12; }
-    .issue-minor { color: #3498db; }
-    table {
-      width: 100%;
-      border-collapse: collapse;
-      background: white;
-      border-radius: 8px;
-      overflow: hidden;
-      box-shadow: 0 2px 4px rgba(0,0,0,0.1);
-    }
-    th, td {
-      padding: 12px;
-      text-align: left;
-      border-bottom: 1px solid #eee;
-    }
-    th {
-      background-color: #667eea;
-      color: white;
-      font-weight: 600;
-    }
-    tr:hover { background-color: #f8f9fa; }
-    .footer {
-      text-align: center;
-      margin-top: 40px;
-      padding: 20px;
-      color: #666;
-      font-size: 0.9em;
+  const head = (0, import_core11.generateReportHead)("AIReady Context Analysis Report");
+  const stats = (0, import_core11.generateStatCards)([
+    { value: summary.totalFiles, label: "Files Analyzed" },
+    { value: summary.totalTokens.toLocaleString(), label: "Total Tokens" },
+    { value: summary.avgContextBudget.toFixed(0), label: "Avg Context Budget" },
+    {
+      value: totalIssues,
+      label: "Total Issues",
+      color: totalIssues > 0 ? "#f39c12" : void 0
     }
-  </style>
-</head>
+  ]);
+  const hero = (0, import_core11.generateReportHero)(
+    "\u{1F50D} AIReady Context Analysis Report",
+    `Generated on ${(/* @__PURE__ */ new Date()).toLocaleString()}`
+  );
+  let body = `${hero}
+${stats}`;
+  if (totalIssues > 0) {
+    body += (0, import_core11.generateIssueSummary)(
+      summary.criticalIssues,
+      summary.majorIssues,
+      summary.minorIssues,
+      summary.totalPotentialSavings
+    );
+  }
+  if (summary.fragmentedModules.length > 0) {
+    const fragmentedRows = summary.fragmentedModules.map((m) => [
+      m.domain,
+      String(m.files.length),
+      `${(m.fragmentationScore * 100).toFixed(0)}%`,
+      m.totalTokens.toLocaleString()
+    ]);
+    body += (0, import_core11.wrapInCard)(
+      (0, import_core11.generateTable)({
+        headers: ["Domain", "Files", "Fragmentation", "Token Cost"],
+        rows: fragmentedRows
+      }),
+      "\u{1F9E9} Fragmented Modules"
+    );
+  }
+  if (summary.topExpensiveFiles.length > 0) {
+    const expensiveRows = summary.topExpensiveFiles.map((f) => [
+      f.file,
+      `${f.contextBudget.toLocaleString()} tokens`,
+      `<span class="issue-${f.severity}">${f.severity.toUpperCase()}</span>`
+    ]);
+    body += (0, import_core11.wrapInCard)(
+      (0, import_core11.generateTable)({
+        headers: ["File", "Context Budget", "Severity"],
+        rows: expensiveRows
+      }),
+      "\u{1F4B8} Most Expensive Files"
+    );
+  }
+  const footer = (0, import_core11.generateReportFooter)({
+    title: "Context Analysis Report",
+    packageName: "context-analyzer",
+    packageUrl: "https://github.com/caopengau/aiready-context-analyzer",
+    bugUrl: "https://github.com/caopengau/aiready-context-analyzer/issues"
+  });
+  return `${head}
 <body>
-  <div class="header">
-    <h1>\u{1F50D} AIReady Context Analysis Report</h1>
-    <p>Generated on ${(/* @__PURE__ */ new Date()).toLocaleString()}</p>
-  </div>
-  <div class="summary">
-    <div class="card">
-      <div class="metric">${summary.totalFiles}</div>
-      <div class="label">Files Analyzed</div>
-    </div>
-    <div class="card">
-      <div class="metric">${summary.totalTokens.toLocaleString()}</div>
-      <div class="label">Total Tokens</div>
-    </div>
-    <div class="card">
-      <div class="metric">${summary.avgContextBudget.toFixed(0)}</div>
-      <div class="label">Avg Context Budget</div>
-    </div>
-    <div class="card">
-      <div class="metric ${totalIssues > 0 ? "issue-major" : ""}">${totalIssues}</div>
-      <div class="label">Total Issues</div>
-    </div>
-  </div>
-  ${totalIssues > 0 ? `
-  <div class="card" style="margin-bottom: 30px;">
-    <h2>\u26A0\uFE0F Issues Summary</h2>
-    <p>
-      <span class="issue-critical">\u{1F534} Critical: ${summary.criticalIssues}</span> &nbsp;
-      <span class="issue-major">\u{1F7E1} Major: ${summary.majorIssues}</span> &nbsp;
-      <span class="issue-minor">\u{1F535} Minor: ${summary.minorIssues}</span>
-    </p>
-    <p><strong>Potential Savings:</strong> ${summary.totalPotentialSavings.toLocaleString()} tokens</p>
-  </div>
-  ` : ""}
-  ${summary.fragmentedModules.length > 0 ? `
-  <div class="card" style="margin-bottom: 30px;">
-    <h2>\u{1F9E9} Fragmented Modules</h2>
-    <table>
-      <thead>
-        <tr>
-          <th>Domain</th>
-          <th>Files</th>
-          <th>Fragmentation</th>
-          <th>Token Cost</th>
-        </tr>
-      </thead>
-      <tbody>
-        ${summary.fragmentedModules.map(
-    (m) => `
-          <tr>
-            <td>${m.domain}</td>
-            <td>${m.files.length}</td>
-            <td>${(m.fragmentationScore * 100).toFixed(0)}%</td>
-            <td>${m.totalTokens.toLocaleString()}</td>
-          </tr>
-        `
-  ).join("")}
-      </tbody>
-    </table>
-  </div>
-  ` : ""}
-  ${summary.topExpensiveFiles.length > 0 ? `
-  <div class="card" style="margin-bottom: 30px;">
-    <h2>\u{1F4B8} Most Expensive Files</h2>
-    <table>
-      <thead>
-        <tr>
-          <th>File</th>
-          <th>Context Budget</th>
-          <th>Severity</th>
-        </tr>
-      </thead>
-      <tbody>
-        ${summary.topExpensiveFiles.map(
-    (f) => `
-          <tr>
-            <td>${f.file}</td>
-            <td>${f.contextBudget.toLocaleString()} tokens</td>
-            <td class="issue-${f.severity}">${f.severity.toUpperCase()}</td>
-          </tr>
-        `
-  ).join("")}
-      </tbody>
-    </table>
-  </div>
-  ` : ""}
-  <div class="footer">
-    <p>Generated by <strong>@aiready/context-analyzer</strong></p>
-    <p>Like AIReady? <a href="https://github.com/caopengau/aiready-context-analyzer">Star us on GitHub</a></p>
-    <p>Found a bug? <a href="https://github.com/caopengau/aiready-context-analyzer/issues">Report it here</a></p>
-  </div>
+  ${body}
+  ${footer}
 </body>
 </html>`;
 }
@@ -2458,7 +2350,7 @@ async function runInteractiveSetup(directory, current) {
 }
 // src/index.ts
-import_core11.ToolRegistry.register(ContextAnalyzerProvider);
+import_core12.ToolRegistry.register(ContextAnalyzerProvider);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   BARREL_EXPORT_MIN_EXPORTS,

package/dist/index.mjs CHANGED Viewed

@@ -52,7 +52,7 @@ import {
   isTypeDefinition,
   isUtilityModule,
   runInteractiveSetup
-} from "./chunk-CBWM3EK5.mjs";
+} from "./chunk-Z5WY6A4P.mjs";
 import "./chunk-64U3PNO3.mjs";
 // src/index.ts

package/dist/python-context-BWDC4E5Z.mjs ADDED Viewed

@@ -0,0 +1,162 @@
+import {
+  calculateImportDepthFromEdges,
+  detectGraphCyclesFromFile
+} from "./chunk-64U3PNO3.mjs";
+// src/analyzers/python-context.ts
+import { getParser, estimateTokens } from "@aiready/core";
+import { resolve, relative, dirname, join } from "path";
+import fs from "fs";
+async function analyzePythonContext(files, rootDir) {
+  const results = [];
+  const parser = await getParser("dummy.py");
+  if (!parser) {
+    console.warn("Python parser not available");
+    return results;
+  }
+  const pythonFiles = files.filter((f) => f.toLowerCase().endsWith(".py"));
+  void relative;
+  void join;
+  const dependencyGraph = await buildPythonDependencyGraph(
+    pythonFiles,
+    rootDir
+  );
+  for (const file of pythonFiles) {
+    try {
+      const code = await fs.promises.readFile(file, "utf-8");
+      const result = parser.parse(code, file);
+      const imports = result.imports.map((imp) => ({
+        source: imp.source,
+        specifiers: imp.specifiers,
+        isRelative: imp.source.startsWith("."),
+        resolvedPath: resolvePythonImport(file, imp.source, rootDir)
+      }));
+      const exports = result.exports.map((exp) => ({
+        name: exp.name,
+        type: exp.type
+      }));
+      const linesOfCode = code.split("\n").length;
+      const importDepth = calculateImportDepthFromEdges(
+        file,
+        dependencyGraph,
+        /* @__PURE__ */ new Set()
+      );
+      const contextBudget = estimateContextBudget(
+        code,
+        imports,
+        dependencyGraph
+      );
+      const cohesion = calculatePythonCohesion(exports, imports);
+      const circularDependencies = detectGraphCyclesFromFile(
+        file,
+        dependencyGraph
+      ).map((cycle) => cycle.join(" -> "));
+      results.push({
+        file,
+        importDepth,
+        contextBudget,
+        cohesion,
+        imports,
+        exports,
+        metrics: {
+          linesOfCode,
+          importCount: imports.length,
+          exportCount: exports.length,
+          circularDependencies
+        }
+      });
+    } catch (error) {
+      console.warn(`Failed to analyze ${file}:`, error);
+    }
+  }
+  return results;
+}
+async function buildPythonDependencyGraph(files, rootDir) {
+  const graph = /* @__PURE__ */ new Map();
+  const parser = await getParser("dummy.py");
+  if (!parser) return graph;
+  for (const file of files) {
+    try {
+      const code = await fs.promises.readFile(file, "utf-8");
+      const result = parser.parse(code, file);
+      const dependencies = /* @__PURE__ */ new Set();
+      for (const imp of result.imports) {
+        const resolved = resolvePythonImport(file, imp.source, rootDir);
+        if (resolved && files.includes(resolved)) {
+          dependencies.add(resolved);
+        }
+      }
+      graph.set(file, dependencies);
+    } catch (error) {
+      void error;
+    }
+  }
+  return graph;
+}
+function resolvePythonImport(fromFile, importPath, rootDir) {
+  const dir = dirname(fromFile);
+  if (importPath.startsWith(".")) {
+    const parts = importPath.split(".");
+    let upCount = 0;
+    while (parts[0] === "") {
+      upCount++;
+      parts.shift();
+    }
+    let targetDir = dir;
+    for (let i = 0; i < upCount - 1; i++) {
+      targetDir = dirname(targetDir);
+    }
+    const modulePath = parts.join("/");
+    const possiblePaths = [
+      resolve(targetDir, `${modulePath}.py`),
+      resolve(targetDir, modulePath, "__init__.py")
+    ];
+    for (const path of possiblePaths) {
+      if (fs.existsSync(path)) {
+        return path;
+      }
+    }
+  } else {
+    const modulePath = importPath.replace(/\./g, "/");
+    const possiblePaths = [
+      resolve(rootDir, `${modulePath}.py`),
+      resolve(rootDir, modulePath, "__init__.py")
+    ];
+    for (const path of possiblePaths) {
+      if (fs.existsSync(path)) {
+        return path;
+      }
+    }
+  }
+  return void 0;
+}
+function estimateContextBudget(code, imports, dependencyGraph) {
+  void dependencyGraph;
+  let budget = estimateTokens(code);
+  const avgTokensPerDep = 500;
+  budget += imports.length * avgTokensPerDep;
+  return budget;
+}
+function calculatePythonCohesion(exports, imports) {
+  if (exports.length === 0) return 1;
+  const exportCount = exports.length;
+  const importCount = imports.length;
+  let cohesion = 1;
+  if (exportCount > 10) {
+    cohesion *= 0.6;
+  } else if (exportCount > 5) {
+    cohesion *= 0.8;
+  }
+  if (exportCount > 0) {
+    const ratio = importCount / exportCount;
+    if (ratio > 2) {
+      cohesion *= 1.1;
+    } else if (ratio < 0.5) {
+      cohesion *= 0.9;
+    }
+  }
+  return Math.min(1, Math.max(0, cohesion));
+}
+export {
+  analyzePythonContext
+};

package/package.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "name": "@aiready/context-analyzer",
-  "version": "0.21.22",
+  "version": "0.21.24",
   "description": "AI context window cost analysis - detect fragmented code, deep import chains, and expensive context budgets",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -49,7 +49,7 @@
     "commander": "^14.0.0",
     "chalk": "^5.3.0",
     "prompts": "^2.4.2",
-    "@aiready/core": "0.23.19"
+    "@aiready/core": "0.23.21"
   },
   "devDependencies": {
     "@types/node": "^24.0.0",
@@ -65,7 +65,7 @@
   "scripts": {
     "build": "tsup src/index.ts src/cli.ts --format cjs,esm --dts",
     "dev": "tsup src/index.ts src/cli.ts --format cjs,esm --dts --watch",
-    "test": "vitest run",
+    "test": "vitest run --exclude \"**/dist/**\"",
     "lint": "eslint src",
     "clean": "rm -rf dist",
     "release": "pnpm build && pnpm publish --no-git-checks"

package/src/__tests__/analyzer.test.ts CHANGED Viewed

@@ -10,7 +10,7 @@ import {
 } from '../index';
 describe('buildDependencyGraph', () => {
-  it('should build a basic dependency graph', () => {
+  it('should build a basic dependency graph', async () => {
     const files = [
       {
         file: 'a.ts',
@@ -27,7 +27,7 @@ export const b = a + 1;
       },
     ];
-    const graph = buildDependencyGraph(files);
+    const graph = await buildDependencyGraph(files);
     expect(graph.nodes.size).toBe(2);
     expect(graph.edges.get('b.ts')?.has('a.ts')).toBe(true);
@@ -35,7 +35,7 @@ export const b = a + 1;
 });
 describe('calculateImportDepth', () => {
-  it('should calculate import depth correctly', () => {
+  it('should calculate import depth correctly', async () => {
     const files = [
       { file: 'a.ts', content: 'export const a = 1;' },
       {
@@ -48,14 +48,14 @@ describe('calculateImportDepth', () => {
       },
     ];
-    const graph = buildDependencyGraph(files);
+    const graph = await buildDependencyGraph(files);
     expect(calculateImportDepth('a.ts', graph)).toBe(0);
     expect(calculateImportDepth('b.ts', graph)).toBe(1);
     expect(calculateImportDepth('c.ts', graph)).toBe(2);
   });
-  it('should handle circular dependencies gracefully', () => {
+  it('should handle circular dependencies gracefully', async () => {
     const files = [
       {
         file: 'a.ts',
@@ -67,7 +67,7 @@ describe('calculateImportDepth', () => {
       },
     ];
-    const graph = buildDependencyGraph(files);
+    const graph = await buildDependencyGraph(files);
     // Should not infinite loop
     const depth = calculateImportDepth('a.ts', graph);
@@ -77,7 +77,7 @@ describe('calculateImportDepth', () => {
 });
 describe('getTransitiveDependencies', () => {
-  it('should get all transitive dependencies', () => {
+  it('should get all transitive dependencies', async () => {
     const files = [
       { file: 'a.ts', content: 'export const a = 1;' },
       {
@@ -90,7 +90,7 @@ describe('getTransitiveDependencies', () => {
       },
     ];
-    const graph = buildDependencyGraph(files);
+    const graph = await buildDependencyGraph(files);
     const deps = getTransitiveDependencies('c.ts', graph);
     expect(deps).toContain('b.ts');
@@ -100,7 +100,7 @@ describe('getTransitiveDependencies', () => {
 });
 describe('calculateContextBudget', () => {
-  it('should calculate total token cost including dependencies', () => {
+  it('should calculate total token cost including dependencies', async () => {
     const files = [
       { file: 'a.ts', content: 'export const a = 1;'.repeat(10) }, // ~40 tokens
       {
@@ -109,7 +109,7 @@ describe('calculateContextBudget', () => {
       }, // ~60 tokens
     ];
-    const graph = buildDependencyGraph(files);
+    const graph = await buildDependencyGraph(files);
     const budget = calculateContextBudget('b.ts', graph);
     // Should include both files' tokens
@@ -118,7 +118,7 @@ describe('calculateContextBudget', () => {
 });
 describe('detectCircularDependencies', () => {
-  it('should detect circular dependencies', () => {
+  it('should detect circular dependencies', async () => {
     const files = [
       {
         file: 'a.ts',
@@ -130,13 +130,13 @@ describe('detectCircularDependencies', () => {
       },
     ];
-    const graph = buildDependencyGraph(files);
+    const graph = await buildDependencyGraph(files);
     const cycles = detectCircularDependencies(graph);
     expect(cycles.length).toBeGreaterThan(0);
   });
-  it('should return empty for no circular dependencies', () => {
+  it('should return empty for no circular dependencies', async () => {
     const files = [
       { file: 'a.ts', content: 'export const a = 1;' },
       {
@@ -145,7 +145,7 @@ describe('detectCircularDependencies', () => {
       },
     ];
-    const graph = buildDependencyGraph(files);
+    const graph = await buildDependencyGraph(files);
     const cycles = detectCircularDependencies(graph);
     expect(cycles.length).toBe(0);