120 lines
4.8 KiB
Python
120 lines
4.8 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from sqlalchemy import create_engine
|
||
|
|
from sqlalchemy.orm import Session, sessionmaker
|
||
|
|
from sqlalchemy.pool import StaticPool
|
||
|
|
|
||
|
|
from app.core.agent_enums import AgentAssetDomain
|
||
|
|
from app.db.base import Base
|
||
|
|
from app.models.agent_asset import AgentAsset
|
||
|
|
from app.schemas.agent_asset import (
|
||
|
|
AgentAssetRiskRuleGenerateRequest,
|
||
|
|
AgentAssetRiskRuleSimulationRequest,
|
||
|
|
)
|
||
|
|
from app.services.agent_asset_rule_library import AgentAssetRuleLibraryManager
|
||
|
|
from app.services.agent_asset_spreadsheet import RISK_RULES_LIBRARY
|
||
|
|
from app.services.agent_assets import AgentAssetService
|
||
|
|
from app.services.risk_rule_generation import RiskRuleGenerationService
|
||
|
|
|
||
|
|
|
||
|
|
class NullRuntimeChatService:
|
||
|
|
def complete(self, *args, **kwargs) -> None:
|
||
|
|
return None
|
||
|
|
|
||
|
|
|
||
|
|
def build_session() -> Session:
|
||
|
|
engine = create_engine(
|
||
|
|
"sqlite+pysqlite:///:memory:",
|
||
|
|
connect_args={"check_same_thread": False},
|
||
|
|
poolclass=StaticPool,
|
||
|
|
)
|
||
|
|
Base.metadata.create_all(bind=engine)
|
||
|
|
session_factory = sessionmaker(bind=engine, autoflush=False, autocommit=False)
|
||
|
|
return session_factory()
|
||
|
|
|
||
|
|
|
||
|
|
def test_generated_risk_rule_contains_semantic_plan_and_flow_model(tmp_path) -> None:
|
||
|
|
with build_session() as db:
|
||
|
|
manager = AgentAssetRuleLibraryManager(rule_root=tmp_path / "rules")
|
||
|
|
asset_id = RiskRuleGenerationService(
|
||
|
|
db,
|
||
|
|
rule_library_manager=manager,
|
||
|
|
runtime_chat_service=NullRuntimeChatService(),
|
||
|
|
).generate_rule_asset(
|
||
|
|
AgentAssetRiskRuleGenerateRequest(
|
||
|
|
business_domain=AgentAssetDomain.EXPENSE,
|
||
|
|
expense_category="travel",
|
||
|
|
rule_title="差旅票据城市一致性校验",
|
||
|
|
natural_language=(
|
||
|
|
"差旅报销时，交通票或住宿票据中的城市必须与申报目的地一致；"
|
||
|
|
"未说明绕行、跨城或改签原因时标记高风险。"
|
||
|
|
),
|
||
|
|
requires_attachment=True,
|
||
|
|
),
|
||
|
|
actor="pytest",
|
||
|
|
)
|
||
|
|
asset = db.get(AgentAsset, asset_id)
|
||
|
|
assert asset is not None
|
||
|
|
payload = manager.read_rule_library_json(
|
||
|
|
library=RISK_RULES_LIBRARY,
|
||
|
|
file_name=asset.config_json["rule_document"]["file_name"],
|
||
|
|
)
|
||
|
|
|
||
|
|
assert payload["semantic_plan"]["required_fields"]
|
||
|
|
assert payload["semantic_plan"]["risk_action"]["risk_level"] == payload["outcomes"]["fail"]["severity"]
|
||
|
|
assert payload["flow_model"]["source"] == "json_dsl"
|
||
|
|
assert payload["flow_model"]["nodes"][0]["id"] == "start"
|
||
|
|
assert any(node["type"] == "risk" for node in payload["flow_model"]["nodes"])
|
||
|
|
assert payload["metadata"]["flow_model"]["nodes"] == payload["flow_model"]["nodes"]
|
||
|
|
assert payload["flow_diagram_svg"].startswith("<svg")
|
||
|
|
assert "风险关键词" not in payload["semantic_plan"]["judgment_steps"][0]["description"]
|
||
|
|
|
||
|
|
|
||
|
|
def test_simulation_returns_execution_trace_for_ticket_city_mismatch(tmp_path) -> None:
|
||
|
|
with build_session() as db:
|
||
|
|
manager = AgentAssetRuleLibraryManager(rule_root=tmp_path / "rules")
|
||
|
|
asset_id = RiskRuleGenerationService(
|
||
|
|
db,
|
||
|
|
rule_library_manager=manager,
|
||
|
|
runtime_chat_service=NullRuntimeChatService(),
|
||
|
|
).generate_rule_asset(
|
||
|
|
AgentAssetRiskRuleGenerateRequest(
|
||
|
|
business_domain=AgentAssetDomain.EXPENSE,
|
||
|
|
expense_category="travel",
|
||
|
|
rule_title="当前差旅票据城市一致性规则",
|
||
|
|
natural_language=(
|
||
|
|
"差旅报销时，交通票或住宿票据中的城市必须与申报目的地一致；"
|
||
|
|
"未说明绕行、跨城或改签原因时标记高风险。"
|
||
|
|
),
|
||
|
|
requires_attachment=True,
|
||
|
|
),
|
||
|
|
actor="pytest",
|
||
|
|
)
|
||
|
|
service = AgentAssetService(db)
|
||
|
|
service.rule_library_manager = manager
|
||
|
|
|
||
|
|
simulation = service.simulate_risk_rule_message(
|
||
|
|
asset_id,
|
||
|
|
AgentAssetRiskRuleSimulationRequest(
|
||
|
|
message="去北京出差3天",
|
||
|
|
attachments=[
|
||
|
|
{
|
||
|
|
"name": "train-ticket.pdf",
|
||
|
|
"content_type": "application/pdf",
|
||
|
|
"ocr_text": "武汉 到 上海",
|
||
|
|
"summary": "高铁票 武汉-上海",
|
||
|
|
"document_fields": [
|
||
|
|
{"key": "route", "label": "行程路线", "value": "武汉-上海"}
|
||
|
|
],
|
||
|
|
}
|
||
|
|
],
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
assert simulation.ready is True
|
||
|
|
assert simulation.hit is True
|
||
|
|
assert simulation.normalized_fields["claim.location"] == "北京"
|
||
|
|
assert simulation.trace["matched"] is True
|
||
|
|
assert "hit" in simulation.trace["path_node_ids"]
|
||
|
|
assert simulation.trace["steps"]
|