Initial commit: Comfyui_o1key v1.10.0

models_config.py

@@ -0,0 +1,449 @@
+"""
+模型配置中心
+用于集中管理所有支持的 Gemini 模型
+
+使用方式:
+    1. 添加新模型: 在对应的模型列表中添加新的模型字典
+    2. 临时关闭模型: 将模型的 enabled 字段设为 False
+    3. 重新启用模型: 将模型的 enabled 字段改回 True
+
+模型类型:
+    - GEMINI_MODELS: Nano Banana Pro 图像生成模型
+    - GEMINI_FLASH_MODELS: Google Gemini Flash 文本生成模型
+
+示例:
+    添加新模型:
+    {
+        "id": "gemini-新模型名称",
+        "description": "模型说明和特点",
+        "enabled": True,
+        "endpoint_type": "standard"  # 端点类型: "dynamic", "standard", "flatfee"
+    }
+    
+    临时关闭模型:
+    将对应模型的 "enabled": True 改为 "enabled": False
+"""
+
+from typing import List, Dict, Optional
+
+
+# ============================================================
+# 模型配置列表
+# ============================================================
+
+# ============================================================
+# Nano Banana Pro 图像生成模型
+# ============================================================
+
+GEMINI_MODELS = [
+    {
+        "id": "nano-banana-pro",
+        "description": "Nano Banana Pro,根据分辨率自动选择端点 (1K/2K/4K),高性能图像生成模型",
+        "enabled": True,
+        "endpoint_type": "dynamic",
+        "endpoint": None  # 动态端点，由代码根据分辨率选择
+    },
+    {
+        "id": "gemini-3-pro-image-preview-url",
+        "description": "URL 模式,根据分辨率自动选择端点 (1K/2K/4K),推荐用于需要不同分辨率的场景",
+        "enabled": False,
+        "endpoint_type": "dynamic",
+        "endpoint": None  # 动态端点，由代码根据分辨率选择
+    },
+    {
+        "id": "gemini-3-pro-image-preview",
+        "description": "标准模式,固定端点,适用于常规图像生成",
+        "enabled": True,
+        "endpoint_type": "standard",
+        "endpoint": "/v1beta/models/gemini-3-pro-image-preview:generateContent"
+    },
+    {
+        "id": "gemini-3-pro-image-preview-flatfee",
+        "description": "固定费用模式,固定端点,按固定价格计费 (暂时不可用-504错误)",
+        "enabled": False,  # 暂时禁用：端点返回 504 错误
+        "endpoint_type": "flatfee",
+        "endpoint": "/v1beta/models/gemini-3-pro-image-preview-flatfee:generateContent"
+    },
+    {
+        "id": "nano-banana-2",
+        "description": "Nano Banana 2 模型,固定端点,适用于高质量图像生成",
+        "enabled": False,
+        "endpoint_type": "standard",
+        "endpoint": "/v1beta/models/nano-banana-2:generateContent"
+    }
+]
+
+
+# ============================================================
+# Google Gemini Flash 文本生成模型
+# ============================================================
+
+GEMINI_FLASH_MODELS = [
+    {
+        "id": "gemini-3-flash-preview",
+        "description": "Gemini 3 Flash,快速多模态文本生成,支持图片和视频输入",
+        "enabled": True,
+        "endpoints": {
+            "不思考": "/v1beta/models/gemini-3-flash-preview-nothinking:generateContent",
+            "高": "/v1beta/models/gemini-3-flash-preview-high:generateContent"
+        }
+    }
+]
+
+
+# ============================================================
+# 工具函数
+# ============================================================
+
+def get_enabled_models() -> List[str]:
+    """
+    获取所有启用的模型 ID 列表
+    
+    Returns:
+        启用的模型 ID 列表
+    
+    Example:
+        >>> get_enabled_models()
+        ['gemini-3-pro-image-preview-url', 'gemini-3-pro-image-preview', ...]
+    """
+    return [model["id"] for model in GEMINI_MODELS if model.get("enabled", False)]
+
+
+def get_all_models() -> List[str]:
+    """
+    获取所有模型 ID 列表（包括已禁用的）
+    
+    Returns:
+        所有模型 ID 列表
+    
+    Example:
+        >>> get_all_models()
+        ['gemini-3-pro-image-preview-url', 'gemini-3-pro-image-preview', ...]
+    """
+    return [model["id"] for model in GEMINI_MODELS]
+
+
+def get_model_config(model_id: str) -> Optional[Dict]:
+    """
+    根据模型 ID 获取完整的模型配置
+    
+    Args:
+        model_id: 模型 ID
+    
+    Returns:
+        模型配置字典，如果未找到则返回 None
+    
+    Example:
+        >>> config = get_model_config("gemini-3-pro-image-preview-url")
+        >>> print(config["description"])
+        URL 模式,根据分辨率自动选择端点 (1K/2K/4K)
+    """
+    for model in GEMINI_MODELS:
+        if model["id"] == model_id:
+            return model
+    return None
+
+
+def is_model_enabled(model_id: str) -> bool:
+    """
+    检查指定模型是否启用
+    
+    Args:
+        model_id: 模型 ID
+    
+    Returns:
+        True 如果模型启用，False 如果禁用或不存在
+    
+    Example:
+        >>> is_model_enabled("gemini-3-pro-image-preview-url")
+        True
+    """
+    config = get_model_config(model_id)
+    if config is None:
+        return False
+    return config.get("enabled", False)
+
+
+def get_model_description(model_id: str) -> str:
+    """
+    获取模型的描述信息
+    
+    Args:
+        model_id: 模型 ID
+    
+    Returns:
+        模型描述，如果未找到则返回空字符串
+    
+    Example:
+        >>> get_model_description("gemini-3-pro-image-preview")
+        '标准模式,固定端点,适用于常规图像生成'
+    """
+    config = get_model_config(model_id)
+    if config is None:
+        return ""
+    return config.get("description", "")
+
+
+def get_endpoint_type(model_id: str) -> Optional[str]:
+    """
+    获取模型的端点类型
+    
+    Args:
+        model_id: 模型 ID
+    
+    Returns:
+        端点类型 ("dynamic", "standard", "flatfee")，如果未找到则返回 None
+    
+    Example:
+        >>> get_endpoint_type("gemini-3-pro-image-preview-url")
+        'dynamic'
+    """
+    config = get_model_config(model_id)
+    if config is None:
+        return None
+    return config.get("endpoint_type")
+
+
+def get_model_endpoint(model_id: str) -> Optional[str]:
+    """
+    获取模型的 API 端点
+    
+    Args:
+        model_id: 模型 ID
+    
+    Returns:
+        API 端点路径，如果未找到或为动态端点则返回 None
+    
+    Example:
+        >>> get_model_endpoint("gemini-3-pro-image-preview")
+        '/v1beta/models/gemini-3-pro-image-preview:generateContent'
+        >>> get_model_endpoint("gemini-3-pro-image-preview-url")
+        None  # 动态端点
+    """
+    config = get_model_config(model_id)
+    if config is None:
+        return None
+    return config.get("endpoint")
+
+
+# ============================================================
+# Gemini Flash 模型工具函数
+# ============================================================
+
+def get_enabled_flash_models() -> List[str]:
+    """
+    获取所有启用的 Flash 模型 ID 列表
+    
+    Returns:
+        启用的 Flash 模型 ID 列表
+    
+    Example:
+        >>> get_enabled_flash_models()
+        ['gemini-3-flash-preview']
+    """
+    return [model["id"] for model in GEMINI_FLASH_MODELS if model.get("enabled", False)]
+
+
+def get_all_flash_models() -> List[str]:
+    """
+    获取所有 Flash 模型 ID 列表（包括已禁用的）
+    
+    Returns:
+        所有 Flash 模型 ID 列表
+    """
+    return [model["id"] for model in GEMINI_FLASH_MODELS]
+
+
+def get_flash_model_config(model_id: str) -> Optional[Dict]:
+    """
+    根据模型 ID 获取 Flash 模型的完整配置
+    
+    Args:
+        model_id: 模型 ID
+    
+    Returns:
+        模型配置字典，如果未找到则返回 None
+    
+    Example:
+        >>> config = get_flash_model_config("gemini-3-flash-preview")
+        >>> print(config["description"])
+        'Gemini 3 Flash,快速多模态文本生成,支持图片和视频输入'
+    """
+    for model in GEMINI_FLASH_MODELS:
+        if model["id"] == model_id:
+            return model
+    return None
+
+
+def is_flash_model_enabled(model_id: str) -> bool:
+    """
+    检查指定 Flash 模型是否启用
+    
+    Args:
+        model_id: 模型 ID
+    
+    Returns:
+        True 如果模型启用，False 如果禁用或不存在
+    """
+    config = get_flash_model_config(model_id)
+    if config is None:
+        return False
+    return config.get("enabled", False)
+
+
+def get_flash_model_endpoint(model_id: str, thinking_depth: str = "不思考") -> Optional[str]:
+    """
+    获取 Flash 模型的 API 端点
+    
+    Args:
+        model_id: 模型 ID
+        thinking_depth: 思考深度 ("不思考" 或 "高")
+    
+    Returns:
+        API 端点路径，如果未找到则返回 None
+    
+    Example:
+        >>> get_flash_model_endpoint("gemini-3-flash-preview", "不思考")
+        '/v1beta/models/gemini-3-flash-preview-nothinking:generateContent'
+        >>> get_flash_model_endpoint("gemini-3-flash-preview", "高")
+        '/v1beta/models/gemini-3-flash-preview-high:generateContent'
+    """
+    config = get_flash_model_config(model_id)
+    if config is None:
+        return None
+    
+    endpoints = config.get("endpoints", {})
+    return endpoints.get(thinking_depth)
+
+
+def get_flash_model_description(model_id: str) -> str:
+    """
+    获取 Flash 模型的描述信息
+    
+    Args:
+        model_id: 模型 ID
+    
+    Returns:
+        模型描述，如果未找到则返回空字符串
+    """
+    config = get_flash_model_config(model_id)
+    if config is None:
+        return ""
+    return config.get("description", "")
+
+
+# ============================================================
+# 向后兼容性检查
+# ============================================================
+
+def validate_models_config() -> None:
+    """
+    验证模型配置的完整性
+    
+    检查:
+    - 每个模型必须有 id, description, enabled, endpoint_type, endpoint 字段
+    - 非动态端点模型必须配置有效的 endpoint
+    - 至少有一个模型是启用的
+    
+    Raises:
+        ValueError: 如果配置不合法
+    """
+    if not GEMINI_MODELS:
+        raise ValueError("GEMINI_MODELS 列表不能为空")
+    
+    required_fields = ["id", "description", "enabled", "endpoint_type", "endpoint"]
+    valid_endpoint_types = ["dynamic", "standard", "flatfee"]
+    
+    for i, model in enumerate(GEMINI_MODELS):
+        # 检查必需字段
+        for field in required_fields:
+            if field not in model:
+                raise ValueError(f"模型 #{i} 缺少必需字段: {field}")
+        
+        # 检查 endpoint_type 是否合法
+        if model["endpoint_type"] not in valid_endpoint_types:
+            raise ValueError(
+                f"模型 {model['id']} 的 endpoint_type '{model['endpoint_type']}' 不合法。"
+                f"必须是: {', '.join(valid_endpoint_types)}"
+            )
+        
+        # 检查非动态端点模型必须有有效的 endpoint
+        if model["endpoint_type"] != "dynamic" and not model.get("endpoint"):
+            raise ValueError(
+                f"模型 {model['id']} 的 endpoint_type 为 '{model['endpoint_type']}'，"
+                f"但未配置有效的 endpoint 字段"
+            )
+        
+        # 检查端点格式（如果配置了）
+        endpoint = model.get("endpoint")
+        if endpoint and not endpoint.startswith("/v1beta/models/"):
+            raise ValueError(
+                f"模型 {model['id']} 的 endpoint '{endpoint}' 格式不正确。"
+                f"应以 '/v1beta/models/' 开头"
+            )
+    
+    # 检查至少有一个启用的模型
+    if not get_enabled_models():
+        raise ValueError("至少需要启用一个模型")
+
+
+def validate_flash_models_config() -> None:
+    """
+    验证 Flash 模型配置的完整性
+    
+    检查:
+    - 每个模型必须有 id, description, enabled, endpoints 字段
+    - endpoints 必须包含所有思考深度选项
+    - 至少有一个模型是启用的
+    
+    Raises:
+        ValueError: 如果配置不合法
+    """
+    if not GEMINI_FLASH_MODELS:
+        raise ValueError("GEMINI_FLASH_MODELS 列表不能为空")
+    
+    required_fields = ["id", "description", "enabled", "endpoints"]
+    required_thinking_depths = ["不思考", "高"]
+    
+    for i, model in enumerate(GEMINI_FLASH_MODELS):
+        # 检查必需字段
+        for field in required_fields:
+            if field not in model:
+                raise ValueError(f"Flash 模型 #{i} 缺少必需字段: {field}")
+        
+        # 检查 endpoints 字典
+        endpoints = model.get("endpoints", {})
+        if not isinstance(endpoints, dict):
+            raise ValueError(f"Flash 模型 {model['id']} 的 endpoints 必须是字典")
+        
+        # 检查所有思考深度选项都有对应端点
+        for depth in required_thinking_depths:
+            if depth not in endpoints:
+                raise ValueError(
+                    f"Flash 模型 {model['id']} 的 endpoints 缺少 '{depth}' 思考深度"
+                )
+            
+            endpoint = endpoints[depth]
+            if not endpoint or not endpoint.startswith("/v1beta/models/"):
+                raise ValueError(
+                    f"Flash 模型 {model['id']} 的端点 '{endpoint}' 格式不正确。"
+                    f"应以 '/v1beta/models/' 开头"
+                )
+    
+    # 检查至少有一个启用的模型
+    if not get_enabled_flash_models():
+        raise ValueError("至少需要启用一个 Flash 模型")
+
+
+# 在模块加载时验证配置
+try:
+    validate_models_config()
+except ValueError as e:
+    print(f"⚠️ 图像模型配置验证失败: {str(e)}")
+    print(f"⚠️ 请检查 models_config.py 文件")
+
+try:
+    validate_flash_models_config()
+except ValueError as e:
+    print(f"⚠️ Flash 模型配置验证失败: {str(e)}")
+    print(f"⚠️ 请检查 models_config.py 文件")