diff --git a/README.md b/README.md index 0597e77..02153fc 100644 --- a/README.md +++ b/README.md @@ -150,7 +150,7 @@ Or from the CLI: ```bash pip install markitdown -markitdown --llm-client OpenAI --llm-model gpt-4o example.jpg +markitdown --llm-model gpt-4o example.jpg ``` ### Docker diff --git a/packages/markitdown/src/markitdown/__main__.py b/packages/markitdown/src/markitdown/__main__.py index 1436fb8..d267766 100644 --- a/packages/markitdown/src/markitdown/__main__.py +++ b/packages/markitdown/src/markitdown/__main__.py @@ -56,9 +56,10 @@ parser.add_argument( action="store_true", help="list installed 3rd-party plugins (loaded with `--use-plugin`)", ) -parser.add_argument("--llm-client", choices={"OpenAI"}, help="default None") -parser.add_argument("--llm-client-url", help="base URL for --llm-client") -parser.add_argument("--llm-model", help="required for --llm-client") +parser.add_argument("--llm-model", metavar="MODEL", help="e.g. gpt-4o") +parser.add_argument( + "--llm-client-url", metavar="URL", help="base URL for OpenAI LLM client" +) parser.add_argument( "filename", metavar="FILENAME", nargs="?", help="if unspecified, defaults to stdin" ) @@ -92,8 +93,9 @@ def main(args=None): elif args.filename is None: raise ValueError("Filename is required when using Document Intelligence.") - if args.llm_client == "OpenAI": + if args.llm_model: from openai import OpenAI + llm_client = OpenAI(base_url=args.llm_client_url) else: llm_client = None