security_demo/install_llm_deps.py

#!/usr/bin/env python3
"""
安装LLM集成所需的依赖包
"""
import subprocess
import sys
import os

def install_package(package_name):
    """安装Python包"""
    try:
        print(f"正在安装 {package_name}...")
        subprocess.check_call([sys.executable, "-m", "pip", "install", package_name])
        print(f"✅ {package_name} 安装成功")
        return True
    except subprocess.CalledProcessError as e:
        print(f"❌ {package_name} 安装失败: {e}")
        return False

def main():
    """主函数"""
    print("🚀 开始安装LLM集成依赖...")
    
    # 必需的依赖包
    dependencies = [
        "anthropic>=0.34.0",  # Claude API
        "openai>=1.3.0",      # OpenAI API (也用于本地模型)
        "python-dotenv",      # 环境变量管理
    ]
    
    success_count = 0
    
    for dep in dependencies:
        if install_package(dep):
            success_count += 1
    
    print(f"\n📊 安装结果: {success_count}/{len(dependencies)} 个包安装成功")
    
    if success_count == len(dependencies):
        print("\n🎉 所有依赖安装完成！")
        print("\n📝 接下来的配置步骤:")
        print("1. 复制 .env.example 为 .env")
        print("2. 在 .env 中配置你的API密钥")
        print("3. 在 configs/llm.yaml 中选择要使用的模型提供商")
        print("4. 重启后端服务")
    else:
        print("\n⚠️  部分依赖安装失败，请检查错误信息")
        return 1
    
    return 0

if __name__ == "__main__":
    sys.exit(main())