add separators and sort_keys plus import fix

CISC · web-flow · commit ea52ae8ed42c · 2024-10-08T20:04:27.000+02:00
diff --git a/llama_cpp/llama_chat_format.py b/llama_cpp/llama_chat_format.py
@@ -24,6 +24,7 @@
 )
 
 import jinja2
+import jinja2.ext as jinja2_ext
 from jinja2.sandbox import ImmutableSandboxedEnvironment
 
 import numpy as np
@@ -211,9 +212,9 @@ def __init__(
         environment = ImmutableSandboxedEnvironment(
             trim_blocks=True,
             lstrip_blocks=True,
-            extensions=[jinja2.ext.loopcontrols],
+            extensions=[jinja2_ext.loopcontrols],
         )
-        environment.filters["tojson"] = lambda x, indent=None: json.dumps(x, indent=indent, ensure_ascii=False)
+        environment.filters["tojson"] = lambda x, indent=None, separators=None, sort_keys=False: json.dumps(x, indent=indent, separators=separators, sort_keys=sort_keys, ensure_ascii=False)
         environment.globals["strftime_now"] = lambda format: datetime.now().strftime(format)
         self._environment = environment.from_string(self.template)
 
