chainlit 2.0.dev0__py3-none-any.whl → 2.0.dev2__py3-none-any.whl

chainlit/__init__.py

@@ -23,6 +23,7 @@ from chainlit.context import context
 from chainlit.element import (
     Audio,
     Component,
+    Dataframe,
     File,
     Image,
     Pdf,

chainlit/auth.py

@@ -52,7 +52,9 @@ def create_jwt(data: User) -> str:
     to_encode: Dict[str, Any] = data.to_dict()
     to_encode.update(
         {
-            "exp": datetime.utcnow() + timedelta(minutes=60 * 24 * 15),  # 15 days
+            "exp": datetime.utcnow() + timedelta(
+                seconds=config.project.user_session_timeout
+            ),
         }
     )
     encoded_jwt = jwt.encode(to_encode, get_jwt_secret(), algorithm="HS256")
@@ -69,8 +71,10 @@ async def authenticate_user(token: str = Depends(reuseable_oauth)):
         )
         del dict["exp"]
         user = User(**dict)
-    except Exception:
-        raise HTTPException(status_code=401, detail="Invalid authentication token")
+    except Exception as e:
+        raise HTTPException(
+            status_code=401, detail="Invalid authentication token"
+        ) from e
     if data_layer := get_data_layer():
         try:
             persisted_user = await data_layer.get_user(user.identifier)
@@ -79,6 +83,8 @@ async def authenticate_user(token: str = Depends(reuseable_oauth)):
         except Exception:
             return user
 
+        if user and user.display_name:
+            persisted_user.display_name = user.display_name
         return persisted_user
     else:
         return user

chainlit/cache.py

@@ -1,3 +1,4 @@
+import importlib.util
 import os
 import threading
 
@@ -8,11 +9,7 @@ from chainlit.logger import logger
 def init_lc_cache():
     use_cache = config.project.cache is True and config.run.no_cache is False
 
-    if use_cache:
-        try:
-            import langchain
-        except ImportError:
-            return
+    if use_cache and importlib.util.find_spec("langchain") is not None:
         from langchain.cache import SQLiteCache
         from langchain.globals import set_llm_cache
 

chainlit/cli/__init__.py

@@ -186,7 +186,7 @@ def chainlit_run(
         # This is required to have OpenAI LLM providers available for the CI run
         os.environ["OPENAI_API_KEY"] = "sk-FAKE-OPENAI-API-KEY"
         # This is required for authentication tests
-        os.environ["CHAINLIT_AUTH_SECRET"] = "SUPER_SECRET"
+        os.environ["CHAINLIT_AUTH_SECRET"] = "SUPER_SECRET"  # nosec B105
     else:
         trace_event("chainlit run")
 

chainlit/config.py

@@ -62,6 +62,9 @@ user_env = []
 # Duration (in seconds) during which the session is saved when the connection is lost
 session_timeout = 3600
 
+# Duration (in seconds) of the user session expiry
+user_session_timeout = 1296000  # 15 days
+
 # Enable third parties caching (e.g LangChain cache)
 cache = false
 
@@ -306,6 +309,8 @@ class ProjectSettings(DataClassJsonMixin):
     # Path to the local chat db
     # Duration (in seconds) during which the session is saved when the connection is lost
     session_timeout: int = 3600
+    # Duration (in seconds) of the user session expiry
+    user_session_timeout: int = 1296000  # 15 days
     # Enable third parties caching (e.g LangChain cache)
     cache: bool = False
     # Follow symlink for asset mount (see https://github.com/Chainlit/chainlit/issues/317)
@@ -447,7 +452,7 @@ def load_settings():
 
         if not meta or meta.get("generated_by") <= "0.3.0":
             raise ValueError(
-                "Your config file is outdated. Please delete it and restart the app to regenerate it."
+                f"Your config file '{config_file}' is outdated. Please delete it and restart the app to regenerate it."
             )
 
         lc_cache_path = os.path.join(config_dir, ".langchain.db")

chainlit/context.py

@@ -23,18 +23,17 @@ class ChainlitContext:
     loop: asyncio.AbstractEventLoop
     emitter: "BaseChainlitEmitter"
     session: Union["HTTPSession", "WebsocketSession"]
-    active_steps: List["Step"]
 
     @property
     def current_step(self):
-        if self.active_steps:
-            return self.active_steps[-1]
+        if previous_steps := local_steps.get():
+            return previous_steps[-1]
 
     @property
     def current_run(self):
-        if self.active_steps:
+        if previous_steps := local_steps.get():
             return next(
-                (step for step in self.active_steps if step.name in CL_RUN_NAMES), None
+                (step for step in previous_steps if step.name in CL_RUN_NAMES), None
             )
 
     def __init__(
@@ -46,7 +45,6 @@ class ChainlitContext:
 
         self.loop = asyncio.get_running_loop()
         self.session = session
-        self.active_steps = []
 
         if emitter:
             self.emitter = emitter
@@ -105,8 +103,8 @@ def init_http_context(
 def get_context() -> ChainlitContext:
     try:
         return context_var.get()
-    except LookupError:
-        raise ChainlitContextException()
+    except LookupError as e:
+        raise ChainlitContextException() from e
 
 
 context: ChainlitContext = LazyProxy(get_context, enable_cache=False)