35 lines
1.1 KiB
TypeScript
35 lines
1.1 KiB
TypeScript
|
import { extractBytes, initWasm } from "@kreuzberg/wasm";
|
||
|
|
|
||
|
|
async function extractWithRetry(bytes: Uint8Array, mimeType: string, maxRetries: number = 3) {
|
||
|
|
await initWasm();
|
||
|
|
|
||
|
|
let lastError: Error | null = null;
|
||
|
|
|
||
|
|
for (let attempt = 1; attempt <= maxRetries; attempt++) {
|
||
|
|
try {
|
||
|
|
console.log(`Extraction attempt ${attempt}/${maxRetries}`);
|
||
|
|
const result = await extractBytes(bytes, mimeType);
|
||
|
|
console.log("Extraction successful");
|
||
|
|
return result;
|
||
|
|
} catch (error) {
|
||
|
|
lastError = error instanceof Error ? error : new Error(String(error));
|
||
|
|
console.warn(`Attempt ${attempt} failed:`, lastError.message);
|
||
|
|
|
||
|
|
if (attempt < maxRetries) {
|
||
|
|
const delay = 2 ** attempt * 100;
|
||
|
|
console.log(`Retrying in ${delay}ms...`);
|
||
|
|
await new Promise((resolve) => setTimeout(resolve, delay));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
throw new Error(`Extraction failed after ${maxRetries} attempts: ${lastError?.message}`);
|
||
|
|
}
|
||
|
|
|
||
|
|
extractWithRetry(
|
||
|
|
new Uint8Array(await fetch("doc.pdf").then((r) => r.arrayBuffer())),
|
||
|
|
"application/pdf",
|
||
|
|
)
|
||
|
|
.then((r) => console.log("Final result:", r))
|
||
|
|
.catch(console.error);
|