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

update: cli options added for engine selection

Browse source
This commit is contained in:
tungsten106 2025-01-07 15:05:30 +08:00
parent f07ea3e09b
commit a1766c5981
1 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/markitdown/__main__.py
View file

@ -57,15 +57,26 @@ def main():
"--output",
help="Output file name. If not provided, output is written to stdout.",
)
# adding CLI option for extra parameters for PdfConverter
parser.add_argument(
"-e",
"--engine",
help="Engine name for converters. If not provided will use default.",
)
args = parser.parse_args()
kwargs = {}
if args.engine:
kwargs.update({"engine": args.engine})
if args.filename is None:
markitdown = MarkItDown()
result = markitdown.convert_stream(sys.stdin.buffer)
_handle_output(args, result)
else:
markitdown = MarkItDown()
result = markitdown.convert(args.filename)
result = markitdown.convert(args.filename, **kwargs)
_handle_output(args, result)