apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-24 21:38:09 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat(error-handling): implement LLM error adaptation and classification for chat streaming

Browse source 
- Introduced LLMErrorCategory and adapt_llm_exception to normalize LLM exceptions.
- Updated llm_retryable_message and llm_permanent_message to utilize the new adaptation logic.
- Enhanced classify_stream_exception to classify provider errors and return user-friendly messages.
- Added tests for error classification and adaptation to ensure robustness.
- Updated frontend error handling to display appropriate messages based on new classifications.
This commit is contained in:
Anish Sarkar 2026-06-12 05:03:14 +05:30
parent 203ef78346
commit 8e8cf96faa
9 changed files with 533 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -5,6 +5,10 @@ export type ChatErrorKind =
| "thread_busy"
| "send_failed_pre_accept"
| "auth_expired"
| "model_auth_failed"
| "model_not_found"
| "model_context_limit"
| "model_provider_unavailable"
| "rate_limited"
| "network_offline"
| "stream_interrupted"
@ -14,7 +18,7 @@ export type ChatErrorKind =
| "server_error"
| "unknown";
export type ChatErrorChannel = "pinned_inline" | "toast" | "silent";
export type ChatErrorChannel = "pinned_inline" | "inline" | "toast" | "silent";
export type ChatTelemetryEvent = "chat_blocked" | "chat_error";
export type ChatErrorSeverity = "info" | "warn" | "error";
@ -206,6 +210,66 @@ export function classifyChatError(input: RawChatErrorInput): NormalizedChatError
};
}
if (errorCode === "MODEL_AUTH_FAILED") {
return {
kind: "model_auth_failed",
channel: "toast",
severity: "warn",
telemetryEvent: "chat_blocked",
isExpected: true,
userMessage:
"This model’s API key is invalid or expired. Switch models, or update the API key.",
rawMessage,
errorCode: errorCode ?? "MODEL_AUTH_FAILED",
details: { flow: input.flow, providerErrorType },
};
}
if (errorCode === "MODEL_NOT_FOUND") {
return {
kind: "model_not_found",
channel: "toast",
severity: "warn",
telemetryEvent: "chat_blocked",
isExpected: true,
userMessage:
"This model is unavailable or no longer exists. Switch to another model and try again.",
rawMessage,
errorCode: errorCode ?? "MODEL_NOT_FOUND",
details: { flow: input.flow, providerErrorType },
};
}
if (errorCode === "MODEL_CONTEXT_LIMIT") {
return {
kind: "model_context_limit",
channel: "toast",
severity: "warn",
telemetryEvent: "chat_blocked",
isExpected: true,
userMessage:
"This request is too large for the selected model. Reduce the input or switch models.",
rawMessage,
errorCode: errorCode ?? "MODEL_CONTEXT_LIMIT",
details: { flow: input.flow, providerErrorType },
};
}
if (errorCode === "MODEL_PROVIDER_UNAVAILABLE") {
return {
kind: "model_provider_unavailable",
channel: "toast",
severity: "warn",
telemetryEvent: "chat_blocked",
isExpected: true,
userMessage:
"The selected model provider is temporarily unavailable. Please try again or switch models.",
rawMessage,
errorCode: errorCode ?? "MODEL_PROVIDER_UNAVAILABLE",
details: { flow: input.flow, providerErrorType },
};
}
if (errorCode === "RATE_LIMITED" || providerTypeNormalized === "rate_limit_error") {
return {
kind: "rate_limited",

4
surfsense_web/lib/chat/chat-request-errors.ts
View file

@ -91,6 +91,10 @@ export function tagPreAcceptSendFailure(error: unknown): unknown {
"TURN_CANCELLING",
"AUTH_EXPIRED",
"UNAUTHORIZED",
"MODEL_AUTH_FAILED",
"MODEL_NOT_FOUND",
"MODEL_CONTEXT_LIMIT",
"MODEL_PROVIDER_UNAVAILABLE",
"RATE_LIMITED",
"NETWORK_ERROR",
"STREAM_PARSE_ERROR",