ai_architecture/run_examples.py

#!/usr/bin/env python3
"""
AI Architecture 示例运行脚本
"""

import sys
import os
from pathlib import Path

# 添加项目根目录到Python路径
project_root = Path(__file__).parent
sys.path.insert(0, str(project_root))

def main():
    """主函数"""
    print("AI Architecture 示例运行器")
    print("=" * 40)
    
    # 检查环境变量
    if not os.getenv("OPENROUTER_API_KEY"):
        print("❌ 错误: 未设置 OPENROUTER_API_KEY 环境变量")
        print("\n请按以下步骤配置:")
        print("1. 复制 env.example 为 .env")
        print("2. 在 .env 文件中设置您的 OpenRouter API 密钥")
        print("3. 重新运行此脚本")
        return
    
    print("✅ 环境变量检查通过")
    
    # 选择要运行的示例
    print("\n请选择要运行的示例:")
    print("1. 基本使用示例 (basic_usage.py)")
    print("2. 高级使用示例 (advanced_usage.py)")
    print("3. 运行所有示例")
    
    try:
        choice = input("\n请输入选择 (1-3): ").strip()
        
        if choice == "1":
            print("\n运行基本使用示例...")
            from examples.basic_usage import main as basic_main
            basic_main()
        elif choice == "2":
            print("\n运行高级使用示例...")
            from examples.advanced_usage import main as advanced_main
            advanced_main()
        elif choice == "3":
            print("\n运行所有示例...")
            from examples.basic_usage import main as basic_main
            from examples.advanced_usage import main as advanced_main
            
            print("\n" + "="*50)
            basic_main()
            print("\n" + "="*50)
            advanced_main()
        else:
            print("❌ 无效选择")
            return
            
    except KeyboardInterrupt:
        print("\n\n👋 用户取消操作")
    except Exception as e:
        print(f"\n❌ 运行失败: {e}")
        print("请检查配置和依赖是否正确安装")

if __name__ == "__main__":
    main()