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
@@ -4,6 +4,7 @@ exports.cloneInvestigationState = cloneInvestigationState;
 exports.createEmptyInvestigationState = createEmptyInvestigationState;
 exports.updateInvestigationState = updateInvestigationState;
 const stage1Contracts_1 = require("../types/stage1Contracts");
+const stage2ProblemUnits_1 = require("../types/stage2ProblemUnits");
 function uniqueStrings(values) {
    return Array.from(new Set(values.map((item) => item.trim()).filter(Boolean)));
 }
@@ -74,10 +75,101 @@ function collectOpenUncertainties(coverageReport, retrievalResults) {
    const limitationNotes = retrievalResults.flatMap((result) => result.limitations).slice(0, 6);
    return capStrings([...requirementNotes, ...limitationNotes], stage1Contracts_1.INVESTIGATION_MAX_UNCERTAINTIES);
 }
+function normalizeEntityBacklinks(values) {
+    const result = [];
+    const seen = new Set();
+    for (const item of values) {
+        const entity = String(item.entity ?? "").trim();
+        const id = String(item.id ?? "").trim();
+        if (!entity || !id) {
+            continue;
+        }
+        const key = `${entity}::${id}`;
+        if (seen.has(key)) {
+            continue;
+        }
+        seen.add(key);
+        result.push({
+            entity,
+            id
+        });
+    }
+    return result;
+}
+function collectProblemUnits(retrievalResults) {
+    return retrievalResults.flatMap((result) => result.problem_units ?? []);
+}
+function capProblemUnitState(state) {
+    return {
+        active_problem_units: capStrings(state.active_problem_units, stage2ProblemUnits_1.INVESTIGATION_MAX_ACTIVE_PROBLEM_UNITS),
+        resolved_problem_units: capStrings(state.resolved_problem_units, stage2ProblemUnits_1.INVESTIGATION_MAX_RESOLVED_PROBLEM_UNITS),
+        problem_unit_backlinks: state.problem_unit_backlinks
+            .map((item) => ({
+            problem_unit_id: String(item.problem_unit_id ?? "").trim(),
+            entity_backlinks: normalizeEntityBacklinks(item.entity_backlinks ?? [])
+        }))
+            .filter((item) => Boolean(item.problem_unit_id) && item.entity_backlinks.length > 0)
+            .slice(0, stage2ProblemUnits_1.INVESTIGATION_MAX_PROBLEM_UNIT_BACKLINKS),
+        focus_problem_types: capStrings(state.focus_problem_types.map((item) => String(item)), stage2ProblemUnits_1.INVESTIGATION_MAX_FOCUS_PROBLEM_TYPES)
+    };
+}
+function updateProblemUnitState(previous, retrievalResults) {
+    const previousState = previous.problem_unit_state;
+    const currentProblemUnits = collectProblemUnits(retrievalResults);
+    const currentIds = capStrings(currentProblemUnits.map((item) => String(item.problem_unit_id ?? "")), stage2ProblemUnits_1.INVESTIGATION_MAX_ACTIVE_PROBLEM_UNITS);
+    const currentTypes = capStrings(currentProblemUnits.map((item) => String(item.problem_unit_type ?? "")), stage2ProblemUnits_1.INVESTIGATION_MAX_FOCUS_PROBLEM_TYPES);
+    const currentBacklinksRaw = currentProblemUnits
+        .filter((item) => currentIds.includes(item.problem_unit_id))
+        .map((item) => ({
+        problem_unit_id: item.problem_unit_id,
+        entity_backlinks: normalizeEntityBacklinks(item.entity_backlinks ?? [])
+    }))
+        .filter((item) => item.entity_backlinks.length > 0);
+    const currentBacklinksById = new Map(currentBacklinksRaw.map((item) => [item.problem_unit_id, item.entity_backlinks]));
+    const previousBacklinksById = new Map((previousState?.problem_unit_backlinks ?? []).map((item) => [item.problem_unit_id, item.entity_backlinks]));
+    const active_problem_units = currentIds.length > 0
+        ? currentIds
+        : capStrings(previousState?.active_problem_units ?? [], stage2ProblemUnits_1.INVESTIGATION_MAX_ACTIVE_PROBLEM_UNITS);
+    const resolved_problem_units = currentIds.length > 0
+        ? capStrings([
+            ...(previousState?.active_problem_units ?? []).filter((item) => !currentIds.includes(item)),
+            ...(previousState?.resolved_problem_units ?? [])
+        ], stage2ProblemUnits_1.INVESTIGATION_MAX_RESOLVED_PROBLEM_UNITS)
+        : capStrings(previousState?.resolved_problem_units ?? [], stage2ProblemUnits_1.INVESTIGATION_MAX_RESOLVED_PROBLEM_UNITS);
+    const problem_unit_backlinks = active_problem_units
+        .map((problemUnitId) => {
+        const entity_backlinks = normalizeEntityBacklinks(currentBacklinksById.get(problemUnitId) ?? previousBacklinksById.get(problemUnitId) ?? []);
+        if (entity_backlinks.length === 0) {
+            return null;
+        }
+        return {
+            problem_unit_id: problemUnitId,
+            entity_backlinks
+        };
+    })
+        .filter((item) => item !== null)
+        .slice(0, stage2ProblemUnits_1.INVESTIGATION_MAX_PROBLEM_UNIT_BACKLINKS);
+    const focus_problem_types = currentTypes.length > 0
+        ? currentTypes
+        : capStrings((previousState?.focus_problem_types ?? []).map((item) => String(item)), stage2ProblemUnits_1.INVESTIGATION_MAX_FOCUS_PROBLEM_TYPES);
+    const nextState = capProblemUnitState({
+        active_problem_units,
+        resolved_problem_units,
+        problem_unit_backlinks,
+        focus_problem_types
+    });
+    if (nextState.active_problem_units.length === 0 &&
+        nextState.resolved_problem_units.length === 0 &&
+        nextState.problem_unit_backlinks.length === 0 &&
+        nextState.focus_problem_types.length === 0) {
+        return undefined;
+    }
+    return nextState;
+}
 function cloneInvestigationState(state) {
    if (!state)
        return null;
-    return {
+    const cloned = {
        ...state,
        focus: {
            ...state.focus,
@@ -92,6 +184,18 @@ function cloneInvestigationState(state) {
            }
            : null
    };
+    if (state.problem_unit_state) {
+        cloned.problem_unit_state = capProblemUnitState({
+            active_problem_units: [...state.problem_unit_state.active_problem_units],
+            resolved_problem_units: [...state.problem_unit_state.resolved_problem_units],
+            problem_unit_backlinks: state.problem_unit_state.problem_unit_backlinks.map((item) => ({
+                problem_unit_id: item.problem_unit_id,
+                entity_backlinks: [...item.entity_backlinks]
+            })),
+            focus_problem_types: [...state.problem_unit_state.focus_problem_types]
+        });
+    }
+    return cloned;
 }
 function createEmptyInvestigationState(sessionId, timestamp = new Date().toISOString()) {
    return {
@@ -120,6 +224,7 @@ function updateInvestigationState(input) {
    const focusFromMessage = capStrings(detectAccounts(input.userMessage), stage1Contracts_1.INVESTIGATION_MAX_PRIMARY_ACCOUNTS);
    const requirementIds = capStrings(input.requirements.map((item) => item.requirement_id), stage1Contracts_1.INVESTIGATION_MAX_REQUIREMENT_LINKS);
    const mainRequirement = input.requirements[0]?.requirement_text ?? input.userMessage;
+    const problemUnitState = updateProblemUnitState(previous, input.retrievalResults);
    return {
        schema_version: stage1Contracts_1.INVESTIGATION_STATE_SCHEMA_VERSION,
        session_id: previous.session_id,
@@ -142,6 +247,11 @@ function updateInvestigationState(input) {
            last_user_message: input.userMessage.slice(0, 240),
            referenced_requirement_ids: requirementIds
        },
-        query_mode_hint: deriveQueryModeHint(input.routeSummary)
+        query_mode_hint: deriveQueryModeHint(input.routeSummary),
+        ...(problemUnitState
+            ? {
+                problem_unit_state: problemUnitState
+            }
+            : {})
    };
 }