check_cache.py

"""
验证模型缓存配置
"""

import os
import sys
from pathlib import Path

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

from src.utils.config_manager import ConfigManager


def check_cache_dir():
    """检查缓存目录配置"""
    print("=" * 60)
    print("📦 模型缓存配置检查")
    print("=" * 60)
    print()
    
    # 加载配置
    config = ConfigManager()
    cache_dir = config.get("advanced.cache_dir", "./models")
    
    # 转换为绝对路径
    cache_path = Path(cache_dir).resolve()
    
    print(f"配置的缓存目录: {cache_dir}")
    print(f"绝对路径: {cache_path}")
    print()
    
    # 检查目录是否存在
    if cache_path.exists():
        print(f"✅ 目录存在")
        
        # 检查是否有写权限
        if os.access(cache_path, os.W_OK):
            print(f"✅ 有写入权限")
        else:
            print(f"❌ 没有写入权限")
        
        # 列出已下载的模型
        model_dirs = list(cache_path.glob("models--*"))
        if model_dirs:
            print()
            print(f"已下载的模型 ({len(model_dirs)} 个):")
            for model_dir in model_dirs:
                # 解析模型名称
                model_name = model_dir.name.replace("models--", "").replace("--", "/")
                
                # 计算目录大小
                total_size = sum(f.stat().st_size for f in model_dir.rglob('*') if f.is_file())
                size_gb = total_size / (1024 ** 3)
                
                print(f"  - {model_name} ({size_gb:.2f} GB)")
        else:
            print()
            print("📭 还没有下载任何模型")
            print("首次运行时会自动下载")
    else:
        print(f"⚠️  目录不存在，将在首次运行时自动创建")
        
        # 检查父目录是否存在且有写权限
        parent_dir = cache_path.parent
        if parent_dir.exists() and os.access(parent_dir, os.W_OK):
            print(f"✅ 父目录存在且有写入权限")
        else:
            print(f"❌ 无法创建目录，请检查权限")
    
    print()
    print("=" * 60)
    print("配置信息:")
    print(f"  默认模型: {config.get('model.default_model')}")
    print(f"  缓存目录: {cache_dir}")
    print(f"  后端偏好: {config.get('system.backend_preference')}")
    print("=" * 60)


if __name__ == "__main__":
    check_cache_dir()