Spaces:
Runtime error
Runtime error
File size: 2,913 Bytes
e4f9cbe 7c35231 e4f9cbe c14732f cbf6a91 c14732f e4f9cbe 7c35231 |
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 63 64 65 66 67 68 69 |
"""Load environment variables from .env file."""
import os
from typing import Any, Literal, Optional, Union, cast
from dotenv import dotenv_values
EnvironmentKeys = Union[Literal['LILAC_DATA_PATH'],
# Authentication on the demo.
Literal['LILAC_AUTH_ENABLED'], Literal['GOOGLE_CLIENT_ID'],
Literal['GOOGLE_CLIENT_SECRET'], Literal['LILAC_OAUTH_SECRET_KEY'],
# DuckDB accessing GCS.
Literal['GCS_REGION'], Literal['GCS_ACCESS_KEY'], Literal['GCS_SECRET_KEY'],
# Embedding API keys.
Literal['OPENAI_API_KEY'], Literal['COHERE_API_KEY'],
Literal['PALM_API_KEY'],
# HuggingFace demos.
Literal['HF_USERNAME'], Literal['HF_STAGING_DEMO_REPO'],
Literal['SPACE_ID'],
# DuckDB
Literal['DUCKDB_USE_VIEWS'],
# Debugging
Literal['DEBUG'], Literal['DISABLE_LOGS']]
_ENV: dict[str, Optional[str]] = {}
# Private config variables. Saved separately to remove private env variables from tests.
_ENV_LOCAL: dict[str, Optional[str]] = {}
def env(key: EnvironmentKeys, default: Optional[Any] = None) -> Any:
"""Return the value of an environment variable."""
global _ENV
global _ENV_LOCAL
first_load = False
# This is done lazily so we can prevent loading local environment variables when testing. The
# 'PYTEST_CURRENT_TEST' environment variable is only set after module initialization by pytest.
if not _ENV or _ENV_LOCAL:
_ENV = {
**dotenv_values('.env'), # load shared variables
**dotenv_values('.env.demo'), # load demo-specific environment flags.
}
_ENV_LOCAL = {
# Load locally set variables when not in a testing environment.
**(dotenv_values('.env.local') if 'PYTEST_CURRENT_TEST' not in os.environ else {}),
}
first_load = True
# Override the file based configs with the current environment, in case flags have changed.
environment = {
**_ENV,
**(_ENV_LOCAL if 'PYTEST_CURRENT_TEST' not in os.environ else {}),
**os.environ
}
if first_load:
if environment.get('LILAC_AUTH_ENABLED', None):
if not environment.get('GOOGLE_CLIENT_ID', None) or not environment.get(
'GOOGLE_CLIENT_SECRET', None):
raise ValueError(
'Missing `GOOGLE_CLIENT_ID` or `GOOGLE_CLIENT_SECRET` when `LILAC_AUTH_ENABLED=true`')
SECRET_KEY = environment.get('LILAC_OAUTH_SECRET_KEY', None)
if not SECRET_KEY:
raise ValueError('Missing `LILAC_OAUTH_SECRET_KEY` when `LILAC_AUTH_ENABLED=true`')
return environment.get(key, default)
def data_path() -> str:
"""Return the base path for data."""
return cast(str, env('LILAC_DATA_PATH', './data'))
|