Files
dobetternorge-tools/api/timeline-stream.php
T
daveadmin f00d3d68e5 Add Quick mode (nova-lite/Bedrock) as 3rd tier for timeline tool
Timeline now offers Quick/Standard/Deep: nova_lite routes to Amazon
Bedrock nova-lite via LiteLLM (1 credit, ~2s faster), azure_mini stays
gpt-4o-mini (1 credit), azure_full stays gpt-4o (2 credits, Pro only).
ToolModels tier rules: free→nova_lite only, plus→quick/standard,
pro→all three.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-05-25 10:26:07 +02:00

122 lines
4.4 KiB
PHP

<?php
declare(strict_types=1);
require_once __DIR__ . '/../includes/LegalTools.php';
require_once __DIR__ . '/../includes/ToolModels.php';
dbnToolsRequireMethod('POST');
dbnToolsRequireAuth();
// Parse input and run credit pre-check BEFORE emitting SSE headers so that
// auth/credit errors can still return JSON (dbnToolsError / dbnToolsAbort).
$input = dbnToolsJsonInput(400000);
$language = dbnToolsNormalizeLanguage($input['language'] ?? 'en');
$_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);
// Only switch to SSE mode once auth + credit checks pass.
header('Content-Type: text/event-stream');
header('Cache-Control: no-cache, no-transform');
header('X-Accel-Buffering: no');
while (ob_get_level() > 0) ob_end_flush();
ob_implicit_flush(true);
function sseEmit(string $event, array $data): void
{
echo "event: {$event}\n";
echo 'data: ' . json_encode($data, JSON_UNESCAPED_UNICODE | JSON_UNESCAPED_SLASHES) . "\n\n";
if (function_exists('flush')) @flush();
}
$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;
}
$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);
$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,
fn(string $msg) => sseEmit('status', ['msg' => $msg])
);
$latency = (int)round((microtime(true) - $start) * 1000);
if ($ftUid > 0) {
$balance = dbnToolsFreeTierDeductAmount($ftUid, 'timeline', $_engineCredits);
$result['balance'] = $balance;
if (!headers_sent()) {
header('X-Credits-Remaining: ' . $balance);
}
}
$result['ok'] = true;
$result['latency_ms'] = $latency;
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,
]);
sseEmit('result', $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,
]);
sseEmit('error', ['code' => $e->errorCode, 'message' => $e->getMessage()]);
} 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-stream error: ' . $e->getMessage());
sseEmit('error', ['code' => 'internal_error', 'message' => 'The tool could not complete this request.']);
}