mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-07-26 20:14:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

restore guess format

Browse Source 
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi 2025-07-25 15:15:05 +02:00
parent bbb735d2de
commit 79c59cb2b0
1 changed files with 9 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
docling/datamodel/document.py
View File

@ -325,15 +325,15 @@ class _DocumentConversionInput(BaseModel):
formats = MimeTypeToFormat.get(mime, [])
_log.info(f"detected formats: {formats}")
input_format: Optional[InputFormat] = None
if len(formats) == 1:
input_format = formats[0]
if content:
input_format = _DocumentConversionInput._guess_from_content(
content, mime, formats
)
return input_format
if formats:
if len(formats) == 1 and mime not in ("text/plain"):
return formats[0]
else: # ambiguity in formats
return _DocumentConversionInput._guess_from_content(
content, mime, formats
)
else:
return None
@staticmethod
def _guess_from_content(
@ -342,9 +342,6 @@ class _DocumentConversionInput(BaseModel):
"""Guess the input format of a document by checking part of its content."""
input_format: Optional[InputFormat] = None
if len(formats) == 1:
input_format = formats[0]
if mime == "application/xml":
content_str = content.decode("utf-8")
match_doctype = re.search(r"<!DOCTYPE [^>]+>", content_str)