config.py 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429
  1. import os
  2. import dotenv
  3. dotenv.load_dotenv()
  4. DEFAULTS = {
  5. 'EDITION': 'SELF_HOSTED',
  6. 'DB_USERNAME': 'postgres',
  7. 'DB_PASSWORD': '',
  8. 'DB_HOST': 'localhost',
  9. 'DB_PORT': '5432',
  10. 'DB_DATABASE': 'dify',
  11. 'DB_CHARSET': '',
  12. 'REDIS_HOST': 'localhost',
  13. 'REDIS_PORT': '6379',
  14. 'REDIS_DB': '0',
  15. 'REDIS_USE_SSL': 'False',
  16. 'OAUTH_REDIRECT_PATH': '/console/api/oauth/authorize',
  17. 'OAUTH_REDIRECT_INDEX_PATH': '/',
  18. 'CONSOLE_WEB_URL': 'https://cloud.dify.ai',
  19. 'CONSOLE_API_URL': 'https://cloud.dify.ai',
  20. 'SERVICE_API_URL': 'https://api.dify.ai',
  21. 'APP_WEB_URL': 'https://udify.app',
  22. 'FILES_URL': '',
  23. 'FILES_ACCESS_TIMEOUT': 300,
  24. 'S3_ADDRESS_STYLE': 'auto',
  25. 'STORAGE_TYPE': 'local',
  26. 'STORAGE_LOCAL_PATH': 'storage',
  27. 'CHECK_UPDATE_URL': 'https://updates.dify.ai',
  28. 'DEPLOY_ENV': 'PRODUCTION',
  29. 'SQLALCHEMY_DATABASE_URI_SCHEME': 'postgresql',
  30. 'SQLALCHEMY_POOL_SIZE': 30,
  31. 'SQLALCHEMY_MAX_OVERFLOW': 10,
  32. 'SQLALCHEMY_POOL_RECYCLE': 3600,
  33. 'SQLALCHEMY_POOL_PRE_PING': 'False',
  34. 'SQLALCHEMY_ECHO': 'False',
  35. 'SENTRY_TRACES_SAMPLE_RATE': 1.0,
  36. 'SENTRY_PROFILES_SAMPLE_RATE': 1.0,
  37. 'WEAVIATE_GRPC_ENABLED': 'True',
  38. 'WEAVIATE_BATCH_SIZE': 100,
  39. 'QDRANT_CLIENT_TIMEOUT': 20,
  40. 'QDRANT_GRPC_ENABLED': 'False',
  41. 'QDRANT_GRPC_PORT': '6334',
  42. 'CELERY_BACKEND': 'database',
  43. 'LOG_LEVEL': 'INFO',
  44. 'LOG_FILE': '',
  45. 'LOG_FORMAT': '%(asctime)s.%(msecs)03d %(levelname)s [%(threadName)s] [%(filename)s:%(lineno)d] - %(message)s',
  46. 'LOG_DATEFORMAT': '%Y-%m-%d %H:%M:%S',
  47. 'HOSTED_OPENAI_QUOTA_LIMIT': 200,
  48. 'HOSTED_OPENAI_TRIAL_ENABLED': 'False',
  49. 'HOSTED_OPENAI_TRIAL_MODELS': 'gpt-3.5-turbo,gpt-3.5-turbo-1106,gpt-3.5-turbo-instruct,gpt-3.5-turbo-16k,gpt-3.5-turbo-16k-0613,gpt-3.5-turbo-0613,gpt-3.5-turbo-0125,text-davinci-003',
  50. 'HOSTED_OPENAI_PAID_ENABLED': 'False',
  51. 'HOSTED_OPENAI_PAID_MODELS': 'gpt-4,gpt-4-turbo-preview,gpt-4-turbo-2024-04-09,gpt-4-1106-preview,gpt-4-0125-preview,gpt-3.5-turbo,gpt-3.5-turbo-16k,gpt-3.5-turbo-16k-0613,gpt-3.5-turbo-1106,gpt-3.5-turbo-0613,gpt-3.5-turbo-0125,gpt-3.5-turbo-instruct,text-davinci-003',
  52. 'HOSTED_AZURE_OPENAI_ENABLED': 'False',
  53. 'HOSTED_AZURE_OPENAI_QUOTA_LIMIT': 200,
  54. 'HOSTED_ANTHROPIC_QUOTA_LIMIT': 600000,
  55. 'HOSTED_ANTHROPIC_TRIAL_ENABLED': 'False',
  56. 'HOSTED_ANTHROPIC_PAID_ENABLED': 'False',
  57. 'HOSTED_MODERATION_ENABLED': 'False',
  58. 'HOSTED_MODERATION_PROVIDERS': '',
  59. 'HOSTED_FETCH_APP_TEMPLATES_MODE': 'remote',
  60. 'HOSTED_FETCH_APP_TEMPLATES_REMOTE_DOMAIN': 'https://tmpl.dify.ai',
  61. 'CLEAN_DAY_SETTING': 30,
  62. 'UPLOAD_FILE_SIZE_LIMIT': 15,
  63. 'UPLOAD_FILE_BATCH_LIMIT': 5,
  64. 'UPLOAD_IMAGE_FILE_SIZE_LIMIT': 10,
  65. 'OUTPUT_MODERATION_BUFFER_SIZE': 300,
  66. 'MULTIMODAL_SEND_IMAGE_FORMAT': 'base64',
  67. 'INVITE_EXPIRY_HOURS': 72,
  68. 'BILLING_ENABLED': 'False',
  69. 'CAN_REPLACE_LOGO': 'False',
  70. 'MODEL_LB_ENABLED': 'False',
  71. 'ETL_TYPE': 'dify',
  72. 'KEYWORD_STORE': 'jieba',
  73. 'BATCH_UPLOAD_LIMIT': 20,
  74. 'CODE_EXECUTION_ENDPOINT': 'http://sandbox:8194',
  75. 'CODE_EXECUTION_API_KEY': 'dify-sandbox',
  76. 'TOOL_ICON_CACHE_MAX_AGE': 3600,
  77. 'MILVUS_DATABASE': 'default',
  78. 'KEYWORD_DATA_SOURCE_TYPE': 'database',
  79. 'INNER_API': 'False',
  80. 'ENTERPRISE_ENABLED': 'False',
  81. 'INDEXING_MAX_SEGMENTATION_TOKENS_LENGTH': 1000,
  82. 'WORKFLOW_MAX_EXECUTION_STEPS': 500,
  83. 'WORKFLOW_MAX_EXECUTION_TIME': 1200,
  84. 'WORKFLOW_CALL_MAX_DEPTH': 5,
  85. }
  86. def get_env(key):
  87. return os.environ.get(key, DEFAULTS.get(key))
  88. def get_bool_env(key):
  89. value = get_env(key)
  90. return value.lower() == 'true' if value is not None else False
  91. def get_cors_allow_origins(env, default):
  92. cors_allow_origins = []
  93. if get_env(env):
  94. for origin in get_env(env).split(','):
  95. cors_allow_origins.append(origin)
  96. else:
  97. cors_allow_origins = [default]
  98. return cors_allow_origins
  99. class Config:
  100. """Application configuration class."""
  101. def __init__(self):
  102. # ------------------------
  103. # General Configurations.
  104. # ------------------------
  105. self.CURRENT_VERSION = "0.6.10"
  106. self.COMMIT_SHA = get_env('COMMIT_SHA')
  107. self.EDITION = get_env('EDITION')
  108. self.DEPLOY_ENV = get_env('DEPLOY_ENV')
  109. self.TESTING = False
  110. self.LOG_LEVEL = get_env('LOG_LEVEL')
  111. self.LOG_FILE = get_env('LOG_FILE')
  112. self.LOG_FORMAT = get_env('LOG_FORMAT')
  113. self.LOG_DATEFORMAT = get_env('LOG_DATEFORMAT')
  114. self.API_COMPRESSION_ENABLED = get_bool_env('API_COMPRESSION_ENABLED')
  115. # The backend URL prefix of the console API.
  116. # used to concatenate the login authorization callback or notion integration callback.
  117. self.CONSOLE_API_URL = get_env('CONSOLE_API_URL')
  118. # The front-end URL prefix of the console web.
  119. # used to concatenate some front-end addresses and for CORS configuration use.
  120. self.CONSOLE_WEB_URL = get_env('CONSOLE_WEB_URL')
  121. # WebApp Url prefix.
  122. # used to display WebAPP API Base Url to the front-end.
  123. self.APP_WEB_URL = get_env('APP_WEB_URL')
  124. # Service API Url prefix.
  125. # used to display Service API Base Url to the front-end.
  126. self.SERVICE_API_URL = get_env('SERVICE_API_URL')
  127. # File preview or download Url prefix.
  128. # used to display File preview or download Url to the front-end or as Multi-model inputs;
  129. # Url is signed and has expiration time.
  130. self.FILES_URL = get_env('FILES_URL') if get_env('FILES_URL') else self.CONSOLE_API_URL
  131. # File Access Time specifies a time interval in seconds for the file to be accessed.
  132. # The default value is 300 seconds.
  133. self.FILES_ACCESS_TIMEOUT = int(get_env('FILES_ACCESS_TIMEOUT'))
  134. # Your App secret key will be used for securely signing the session cookie
  135. # Make sure you are changing this key for your deployment with a strong key.
  136. # You can generate a strong key using `openssl rand -base64 42`.
  137. # Alternatively you can set it with `SECRET_KEY` environment variable.
  138. self.SECRET_KEY = get_env('SECRET_KEY')
  139. # Enable or disable the inner API.
  140. self.INNER_API = get_bool_env('INNER_API')
  141. # The inner API key is used to authenticate the inner API.
  142. self.INNER_API_KEY = get_env('INNER_API_KEY')
  143. # cors settings
  144. self.CONSOLE_CORS_ALLOW_ORIGINS = get_cors_allow_origins(
  145. 'CONSOLE_CORS_ALLOW_ORIGINS', self.CONSOLE_WEB_URL)
  146. self.WEB_API_CORS_ALLOW_ORIGINS = get_cors_allow_origins(
  147. 'WEB_API_CORS_ALLOW_ORIGINS', '*')
  148. # check update url
  149. self.CHECK_UPDATE_URL = get_env('CHECK_UPDATE_URL')
  150. # ------------------------
  151. # Database Configurations.
  152. # ------------------------
  153. db_credentials = {
  154. key: get_env(key) for key in
  155. ['DB_USERNAME', 'DB_PASSWORD', 'DB_HOST', 'DB_PORT', 'DB_DATABASE', 'DB_CHARSET']
  156. }
  157. self.SQLALCHEMY_DATABASE_URI_SCHEME = get_env('SQLALCHEMY_DATABASE_URI_SCHEME')
  158. db_extras = f"?client_encoding={db_credentials['DB_CHARSET']}" if db_credentials['DB_CHARSET'] else ""
  159. self.SQLALCHEMY_DATABASE_URI = f"{self.SQLALCHEMY_DATABASE_URI_SCHEME}://{db_credentials['DB_USERNAME']}:{db_credentials['DB_PASSWORD']}@{db_credentials['DB_HOST']}:{db_credentials['DB_PORT']}/{db_credentials['DB_DATABASE']}{db_extras}"
  160. self.SQLALCHEMY_ENGINE_OPTIONS = {
  161. 'pool_size': int(get_env('SQLALCHEMY_POOL_SIZE')),
  162. 'max_overflow': int(get_env('SQLALCHEMY_MAX_OVERFLOW')),
  163. 'pool_recycle': int(get_env('SQLALCHEMY_POOL_RECYCLE')),
  164. 'pool_pre_ping': get_bool_env('SQLALCHEMY_POOL_PRE_PING'),
  165. 'connect_args': {'options': '-c timezone=UTC'},
  166. }
  167. self.SQLALCHEMY_ECHO = get_bool_env('SQLALCHEMY_ECHO')
  168. # ------------------------
  169. # Redis Configurations.
  170. # ------------------------
  171. self.REDIS_HOST = get_env('REDIS_HOST')
  172. self.REDIS_PORT = get_env('REDIS_PORT')
  173. self.REDIS_USERNAME = get_env('REDIS_USERNAME')
  174. self.REDIS_PASSWORD = get_env('REDIS_PASSWORD')
  175. self.REDIS_DB = get_env('REDIS_DB')
  176. self.REDIS_USE_SSL = get_bool_env('REDIS_USE_SSL')
  177. # ------------------------
  178. # Celery worker Configurations.
  179. # ------------------------
  180. self.CELERY_BROKER_URL = get_env('CELERY_BROKER_URL')
  181. self.CELERY_BACKEND = get_env('CELERY_BACKEND')
  182. self.CELERY_RESULT_BACKEND = 'db+{}'.format(self.SQLALCHEMY_DATABASE_URI) \
  183. if self.CELERY_BACKEND == 'database' else self.CELERY_BROKER_URL
  184. self.BROKER_USE_SSL = self.CELERY_BROKER_URL.startswith('rediss://')
  185. # ------------------------
  186. # Code Execution Sandbox Configurations.
  187. # ------------------------
  188. self.CODE_EXECUTION_ENDPOINT = get_env('CODE_EXECUTION_ENDPOINT')
  189. self.CODE_EXECUTION_API_KEY = get_env('CODE_EXECUTION_API_KEY')
  190. # ------------------------
  191. # File Storage Configurations.
  192. # ------------------------
  193. self.STORAGE_TYPE = get_env('STORAGE_TYPE')
  194. self.STORAGE_LOCAL_PATH = get_env('STORAGE_LOCAL_PATH')
  195. # S3 Storage settings
  196. self.S3_ENDPOINT = get_env('S3_ENDPOINT')
  197. self.S3_BUCKET_NAME = get_env('S3_BUCKET_NAME')
  198. self.S3_ACCESS_KEY = get_env('S3_ACCESS_KEY')
  199. self.S3_SECRET_KEY = get_env('S3_SECRET_KEY')
  200. self.S3_REGION = get_env('S3_REGION')
  201. self.S3_ADDRESS_STYLE = get_env('S3_ADDRESS_STYLE')
  202. # Azure Blob Storage settings
  203. self.AZURE_BLOB_ACCOUNT_NAME = get_env('AZURE_BLOB_ACCOUNT_NAME')
  204. self.AZURE_BLOB_ACCOUNT_KEY = get_env('AZURE_BLOB_ACCOUNT_KEY')
  205. self.AZURE_BLOB_CONTAINER_NAME = get_env('AZURE_BLOB_CONTAINER_NAME')
  206. self.AZURE_BLOB_ACCOUNT_URL = get_env('AZURE_BLOB_ACCOUNT_URL')
  207. # Aliyun Storage settings
  208. self.ALIYUN_OSS_BUCKET_NAME = get_env('ALIYUN_OSS_BUCKET_NAME')
  209. self.ALIYUN_OSS_ACCESS_KEY = get_env('ALIYUN_OSS_ACCESS_KEY')
  210. self.ALIYUN_OSS_SECRET_KEY = get_env('ALIYUN_OSS_SECRET_KEY')
  211. self.ALIYUN_OSS_ENDPOINT = get_env('ALIYUN_OSS_ENDPOINT')
  212. self.ALIYUN_OSS_REGION = get_env('ALIYUN_OSS_REGION')
  213. self.ALIYUN_OSS_AUTH_VERSION = get_env('ALIYUN_OSS_AUTH_VERSION')
  214. # Google Cloud Storage settings
  215. self.GOOGLE_STORAGE_BUCKET_NAME = get_env('GOOGLE_STORAGE_BUCKET_NAME')
  216. self.GOOGLE_STORAGE_SERVICE_ACCOUNT_JSON_BASE64 = get_env('GOOGLE_STORAGE_SERVICE_ACCOUNT_JSON_BASE64')
  217. # ------------------------
  218. # Vector Store Configurations.
  219. # Currently, only support: qdrant, milvus, zilliz, weaviate, relyt, pgvector
  220. # ------------------------
  221. self.VECTOR_STORE = get_env('VECTOR_STORE')
  222. self.KEYWORD_STORE = get_env('KEYWORD_STORE')
  223. # qdrant settings
  224. self.QDRANT_URL = get_env('QDRANT_URL')
  225. self.QDRANT_API_KEY = get_env('QDRANT_API_KEY')
  226. self.QDRANT_CLIENT_TIMEOUT = get_env('QDRANT_CLIENT_TIMEOUT')
  227. self.QDRANT_GRPC_ENABLED = get_env('QDRANT_GRPC_ENABLED')
  228. self.QDRANT_GRPC_PORT = get_env('QDRANT_GRPC_PORT')
  229. # milvus / zilliz setting
  230. self.MILVUS_HOST = get_env('MILVUS_HOST')
  231. self.MILVUS_PORT = get_env('MILVUS_PORT')
  232. self.MILVUS_USER = get_env('MILVUS_USER')
  233. self.MILVUS_PASSWORD = get_env('MILVUS_PASSWORD')
  234. self.MILVUS_SECURE = get_env('MILVUS_SECURE')
  235. self.MILVUS_DATABASE = get_env('MILVUS_DATABASE')
  236. # weaviate settings
  237. self.WEAVIATE_ENDPOINT = get_env('WEAVIATE_ENDPOINT')
  238. self.WEAVIATE_API_KEY = get_env('WEAVIATE_API_KEY')
  239. self.WEAVIATE_GRPC_ENABLED = get_bool_env('WEAVIATE_GRPC_ENABLED')
  240. self.WEAVIATE_BATCH_SIZE = int(get_env('WEAVIATE_BATCH_SIZE'))
  241. # relyt settings
  242. self.RELYT_HOST = get_env('RELYT_HOST')
  243. self.RELYT_PORT = get_env('RELYT_PORT')
  244. self.RELYT_USER = get_env('RELYT_USER')
  245. self.RELYT_PASSWORD = get_env('RELYT_PASSWORD')
  246. self.RELYT_DATABASE = get_env('RELYT_DATABASE')
  247. # pgvecto rs settings
  248. self.PGVECTO_RS_HOST = get_env('PGVECTO_RS_HOST')
  249. self.PGVECTO_RS_PORT = get_env('PGVECTO_RS_PORT')
  250. self.PGVECTO_RS_USER = get_env('PGVECTO_RS_USER')
  251. self.PGVECTO_RS_PASSWORD = get_env('PGVECTO_RS_PASSWORD')
  252. self.PGVECTO_RS_DATABASE = get_env('PGVECTO_RS_DATABASE')
  253. # pgvector settings
  254. self.PGVECTOR_HOST = get_env('PGVECTOR_HOST')
  255. self.PGVECTOR_PORT = get_env('PGVECTOR_PORT')
  256. self.PGVECTOR_USER = get_env('PGVECTOR_USER')
  257. self.PGVECTOR_PASSWORD = get_env('PGVECTOR_PASSWORD')
  258. self.PGVECTOR_DATABASE = get_env('PGVECTOR_DATABASE')
  259. # tidb-vector settings
  260. self.TIDB_VECTOR_HOST = get_env('TIDB_VECTOR_HOST')
  261. self.TIDB_VECTOR_PORT = get_env('TIDB_VECTOR_PORT')
  262. self.TIDB_VECTOR_USER = get_env('TIDB_VECTOR_USER')
  263. self.TIDB_VECTOR_PASSWORD = get_env('TIDB_VECTOR_PASSWORD')
  264. self.TIDB_VECTOR_DATABASE = get_env('TIDB_VECTOR_DATABASE')
  265. # ------------------------
  266. # Mail Configurations.
  267. # ------------------------
  268. self.MAIL_TYPE = get_env('MAIL_TYPE')
  269. self.MAIL_DEFAULT_SEND_FROM = get_env('MAIL_DEFAULT_SEND_FROM')
  270. self.RESEND_API_KEY = get_env('RESEND_API_KEY')
  271. self.RESEND_API_URL = get_env('RESEND_API_URL')
  272. # SMTP settings
  273. self.SMTP_SERVER = get_env('SMTP_SERVER')
  274. self.SMTP_PORT = get_env('SMTP_PORT')
  275. self.SMTP_USERNAME = get_env('SMTP_USERNAME')
  276. self.SMTP_PASSWORD = get_env('SMTP_PASSWORD')
  277. self.SMTP_USE_TLS = get_bool_env('SMTP_USE_TLS')
  278. self.SMTP_OPPORTUNISTIC_TLS = get_bool_env('SMTP_OPPORTUNISTIC_TLS')
  279. # ------------------------
  280. # Workspace Configurations.
  281. # ------------------------
  282. self.INVITE_EXPIRY_HOURS = int(get_env('INVITE_EXPIRY_HOURS'))
  283. # ------------------------
  284. # Sentry Configurations.
  285. # ------------------------
  286. self.SENTRY_DSN = get_env('SENTRY_DSN')
  287. self.SENTRY_TRACES_SAMPLE_RATE = float(get_env('SENTRY_TRACES_SAMPLE_RATE'))
  288. self.SENTRY_PROFILES_SAMPLE_RATE = float(get_env('SENTRY_PROFILES_SAMPLE_RATE'))
  289. # ------------------------
  290. # Business Configurations.
  291. # ------------------------
  292. # multi model send image format, support base64, url, default is base64
  293. self.MULTIMODAL_SEND_IMAGE_FORMAT = get_env('MULTIMODAL_SEND_IMAGE_FORMAT')
  294. # Dataset Configurations.
  295. self.CLEAN_DAY_SETTING = get_env('CLEAN_DAY_SETTING')
  296. # File upload Configurations.
  297. self.UPLOAD_FILE_SIZE_LIMIT = int(get_env('UPLOAD_FILE_SIZE_LIMIT'))
  298. self.UPLOAD_FILE_BATCH_LIMIT = int(get_env('UPLOAD_FILE_BATCH_LIMIT'))
  299. self.UPLOAD_IMAGE_FILE_SIZE_LIMIT = int(get_env('UPLOAD_IMAGE_FILE_SIZE_LIMIT'))
  300. self.BATCH_UPLOAD_LIMIT = get_env('BATCH_UPLOAD_LIMIT')
  301. # RAG ETL Configurations.
  302. self.ETL_TYPE = get_env('ETL_TYPE')
  303. self.UNSTRUCTURED_API_URL = get_env('UNSTRUCTURED_API_URL')
  304. self.UNSTRUCTURED_API_KEY = get_env('UNSTRUCTURED_API_KEY')
  305. self.KEYWORD_DATA_SOURCE_TYPE = get_env('KEYWORD_DATA_SOURCE_TYPE')
  306. # Indexing Configurations.
  307. self.INDEXING_MAX_SEGMENTATION_TOKENS_LENGTH = get_env('INDEXING_MAX_SEGMENTATION_TOKENS_LENGTH')
  308. # Tool Configurations.
  309. self.TOOL_ICON_CACHE_MAX_AGE = get_env('TOOL_ICON_CACHE_MAX_AGE')
  310. self.WORKFLOW_MAX_EXECUTION_STEPS = int(get_env('WORKFLOW_MAX_EXECUTION_STEPS'))
  311. self.WORKFLOW_MAX_EXECUTION_TIME = int(get_env('WORKFLOW_MAX_EXECUTION_TIME'))
  312. self.WORKFLOW_CALL_MAX_DEPTH = int(get_env('WORKFLOW_CALL_MAX_DEPTH'))
  313. # Moderation in app Configurations.
  314. self.OUTPUT_MODERATION_BUFFER_SIZE = int(get_env('OUTPUT_MODERATION_BUFFER_SIZE'))
  315. # Notion integration setting
  316. self.NOTION_CLIENT_ID = get_env('NOTION_CLIENT_ID')
  317. self.NOTION_CLIENT_SECRET = get_env('NOTION_CLIENT_SECRET')
  318. self.NOTION_INTEGRATION_TYPE = get_env('NOTION_INTEGRATION_TYPE')
  319. self.NOTION_INTERNAL_SECRET = get_env('NOTION_INTERNAL_SECRET')
  320. self.NOTION_INTEGRATION_TOKEN = get_env('NOTION_INTEGRATION_TOKEN')
  321. # ------------------------
  322. # Platform Configurations.
  323. # ------------------------
  324. self.GITHUB_CLIENT_ID = get_env('GITHUB_CLIENT_ID')
  325. self.GITHUB_CLIENT_SECRET = get_env('GITHUB_CLIENT_SECRET')
  326. self.GOOGLE_CLIENT_ID = get_env('GOOGLE_CLIENT_ID')
  327. self.GOOGLE_CLIENT_SECRET = get_env('GOOGLE_CLIENT_SECRET')
  328. self.OAUTH_REDIRECT_PATH = get_env('OAUTH_REDIRECT_PATH')
  329. self.HOSTED_OPENAI_API_KEY = get_env('HOSTED_OPENAI_API_KEY')
  330. self.HOSTED_OPENAI_API_BASE = get_env('HOSTED_OPENAI_API_BASE')
  331. self.HOSTED_OPENAI_API_ORGANIZATION = get_env('HOSTED_OPENAI_API_ORGANIZATION')
  332. self.HOSTED_OPENAI_TRIAL_ENABLED = get_bool_env('HOSTED_OPENAI_TRIAL_ENABLED')
  333. self.HOSTED_OPENAI_TRIAL_MODELS = get_env('HOSTED_OPENAI_TRIAL_MODELS')
  334. self.HOSTED_OPENAI_QUOTA_LIMIT = int(get_env('HOSTED_OPENAI_QUOTA_LIMIT'))
  335. self.HOSTED_OPENAI_PAID_ENABLED = get_bool_env('HOSTED_OPENAI_PAID_ENABLED')
  336. self.HOSTED_OPENAI_PAID_MODELS = get_env('HOSTED_OPENAI_PAID_MODELS')
  337. self.HOSTED_AZURE_OPENAI_ENABLED = get_bool_env('HOSTED_AZURE_OPENAI_ENABLED')
  338. self.HOSTED_AZURE_OPENAI_API_KEY = get_env('HOSTED_AZURE_OPENAI_API_KEY')
  339. self.HOSTED_AZURE_OPENAI_API_BASE = get_env('HOSTED_AZURE_OPENAI_API_BASE')
  340. self.HOSTED_AZURE_OPENAI_QUOTA_LIMIT = int(get_env('HOSTED_AZURE_OPENAI_QUOTA_LIMIT'))
  341. self.HOSTED_ANTHROPIC_API_BASE = get_env('HOSTED_ANTHROPIC_API_BASE')
  342. self.HOSTED_ANTHROPIC_API_KEY = get_env('HOSTED_ANTHROPIC_API_KEY')
  343. self.HOSTED_ANTHROPIC_TRIAL_ENABLED = get_bool_env('HOSTED_ANTHROPIC_TRIAL_ENABLED')
  344. self.HOSTED_ANTHROPIC_QUOTA_LIMIT = int(get_env('HOSTED_ANTHROPIC_QUOTA_LIMIT'))
  345. self.HOSTED_ANTHROPIC_PAID_ENABLED = get_bool_env('HOSTED_ANTHROPIC_PAID_ENABLED')
  346. self.HOSTED_MINIMAX_ENABLED = get_bool_env('HOSTED_MINIMAX_ENABLED')
  347. self.HOSTED_SPARK_ENABLED = get_bool_env('HOSTED_SPARK_ENABLED')
  348. self.HOSTED_ZHIPUAI_ENABLED = get_bool_env('HOSTED_ZHIPUAI_ENABLED')
  349. self.HOSTED_MODERATION_ENABLED = get_bool_env('HOSTED_MODERATION_ENABLED')
  350. self.HOSTED_MODERATION_PROVIDERS = get_env('HOSTED_MODERATION_PROVIDERS')
  351. # fetch app templates mode, remote, builtin, db(only for dify SaaS), default: remote
  352. self.HOSTED_FETCH_APP_TEMPLATES_MODE = get_env('HOSTED_FETCH_APP_TEMPLATES_MODE')
  353. self.HOSTED_FETCH_APP_TEMPLATES_REMOTE_DOMAIN = get_env('HOSTED_FETCH_APP_TEMPLATES_REMOTE_DOMAIN')
  354. # Model Load Balancing Configurations.
  355. self.MODEL_LB_ENABLED = get_bool_env('MODEL_LB_ENABLED')
  356. # Platform Billing Configurations.
  357. self.BILLING_ENABLED = get_bool_env('BILLING_ENABLED')
  358. # ------------------------
  359. # Enterprise feature Configurations.
  360. # **Before using, please contact business@dify.ai by email to inquire about licensing matters.**
  361. # ------------------------
  362. self.ENTERPRISE_ENABLED = get_bool_env('ENTERPRISE_ENABLED')
  363. self.CAN_REPLACE_LOGO = get_bool_env('CAN_REPLACE_LOGO')