mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-12-11 22:28:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(asr): Implement robust status check in AsrPipeline (#2442)

Browse Source 
* test: Add failing test case for silent audio file

* fix: Implement robust status check in AsrPipeline

* DCO Remediation Commit for mastermaxx03 <srivastavaanimesh22@gmail.com>I, mastermaxx03 <srivastavaanimesh22@gmail.com>, hereby add my Signed-off-by to this commit: 5fc4d512b330bb0cd347da4cbcca0fbe9687898aI, mastermaxx03 <srivastavaanimesh22@gmail.com>, hereby add my Signed-off-by to this commit: 31a4e9a5f1

Signed-off-by: mastermaxx03 <srivastavaanimesh22@gmail.com>

* DCO Remediation Commit for mastermaxx03 <srivastavaanimesh22@gmail.com>

I, mastermaxx03 <srivastavaanimesh22@gmail.com>, hereby add my Signed-off-by to this commit: 5fc4d512b3
I, mastermaxx03 <srivastavaanimesh22@gmail.com>, hereby add my Signed-off-by to this commit: 31a4e9a5f1

Signed-off-by: mastermaxx03 <srivastavaanimesh22@gmail.com>

* DCO Remediation Commit for mastermaxx03 <srivastavaanimesh22@gmail.com>

I, mastermaxx03 <srivastavaanimesh22@gmail.com>, hereby add my Signed-off-by to this commit: 5fc4d512b3
I, mastermaxx03 <srivastavaanimesh22@gmail.com>, hereby add my Signed-off-by to this commit: 31a4e9a5f1

Signed-off-by: mastermaxx03 <srivastavaanimesh22@gmail.com>

---------

Signed-off-by: mastermaxx03 <srivastavaanimesh22@gmail.com>
This commit is contained in:
Animesh
2025-10-13 13:21:31 +05:30
committed by GitHub
parent 90200443bc
commit db985bb159
3 changed files with 48 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
docling/pipeline/asr_pipeline.py
View File

@@ -186,7 +186,6 @@ class _NativeWhisperModel:
label=DocItemLabel.TEXT, text=citem.to_string()
)
conv_res.status = ConversionStatus.SUCCESS
return conv_res
except Exception as exc:
@@ -249,9 +248,29 @@ class AsrPipeline(BasePipeline):
else:
_log.error(f"No model support for {self.pipeline_options.asr_options}")
def _has_text(self, document: "DoclingDocument") -> bool:
"""
Helper method to check if the document contains any transcribed text.
A transcription is considered non-empty if the .texts list contains items with actual, non whitespace content.
"""
if not document or not document.texts:
return False
for item in document.texts:
if item.text and item.text.strip():
return True
return False
def _determine_status(self, conv_res: ConversionResult) -> ConversionStatus:
status = ConversionStatus.SUCCESS
return status
"""Determines the final status of ASR Conversion based on its result."""
if conv_res.status == ConversionStatus.FAILURE or conv_res.errors:
return ConversionStatus.FAILURE
if not self._has_text(conv_res.document):
_log.warning(
"ASR conversion resulted in an empty document."
f"File: {conv_res.input.file.name}"
)
return ConversionStatus.PARTIAL_SUCCESS
return ConversionStatus.SUCCESS
@classmethod
def get_default_options(cls) -> AsrPipelineOptions: