338 lines
12 KiB
Python
338 lines
12 KiB
Python
|
"""已完成业务动作的计划员语言说明必须走只读 Pi 叙事段。"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
import pytest
|
|||
|
|
from fastapi.testclient import TestClient
|
|||
|
|
|
|||
|
|
from server.agent_core import fallback_lane, harness
|
|||
|
|
from server.auth.context import bind_identity, reset_identity
|
|||
|
|
from server.contracts import AgentReply
|
|||
|
|
from server.state.seed import seed_world
|
|||
|
|
from tests.auth_provider import install_test_auth
|
|||
|
|
|
|||
|
|
|
|||
|
|
class _Store:
|
|||
|
|
def __init__(self) -> None:
|
|||
|
|
self.data = seed_world()
|
|||
|
|
self._next = 0
|
|||
|
|
|
|||
|
|
def next_id(self, _kind: str) -> int:
|
|||
|
|
self._next += 1
|
|||
|
|
return self._next
|
|||
|
|
|
|||
|
|
def save(self) -> None:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
@pytest.fixture(autouse=True)
|
|||
|
|
def _clear_pending_confirmations():
|
|||
|
|
harness._approval_store.clear()
|
|||
|
|
yield
|
|||
|
|
harness._approval_store.clear()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_action_result_uses_readonly_pi_without_business_tools(tmp_path, monkeypatch):
|
|||
|
|
monkeypatch.setenv("APS_FALLBACK_DIR", str(tmp_path / "fallback"))
|
|||
|
|
captured: dict[str, object] = {}
|
|||
|
|
|
|||
|
|
def runner(task: str, work_dir: Path):
|
|||
|
|
captured["task"] = task
|
|||
|
|
facts_path = work_dir.parent / "inbox" / "completed-action.json"
|
|||
|
|
raw = facts_path.read_text(encoding="utf-8")
|
|||
|
|
captured["payload"] = json.loads(raw.split("\n", 1)[1])
|
|||
|
|
yield {
|
|||
|
|
"type": "message_end",
|
|||
|
|
"message": {
|
|||
|
|
"role": "assistant",
|
|||
|
|
"stopReason": "stop",
|
|||
|
|
"content": [{
|
|||
|
|
"type": "text",
|
|||
|
|
"text": (
|
|||
|
|
"status: success\n\n"
|
|||
|
|
"资料已经采用完成。本次保存了物料和设备资料，"
|
|||
|
|
"下一步可以生成试排方案。"
|
|||
|
|
),
|
|||
|
|
}],
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
yield {"type": "agent_end", "messages": []}
|
|||
|
|
|
|||
|
|
reply = fallback_lane.propose_action_result(
|
|||
|
|
_Store(),
|
|||
|
|
"s-action-result",
|
|||
|
|
action="采用排产资料",
|
|||
|
|
fallback_text=(
|
|||
|
|
"排产资料已采用完成。本次写入：物料 45 项、设备 15 台。\n\n"
|
|||
|
|
"这次操作只保存资料，没有生成排产方案。"
|
|||
|
|
),
|
|||
|
|
facts={
|
|||
|
|
"summary": "排产资料已采用完成。",
|
|||
|
|
"items": [
|
|||
|
|
{"label": "物料", "count": 45, "unit": "项"},
|
|||
|
|
{"label": "设备", "count": 15, "unit": "台"},
|
|||
|
|
],
|
|||
|
|
"nextStep": "下一步可以生成试排方案。",
|
|||
|
|
},
|
|||
|
|
runner=runner,
|
|||
|
|
config=fallback_lane.FallbackConfig(pi_home=str(tmp_path / "pi-home")),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
assert reply.text == (
|
|||
|
|
"资料已经采用完成。本次保存了物料和设备资料，下一步可以生成试排方案。"
|
|||
|
|
)
|
|||
|
|
payload = captured["payload"]
|
|||
|
|
assert payload["本次写入"] == [
|
|||
|
|
{"label": "物料", "count": 45, "unit": "项"},
|
|||
|
|
{"label": "设备", "count": 15, "unit": "台"},
|
|||
|
|
]
|
|||
|
|
assert "materials×45" not in captured["task"]
|
|||
|
|
assert "APS 业务工具" not in captured["task"]
|
|||
|
|
assert "aps_invoke" not in captured["task"]
|
|||
|
|
assert "completed-action.json" in captured["task"]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_action_result_keeps_success_when_pi_is_unavailable(tmp_path, monkeypatch):
|
|||
|
|
monkeypatch.setenv("APS_FALLBACK_DIR", str(tmp_path / "fallback"))
|
|||
|
|
|
|||
|
|
def unavailable(*_args, **_kwargs):
|
|||
|
|
raise fallback_lane.FallbackUnavailable("模型服务未配置")
|
|||
|
|
|
|||
|
|
monkeypatch.setattr(fallback_lane, "build_pi_runner", unavailable)
|
|||
|
|
fallback_text = (
|
|||
|
|
"排产资料已采用完成。本次写入：物料 45 项、设备 15 台。\n\n"
|
|||
|
|
"这次操作只保存资料，没有生成排产方案。"
|
|||
|
|
)
|
|||
|
|
reply = fallback_lane.propose_action_result(
|
|||
|
|
_Store(),
|
|||
|
|
"s-action-result",
|
|||
|
|
action="采用排产资料",
|
|||
|
|
fallback_text=fallback_text,
|
|||
|
|
facts={
|
|||
|
|
"summary": "排产资料已采用完成。",
|
|||
|
|
"items": [
|
|||
|
|
{"label": "物料", "count": 45, "unit": "项"},
|
|||
|
|
{"label": "设备", "count": 15, "unit": "台"},
|
|||
|
|
],
|
|||
|
|
"nextStep": "下一步可以生成试排方案。",
|
|||
|
|
},
|
|||
|
|
config=fallback_lane.FallbackConfig(pi_home=str(tmp_path / "pi-home")),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
assert reply.text == fallback_text
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_action_result_rejects_internal_field_names_from_pi(tmp_path, monkeypatch):
|
|||
|
|
monkeypatch.setenv("APS_FALLBACK_DIR", str(tmp_path / "fallback"))
|
|||
|
|
|
|||
|
|
def runner(_task: str, _work_dir: Path):
|
|||
|
|
yield {
|
|||
|
|
"type": "message_end",
|
|||
|
|
"message": {
|
|||
|
|
"role": "assistant",
|
|||
|
|
"stopReason": "stop",
|
|||
|
|
"content": [{
|
|||
|
|
"type": "text",
|
|||
|
|
"text": "status: success\n\n已采用 materials×45、equipment×15。",
|
|||
|
|
}],
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
yield {"type": "agent_end", "messages": []}
|
|||
|
|
|
|||
|
|
fallback_text = "排产资料已采用完成。本次写入：物料 45 项、设备 15 台。"
|
|||
|
|
reply = fallback_lane.propose_action_result(
|
|||
|
|
_Store(),
|
|||
|
|
"s-action-result",
|
|||
|
|
action="采用排产资料",
|
|||
|
|
fallback_text=fallback_text,
|
|||
|
|
facts={
|
|||
|
|
"summary": "排产资料已采用完成。",
|
|||
|
|
"items": [
|
|||
|
|
{"label": "物料", "count": 45, "unit": "项"},
|
|||
|
|
{"label": "设备", "count": 15, "unit": "台"},
|
|||
|
|
],
|
|||
|
|
"nextStep": "下一步可以生成试排方案。",
|
|||
|
|
},
|
|||
|
|
runner=runner,
|
|||
|
|
config=fallback_lane.FallbackConfig(pi_home=str(tmp_path / "pi-home")),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
assert reply.text == fallback_text
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_import_commit_confirmation_route_returns_pi_explanation(tmp_path, monkeypatch):
|
|||
|
|
monkeypatch.setenv("APS_SEED_DEMO", "0")
|
|||
|
|
monkeypatch.setenv("APS_HOME", str(tmp_path))
|
|||
|
|
monkeypatch.setenv("APS_DB_PATH", str(tmp_path / "master.db"))
|
|||
|
|
monkeypatch.setenv("APS_WORLD_PATH", str(tmp_path / "world.json"))
|
|||
|
|
monkeypatch.setenv("APS_DATA_DIR", str(tmp_path / "data"))
|
|||
|
|
provider = install_test_auth(monkeypatch, "action-result-test")
|
|||
|
|
|
|||
|
|
from server.gateway.app import create_app
|
|||
|
|
from server.state.store import get_store
|
|||
|
|
|
|||
|
|
with TestClient(create_app()) as client:
|
|||
|
|
login = client.post(
|
|||
|
|
"/api/auth/login",
|
|||
|
|
json={"username": "planner", "password": "test"},
|
|||
|
|
)
|
|||
|
|
assert login.status_code == 200, login.text
|
|||
|
|
store = get_store()
|
|||
|
|
token = bind_identity(provider._identity("planner"))
|
|||
|
|
try:
|
|||
|
|
block = harness.stage_confirmation(
|
|||
|
|
"s-action-result",
|
|||
|
|
"import.commit",
|
|||
|
|
{
|
|||
|
|
"filename": "测试资料.xlsx",
|
|||
|
|
"analysisIngest": {
|
|||
|
|
"projectName": "测试项目",
|
|||
|
|
"sources": ["测试资料.xlsx"],
|
|||
|
|
},
|
|||
|
|
"batches": [{
|
|||
|
|
"kind": "materials",
|
|||
|
|
"sheet": "物料",
|
|||
|
|
"okCount": 1,
|
|||
|
|
"errorCount": 0,
|
|||
|
|
"okRows": [{
|
|||
|
|
"code": "M-100",
|
|||
|
|
"name": "测试物料",
|
|||
|
|
"type": "RAW_MATERIAL",
|
|||
|
|
"unit": "kg",
|
|||
|
|
"stock": 0,
|
|||
|
|
}],
|
|||
|
|
}],
|
|||
|
|
},
|
|||
|
|
title="核对并采用本次排产资料",
|
|||
|
|
summary_lines=["确认后保存资料，本步不会排产。"],
|
|||
|
|
)
|
|||
|
|
finally:
|
|||
|
|
reset_identity(token)
|
|||
|
|
store.save()
|
|||
|
|
|
|||
|
|
captured: dict[str, object] = {}
|
|||
|
|
|
|||
|
|
def fake_result(
|
|||
|
|
store, session_id, *, action, fallback_text, facts,
|
|||
|
|
actor="planner", **_kwargs,
|
|||
|
|
):
|
|||
|
|
captured.update({
|
|||
|
|
"session_id": session_id,
|
|||
|
|
"action": action,
|
|||
|
|
"fallback_text": fallback_text,
|
|||
|
|
"facts": facts,
|
|||
|
|
"actor": actor,
|
|||
|
|
})
|
|||
|
|
return AgentReply(text="资料已经采用，下一步可以生成试排方案。")
|
|||
|
|
|
|||
|
|
monkeypatch.setattr(fallback_lane, "propose_action_result", fake_result)
|
|||
|
|
response = client.post(
|
|||
|
|
"/api/actions/confirm",
|
|||
|
|
json={
|
|||
|
|
"sessionId": "s-action-result",
|
|||
|
|
"confirmId": block.props["confirmId"],
|
|||
|
|
"approve": True,
|
|||
|
|
},
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
assert response.status_code == 200, response.text
|
|||
|
|
assert response.json()["message"] == "资料已经采用，下一步可以生成试排方案。"
|
|||
|
|
assert captured["session_id"] == "s-action-result"
|
|||
|
|
assert captured["action"] == "采用排产资料"
|
|||
|
|
assert str(captured["fallback_text"]).startswith("排产资料已采用完成。")
|
|||
|
|
assert captured["facts"]["items"] == [
|
|||
|
|
{"label": "物料", "count": 1, "unit": "项"},
|
|||
|
|
]
|
|||
|
|
assert "materials×" not in json.dumps(captured["facts"], ensure_ascii=False)
|
|||
|
|
assert captured["actor"] == "planner"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_import_commit_batch_route_returns_pi_explanation(tmp_path, monkeypatch):
|
|||
|
|
monkeypatch.setenv("APS_SEED_DEMO", "0")
|
|||
|
|
monkeypatch.setenv("APS_HOME", str(tmp_path))
|
|||
|
|
monkeypatch.setenv("APS_DB_PATH", str(tmp_path / "master.db"))
|
|||
|
|
monkeypatch.setenv("APS_WORLD_PATH", str(tmp_path / "world.json"))
|
|||
|
|
monkeypatch.setenv("APS_DATA_DIR", str(tmp_path / "data"))
|
|||
|
|
provider = install_test_auth(monkeypatch, "action-result-batch-test")
|
|||
|
|
|
|||
|
|
from server.gateway.app import create_app
|
|||
|
|
from server.state.store import get_store
|
|||
|
|
|
|||
|
|
with TestClient(create_app()) as client:
|
|||
|
|
login = client.post(
|
|||
|
|
"/api/auth/login",
|
|||
|
|
json={"username": "planner", "password": "test"},
|
|||
|
|
)
|
|||
|
|
assert login.status_code == 200, login.text
|
|||
|
|
store = get_store()
|
|||
|
|
token = bind_identity(provider._identity("planner"))
|
|||
|
|
try:
|
|||
|
|
block = harness.stage_confirmation(
|
|||
|
|
"s-action-result-batch",
|
|||
|
|
"import.commit",
|
|||
|
|
{
|
|||
|
|
"filename": "批量测试资料.xlsx",
|
|||
|
|
"analysisIngest": {
|
|||
|
|
"projectName": "批量测试项目",
|
|||
|
|
"sources": ["批量测试资料.xlsx"],
|
|||
|
|
},
|
|||
|
|
"batches": [{
|
|||
|
|
"kind": "equipment",
|
|||
|
|
"sheet": "设备",
|
|||
|
|
"okCount": 1,
|
|||
|
|
"errorCount": 0,
|
|||
|
|
"okRows": [{
|
|||
|
|
"code": "EQ-100",
|
|||
|
|
"name": "测试设备",
|
|||
|
|
"type": "MACHINE",
|
|||
|
|
}],
|
|||
|
|
}],
|
|||
|
|
},
|
|||
|
|
title="核对并采用本次排产资料",
|
|||
|
|
summary_lines=["确认后保存资料，本步不会排产。"],
|
|||
|
|
)
|
|||
|
|
finally:
|
|||
|
|
reset_identity(token)
|
|||
|
|
store.save()
|
|||
|
|
|
|||
|
|
captured: list[dict[str, object]] = []
|
|||
|
|
|
|||
|
|
def fake_result(
|
|||
|
|
store, session_id, *, action, fallback_text, facts,
|
|||
|
|
actor="planner", **_kwargs,
|
|||
|
|
):
|
|||
|
|
captured.append({
|
|||
|
|
"session_id": session_id,
|
|||
|
|
"action": action,
|
|||
|
|
"fallback_text": fallback_text,
|
|||
|
|
"facts": facts,
|
|||
|
|
"actor": actor,
|
|||
|
|
})
|
|||
|
|
return AgentReply(text="资料已经采用，下一步可以生成试排方案。")
|
|||
|
|
|
|||
|
|
monkeypatch.setattr(fallback_lane, "propose_action_result", fake_result)
|
|||
|
|
response = client.post(
|
|||
|
|
"/api/actions/confirm-batch",
|
|||
|
|
json={
|
|||
|
|
"sessionId": "s-action-result-batch",
|
|||
|
|
"confirmIds": [block.props["confirmId"]],
|
|||
|
|
"approve": True,
|
|||
|
|
},
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
assert response.status_code == 200, response.text
|
|||
|
|
results = response.json()["results"]
|
|||
|
|
assert len(results) == 1
|
|||
|
|
assert results[0]["ok"] is True
|
|||
|
|
assert results[0]["message"] == "资料已经采用，下一步可以生成试排方案。"
|
|||
|
|
assert len(captured) == 1
|
|||
|
|
assert captured[0]["session_id"] == "s-action-result-batch"
|
|||
|
|
assert captured[0]["action"] == "采用排产资料"
|
|||
|
|
assert captured[0]["facts"]["items"] == [
|
|||
|
|
{"label": "设备", "count": 1, "unit": "台"},
|
|||
|
|
]
|
|||
|
|
assert "equipment×" not in json.dumps(captured[0]["facts"], ensure_ascii=False)
|
|||
|
|
assert captured[0]["actor"] == "planner"
|