From f2bd0028ea9bc094c8d7597d79fe0a04efa6fd2d Mon Sep 17 00:00:00 2001 From: Asankhaya Sharma Date: Thu, 14 Nov 2024 10:41:50 +0800 Subject: [PATCH] Update optillm.py --- optillm.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/optillm.py b/optillm.py index c5b63781..b7f0d35a 100644 --- a/optillm.py +++ b/optillm.py @@ -12,6 +12,7 @@ import re from concurrent.futures import ThreadPoolExecutor from typing import Tuple, Optional, Union, Dict, Any, List +from importlib.metadata import version # Import approach modules from optillm.mcts import chat_with_mcts @@ -507,6 +508,16 @@ def health(): def parse_args(): parser = argparse.ArgumentParser(description="Run LLM inference with various approaches.") + + # Add version argument using importlib.metadata + try: + package_version = version('optillm') + except Exception: + package_version = "unknown" # Fallback if package is not installed + + parser.add_argument('--version', action='version', + version=f'%(prog)s {package_version}', + help="Show program's version number and exit") # Define arguments and their corresponding environment variables args_env = [