LLM/docling-serve
1
0
Fork 0
mirror of https://github.com/docling-project/docling-serve.git synced 2025-11-29 08:33:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Use bytes as options key (#58)

Browse Source 
Signed-off-by: Eugene <fogaprod@gmail.com>
This commit is contained in:
Eugene
2025-02-21 21:03:27 +04:00
committed by GitHub
parent 36967f7f61
commit d5557fad9f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docling_serve/docling_conversion.py
View File

@@ -265,7 +265,7 @@ ConvertDocumentsRequest = Union[
# Document converters will be preloaded and stored in a dictionary
converters: Dict[str, DocumentConverter] = {}
converters: Dict[bytes, DocumentConverter] = {}
# Custom serializer for PdfFormatOption
@@ -301,7 +301,7 @@ def _serialize_pdf_format_option(pdf_format_option: PdfFormatOption) -> str:
# Computes the PDF pipeline options and returns the PdfFormatOption and its hash
def get_pdf_pipeline_opts( # noqa: C901
request: ConvertDocumentsOptions,
) -> Tuple[PdfFormatOption, str]:
) -> Tuple[PdfFormatOption, bytes]:
if request.ocr_engine == OcrEngine.EASYOCR:
try:
import easyocr # noqa: F401
@@ -401,7 +401,7 @@ def get_pdf_pipeline_opts( # noqa: C901
serialized_data = _serialize_pdf_format_option(pdf_format_option)
options_hash = hashlib.sha1(serialized_data.encode()).hexdigest()
options_hash = hashlib.sha1(serialized_data.encode()).digest()
return pdf_format_option, options_hash