fix: update templates in dictionary after validation

app.py

@@ -87,7 +87,7 @@ with open("prompts.yaml", 'r') as stream:
     
     logger.debug("Starting template validation...")
     try:
-        # Validate templates
+        # Validate and update templates
         for key, value in prompt_templates.items():
             if isinstance(value, dict):
                 for subkey, subvalue in value.items():
@@ -95,7 +95,7 @@ with open("prompts.yaml", 'r') as stream:
                         logger.debug(f"Validating template: {key}.{subkey}")
                         logger.debug(f"Template content: {subvalue[:200]}...")
                         
-                        # Validate template
+                        # Validate and update template
                         try:
                             # Create a template with at least one template node
                             template_str = subvalue
@@ -111,6 +111,8 @@ 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)}")
@@ -119,7 +121,7 @@ with open("prompts.yaml", 'r') as stream:
                 logger.debug(f"Validating template: {key}")
                 logger.debug(f"Template content: {value[:200]}...")
                 
-                # Validate template
+                # Validate and update template
                 try:
                     # Create a template with at least one template node
                     template_str = value
@@ -135,6 +137,8 @@ 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)}")
@@ -146,7 +150,7 @@ with open("prompts.yaml", 'r') as stream:
         logger.error(f"Error during template validation: {str(e)}")
         raise
 
-# Create the agent with the templates
+# Create the agent with the updated templates
 agent = CodeAgent(
     model=model,
     tools=[final_answer, DuckDuckGoSearchTool(), calculate_min_price, extract_price_from_snippet, get_current_time_in_timezone],