🔧 fix template vars with jinja2

shroominic · May 17, 2024 · c7504e4 · c7504e4
1 parent 2b5fec2
commit c7504e4
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/funcchain/backend/prompt.py b/src/funcchain/backend/prompt.py
@@ -30,7 +30,14 @@ def create_instruction_prompt(
 
     _filter_fstring_vars(input_kwargs)
 
-    inject_vars = [f"{var.upper()}:\n{{{var}}}\n" for var, _ in input_kwargs.items() if var not in required_f_str_vars]
+    if template_format == "jinja2":
+        inject_vars = [
+            f"{var.upper()}:\n{{{{{var}}}}}\n" for var, _ in input_kwargs.items() if var not in required_f_str_vars
+        ]
+    else:
+        inject_vars = [
+            f"{var.upper()}:\n{{{var}}}\n" for var, _ in input_kwargs.items() if var not in required_f_str_vars
+        ]
 
     added_instruction = "\n".join(inject_vars)
     instruction = added_instruction + instruction