apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-03 21:02:40 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

feat: unified credits and its cost calculations

Browse source
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2026-05-02 14:34:23 -07:00
parent 451a98936e
commit ae9d36d77f
61 changed files with 5835 additions and 272 deletions
Download patch file Download diff file Expand all files Collapse all files

2
surfsense_web/lib/chat/chat-error-classifier.ts
View file

@ -41,7 +41,7 @@ export interface RawChatErrorInput {
}
export const PREMIUM_QUOTA_ASSISTANT_MESSAGE =
"I can’t continue with the current premium model because your premium tokens are exhausted. Switch to a free model or buy more tokens to continue.";
"I can’t continue with the current premium model because your premium credit is exhausted. Switch to a free model or top up your credit to continue.";
function getErrorMessage(error: unknown): string {
if (error instanceof Error) return error.message;

9
surfsense_web/lib/chat/streaming-state.ts
View file

@ -541,16 +541,23 @@ export type SSEEvent =
data: {
usage: Record<
string,
{ prompt_tokens: number; completion_tokens: number; total_tokens: number }
{
prompt_tokens: number;
completion_tokens: number;
total_tokens: number;
cost_micros?: number;
}
>;
prompt_tokens: number;
completion_tokens: number;
total_tokens: number;
cost_micros?: number;
call_details: Array<{
model: string;
prompt_tokens: number;
completion_tokens: number;
total_tokens: number;
cost_micros?: number;
}>;
};
}

13
surfsense_web/lib/chat/thread-persistence.ts
View file

@ -30,9 +30,20 @@ export interface TokenUsageSummary {
prompt_tokens: number;
completion_tokens: number;
total_tokens: number;
/**
* Total provider USD cost for this assistant turn, in micro-USD
* (1_000_000 = $1.00). Optional because rows persisted before the
* cost-credits migration won't have it.
*/
cost_micros?: number;
model_breakdown?: Record<
string,
{ prompt_tokens: number; completion_tokens: number; total_tokens: number }
{
prompt_tokens: number;
completion_tokens: number;
total_tokens: number;
cost_micros?: number;
}
> | null;
}