codeboarding 0.13.10__tar.gz → 0.13.11__tar.gz

{codeboarding-0.13.10/codeboarding.egg-info → codeboarding-0.13.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codeboarding
-Version: 0.13.10
+Version: 0.13.11
 Summary: Interactive Diagrams for Code
 Author: CodeBoarding Team
 License-Expression: MIT

{codeboarding-0.13.10 → codeboarding-0.13.11}/agents/llm_config.py

@@ -91,6 +91,8 @@ class LLMConfig:
         selection_envs: Env vars that select this provider — any one being set selects it.
         api_key_env: Env var holding the provider's secret, or None when the
                      underlying SDK reads its credentials from the environment itself.
+        base_url_env: Env var that overrides the provider endpoint, or an empty string when unsupported.
+        default_base_url: Built-in endpoint used when base_url_env is unset, or an empty string for the SDK default.
         agent_model: The "agent" model used for complex reasoning and agentic tasks.
         parsing_model: The "parsing" model used for fast, cost-effective extraction and parsing tasks.
         agent_temperature: Temperature for the agent model. Defaults to 0 for deterministic behavior
@@ -109,6 +111,8 @@ class LLMConfig:
     parsing_temperature: float = LLMDefaults.DEFAULT_PARSING_TEMPERATURE
     extra_args: dict[str, Any] = field(default_factory=dict)
     api_key_env: str | None = None
+    base_url_env: str = ""
+    default_base_url: str = ""
     keyless_capable: bool = False
     """Whether this provider can run without a real API key.
 
@@ -139,6 +143,10 @@ class LLMConfig:
             value = v() if callable(v) else v
             if value is not None:
                 resolved[k] = value
+        if self.base_url_env:
+            base_url = os.getenv(self.base_url_env, self.default_base_url or None)
+            if base_url is not None:
+                resolved["base_url"] = base_url
         return resolved
 
 
@@ -152,8 +160,8 @@ LLM_PROVIDERS = {
         parsing_model="gpt-4o-mini",
         llm_type=LLMType.GPT4,
         keyless_capable=True,
+        base_url_env="OPENAI_BASE_URL",
         extra_args={
-            "base_url": lambda: os.getenv("OPENAI_BASE_URL"),
             "max_tokens": None,
             "timeout": None,
             "max_retries": 0,
@@ -166,8 +174,9 @@ LLM_PROVIDERS = {
         agent_model="google/gemini-3.7-flash",
         parsing_model="openai/gpt-5-mini",
         llm_type=LLMType.GEMINI_FLASH,
+        base_url_env="VERCEL_BASE_URL",
+        default_base_url="https://ai-gateway.vercel.sh/v1",
         extra_args={
-            "base_url": lambda: os.getenv("VERCEL_BASE_URL", f"https://ai-gateway.vercel.sh/v1"),
             "max_tokens": None,
             "timeout": None,
             "max_retries": 0,
@@ -180,8 +189,8 @@ LLM_PROVIDERS = {
         agent_model="claude-sonnet-5",
         parsing_model="claude-haiku-4-5",
         llm_type=LLMType.CLAUDE,
+        base_url_env="ANTHROPIC_BASE_URL",
         extra_args={
-            "base_url": lambda: os.getenv("ANTHROPIC_BASE_URL"),
             "thinking": {"type": "disabled"},
             "max_tokens": 8192,
             "timeout": None,
@@ -239,9 +248,7 @@ LLM_PROVIDERS = {
         llm_type=LLMType.GEMINI_FLASH,
         agent_temperature=LLMDefaults.DEFAULT_AGENT_TEMPERATURE,
         parsing_temperature=LLMDefaults.DEFAULT_PARSING_TEMPERATURE,
-        extra_args={
-            "base_url": lambda: os.getenv("OLLAMA_BASE_URL"),
-        },
+        base_url_env="OLLAMA_BASE_URL",
     ),
     "deepseek": LLMConfig(
         chat_class=ChatOpenAI,
@@ -250,8 +257,9 @@ LLM_PROVIDERS = {
         agent_model="deepseek-v4-flash",
         parsing_model="deepseek-v4-flash",
         llm_type=LLMType.DEEPSEEK,
+        base_url_env="DEEPSEEK_BASE_URL",
+        default_base_url="https://api.deepseek.com/v1",
         extra_args={
-            "base_url": lambda: os.getenv("DEEPSEEK_BASE_URL", "https://api.deepseek.com/v1"),
             "max_tokens": None,
             "timeout": None,
             "max_retries": 0,
@@ -264,8 +272,9 @@ LLM_PROVIDERS = {
         agent_model="glm-4.7-flash",
         parsing_model="glm-4.7-flash",
         llm_type=LLMType.GLM,
+        base_url_env="GLM_BASE_URL",
+        default_base_url="https://open.bigmodel.cn/api/paas/v4",
         extra_args={
-            "base_url": lambda: os.getenv("GLM_BASE_URL", "https://open.bigmodel.cn/api/paas/v4"),
             "max_tokens": None,
             "timeout": None,
             "max_retries": 0,
@@ -278,8 +287,9 @@ LLM_PROVIDERS = {
         agent_model="kimi-k2.6",
         parsing_model="kimi-k2.6",
         llm_type=LLMType.KIMI,
+        base_url_env="KIMI_BASE_URL",
+        default_base_url="https://api.moonshot.cn/v1",
         extra_args={
-            "base_url": lambda: os.getenv("KIMI_BASE_URL", "https://api.moonshot.cn/v1"),
             "max_tokens": None,
             "timeout": None,
             "max_retries": 0,
@@ -292,8 +302,9 @@ LLM_PROVIDERS = {
         agent_model="google/gemini-3.7-flash",
         parsing_model="google/gemini-3.7-flash",
         llm_type=LLMType.GEMINI_FLASH,
+        base_url_env="OPENROUTER_BASE_URL",
+        default_base_url="https://openrouter.ai/api/v1",
         extra_args={
-            "base_url": lambda: os.getenv("OPENROUTER_BASE_URL", "https://openrouter.ai/api/v1"),
             "max_tokens": None,
             "timeout": None,
             "max_retries": 0,
@@ -306,8 +317,9 @@ LLM_PROVIDERS = {
         agent_model="openai/gpt-5.4-mini",
         parsing_model="openai/gpt-5.4-mini",
         llm_type=LLMType.GPT4,
+        base_url_env="ORCAROUTER_BASE_URL",
+        default_base_url="https://api.orcarouter.ai/v1",
         extra_args={
-            "base_url": lambda: os.getenv("ORCAROUTER_BASE_URL", "https://api.orcarouter.ai/v1"),
             "max_tokens": None,
             "timeout": None,
             "max_retries": 0,
@@ -323,8 +335,8 @@ LLM_PROVIDERS = {
         parsing_model="gpt-4o-mini",
         llm_type=LLMType.GPT4,
         keyless_capable=True,
+        base_url_env="LITELLM_BASE_URL",
         extra_args={
-            "base_url": lambda: os.getenv("LITELLM_BASE_URL"),
             "max_tokens": None,
             "timeout": None,
             "max_retries": 0,
@@ -333,6 +345,15 @@ LLM_PROVIDERS = {
 }
 
 
+LLM_PROVIDER_ENV_VARS: frozenset[str] = frozenset(
+    var
+    for config in LLM_PROVIDERS.values()
+    for var in [*config.selection_envs, config.api_key_env, config.base_url_env]
+    if var
+)
+LLM_ENV_VARS: frozenset[str] = LLM_PROVIDER_ENV_VARS | frozenset({"AGENT_MODEL", "PARSING_MODEL"})
+
+
 def _all_selection_envs() -> list[str]:
     return sorted({var for config in LLM_PROVIDERS.values() for var in config.selection_envs})
 

{codeboarding-0.13.10 → codeboarding-0.13.11/codeboarding.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codeboarding
-Version: 0.13.10
+Version: 0.13.11
 Summary: Interactive Diagrams for Code
 Author: CodeBoarding Team
 License-Expression: MIT

{codeboarding-0.13.10 → codeboarding-0.13.11}/codeboarding.egg-info/SOURCES.txt

@@ -4,7 +4,6 @@ README.md
 clustering_ids.py
 constants.py
 github_action.py
-health_main.py
 install.py
 logging_config.py
 main.py

{codeboarding-0.13.10 → codeboarding-0.13.11}/diagram_analysis/file_index.py

@@ -27,7 +27,7 @@ def build_file_methods_from_nodes(
     source_cache: SourceCache | None = None,
 ) -> list[FileMethodGroup]:
     """Build sorted file/method groups from assigned CFG nodes."""
-    by_file: dict[str, dict[tuple[int, int, str, str], MethodEntry]] = defaultdict(dict)
+    by_file: dict[str, dict[tuple[int, int, int, str, str], MethodEntry]] = defaultdict(dict)
     file_cache = source_cache if source_cache is not None else {}
 
     for node in nodes:
@@ -35,7 +35,7 @@ def build_file_methods_from_nodes(
             continue
         file_path = normalize_repo_path(node.file_path, repo_dir)
         method_name = node.fully_qualified_name.split(".")[-1]
-        dedupe_key = (node.line_start, node.line_end, node.type.name, method_name)
+        dedupe_key = (node.line_start, node.line_end, node.col_start, node.type.name, method_name)
         candidate = MethodEntry(
             qualified_name=node.fully_qualified_name,
             start_line=node.line_start,

{codeboarding-0.13.10 → codeboarding-0.13.11}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "codeboarding"
-version = "0.13.10"
+version = "0.13.11"
 description = "Interactive Diagrams for Code"
 readme = "PYPI.md"
 license = "MIT"

codeboarding-0.13.10/health_main.py

@@ -1,152 +0,0 @@
-"""Standalone entry point for running health checks on a repository.
-
-Runs static analysis and health checks only — no LLM agents or diagram generation.
-Useful for testing health checks in isolation and for CI/CD health gates.
-
-Usage:
-    # Local repository (output written to <repo>/.codeboarding/health/)
-    python health_main.py --local /path/to/repo
-
-    # Remote repository (cloned to cwd/repos/, output to cwd/<repo_name>/.codeboarding/health/)
-    python health_main.py https://github.com/user/repo
-"""
-
-import argparse
-import logging
-from pathlib import Path
-
-from health.runner import run_health_checks
-from health.config import initialize_health_dir, load_health_config
-from logging_config import setup_logging
-from repo_utils import clone_repository, get_repo_name
-from static_analyzer import get_static_analysis
-from utils import CODEBOARDING_DIR_NAME, get_artifact_dir
-from vscode_constants import update_config
-
-logger = logging.getLogger(__name__)
-
-
-def run_health_check_local(
-    repo_path: Path,
-    project_name: str | None = None,
-) -> None:
-    """Run health checks on a local repository.
-
-    Args:
-        repo_path: Path to a local repository
-        project_name: Optional project name (default: repo directory name)
-    """
-    resolved_repo_path = repo_path.resolve()
-    if not resolved_repo_path.is_dir():
-        raise ValueError(f"Repository path does not exist: {resolved_repo_path}")
-
-    resolved_project_name = project_name or resolved_repo_path.name
-    output_dir = resolved_repo_path / CODEBOARDING_DIR_NAME
-
-    output_dir.mkdir(parents=True, exist_ok=True)
-    setup_logging(log_dir=output_dir)
-
-    _run_health_checks(resolved_repo_path, resolved_project_name, output_dir)
-
-
-def run_health_check_remote(
-    repo_url: str,
-    project_name: str | None = None,
-) -> None:
-    """Run health checks on a remote repository.
-
-    Args:
-        repo_url: URL of a remote Git repository
-        project_name: Optional project name (default: extracted from URL)
-    """
-    repo_root = Path("repos")
-    repo_name = clone_repository(repo_url, repo_root)
-    resolved_repo_path = repo_root / repo_name
-    resolved_project_name = project_name or get_repo_name(repo_url)
-    output_dir = Path.cwd() / repo_name / CODEBOARDING_DIR_NAME
-
-    output_dir.mkdir(parents=True, exist_ok=True)
-    setup_logging(log_dir=output_dir)
-
-    _run_health_checks(resolved_repo_path, resolved_project_name, output_dir)
-
-
-def _run_health_checks(repo_path: Path, project_name: str, output_dir: Path) -> None:
-    """Core health check logic shared by local and remote paths."""
-    logger.info(f"Running health checks on '{project_name}' at {repo_path}")
-
-    static_analysis = get_static_analysis(repo_path, cache_dir=get_artifact_dir(repo_path))
-
-    # Load health check configuration and initialize health config dir
-    health_config_dir = output_dir / "health"
-    initialize_health_dir(health_config_dir)
-    health_config = load_health_config(health_config_dir)
-
-    report = run_health_checks(static_analysis, project_name, config=health_config, repo_path=repo_path)
-
-    if report is None:
-        logger.warning("Health checks skipped: no languages found in static analysis results")
-        return
-
-    report_path = health_config_dir / "health_report.json"
-    report_path.write_text(report.model_dump_json(indent=2, exclude_none=True))
-
-    logger.info(f"Health report written to {report_path} (overall score: {report.overall_score:.3f})")
-
-
-def main():
-    """Main entry point that parses CLI arguments and routes to subcommands."""
-    parser = argparse.ArgumentParser(
-        description="Run static analysis health checks on a repository (local or remote)",
-        formatter_class=argparse.RawDescriptionHelpFormatter,
-        epilog="""
-Examples:
-  # Local repository (output written to <repo>/.codeboarding/health/)
-  python health_main.py --local /path/to/repo
-
-  # Remote repository (cloned to cwd/repos/, output to cwd/<repo_name>/.codeboarding/health/)
-  python health_main.py https://github.com/user/repo
-
-        """,
-    )
-    parser.add_argument(
-        "repositories",
-        nargs="*",
-        help="One or more remote Git repository URLs to run health checks on",
-    )
-    parser.add_argument("--local", type=Path, help="Path to a local repository")
-    parser.add_argument(
-        "--project-name",
-        type=str,
-        default=None,
-        help="Name of the project (default: extracted from repo path or URL)",
-    )
-    parser.add_argument(
-        "--binary-location", type=Path, help="Path to the binary directory for language servers and tools"
-    )
-
-    args = parser.parse_args()
-
-    # Validate: must provide either remote repos or --local, not both
-    has_remote = bool(args.repositories)
-    has_local = args.local is not None
-
-    if has_remote == has_local:
-        parser.error("Provide either one or more remote repositories or --local, but not both.")
-
-    # Resolve binary paths for tools (tokei, LSP servers, etc.)
-    if args.binary_location:
-        update_config(args.binary_location)
-
-    try:
-        if has_local:
-            run_health_check_local(args.local, args.project_name)
-        else:
-            for repo_url in args.repositories:
-                run_health_check_remote(repo_url, args.project_name)
-    except ValueError as e:
-        parser.error(str(e))
-
-
-if __name__ == "__main__":
-    main()