|
|
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' |
