env_overrides.cpp

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#include <entropic/config/loader.h>
#include <entropic/types/logging.h>
#include <cstdlib>
 
static auto s_log = entropic::log::get("config");
 
namespace entropic::config {
 
static std::string get_env(const char* name)
{
    const char* val = std::getenv(name);
    return val != nullptr ? std::string(val) : "";
}
 
static bool parse_bool(const std::string& val)
{
    return val == "true" || val == "True" || val == "TRUE"
           || val == "yes" || val == "1";
}
 
template <typename Setter>
static void apply_env(const char* key, Setter setter) {
    auto val = get_env(key);
    if (val.empty()) { return; }
    s_log->info("Env override: {}={}", key, val);
    setter(val);
}
 
void apply_env_overrides(ParsedConfig& config)
{
    apply_env("ENTROPIC_LOG_LEVEL",
              [&](const std::string& v) { config.log_level = v; });
    apply_env("ENTROPIC_MODELS__DEFAULT",
              [&](const std::string& v) { config.models.default_tier = v; });
    apply_env("ENTROPIC_ROUTING__ENABLED",
              [&](const std::string& v) {
                  config.routing.enabled = parse_bool(v);
              });
    apply_env("ENTROPIC_ROUTING__FALLBACK_TIER",
              [&](const std::string& v) { config.routing.fallback_tier = v; });
    apply_env("ENTROPIC_COMPACTION__THRESHOLD_PERCENT",
              [&](const std::string& v) {
                  config.compaction.threshold_percent = std::stof(v);
              });
    apply_env("ENTROPIC_COMPACTION__ENABLED",
              [&](const std::string& v) {
                  config.compaction.enabled = parse_bool(v);
              });
    apply_env("ENTROPIC_VRAM_RESERVE_MB",
              [&](const std::string& v) {
                  config.vram_reserve_mb = std::stoi(v);
              });
    apply_env("ENTROPIC_CONFIG_DIR",
              [&](const std::string& v) {
                  config.config_dir = std::filesystem::path(v);
              });
}
 
} // namespace entropic::config