From e343ebc212ac13302f4b7599da0ff64f168b88f1 Mon Sep 17 00:00:00 2001 From: Michael Trimarchi Date: Wed, 16 Jul 2025 07:16:35 +0200 Subject: [PATCH] Add openai key parameter and option of the config file Allow to get the openai key from command line or file ini. Signed-off-by: Michael Trimarchi --- main.py | 12 ++++++++++++ src/config_parser.py | 1 + 2 files changed, 13 insertions(+) diff --git a/main.py b/main.py index f796933..422db7e 100644 --- a/main.py +++ b/main.py @@ -54,6 +54,13 @@ def main(): type=str, default="deploy", ) + parser.add_argument( + "-k", + "--openai-key", + help="Pass the openai key for create commit summary by Author and repo.", + type=str, + default=None, + ) args = parser.parse_args() @@ -62,6 +69,7 @@ def main(): months_back = None save_file_name = None deploy_dir = None + openai_key = None config_loaded_successfully = False @@ -73,6 +81,8 @@ def main(): company_identifier = config_data.get("company_identifier", "") months_back = config_data.get("months_back", None) deploy_dir = config_data.get("deploy_dir", None) + openai_apikey = config_data.get("openai_apikey", None) + config_loaded_successfully = True print(f"Configuration loaded from {args.config_file}.") else: @@ -93,6 +103,8 @@ def main(): months_back = args.months_back if args.deploy_dir: deploy_dir = args.deploy_dir + if args.openai_key: + openai_key = args.openai_key if args.deploy_dir: deploy_dir = args.deploy_dir diff --git a/src/config_parser.py b/src/config_parser.py index f90280d..204b449 100644 --- a/src/config_parser.py +++ b/src/config_parser.py @@ -25,6 +25,7 @@ def load_config_from_ini(file_path: str) -> dict | None: ) git_config["months_back"] = config["GitConfig"].getint("months_back", None) git_config["deploy_dir"] = config["GitConfig"].get("deploy_dir", None) + git_config["openai_apikey"] = config["OpenAi"].get("openai_apikey", None) return git_config except Exception as e: print(f"Error reading INI file {file_path}: {e}")