Stage 2 завершён: problem-first ответы и follow-up continuity - ассистент переведён от entity-heavy логики к problem-first ответам с problem-unit слоем, удержанием контекста в follow-up и очисткой пользовательского ответа от сырых технических ссылок.

2026-03-26 14:53:52 +03:00
parent ece1abed76
commit 96353cfd48
2474 changed files with 21678 additions and 3292445 deletions
@@ -0,0 +1,86 @@
+import request from "supertest";
+import { afterEach, describe, expect, it, vi } from "vitest";
+import {
+  INVESTIGATION_MAX_ACTIVE_PROBLEM_UNITS,
+  INVESTIGATION_MAX_FOCUS_PROBLEM_TYPES,
+  INVESTIGATION_MAX_PROBLEM_UNIT_BACKLINKS,
+  INVESTIGATION_MAX_RESOLVED_PROBLEM_UNITS
+} from "../src/types/stage2ProblemUnits";
+
+const FLAG_KEYS = [
+  "FEATURE_ASSISTANT_INVESTIGATION_STATE_V1",
+  "FEATURE_ASSISTANT_STATE_FOLLOWUP_BINDING_V1",
+  "FEATURE_ASSISTANT_PROBLEM_UNITS_V1",
+  "FEATURE_ASSISTANT_PROBLEM_UNIT_CONTINUITY_V1"
+] as const;
+
+const ORIGINAL_FLAGS: Record<string, string | undefined> = Object.fromEntries(
+  FLAG_KEYS.map((key) => [key, process.env[key]])
+);
+
+function restoreFlags(): void {
+  for (const key of FLAG_KEYS) {
+    const original = ORIGINAL_FLAGS[key];
+    if (original === undefined) {
+      delete process.env[key];
+    } else {
+      process.env[key] = original;
+    }
+  }
+}
+
+async function createAppWithWave4Flags() {
+  process.env.FEATURE_ASSISTANT_INVESTIGATION_STATE_V1 = "1";
+  process.env.FEATURE_ASSISTANT_STATE_FOLLOWUP_BINDING_V1 = "1";
+  process.env.FEATURE_ASSISTANT_PROBLEM_UNITS_V1 = "1";
+  process.env.FEATURE_ASSISTANT_PROBLEM_UNIT_CONTINUITY_V1 = "1";
+  vi.resetModules();
+  const { createApp } = await import("../src/server");
+  return createApp();
+}
+
+describe.sequential("assistant problem-unit continuity state", () => {
+  afterEach(() => {
+    restoreFlags();
+    vi.resetModules();
+  });
+
+  it("stores bounded problem_unit_state in investigation snapshot and session state", async () => {
+    const app = await createAppWithWave4Flags();
+    const sessionId = `asst-wave4-state-${Date.now()}`;
+
+    const first = await request(app).post("/api/assistant/message").send({
+      session_id: sessionId,
+      useMock: true,
+      promptVersion: "normalizer_v2_0_2",
+      user_message: "Разложи цепочку документов и оплат по контрагентам за 2020-06, где есть разрыв закрытия."
+    });
+
+    expect(first.status).toBe(200);
+    expect(first.body.debug?.investigation_state_snapshot?.problem_unit_state).toBeTruthy();
+
+    const second = await request(app).post("/api/assistant/message").send({
+      session_id: sessionId,
+      useMock: true,
+      promptVersion: "normalizer_v2_0_2",
+      user_message: "И по тому же кейсу уточни, что влияет на закрытие периода."
+    });
+
+    expect(second.status).toBe(200);
+    const problemState = second.body.debug?.investigation_state_snapshot?.problem_unit_state;
+    expect(problemState).toBeTruthy();
+    expect(problemState.active_problem_units.length).toBeLessThanOrEqual(INVESTIGATION_MAX_ACTIVE_PROBLEM_UNITS);
+    expect(problemState.resolved_problem_units.length).toBeLessThanOrEqual(INVESTIGATION_MAX_RESOLVED_PROBLEM_UNITS);
+    expect(problemState.problem_unit_backlinks.length).toBeLessThanOrEqual(INVESTIGATION_MAX_PROBLEM_UNIT_BACKLINKS);
+    expect(problemState.focus_problem_types.length).toBeLessThanOrEqual(INVESTIGATION_MAX_FOCUS_PROBLEM_TYPES);
+
+    const sessionResponse = await request(app).get(`/api/assistant/session/${sessionId}`);
+    expect(sessionResponse.status).toBe(200);
+    const sessionProblemState = sessionResponse.body.session?.investigation_state?.problem_unit_state;
+    expect(sessionProblemState).toBeTruthy();
+    expect(sessionProblemState.active_problem_units.length).toBeLessThanOrEqual(INVESTIGATION_MAX_ACTIVE_PROBLEM_UNITS);
+    expect(sessionProblemState.resolved_problem_units.length).toBeLessThanOrEqual(INVESTIGATION_MAX_RESOLVED_PROBLEM_UNITS);
+    expect(sessionProblemState.problem_unit_backlinks.length).toBeLessThanOrEqual(INVESTIGATION_MAX_PROBLEM_UNIT_BACKLINKS);
+    expect(sessionProblemState.focus_problem_types.length).toBeLessThanOrEqual(INVESTIGATION_MAX_FOCUS_PROBLEM_TYPES);
+  });
+});