codegraph-gen 0.2.0__tar.gz → 1.1.0__tar.gz

{codegraph_gen-0.2.0 → codegraph_gen-1.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: codegraph-gen
-Version: 0.2.0
+Version: 1.1.0
 Summary: AST-based codebase knowledge graph generator in Markdown
 Keywords: knowledge-graph,ast,codebase,markdown,tree-sitter,visualization,static-analysis,ai-agent,obsidian
 Author: twn39
@@ -10,6 +10,7 @@ Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Software Development :: Code Generators
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
@@ -27,23 +28,30 @@ Requires-Dist: pydantic>=2.0.0
 Requires-Dist: tree-sitter-c>=0.24.2
 Requires-Dist: tree-sitter-cpp>=0.23.4
 Requires-Dist: tree-sitter-kotlin>=1.1.0
-Requires-Python: >=3.12
+Requires-Python: >=3.11
 Project-URL: Homepage, https://github.com/twn39/codegraph
 Project-URL: Repository, https://github.com/twn39/codegraph
 Project-URL: Issues, https://github.com/twn39/codegraph/issues
 Description-Content-Type: text/markdown
 
-# codegraph
+<h1 align="center">codegraph-gen</h1>
 
-`codegraph` 是一个面向 AI Agent（如 Antigravity、Codex、Claude Code 等）的静态代码知识图谱生成工具。它能够静态解析多语言 codebase，通过社区发现算法自动进行组件聚类，并导出为由标准 Markdown 文件组成的关联图谱库（Obsidian-like vault），极大地辅助 AI Agent 在本地进行精准的架构理解、逻辑导航与深度洞察分析。
+<p align="center">
+  <a href="https://pypi.org/project/codegraph-gen/"><img src="https://img.shields.io/pypi/v/codegraph-gen.svg" alt="PyPI Version"></a>
+  <img src="https://img.shields.io/badge/python-3.11+-blue.svg" alt="Python Version">
+  <img src="https://img.shields.io/badge/code%20style-ruff-26b22f.svg" alt="Ruff">
+  <img src="https://img.shields.io/badge/license-MIT-green.svg" alt="License">
+</p>
 
-与基于图形化 Canvas 渲染的知识图谱不同，`codegraph` 采用全 Markdown 的扁平结构存储。它专门为 LLM 设计，摒弃了昂贵且复杂的数据库依赖，让 AI Agent 可以通过标准文件读取与路径导航（Relative Links）轻松周游整个代码库。
+`codegraph-gen` 是一个面向 AI Agent（如 Antigravity、Codex、Claude Code 等）的静态代码知识图谱生成工具。它能够静态解析多语言 codebase，通过社区发现算法自动进行组件聚类，并导出为由标准 Markdown 文件组成的关联图谱库（Obsidian-like vault），极大地辅助 AI Agent 在本地进行精准的架构理解、逻辑导航与深度洞察分析。
+
+与基于图形化 Canvas 渲染的知识图谱不同，`codegraph-gen` 采用全 Markdown 的扁平结构存储。它专门为 LLM 设计，摒弃了昂贵且复杂的数据库依赖，让 AI Agent 可以通过标准文件读取与路径导航（Relative Links）轻松周游整个代码库。
 
 ---
 
 ## 🚀 核心特性
 
-- **多语言 AST 解析**：基于 `tree-sitter`，原生支持 **Python, JavaScript, TypeScript, Go, Rust, Swift**。
+- **多语言 AST 解析**：基于 `tree-sitter`，原生支持 **Python, JavaScript, TypeScript, Kotlin, Go, Rust, Swift**。
 - **语义边解析与绑定**：静态解析跨文件的函数/方法调用（`calls`）、类型继承/接口实现（`inherits`/`implements`）以及文件导入关系（`imports`）。
 - **逻辑组件自动聚类**：利用贪心模块度社区发现算法（Louvain Modularity Clustering）将紧密耦合的文件和符号自动聚类为 **Component（逻辑组件）**，并根据组件核心节点智能命名。
 - **架构脆弱性分析**：自动识别 **God Nodes（度数最高的核心抽象）**，并静态检测文件级别的 **循环导入依赖（Circular Imports）**。
@@ -53,39 +61,29 @@ Description-Content-Type: text/markdown
 
 ## 📦 架构概览
 
-```mermaid
-graph TD
-    A[工作区源码 Workspace] --> B[detect: 语言识别与过滤]
-    B --> C[parser: Tree-Sitter AST 符号提取]
-    C --> D[builder: NetworkX 语义图组装与绑定]
-    D --> E[cluster: 社区模块度聚类命名]
-    E --> F[analyze: 上帝节点与循环导入分析]
-    F --> G[export: 导出至 .codegraph/]
-    G --> H[AGENT_PROMPT.md / AGENTS.md / README.md / nodes / components]
-```
+- **工作区源码 Workspace** -> **detect: 语言识别与过滤** -> **parser: Tree-Sitter AST 符号提取** -> **builder: NetworkX 语义图组装与绑定** -> **cluster: 社区模块度聚类命名** -> **analyze: 上帝节点与循环导入分析** -> **export: 导出至 .codegraph/** -> **生成 AGENT_PROMPT.md / AGENTS.md / README.md / nodes / components**
 
 ---
 
 ## 🛠️ 安装指南
 
-推荐使用 [uv](https://github.com/astral-sh/uv) 管理项目依赖与虚拟环境：
+推荐使用 [uv](https://github.com/astral-sh/uv) 直接安装工具：
 
 ```bash
-# 克隆仓库
-git clone <repository-url>
-cd codegraph
+# 全局安装 (推荐)
+uv tool install codegraph-gen
+```
 
-# 同步依赖并激活虚拟环境
-uv sync
-source .venv/bin/activate
+如果需要在当前虚拟环境中作为库依赖安装：
 
-# 全局安装 (推荐)
-uv tool install --force --no-cache .
+```bash
+# 安装至当前项目/环境
+uv pip install codegraph-gen
 ```
 
-### 2. 注入 AI Agent 斜杠命令集成
+### 注册 AI Agent 斜杠命令
 
-`codegraph` 支持一键将 `/codegraph` 自定义斜杠命令注册到您的 AI Agent（如 Codex 或 Antigravity）的全局配置中：
+`codegraph-gen` 支持一键将 `/codegraph` 自定义斜杠命令注册到您的 AI Agent（如 Codex 或 Antigravity）的全局配置中：
 
 ```bash
 # 为 Codex / Antigravity 注入 /codegraph 全局斜杠命令
@@ -130,7 +128,7 @@ codegraph build . --exclude extra_folder/ --exclude docs/
 
 ## 🤖 与 AI Agent（Codex / Antigravity / Claude Code）协同分析
 
-`codegraph` 的核心设计思想是**离线构建，Agent 分析**。这避免了在 CLI 中直接硬编码大模型 API，降低了使用成本，并充分利用了你当前对话中功能更强、带有上下文读取能力的外部 Agent。
+`codegraph-gen` 的核心设计思想是**离线构建，Agent 分析**。这避免了在 CLI 中直接硬编码大模型 API，降低了使用成本，并充分利用了你当前对话中功能更强、带有上下文读取能力的外部 Agent。
 
 ### 步骤 1：生成本地图谱
 

{codegraph_gen-0.2.0 → codegraph_gen-1.1.0}/README.md

@@ -1,14 +1,21 @@
-# codegraph
+<h1 align="center">codegraph-gen</h1>
 
-`codegraph` 是一个面向 AI Agent（如 Antigravity、Codex、Claude Code 等）的静态代码知识图谱生成工具。它能够静态解析多语言 codebase，通过社区发现算法自动进行组件聚类，并导出为由标准 Markdown 文件组成的关联图谱库（Obsidian-like vault），极大地辅助 AI Agent 在本地进行精准的架构理解、逻辑导航与深度洞察分析。
+<p align="center">
+  <a href="https://pypi.org/project/codegraph-gen/"><img src="https://img.shields.io/pypi/v/codegraph-gen.svg" alt="PyPI Version"></a>
+  <img src="https://img.shields.io/badge/python-3.11+-blue.svg" alt="Python Version">
+  <img src="https://img.shields.io/badge/code%20style-ruff-26b22f.svg" alt="Ruff">
+  <img src="https://img.shields.io/badge/license-MIT-green.svg" alt="License">
+</p>
 
-与基于图形化 Canvas 渲染的知识图谱不同，`codegraph` 采用全 Markdown 的扁平结构存储。它专门为 LLM 设计，摒弃了昂贵且复杂的数据库依赖，让 AI Agent 可以通过标准文件读取与路径导航（Relative Links）轻松周游整个代码库。
+`codegraph-gen` 是一个面向 AI Agent（如 Antigravity、Codex、Claude Code 等）的静态代码知识图谱生成工具。它能够静态解析多语言 codebase，通过社区发现算法自动进行组件聚类，并导出为由标准 Markdown 文件组成的关联图谱库（Obsidian-like vault），极大地辅助 AI Agent 在本地进行精准的架构理解、逻辑导航与深度洞察分析。
+
+与基于图形化 Canvas 渲染的知识图谱不同，`codegraph-gen` 采用全 Markdown 的扁平结构存储。它专门为 LLM 设计，摒弃了昂贵且复杂的数据库依赖，让 AI Agent 可以通过标准文件读取与路径导航（Relative Links）轻松周游整个代码库。
 
 ---
 
 ## 🚀 核心特性
 
-- **多语言 AST 解析**：基于 `tree-sitter`，原生支持 **Python, JavaScript, TypeScript, Go, Rust, Swift**。
+- **多语言 AST 解析**：基于 `tree-sitter`，原生支持 **Python, JavaScript, TypeScript, Kotlin, Go, Rust, Swift**。
 - **语义边解析与绑定**：静态解析跨文件的函数/方法调用（`calls`）、类型继承/接口实现（`inherits`/`implements`）以及文件导入关系（`imports`）。
 - **逻辑组件自动聚类**：利用贪心模块度社区发现算法（Louvain Modularity Clustering）将紧密耦合的文件和符号自动聚类为 **Component（逻辑组件）**，并根据组件核心节点智能命名。
 - **架构脆弱性分析**：自动识别 **God Nodes（度数最高的核心抽象）**，并静态检测文件级别的 **循环导入依赖（Circular Imports）**。
@@ -18,39 +25,29 @@
 
 ## 📦 架构概览
 
-```mermaid
-graph TD
-    A[工作区源码 Workspace] --> B[detect: 语言识别与过滤]
-    B --> C[parser: Tree-Sitter AST 符号提取]
-    C --> D[builder: NetworkX 语义图组装与绑定]
-    D --> E[cluster: 社区模块度聚类命名]
-    E --> F[analyze: 上帝节点与循环导入分析]
-    F --> G[export: 导出至 .codegraph/]
-    G --> H[AGENT_PROMPT.md / AGENTS.md / README.md / nodes / components]
-```
+- **工作区源码 Workspace** -> **detect: 语言识别与过滤** -> **parser: Tree-Sitter AST 符号提取** -> **builder: NetworkX 语义图组装与绑定** -> **cluster: 社区模块度聚类命名** -> **analyze: 上帝节点与循环导入分析** -> **export: 导出至 .codegraph/** -> **生成 AGENT_PROMPT.md / AGENTS.md / README.md / nodes / components**
 
 ---
 
 ## 🛠️ 安装指南
 
-推荐使用 [uv](https://github.com/astral-sh/uv) 管理项目依赖与虚拟环境：
+推荐使用 [uv](https://github.com/astral-sh/uv) 直接安装工具：
 
 ```bash
-# 克隆仓库
-git clone <repository-url>
-cd codegraph
+# 全局安装 (推荐)
+uv tool install codegraph-gen
+```
 
-# 同步依赖并激活虚拟环境
-uv sync
-source .venv/bin/activate
+如果需要在当前虚拟环境中作为库依赖安装：
 
-# 全局安装 (推荐)
-uv tool install --force --no-cache .
+```bash
+# 安装至当前项目/环境
+uv pip install codegraph-gen
 ```
 
-### 2. 注入 AI Agent 斜杠命令集成
+### 注册 AI Agent 斜杠命令
 
-`codegraph` 支持一键将 `/codegraph` 自定义斜杠命令注册到您的 AI Agent（如 Codex 或 Antigravity）的全局配置中：
+`codegraph-gen` 支持一键将 `/codegraph` 自定义斜杠命令注册到您的 AI Agent（如 Codex 或 Antigravity）的全局配置中：
 
 ```bash
 # 为 Codex / Antigravity 注入 /codegraph 全局斜杠命令
@@ -95,7 +92,7 @@ codegraph build . --exclude extra_folder/ --exclude docs/
 
 ## 🤖 与 AI Agent（Codex / Antigravity / Claude Code）协同分析
 
-`codegraph` 的核心设计思想是**离线构建，Agent 分析**。这避免了在 CLI 中直接硬编码大模型 API，降低了使用成本，并充分利用了你当前对话中功能更强、带有上下文读取能力的外部 Agent。
+`codegraph-gen` 的核心设计思想是**离线构建，Agent 分析**。这避免了在 CLI 中直接硬编码大模型 API，降低了使用成本，并充分利用了你当前对话中功能更强、带有上下文读取能力的外部 Agent。
 
 ### 步骤 1：生成本地图谱
 

{codegraph_gen-0.2.0 → codegraph_gen-1.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codegraph-gen"
-version = "0.2.0"
+version = "1.1.0"
 description = "AST-based codebase knowledge graph generator in Markdown"
 readme = "README.md"
 authors = [
@@ -23,11 +23,12 @@ classifiers = [
     "Intended Audience :: Developers",
     "License :: OSI Approved :: MIT License",
     "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
     "Topic :: Software Development :: Code Generators",
     "Topic :: Software Development :: Libraries :: Python Modules",
 ]
-requires-python = ">=3.12"
+requires-python = ">=3.11"
 dependencies = [
     "networkx>=3.0",
     "tree-sitter>=0.23.0",
@@ -59,7 +60,6 @@ Issues = "https://github.com/twn39/codegraph/issues"
 
 [project.scripts]
 codegraph = "codegraph_gen.__main__:main"
-codegraph-gen = "codegraph_gen.__main__:main"
 
 [build-system]
 requires = ["uv_build>=0.9.26,<0.10.0"]

{codegraph_gen-0.2.0 → codegraph_gen-1.1.0}/src/codegraph_gen/__main__.py

@@ -93,7 +93,7 @@ def build(
 
     from codegraph_gen.engine import CodegraphEngine, PipelineStage
 
-    engine = CodegraphEngine(config)
+    engine = CodegraphEngine()
 
     # Run pipeline with click progress bar
     with Progress(
@@ -129,7 +129,7 @@ def build(
             elif stage == PipelineStage.COMPLETED:
                 progress.update(task, description="Done!")
 
-        result = engine.run_pipeline(progress_callback=progress_callback)
+        result = engine.run_pipeline(config, progress_callback=progress_callback)
 
     G = result.graph
     if G.number_of_nodes() == 0:
@@ -294,12 +294,12 @@ def info():
     try:
         from importlib.metadata import version
 
-        ver = version("codegraph")
+        ver = version("codegraph-gen")
     except Exception:
-        ver = "0.2.0"
+        ver = "1.1.0"
     console.print(f"[bold]codegraph v{ver}[/bold]")
     console.print(
-        "Supported languages: Python, JavaScript, TypeScript, Go, Rust, Swift"
+        "Supported languages: Python, JavaScript, TypeScript, Kotlin, Go, Rust, Swift"
     )
 
 

codegraph_gen-1.1.0/src/codegraph_gen/builder.py

@@ -0,0 +1,27 @@
+import logging
+from pathlib import Path
+import networkx as nx
+from codegraph_gen.schema import ExtractionResult
+from codegraph_gen.resolver import TypeResolver
+
+logger = logging.getLogger(__name__)
+
+
+def build_graph(extractions: list[ExtractionResult], workspace_dir: Path) -> nx.DiGraph:
+    """
+    Assembles a list of ExtractionResults into a single directed graph
+    and resolves call, inherit, and import edges using a two-pass scope resolver.
+    """
+    G = nx.DiGraph()
+
+    # 1. Add all nodes to the graph
+    for ext in extractions:
+        for node in ext.nodes:
+            G.add_node(node.id, **node.model_dump())
+
+    # 2. Run Type Resolver (Two-pass type inference & scope/edge resolution)
+    resolver = TypeResolver(G, extractions, workspace_dir)
+    resolver.propagate_types()
+    resolver.resolve_all_edges()
+
+    return G

{codegraph_gen-0.2.0 → codegraph_gen-1.1.0}/src/codegraph_gen/config.py

@@ -1,7 +1,7 @@
 import os
 from pathlib import Path
 from pydantic import BaseModel, Field
-from codegraph_gen.parser.base import ExtractionResult
+from codegraph_gen.schema import ExtractionResult
 
 # Default exclusions for files and directories we want to ignore
 DEFAULT_EXCLUSIONS = {

{codegraph_gen-0.2.0 → codegraph_gen-1.1.0}/src/codegraph_gen/detect.py

@@ -1,11 +1,15 @@
 import logging
 from pathlib import Path
-from codegraph_gen.config import CodegraphConfig, LANGUAGE_EXTENSIONS
+from codegraph_gen.config import LANGUAGE_EXTENSIONS
 
 logger = logging.getLogger(__name__)
 
 
-def discover_files(config: CodegraphConfig) -> list[tuple[Path, str]]:
+def discover_files(
+    workspace_dir: Path,
+    languages: set[str],
+    exclusions: set[str],
+) -> list[tuple[Path, str]]:
     """
     Recursively discovers source files in the workspace directory.
     Filters by allowed languages and ignores files/directories in exclusions.
@@ -14,17 +18,17 @@ def discover_files(config: CodegraphConfig) -> list[tuple[Path, str]]:
         List of tuples: (absolute_file_path, language_name)
     """
     found_files = []
-    workspace = config.workspace_dir.resolve()
+    workspace = workspace_dir.resolve()
 
     # Map extension -> language
     ext_to_lang = {}
-    for lang in config.languages:
+    for lang in languages:
         if lang in LANGUAGE_EXTENSIONS:
             for ext in LANGUAGE_EXTENSIONS[lang]:
                 ext_to_lang[ext] = lang
 
     # Normalize exclusions to lowercase for case-insensitive matching
-    exclusions_lower = {exc.lower() for exc in config.exclusions}
+    exclusions_lower = {exc.lower() for exc in exclusions}
 
     def is_ignored(path: Path) -> bool:
         # Check if any part of the path is in exclusions_lower

{codegraph_gen-0.2.0 → codegraph_gen-1.1.0}/src/codegraph_gen/engine.py

@@ -9,7 +9,7 @@ import networkx as nx
 from pydantic import BaseModel, ConfigDict
 
 from codegraph_gen.config import CodegraphConfig, CacheEntry
-from codegraph_gen.parser.base import ExtractionResult
+from codegraph_gen.schema import ExtractionResult
 from codegraph_gen.detect import discover_files
 from codegraph_gen.parser import get_parser
 from codegraph_gen.builder import build_graph
@@ -77,13 +77,12 @@ class PipelineResult(BaseModel):
 
 
 class CodegraphEngine:
-    def __init__(self, config: CodegraphConfig):
-        self.config = config
-        self.renderer = MarkdownRenderer(config.workspace_dir)
+    def __init__(self):
         self.writer = VaultWriter()
 
     def run_pipeline(
         self,
+        config: CodegraphConfig,
         progress_callback: Optional[
             Callable[[PipelineStage, Any, int, int], None]
         ] = None,
@@ -91,14 +90,18 @@ class CodegraphEngine:
         """
         Runs the full codegraph generation pipeline.
         Args:
+            config: Configuration settings.
             progress_callback: A function taking (stage, current_item, index, total)
         """
         logger.info("Starting codegraph engine pipeline...")
+        renderer = MarkdownRenderer(config.workspace_dir)
 
         # 1. Discover files
         if progress_callback:
             progress_callback(PipelineStage.DISCOVERING, None, 0, 0)
-        files = discover_files(self.config)
+        files = discover_files(
+            config.workspace_dir, config.languages, config.exclusions
+        )
         if not files:
             logger.warning("No supported files found.")
             if progress_callback:
@@ -116,9 +119,9 @@ class CodegraphEngine:
         extractions = []
         total_files = len(files)
 
-        cache_path = self.config.absolute_output_dir / "cache.json"
+        cache_path = config.absolute_output_dir / "cache.json"
         cache_entries = {}
-        if self.config.use_cache and cache_path.exists():
+        if config.use_cache and cache_path.exists():
             try:
                 with open(cache_path, "r", encoding="utf-8") as f:
                     cache_data = json.load(f)
@@ -132,7 +135,7 @@ class CodegraphEngine:
         new_cache_entries = {}
 
         for file_path, lang in files:
-            rel_path = str(file_path.relative_to(self.config.workspace_dir))
+            rel_path = str(file_path.relative_to(config.workspace_dir))
             try:
                 stat = file_path.stat()
                 mtime = stat.st_mtime
@@ -170,7 +173,7 @@ class CodegraphEngine:
             if progress_callback:
                 progress_callback(PipelineStage.PARSING, None, total_files, total_files)
         else:
-            max_workers = self.config.max_workers
+            max_workers = config.max_workers
             if max_workers > 1 and len(files_to_parse) > 1:
                 logger.info(
                     f"Parsing {len(files_to_parse)} files in parallel with {max_workers} workers..."
@@ -183,7 +186,7 @@ class CodegraphEngine:
                             _parse_file_worker,
                             file_path,
                             lang,
-                            self.config.workspace_dir,
+                            config.workspace_dir,
                         ): (file_path, rel_path, mtime, size, file_hash)
                         for file_path, lang, rel_path, mtime, size, file_hash in files_to_parse
                     }
@@ -235,7 +238,7 @@ class CodegraphEngine:
                         )
                     try:
                         parser = get_parser(lang)
-                        result = parser.parse_file(file_path, self.config.workspace_dir)
+                        result = parser.parse_file(file_path, config.workspace_dir)
                         extractions.append(result)
                         if file_hash:
                             new_cache_entries[rel_path] = CacheEntry(
@@ -247,7 +250,7 @@ class CodegraphEngine:
         # 3. Build graph
         if progress_callback:
             progress_callback(PipelineStage.BUILDING, None, 0, 0)
-        G = build_graph(extractions, self.config.workspace_dir)
+        G = build_graph(extractions, config.workspace_dir)
 
         # 4. Component clustering
         if progress_callback:
@@ -271,7 +274,7 @@ class CodegraphEngine:
         rendered_nodes = {}
         for nid, ndata in G.nodes(data=True):
             fname = get_node_filename(nid)
-            content = self.renderer.render_node_page(nid, ndata, G, node_component_map)
+            content = renderer.render_node_page(nid, ndata, G, node_component_map)
             rendered_nodes[fname] = content
 
         rendered_components = {}
@@ -279,7 +282,7 @@ class CodegraphEngine:
             comp_name = component_names[cid]
             cohesion = cohesion_scores[cid]
             fname = get_component_filename(comp_name)
-            content = self.renderer.render_component_page(
+            content = renderer.render_component_page(
                 cid,
                 members,
                 G,
@@ -292,7 +295,7 @@ class CodegraphEngine:
 
         # Check if README already has AI Insights and preserve it
         ai_insights = None
-        readme_path = self.config.absolute_output_dir / "README.md"
+        readme_path = config.absolute_output_dir / "README.md"
         if readme_path.exists():
             try:
                 old_readme = readme_path.read_text(encoding="utf-8")
@@ -315,7 +318,7 @@ class CodegraphEngine:
                     f"Could not read existing README.md to preserve AI insights: {e}"
                 )
 
-        readme_content = self.renderer.render_readme(
+        readme_content = renderer.render_readme(
             G,
             components,
             cohesion_scores,
@@ -324,7 +327,7 @@ class CodegraphEngine:
             ai_insights=ai_insights,
         )
 
-        prompt_content = self.renderer.render_agent_prompt(
+        prompt_content = renderer.render_agent_prompt(
             G, components, cohesion_scores, component_names, analysis
         )
 
@@ -332,7 +335,7 @@ class CodegraphEngine:
         if progress_callback:
             progress_callback(PipelineStage.WRITING, None, 0, 0)
         self.writer.write_vault(
-            self.config.absolute_output_dir,
+            config.absolute_output_dir,
             rendered_nodes,
             rendered_components,
             readme_content,
@@ -340,9 +343,9 @@ class CodegraphEngine:
         )
 
         # Write updated cache back to disk
-        if self.config.use_cache:
+        if config.use_cache:
             try:
-                self.config.absolute_output_dir.mkdir(parents=True, exist_ok=True)
+                config.absolute_output_dir.mkdir(parents=True, exist_ok=True)
                 with open(cache_path, "w", encoding="utf-8") as f:
                     json.dump(
                         {k: v.model_dump() for k, v in new_cache_entries.items()},

codegraph_gen-1.1.0/src/codegraph_gen/parser/__init__.py

@@ -0,0 +1,31 @@
+import importlib
+import logging
+import pkgutil
+import sys
+from pathlib import Path
+from codegraph_gen.parser.base import BaseParser, _PARSER_REGISTRY
+
+logger = logging.getLogger(__name__)
+
+# Dynamic package scan & load to trigger @register_parser registrations
+package_dir = str(Path(__file__).parent)
+for _, module_name, _ in pkgutil.iter_modules([package_dir]):
+    if module_name == "base":
+        continue
+    full_module_name = f"{__name__}.{module_name}"
+    if full_module_name not in sys.modules:
+        try:
+            importlib.import_module(full_module_name)
+        except Exception as e:
+            logger.error(
+                f"Defensive Loading: Failed to import parser module {full_module_name}: {e}",
+                exc_info=True,
+            )
+
+
+def get_parser(language: str) -> BaseParser:
+    """Returns an instance of the parser for the given language."""
+    lang_lower = language.lower()
+    if lang_lower not in _PARSER_REGISTRY:
+        raise ValueError(f"Unsupported language: {language}")
+    return _PARSER_REGISTRY[lang_lower]()