Method get_sources_from_ranking

LoadBalancing.py:242–297

基于预设排名获取源和模型配置 当健康检查数据为空时，基于预设的源排名获取配置。 Args: model_name (str): 模型名称 Returns: tuple: 包含主源和备用源的配置信息 Raises: ValueError: 如果找不到可用的源

(self, model_name)

            return True  # 出错时假设数据为空，使用预设排名
    
    def get_sources_from_ranking(self, model_name):
        """基于预设排名获取源和模型配置
        
        当健康检查数据为空时，基于预设的源排名获取配置。
        
        Args:
            model_name (str): 模型名称
            
        Returns:
            tuple: 包含主源和备用源的配置信息
            
        Raises:
            ValueError: 如果找不到可用的源
        """
        try:
            # 检查是否为屏蔽模型
            if is_model_health_check_blacklisted(model_name):
                self.logger.info(f"🚫 模型 {model_name} 在健康检测屏蔽清单中，基于预设排名选择源")
            else:
                self.logger.info(f"基于预设排名为模型 {model_name} 选择源")
            
            # 根据source_ranking选择排名靠前的源
            sorted_sources = sorted(self.source_ranking.keys(), key=lambda s: self.source_ranking[s])
            
            # 找出可用的源（即映射中有此模型的源）
            available_sources = []
            for source in sorted_sources:
                if self._check_valid_model(source, model_name):
                    available_sources.append(source)

            if not available_sources:
                error_msg = f"在预设排名中找不到模型 {model_name} 的可用源"
                self.logger.error(error_msg)
                raise ValueError(error_msg)
            
            # 选择主源和备用源
            main_source = available_sources[0]
            backup_source = available_sources[1] if len(available_sources) > 1 else main_source
            
            # 获取模型名称和API密钥
            main_source_model_name = self._get_actual_model_name(main_source, model_name)
            backup_source_model_name = self._get_actual_model_name(backup_source, model_name)
            
            self.logger.info(f"选择主源 {main_source} 和备用源 {backup_source} 为模型 {model_name}")
            
            try:
                main_api_key = Harness_localAPI.get_api_key(main_source)
                backup_api_key = Harness_localAPI.get_api_key(backup_source)
            except Exception as e:
                self.logger.error(f"获取API密钥时出错: {str(e)}")
                raise ValueError(f"无法获取API密钥: {str(e)}")
            
            return main_source, main_source_model_name, main_api_key, backup_source, backup_source_model_name, backup_api_key
        except Exception as e:
            self.logger.error(f"基于预设排名选择源时出错: {str(e)}")
            raise

    def _get_actual_model_name(self, source_name, model_name):