daveadmin/dobetternorge-tools
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add timeline engine size routing

Browse Source
This commit is contained in:
davegilligan
2026-05-25 11:33:47 +02:00
parent 3ad8f4843c
commit 75b19f1dcf
4 changed files with 208 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/timeline-stream.php
+49 -21
View File
@@ -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',
api/timeline.php
+80 -23
View File
@@ -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');
}