use an environment-specific __init__

Agenta-AI · Dec 1, 2023 · 86c2088 · 86c2088
1 parent 99becd0
commit 86c2088
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/agenta-backend/agenta_backend/__init__.py b/agenta-backend/agenta_backend/__init__.py
@@ -1,4 +1,6 @@
 import os
 
-if os.environ["FEATURE_FLAG"] in ["cloud", "ee"]:
-    import agenta_backend.commons.__init__
+if os.environ["FEATURE_FLAG"] in ["cloud"]:
+    import agenta_backend.cloud.__init__
+if os.environ["FEATURE_FLAG"] in ["ee"]:
+    import agenta_backend.ee.__init__