fix: tests

1 year ago · 4b91ba422b
parent 40b3031215
commit 4b91ba422b
1 changed files with 1 additions and 2 deletions
--- a/api/core/workflow/nodes/document_extractor/node.py
+++ b/api/core/workflow/nodes/document_extractor/node.py
@ -468,8 +468,7 @@ def _extract_text_from_msg(file_content: bytes) -> str:
    except Exception as e:
        raise TextExtractionError(f"Failed to extract text from MSG: {str(e)}") from e

-
-def _extract_text_from_vtt(vtt_bytes: bytes):
+def _extract_text_from_vtt(vtt_bytes: bytes) -> str:
    text = _extract_text_from_plain_text(vtt_bytes)

    # remove bom