Function load_config

config.py:451–470

Load configuration from a YAML file, or create a default config.

(config_path: Optional[Union[str, Path]] = None)

def load_config(config_path: Optional[Union[str, Path]] = None) -> Config:
    """Load configuration from a YAML file, or create a default config."""
    if config_path and os.path.exists(config_path):
        # If a path is provided and the file exists, load from YAML
        config = Config.from_yaml(config_path)
    else:
        # Otherwise, create a default Config object
        config = Config()

        # Try to populate API key and base URL from environment variables as a fallback
        api_key = os.environ.get("OPENAI_API_KEY")
        api_base = os.environ.get("OPENAI_API_BASE", "https://api.openai.com/v1")

        config.llm.update_model_params({"api_key": api_key, "api_base": api_base})

    # Ensure the system message from the prompt config is available to individual models,
    # in case it's not provided directly by the prompt sampler during a call.
    config.llm.update_model_params({"system_message": config.prompt.system_message})

    return config