def get_generator()

in evalbench/generators/models/__init__.py [0:0]

• 19 lines of code
• 7 McCabe index (conditional complexity)

def get_generator(global_models, model_config_path: str, db: DB = None):
    with global_models.get("lock"):
        global_model_configs = global_models.get("registered_models")
        if model_config_path in global_model_configs:
            return global_model_configs[model_config_path]

        config = load_yaml_config(model_config_path)
        # Create a new model_config
        model: QueryGenerator | None = None
        if config["generator"] == "gcp_vertex_gemini":
            model = GeminiGenerator(config)
        if config["generator"] == "gcp_vertex_claude":
            model = ClaudeGenerator(config)
        if config["generator"] == "noop":
            model = NOOPGenerator(config)
        if config["generator"] == "alloydb_ai_nl":
            model = AlloyDBGenerator(db, config)
        if not model:
            raise ValueError(f"Unknown Generator {config['generator']}")

        global_model_configs[model_config_path] = model
    return model