diff --git a/api/timeline-stream.php b/api/timeline-stream.php index 7e50c70..49325de 100644 --- a/api/timeline-stream.php +++ b/api/timeline-stream.php @@ -13,11 +13,33 @@ $input = dbnToolsJsonInput(400000); $language = dbnToolsNormalizeLanguage($input['language'] ?? 'en'); $_validEngines = ['nova_lite', 'azure_mini', 'azure_full']; -$_engine = in_array((string)($input['engine'] ?? ''), $_validEngines, true) +$_requestedEngine = in_array((string)($input['engine'] ?? ''), $_validEngines, true) ? (string)$input['engine'] : 'azure_mini'; -$_engineCredits = $_engine === 'azure_full' ? 2 : 1; -$ftUid = dbnToolsFreeTierCheckAmount('timeline', $_engineCredits); +try { + $text = dbnToolsInjectDocContent($input, dbnToolsString($input, 'text', 128000, false)); + if (mb_strlen(trim($text), 'UTF-8') < 10) { + dbnToolsError('Paste text, upload a file, or select a document before running.', 422, 'empty_text'); + } + + $ftUid = dbnToolsIsFreeTier() ? (int)($_SESSION['dbn_tools_sso_uid'] ?? 0) : 0; + + $useMyCase = !empty($input['use_my_case']); + if ($useMyCase) { + $caseBlock = dbnToolsCaseContext(true, $text, 5); + if ($caseBlock !== '') { + $text = $text . "\n\n" . $caseBlock; + } + } + + $timelineRoute = ToolModels::timelineRoute($ftUid, $_requestedEngine, $text); + $ftUid = dbnToolsFreeTierCheckAmount('timeline', (int)$timelineRoute['credits']); +} catch (DbnToolsHttpException $e) { + dbnToolsError($e->getMessage(), $e->status, $e->errorCode, $e->extra); +} catch (Throwable $e) { + error_log('timeline-stream preparation error: ' . $e->getMessage()); + dbnToolsError('The tool could not prepare this request.', 500, 'internal_error'); +} // Only switch to SSE mode once auth + credit checks pass. header('Content-Type: text/event-stream'); @@ -36,17 +58,18 @@ function sseEmit(string $event, array $data): void $start = microtime(true); try { - $text = dbnToolsInjectDocContent($input, dbnToolsString($input, 'text', 128000, false)); - if (mb_strlen(trim($text), 'UTF-8') < 10) { - sseEmit('error', ['code' => 'empty_text', 'message' => 'Paste text, upload a file, or select a document before running.']); - exit; + $engine = (string)$timelineRoute['effective_engine']; + if (!empty($timelineRoute['auto_upgraded_engine'])) { + $label = match ($engine) { + 'azure_full' => 'Deep', + 'azure_mini' => 'Standard', + default => 'Quick', + }; + sseEmit('status', [ + 'msg' => 'This input is ' . number_format((int)$timelineRoute['input_char_count']) . " characters, so Timeline is using {$label} for reliability.", + ]); } - $validEngines = ['nova_lite', 'azure_mini', 'azure_full']; - $engine = in_array((string)($input['engine'] ?? ''), $validEngines, true) - ? (string)$input['engine'] : 'azure_mini'; - $engine = ToolModels::engineForUser($ftUid, $engine); - $validFocus = ['all', 'deadlines', 'hearings', 'cps']; $focus = in_array((string)($input['focus'] ?? ''), $validFocus, true) ? (string)$input['focus'] : 'all'; @@ -58,14 +81,6 @@ try { $includeBackground = ($input['include_background'] ?? true) !== false; $userNotes = dbnToolsString($input, 'user_notes', 2000, false); - $useMyCase = !empty($input['use_my_case']); - if ($useMyCase) { - $caseBlock = dbnToolsCaseContext(true, $text, 5); - if ($caseBlock !== '') { - $text = $text . "\n\n" . $caseBlock; - } - } - $result = (new DbnLegalToolsService())->timeline( $text, $language, $engine, $focus, $confidenceFilter, $includeRelative, $includeBackground, $userNotes, @@ -75,7 +90,12 @@ try { $latency = (int)round((microtime(true) - $start) * 1000); if ($ftUid > 0) { - $balance = dbnToolsFreeTierDeductAmount($ftUid, 'timeline', $_engineCredits); + $balance = dbnToolsFreeTierDeductAmount($ftUid, 'timeline', (int)$timelineRoute['credits'], [ + 'requested_engine' => $timelineRoute['requested_engine'], + 'effective_engine' => $timelineRoute['effective_engine'], + 'auto_upgraded_engine' => $timelineRoute['auto_upgraded_engine'], + 'input_char_count' => $timelineRoute['input_char_count'], + ]); $result['balance'] = $balance; if (!headers_sent()) { header('X-Credits-Remaining: ' . $balance); @@ -84,6 +104,14 @@ try { $result['ok'] = true; $result['latency_ms'] = $latency; + $result['trace_metadata'] = array_merge($result['trace_metadata'] ?? [], [ + 'requested_engine' => $timelineRoute['requested_engine'], + 'effective_engine' => $timelineRoute['effective_engine'], + 'auto_upgraded_engine' => $timelineRoute['auto_upgraded_engine'], + 'input_char_count' => $timelineRoute['input_char_count'], + 'engine_limit_chars' => $timelineRoute['engine_limit_chars'], + 'credits_charged' => $timelineRoute['credits'], + ]); dbnToolsLogMetadata([ 'tool' => 'timeline', diff --git a/api/timeline.php b/api/timeline.php index b2c9f29..6ef027c 100644 --- a/api/timeline.php +++ b/api/timeline.php @@ -7,34 +7,20 @@ require_once __DIR__ . '/../includes/ToolModels.php'; dbnToolsRequireMethod('POST'); dbnToolsRequireAuth(); $input = dbnToolsJsonInput(400000); -$_validEngines = ['nova_lite', 'azure_mini', 'azure_full']; -$_engine = in_array((string)($input['engine'] ?? ''), $_validEngines, true) - ? (string)$input['engine'] : 'azure_mini'; -$_engineCredits = $_engine === 'azure_full' ? 2 : 1; -$ftUid = dbnToolsFreeTierCheckAmount('timeline', $_engineCredits); $language = dbnToolsNormalizeLanguage($input['language'] ?? 'en'); +$_validEngines = ['nova_lite', 'azure_mini', 'azure_full']; +$_requestedEngine = in_array((string)($input['engine'] ?? ''), $_validEngines, true) + ? (string)$input['engine'] : 'azure_mini'; -dbnToolsWithChargedTelemetry('timeline', $language, $ftUid, function () use ($input, $language, $ftUid): array { +$start = microtime(true); + +try { $text = dbnToolsInjectDocContent($input, dbnToolsString($input, 'text', 128000, false)); if (mb_strlen(trim($text), 'UTF-8') < 10) { dbnToolsAbort('Paste text, upload a file, or select a document before running.', 422, 'empty_text'); } - $validEngines = ['nova_lite', 'azure_mini', 'azure_full']; - $engine = in_array((string)($input['engine'] ?? ''), $validEngines, true) - ? (string)$input['engine'] : 'azure_mini'; - $engine = ToolModels::engineForUser($ftUid, $engine); - - $validFocus = ['all', 'deadlines', 'hearings', 'cps']; - $focus = in_array((string)($input['focus'] ?? ''), $validFocus, true) - ? (string)$input['focus'] : 'all'; - - $confidenceFilter = (string)($input['confidence_filter'] ?? '') === 'high_medium' - ? 'high_medium' : 'all'; - - $includeRelative = ($input['include_relative'] ?? true) !== false; - $includeBackground = ($input['include_background'] ?? true) !== false; - $userNotes = dbnToolsString($input, 'user_notes', 2000, false); + $ftUid = dbnToolsIsFreeTier() ? (int)($_SESSION['dbn_tools_sso_uid'] ?? 0) : 0; // Optional: prepend the user's case-context chunks so the timeline includes events // referenced in their uploaded case documents. @@ -46,7 +32,78 @@ dbnToolsWithChargedTelemetry('timeline', $language, $ftUid, function () use ($in } } + $timelineRoute = ToolModels::timelineRoute($ftUid, $_requestedEngine, $text); + $ftUid = dbnToolsFreeTierCheckAmount('timeline', (int)$timelineRoute['credits']); + + $validFocus = ['all', 'deadlines', 'hearings', 'cps']; + $focus = in_array((string)($input['focus'] ?? ''), $validFocus, true) + ? (string)$input['focus'] : 'all'; + + $confidenceFilter = (string)($input['confidence_filter'] ?? '') === 'high_medium' + ? 'high_medium' : 'all'; + + $includeRelative = ($input['include_relative'] ?? true) !== false; + $includeBackground = ($input['include_background'] ?? true) !== false; + $userNotes = dbnToolsString($input, 'user_notes', 2000, false); + $engine = (string)$timelineRoute['effective_engine']; + $result = (new DbnLegalToolsService())->timeline($text, $language, $engine, $focus, $confidenceFilter, $includeRelative, $includeBackground, $userNotes, null); - return $result; -}, $_engineCredits); + if ($ftUid > 0) { + $balance = dbnToolsFreeTierDeductAmount($ftUid, 'timeline', (int)$timelineRoute['credits'], [ + 'requested_engine' => $timelineRoute['requested_engine'], + 'effective_engine' => $timelineRoute['effective_engine'], + 'auto_upgraded_engine' => $timelineRoute['auto_upgraded_engine'], + 'input_char_count' => $timelineRoute['input_char_count'], + ]); + if ($balance >= 0 && !headers_sent()) { + header('X-Credits-Remaining: ' . $balance); + } + $result['balance'] = $balance; + } + + $latency = (int)round((microtime(true) - $start) * 1000); + $result['ok'] = $result['ok'] ?? true; + $result['latency_ms'] = $latency; + $result['trace_metadata'] = array_merge($result['trace_metadata'] ?? [], [ + 'requested_engine' => $timelineRoute['requested_engine'], + 'effective_engine' => $timelineRoute['effective_engine'], + 'auto_upgraded_engine' => $timelineRoute['auto_upgraded_engine'], + 'input_char_count' => $timelineRoute['input_char_count'], + 'engine_limit_chars' => $timelineRoute['engine_limit_chars'], + 'credits_charged' => $timelineRoute['credits'], + ]); + + dbnToolsLogMetadata([ + 'tool' => 'timeline', + 'language' => $language, + 'ok' => true, + 'latency_ms' => $latency, + 'chunk_count' => (int)($result['trace_metadata']['chunk_count'] ?? 0), + 'source_count' => (int)($result['trace_metadata']['source_count'] ?? 0), + 'deployment' => $result['trace_metadata']['deployment'] ?? null, + ]); + + dbnToolsRespond($result); +} catch (DbnToolsHttpException $e) { + $latency = (int)round((microtime(true) - $start) * 1000); + dbnToolsLogMetadata([ + 'tool' => 'timeline', + 'language' => $language, + 'ok' => false, + 'latency_ms' => $latency, + 'error_code' => $e->errorCode, + ]); + dbnToolsError($e->getMessage(), $e->status, $e->errorCode, $e->extra); +} catch (Throwable $e) { + $latency = (int)round((microtime(true) - $start) * 1000); + dbnToolsLogMetadata([ + 'tool' => 'timeline', + 'language' => $language, + 'ok' => false, + 'latency_ms' => $latency, + 'error_code' => 'internal_error', + ]); + error_log('timeline error: ' . $e->getMessage()); + dbnToolsError('The tool could not complete this request.', 500, 'internal_error'); +} diff --git a/assets/js/tools.js b/assets/js/tools.js index 39b6c1a..f5ffe2b 100644 --- a/assets/js/tools.js +++ b/assets/js/tools.js @@ -755,6 +755,21 @@ function currentTimelineEngine() { return document.querySelector('input[name="timelineEngine"]:checked')?.value || 'azure_mini'; } +function timelineEngineLabel(engine) { + return { + nova_lite: 'Quick', + azure_mini: 'Standard', + azure_full: 'Deep', + }[engine] || 'Timeline'; +} + +function timelineClientRoute(engine, charCount) { + let effective = engine; + if (charCount > 55000) effective = 'azure_full'; + else if (charCount > 25000 && effective === 'nova_lite') effective = 'azure_mini'; + return { effective, upgraded: effective !== engine }; +} + function currentTimelineFocus() { return document.querySelector('input[name="timelineFocus"]:checked')?.value || 'all'; } @@ -1104,14 +1119,19 @@ async function runTool(event) { payload.redact_types = currentRedactTypes(); lastRedactPayload = { ...payload }; } + let timelineRouteNotice = ''; if (state.activeTool === 'timeline') { payload.engine = currentTimelineEngine(); + const clientRoute = timelineClientRoute(payload.engine, text.length); payload.focus = currentTimelineFocus(); payload.confidence_filter = currentConfidenceFilter(); payload.include_relative = currentIncludeRelative(); payload.include_background = currentIncludeBackground(); payload.user_notes = (document.getElementById('timelineNotes')?.value || '').trim(); payload.use_my_case = (typeof window.dbnGetUseMyCase === 'function') ? window.dbnGetUseMyCase() : false; + timelineRouteNotice = clientRoute.upgraded + ? `This input is ${text.length.toLocaleString()} characters, so Timeline will use ${timelineEngineLabel(clientRoute.effective)} for reliability.` + : ''; } lastToolPayload = { ...payload }; @@ -1121,7 +1141,8 @@ async function runTool(event) { els.results.innerHTML = '
Redacting document…
Building timeline…
${escapeHtml(timelineRouteNotice)}
` : ''; + els.results.innerHTML = `Building timeline…
${routeNotice}