From 958e18153673fb295f3f04f8244d4adb6b2414d1 Mon Sep 17 00:00:00 2001 From: samelhousseini <108335186+samelhousseini@users.noreply.github.com> Date: Thu, 6 Mar 2025 13:20:27 +0400 Subject: [PATCH] Bug fix --- .../natural_language_to_SQL/src/steps/business_rules_step.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/templates/natural_language_to_SQL/src/steps/business_rules_step.py b/templates/natural_language_to_SQL/src/steps/business_rules_step.py index 99b420e..edaa970 100644 --- a/templates/natural_language_to_SQL/src/steps/business_rules_step.py +++ b/templates/natural_language_to_SQL/src/steps/business_rules_step.py @@ -14,7 +14,7 @@ ValidationResult ) from src.utils.chat_helpers import call_chat_completion_structured_outputs -from src.constants.data_model import markdown_rules +from src.constants.data_model import json_rules from src.constants.prompts import business_rules_prompt console = Console() @@ -24,7 +24,7 @@ class BusinessRulesStep(KernelProcessStep): async def _apply_business_rules(self, kernel: Kernel, data: BusinessRulesStepInput) -> ValidationResult: """Apply business rules to the generated SQL statement.""" - rules_with_query = markdown_rules.format(question=data.user_query) + rules_with_query = json_rules.format(question=data.user_query) prompt = business_rules_prompt.format( question=data.user_query,