[Misc] Rename env in runtime (#176)

* misc: rename env variable name
vllm-project · Sep 13, 2024 · e2ad163 · e2ad163
1 parent de7fc99
commit e2ad163
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/python/aibrix/aibrix/downloader/s3.py b/python/aibrix/aibrix/downloader/s3.py
@@ -35,9 +35,9 @@ def _parse_bucket_info_from_uri(uri: str) -> Tuple[str, str]:
 class S3Downloader(BaseDownloader):
     def __init__(self, model_uri):
         model_name = envs.DOWNLOADER_MODEL_NAME
-        ak = envs.DOWNLOADER_AWS_ACCESS_KEY
-        sk = envs.DOWNLOADER_AWS_SECRET_KEY
-        endpoint = envs.DOWNLOADER_AWS_ENDPOINT
+        ak = envs.DOWNLOADER_AWS_ACCESS_KEY_ID
+        sk = envs.DOWNLOADER_AWS_SECRET_ACCESS_KEY
+        endpoint = envs.DOWNLOADER_AWS_ENDPOINT_URL
         region = envs.DOWNLOADER_AWS_REGION
         bucket_name, bucket_path = _parse_bucket_info_from_uri(model_uri)
 

diff --git a/python/aibrix/aibrix/envs.py b/python/aibrix/aibrix/envs.py
@@ -70,9 +70,9 @@ def _parse_int_or_none(value: Optional[str]) -> Optional[int]:
 DOWNLOADER_TOS_ENABLE_CRC = _is_true(os.getenv("TOS_ENABLE_CRC"))
 
 # Downloader AWS S3 Envs
-DOWNLOADER_AWS_ACCESS_KEY = os.getenv("AWS_ACCESS_KEY_ID")
-DOWNLOADER_AWS_SECRET_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
-DOWNLOADER_AWS_ENDPOINT = os.getenv("AWS_ENDPOINT_URL")
+DOWNLOADER_AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")
+DOWNLOADER_AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
+DOWNLOADER_AWS_ENDPOINT_URL = os.getenv("AWS_ENDPOINT_URL")
 DOWNLOADER_AWS_REGION = os.getenv("AWS_REGION")
 
 # Metric Standardizing Related Config