code2llm 0.5.145__tar.gz → 0.5.146__tar.gz

{code2llm-0.5.145/code2llm.egg-info → code2llm-0.5.146}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code2llm
-Version: 0.5.145
+Version: 0.5.146
 Summary: High-performance Python code flow analysis with optimized TOON format - CFG, DFG, call graphs, and intelligent code queries
 Home-page: https://github.com/wronai/stts
 Author: STTS Project
@@ -67,18 +67,16 @@ Dynamic: requires-python
 
 ## AI Cost Tracking
 
-![PyPI](https://img.shields.io/badge/pypi-costs-blue) ![Version](https://img.shields.io/badge/version-0.5.145-blue) ![Python](https://img.shields.io/badge/python-3.9+-blue) ![License](https://img.shields.io/badge/license-Apache--2.0-green)
-![AI Cost](https://img.shields.io/badge/AI%20Cost-$7.50-orange) ![Human Time](https://img.shields.io/badge/Human%20Time-71.2h-blue) ![Model](https://img.shields.io/badge/Model-openrouter%2Fqwen%2Fqwen3--coder--next-lightgrey)
+![PyPI](https://img.shields.io/badge/pypi-costs-blue) ![Version](https://img.shields.io/badge/version-0.5.146-blue) ![Python](https://img.shields.io/badge/python-3.9+-blue) ![License](https://img.shields.io/badge/license-Apache--2.0-green)
+![AI Cost](https://img.shields.io/badge/AI%20Cost-$7.50-orange) ![Human Time](https://img.shields.io/badge/Human%20Time-72.2h-blue) ![Model](https://img.shields.io/badge/Model-openrouter%2Fqwen%2Fqwen3--coder--next-lightgrey)
 
-- 🤖 **LLM usage:** $7.5000 (197 commits)
-- 👤 **Human dev:** ~$7124 (71.2h @ $100/h, 30min dedup)
+- 🤖 **LLM usage:** $7.5000 (198 commits)
+- 👤 **Human dev:** ~$7224 (72.2h @ $100/h, 30min dedup)
 
 Generated on 2026-05-06 using [openrouter/qwen/qwen3-coder-next](https://openrouter.ai/qwen/qwen3-coder-next)
 
 ---
 
-
-
 This directory contains the complete analysis of your project generated by `code2llm`. Each file serves a specific purpose for understanding, refactoring, and documenting your codebase.
 
 ## 📁 Generated Files Overview

{code2llm-0.5.145 → code2llm-0.5.146}/README.md

@@ -3,18 +3,16 @@
 
 ## AI Cost Tracking
 
-![PyPI](https://img.shields.io/badge/pypi-costs-blue) ![Version](https://img.shields.io/badge/version-0.5.145-blue) ![Python](https://img.shields.io/badge/python-3.9+-blue) ![License](https://img.shields.io/badge/license-Apache--2.0-green)
-![AI Cost](https://img.shields.io/badge/AI%20Cost-$7.50-orange) ![Human Time](https://img.shields.io/badge/Human%20Time-71.2h-blue) ![Model](https://img.shields.io/badge/Model-openrouter%2Fqwen%2Fqwen3--coder--next-lightgrey)
+![PyPI](https://img.shields.io/badge/pypi-costs-blue) ![Version](https://img.shields.io/badge/version-0.5.146-blue) ![Python](https://img.shields.io/badge/python-3.9+-blue) ![License](https://img.shields.io/badge/license-Apache--2.0-green)
+![AI Cost](https://img.shields.io/badge/AI%20Cost-$7.50-orange) ![Human Time](https://img.shields.io/badge/Human%20Time-72.2h-blue) ![Model](https://img.shields.io/badge/Model-openrouter%2Fqwen%2Fqwen3--coder--next-lightgrey)
 
-- 🤖 **LLM usage:** $7.5000 (197 commits)
-- 👤 **Human dev:** ~$7124 (71.2h @ $100/h, 30min dedup)
+- 🤖 **LLM usage:** $7.5000 (198 commits)
+- 👤 **Human dev:** ~$7224 (72.2h @ $100/h, 30min dedup)
 
 Generated on 2026-05-06 using [openrouter/qwen/qwen3-coder-next](https://openrouter.ai/qwen/qwen3-coder-next)
 
 ---
 
-
-
 This directory contains the complete analysis of your project generated by `code2llm`. Each file serves a specific purpose for understanding, refactoring, and documenting your codebase.
 
 ## 📁 Generated Files Overview

code2llm-0.5.146/VERSION

@@ -0,0 +1 @@
+0.5.146

{code2llm-0.5.145 → code2llm-0.5.146}/code2llm/__init__.py

@@ -8,7 +8,7 @@ Includes NLP Processing Pipeline for query normalization, intent matching,
 and entity resolution with multilingual support.
 """
 
-__version__ = "0.5.145"
+__version__ = "0.5.146"
 __author__ = "STTS Project"
 
 # Core analysis components (lightweight, always needed)

{code2llm-0.5.145 → code2llm-0.5.146}/code2llm/cli_exports/formats.py

@@ -2,6 +2,7 @@
 
 import os
 import sys
+import time
 from pathlib import Path
 from typing import Optional
 
@@ -13,6 +14,7 @@ from code2llm.exporters import (
     ArticleViewGenerator, HTMLDashboardGenerator,
     load_project_yaml, IndexHTMLGenerator,
 )
+from .orchestrator import _inject_generation_time as _inject_time
 
 
 def _export_evolution(args, result, output_dir: Path):
@@ -21,9 +23,12 @@ def _export_evolution(args, result, output_dir: Path):
         return
     exporter = EvolutionExporter()
     filepath = output_dir / 'evolution.toon.yaml'
+    t0 = time.monotonic()
     exporter.export(result, str(filepath))
+    elapsed = time.monotonic() - t0
+    _inject_time(filepath, elapsed)
     if args.verbose:
-        print(f"  - EVOLUTION (refactoring queue): {filepath}")
+        print(f"  - EVOLUTION (refactoring queue): {filepath} ({elapsed:.2f}s)")
 
 
 def _export_data_structures(args, result, output_dir: Path):
@@ -43,9 +48,12 @@ def _export_context_fallback(args, result, output_dir: Path, formats: list):
         return
     exporter = ContextExporter()
     filepath = output_dir / 'context.md'
+    t0 = time.monotonic()
     exporter.export(result, str(filepath))
+    elapsed = time.monotonic() - t0
+    _inject_time(filepath, elapsed)
     if args.verbose:
-        print(f"  - CONTEXT (LLM narrative): {filepath}")
+        print(f"  - CONTEXT (LLM narrative): {filepath} ({elapsed:.2f}s)")
 
 
 def _export_readme(args, result, output_dir: Path):
@@ -54,18 +62,24 @@ def _export_readme(args, result, output_dir: Path):
         return
     exporter = READMEExporter()
     filepath = output_dir / 'README.md'
+    t0 = time.monotonic()
     exporter.export(result, str(filepath))
+    elapsed = time.monotonic() - t0
+    _inject_time(filepath, elapsed)
     if args.verbose:
-        print(f"  - README (documentation): {filepath}")
+        print(f"  - README (documentation): {filepath} ({elapsed:.2f}s)")
 
 
 def _export_project_yaml(args, result, output_dir: Path):
     """Export unified project.yaml — single source of truth."""
     exporter = ProjectYAMLExporter()
     filepath = output_dir / 'project.yaml'
+    t0 = time.monotonic()
     exporter.export(result, str(filepath))
+    elapsed = time.monotonic() - t0
+    _inject_time(filepath, elapsed)
     if getattr(args, 'verbose', False):
-        print(f"  - PROJECT-YAML (single source of truth): {filepath}")
+        print(f"  - PROJECT-YAML (single source of truth): {filepath} ({elapsed:.2f}s)")
     return filepath
 
 
@@ -79,10 +93,13 @@ def _export_project_toon(args, result, output_dir: Path):
 
     exporter = ToonViewGenerator()
     filepath = output_dir / 'project.toon.yaml'
+    t0 = time.monotonic()
     exporter.generate(data, str(filepath))
+    elapsed = time.monotonic() - t0
+    _inject_time(filepath, elapsed)
 
     if getattr(args, 'verbose', False):
-        print(f"  - PROJECT-TOON (project overview): {filepath}")
+        print(f"  - PROJECT-TOON (project overview): {filepath} ({elapsed:.2f}s)")
 
     return filepath
 
@@ -129,9 +146,12 @@ def _export_simple_formats(args, result, output_dir: Path, formats):
             exporter = exporter_cls()
             # Export as plain text TOON format with .toon.yaml extension
             filepath = output_dir / filename
+            t0 = time.monotonic()
             exporter.export(result, str(filepath))
+            elapsed = time.monotonic() - t0
+            _inject_time(filepath, elapsed)
             if args.verbose:
-                print(f"  - {label}: {filepath}")
+                print(f"  - {label}: {filepath} ({elapsed:.2f}s)")
 
     # Unified project.yaml (single source of truth)
     if 'project-yaml' in formats:

{code2llm-0.5.145 → code2llm-0.5.146}/code2llm/cli_exports/orchestrator.py

@@ -7,6 +7,7 @@ Maintains backward compatibility with all existing --format values.
 import os
 import shutil
 import sys
+import time
 from pathlib import Path
 from typing import Optional, List, Dict, Any
 
@@ -280,9 +281,12 @@ def _export_registry_formats(args, result, output_dir: Path, formats: List[str])
         kwargs = _get_format_kwargs(fmt, args)
 
         try:
+            t0 = time.monotonic()
             exporter.export(result, str(filepath), **kwargs)
+            elapsed = time.monotonic() - t0
+            _inject_generation_time(filepath, elapsed)
             if args.verbose:
-                print(f"  - {label}: {filepath}")
+                print(f"  - {label}: {filepath} ({elapsed:.2f}s)")
         except Exception as e:
             if args.verbose:
                 print(f"  - {label} export failed: {e}", file=sys.stderr)
@@ -306,6 +310,53 @@ def _export_chunked(
     _chunked_impl(args, result, output_dir, source_path, formats, requested_formats)
 
 
+def _inject_generation_time(filepath: Path, elapsed: float) -> None:
+    """Inject generation time comment into the second line of an exported file."""
+    try:
+        path = Path(filepath)
+        if not path.exists():
+            return
+        suffix = path.suffix.lower()
+        name = path.name.lower()
+
+        # Only inject into text-based files
+        if suffix not in ('.yaml', '.yml', '.md', '.txt', '.mmd', '.html', '.json', '.export'):
+            return
+
+        content = path.read_text(encoding='utf-8')
+        if not content:
+            return
+
+        tag = f"generated in {elapsed:.2f}s"
+
+        if suffix in ('.yaml', '.yml', '.mmd', '.export', '.txt') or name.endswith('.toon.yaml'):
+            # YAML/Mermaid/text: insert '# generated in X.XXs' after first line
+            lines = content.split('\n', 1)
+            if len(lines) == 2:
+                content = f"{lines[0]}\n# {tag}\n{lines[1]}"
+            else:
+                content = f"{lines[0]}\n# {tag}\n"
+        elif suffix == '.md':
+            # Markdown: insert HTML comment after first line
+            lines = content.split('\n', 1)
+            if len(lines) == 2:
+                content = f"{lines[0]}\n<!-- {tag} -->\n{lines[1]}"
+            else:
+                content = f"{lines[0]}\n<!-- {tag} -->\n"
+        elif suffix == '.html':
+            # HTML: insert comment after <!DOCTYPE or <html>
+            content = content.replace('\n', f'\n<!-- {tag} -->\n', 1)
+        elif suffix == '.json':
+            # JSON doesn't support comments — skip
+            return
+        else:
+            return
+
+        path.write_text(content, encoding='utf-8')
+    except Exception:
+        pass  # Never fail the export pipeline for a comment
+
+
 # Backward-compatible aliases
 _export_single_project = _export_single
 _export_chunked_results = _export_chunked

{code2llm-0.5.145 → code2llm-0.5.146}/code2llm/cli_exports/orchestrator_handlers.py

@@ -4,6 +4,7 @@ This module contains all the individual export handler functions
 that were extracted from orchestrator.py to reduce its size.
 """
 
+import time
 from pathlib import Path
 from typing import Any, Dict, List
 
@@ -20,10 +21,14 @@ from .orchestrator_constants import FORMAT_LABELS
 
 def _export_mermaid(args, result, output_dir: Path):
     """Export mermaid diagrams."""
+    from .orchestrator import _inject_generation_time
+
     exporter = MermaidExporter()
     include_examples = getattr(args, 'flow_include_examples', False)
 
     # Core diagrams
+    mmd_files = ['flow.mmd', 'calls.mmd', 'compact_flow.mmd']
+    t0 = time.monotonic()
     exporter.export_flow_compact(result, str(output_dir / 'flow.mmd'), include_examples)
     exporter.export_call_graph(result, str(output_dir / 'calls.mmd'))
     exporter.export_compact(result, str(output_dir / 'compact_flow.mmd'))
@@ -31,21 +36,27 @@ def _export_mermaid(args, result, output_dir: Path):
     # Optional detailed diagrams
     if getattr(args, 'flow_detail', False):
         exporter.export_flow_detailed(result, str(output_dir / 'flow_detailed.mmd'), include_examples)
+        mmd_files.append('flow_detailed.mmd')
     if getattr(args, 'flow_full', False):
         exporter.export_flow_full(result, str(output_dir / 'flow_full.mmd'), include_examples)
+        mmd_files.append('flow_full.mmd')
+    elapsed_mmd = time.monotonic() - t0
+
+    # Inject timing into each .mmd file
+    for mf in mmd_files:
+        _inject_generation_time(output_dir / mf, elapsed_mmd)
 
     # Also export calls.yaml/toon
     yaml_exporter = YAMLExporter()
+    t0 = time.monotonic()
     yaml_exporter.export_calls(result, str(output_dir / 'calls.yaml'))
     yaml_exporter.export_calls_toon(result, str(output_dir / 'calls.toon.yaml'))
+    elapsed_calls = time.monotonic() - t0
+    _inject_generation_time(output_dir / 'calls.toon.yaml', elapsed_calls)
 
     if args.verbose:
-        files = ['flow.mmd', 'calls.mmd', 'compact_flow.mmd', 'calls.yaml']
-        if getattr(args, 'flow_detail', False):
-            files.append('flow_detailed.mmd')
-        if getattr(args, 'flow_full', False):
-            files.append('flow_full.mmd')
-        print(f"  - Mermaid: {output_dir}/*.mmd ({', '.join(files)})")
+        files = mmd_files + ['calls.yaml']
+        print(f"  - Mermaid: {output_dir}/*.mmd ({', '.join(files)}) ({elapsed_mmd:.2f}s)")
 
     # PNG generation
     _export_mermaid_pngs(args, output_dir)
@@ -99,28 +110,38 @@ def _export_data_structures(args, result, output_dir: Path):
 def _export_project_toon(args, result, output_dir: Path):
     """Export project.toon.yaml from project.yaml data."""
     from ..exporters.project_yaml_exporter import ProjectYAMLExporter
+    from .orchestrator import _inject_generation_time
 
     project_yaml_exporter = ProjectYAMLExporter()
     prev_evolution = load_previous_evolution(output_dir / 'project.yaml')
     data = project_yaml_exporter._build_project_yaml(result, prev_evolution)
 
     generator = ToonViewGenerator()
-    generator.generate(data, str(output_dir / 'project.toon.yaml'))
+    filepath = output_dir / 'project.toon.yaml'
+    t0 = time.monotonic()
+    generator.generate(data, str(filepath))
+    elapsed = time.monotonic() - t0
+    _inject_generation_time(filepath, elapsed)
 
     if args.verbose:
-        print(f"  - PROJECT-TOON (project overview): {output_dir / 'project.toon.yaml'}")
+        print(f"  - PROJECT-TOON (project overview): {filepath} ({elapsed:.2f}s)")
 
 
 def _export_readme(args, result, output_dir: Path):
     """Export README.md."""
     if getattr(args, 'no_readme', False):
         return
+    from .orchestrator import _inject_generation_time
     exporter_cls = get_exporter('readme')
     if exporter_cls:
         exporter = exporter_cls()
-        exporter.export(result, str(output_dir / 'README.md'))
+        filepath = output_dir / 'README.md'
+        t0 = time.monotonic()
+        exporter.export(result, str(filepath))
+        elapsed = time.monotonic() - t0
+        _inject_generation_time(filepath, elapsed)
         if args.verbose:
-            print(f"  - README (documentation): {output_dir / 'README.md'}")
+            print(f"  - README (documentation): {filepath} ({elapsed:.2f}s)")
 
 
 def _export_index_html(args, output_dir: Path):
@@ -128,10 +149,14 @@ def _export_index_html(args, output_dir: Path):
     if 'all' not in getattr(args, 'format', ''):
         return
     try:
+        from .orchestrator import _inject_generation_time
         generator = IndexHTMLGenerator(output_dir)
+        t0 = time.monotonic()
         index_path = generator.generate()
+        elapsed = time.monotonic() - t0
+        _inject_generation_time(index_path, elapsed)
         if args.verbose:
-            print(f"  - INDEX (file browser): {index_path}")
+            print(f"  - INDEX (file browser): {index_path} ({elapsed:.2f}s)")
     except Exception as e:
         if args.verbose:
             print(f"  - INDEX generation failed: {e}", file=__import__('sys').stderr)

{code2llm-0.5.145 → code2llm-0.5.146}/code2llm/cli_exports/prompt.py

@@ -1,6 +1,7 @@
 """Prompt generation — prompt.txt for LLM consumption (regular and chunked)."""
 
 import sys
+import time
 from pathlib import Path
 from typing import List, Optional, Tuple
 
@@ -30,9 +31,13 @@ def _export_prompt_txt(args, output_dir: Path, formats: list[str], source_path:
     lines.extend(_build_prompt_footer(chunked=False, file_analysis=file_analysis))
 
     prompt_path = output_dir / 'prompt.txt'
+    t0 = time.monotonic()
     prompt_path.write_text("\n".join(lines) + "\n", encoding='utf-8')
+    elapsed = time.monotonic() - t0
+    from .orchestrator import _inject_generation_time
+    _inject_generation_time(prompt_path, elapsed)
     if args.verbose:
-        print(f"  - PROMPT: {prompt_path}")
+        print(f"  - PROMPT: {prompt_path} ({elapsed:.2f}s)")
 
 
 def _export_chunked_prompt_txt(args, output_dir: Path, formats: list[str], source_path: Optional[Path] = None, subprojects: list = None) -> None:

{code2llm-0.5.145 → code2llm-0.5.146}/code2llm/exporters/project_yaml/core.py

@@ -48,7 +48,7 @@ class ProjectYAMLExporter(BaseExporter):
         self, result: AnalysisResult, prev_evolution: List[Dict]
     ) -> Dict[str, Any]:
         """Build complete project.yaml structure."""
-        line_counts = _scan_line_counts(result.project_path)
+        line_counts = _scan_line_counts(result.project_path, result=result)
         # Filter out venv/site-packages/etc — only count lines of non-excluded files
         filtered_lines = {
             k: v for k, v in line_counts.items()

{code2llm-0.5.145 → code2llm-0.5.146}/code2llm/exporters/toon/helpers.py

@@ -83,15 +83,12 @@ def _hotspot_description(fi: FunctionInfo, fan_out: int) -> str:
     return f"calls {fan_out} functions"
 
 
-@lru_cache(maxsize=8)
-def _scan_line_counts(project_path) -> Dict[str, int]:
-    """Scan project directory for all source file line counts.
+def _scan_line_counts(project_path, result=None) -> Dict[str, int]:
+    """Get line counts for project files.
 
-    Cached per project_path — callers within the same process reuse the result
-    instead of repeating the rglob + read_text I/O.
+    Fast path: derive from AnalysisResult modules (already parsed, no extra I/O).
+    Slow fallback: single os.walk pass reading files from disk.
     """
-    from ...core.config import ALL_EXTENSIONS
-
     line_counts: Dict[str, int] = {}
     if not project_path:
         return line_counts
@@ -99,8 +96,35 @@ def _scan_line_counts(project_path) -> Dict[str, int]:
     if not pp.is_dir():
         return line_counts
 
-    for ext in ALL_EXTENSIONS:
-        for src_file in pp.rglob(f"*{ext}"):
+    # Fast path: use already-analyzed file data when available
+    if result is not None:
+        for mname, mi in getattr(result, 'modules', {}).items():
+            fpath = mi.file
+            if not fpath:
+                continue
+            try:
+                lc = len(Path(fpath).read_text(encoding="utf-8", errors="ignore").splitlines())
+                rel = str(Path(fpath).relative_to(pp))
+                line_counts[str(fpath)] = lc
+                line_counts[rel] = lc
+            except Exception:
+                pass
+        return line_counts
+
+    # Slow fallback: single walk instead of 73 rglob calls
+    from ...core.config import ALL_EXTENSIONS
+    ext_set = set(ALL_EXTENSIONS)
+    for root, dirs, files in Path(project_path).walk() if hasattr(Path, 'walk') else _walk_compat(pp):
+        # Prune excluded directories
+        dirs[:] = [d for d in dirs if d not in {
+            'venv', '.venv', 'node_modules', '__pycache__', '.git',
+            'dist', 'build', '.tox', '.mypy_cache', 'egg-info',
+        }]
+        for fname in files:
+            ext = Path(fname).suffix
+            if ext not in ext_set:
+                continue
+            src_file = root / fname
             try:
                 lc = len(src_file.read_text(encoding="utf-8", errors="ignore").splitlines())
                 rel = str(src_file.relative_to(pp))
@@ -109,3 +133,10 @@ def _scan_line_counts(project_path) -> Dict[str, int]:
             except Exception:
                 pass
     return line_counts
+
+
+def _walk_compat(path):
+    """os.walk compatibility wrapper for Path (Python < 3.12)."""
+    import os
+    for root, dirs, files in os.walk(path):
+        yield Path(root), dirs, files

{code2llm-0.5.145 → code2llm-0.5.146}/code2llm/exporters/toon/metrics.py

@@ -35,7 +35,7 @@ class MetricsComputer:
         """Compute all metrics and return context dict."""
         self.result = result
         self.project_path = result.project_path
-        self.line_counts = _scan_line_counts(self.project_path)
+        self.line_counts = _scan_line_counts(self.project_path, result=result)
 
         # Initialize specialized computers
         self._core = CoreMetricsComputer(self.line_counts, self.project_path)

{code2llm-0.5.145 → code2llm-0.5.146}/code2llm/exporters/toon/metrics_core.py

@@ -80,55 +80,46 @@ class CoreMetricsComputer:
             "fan_in": 0,
         }
 
+    @staticmethod
+    def _build_suffix_index(result: AnalysisResult) -> Dict[str, List[Tuple[str, FunctionInfo]]]:
+        """Build reverse lookup: simple_name -> [(qualified_name, FunctionInfo)].
+
+        Replaces O(F) scans per call with O(1) dict lookups.
+        """
+        idx: Dict[str, List[Tuple[str, FunctionInfo]]] = defaultdict(list)
+        for qn, fi in result.functions.items():
+            simple = qn.rsplit(".", 1)[-1]
+            idx[simple].append((qn, fi))
+        return idx
+
     @staticmethod
     def _compute_fan_in(files: Dict, result: AnalysisResult) -> None:
         """Compute fan-in per file (how many other files call into this file)."""
         importers: Dict[str, set] = defaultdict(set)
+        suffix_idx = CoreMetricsComputer._build_suffix_index(result)
 
         for fname, fi in result.functions.items():
-            CoreMetricsComputer._process_function_calls(fi, result, importers)
+            src_file = fi.file
+            # Forward: who calls me? (called_by)
+            for callee in fi.called_by:
+                callee_info = result.functions.get(callee)
+                if callee_info and callee_info.file != src_file:
+                    importers[src_file].add(callee_info.file)
+            # Reverse: who do I call? → target file gets fan-in
+            for callee in fi.calls:
+                callee_info = result.functions.get(callee)
+                if callee_info and callee_info.file != src_file:
+                    importers[callee_info.file].add(src_file)
+                elif not callee_info:
+                    # O(1) suffix lookup instead of O(F) scan
+                    for qn, ci in suffix_idx.get(callee, []):
+                        if ci.file != src_file:
+                            importers[ci.file].add(src_file)
+                            break
 
         for fpath in files:
             files[fpath]["fan_in"] = len(importers.get(fpath, set()))
 
-    @staticmethod
-    def _process_function_calls(fi: FunctionInfo, result: AnalysisResult, importers: Dict[str, set]) -> None:
-        """Process calls for a single function to compute fan-in."""
-        src_file = fi.file
-
-        # Forward: who calls me? (called_by)
-        CoreMetricsComputer._process_called_by(fi, result, src_file, importers)
-
-        # Reverse: who do I call? → target file gets fan-in
-        CoreMetricsComputer._process_callee_calls(fi, result, src_file, importers)
-
-    @staticmethod
-    def _process_called_by(fi: FunctionInfo, result: AnalysisResult, src_file: str, importers: Dict[str, set]) -> None:
-        """Process called_by relationships."""
-        for callee in fi.called_by:
-            callee_info = result.functions.get(callee)
-            if callee_info and callee_info.file != src_file:
-                importers[src_file].add(callee_info.file)
-
-    @staticmethod
-    def _process_callee_calls(fi: FunctionInfo, result: AnalysisResult, src_file: str, importers: Dict[str, set]) -> None:
-        """Process callee relationships."""
-        for callee in fi.calls:
-            callee_info = result.functions.get(callee)
-            if callee_info and callee_info.file != src_file:
-                importers[callee_info.file].add(src_file)
-            else:
-                # Suffix match for unqualified names
-                CoreMetricsComputer._handle_suffix_match(callee, result, src_file, importers)
-
-    @staticmethod
-    def _handle_suffix_match(callee: str, result: AnalysisResult, src_file: str, importers: Dict[str, set]) -> None:
-        """Handle suffix matching for unqualified names."""
-        for qn, ci in result.functions.items():
-            if qn.endswith(f".{callee}") and ci.file != src_file:
-                importers[ci.file].add(src_file)
-                break
-
     def compute_package_metrics(
         self, files: Dict[str, Dict], result: AnalysisResult
     ) -> Dict[str, Dict[str, Any]]:
@@ -248,6 +239,12 @@ class CoreMetricsComputer:
         """Build coupling matrix from cross-module calls."""
         matrix: Dict[Tuple[str, str], int] = defaultdict(int)
 
+        # Build reverse index: simple_name -> [(qualified, module)] for O(1) lookup
+        suffix_idx: Dict[str, List[Tuple[str, str]]] = defaultdict(list)
+        for qn, mod in func_to_module.items():
+            simple = qn.rsplit(".", 1)[-1]
+            suffix_idx[simple].append((qn, mod))
+
         for qname, fi in result.functions.items():
             if _is_excluded(fi.file):
                 continue
@@ -255,7 +252,9 @@ class CoreMetricsComputer:
             src_pkg = _package_of_module(src_mod)
 
             for callee in fi.calls:
-                callee_mod = self._resolve_callee_module(callee, func_to_module, src_pkg)
+                callee_mod = self._resolve_callee_module(
+                    callee, func_to_module, src_pkg, suffix_idx
+                )
                 if callee_mod and callee_mod != src_mod:
                     dst_pkg = _package_of_module(callee_mod)
                     if dst_pkg and dst_pkg != src_pkg:
@@ -263,31 +262,24 @@ class CoreMetricsComputer:
 
         return matrix
 
-    def _resolve_callee_module(self, callee: str, func_to_module: Dict[str, str], src_pkg: str) -> Optional[str]:
-        """Resolve callee to a known function module."""
+    def _resolve_callee_module(
+        self, callee: str, func_to_module: Dict[str, str],
+        src_pkg: str, suffix_idx: Dict[str, List[Tuple[str, str]]],
+    ) -> Optional[str]:
+        """Resolve callee to a known function module (O(1) via suffix index)."""
         callee_mod = func_to_module.get(callee)
         if callee_mod:
             return callee_mod
 
-        # Try suffix match — collect all candidates
-        candidates = [
-            (qn, mod) for qn, mod in func_to_module.items()
-            if qn.endswith(f".{callee}")
-        ]
-
+        candidates = suffix_idx.get(callee, [])
         if len(candidates) == 1:
             return candidates[0][1]
         elif candidates:
             # Prefer callee in same package as caller
-            same_pkg = [
-                (qn, mod) for qn, mod in candidates
-                if _package_of_module(mod) == src_pkg
-            ]
-            if same_pkg:
-                return same_pkg[0][1]
-            else:
-                # Pick first cross-package candidate
-                return candidates[0][1]
+            for qn, mod in candidates:
+                if _package_of_module(mod) == src_pkg:
+                    return mod
+            return candidates[0][1]
 
         return None
 

{code2llm-0.5.145 → code2llm-0.5.146}/code2llm/nlp/__init__.py

@@ -4,7 +4,7 @@ Provides query normalization, intent matching, and entity resolution
 with multilingual support and fuzzy matching.
 """
 
-__version__ = "0.5.145"
+__version__ = "0.5.146"
 
 from .pipeline import NLPPipeline
 from .normalization import QueryNormalizer