theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-04-28 02:01:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Templating: Add loopcontrols extension

Browse Source 
Inbuilt jinja extension to allow for break and continue in loops.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2024-09-08 12:21:42 -04:00
parent dffceab777
commit df11890851
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
common/templating.py
View File

@@ -5,6 +5,7 @@ import pathlib
from importlib.metadata import version as package_version from importlib.metadata import version as package_version
from typing import List, Optional from typing import List, Optional
from jinja2 import Template, TemplateError from jinja2 import Template, TemplateError
from jinja2.ext import loopcontrols
from jinja2.sandbox import ImmutableSandboxedEnvironment from jinja2.sandbox import ImmutableSandboxedEnvironment
from loguru import logger from loguru import logger
from packaging import version from packaging import version
@@ -32,7 +33,10 @@ class PromptTemplate:
raw_template: str raw_template: str
template: Template template: Template
environment: ImmutableSandboxedEnvironment = ImmutableSandboxedEnvironment( environment: ImmutableSandboxedEnvironment = ImmutableSandboxedEnvironment(
trim_blocks=True, lstrip_blocks=True, enable_async=True trim_blocks=True,
lstrip_blocks=True,
enable_async=True,
extensions=[loopcontrols],
) )
metadata: Optional[TemplateMetadata] = None metadata: Optional[TemplateMetadata] = None