mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-22 06:18:14 +00:00
d9b29daa27
CrewAI initializes several I18N instances deep within the library at package initialization time, using the default prompts. This makes overriding the defaults consistently throughout the package very difficult, requiring monkeypatching. This change will allow overriding the default prompt file location in the $CREWAI_PROMPT_FILE environment variable, allowing consistency throughout the library.
66 lines
1.6 KiB
Python
66 lines
1.6 KiB
Python
import pytest
|
|
|
|
from crewai.utilities.i18n import I18N
|
|
|
|
|
|
def test_load_prompts():
|
|
i18n = I18N()
|
|
i18n.load_prompts()
|
|
assert i18n._prompts is not None
|
|
|
|
|
|
def test_slice():
|
|
i18n = I18N()
|
|
i18n.load_prompts()
|
|
assert isinstance(i18n.slice("role_playing"), str)
|
|
|
|
|
|
def test_tools():
|
|
i18n = I18N()
|
|
i18n.load_prompts()
|
|
assert isinstance(i18n.tools("ask_question"), str)
|
|
|
|
|
|
def test_retrieve():
|
|
i18n = I18N()
|
|
i18n.load_prompts()
|
|
assert isinstance(i18n.retrieve("slices", "role_playing"), str)
|
|
|
|
|
|
def test_retrieve_not_found():
|
|
i18n = I18N()
|
|
i18n.load_prompts()
|
|
with pytest.raises(Exception):
|
|
i18n.retrieve("nonexistent_kind", "nonexistent_key")
|
|
|
|
|
|
def test_prompt_file():
|
|
import os
|
|
|
|
path = os.path.join(os.path.dirname(__file__), "prompts.json")
|
|
i18n = I18N(prompt_file=path)
|
|
i18n.load_prompts()
|
|
assert isinstance(i18n.retrieve("slices", "role_playing"), str)
|
|
assert i18n.retrieve("slices", "role_playing") == "Lorem ipsum dolor sit amet"
|
|
|
|
|
|
def test_prompt_file_env():
|
|
import os
|
|
|
|
path = os.path.join(os.path.dirname(__file__), "prompts.json")
|
|
old_env = os.environ.get("CREWAI_PROMPT_FILE")
|
|
try:
|
|
os.environ["CREWAI_PROMPT_FILE"] = path
|
|
i18n = I18N()
|
|
i18n.load_prompts()
|
|
assert i18n.retrieve("slices", "role_playing") == "Lorem ipsum dolor sit amet"
|
|
finally:
|
|
if old_env:
|
|
os.environ["CREWAI_PROMPT_FILE"] = old_env
|
|
else:
|
|
del os.environ["CREWAI_PROMPT_FILE"]
|
|
|
|
i18n = I18N()
|
|
i18n.load_prompts()
|
|
assert i18n.retrieve("slices", "role_playing") != "Lorem ipsum dolor sit amet"
|