♻️ Consolidate vllm utils

Signed-off-by: Evaline Ju <[email protected]>

Author: evaline-ju

src/vllm_tgis_adapter/tgis_utils/args.py

@@ -3,8 +3,7 @@
 import argparse
 import os
 
-from vllm.engine.arg_utils import StoreBoolean
-from vllm.utils import FlexibleArgumentParser
+from vllm.utils import FlexibleArgumentParser, StoreBoolean
 
 from vllm_tgis_adapter.grpc.validation import MAX_TOP_N_TOKENS
 from vllm_tgis_adapter.logging import init_logger

tests/test_tgis_utils.py

@@ -1,8 +1,7 @@
 import sys
 
 import pytest
-from vllm.engine.arg_utils import StoreBoolean
-from vllm.utils import FlexibleArgumentParser
+from vllm.utils import FlexibleArgumentParser, StoreBoolean
 
 from vllm_tgis_adapter.tgis_utils.args import (
     EnvVarArgumentParser,