feat: Support unlock pymupdf (#49)

Author: Wh1isper

lightblue_ai/tools/document/pdf.py

@@ -1,3 +1,4 @@
+import os
 from pathlib import Path
 from typing import Annotated, Any
 
@@ -82,6 +83,11 @@ def __init__(self):
             "This tool will also convert the PDF to images and save them in the `image_path` directory. "
             "You can View the images using the `view` tool. "
         )
+        if pymupdf_pro_password := os.getenv("PYMUPDF_PRO_PASSWORD"):
+            import pymupdf.pro
+
+            pymupdf.pro.unlock(pymupdf_pro_password)
+            self.description += "\n **This tool now supports DOC/DOCX, XLS/XLSX, PPT/PPTX, HWP/HWPX as PRO version.**"
 
     async def call(
         self,

pyproject.toml

@@ -32,6 +32,7 @@ dependencies = [
     "pydantic-ai>=0.2.0",
     "pydantic-settings>=2.8.1",
     "pymupdf4llm>=0.0.21",
+    "pymupdfpro>=1.25.5",
     "python-dotenv>=1.1.0",
     "rich>=14.0.0",
     "tavily-python>=0.5.3",