119 lines
4.1 KiB
Python
119 lines
4.1 KiB
Python
# ============================================================
|
|
# 通用助理：拒识不再甩清单；概念题正面回答
|
|
# ============================================================
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from server.agent_core.assistant import reply as assistant_reply
|
|
from server.agent_core.intent import parse_fast, recognize
|
|
from server.agent_core.providers import reset_provider
|
|
from server.contracts import IntentResult
|
|
from server.aps_domain.workflow import handle_intent
|
|
from server.state.seed import seed_world
|
|
|
|
|
|
class FakeStore:
|
|
def __init__(self):
|
|
self.data = seed_world()
|
|
self._counters: dict[str, int] = {}
|
|
|
|
def next_id(self, kind: str) -> int:
|
|
self._counters[kind] = self._counters.get(kind, 0) + 1
|
|
return self._counters[kind]
|
|
|
|
def save(self) -> None:
|
|
pass
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _no_llm(monkeypatch):
|
|
monkeypatch.delenv("LLM_API_KEY", raising=False)
|
|
monkeypatch.delenv("LLM_PROVIDER", raising=False)
|
|
reset_provider()
|
|
yield
|
|
reset_provider()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_recognize_falls_back_to_assistant():
|
|
world = seed_world()
|
|
r = await recognize("今天天气怎么样顺便聊聊排产", world)
|
|
assert r.intent == "assistant.reply"
|
|
assert r.params.get("query")
|
|
|
|
|
|
def test_fast_greeting_and_howto():
|
|
world = seed_world()
|
|
assert parse_fast("你好", world).intent == "assistant.reply"
|
|
assert parse_fast("你能做什么", world).intent == "assistant.reply"
|
|
assert parse_fast("现在项目怎么样", world).intent == "assistant.reply"
|
|
assert parse_fast("你给我解释什么叫排产", world).intent == "assistant.reply"
|
|
assert parse_fast("你会排产吗", world).intent == "assistant.reply"
|
|
|
|
|
|
def test_imperative_schedule_not_assistant_menu():
|
|
"""「给我排产」必须直接开排，不能落到助理四选一。"""
|
|
world = seed_world()
|
|
for phrase in ("给我排产", "我要你排产", "你给我排产", "开始排产", "排产吧"):
|
|
r = parse_fast(phrase, world)
|
|
assert r is not None, phrase
|
|
assert r.intent in ("flex.schedule", "folder.schedule", "schedule.run"), (
|
|
phrase, r.intent
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_what_time_is_it():
|
|
world = seed_world()
|
|
reply = await assistant_reply(world, "现在几点了")
|
|
assert "现在是" in reply.text
|
|
assert "排产无关" not in reply.text
|
|
assert "帮不上" not in reply.text
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_unknown_handle_gives_project_brief_not_tip_spam():
|
|
store = FakeStore()
|
|
intent = IntentResult(intent="unknown", params={"query": "随便说说"}, confidence=0.1, source="LLM")
|
|
reply = await handle_intent(store, "s1", intent)
|
|
assert "置信度不足" not in reply.text
|
|
assert "我还没锁定" not in reply.text
|
|
assert "排产无关" not in reply.text
|
|
assert "你可以任选一句" not in reply.text
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_explain_scheduling_not_inventory_dump():
|
|
world = seed_world()
|
|
reply = await assistant_reply(world, "你给我解释什么叫排产")
|
|
assert "排产是啥" in reply.text or "把这些事提前安排清楚" in reply.text
|
|
assert "要排的单子" not in reply.text
|
|
assert "你可以任选一句" not in reply.text
|
|
assert not reply.blocks
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_can_schedule_not_inventory_dump():
|
|
world = seed_world()
|
|
reply = await assistant_reply(world, "你会排产吗？")
|
|
assert "本职" in reply.text or "会" in reply.text
|
|
assert "要排的单子" not in reply.text
|
|
assert not reply.blocks
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_assistant_offline_analyze_style():
|
|
world = seed_world()
|
|
reply = await assistant_reply(world, "分析一下现状")
|
|
assert "手头有啥" in reply.text or "我帮你看了看" in reply.text
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_assistant_chitchat_warm_guide():
|
|
world = seed_world()
|
|
reply = await assistant_reply(world, "今天好累不想干")
|
|
assert "置信度不足" not in reply.text
|
|
assert "排产" in reply.text or "单子" in reply.text
|
|
assert "你可以任选一句" not in reply.text
|