如何在训练与测试运行时为dbt模型生成不同的SQL语句？

刚好之前做过类似的dbt场景，给你两种靠谱的实现方式，都能轻松实现根据训练/测试阶段切换SQL逻辑：

这种方式可以通过命令行或者配置文件灵活指定阶段，完全自定义逻辑。

首先在你的dbt_project.yml里定义好各阶段的配置和默认阶段：

vars:
  # 默认使用train阶段，可通过命令行覆盖
  current_stage: "train"
  # 把你提供的JSON配置转成dbt变量格式
  stage_settings:
    train:
      from_inclusive: "15 days"
    train-extra:
      from_inclusive: "1 days"
    eval-test:
      from_inclusive: "1 days"

然后在你的dbt模型文件里，用Jinja语法读取对应阶段的参数：

{%- set current_config = var('stage_settings')[var('current_stage')] -%}

SELECT * 
FROM tracks tr 
WHERE event_timestamp >= (CURRENT_DATE - INTERVAL '{{ current_config.from_inclusive }}') - INTERVAL '60 days'

怎么切换阶段？
运行dbt的时候，通过--vars参数指定即可：

# 用eval-test阶段运行
dbt run --vars '{current_stage: "eval-test"}'

# 用train-extra阶段运行
dbt run --vars '{current_stage: "train-extra"}'

如果你的训练/测试阶段和dbt的运行环境（比如dev、prod）强绑定，比如dev环境默认跑测试逻辑，prod环境默认跑训练逻辑，可以把阶段配置和环境绑定，不用每次手动传变量。

修改dbt_project.yml，给不同环境设置默认阶段：

vars:
  stage_settings:
    train:
      from_inclusive: "15 days"
    train-extra:
      from_inclusive: "1 days"
    eval-test:
      from_inclusive: "1 days"

# 给不同target环境设置默认阶段
environments:
  dev:
    vars:
      current_stage: "eval-test"
  prod:
    vars:
      current_stage: "train"

这样运行的时候，只要指定target，就自动用对应阶段的配置：

# dev环境自动用eval-test逻辑
dbt run --target dev

# prod环境自动用train逻辑
dbt run --target prod

如果担心传了不存在的阶段名称导致模型报错，可以加个默认 fallback 逻辑，确保模型能正常编译：

{%- set current_stage = var('current_stage', 'train') -%}
{%- set stage_settings = var('stage_settings') -%}
{%- set current_config = stage_settings.get(current_stage, stage_settings['train']) -%}

SELECT * 
FROM tracks tr 
WHERE event_timestamp >= (CURRENT_DATE - INTERVAL '{{ current_config.from_inclusive }}') - INTERVAL '60 days'

这样如果传入了未定义的阶段，会自动 fallback 到train阶段的配置，避免编译失败。

内容的提问来源于stack exchange，提问作者Leo Folsom