fix: preserve original template structure from prompts.yaml

app.py

@@ -87,7 +87,7 @@ with open("prompts.yaml", 'r') as stream:
     
     logger.debug("Starting template validation...")
     try:
-        # Validate and update templates
+        # Validate templates
         for key, value in prompt_templates.items():
             if isinstance(value, dict):
                 for subkey, subvalue in value.items():
@@ -95,13 +95,14 @@ with open("prompts.yaml", 'r') as stream:
                         logger.debug(f"Validating template: {key}.{subkey}")
                         logger.debug(f"Template content: {subvalue[:200]}...")
                         
-                        # Validate and update template
+                        # Validate template
                         try:
                             # Create a template with at least one template node
                             template_str = subvalue
                             if '{{' not in template_str and '{%' not in template_str:
                                 template_str = "{{ task }}\n" + template_str
                             
+                            # Create a template with the task variable
                             template = Template(template_str, undefined=StrictUndefined)
                             rendered = template.render(
                                 tools=[],
@@ -111,8 +112,6 @@ with open("prompts.yaml", 'r') as stream:
                                 remaining_steps=1,
                                 answer_facts="test"
                             )
-                            # Update the template in the dictionary
-                            prompt_templates[key][subkey] = template_str
                             logger.debug(f"Template render test successful for {key}.{subkey}")
                         except Exception as e:
                             logger.error(f"Template render test failed for {key}.{subkey}: {str(e)}")
@@ -121,13 +120,14 @@ with open("prompts.yaml", 'r') as stream:
                 logger.debug(f"Validating template: {key}")
                 logger.debug(f"Template content: {value[:200]}...")
                 
-                # Validate and update template
+                # Validate template
                 try:
                     # Create a template with at least one template node
                     template_str = value
                     if '{{' not in template_str and '{%' not in template_str:
                         template_str = "{{ task }}\n" + template_str
                     
+                    # Create a template with the task variable
                     template = Template(template_str, undefined=StrictUndefined)
                     rendered = template.render(
                         tools=[],
@@ -137,8 +137,6 @@ with open("prompts.yaml", 'r') as stream:
                         remaining_steps=1,
                         answer_facts="test"
                     )
-                    # Update the template in the dictionary
-                    prompt_templates[key] = template_str
                     logger.debug(f"Template render test successful for {key}")
                 except Exception as e:
                     logger.error(f"Template render test failed for {key}: {str(e)}")
@@ -150,7 +148,7 @@ with open("prompts.yaml", 'r') as stream:
         logger.error(f"Error during template validation: {str(e)}")
         raise
 
-# Create the agent with the updated templates
+# Create the agent with the templates
 agent = CodeAgent(
     model=model,
     tools=[final_answer, DuckDuckGoSearchTool(), calculate_min_price, extract_price_from_snippet, get_current_time_in_timezone],