Add default value for llm_model_name parameter

oravi · oravi · commit eb207be33b90 · 2025-03-12T12:17:50.000+02:00
Updated `test_ai_data_validation` and `test_unstructured_data_validation` macros to set a default value of `none` for the `llm_model_name` parameter, providing more flexibility in AI data validation tests.
diff --git a/macros/edr/tests/test_ai_data_validation.sql b/macros/edr/tests/test_ai_data_validation.sql
@@ -1,4 +1,4 @@
-{% test ai_data_validation(model, column_name, expectation_prompt, llm_model_name, context='') %}
+{% test ai_data_validation(model, column_name, expectation_prompt, llm_model_name=none, context='') %}
     {{ config(tags = ['elementary-tests']) }}
     {%- if execute and elementary.is_test_command() and elementary.is_elementary_enabled() %}
        {% set model_relation = elementary.get_model_relation_for_test(model, context["model"]) %}
diff --git a/macros/edr/tests/test_unstructured_data_validation.sql b/macros/edr/tests/test_unstructured_data_validation.sql
@@ -1,4 +1,4 @@
-{% test unstructured_data_validation(model, column_name, expectation_prompt, llm_model_name) %}
+{% test unstructured_data_validation(model, column_name, expectation_prompt, llm_model_name=none) %}
     {{ config(tags = ['elementary-tests']) }}
     {% set context = "You are a data validator specializing in validating unstructured data." %}
     {{ return(elementary.test_ai_data_validation(model, column_name, expectation_prompt, llm_model_name, context)) }}

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-{% test ai_data_validation(model, column_name, expectation_prompt, llm_model_name, context='') %}`
	`1`	`+{% test ai_data_validation(model, column_name, expectation_prompt, llm_model_name=none, context='') %}`
`2`	`2`	`{{ config(tags = ['elementary-tests']) }}`
`3`	`3`	`{%- if execute and elementary.is_test_command() and elementary.is_elementary_enabled() %}`
`4`	`4`	`{% set model_relation = elementary.get_model_relation_for_test(model, context["model"]) %}`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-{% test unstructured_data_validation(model, column_name, expectation_prompt, llm_model_name) %}`
	`1`	`+{% test unstructured_data_validation(model, column_name, expectation_prompt, llm_model_name=none) %}`
`2`	`2`	`{{ config(tags = ['elementary-tests']) }}`
`3`	`3`	`{% set context = "You are a data validator specializing in validating unstructured data." %}`
`4`	`4`	`{{ return(elementary.test_ai_data_validation(model, column_name, expectation_prompt, llm_model_name, context)) }}`