mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 23:58:34 +00:00
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
import re
|
|
|
|
import pytest
|
|
|
|
from crewai.agent import Agent
|
|
from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource
|
|
|
|
|
|
def test_agent_with_chinese_role_name():
|
|
"""Test that an agent with a Chinese role name works correctly."""
|
|
# Create a knowledge source with some content
|
|
content = "This is some test content."
|
|
string_source = StringKnowledgeSource(content=content)
|
|
|
|
# Create an agent with a Chinese role name
|
|
agent = Agent(
|
|
role="中文角色", # Chinese role name
|
|
goal="Test Chinese character support",
|
|
backstory="Testing Chinese character support in agent role names.",
|
|
knowledge_sources=[string_source],
|
|
)
|
|
|
|
# Test that the regex pattern in agent.py correctly preserves Chinese characters
|
|
full_pattern = re.compile(r"[^a-zA-Z0-9\u4e00-\u9fa5\-_\r\n]|(\.\.)")
|
|
knowledge_agent_name = f"{re.sub(full_pattern, '_', agent.role)}"
|
|
|
|
# Verify that the agent was created successfully
|
|
assert agent.role == "中文角色"
|
|
|
|
# Verify that the Chinese characters are preserved in the knowledge_agent_name
|
|
assert knowledge_agent_name == "中文角色"
|