From a77c4f04154200f885fccef577a3341fbb2a2338 Mon Sep 17 00:00:00 2001 From: Casper da Costa-Luis Date: Wed, 18 Dec 2024 13:29:53 +0000 Subject: [PATCH] CLI: add --llm-client-header --- packages/markitdown/src/markitdown/__main__.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/packages/markitdown/src/markitdown/__main__.py b/packages/markitdown/src/markitdown/__main__.py index d267766..7861857 100644 --- a/packages/markitdown/src/markitdown/__main__.py +++ b/packages/markitdown/src/markitdown/__main__.py @@ -60,6 +60,14 @@ 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( + "-H", + "--llm-client-header", + metavar="HEADER", + nargs="*", + default=[], + help="may be specified multiple times", +) parser.add_argument( "filename", metavar="FILENAME", nargs="?", help="if unspecified, defaults to stdin" ) @@ -96,7 +104,11 @@ def main(args=None): if args.llm_model: from openai import OpenAI - llm_client = OpenAI(base_url=args.llm_client_url) + headers = {} + for header in args.llm_client_header: + key, value = header.split(":", 1) + headers[key] = value.lstrip() + llm_client = OpenAI(base_url=args.llm_client_url, default_headers=headers) else: llm_client = None