luciano/markitdown
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: reformatted to unify syntax

Browse source
This commit is contained in:
Hieu Lam 2025-03-10 10:31:21 +07:00
parent 65b3f4a152
commit 9fe8507906
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
packages/markitdown/src/markitdown/converters/_markdownify.py
View file

@ -7,6 +7,7 @@ from magika import Magika
magika = Magika()
class _CustomMarkdownify(markdownify.MarkdownConverter):
"""
A custom version of markdownify's MarkdownConverter. Changes include:
@ -26,7 +27,10 @@ class _CustomMarkdownify(markdownify.MarkdownConverter):
if not extracted_code_snippet:
return ""
result = magika.identify_bytes(extracted_code_snippet.encode())
if result.status == "ok" and result.prediction.output.group in ["text", "code"]:
if result.status == "ok" and result.prediction.output.group in [
"text",
"code",
]:
language = result.prediction.output.label
return language
return ""