Initial Commit
This commit is contained in:
15
src/modules/pdf/tools/metadata.pdf.tool.py
Normal file
15
src/modules/pdf/tools/metadata.pdf.tool.py
Normal file
@@ -0,0 +1,15 @@
|
||||
import fitz
|
||||
from langchain_community.tools import BaseTool
|
||||
|
||||
class PDFMetadataTool(BaseTool):
|
||||
name="pdf_metadata"
|
||||
description="Extract metadata from PDF file"
|
||||
|
||||
def _run(self, path: str):
|
||||
try:
|
||||
with fitz.open(path) as doc:
|
||||
metadata = {
|
||||
"páginas": len(doc)
|
||||
}
|
||||
except Exception as e:
|
||||
return f"Error: {e}"
|
||||
Reference in New Issue
Block a user