|
1 |
| -import { APICallError } from 'ai' |
| 1 | +import { APICallError, RetryError } from 'ai' |
| 2 | + |
| 3 | +function handleApiCallError(operation: string, error: APICallError) { |
| 4 | + let message = error.message |
| 5 | + if (error.statusCode) message += ` (${error.statusCode})` |
| 6 | + if (error.cause) message += `\nCause: ${error.cause}` |
| 7 | + if (error.responseBody) message += `\nResponse: ${error.responseBody}` |
| 8 | + if (error.url) message += `\nURL: ${error.url}` |
| 9 | + |
| 10 | + console.error(`[${operation}]`, error, { |
| 11 | + statusCode: error.statusCode, |
| 12 | + response: error.responseBody, |
| 13 | + cause: error.cause, |
| 14 | + stack: error.stack, |
| 15 | + isRetryable: error.isRetryable, |
| 16 | + url: error.url, |
| 17 | + }) |
| 18 | + throw new Error(message) |
| 19 | +} |
| 20 | + |
| 21 | +function handleRetryError(operation: string, error: RetryError) { |
| 22 | + if (APICallError.isInstance(error.lastError)) { |
| 23 | + handleApiCallError(operation, error.lastError) |
| 24 | + } |
| 25 | + let message = error.message |
| 26 | + if (error.cause) message += `\nCause: ${error.cause}` |
| 27 | + if (error.stack) message += `\nStack: ${error.stack}` |
| 28 | + if (error.reason) message += `\nReason: ${error.reason}` |
| 29 | + |
| 30 | + console.error(`[${operation}]`, error, { |
| 31 | + cause: error.cause, |
| 32 | + stack: error.stack, |
| 33 | + lastError: error.lastError, |
| 34 | + reason: error.reason, |
| 35 | + errors: error.errors, |
| 36 | + }) |
| 37 | + throw new Error(message) |
| 38 | +} |
2 | 39 |
|
3 | 40 | /**
|
4 | 41 | * Parse an error thrown by the AI SDK, and re-throw it with a human-readable message
|
5 | 42 | */
|
6 | 43 | export function throwAiError(operation: string, error: unknown) {
|
7 | 44 | if (APICallError.isInstance(error)) {
|
8 |
| - let message = error.message |
9 |
| - if (error.statusCode) message += ` (${error.statusCode})` |
10 |
| - if (error.cause) message += `\nCause: ${error.cause}` |
11 |
| - if (error.responseBody) message += `\nResponse: ${error.responseBody}` |
12 |
| - if (error.url) message += `\nURL: ${error.url}` |
13 |
| - |
14 |
| - console.error(`[${operation}]`, error, { |
15 |
| - statusCode: error.statusCode, |
16 |
| - response: error.responseBody, |
17 |
| - cause: error.cause, |
18 |
| - stack: error.stack, |
19 |
| - isRetryable: error.isRetryable, |
20 |
| - url: error.url, |
21 |
| - }) |
22 |
| - throw new Error(message) |
| 45 | + handleApiCallError(operation, error) |
| 46 | + } else if (RetryError.isInstance(error)) { |
| 47 | + handleRetryError(operation, error) |
23 | 48 | } else {
|
24 | 49 | console.error(`[${operation}]`, error)
|
25 | 50 | }
|
|
0 commit comments