@make-u-free/migi 0.5.11 → 0.5.13

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@make-u-free/migi",
-  "version": "0.5.11",
+  "version": "0.5.13",
   "description": "Your AI right-hand agent. Works anywhere, with any LLM API.",
   "type": "module",
   "bin": {
@@ -17,6 +17,7 @@
     "glob": "^11.0.0",
     "openai": "^4.0.0",
     "pdf-parse": "^2.4.5",
+    "pdfjs-dist": "^5.5.207",
     "xlsx": "^0.18.5"
   },
   "engines": {

package/src/tools.js

@@ -8,8 +8,10 @@ import chalk from 'chalk'
 import xlsxPkg from 'xlsx'
 import { createRequire } from 'module'
 const require = createRequire(import.meta.url)
-const pdfParse = require('pdf-parse')
+const _pdfParseModule = require('pdf-parse')
+const pdfParse = typeof _pdfParseModule === 'function' ? _pdfParseModule : _pdfParseModule.default
 import AdmZip from 'adm-zip'
+import * as pdfjsLib from 'pdfjs-dist/legacy/build/pdf.mjs'
 import OpenAI from 'openai'
 import { httpsAgent } from './tls.js'
 const { readFile: xlsxReadFile, utils: xlsxUtils } = xlsxPkg
@@ -232,14 +234,29 @@ export async function executeTool(name, args, opts = {}) {
       if (ext === '.pdf') {
         const buf = readFileSync(args.path)
 
-        // Step 1: テキストPDFとして抽出を試みる
+        // Step 1: pdfjs-dist でテキスト抽出（最も信頼性が高い）
         try {
-          const data = await pdfParse(buf)
-          const text = data.text?.trim()
+          const doc = await pdfjsLib.getDocument({ data: new Uint8Array(buf) }).promise
+          const pages = []
+          for (let p = 1; p <= doc.numPages; p++) {
+            const page = await doc.getPage(p)
+            const content = await page.getTextContent()
+            pages.push(content.items.map(item => item.str).join(''))
+          }
+          const text = pages.join('\n').trim()
           if (text) return text
         } catch (_) {}
 
-        // Step 2: 画像PDFとしてVision APIでOCR（ネイティブ依存なし）
+        // Step 2: pdf-parse でも試みる（フォールバック）
+        if (typeof pdfParse === 'function') {
+          try {
+            const data = await pdfParse(buf)
+            const text = data.text?.trim()
+            if (text) return text
+          } catch (_) {}
+        }
+
+        // Step 3: 画像PDFとしてVision APIでOCR（ネイティブ依存なし）
         if (!opts.apiKey) return '(テキストが抽出できませんでした)'
         const images = extractImagesFromPdf(buf)
         if (images.length === 0) return '(テキストも画像も抽出できませんでした)'