20 lines
659 B
Python
20 lines
659 B
Python
import pytest
|
|
|
|
from review_agent.models import Conversation
|
|
from review_agent.skill_router import route_message_intent
|
|
|
|
|
|
pytestmark = pytest.mark.django_db
|
|
|
|
|
|
def test_fixed_keyword_routes_to_regulatory_info_package(django_user_model):
|
|
user = django_user_model.objects.create_user(username="owner", password="pass")
|
|
conversation = Conversation.objects.create(user=user, title="会话")
|
|
|
|
route = route_message_intent(conversation, "请根据说明书生成第1章监管信息")
|
|
|
|
assert route.action == "regulatory_info_package"
|
|
assert route.workflow_type == "regulatory_info_package"
|
|
assert route.starts_regulatory_info_package is True
|
|
|