Spaces:
Build error
Build error
File size: 2,747 Bytes
b6af722 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from cosmos_predict1.utils.env_parsers.env_parser import EnvParser
from cosmos_predict1.utils.validator import String
class CredentialEnvParser(EnvParser):
APP_ENV = String(default="")
PROD_FT_AWS_CREDS_ACCESS_KEY_ID = String(default="")
PROD_FT_AWS_CREDS_SECRET_ACCESS_KEY = String(default="")
PROD_FT_AWS_CREDS_ENDPOINT_URL = String(default="https://s3.us-west-2.amazonaws.com")
PROD_FT_AWS_CREDS_REGION_NAME = String(default="us-west-2")
PROD_S3_CHECKPOINT_ACCESS_KEY_ID = String(default="")
PROD_S3_CHECKPOINT_SECRET_ACCESS_KEY = String(default="")
PROD_S3_CHECKPOINT_ENDPOINT_URL = String(default="")
PROD_S3_CHECKPOINT_REGION_NAME = String(default="")
PROD_TEAM_DIR_ACCESS_KEY_ID = String(default="")
PROD_TEAM_DIR_SECRET_ACCESS_KEY = String(default="")
PROD_TEAM_DIR_ENDPOINT_URL = String(default="")
PROD_TEAM_DIR_REGION_NAME = String(default="")
PICASSO_AUTH_MODEL_REGISTRY_API_KEY = String(default="")
PICASSO_API_ENDPOINT_URL = String(default="https://meeocvslt2.execute-api.us-west-2.amazonaws.com")
CRED_ENVS = CredentialEnvParser()
CRED_ENVS_DICT = {
"PROD_FT_AWS_CREDS": {
"aws_access_key_id": CRED_ENVS.PROD_FT_AWS_CREDS_ACCESS_KEY_ID,
"aws_secret_access_key": CRED_ENVS.PROD_FT_AWS_CREDS_SECRET_ACCESS_KEY,
"endpoint_url": CRED_ENVS.PROD_FT_AWS_CREDS_ENDPOINT_URL,
"region_name": CRED_ENVS.PROD_FT_AWS_CREDS_REGION_NAME,
},
"PROD_S3_CHECKPOINT": {
"aws_access_key_id": CRED_ENVS.PROD_S3_CHECKPOINT_ACCESS_KEY_ID,
"aws_secret_access_key": CRED_ENVS.PROD_S3_CHECKPOINT_SECRET_ACCESS_KEY,
"endpoint_url": CRED_ENVS.PROD_S3_CHECKPOINT_ENDPOINT_URL,
"region_name": CRED_ENVS.PROD_S3_CHECKPOINT_REGION_NAME,
},
"PROD_TEAM_DIR": {
"aws_access_key_id": CRED_ENVS.PROD_TEAM_DIR_ACCESS_KEY_ID,
"aws_secret_access_key": CRED_ENVS.PROD_TEAM_DIR_SECRET_ACCESS_KEY,
"endpoint_url": CRED_ENVS.PROD_TEAM_DIR_ENDPOINT_URL,
"region_name": CRED_ENVS.PROD_TEAM_DIR_REGION_NAME,
},
}
|