139 lines
3.3 KiB
TypeScript
139 lines
3.3 KiB
TypeScript
import { APICallError } from '@ai-sdk/provider';
|
|
import { ResponseHandler } from './response-handler';
|
|
|
|
export const postJsonToApi = async <T>({
|
|
url,
|
|
headers,
|
|
body,
|
|
failedResponseHandler,
|
|
successfulResponseHandler,
|
|
abortSignal,
|
|
}: {
|
|
url: string;
|
|
headers?: Record<string, string | undefined>;
|
|
body: unknown;
|
|
failedResponseHandler: ResponseHandler<APICallError>;
|
|
successfulResponseHandler: ResponseHandler<T>;
|
|
abortSignal?: AbortSignal;
|
|
}) =>
|
|
postToApi({
|
|
url,
|
|
headers: {
|
|
...headers,
|
|
'Content-Type': 'application/json',
|
|
},
|
|
body: {
|
|
content: JSON.stringify(body),
|
|
values: body,
|
|
},
|
|
failedResponseHandler,
|
|
successfulResponseHandler,
|
|
abortSignal,
|
|
});
|
|
|
|
export const postToApi = async <T>({
|
|
url,
|
|
headers = {},
|
|
body,
|
|
successfulResponseHandler,
|
|
failedResponseHandler,
|
|
abortSignal,
|
|
}: {
|
|
url: string;
|
|
headers?: Record<string, string | undefined>;
|
|
body: {
|
|
content: string | FormData | Uint8Array;
|
|
values: unknown;
|
|
};
|
|
failedResponseHandler: ResponseHandler<Error>;
|
|
successfulResponseHandler: ResponseHandler<T>;
|
|
abortSignal?: AbortSignal;
|
|
}) => {
|
|
try {
|
|
// remove undefined headers:
|
|
const definedHeaders = Object.fromEntries(
|
|
Object.entries(headers).filter(([_key, value]) => value != null),
|
|
) as Record<string, string>;
|
|
|
|
const response = await fetch(url, {
|
|
method: 'POST',
|
|
headers: definedHeaders,
|
|
body: body.content,
|
|
signal: abortSignal,
|
|
});
|
|
|
|
if (!response.ok) {
|
|
try {
|
|
throw await failedResponseHandler({
|
|
response,
|
|
url,
|
|
requestBodyValues: body.values,
|
|
});
|
|
} catch (error) {
|
|
if (error instanceof Error) {
|
|
if (
|
|
error.name === 'AbortError' ||
|
|
APICallError.isAPICallError(error)
|
|
) {
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
throw new APICallError({
|
|
message: 'Failed to process error response',
|
|
cause: error,
|
|
statusCode: response.status,
|
|
url,
|
|
requestBodyValues: body.values,
|
|
});
|
|
}
|
|
}
|
|
|
|
try {
|
|
return await successfulResponseHandler({
|
|
response,
|
|
url,
|
|
requestBodyValues: body.values,
|
|
});
|
|
} catch (error) {
|
|
if (error instanceof Error) {
|
|
if (error.name === 'AbortError' || APICallError.isAPICallError(error)) {
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
throw new APICallError({
|
|
message: 'Failed to process successful response',
|
|
cause: error,
|
|
statusCode: response.status,
|
|
url,
|
|
requestBodyValues: body.values,
|
|
});
|
|
}
|
|
} catch (error) {
|
|
if (error instanceof Error) {
|
|
if (error.name === 'AbortError') {
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
// unwrap original error when fetch failed (for easier debugging):
|
|
if (error instanceof TypeError && error.message === 'fetch failed') {
|
|
const cause = (error as any).cause;
|
|
|
|
if (cause != null) {
|
|
// Failed to connect to server:
|
|
throw new APICallError({
|
|
message: `Cannot connect to API: ${cause.message}`,
|
|
cause,
|
|
url,
|
|
requestBodyValues: body.values,
|
|
isRetryable: true, // retry when network error
|
|
});
|
|
}
|
|
}
|
|
|
|
throw error;
|
|
}
|
|
};
|