mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 23:58:34 +00:00
9f0f668d12
- Replace pytest.raises match parameter with exc_info and string assertion - Avoids potential regex matching differences between Python versions - Should resolve Python 3.10 CI test failure while maintaining test effectiveness Co-Authored-By: João <joao@crewai.com>
134 lines
4.1 KiB
Python
134 lines
4.1 KiB
Python
import json
|
|
import os
|
|
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_global_i18n_file_environment_variable(monkeypatch):
|
|
"""Test that CREWAI_I18N_FILE environment variable is respected"""
|
|
test_translations = {
|
|
"slices": {"role_playing": "Test role playing message"},
|
|
"tools": {"ask_question": "Test ask question message"}
|
|
}
|
|
|
|
test_file_path = os.path.join(os.path.dirname(__file__), "test_env_prompts.json")
|
|
with open(test_file_path, "w", encoding="utf-8") as f:
|
|
json.dump(test_translations, f)
|
|
|
|
try:
|
|
monkeypatch.setenv("CREWAI_I18N_FILE", test_file_path)
|
|
|
|
i18n = I18N()
|
|
i18n.load_prompts()
|
|
|
|
assert i18n.slice("role_playing") == "Test role playing message"
|
|
assert i18n.tools("ask_question") == "Test ask question message"
|
|
|
|
finally:
|
|
if os.path.exists(test_file_path):
|
|
os.remove(test_file_path)
|
|
|
|
|
|
def test_prompt_file_priority_over_environment_variable(monkeypatch):
|
|
"""Test that explicit prompt_file takes priority over environment variable"""
|
|
monkeypatch.setenv("CREWAI_I18N_FILE", "/nonexistent/path.json")
|
|
|
|
path = os.path.join(os.path.dirname(__file__), "prompts.json")
|
|
i18n = I18N(prompt_file=path)
|
|
i18n.load_prompts()
|
|
|
|
assert i18n.retrieve("slices", "role_playing") == "Lorem ipsum dolor sit amet"
|
|
|
|
|
|
def test_environment_variable_file_not_found(monkeypatch):
|
|
"""Test proper error handling when environment variable points to non-existent file"""
|
|
monkeypatch.setenv("CREWAI_I18N_FILE", "/nonexistent/file.json")
|
|
|
|
with pytest.raises(Exception) as exc_info:
|
|
I18N()
|
|
|
|
assert "Prompt file '/nonexistent/file.json' not found" in str(exc_info.value)
|
|
|
|
|
|
def test_fallback_to_default_when_no_environment_variable(monkeypatch):
|
|
"""Test that it falls back to default en.json when no environment variable is set"""
|
|
monkeypatch.delenv("CREWAI_I18N_FILE", raising=False)
|
|
|
|
i18n = I18N()
|
|
i18n.load_prompts()
|
|
|
|
assert isinstance(i18n.slice("role_playing"), str)
|
|
assert len(i18n.slice("role_playing")) > 0
|
|
|
|
|
|
def test_chinese_translation_file():
|
|
"""Test loading Chinese translation file"""
|
|
import os
|
|
|
|
zh_path = os.path.join(os.path.dirname(__file__), "../../src/crewai/translations/zh.json")
|
|
zh_path = os.path.abspath(zh_path)
|
|
|
|
i18n = I18N(prompt_file=zh_path)
|
|
i18n.load_prompts()
|
|
|
|
assert i18n.retrieve("hierarchical_manager_agent", "role") == "团队经理"
|
|
assert i18n.slice("observation") == "\n观察:"
|
|
assert i18n.errors("tool_usage_error") == "我遇到了错误: {error}"
|
|
|
|
|
|
def test_spanish_translation_file():
|
|
"""Test loading Spanish translation file"""
|
|
import os
|
|
|
|
es_path = os.path.join(os.path.dirname(__file__), "../../src/crewai/translations/es.json")
|
|
es_path = os.path.abspath(es_path)
|
|
|
|
i18n = I18N(prompt_file=es_path)
|
|
i18n.load_prompts()
|
|
|
|
assert i18n.retrieve("hierarchical_manager_agent", "role") == "Gerente del Equipo"
|
|
assert i18n.slice("observation") == "\nObservación:"
|
|
assert i18n.errors("tool_usage_error") == "Encontré un error: {error}"
|