alita-sdk 0.3.379__py3-none-any.whl → 0.3.627__py3-none-any.whl

alita_sdk/cli/inventory.py

@@ -0,0 +1,1073 @@
+"""
+CLI commands for Inventory Ingestion Pipeline.
+
+Provides command-line interface for running knowledge graph ingestion
+from various source toolkits (GitHub, ADO, LocalGit, etc.).
+
+Usage:
+    # List available presets
+    alita inventory presets
+    
+    # Ingest using a preset (recommended!)
+    alita inventory ingest --dir ./my-project --graph ./graph.json --preset python
+    
+    # Ingest using a toolkit config file
+    alita inventory ingest --toolkit .alita/tools/github.json --graph ./graph.json -w "*.md"
+    
+    # Ingest from a local git repository
+    alita inventory ingest --source localgit --path /path/to/repo --graph ./graph.json
+    
+    # Use a config file for LLM/embedding/guardrails settings
+    alita inventory ingest --toolkit ./github.json -g ./graph.json --config ingestion-config.yml
+    
+    # Check ingestion status (failed files, progress)
+    alita inventory status --graph ./graph.json --name my-source
+    
+    # Retry failed files from previous ingestion
+    alita inventory retry --dir ./my-project -g ./graph.json --name my-source
+    alita inventory retry --dir ./my-project -g ./graph.json --name my-source --force
+    
+    # Generate config template
+    alita inventory init-config
+    
+    # Show graph stats
+    alita inventory stats --graph ./graph.json
+    
+    # Search the graph
+    alita inventory search "PaymentService" --graph ./graph.json
+"""
+
+import click
+import json
+import logging
+import os
+import re
+import sys
+from pathlib import Path
+from typing import Optional, List, Dict, Any
+
+logger = logging.getLogger(__name__)
+
+
+@click.group()
+def inventory():
+    """Inventory knowledge graph commands."""
+    pass
+
+
+@inventory.command('presets')
+def presets():
+    """
+    List available ingestion presets.
+    
+    Presets provide pre-configured whitelist/blacklist patterns for common
+    programming languages and project types.
+    
+    Example:
+        alita inventory presets
+    """
+    from alita_sdk.community.inventory import list_presets, get_preset
+    
+    available = list_presets()
+    
+    click.echo(f"\n📋 Available Presets ({len(available)} total):\n")
+    
+    # Group by category
+    categories = {
+        'Python': [p for p in available if 'python' in p.lower()],
+        'JavaScript/TypeScript': [p for p in available if any(x in p.lower() for x in ['javascript', 'typescript', 'react', 'next', 'node'])],
+        'Java': [p for p in available if 'java' in p.lower() or 'maven' in p.lower() or 'gradle' in p.lower() or 'spring' in p.lower()],
+        '.NET/C#': [p for p in available if 'dotnet' in p.lower() or 'csharp' in p.lower() or 'aspnet' in p.lower()],
+        'Multi-Language': [p for p in available if any(x in p.lower() for x in ['fullstack', 'monorepo', 'documentation'])],
+    }
+    
+    for category, preset_names in categories.items():
+        if not preset_names:
+            continue
+        
+        click.echo(f"  {category}:")
+        for preset_name in sorted(preset_names):
+            preset_config = get_preset(preset_name)
+            whitelist = preset_config.get('whitelist', [])
+            blacklist = preset_config.get('blacklist', [])
+            
+            # Format whitelist (show first 3 patterns)
+            wl_display = ', '.join(whitelist[:3])
+            if len(whitelist) > 3:
+                wl_display += f', ... (+{len(whitelist)-3})'
+            
+            click.echo(f"    • {preset_name:20} - {wl_display}")
+        
+        click.echo()
+    
+    click.echo("Usage:")
+    click.echo("  alita inventory ingest --dir ./my-project -g ./graph.json --preset python")
+    click.echo("  alita inventory ingest --dir ./src -g ./graph.json -p typescript -w '*.json'")
+    click.echo()
+
+
+@inventory.command('init-config')
+@click.option('--output', '-o', default='./ingestion-config.yml', type=click.Path(),
+              help='Output path for config template')
+def init_config(output: str):
+    """
+    Generate a configuration template file.
+    
+    Example:
+        alita inventory init-config -o ./my-config.yml
+    """
+    from alita_sdk.community.inventory import generate_config_template
+    
+    path = generate_config_template(output)
+    click.echo(f"✅ Configuration template created: {path}")
+    click.echo("\nEdit this file to configure:")
+    click.echo("  - LLM provider and model (openai, azure, anthropic, ollama)")
+    click.echo("  - Embeddings for semantic search")
+    click.echo("  - Guardrails (rate limits, content filtering, thresholds)")
+
+
+@inventory.command('ingest')
+@click.option('--toolkit', '-t', type=click.Path(exists=True),
+              help='Path to toolkit config JSON (e.g., .alita/tools/github.json)')
+@click.option('--dir', '-d', 'directory', type=click.Path(exists=True, file_okay=False, dir_okay=True),
+              help='Local directory to ingest (alternative to --toolkit for local files)')
+@click.option('--graph', '-g', required=True, type=click.Path(),
+              help='Path to output graph JSON file')
+@click.option('--config', '-c', type=click.Path(exists=True),
+              help='Path to YAML/JSON config file for LLM, embeddings, guardrails')
+@click.option('--preset', '-p', default=None,
+              help='Use a preset configuration (e.g., python, typescript, java, dotnet)')
+@click.option('--whitelist', '-w', multiple=True,
+              help='File patterns to include (e.g., -w "*.py" -w "*.md")')
+@click.option('--blacklist', '-x', multiple=True,
+              help='File patterns to exclude (e.g., -x "*test*" -x "*vendor*")')
+@click.option('--no-relations', is_flag=True,
+              help='Skip relation extraction (faster)')
+@click.option('--model', '-m', default=None,
+              help='LLM model name (overrides config file)')
+@click.option('--limit', '-l', type=int, default=None,
+              help='Limit number of documents to process (for testing)')
+@click.option('--fresh', '-f', is_flag=True,
+              help='Start fresh - delete existing graph and create new one')
+@click.option('--name', '-n', default=None,
+              help='Source name for the graph (default: directory name or toolkit_name)')
+@click.option('--recursive/--no-recursive', default=True,
+              help='Recursively scan subdirectories (default: recursive)')
+@click.pass_context
+def ingest(ctx, toolkit: Optional[str], directory: Optional[str], graph: str, 
+           config: Optional[str], preset: Optional[str], whitelist: tuple, blacklist: tuple, 
+           no_relations: bool, model: Optional[str], limit: Optional[int], 
+           fresh: bool, name: Optional[str], recursive: bool):
+    """Run ingestion pipeline to build/update a knowledge graph.
+    
+    Use --toolkit for configured sources (GitHub, ADO, etc.) or --dir for
+    local directories (simpler, no config needed).
+    
+    \b
+    Examples:
+      alita inventory ingest --dir ./src -g graph.json --preset python
+      alita inventory ingest --dir ./src -g graph.json -w "*.py" -w "*.md"
+      alita inventory ingest --dir ./src -g graph.json -p typescript -w "*.json"
+      alita inventory ingest --dir ./docs -g graph.json --name my-docs
+      alita inventory ingest -t github.json -g graph.json -w "*.md"
+      alita inventory ingest --dir ./src -g graph.json -c config.yml
+    """
+    # Load preset configuration if specified
+    preset_whitelist = []
+    preset_blacklist = []
+    
+    if preset:
+        from alita_sdk.community.inventory import get_preset, list_presets
+        
+        try:
+            preset_config = get_preset(preset)
+            preset_whitelist = preset_config.get('whitelist', [])
+            preset_blacklist = preset_config.get('blacklist', [])
+            
+            click.echo(f"📋 Using preset: {preset}")
+            if preset_whitelist:
+                click.echo(f"   Whitelist: {', '.join(preset_whitelist)}")
+            if preset_blacklist:
+                click.echo(f"   Blacklist: {', '.join(preset_blacklist)}")
+        except ValueError as e:
+            available = ', '.join(list_presets())
+            raise click.ClickException(f"Unknown preset '{preset}'. Available: {available}")
+    
+    # Merge preset patterns with user-provided patterns
+    # User patterns are added after preset patterns (more specific)
+    final_whitelist = list(preset_whitelist) + list(whitelist)
+    final_blacklist = list(preset_blacklist) + list(blacklist)
+    
+    # Validate: must have either --toolkit or --dir
+    if not toolkit and not directory:
+        raise click.ClickException("Must specify either --toolkit or --dir")
+    
+    if toolkit and directory:
+        raise click.ClickException("Cannot use both --toolkit and --dir. Choose one.")
+    
+    # Handle --dir mode (simple local directory ingestion)
+    if directory:
+        from pathlib import Path
+        dir_path = Path(directory).resolve()
+        source_name = name or dir_path.name
+        source_type = 'filesystem'
+        
+        click.echo(f"📂 Ingesting local directory: {dir_path}")
+        click.echo(f"   Name: {source_name}")
+        click.echo(f"   Recursive: {recursive}")
+        
+        # Create a simple toolkit config for the directory
+        toolkit_config = {
+            'type': 'filesystem',
+            'toolkit_name': source_name,
+            'base_directory': str(dir_path),
+            'recursive': recursive,
+        }
+        branch = None  # No branch for filesystem
+    else:
+        # Load toolkit config
+        toolkit_config = _load_toolkit_config(toolkit)
+        click.echo(f"📦 Loaded toolkit config: {toolkit}")
+        
+        # Get source type from toolkit
+        source_type = toolkit_config.get('type')
+        if not source_type:
+            raise click.ClickException(f"Toolkit config missing 'type' field: {toolkit}")
+        click.echo(f"   Type: {source_type}")
+        
+        # Get toolkit name (used as source identifier in the graph)
+        source_name = name or toolkit_config.get('toolkit_name') or source_type
+        click.echo(f"   Name: {source_name}")
+        
+        # Get repo/branch from toolkit config
+        repo = toolkit_config.get('repository')
+        if repo:
+            click.echo(f"   Repository: {repo}")
+        
+        branch = toolkit_config.get('active_branch') or toolkit_config.get('base_branch') or 'main'
+        click.echo(f"   Branch: {branch}")
+        
+        # Get path for local sources (filesystem or localgit)
+        path = (
+            toolkit_config.get('base_directory') or  # filesystem toolkit
+            toolkit_config.get('git_root_dir') or    # localgit toolkit
+            toolkit_config.get('path')               # generic path
+        )
+        if path:
+            click.echo(f"   Path: {path}")
+        
+        # Validate required fields based on source type
+        if source_type in ('github', 'ado') and not repo:
+            raise click.ClickException(f"Toolkit config missing 'repository' for source '{source_type}'")
+        
+        if source_type == 'filesystem' and not path:
+            raise click.ClickException(f"Toolkit config missing 'base_directory' or 'path' for source '{source_type}'")
+        
+        if source_type == 'localgit' and not path:
+            raise click.ClickException(f"Toolkit config missing 'git_root_dir' or 'path' for source '{source_type}'")
+    
+    # Handle --fresh option: delete existing graph
+    if fresh and os.path.exists(graph):
+        click.echo(f"🗑️  Fresh mode: deleting existing graph at {graph}")
+        os.remove(graph)
+    
+    click.echo(f"🚀 Starting ingestion from {source_name} ({source_type})...")
+    
+    # Progress callback
+    def progress(message: str, phase: str):
+        click.echo(f"  [{phase}] {message}")
+    
+    try:
+        from alita_sdk.community.inventory import IngestionPipeline, IngestionConfig
+        
+        # Load configuration
+        if config:
+            click.echo(f"📋 Loading config from {config}")
+            if config.endswith('.yml') or config.endswith('.yaml'):
+                ingestion_config = IngestionConfig.from_yaml(config)
+            else:
+                ingestion_config = IngestionConfig.from_json(config)
+            
+            # Override model if specified on command line
+            if model:
+                ingestion_config.llm_model = model
+            
+            # Override graph path
+            ingestion_config.graph_path = graph
+            
+            # Get LLM using the model name and temperature from config
+            llm = _get_llm(ctx, ingestion_config.llm_model, ingestion_config.temperature)
+            
+            pipeline = IngestionPipeline(
+                llm=llm,
+                graph_path=ingestion_config.graph_path,
+                guardrails=ingestion_config.guardrails,
+            )
+        else:
+            # Fall back to environment-based config
+            click.echo("📋 Loading config from environment")
+            llm = _get_llm(ctx, model)
+            pipeline = IngestionPipeline(
+                llm=llm,
+                graph_path=graph,
+            )
+        
+        # Set progress callback
+        pipeline.progress_callback = progress
+        
+        # Show existing graph status
+        graph_stats = pipeline.get_stats()
+        if graph_stats['node_count'] > 0:
+            click.echo(f"📊 Existing graph: {graph_stats['node_count']} entities, {graph_stats['edge_count']} relations")
+            click.echo("   New entities will be ADDED to existing graph")
+        
+        # Get source toolkit from config and register it
+        source_toolkit = _get_source_toolkit(toolkit_config)
+        
+        # Create a RunnableConfig for CLI context - this allows dispatch_custom_event to work
+        # without being inside a LangChain agent run
+        import uuid
+        cli_runnable_config = {
+            'run_id': uuid.uuid4(),
+            'tags': ['cli', 'inventory', 'ingest'],
+        }
+        
+        # Set the runnable config on the toolkit if it supports it
+        if hasattr(source_toolkit, 'set_runnable_config'):
+            source_toolkit.set_runnable_config(cli_runnable_config)
+        
+        pipeline.register_toolkit(source_name, source_toolkit)
+        
+        # Run ingestion
+        if limit:
+            click.echo(f"⚠️  Limiting to {limit} documents (test mode)")
+        
+        result = pipeline.run(
+            source=source_name,
+            branch=branch,
+            whitelist=final_whitelist if final_whitelist else None,
+            blacklist=final_blacklist if final_blacklist else None,
+            extract_relations=not no_relations,
+            max_documents=limit,
+        )
+        
+        # Show result
+        if result.success:
+            click.echo(f"\n✅ Ingestion complete!")
+            click.echo(f"   Documents processed: {result.documents_processed}")
+            click.echo(f"   Entities extracted: {result.entities_added}")
+            click.echo(f"   Relations extracted: {result.relations_added}")
+            click.echo(f"   Duration: {result.duration_seconds:.1f}s")
+            click.echo(f"   Graph saved to: {graph}")
+            
+            # Show failed documents info if any
+            if result.failed_documents:
+                click.echo(f"\n⚠️  {len(result.failed_documents)} documents failed to process")
+                click.echo(f"   Run 'alita inventory status -g {graph} -n {source_name}' to see details")
+                click.echo(f"   Run 'alita inventory retry ...' to retry failed files")
+        else:
+            click.echo(f"\n❌ Ingestion failed!")
+            for error in result.errors:
+                click.echo(f"   Error: {error}")
+            sys.exit(1)
+            
+    except Exception as e:
+        logger.exception("Ingestion failed")
+        raise click.ClickException(str(e))
+
+
+@inventory.command('retry')
+@click.option('--toolkit', '-t', type=click.Path(exists=True),
+              help='Path to toolkit config JSON (e.g., .alita/tools/github.json)')
+@click.option('--dir', '-d', 'directory', type=click.Path(exists=True, file_okay=False, dir_okay=True),
+              help='Local directory to ingest (alternative to --toolkit for local files)')
+@click.option('--graph', '-g', required=True, type=click.Path(exists=True),
+              help='Path to graph JSON file')
+@click.option('--config', '-c', type=click.Path(exists=True),
+              help='Path to YAML/JSON config file for LLM, embeddings, guardrails')
+@click.option('--no-relations', is_flag=True,
+              help='Skip relation extraction (faster)')
+@click.option('--model', '-m', default=None,
+              help='LLM model name (overrides config file)')
+@click.option('--name', '-n', required=True,
+              help='Source name (must match the name used during original ingestion)')
+@click.option('--force', '-f', is_flag=True,
+              help='Retry all failed files regardless of attempt count')
+@click.option('--recursive/--no-recursive', default=True,
+              help='Recursively scan subdirectories (default: recursive)')
+@click.pass_context
+def retry(ctx, toolkit: Optional[str], directory: Optional[str], graph: str,
+          config: Optional[str], no_relations: bool, model: Optional[str],
+          name: str, force: bool, recursive: bool):
+    """Retry ingestion for files that failed in a previous run.
+    
+    Reads the checkpoint file to find failed files and re-ingests them.
+    Use --force to retry all failed files regardless of previous attempt count.
+    
+    \b
+    Examples:
+      alita inventory retry --dir ./src -g graph.json -n my-source
+      alita inventory retry --dir ./src -g graph.json -n my-source --force
+      alita inventory retry -t github.json -g graph.json -n github-repo
+    """
+    # Validate: must have either --toolkit or --dir
+    if not toolkit and not directory:
+        raise click.ClickException("Must specify either --toolkit or --dir")
+    
+    if toolkit and directory:
+        raise click.ClickException("Cannot use both --toolkit and --dir. Choose one.")
+    
+    # Check if checkpoint exists
+    checkpoint_path = _get_checkpoint_path(graph, name)
+    if not os.path.exists(checkpoint_path):
+        click.echo(f"\n❌ No checkpoint found for source '{name}'")
+        click.echo(f"   Expected checkpoint: {checkpoint_path}")
+        click.echo(f"\n   This could mean:")
+        click.echo(f"   - No previous ingestion was run with --name '{name}'")
+        click.echo(f"   - The previous ingestion completed successfully (checkpoint cleared)")
+        click.echo(f"   - The checkpoint was manually deleted")
+        sys.exit(1)
+    
+    # Load checkpoint to get failed files
+    try:
+        with open(checkpoint_path, 'r') as f:
+            checkpoint_data = json.load(f)
+    except Exception as e:
+        raise click.ClickException(f"Failed to load checkpoint: {e}")
+    
+    failed_files = checkpoint_data.get('failed_files', [])
+    
+    if not failed_files:
+        click.echo(f"\n✅ No failed files to retry for source '{name}'")
+        click.echo(f"   Processed files: {len(checkpoint_data.get('processed_files', []))}")
+        # Clear checkpoint since there's nothing to retry
+        os.remove(checkpoint_path)
+        click.echo(f"   Checkpoint cleared.")
+        return
+    
+    # Get files to retry
+    if force:
+        # Retry all failed files
+        files_to_retry = [f['file_path'] for f in failed_files]
+        click.echo(f"\n🔄 Force retrying ALL {len(files_to_retry)} failed files...")
+    else:
+        # Only retry files under max attempts (default: 3)
+        max_attempts = 3
+        files_to_retry = [
+            f['file_path'] for f in failed_files
+            if f.get('attempts', 1) < max_attempts
+        ]
+        skipped = len(failed_files) - len(files_to_retry)
+        if skipped > 0:
+            click.echo(f"\n⚠️  Skipping {skipped} files that exceeded {max_attempts} attempts")
+            click.echo(f"   Use --force to retry all failed files")
+        
+        if not files_to_retry:
+            click.echo(f"\n❌ No files eligible for retry (all exceeded max attempts)")
+            click.echo(f"   Use --force to retry anyway")
+            sys.exit(1)
+        
+        click.echo(f"\n🔄 Retrying {len(files_to_retry)} failed files...")
+    
+    # Handle --dir mode (simple local directory ingestion)
+    if directory:
+        from pathlib import Path
+        dir_path = Path(directory).resolve()
+        source_type = 'filesystem'
+        
+        click.echo(f"📂 Source directory: {dir_path}")
+        
+        # Create a simple toolkit config for the directory
+        toolkit_config = {
+            'type': 'filesystem',
+            'toolkit_name': name,
+            'base_directory': str(dir_path),
+            'recursive': recursive,
+        }
+    else:
+        # Load toolkit config
+        toolkit_config = _load_toolkit_config(toolkit)
+        source_type = toolkit_config.get('type', 'unknown')
+        click.echo(f"📦 Source toolkit: {source_type}")
+    
+    # Progress callback
+    def progress(message: str, phase: str):
+        click.echo(f"  [{phase}] {message}")
+    
+    try:
+        from alita_sdk.community.inventory import IngestionPipeline, IngestionConfig
+        
+        # Load configuration
+        if config:
+            click.echo(f"📋 Loading config from {config}")
+            if config.endswith('.yml') or config.endswith('.yaml'):
+                ingestion_config = IngestionConfig.from_yaml(config)
+            else:
+                ingestion_config = IngestionConfig.from_json(config)
+            
+            if model:
+                ingestion_config.llm_model = model
+            
+            ingestion_config.graph_path = graph
+            llm = _get_llm(ctx, ingestion_config.llm_model, ingestion_config.temperature)
+            
+            pipeline = IngestionPipeline(
+                llm=llm,
+                graph_path=ingestion_config.graph_path,
+                guardrails=ingestion_config.guardrails,
+            )
+        else:
+            click.echo("📋 Loading config from environment")
+            llm = _get_llm(ctx, model)
+            pipeline = IngestionPipeline(
+                llm=llm,
+                graph_path=graph,
+            )
+        
+        pipeline.progress_callback = progress
+        
+        # Get source toolkit and register it
+        source_toolkit = _get_source_toolkit(toolkit_config)
+        
+        import uuid
+        cli_runnable_config = {
+            'run_id': uuid.uuid4(),
+            'tags': ['cli', 'inventory', 'retry'],
+        }
+        
+        if hasattr(source_toolkit, 'set_runnable_config'):
+            source_toolkit.set_runnable_config(cli_runnable_config)
+        
+        pipeline.register_toolkit(name, source_toolkit)
+        
+        # Run delta update for failed files
+        result = pipeline.delta_update(
+            source=name,
+            file_paths=files_to_retry,
+            extract_relations=not no_relations,
+        )
+        
+        # Show result
+        if result.success:
+            click.echo(f"\n✅ Retry complete!")
+            click.echo(f"   Files retried: {len(files_to_retry)}")
+            click.echo(f"   Documents processed: {result.documents_processed}")
+            click.echo(f"   Entities added: {result.entities_added}")
+            click.echo(f"   Relations added: {result.relations_added}")
+            click.echo(f"   Duration: {result.duration_seconds:.1f}s")
+            
+            # Check if there are still failed files
+            if result.failed_documents:
+                click.echo(f"\n⚠️  {len(result.failed_documents)} files still failing")
+                click.echo(f"   Run 'alita inventory status -g {graph} -n {name}' to see details")
+            else:
+                # All retries succeeded - clear checkpoint
+                if os.path.exists(checkpoint_path):
+                    os.remove(checkpoint_path)
+                    click.echo(f"\n🧹 Checkpoint cleared (all files processed successfully)")
+        else:
+            click.echo(f"\n❌ Retry failed!")
+            for error in result.errors:
+                click.echo(f"   Error: {error}")
+            sys.exit(1)
+            
+    except Exception as e:
+        logger.exception("Retry failed")
+        raise click.ClickException(str(e))
+
+
+@inventory.command('status')
+@click.option('--graph', '-g', required=True, type=click.Path(),
+              help='Path to graph JSON file')
+@click.option('--name', '-n', required=True,
+              help='Source name to check status for')
+def status(graph: str, name: str):
+    """
+    Show ingestion checkpoint status for a source.
+    
+    Displays information about the last ingestion run including:
+    - Number of processed files
+    - Number of failed files
+    - Current phase
+    - Timestamps
+    
+    Example:
+        alita inventory status -g ./graph.json -n my-source
+    """
+    checkpoint_path = _get_checkpoint_path(graph, name)
+    
+    if not os.path.exists(checkpoint_path):
+        click.echo(f"\n❌ No checkpoint found for source '{name}'")
+        click.echo(f"   Expected: {checkpoint_path}")
+        click.echo(f"\n   No active or failed ingestion for this source.")
+        sys.exit(1)
+    
+    try:
+        with open(checkpoint_path, 'r') as f:
+            checkpoint = json.load(f)
+    except Exception as e:
+        raise click.ClickException(f"Failed to load checkpoint: {e}")
+    
+    click.echo(f"\n📋 Ingestion Status for '{name}'")
+    click.echo(f"   Checkpoint: {checkpoint_path}")
+    
+    click.echo(f"\n   Run ID: {checkpoint.get('run_id', 'unknown')}")
+    click.echo(f"   Phase: {checkpoint.get('phase', 'unknown')}")
+    click.echo(f"   Completed: {'Yes' if checkpoint.get('completed') else 'No'}")
+    
+    click.echo(f"\n   Started: {checkpoint.get('started_at', 'unknown')}")
+    click.echo(f"   Updated: {checkpoint.get('updated_at', 'unknown')}")
+    
+    processed_files = checkpoint.get('processed_files', [])
+    failed_files = checkpoint.get('failed_files', [])
+    
+    click.echo(f"\n   📊 Progress:")
+    click.echo(f"      Documents processed: {checkpoint.get('documents_processed', 0)}")
+    click.echo(f"      Entities added: {checkpoint.get('entities_added', 0)}")
+    click.echo(f"      Relations added: {checkpoint.get('relations_added', 0)}")
+    
+    click.echo(f"\n   📁 Files:")
+    click.echo(f"      Processed: {len(processed_files)}")
+    click.echo(f"      Failed: {len(failed_files)}")
+    
+    if failed_files:
+        # Count by attempts
+        by_attempts = {}
+        for f in failed_files:
+            attempts = f.get('attempts', 1)
+            by_attempts[attempts] = by_attempts.get(attempts, 0) + 1
+        
+        click.echo(f"\n   ❌ Failed files by attempt count:")
+        for attempts, count in sorted(by_attempts.items()):
+            click.echo(f"      {attempts} attempt(s): {count} files")
+        
+        # Show sample errors
+        click.echo(f"\n   📝 Sample errors (first 3):")
+        for f in failed_files[:3]:
+            file_path = f.get('file_path', 'unknown')
+            error = f.get('error', f.get('last_error', 'unknown error'))
+            # Truncate long paths and errors
+            if len(file_path) > 50:
+                file_path = '...' + file_path[-47:]
+            if len(error) > 60:
+                error = error[:57] + '...'
+            click.echo(f"      - {file_path}")
+            click.echo(f"        Error: {error}")
+    
+    errors = checkpoint.get('errors', [])
+    if errors:
+        click.echo(f"\n   ⚠️  Run errors:")
+        for error in errors[:3]:
+            click.echo(f"      - {error[:80]}{'...' if len(error) > 80 else ''}")
+    
+    if failed_files:
+        click.echo(f"\n   💡 To retry failed files:")
+        click.echo(f"      alita inventory retry --dir <path> -g {graph} -n {name}")
+        click.echo(f"      alita inventory retry --dir <path> -g {graph} -n {name} --force")
+    
+    click.echo()
+
+
+@inventory.command('stats')
+@click.option('--graph', '-g', required=True, type=click.Path(exists=True),
+              help='Path to graph JSON file')
+def stats(graph: str):
+    """
+    Show knowledge graph statistics.
+    
+    Example:
+        alita inventory stats -g ./graph.json
+    """
+    try:
+        from alita_sdk.community.inventory import KnowledgeGraph
+        
+        kg = KnowledgeGraph()
+        kg.load_from_json(graph)
+        stats = kg.get_stats()
+        
+        click.echo(f"\n📊 Knowledge Graph Statistics")
+        click.echo(f"   Path: {graph}")
+        click.echo(f"\n   Entities: {stats['node_count']}")
+        click.echo(f"   Relations: {stats['edge_count']}")
+        
+        if stats['entity_types']:
+            click.echo(f"\n   Entity Types:")
+            for etype, count in sorted(stats['entity_types'].items(), key=lambda x: -x[1]):
+                click.echo(f"     - {etype}: {count}")
+        
+        if stats['relation_types']:
+            click.echo(f"\n   Relation Types:")
+            for rtype, count in sorted(stats['relation_types'].items(), key=lambda x: -x[1]):
+                click.echo(f"     - {rtype}: {count}")
+        
+        if stats['source_toolkits']:
+            click.echo(f"\n   Sources: {', '.join(stats['source_toolkits'])}")
+        
+        if stats['last_saved']:
+            click.echo(f"\n   Last updated: {stats['last_saved']}")
+        
+        click.echo()
+        
+    except FileNotFoundError:
+        raise click.ClickException(f"Graph file not found: {graph}")
+    except Exception as e:
+        raise click.ClickException(str(e))
+
+
+@inventory.command('search')
+@click.argument('query')
+@click.option('--graph', '-g', required=True, type=click.Path(exists=True),
+              help='Path to graph JSON file')
+@click.option('--type', '-t', 'entity_type', default=None,
+              help='Filter by entity type')
+@click.option('--limit', '-n', default=10, type=int,
+              help='Maximum results (default: 10)')
+def search(query: str, graph: str, entity_type: Optional[str], limit: int):
+    """
+    Search for entities in the knowledge graph.
+    
+    Example:
+        alita inventory search "Payment" -g ./graph.json
+        alita inventory search "User" -g ./graph.json --type class
+    """
+    try:
+        from alita_sdk.community.inventory import KnowledgeGraph
+        
+        kg = KnowledgeGraph()
+        kg.load_from_json(graph)
+        
+        results = kg.search(query, top_k=limit, entity_type=entity_type)
+        
+        if not results:
+            click.echo(f"No entities found matching '{query}'")
+            return
+        
+        click.echo(f"\n🔍 Found {len(results)} entities matching '{query}':\n")
+        
+        for i, result in enumerate(results, 1):
+            entity = result['entity']
+            citation = entity.get('citation', {})
+            
+            click.echo(f"{i}. {entity.get('name')} ({entity.get('type')})")
+            
+            if citation:
+                file_path = citation.get('file_path', 'unknown')
+                line_info = ""
+                if citation.get('line_start'):
+                    line_info = f":{citation['line_start']}"
+                    if citation.get('line_end'):
+                        line_info += f"-{citation['line_end']}"
+                click.echo(f"   📍 {file_path}{line_info}")
+            
+            # Show description if available
+            if entity.get('description'):
+                desc = entity['description'][:80]
+                if len(entity['description']) > 80:
+                    desc += "..."
+                click.echo(f"   {desc}")
+            
+            click.echo()
+        
+    except FileNotFoundError:
+        raise click.ClickException(f"Graph file not found: {graph}")
+    except Exception as e:
+        raise click.ClickException(str(e))
+
+
+@inventory.command('entity')
+@click.argument('name')
+@click.option('--graph', '-g', required=True, type=click.Path(exists=True),
+              help='Path to graph JSON file')
+@click.option('--relations/--no-relations', default=True,
+              help='Include relations (default: yes)')
+def entity(name: str, graph: str, relations: bool):
+    """
+    Get detailed information about an entity.
+    
+    Example:
+        alita inventory entity "PaymentProcessor" -g ./graph.json
+    """
+    try:
+        from alita_sdk.community.inventory import InventoryRetrievalApiWrapper
+        
+        api = InventoryRetrievalApiWrapper(graph_path=graph)
+        result = api.get_entity(name, include_relations=relations)
+        
+        click.echo(f"\n{result}")
+        
+    except FileNotFoundError:
+        raise click.ClickException(f"Graph file not found: {graph}")
+    except Exception as e:
+        raise click.ClickException(str(e))
+
+
+@inventory.command('impact')
+@click.argument('name')
+@click.option('--graph', '-g', required=True, type=click.Path(exists=True),
+              help='Path to graph JSON file')
+@click.option('--direction', '-d', type=click.Choice(['upstream', 'downstream']),
+              default='downstream', help='Analysis direction (default: downstream)')
+@click.option('--depth', default=3, type=int,
+              help='Maximum traversal depth (default: 3)')
+def impact(name: str, graph: str, direction: str, depth: int):
+    """
+    Analyze impact of changes to an entity.
+    
+    Example:
+        alita inventory impact "UserService" -g ./graph.json
+        alita inventory impact "Database" -g ./graph.json --direction upstream
+    """
+    try:
+        from alita_sdk.community.inventory import InventoryRetrievalApiWrapper
+        
+        api = InventoryRetrievalApiWrapper(graph_path=graph)
+        result = api.impact_analysis(name, direction=direction, max_depth=depth)
+        
+        click.echo(f"\n{result}")
+        
+    except FileNotFoundError:
+        raise click.ClickException(f"Graph file not found: {graph}")
+    except Exception as e:
+        raise click.ClickException(str(e))
+
+
+@inventory.command('visualize')
+@click.option('--graph', '-g', required=True, type=click.Path(exists=True),
+              help='Path to graph JSON file')
+@click.option('--output', '-o', default=None, type=click.Path(),
+              help='Output HTML file path (default: graph_visualization.html in same dir)')
+@click.option('--open/--no-open', 'open_browser', default=True,
+              help='Open in browser after generation (default: yes)')
+@click.option('--title', '-t', default=None,
+              help='Title for the visualization')
+def visualize(graph: str, output: Optional[str], open_browser: bool, title: Optional[str]):
+    """
+    Generate an interactive visualization of the knowledge graph.
+    
+    Creates a standalone HTML file with D3.js-powered graph visualization.
+    Features include:
+    - Force-directed layout
+    - Color-coded entity types
+    - Node size based on connections
+    - Interactive drag, zoom, and pan
+    - Search and filter by entity type
+    - Click nodes for detailed info
+    
+    Example:
+        alita inventory visualize -g ./graph.json
+        alita inventory visualize -g ./graph.json -o my_graph.html
+        alita inventory visualize -g ./graph.json --no-open
+    """
+    try:
+        from alita_sdk.community.inventory.visualize import generate_visualization
+        from alita_sdk.community.inventory import KnowledgeGraph
+        import webbrowser
+        import os
+        
+        # Default output path
+        if output is None:
+            graph_dir = os.path.dirname(os.path.abspath(graph))
+            graph_name = os.path.splitext(os.path.basename(graph))[0]
+            output = os.path.join(graph_dir, f"{graph_name}_visualization.html")
+        
+        # Default title
+        if title is None:
+            title = os.path.splitext(os.path.basename(graph))[0].replace('_', ' ').title()
+        
+        click.echo(f"\n🎨 Generating graph visualization...")
+        click.echo(f"   Source: {graph}")
+        
+        # Generate visualization
+        html_path = generate_visualization(graph, output, title)
+        
+        click.echo(f"   Output: {html_path}")
+        
+        # Show graph stats
+        kg = KnowledgeGraph()
+        kg.load_from_json(graph)
+        stats = kg.get_stats()
+        click.echo(f"\n   📊 Graph contains:")
+        click.echo(f"      - {stats['node_count']} entities")
+        click.echo(f"      - {stats['edge_count']} relations")
+        if stats['entity_types']:
+            click.echo(f"      - {len(stats['entity_types'])} entity types")
+        
+        if open_browser:
+            click.echo(f"\n   Opening in browser...")
+            webbrowser.open(f"file://{os.path.abspath(html_path)}")
+        
+        click.echo(f"\n✅ Visualization complete!")
+        click.echo()
+        
+    except FileNotFoundError:
+        raise click.ClickException(f"Graph file not found: {graph}")
+    except ImportError as e:
+        raise click.ClickException(f"Visualization module not available: {e}")
+    except Exception as e:
+        raise click.ClickException(str(e))
+
+
+@inventory.command('enrich')
+@click.option('--graph', '-g', required=True, type=click.Path(exists=True),
+              help='Path to graph JSON file')
+@click.option('--output', '-o', default=None, type=click.Path(),
+              help='Output graph file (default: overwrite input)')
+@click.option('--deduplicate/--no-deduplicate', default=False,
+              help='Merge entities with exact same name (DISABLED by default, use with caution)')
+@click.option('--cross-source/--no-cross-source', default=True,
+              help='Link same-named entities across sources (default: yes)')
+@click.option('--semantic/--no-semantic', default=True,
+              help='Create semantic cross-links based on shared concepts (default: yes)')
+@click.option('--orphans/--no-orphans', default=True,
+              help='Connect orphan nodes to related entities (default: yes)')
+@click.option('--similarity/--no-similarity', default=False,
+              help='Link entities with similar names (default: no)')
+@click.option('--dry-run', is_flag=True, default=False,
+              help='Show what would be done without saving')
+def enrich(graph: str, output: Optional[str], deduplicate: bool, cross_source: bool,
+           semantic: bool, orphans: bool, similarity: bool, dry_run: bool):
+    """
+    Enrich a knowledge graph with cross-linking.
+    
+    Post-processes the graph to improve connectivity by creating links:
+    
+    1. CROSS-SOURCE LINKING: Link entities across sources
+       - SDK class ↔ docs concept, code ↔ documentation
+       - Automatically determines relationship type
+    
+    2. SEMANTIC LINKING: Link entities sharing concepts
+       - Finds entities with overlapping significant words
+       - Creates LINKS between related entities
+       - Example: "Artifact Toolkit" --[related_to]--> "Configure Artifact Toolkit"
+    
+    3. ORPHAN LINKING: Connect isolated nodes
+       - Links unconnected nodes to related entities
+    
+    4. DEDUPLICATION (optional, disabled by default):
+       - Use --deduplicate to merge exact name matches
+       - Use with caution - can lose semantic meaning
+    
+    Example:
+        alita inventory enrich -g ./graph.json
+        alita inventory enrich -g ./graph.json -o enriched.json
+        alita inventory enrich -g ./graph.json --deduplicate
+        alita inventory enrich -g ./graph.json --dry-run
+    """
+    try:
+        from alita_sdk.community.inventory.enrichment import GraphEnricher
+        
+        click.echo(f"\n🔗 Enriching knowledge graph...")
+        click.echo(f"   Source: {graph}")
+        
+        enricher = GraphEnricher(graph)
+        
+        # Show initial stats
+        initial_nodes = len(enricher.nodes_by_id)
+        initial_links = len(enricher.graph_data.get("links", []))
+        click.echo(f"   Initial: {initial_nodes} nodes, {initial_links} links")
+        
+        # Run enrichment
+        stats = enricher.enrich(
+            deduplicate=deduplicate,
+            cross_source=cross_source,
+            semantic_links=semantic,
+            orphans=orphans,
+            similarity=similarity,
+        )
+        
+        click.echo(f"\n   📊 Enrichment results:")
+        
+        if deduplicate:
+            click.echo(f"      Entities merged:    {stats.get('entities_merged', 0)} (exact name matches into {stats.get('merge_groups', 0)} groups)")
+            final_nodes = len(enricher.nodes_by_id)
+            click.echo(f"      Node reduction:     {initial_nodes} → {final_nodes}")
+        
+        click.echo(f"      Cross-source links: +{stats.get('cross_source_links', 0)}")
+        
+        if semantic:
+            click.echo(f"      Semantic links:     +{stats.get('semantic_links', 0)}")
+        
+        click.echo(f"      Orphan connections: +{stats.get('orphan_links', 0)}")
+        
+        if similarity:
+            click.echo(f"      Similarity links:   +{stats.get('similarity_links', 0)}")
+        
+        click.echo(f"      Total new links:    +{len(enricher.new_links)}")
+        
+        if dry_run:
+            click.echo(f"\n   🔍 Dry run - no changes saved")
+            
+            # Show merge examples
+            if deduplicate and enricher.merged_nodes:
+                click.echo(f"\n   Sample merged entities:")
+                for merge in enricher.merged_nodes[:5]:
+                    new_node = merge["new_node"]
+                    types = merge.get("merged_types", [])
+                    click.echo(f"      '{new_node['name']}' [{' + '.join(set(types))}] → [{new_node['type']}]")
+            
+            # Show link examples
+            click.echo(f"\n   Sample new links:")
+            for link in enricher.new_links[:10]:
+                src = enricher.nodes_by_id.get(link['source'], {})
+                tgt = enricher.nodes_by_id.get(link['target'], {})
+                click.echo(f"      {src.get('name', '?')[:25]:25} --[{link['relation_type']}]--> {tgt.get('name', '?')[:25]}")
+        else:
+            output_path = enricher.save(output)
+            click.echo(f"\n   💾 Saved to: {output_path}")
+        
+        click.echo(f"\n✅ Enrichment complete!")
+        click.echo()
+        
+    except FileNotFoundError:
+        raise click.ClickException(f"Graph file not found: {graph}")
+    except ImportError as e:
+        raise click.ClickException(f"Enrichment module not available: {e}")
+    except Exception as e:
+        raise click.ClickException(str(e))
+
+
+# ========== Helper Functions ==========
+
+def _get_checkpoint_path(graph: str, source_name: str) -> str:
+    """
+    Get the checkpoint file path for a source.
+    
+    Checkpoint files are stored in the same directory as the graph file,
+    with naming pattern: .ingestion-checkpoint-{source_name}.json
+    
+    Args:
+        graph: Path to the graph JSON file
+        source_name: Name of the source toolkit
+        
+    Returns:
+        Absolute path to the checkpoint file
+    """
+    graph_path = Path(graph).resolve()
+    graph_dir = graph_path.parent
+    return str(graph_dir / f".ingestion-checkpoint-{source_name}.json")
+
+
+def _load_toolkit_config(toolkit_path: str) -> Dict[str, Any]:
+    """Deprecated: Use alita_sdk.community.inventory.toolkit_utils.load_toolkit_config instead."""
+    from alita_sdk.community.inventory.toolkit_utils import load_toolkit_config
+    return load_toolkit_config(toolkit_path)
+
+
+def _get_llm(ctx, model: Optional[str] = None, temperature: float = 0.0):
+    """Deprecated: Use alita_sdk.community.inventory.toolkit_utils.get_llm_for_config instead."""
+    from .cli import get_client
+    from alita_sdk.community.inventory.toolkit_utils import get_llm_for_config
+    
+    client = get_client(ctx)
+    return get_llm_for_config(client, model=model, temperature=temperature)
+
+
+def _get_source_toolkit(toolkit_config: Dict[str, Any]):
+    """Deprecated: Use alita_sdk.community.inventory.toolkit_utils.get_source_toolkit instead."""
+    from alita_sdk.community.inventory.toolkit_utils import get_source_toolkit
+    
+    try:
+        return get_source_toolkit(toolkit_config)
+    except ValueError as e:
+        # Convert ValueError to ClickException for CLI context
+        raise click.ClickException(str(e))