51 lines
1.8 KiB
TypeScript
51 lines
1.8 KiB
TypeScript
import type { ZObject, Bundle, MutableRequest, ZResponse } from './types';
|
|
|
|
// Prepend the configured base URL and inject the auth header on every request.
|
|
export const includeApiKey = (
|
|
request: MutableRequest,
|
|
_z: ZObject,
|
|
bundle: Bundle,
|
|
): MutableRequest => {
|
|
if (bundle.authData && bundle.authData.apiKey) {
|
|
request.headers = request.headers || {};
|
|
request.headers.Authorization = `Token ${bundle.authData.apiKey}`;
|
|
}
|
|
// Resolve relative URLs against the configured base URL.
|
|
if (request.url && request.url.startsWith('/')) {
|
|
const base = (bundle.authData && bundle.authData.baseUrl) || 'https://api.mem0.ai';
|
|
request.url = `${base.replace(/\/$/, '')}${request.url}`;
|
|
}
|
|
return request;
|
|
};
|
|
|
|
// Surface HTTP failures as errors. z.request does NOT throw on non-2xx by
|
|
// default, so without this a 4xx/5xx would flow downstream as a fake success
|
|
// (empty search results / error body returned as a created memory).
|
|
export const handleBadResponses = (
|
|
response: ZResponse,
|
|
z: ZObject,
|
|
_bundle: Bundle,
|
|
): ZResponse => {
|
|
if (response.status === 401 || response.status === 403) {
|
|
throw new z.errors.Error(
|
|
'The Mem0 API key you supplied is invalid or lacks access.',
|
|
'AuthenticationError',
|
|
response.status,
|
|
);
|
|
}
|
|
if (response.status >= 400) {
|
|
const data = (response.data as Record<string, unknown>) || {};
|
|
const raw =
|
|
data.detail || data.error || data.message || response.content || 'unknown error';
|
|
// DRF sometimes returns detail as an object/array; stringify so the
|
|
// thrown message never renders as "[object Object]".
|
|
const detail = typeof raw === 'string' ? raw : JSON.stringify(raw);
|
|
throw new z.errors.Error(
|
|
`Mem0 API request failed (HTTP ${response.status}): ${detail}`,
|
|
'Mem0ApiError',
|
|
response.status,
|
|
);
|
|
}
|
|
return response;
|
|
};
|