med-pdf-nmo 0.1.0

package/dist/cli.js

@@ -0,0 +1,87 @@
+#!/usr/bin/env node
+import fs from "node:fs/promises";
+import path from "node:path";
+import { predict } from "./predictor.js";
+function parseArgs(argv) {
+    const args = {};
+    for (let i = 0; i < argv.length; i += 1) {
+        const arg = argv[i];
+        if (!arg.startsWith("--"))
+            continue;
+        const key = arg.slice(2);
+        const next = argv[i + 1];
+        if (key === "answer") {
+            args.answer ??= [];
+            args.answer.push(next);
+            i += 1;
+        }
+        else if (next && !next.startsWith("--")) {
+            args[key] = next;
+            i += 1;
+        }
+        else {
+            args[key] = true;
+        }
+    }
+    return args;
+}
+function parseAnswers(args) {
+    if (args.answers) {
+        const parsed = JSON.parse(args.answers);
+        if (!Array.isArray(parsed))
+            throw new Error("--answers must be a JSON array");
+        return parsed;
+    }
+    if (args.answer?.length) {
+        return args.answer.map((value, index) => {
+            const separator = value.indexOf("=");
+            if (separator > 0) {
+                return { id: value.slice(0, separator), text: value.slice(separator + 1) };
+            }
+            return { id: String.fromCharCode(65 + index), text: value };
+        });
+    }
+    throw new Error("Provide --answers JSON or one or more --answer ID=text arguments");
+}
+async function readInput(args) {
+    if (args.input) {
+        const inputPath = path.resolve(args.input);
+        const raw = await fs.readFile(inputPath, "utf8");
+        return JSON.parse(raw.replace(/^\uFEFF/, ""));
+    }
+    return {
+        pdfPath: args.pdf ?? args.pdfPath,
+        question: args.question,
+        answers: parseAnswers(args),
+        mode: args.mode ?? "single",
+    };
+}
+async function attachLocalPdfData(input) {
+    if (input.pdfData || input.pdfBuffer || input.pdf || input.file || input.blob || input.pdfUrl || input.url || !input.pdfPath)
+        return input;
+    const absolutePath = path.resolve(input.pdfPath);
+    return {
+        ...input,
+        pdfData: await fs.readFile(absolutePath),
+        cacheKey: absolutePath,
+    };
+}
+async function main() {
+    const args = parseArgs(process.argv.slice(2));
+    if (args.help || args.h) {
+        process.stdout.write([
+            "Usage:",
+            "  npm run predict -- --input request.json",
+            "  npm run predict -- --pdf doc.pdf --question \"...\" --mode single --answer A=\"...\" --answer B=\"...\"",
+            "",
+        ].join("\n"));
+        return;
+    }
+    const input = await attachLocalPdfData(await readInput(args));
+    const output = await predict(input);
+    process.stdout.write(`${JSON.stringify(output, null, 2)}\n`);
+}
+main().catch((error) => {
+    process.stderr.write(`${error.stack || error.message}\n`);
+    process.exit(1);
+});

package/dist/index.d.ts

@@ -0,0 +1,82 @@
+import { predict, clearPredictorCache } from "./predictor.js";
+import { setPdfJsLib } from "./pdf.js";
+export { predict, clearPredictorCache, setPdfJsLib };
+/**
+ * PDF-вход, который принимает высокоуровневый API.
+ *
+ * В Node.js обычно передается Buffer/Uint8Array или ArrayBuffer. В браузере
+ * можно передавать File или Blob. Строка считается URL и загружается через
+ * fetch.
+ */
+export type PdfInput = string | ArrayBuffer | Uint8Array | Blob | {
+    arrayBuffer(): Promise<ArrayBuffer>;
+};
+/**
+ * Вариант ответа, который принимает {@link answerQuestion}.
+ *
+ * Для строковых вариантов автоматически создаются id: A, B, C и далее.
+ * Объектные варианты могут передавать стабильные id, которые вернутся в
+ * `selectedIds`.
+ */
+export type AnswerVariant = string | {
+    id?: string | number;
+    text?: string;
+};
+/**
+ * Параметры ответа на один вопрос НМО по одному PDF с медицинскими
+ * рекомендациями.
+ */
+export interface AnswerQuestionOptions {
+    /** Текст вопроса из задания НМО. */
+    question: string;
+    /** Варианты ответа. Алиас: `answers`. */
+    variants?: AnswerVariant[];
+    /** Альтернативное имя для `variants`. */
+    answers?: AnswerVariant[];
+    /** Режим вопроса: один ответ или точное множество нескольких ответов. */
+    type?: "single" | "multi" | string;
+    /** Альтернативное имя для `type`. */
+    mode?: "single" | "multi" | string;
+    /** Необязательный ключ кеша для повторного использования текста PDF. */
+    cacheKey?: string;
+    /** Явно переданный модуль PDF.js, полезно для браузерного окружения. */
+    pdfjsLib?: any;
+}
+/**
+ * Высокоуровневый результат, который возвращает {@link answerQuestion}.
+ */
+export interface AnswerQuestionResult {
+    /** Тексты выбранных ответов. */
+    selected: string[];
+    /** ID выбранных ответов. */
+    selectedIds: string[];
+    /** Итоговый режим, использованный predictor. */
+    mode: string;
+    /** Относительная уверенность выбранного ответа или набора ответов. */
+    confidence: number;
+    /** Калиброванные и сырые score по каждому варианту. */
+    scores: Array<{
+        id: string;
+        variant: string;
+        score: number;
+        raw: number;
+    }>;
+    /** Исходный низкоуровневый результат predictor. */
+    raw: any;
+    /** Фрагменты PDF и evidence, использованные при скоринге. */
+    evidence: any;
+    /** Runtime-метаданные, например число страниц и признак необходимости OCR. */
+    meta: any;
+}
+/**
+ * Отвечает на один вопрос НМО по содержимому PDF с медицинскими рекомендациями.
+ *
+ * Удобная обертка принимает варианты строками или объектами `{ id, text }`,
+ * вызывает локальный non-LLM predictor и сопоставляет выбранные id обратно с
+ * пользовательскими текстами ответов.
+ *
+ * @param pdf PDF-файл, байты, Blob/File, ArrayBuffer, Uint8Array или URL.
+ * @param options Текст вопроса, варианты ответа и необязательные runtime-настройки.
+ * @returns Выбранные ответы, id, confidence, score по вариантам и evidence.
+ */
+export declare function answerQuestion(pdf: PdfInput, options?: AnswerQuestionOptions): Promise<AnswerQuestionResult>;

package/dist/index.js

@@ -0,0 +1,51 @@
+import { predict, clearPredictorCache } from "./predictor.js";
+import { setPdfJsLib } from "./pdf.js";
+export { predict, clearPredictorCache, setPdfJsLib };
+/**
+ * Отвечает на один вопрос НМО по содержимому PDF с медицинскими рекомендациями.
+ *
+ * Удобная обертка принимает варианты строками или объектами `{ id, text }`,
+ * вызывает локальный non-LLM predictor и сопоставляет выбранные id обратно с
+ * пользовательскими текстами ответов.
+ *
+ * @param pdf PDF-файл, байты, Blob/File, ArrayBuffer, Uint8Array или URL.
+ * @param options Текст вопроса, варианты ответа и необязательные runtime-настройки.
+ * @returns Выбранные ответы, id, confidence, score по вариантам и evidence.
+ */
+export async function answerQuestion(pdf, options = { question: "" }) {
+    const variants = options.variants ?? options.answers ?? [];
+    const answers = variants.map((item, index) => {
+        if (typeof item === "string") {
+            return { id: String.fromCharCode(65 + index), text: item };
+        }
+        return {
+            id: String(item.id ?? String.fromCharCode(65 + index)),
+            text: String(item.text ?? ""),
+        };
+    });
+    const output = await predict({
+        pdfData: pdf,
+        cacheKey: options.cacheKey,
+        question: options.question,
+        answers,
+        mode: options.type ?? options.mode ?? "single",
+    }, { pdfjsLib: options.pdfjsLib });
+    const selectedAnswers = output.selected
+        .map((id) => answers.find((answer) => answer.id === id))
+        .filter(Boolean);
+    return {
+        selected: selectedAnswers.map((answer) => answer.text),
+        selectedIds: selectedAnswers.map((answer) => answer.id),
+        mode: output.mode,
+        confidence: output.confidence,
+        scores: answers.map((answer) => ({
+            id: answer.id,
+            variant: answer.text,
+            score: output.scores[answer.id] ?? 0,
+            raw: output.rawScores[answer.id] ?? 0,
+        })),
+        raw: output,
+        evidence: output.evidence,
+        meta: output.meta,
+    };
+}