АДРЕСНЫЙ РЕЖИМ - ллм декомпоз

llm_normalizer/backend/src/services/address_runtime/decomposeStage.ts

@@ -1,4 +1,4 @@
-import type {
+import type {
   AddressFilterSet,
   AddressIntent,
   AddressIntentResolution,
@@ -45,7 +45,14 @@ function toNonEmptyString(value: unknown): string | null {
 }
 
 function hasAllTimeHint(text: string): boolean {
-  return /(?:за\s+вс[её]\s+время|за\s+весь\s+период|за\s+всю\s+истори(?:ю|и)|за\s+любой\s+период|for\s+all\s+time|all\s+time|for\s+entire\s+period|entire\s+period|for\s+any\s+period|any\s+period|for\s+full\s+history|full\s+history)/iu.test(
+  const normalized = String(text ?? "");
+  return /(?:за\s+вс[её]\s+время|за\s+весь\s+период|за\s+весь\s+срок|за\s+всю\s+истори(?:ю|и)|за\s+любой\s+период|за\s+любой\s+срок|for\s+all\s+time|all\s+time|for\s+entire\s+period|entire\s+period|for\s+any\s+period|any\s+period|for\s+full\s+history|full\s+history)/iu.test(
+    normalized
+  );
+}
+
+function hasSameDateHint(text: string): boolean {
+  return /(?:на\s+ту\s+же\s+дат[ауеы]|на\s+эту\s+же\s+дат[ауеы]|та\s+же\s+дата|same\s+date|as\s+of\s+same\s+date|the\s+same\s+date)/iu.test(
     String(text ?? "")
   );
 }
@@ -58,10 +65,23 @@ export function hasAddressFollowupContextSignal(text: string): boolean {
   if (hasAllTimeHint(normalized)) {
     return true;
   }
-  if (/(?:^|\s)(?:и|а\s+еще|а\s+ещё|еще|ещё|также|по\s+этому|по\s+тому|это\s+же|в\s+этом|тот\s+же|also|same|that)/iu.test(normalized)) {
+  if (
+    /(?:^|\s)(?:и|а\s+еще|а\s+ещё|еще|ещё|также|а\s+теперь|теперь|по\s+этому|по\s+тому|это\s+же|в\s+этом|тот\s+же|also|same|that|then|now)/iu.test(
+      normalized
+    )
+  ) {
     return true;
   }
-  return normalized.split(/\s+/).filter(Boolean).length <= 8;
+  if (hasSameDateHint(normalized)) {
+    return true;
+  }
+
+  const tokenCount = normalized.split(/\s+/).filter(Boolean).length;
+  const hasPeriodLiteral = /\b(?:19|20)\d{2}(?:[./-](?:0?[1-9]|1[0-2]))?\b/.test(normalized);
+  if (tokenCount <= 8 && hasPeriodLiteral) {
+    return true;
+  }
+  return tokenCount <= 6;
 }
 
 function mergeFollowupFilters(
@@ -81,7 +101,11 @@ function mergeFollowupFilters(
   const previousCounterparty = toNonEmptyString(previous.counterparty);
   const previousContract = toNonEmptyString(previous.contract);
   const previousAccount = toNonEmptyString(previous.account);
+  const previousAsOfDate = toNonEmptyString(previous.as_of_date);
+  const previousPeriodFrom = toNonEmptyString(previous.period_from);
+  const previousPeriodTo = toNonEmptyString(previous.period_to);
   const allTimeRequested = hasAllTimeHint(userMessage);
+  const sameDateRequested = hasSameDateHint(userMessage);
 
   if (intent === "list_documents_by_counterparty" || intent === "bank_operations_by_counterparty") {
     if (!toNonEmptyString(merged.counterparty)) {
@@ -97,21 +121,24 @@ function mergeFollowupFilters(
 
   if (intent === "account_balance_snapshot" || intent === "documents_forming_balance") {
     if (!toNonEmptyString(merged.account)) {
-      const inheritedAccount =
-        previousAccount ??
-        (followupContext.previous_anchor_type === "account" ? previousAnchorValue : null);
+      const inheritedAccount = previousAccount ?? (followupContext.previous_anchor_type === "account" ? previousAnchorValue : null);
       if (inheritedAccount) {
         merged.account = inheritedAccount;
         reasons.push("account_from_followup_context");
       }
     }
+    if (sameDateRequested) {
+      const inheritedAsOfDate = previousAsOfDate ?? previousPeriodTo ?? previousPeriodFrom;
+      if (inheritedAsOfDate && merged.as_of_date !== inheritedAsOfDate) {
+        merged.as_of_date = inheritedAsOfDate;
+        reasons.push("as_of_date_from_followup_context");
+      }
+    }
   }
 
   if (intent === "open_items_by_counterparty_or_contract" || intent === "list_open_contracts") {
     if (!toNonEmptyString(merged.contract)) {
-      const inheritedContract =
-        previousContract ??
-        (followupContext.previous_anchor_type === "contract" ? previousAnchorValue : null);
+      const inheritedContract = previousContract ?? (followupContext.previous_anchor_type === "contract" ? previousAnchorValue : null);
       if (inheritedContract) {
         merged.contract = inheritedContract;
         reasons.push("contract_from_followup_context");
@@ -140,11 +167,11 @@ function mergeFollowupFilters(
   const currentHasPeriod = hasExplicitPeriodWindow(merged);
   const previousHasPeriod = hasExplicitPeriodWindow(previous);
   if (!currentHasPeriod && previousHasPeriod && hasAddressFollowupContextSignal(userMessage)) {
-    if (toNonEmptyString(previous.period_from)) {
-      merged.period_from = previous.period_from;
+    if (previousPeriodFrom) {
+      merged.period_from = previousPeriodFrom;
     }
-    if (toNonEmptyString(previous.period_to)) {
-      merged.period_to = previous.period_to;
+    if (previousPeriodTo) {
+      merged.period_to = previousPeriodTo;
     }
     reasons.push("period_from_followup_context");
   }
@@ -241,4 +268,3 @@ export function runAddressDecomposeStage(
     baseReasons
   };
 }
-