show_analysis.py

#!/usr/bin/env python3
"""
Show complete reasoning analysis using the existing framework.
"""

import sys
from pathlib import Path

# Add src to path
sys.path.append(str(Path(__file__).parent / "src"))

from models.model_factory import ModelFactory
from adaptive.adaptive_cot import AdaptiveCoT

def main():
    if len(sys.argv) < 3:
        print("Usage: python show_analysis.py <problem> <model_path> [gpu_id]")
        print("Example: python show_analysis.py 'What is 2+2?' '/path/to/model' 1")
        sys.exit(1)
    
    problem = sys.argv[1]
    model_path = sys.argv[2]
    gpu_id = int(sys.argv[3]) if len(sys.argv) > 3 else 1
    
    print(f"🔬 Loading model: {model_path}")
    
    # Load model using same type as working script
    model = ModelFactory.create_model(
        model_type="deepseek",
        model_name=model_path,
        config={"gpu_id": gpu_id}
    )
    model.load_model()
    
    # Create Adaptive CoT
    config = {
        "adaptive_branching": True,
        "min_branches": 3,
        "max_branches": 5,
        "default_branches": 5,
        "num_fewshot": 8,
        "entropy_threshold": 0.8,
        "kl_threshold": 0.5,
        "confidence_threshold": 0.7,
    }
    
    adaptive_cot = AdaptiveCoT(model, config)
    
    # Show complete analysis with 5 branches
    adaptive_cot.show_complete_analysis(problem, num_branches=5)

if __name__ == "__main__":
    main()