| | import os
|
| | from dotenv import dotenv_values, load_dotenv
|
| |
|
| | print("Loading .env file...")
|
| | env_path = os.path.join(os.path.dirname(__file__), '.env')
|
| | if os.path.exists(env_path):
|
| | env_dict = dotenv_values(env_path)
|
| |
|
| | print("Environment variables to be loaded:")
|
| | for key, value in env_dict.items():
|
| | print(f"{key}={value}")
|
| | print("-" * 50)
|
| |
|
| | os.environ.update(env_dict)
|
| | print(f"Loaded environment variables from: {env_path}")
|
| | else:
|
| | print("Warning: .env file not found")
|
| |
|
| |
|
| |
|
| | MAX_THREAD_NUM = int(os.getenv('MAX_THREAD_NUM', 5))
|
| |
|
| |
|
| | MAX_NOVEL_SUMMARY_LENGTH = int(os.getenv('MAX_NOVEL_SUMMARY_LENGTH', 20000))
|
| |
|
| |
|
| | ENABLE_MONOGODB = os.getenv('ENABLE_MONGODB', 'false').lower() == 'true'
|
| | MONGODB_URI = os.getenv('MONGODB_URI', 'mongodb://127.0.0.1:27017/')
|
| | MONOGODB_DB_NAME = os.getenv('MONGODB_DB_NAME', 'llm_api')
|
| | ENABLE_MONOGODB_CACHE = os.getenv('ENABLE_MONGODB_CACHE', 'true').lower() == 'true'
|
| | CACHE_REPLAY_SPEED = float(os.getenv('CACHE_REPLAY_SPEED', 2))
|
| | CACHE_REPLAY_MAX_DELAY = float(os.getenv('CACHE_REPLAY_MAX_DELAY', 5))
|
| |
|
| |
|
| | API_COST_LIMITS = {
|
| | 'HOURLY_LIMIT_RMB': float(os.getenv('API_HOURLY_LIMIT_RMB', 100)),
|
| | 'DAILY_LIMIT_RMB': float(os.getenv('API_DAILY_LIMIT_RMB', 500)),
|
| | 'USD_TO_RMB_RATE': float(os.getenv('API_USD_TO_RMB_RATE', 7))
|
| | }
|
| |
|
| |
|
| | API_SETTINGS = {
|
| | 'wenxin': {
|
| | 'ak': os.getenv('WENXIN_AK', ''),
|
| | 'sk': os.getenv('WENXIN_SK', ''),
|
| | 'available_models': os.getenv('WENXIN_AVAILABLE_MODELS', '').split(','),
|
| | 'max_tokens': 4096,
|
| | },
|
| | 'doubao': {
|
| | 'api_key': os.getenv('DOUBAO_API_KEY', ''),
|
| | 'endpoint_ids': os.getenv('DOUBAO_ENDPOINT_IDS', '').split(','),
|
| | 'available_models': os.getenv('DOUBAO_AVAILABLE_MODELS', '').split(','),
|
| | 'max_tokens': 4096,
|
| | },
|
| | 'gpt': {
|
| | 'base_url': os.getenv('GPT_BASE_URL', ''),
|
| | 'api_key': os.getenv('GPT_API_KEY', ''),
|
| | 'proxies': os.getenv('GPT_PROXIES', ''),
|
| | 'available_models': os.getenv('GPT_AVAILABLE_MODELS', '').split(','),
|
| | 'max_tokens': 4096,
|
| | },
|
| | 'zhipuai': {
|
| | 'api_key': os.getenv('ZHIPUAI_API_KEY', ''),
|
| | 'available_models': os.getenv('ZHIPUAI_AVAILABLE_MODELS', '').split(','),
|
| | 'max_tokens': 4096,
|
| | },
|
| | 'local': {
|
| | 'base_url': os.getenv('LOCAL_BASE_URL', ''),
|
| | 'api_key': os.getenv('LOCAL_API_KEY', ''),
|
| | 'available_models': os.getenv('LOCAL_AVAILABLE_MODELS', '').split(','),
|
| | 'max_tokens': 4096,
|
| | }
|
| | }
|
| |
|
| | for model in API_SETTINGS.values():
|
| | model['available_models'] = [e.strip() for e in model['available_models']]
|
| |
|
| | DEFAULT_MAIN_MODEL = os.getenv('DEFAULT_MAIN_MODEL', 'wenxin/ERNIE-Novel-8K')
|
| | DEFAULT_SUB_MODEL = os.getenv('DEFAULT_SUB_MODEL', 'wenxin/ERNIE-3.5-8K')
|
| |
|
| | ENABLE_ONLINE_DEMO = os.getenv('ENABLE_ONLINE_DEMO', 'false').lower() == 'true' |
