调整对应的调用

This commit is contained in:
UnCLAS-Prommer
2025-07-30 17:07:55 +08:00
parent 3c40ceda4c
commit 6c0edd0ad7
40 changed files with 580 additions and 1236 deletions

View File

@@ -60,268 +60,6 @@ TEMPLATE_DIR = os.path.join(PROJECT_ROOT, "template")
MMC_VERSION = "0.10.0-snapshot.2"
# def _get_config_version(toml: Dict) -> Version:
# """提取配置文件的 SpecifierSet 版本数据
# Args:
# toml[dict]: 输入的配置文件字典
# Returns:
# Version
# """
# if "inner" in toml and "version" in toml["inner"]:
# config_version: str = toml["inner"]["version"]
# else:
# raise InvalidVersion("配置文件缺少版本信息,请检查配置文件。")
# try:
# return version.parse(config_version)
# except InvalidVersion as e:
# logger.error(
# "配置文件中 inner段 的 version 键是错误的版本描述\n"
# f"请检查配置文件,当前 version 键: {config_version}\n"
# f"错误信息: {e}"
# )
# raise e
# def _request_conf(parent: Dict, config: ModuleConfig):
# request_conf_config = parent.get("request_conf")
# config.req_conf.max_retry = request_conf_config.get(
# "max_retry", config.req_conf.max_retry
# )
# config.req_conf.timeout = request_conf_config.get(
# "timeout", config.req_conf.timeout
# )
# config.req_conf.retry_interval = request_conf_config.get(
# "retry_interval", config.req_conf.retry_interval
# )
# config.req_conf.default_temperature = request_conf_config.get(
# "default_temperature", config.req_conf.default_temperature
# )
# config.req_conf.default_max_tokens = request_conf_config.get(
# "default_max_tokens", config.req_conf.default_max_tokens
# )
# def _api_providers(parent: Dict, config: ModuleConfig):
# api_providers_config = parent.get("api_providers")
# for provider in api_providers_config:
# name = provider.get("name", None)
# base_url = provider.get("base_url", None)
# api_key = provider.get("api_key", None)
# api_keys = provider.get("api_keys", []) # 新增支持多个API Key
# client_type = provider.get("client_type", "openai")
# if name in config.api_providers: # 查重
# logger.error(f"重复的API提供商名称: {name},请检查配置文件。")
# raise KeyError(f"重复的API提供商名称: {name},请检查配置文件。")
# if name and base_url:
# # 处理API Key配置支持单个api_key或多个api_keys
# if api_keys:
# # 使用新格式api_keys列表
# logger.debug(f"API提供商 '{name}' 配置了 {len(api_keys)} 个API Key")
# elif api_key:
# # 向后兼容使用单个api_key
# api_keys = [api_key]
# logger.debug(f"API提供商 '{name}' 使用单个API Key向后兼容模式")
# else:
# logger.warning(f"API提供商 '{name}' 没有配置API Key某些功能可能不可用")
# config.api_providers[name] = APIProvider(
# name=name,
# base_url=base_url,
# api_key=api_key, # 保留向后兼容
# api_keys=api_keys, # 新格式
# client_type=client_type,
# )
# else:
# logger.error(f"API提供商 '{name}' 的配置不完整,请检查配置文件。")
# raise ValueError(f"API提供商 '{name}' 的配置不完整,请检查配置文件。")
# def _models(parent: Dict, config: ModuleConfig):
# models_config = parent.get("models")
# for model in models_config:
# model_identifier = model.get("model_identifier", None)
# name = model.get("name", model_identifier)
# api_provider = model.get("api_provider", None)
# price_in = model.get("price_in", 0.0)
# price_out = model.get("price_out", 0.0)
# force_stream_mode = model.get("force_stream_mode", False)
# task_type = model.get("task_type", "")
# capabilities = model.get("capabilities", [])
# if name in config.models: # 查重
# logger.error(f"重复的模型名称: {name},请检查配置文件。")
# raise KeyError(f"重复的模型名称: {name},请检查配置文件。")
# if model_identifier and api_provider:
# # 检查API提供商是否存在
# if api_provider not in config.api_providers:
# logger.error(f"未声明的API提供商 '{api_provider}' ,请检查配置文件。")
# raise ValueError(
# f"未声明的API提供商 '{api_provider}' ,请检查配置文件。"
# )
# config.models[name] = ModelInfo(
# name=name,
# model_identifier=model_identifier,
# api_provider=api_provider,
# price_in=price_in,
# price_out=price_out,
# force_stream_mode=force_stream_mode,
# task_type=task_type,
# capabilities=capabilities,
# )
# else:
# logger.error(f"模型 '{name}' 的配置不完整,请检查配置文件。")
# raise ValueError(f"模型 '{name}' 的配置不完整,请检查配置文件。")
# def _task_model_usage(parent: Dict, config: ModuleConfig):
# model_usage_configs = parent.get("task_model_usage")
# config.task_model_arg_map = {}
# for task_name, item in model_usage_configs.items():
# if task_name in config.task_model_arg_map:
# logger.error(f"子任务 {task_name} 已存在,请检查配置文件。")
# raise KeyError(f"子任务 {task_name} 已存在,请检查配置文件。")
# usage = []
# if isinstance(item, Dict):
# if "model" in item:
# usage.append(
# ModelUsageArgConfigItem(
# name=item["model"],
# temperature=item.get("temperature", None),
# max_tokens=item.get("max_tokens", None),
# max_retry=item.get("max_retry", None),
# )
# )
# else:
# logger.error(f"子任务 {task_name} 的模型配置不合法,请检查配置文件。")
# raise ValueError(
# f"子任务 {task_name} 的模型配置不合法,请检查配置文件。"
# )
# elif isinstance(item, List):
# for model in item:
# if isinstance(model, Dict):
# usage.append(
# ModelUsageArgConfigItem(
# name=model["model"],
# temperature=model.get("temperature", None),
# max_tokens=model.get("max_tokens", None),
# max_retry=model.get("max_retry", None),
# )
# )
# elif isinstance(model, str):
# usage.append(
# ModelUsageArgConfigItem(
# name=model,
# temperature=None,
# max_tokens=None,
# max_retry=None,
# )
# )
# else:
# logger.error(
# f"子任务 {task_name} 的模型配置不合法,请检查配置文件。"
# )
# raise ValueError(
# f"子任务 {task_name} 的模型配置不合法,请检查配置文件。"
# )
# elif isinstance(item, str):
# usage.append(
# ModelUsageArgConfigItem(
# name=item,
# temperature=None,
# max_tokens=None,
# max_retry=None,
# )
# )
# config.task_model_arg_map[task_name] = ModelUsageArgConfig(
# name=task_name,
# usage=usage,
# )
# def api_ada_load_config(config_path: str) -> ModuleConfig:
# """从TOML配置文件加载配置"""
# config = ModuleConfig()
# include_configs: Dict[str, Dict[str, Any]] = {
# "request_conf": {
# "func": _request_conf,
# "support": ">=0.0.0",
# "necessary": False,
# },
# "api_providers": {"func": _api_providers, "support": ">=0.0.0"},
# "models": {"func": _models, "support": ">=0.0.0"},
# "task_model_usage": {"func": _task_model_usage, "support": ">=0.0.0"},
# }
# if os.path.exists(config_path):
# with open(config_path, "rb") as f:
# try:
# toml_dict = tomlkit.load(f)
# except tomlkit.TOMLDecodeError as e:
# logger.critical(
# f"配置文件model_list.toml填写有误请检查第{e.lineno}行第{e.colno}处:{e.msg}"
# )
# exit(1)
# # 获取配置文件版本
# config.INNER_VERSION = _get_config_version(toml_dict)
# # 检查版本
# if config.INNER_VERSION > Version(NEWEST_VER):
# logger.warning(
# f"当前配置文件版本 {config.INNER_VERSION} 高于支持的最新版本 {NEWEST_VER},可能导致异常,建议更新依赖。"
# )
# # 解析配置文件
# # 如果在配置中找到了需要的项,调用对应项的闭包函数处理
# for key in include_configs:
# if key in toml_dict:
# group_specifier_set: SpecifierSet = SpecifierSet(
# include_configs[key]["support"]
# )
# # 检查配置文件版本是否在支持范围内
# if config.INNER_VERSION in group_specifier_set:
# # 如果版本在支持范围内,检查是否存在通知
# if "notice" in include_configs[key]:
# logger.warning(include_configs[key]["notice"])
# # 调用闭包函数处理配置
# (include_configs[key]["func"])(toml_dict, config)
# else:
# # 如果版本不在支持范围内,崩溃并提示用户
# logger.error(
# f"配置文件中的 '{key}' 字段的版本 ({config.INNER_VERSION}) 不在支持范围内。\n"
# f"当前程序仅支持以下版本范围: {group_specifier_set}"
# )
# raise InvalidVersion(
# f"当前程序仅支持以下版本范围: {group_specifier_set}"
# )
# # 如果 necessary 项目存在,而且显式声明是 False进入特殊处理
# elif (
# "necessary" in include_configs[key]
# and include_configs[key].get("necessary") is False
# ):
# # 通过 pass 处理的项虽然直接忽略也是可以的,但是为了不增加理解困难,依然需要在这里显式处理
# if key == "keywords_reaction":
# pass
# else:
# # 如果用户根本没有需要的配置项,提示缺少配置
# logger.error(f"配置文件中缺少必需的字段: '{key}'")
# raise KeyError(f"配置文件中缺少必需的字段: '{key}'")
# logger.info(f"成功加载配置文件: {config_path}")
# return config
def get_key_comment(toml_table, key):
# 获取key的注释如果有
if hasattr(toml_table, "trivia") and hasattr(toml_table.trivia, "comment"):
@@ -626,9 +364,19 @@ class APIAdapterConfig(ConfigBase):
"""API提供商列表"""
def __post_init__(self):
# 检查API提供商名称是否重复
provider_names = [provider.name for provider in self.api_providers]
if len(provider_names) != len(set(provider_names)):
raise ValueError("API提供商名称存在重复请检查配置文件。")
# 检查模型名称是否重复
model_names = [model.name for model in self.models]
if len(model_names) != len(set(model_names)):
raise ValueError("模型名称存在重复,请检查配置文件。")
self.api_providers_dict = {provider.name: provider for provider in self.api_providers}
self.models_dict = {model.name: model for model in self.models}
def get_model_info(self, model_name: str) -> ModelInfo:
"""根据模型名称获取模型信息"""
if not model_name:
@@ -636,7 +384,7 @@ class APIAdapterConfig(ConfigBase):
if model_name not in self.models_dict:
raise KeyError(f"模型 '{model_name}' 不存在")
return self.models_dict[model_name]
def get_provider(self, provider_name: str) -> APIProvider:
"""根据提供商名称获取API提供商信息"""
if not provider_name: