diff --git a/src/crewai/utilities/jinja_templating.py b/src/crewai/utilities/jinja_templating.py index 0cde9d79d..2c75e800e 100644 --- a/src/crewai/utilities/jinja_templating.py +++ b/src/crewai/utilities/jinja_templating.py @@ -4,6 +4,7 @@ from typing import Any, Dict, List, Optional, Union import jinja2 + def to_jinja_template(input_string: str) -> str: """ Convert CrewAI-style {var} templates to Jinja2-style {{var}} templates. diff --git a/src/crewai/utilities/string_utils.py b/src/crewai/utilities/string_utils.py index 3db166d2a..114f7dcf8 100644 --- a/src/crewai/utilities/string_utils.py +++ b/src/crewai/utilities/string_utils.py @@ -4,6 +4,7 @@ from typing import Any, Dict, List, Optional, Union from crewai.utilities.jinja_templating import render_template + def interpolate_only( input_string: Optional[str], inputs: Dict[str, Any], diff --git a/tests/test_templating.py b/tests/test_templating.py index 768df55a0..9c589f0c4 100644 --- a/tests/test_templating.py +++ b/tests/test_templating.py @@ -1,10 +1,12 @@ import datetime -import pytest from typing import Dict, List +import pytest + from crewai.agent import Agent from crewai.task import Task + class TestTemplating: def test_task_interpolation(self): task = Task( diff --git a/tests/utilities/test_jinja_templating.py b/tests/utilities/test_jinja_templating.py index b689cbdba..8993d06fc 100644 --- a/tests/utilities/test_jinja_templating.py +++ b/tests/utilities/test_jinja_templating.py @@ -1,11 +1,12 @@ import datetime -import pytest from typing import Any, Dict, List +import pytest from pydantic import BaseModel from crewai.utilities.jinja_templating import render_template, to_jinja_template + class Person(BaseModel): name: str age: int