refac

open-webui · Jun 1, 2024 · a2e11db · a2e11db
1 parent 8bab7a2
commit a2e11db
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 14 deletions.
diff --git a/config.py b/config.py
@@ -10,3 +10,6 @@
     load_dotenv(find_dotenv("./.env"))
 except ImportError:
     print("dotenv not installed, skipping...")
+
+API_KEY = os.getenv("PIPELINES_API_KEY", "0p3n-w3bu!")
+PIPELINES_DIR = os.getenv("PIPELINES_DIR", "./pipelines")
diff --git a/main.py b/main.py
@@ -27,20 +27,7 @@
 import sys
 
 
-####################################
-# Load .env file
-####################################
-
-try:
-    from dotenv import load_dotenv, find_dotenv
-
-    load_dotenv(find_dotenv(".env"))
-except ImportError:
-    print("dotenv not installed, skipping...")
-
-API_KEY = os.getenv("PIPELINES_API_KEY", "0p3n-w3bu!")
-
-PIPELINES_DIR = os.getenv("PIPELINES_DIR", "./pipelines")
+from config import API_KEY, PIPELINES_DIR
 
 if not os.path.exists(PIPELINES_DIR):
     os.makedirs(PIPELINES_DIR)