Don't run code during __import__, don't import LiteralDataLayer by de…

…fault. (#1382)
Chainlit · Sep 30, 2024 · dfa858b · dfa858b
1 parent 1c5098c
commit dfa858b
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/backend/chainlit/data/__init__.py b/backend/chainlit/data/__init__.py
@@ -2,19 +2,24 @@
 from typing import Optional
 
 from .base import BaseDataLayer
-from .literalai import LiteralDataLayer
 from .utils import (
     queue_until_user_message as queue_until_user_message,  # TODO: Consider deprecating re-export.; Redundant alias tells type checkers to STFU.
 )
 
 _data_layer: Optional[BaseDataLayer] = None
 
 
-if api_key := os.environ.get("LITERAL_API_KEY"):
-    # support legacy LITERAL_SERVER variable as fallback
-    server = os.environ.get("LITERAL_API_URL") or os.environ.get("LITERAL_SERVER")
-    _data_layer = LiteralDataLayer(api_key=api_key, server=server)
+def get_data_layer():
+    global _data_layer
 
+    if not _data_layer:
+        if api_key := os.environ.get("LITERAL_API_KEY"):
+            from .literalai import LiteralDataLayer
+
+            # support legacy LITERAL_SERVER variable as fallback
+            server = os.environ.get("LITERAL_API_URL") or os.environ.get(
+                "LITERAL_SERVER"
+            )
+            _data_layer = LiteralDataLayer(api_key=api_key, server=server)
 
-def get_data_layer():
     return _data_layer