def _get_jinja_prompts_env()

in databao/executors/lighthouse/utils.py [0:0]

• 11 lines of code
• 4 McCabe index (conditional complexity)

def _get_jinja_prompts_env(prompts_dir: Path | None = None) -> jinja2.Environment:
    if prompts_dir:
        return jinja2.Environment(loader=jinja2.FileSystemLoader(prompts_dir))

    global _jinja_prompts_env
    if _jinja_prompts_env is None:
        # A package loader must be used for using as a library!
        # Use empty string to load from package directory itself, not from 'templates' subdirectory
        _jinja_prompts_env = jinja2.Environment(
            loader=jinja2.PackageLoader("databao.executors.lighthouse", ""),
            trim_blocks=True,  # better whitespace handling
            lstrip_blocks=True,
        )
    return _jinja_prompts_env