MCPcopy Create free account
hub / github.com/THESIS-AGENT/AIRouter / _get_actual_model_name

Method _get_actual_model_name

LoadBalancing.py:299–320  ·  view source on GitHub ↗

获取实际模型名称,处理多模态模型 Args: source_name (str): 源名称 model_name (str): 模型名称 Returns: str: 实际模型名称或None

(self, source_name, model_name)

Source from the content-addressed store, hash-verified

297 raise
298
299 def _get_actual_model_name(self, source_name, model_name):
300 """获取实际模型名称,处理多模态模型
301
302 Args:
303 source_name (str): 源名称
304 model_name (str): 模型名称
305
306 Returns:
307 str: 实际模型名称或None
308 """
309 # 获取基础模型名(去掉_mm后缀)
310 base_model_name = model_name[:-3] if model_name.endswith("_mm") else model_name
311
312 if not (source_name in self.source_mapping and base_model_name in self.source_mapping[source_name]):
313 return None
314
315 # 如果模型映射值为None且是多模态模型(以_mm结尾)
316 if self.source_mapping[source_name][base_model_name] is None and model_name.endswith("_mm"):
317 # 多模态模型映射为None表示不支持
318 return None
319
320 return self.source_mapping[source_name][base_model_name]
321
322 def get_config(self, model_name, mode, input_proportion: int, output_proportion: int):
323 """维护两套策略, 一套是以便宜为导向,

Callers 3

get_configMethod · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected