github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-02 01:23:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Import jinja2 only when used (#3123)

Browse Source 
Addressing #3113
This commit is contained in:
engkheng 2023-04-19 12:23:03 +08:00 committed by GitHub
parent 90ef705ced
commit 8e386613ac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
langchain/prompts/base.py
View File

@ -18,7 +18,7 @@ def jinja2_formatter(template: str, **kwargs: Any) -> str:
try:
from jinja2 import Template
except ImportError:
raise ValueError(
raise ImportError(
"jinja2 not installed, which is needed to use the jinja2_formatter. "
"Please install it with `pip install jinja2`."
)

8
langchain/prompts/prompt.py
View File

@ -5,7 +5,6 @@ from pathlib import Path
from string import Formatter
from typing import Any, Dict, List, Set, Union
from jinja2 import Environment, meta
from pydantic import Extra, root_validator
from langchain.prompts.base import (
@ -16,6 +15,13 @@ from langchain.prompts.base import (
def _get_jinja2_variables_from_template(template: str) -> Set[str]:
try:
from jinja2 import Environment, meta
except ImportError:
raise ImportError(
"jinja2 not installed, which is needed to use the jinja2_formatter. "
"Please install it with `pip install jinja2`."
)
env = Environment()
ast = env.parse(template)
variables = meta.find_undeclared_variables(ast)