run.py

#!/usr/bin/env python3
"""
代码分析Agent系统启动脚本
"""
import os
import sys
import subprocess

def check_python_version():
    """检查Python版本"""
    if sys.version_info < (3, 8):
        print("❌ 需要Python 3.8或更高版本")
        sys.exit(1)
    print(f"✅ Python版本: {sys.version}")

def install_requirements():
    """安装依赖"""
    print("📦 检查并安装依赖...")
    
    try:
        import langchain
        import langgraph
        import rich
        print("✅ 主要依赖已安装")
        return True
    except ImportError:
        print("⚠️  检测到缺少依赖，正在安装...")
        
        try:
            subprocess.check_call([
                sys.executable, "-m", "pip", "install", "-r", "requirements.txt"
            ])
            print("✅ 依赖安装完成")
            return True
        except subprocess.CalledProcessError:
            print("❌ 依赖安装失败")
            return False

def check_environment():
    """检查环境配置"""
    print("🔧 检查环境配置...")
    
    # 检查LLM提供商
    llm_provider = os.getenv('LLM_PROVIDER', 'openai').lower()
    print(f"🤖 LLM提供商: {llm_provider.upper()}")
    
    if llm_provider == 'qwen':
        # 检查Qwen API Key
        api_key = os.getenv('QWEN_API_KEY')
        if not api_key:
            print("⚠️  未设置QWEN_API_KEY环境变量")
            print("请设置您的通义千问API密钥:")
            print("export QWEN_API_KEY='your-qwen-api-key-here'")
            print("或在.env文件中配置")
            return False
    else:
        # 检查OpenAI API Key
        api_key = os.getenv('OPENAI_API_KEY')
        if not api_key:
            print("⚠️  未设置OPENAI_API_KEY环境变量")
            print("请设置您的OpenAI API密钥:")
            print("export OPENAI_API_KEY='your-api-key-here'")
            print("或在.env文件中配置")
            return False
    
    print("✅ 环境配置检查通过")
    return True

def show_usage():
    """显示使用说明"""
    print("""
🤖 代码分析Agent系统
==================

使用方法:
  python run.py demo                    # 运行演示模式
  python run.py analyze [目录] [需求]    # 分析指定目录
  python run.py example                 # 运行使用示例
  python run.py test                    # 运行测试
  python run.py model-info              # 显示模型信息

环境配置:
  设置LLM_PROVIDER=openai 或 qwen
  
  OpenAI: 设置OPENAI_API_KEY
  Qwen:   设置QWEN_API_KEY

示例:
  python run.py demo
  python run.py analyze ./my_project "分析代码质量"
  python run.py model-info
""")

def run_demo():
    """运行演示模式"""
    print("🎯 启动演示模式...")
    os.system(f"{sys.executable} main.py --demo")

def run_analysis(directory=None, request=None):
    """运行代码分析"""
    directory = directory or "."
    request = request or "分析项目代码质量"
    
    print(f"🔍 开始分析目录: {directory}")
    print(f"📋 分析需求: {request}")
    
    cmd = f'{sys.executable} main.py --directory "{directory}" --request "{request}" --save'
    os.system(cmd)

def run_example():
    """运行使用示例"""
    print("💡 运行使用示例...")
    os.system(f"{sys.executable} example.py")

def run_tests():
    """运行测试"""
    print("🧪 运行测试...")
    
    # 检查pytest是否安装
    try:
        import pytest
    except ImportError:
        print("📦 安装pytest...")
        subprocess.check_call([sys.executable, "-m", "pip", "install", "pytest", "pytest-asyncio"])
    
    # 运行测试
    os.system(f"{sys.executable} -m pytest tests/ -v")

def show_model_info():
    """显示模型信息"""
    print("🤖 模型信息查看...")
    os.system(f"{sys.executable} main.py --model-info")

def main():
    """主函数"""
    print("🚀 代码分析Agent系统启动器")
    print("=" * 40)
    
    # 检查Python版本
    check_python_version()
    
    # 解析命令行参数
    if len(sys.argv) < 2:
        show_usage()
        return
    
    command = sys.argv[1].lower()
    
    # 安装依赖
    if not install_requirements():
        return
    
    # 检查环境（除了测试模式）
    if command != 'test' and not check_environment():
        print("⚠️  环境检查失败，但您仍可以运行测试和示例")
    
    # 执行命令
    if command == 'demo':
        run_demo()
    elif command == 'analyze':
        directory = sys.argv[2] if len(sys.argv) > 2 else None
        request = sys.argv[3] if len(sys.argv) > 3 else None
        run_analysis(directory, request)
    elif command == 'example':
        run_example()
    elif command == 'test':
        run_tests()
    elif command == 'model-info':
        show_model_info()
    else:
        print(f"❌ 未知命令: {command}")
        show_usage()

if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        print("\n🛑 用户中断操作")
    except Exception as e:
        print(f"❌ 执行错误: {e}")
        import traceback
        traceback.print_exc()