forked from bai/curriculum-project-hub
hongjr03
5e10419fc8
Co-authored-by: Hong Jiarong <me@jrhim.com> Co-committed-by: Hong Jiarong <me@jrhim.com>
90 lines
2.9 KiB
TypeScript
90 lines
2.9 KiB
TypeScript
/**
|
|
* Smoke test: real Aliyun docmind API call using createReadStream.
|
|
* Run: node --experimental-strip-types scripts/smoke-docmind.ts <pdf-path>
|
|
*/
|
|
import DocmindClient from "@alicloud/docmind-api20220711";
|
|
import { RuntimeOptions } from "@alicloud/tea-util";
|
|
import { createReadStream } from "node:fs";
|
|
import { basename } from "node:path";
|
|
|
|
const accessKeyId = process.env["ALIBABA_CLOUD_ACCESS_KEY_ID"];
|
|
const accessKeySecret = process.env["ALIBABA_CLOUD_ACCESS_KEY_SECRET"];
|
|
if (accessKeyId === undefined || accessKeySecret === undefined) {
|
|
console.error("Set ALIBABA_CLOUD_ACCESS_KEY_ID and ALIBABA_CLOUD_ACCESS_KEY_SECRET env vars.");
|
|
process.exit(1);
|
|
}
|
|
|
|
const pdfPath = process.argv[2] ?? "../examples/trial/prime_sieve.pdf";
|
|
console.log(`Parsing: ${pdfPath}`);
|
|
|
|
const client = new DocmindClient.default({
|
|
endpoint: "docmind-api.cn-hangzhou.aliyuncs.com",
|
|
accessKeyId,
|
|
accessKeySecret,
|
|
type: "access_key",
|
|
regionId: "cn-hangzhou",
|
|
} as never);
|
|
|
|
const fileStream = createReadStream(pdfPath);
|
|
const runtime = new RuntimeOptions({});
|
|
|
|
console.log("Submitting job...");
|
|
const submitResp = await client.submitDocParserJobAdvance(
|
|
new DocmindClient.SubmitDocParserJobAdvanceRequest({
|
|
fileUrlObject: fileStream,
|
|
fileName: basename(pdfPath),
|
|
outputFormat: ["markdown"],
|
|
formulaEnhancement: true,
|
|
}),
|
|
runtime,
|
|
);
|
|
const jobId = submitResp.body?.data?.id;
|
|
console.log("Job ID:", jobId);
|
|
|
|
if (jobId === undefined) {
|
|
console.error("No job id:", JSON.stringify(submitResp.body));
|
|
process.exit(1);
|
|
}
|
|
|
|
console.log("Polling...");
|
|
const deadline = Date.now() + 5 * 60_000;
|
|
let status = "";
|
|
let markdownUrl = "";
|
|
let pageCount = 0;
|
|
while (Date.now() < deadline) {
|
|
await new Promise((r) => setTimeout(r, 10_000));
|
|
const resp = await client.queryDocParserStatus(
|
|
new DocmindClient.QueryDocParserStatusRequest({ id: jobId }),
|
|
);
|
|
const data = (resp.body as { data?: { status?: string; pageCountEstimate?: number; outputFormatResult?: Array<{ outputFileUrl?: string; outputType?: string }> } }).data;
|
|
status = data?.status ?? "";
|
|
console.log(` status: ${status}`);
|
|
if (status === "success") {
|
|
const md = data?.outputFormatResult?.find((r) => r.outputType === "markdown");
|
|
markdownUrl = md?.outputFileUrl ?? "";
|
|
pageCount = data?.pageCountEstimate ?? 0;
|
|
break;
|
|
}
|
|
if (status === "fail") {
|
|
console.error("Job failed!");
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
if (status !== "success" || markdownUrl === "") {
|
|
console.error("Failed or timed out:", status);
|
|
process.exit(1);
|
|
}
|
|
|
|
console.log("Downloading markdown from OSS...");
|
|
const mdResp = await fetch(markdownUrl);
|
|
const markdown = await mdResp.text();
|
|
|
|
console.log("--- Result ---");
|
|
console.log("Pages:", pageCount);
|
|
console.log("Cost (USD, est):", ((pageCount > 0 ? pageCount : 1) * 0.0056).toFixed(4));
|
|
console.log("Job ID:", jobId);
|
|
console.log("Markdown length:", markdown.length, "chars");
|
|
console.log("--- Markdown (first 1000 chars) ---");
|
|
console.log(markdown.slice(0, 1000));
|