@graphai/pdf2text_agent 0.0.1 → 0.0.2

package/lib/index.d.ts

@@ -1,2 +1,2 @@
-import templateAgent from "./template_agent";
-export { templateAgent };
+import pdf2textAgent from "./pdf2text_agent";
+export { pdf2textAgent };

package/lib/index.js

@@ -3,6 +3,6 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.templateAgent = void 0;
-const template_agent_1 = __importDefault(require("./template_agent"));
-exports.templateAgent = template_agent_1.default;
+exports.pdf2textAgent = void 0;
+const pdf2text_agent_1 = __importDefault(require("./pdf2text_agent"));
+exports.pdf2textAgent = pdf2text_agent_1.default;

package/lib/pdf2text_agent.d.ts

@@ -0,0 +1,10 @@
+import { AgentFunction, AgentFunctionInfo } from "graphai";
+export declare const pdf2textAgent: AgentFunction<{
+    type: string;
+}, {
+    text?: string;
+}, null, {
+    buffer: Buffer;
+}>;
+declare const pdf2textAgentInfo: AgentFunctionInfo;
+export default pdf2textAgentInfo;

package/lib/pdf2text_agent.js

@@ -0,0 +1,34 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.pdf2textAgent = void 0;
+const graphai_1 = require("graphai");
+const pdf2md_1 = __importDefault(require("@opendocsg/pdf2md"));
+const unpdf_1 = require("unpdf");
+const pdf2textAgent = async ({ params, namedInputs }) => {
+    const { buffer } = namedInputs;
+    const { type } = params;
+    (0, graphai_1.assert)(!!buffer, "pdf2textAgent required buffer");
+    if (type === "md" || type === "markdown") {
+        const md_text = await (0, pdf2md_1.default)(buffer);
+        return { text: md_text };
+    }
+    const pdf = await (0, unpdf_1.getDocumentProxy)(new Uint8Array(buffer));
+    const { text: pdf_text } = await (0, unpdf_1.extractText)(pdf, { mergePages: true });
+    return { text: pdf_text };
+};
+exports.pdf2textAgent = pdf2textAgent;
+const pdf2textAgentInfo = {
+    name: "pdf2textAgent",
+    agent: exports.pdf2textAgent,
+    mock: exports.pdf2textAgent,
+    samples: [],
+    description: "Pdf2text Agent",
+    category: ["documents"],
+    author: "Receptron team",
+    repository: "https://github.com/receptron/graphai",
+    license: "MIT",
+};
+exports.default = pdf2textAgentInfo;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@graphai/pdf2text_agent",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "description": "",
   "main": "lib/index.js",
   "files": [