刘金宝
109 lines
3.8 KiB
Python
109 lines
3.8 KiB
Python
import sys
|
|
import os
|
|
from pathlib import Path
|
|
|
|
os.environ.setdefault("DATABASE_URL", "sqlite:///./storage/test_core.db")
|
|
os.environ.setdefault("RUNTIME_MEMORY_BACKEND", "memory")
|
|
os.environ.setdefault("LLM_MOCK_ENABLED", "true")
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
|
|
|
|
from app.agents.scoring_agent import ScoringAgent
|
|
from app.agents.hint_agent import HintAgent
|
|
from app.agents.llm_adapter import OpenAICompatibleLLMClient
|
|
from app.agents.patient_agent import PatientAgent
|
|
from app.core.config import settings
|
|
from app.services.runtime_memory import InMemoryRuntimeMemoryService
|
|
|
|
|
|
def test_runtime_memory_lifecycle() -> None:
|
|
"""短期 memory：验证创建、写入、读取和释放流程。"""
|
|
memory = InMemoryRuntimeMemoryService()
|
|
memory.create("test-memory", "开场白")
|
|
memory.add_message("test-memory", "doctor", "孩子发热几天？")
|
|
assert memory.has_doctor_message("test-memory") is True
|
|
assert len(memory.get_messages("test-memory")) == 2
|
|
memory.release("test-memory")
|
|
assert memory.get_messages("test-memory") == []
|
|
|
|
|
|
def test_score_convert_to_five_point() -> None:
|
|
"""评分转换：验证百分制到五分制的结构转换。"""
|
|
agent = ScoringAgent()
|
|
result = agent._convert_to_five_point(
|
|
{
|
|
"score_type": "percentage",
|
|
"total_score": 80,
|
|
"dimension_scores": [{"dimension": "信息获取", "score": 20, "max_score": 25, "comment": "ok"}],
|
|
}
|
|
)
|
|
assert result["score_type"] == "five_point"
|
|
assert result["total_score"] == 4.0
|
|
assert result["dimension_scores"][0]["max_score"] == 5
|
|
|
|
|
|
def test_public_settings() -> None:
|
|
"""配置输出：验证 Demo 前端可读取功能开关。"""
|
|
public = settings.as_public_dict()
|
|
assert "score_types" in public
|
|
assert "percentage" in public["score_types"]
|
|
|
|
|
|
def test_reasoning_effort_disabled_when_thinking_off() -> None:
|
|
"""LLM 参数构造：thinking 关闭时不发送 reasoning_effort，避免 reason 测试流式 400。"""
|
|
client = OpenAICompatibleLLMClient()
|
|
payload = client._build_payload(
|
|
model="deepseek-v4-pro",
|
|
messages=[{"role": "user", "content": "hello"}],
|
|
stream=True,
|
|
thinking_enabled=False,
|
|
reasoning_effort="low",
|
|
max_tokens=128,
|
|
)
|
|
assert "reasoning_effort" not in payload
|
|
|
|
|
|
def test_hint_agent_invalid_json_fallback() -> None:
|
|
"""新手提示：验证模型输出结构不匹配时使用稳定 fallback。"""
|
|
agent = HintAgent()
|
|
payload = {
|
|
"case": {
|
|
"chief_complaint": "发热、咳嗽4天，喘息1天",
|
|
"key_exams": ["blood_routine", "chest_xray"],
|
|
},
|
|
"ordered_results": [],
|
|
}
|
|
result = agent._normalize_output({"score_type": "percentage"}, payload)
|
|
assert result["hints"]
|
|
assert result["next_questions"]
|
|
assert result["recommended_orders"]
|
|
|
|
|
|
def test_patient_agent_config_rule_boundary() -> None:
|
|
"""病人配置提示：验证训练页初始化配置进入 Patient Agent 且不改变病例事实边界。"""
|
|
agent = PatientAgent()
|
|
rule = agent._build_patient_config_rule(
|
|
{
|
|
"labels": {
|
|
"visit_environment": "急诊",
|
|
"age_group": "儿童",
|
|
"education_level": "小学及以下",
|
|
"personality": "焦虑",
|
|
}
|
|
}
|
|
)
|
|
assert "就诊环境=急诊" in rule
|
|
assert "以家属代述为主" in rule
|
|
assert "不能改变病例事实" in rule
|
|
assert "不能编造检查检验结果" in rule
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_runtime_memory_lifecycle()
|
|
test_score_convert_to_five_point()
|
|
test_public_settings()
|
|
test_reasoning_effort_disabled_when_thinking_off()
|
|
test_hint_agent_invalid_json_fallback()
|
|
test_patient_agent_config_rule_boundary()
|
|
print("core logic tests passed")
|