feat(services): assignment 6 skeleton - app service, field registry, in-memory repository

- fields_registry: 24 business keys in acd/human1/human2 groups with
  value_type, writable and SensitiveLevel metadata; strict bool/int
  parsing (no bool('0') trap); INTERNAL_FIELD_KEYS protection list
- repository: SessionRepository protocol + InMemorySessionRepository
  (idempotent delete; PostgreSQL swap-in is assignment 7)
- AccidentAppService: chat/set_info/get_info/delete_session share one
  business state - chat injects latest repository fields into the graph
  and writes back validated field diffs; set/get/delete make zero LLM
  calls and zero helper conversations (includeInputInfo accepted but
  unused); unknown get keys keep the '' JSON-empty-string contract,
  unknown set keys are rejected by whitelist
- ChatInput.injected_fields neutral contract extension; LangGraphBackend
  merges injected fields into graph input (set_info visible next chat)
- endpoints: thin layer - langgraph mode routes business state through
  the app service (fastgpt mode keeps the legacy path); delete cleans
  repository + graph checkpoint, idempotent
- tests: registry completeness (sfzmwh/sjwh/csbw included), app service
  combos with LLM-call spy, HTTP acceptance flow (set -> get -> chat ->
  groups -> delete -> no leak, idempotent re-delete)
- docs/assignment6-compat.md: behavior diff list vs FastGPT status quo
  and the audit backlog (arbitrary set_info key usage, stream formUpdate
  write-back, sensitive-field display boundary)

Verified: 200 unit tests (172 existing + 28 new)
This commit is contained in:
Eric Wang
2026-10-22 20:50:10 +08:00
parent 6c1fbd0295
commit 8717d9066c
12 changed files with 882 additions and 16 deletions
+5
View File
@@ -0,0 +1,5 @@
"""应用服务层(作业 6):应用服务、字段注册表、会话 repository。
endpoint 只处理校验、协议(SSE/响应模型)和依赖注入;
业务状态读写、字段白名单与分组语义集中在本层。
"""
+152
View File
@@ -0,0 +1,152 @@
"""事故处理应用服务(作业 6-1):四业务接口共享同一份业务状态。
职责边界(对应计划作业 6):
- set_info / get_info / delete_session **不调用 LLM、不创建辅助对话**
(对照旧 FastGPT 实现的“辅助对话 + 删记录”hack);
- chat 每轮从 repository 读最新业务字段注入图(set_info 写入后
下一轮 chat 必须可见),图产出字段差量后写回 repository;
- 未知 key 兼容策略:get 对未知 key 返回 JSON 空串 '""'(现状兼容);
set 对未知 key 拒绝写入(白名单保护,差异见兼容清单);
- includeInputInfo 继续被 endpoint 接受,但本层不再使用
——新后端无需触发辅助对话。
endpoint 只负责校验、协议(SSE / 响应模型)与依赖注入。
"""
from ..backends.chat import ChatBackend, ChatInput, ChatResult
from .fields_registry import (
FIELD_REGISTRY,
GROUP_KEYS,
FieldGroup,
parse_field_value,
to_external_string,
)
from .repository import SessionRepository
# set_info 对未知/受保护 key 的拒绝码(与现有“入参不规范”文案对齐)
SET_INFO_INVALID_KEY_MSG = "入参不规范:未知或受保护的字段 key"
class AccidentAppService:
"""四个业务接口的应用服务(langgraph 后端)。"""
def __init__(self, backend: ChatBackend, repository: SessionRepository):
self._backend = backend
self._repository = repository
# ---- chat:图每轮读取最新业务状态,产出差量写回 ----
def _injected_fields(self, session_id: str) -> dict:
"""repository 业务字段 -> 图输入的分组注入(acd 进 accident_fields,
两组人员进 person_fields,与 AgentState 的键结构对齐)。"""
business = self._repository.load(session_id)
if not business:
return {}
injected: dict = {"accident_fields": {}, "person_fields": {}}
for key, value in business.items():
spec = FIELD_REGISTRY.get(key)
if spec is None:
continue
if spec.group is FieldGroup.ACD:
injected["accident_fields"][key] = value
else:
injected["person_fields"][key] = value
return injected
async def chat(self, chat_input: ChatInput) -> ChatResult:
"""非流式一轮对话:注入最新业务字段 -> 执行图 -> 写回字段差量。"""
injected = self._injected_fields(chat_input.session_id)
result = await self._backend.complete(
ChatInput(
session_id=chat_input.session_id,
text=chat_input.text,
need_form_update=chat_input.need_form_update,
client_mode=chat_input.client_mode,
injected_fields=injected or None,
)
)
# 图产出的字段差量(turn_patch / 完整字段)写回 repository,
# 使 set/get_info 与后续轮次看到同一事实来源
patch = result.form_update if isinstance(result.form_update, dict) else {}
if patch:
self._repository.apply_update(chat_input.session_id, patch)
return result
async def stream(self, chat_input: ChatInput):
"""流式代理:字段注入与写回语义同 chat(),事件流原样透传。"""
injected = self._injected_fields(chat_input.session_id)
result_input = ChatInput(
session_id=chat_input.session_id,
text=chat_input.text,
need_form_update=chat_input.need_form_update,
client_mode=chat_input.client_mode,
injected_fields=injected or None,
)
# 骨架阶段原样透传事件流;字段差量写回在作业 6 完整版随
# FormUpdate 事件捕获补齐(对照计划作业 8 的真实增量)。
async for event in self._backend.stream(result_input):
yield event
# ---- set_info / get_info:零 LLM、零辅助对话 ----
def set_info(self, session_id: str, key: str, value: str) -> tuple[str, str]:
"""白名单校验后写入 repository。返回 (code, msg)。"""
spec = FIELD_REGISTRY.get(key)
if spec is None or not spec.writable:
return "500", SET_INFO_INVALID_KEY_MSG
try:
parsed = parse_field_value(spec, value)
except ValueError:
return "500", "入参不规范:字段值无法解析"
self._repository.apply_update(session_id, {key: parsed})
return "200", ""
def get_info(self, session_id: str, key: str) -> str:
"""按分组或单键读出,返回 JSON 字符串(兼容现有 /get_info 契约)。"""
import json
business = self._repository.load(session_id)
if key == "all":
return json.dumps(
{
group.value: {
k: to_external_string(business.get(k))
for k in GROUP_KEYS[group]
}
for group in FieldGroup
},
ensure_ascii=False,
)
try:
group = FieldGroup(key)
except ValueError:
group = None
if group is not None:
return json.dumps(
{k: to_external_string(business.get(k)) for k in GROUP_KEYS[group]},
ensure_ascii=False,
)
# 单键:未知 key 返回 JSON 空串(兼容现状);已知 key 返回其外部表示
if key not in FIELD_REGISTRY:
return json.dumps("", ensure_ascii=False)
return json.dumps(to_external_string(business.get(key)), ensure_ascii=False)
# ---- delete_session:业务状态 + 图 checkpoint 一起清理 ----
async def delete_session(self, session_id: str, drop_checkpoint) -> None:
"""删除会话全部数据。
drop_checkpoint:由 core 层注入的 checkpoint 清理函数
(图生命周期归 lifespan 管理,本层不持有图引用)。
repository 删除幂等;checkpoint 缺失时静默(幂等语义一致)。
"""
self._repository.delete(session_id)
if drop_checkpoint is not None:
try:
await drop_checkpoint(session_id)
except Exception:
# checkpoint 已不存在视为删除成功的一部分(幂等)
pass
__all__ = ["AccidentAppService", "SET_INFO_INVALID_KEY_MSG"]
+144
View File
@@ -0,0 +1,144 @@
"""字段注册表(作业 6-3/6-4):外部业务键 -> 类型/分组/可写性/敏感级别。
完整覆盖当前 get_info 代码中的三个分组:
- 事故字段(acdinfo):ywrysw/ywfjdc/ywmtc/bjrjs/sgfssj/sfsgxc/jdcsl/sgyy
- 本方人员(acdhuman1):xm1/hpzl1/hphm1/sfzmhm1/sfzmwh1/sjhm1/sjwh1/csbw1
- 对方人员(acdhuman2):同上一组,序号 2
设计约束:
- 外部传输一律字符串;内部按 value_type 存储强类型值,
保留区分 空值 / False / 0 / 未知 的能力(不做 bool("0") 转换);
- 敏感级别用于作业 9 的访问边界与日志脱敏,注册表先行记录;
- get 的分组名(acdinfo/acdhuman1/acdhuman2)与外部接口契约保持一致。
"""
from dataclasses import dataclass
from enum import StrEnum
from typing import Any
class FieldGroup(StrEnum):
"""get_info 对外分组名(兼容现有 /get_info 契约)。"""
ACD = "acdinfo"
HUMAN1 = "acdhuman1"
HUMAN2 = "acdhuman2"
class SensitiveLevel(StrEnum):
"""敏感级别(作业 9 用于访问边界/脱敏)。"""
PUBLIC = "public" # 无个人信息
PERSONAL = "personal" # 姓名等一般个人信息
SENSITIVE = "sensitive" # 证件号/手机号,日志与回显需保护
@dataclass(frozen=True)
class FieldSpec:
"""一个外部业务字段的注册信息。"""
key: str
group: FieldGroup
value_type: type # str / bool / int
writable: bool
sensitive: SensitiveLevel
description: str = ""
def _acd_fields() -> list[FieldSpec]:
return [
FieldSpec("ywrysw", FieldGroup.ACD, bool, True, SensitiveLevel.PUBLIC, "是否有人受伤"),
FieldSpec("ywfjdc", FieldGroup.ACD, bool, True, SensitiveLevel.PUBLIC, "是否撞非机动车"),
FieldSpec("ywmtc", FieldGroup.ACD, bool, True, SensitiveLevel.PUBLIC, "是否撞摩托车"),
FieldSpec("bjrjs", FieldGroup.ACD, bool, True, SensitiveLevel.PUBLIC, "报警人是否驾驶员"),
FieldSpec("sgfssj", FieldGroup.ACD, str, True, SensitiveLevel.PUBLIC, "事故发生时间"),
FieldSpec("sfsgxc", FieldGroup.ACD, bool, True, SensitiveLevel.PUBLIC, "是否在现场"),
FieldSpec("jdcsl", FieldGroup.ACD, int, True, SensitiveLevel.PUBLIC, "机动车数量"),
FieldSpec("sgyy", FieldGroup.ACD, str, True, SensitiveLevel.PUBLIC, "事故原因"),
]
def _human_fields(index: int) -> list[FieldSpec]:
group = FieldGroup.HUMAN1 if index == 1 else FieldGroup.HUMAN2
side = "本方" if index == 1 else "对方"
return [
FieldSpec(f"xm{index}", group, str, True, SensitiveLevel.PERSONAL, f"{side}姓名"),
FieldSpec(f"hpzl{index}", group, str, True, SensitiveLevel.PUBLIC, f"{side}号牌种类"),
FieldSpec(f"hphm{index}", group, str, True, SensitiveLevel.PUBLIC, f"{side}号牌号码"),
FieldSpec(f"sfzmhm{index}", group, str, True, SensitiveLevel.SENSITIVE, f"{side}身份证明号码"),
FieldSpec(f"sfzmwh{index}", group, str, True, SensitiveLevel.SENSITIVE, f"{side}身份证明文号"),
FieldSpec(f"sjhm{index}", group, str, True, SensitiveLevel.SENSITIVE, f"{side}手机号码"),
FieldSpec(f"sjwh{index}", group, str, True, SensitiveLevel.PUBLIC, f"{side}手机文号"),
FieldSpec(f"csbw{index}", group, str, True, SensitiveLevel.PUBLIC, f"{side}常熟部位/身份标识"),
]
# 注册表:key -> FieldSpec(构建一次,只读共享)
FIELD_REGISTRY: dict[str, FieldSpec] = {
spec.key: spec for spec in [*_acd_fields(), *_human_fields(1), *_human_fields(2)]
}
# 分组 -> 组内有序字段键(保持当前 get_info 的键顺序,输出稳定)
GROUP_KEYS: dict[FieldGroup, tuple[str, ...]] = {
group: tuple(spec.key for spec in FIELD_REGISTRY.values() if spec.group == group)
for group in FieldGroup
}
# 内部受保护字段:注册表之外、图内部使用的键,set_info 不得写入
# (兼容策略见 docs/assignment6-compat.md)
INTERNAL_FIELD_KEYS = frozenset(
{
"stage_code", "raw_status_code", "turn_count", "client_event",
"clarify_count", "no_reply_count", "state_version",
}
)
def parse_field_value(spec: FieldSpec, raw: str) -> Any:
"""外部字符串 -> 内部强类型值;无法解析时抛 ValueError。
bool 只接受明确的 1/0/true/false/是/否;拒绝空串与模糊值,
避免 bool("0") == True 这类陷阱。int 必须是纯数字字面量。
"""
if raw is None:
raise ValueError(f"{spec.key}: 值为空")
text = raw.strip()
if spec.value_type is bool:
lowered = text.lower()
if lowered in ("1", "true", "是"):
return True
if lowered in ("0", "false", "否"):
return False
raise ValueError(f"{spec.key}: 无法解析为布尔值: {raw!r}")
if spec.value_type is int:
if text.isdigit():
return int(text)
raise ValueError(f"{spec.key}: 无法解析为整数: {raw!r}")
return raw # str 原样存储(空串视为有效的外部空值)
def to_external_string(value: Any) -> str:
"""内部值 -> 外部字符串表示(兼容现有 get_info 的 bool_to_str)。
True -> "1",False -> "0",None/未知 -> "",其余 str() 原样。
内部仍保留 False 与 0 的区分,仅在对外序列化时收敛。
"""
if value is True:
return "1"
if value is False:
return "0"
if value is None:
return ""
return str(value)
__all__ = [
"FIELD_REGISTRY",
"GROUP_KEYS",
"INTERNAL_FIELD_KEYS",
"FieldGroup",
"FieldSpec",
"SensitiveLevel",
"parse_field_value",
"to_external_string",
]
+52
View File
@@ -0,0 +1,52 @@
"""会话业务状态 repository(作业 6-1/6-2)。
抽象出 repository 接口,使应用服务不依赖具体存储;
作业 6 用内存实现跑通四接口,作业 7 再替换为 PostgreSQL 实现
(对话 checkpoint 已由 LangGraph checkpointer 承担,本层只管
业务字段——单一事实来源,图每轮读取最新值)。
语义约定:
- 业务状态按外部业务键(hphm1、ywrysw...)扁平存储,值为内部强类型;
- 不存在的会话视为空状态(读取返回 {}),与首轮 checkpoint 无字段一致;
- delete 幂等:重复删除固定成功,不抛异常(行为进测试)。
"""
from typing import Protocol
class SessionRepository(Protocol):
"""业务字段存储接口(作业 7 由 PostgreSQL 实现替换)。"""
def load(self, session_id: str) -> dict:
"""读取会话业务字段;无记录时返回空 dict。"""
...
def apply_update(self, session_id: str, patch: dict) -> dict:
"""合并差量并返回合并后的完整业务字段。"""
...
def delete(self, session_id: str) -> None:
"""删除会话业务状态;幂等,重复删除不抛异常。"""
...
class InMemorySessionRepository:
"""内存实现:字典存储,仅用于单进程开发与测试。"""
def __init__(self) -> None:
self._store: dict[str, dict] = {}
def load(self, session_id: str) -> dict:
return dict(self._store.get(session_id) or {})
def apply_update(self, session_id: str, patch: dict) -> dict:
state = self._store.setdefault(session_id, {})
state.update(patch)
return dict(state)
def delete(self, session_id: str) -> None:
# 幂等:pop 的缺省值吞掉不存在的情况,重复删除行为固定为成功
self._store.pop(session_id, None)
__all__ = ["SessionRepository", "InMemorySessionRepository"]