chainlit 1.2.0__py3-none-any.whl → 1.3.0__py3-none-any.whl

chainlit/auth.py

@@ -42,14 +42,14 @@ def get_configuration():
         "requireLogin": require_login(),
         "passwordAuth": config.code.password_auth_callback is not None,
         "headerAuth": config.code.header_auth_callback is not None,
-        "oauthProviders": get_configured_oauth_providers()
-        if is_oauth_enabled()
-        else [],
+        "oauthProviders": (
+            get_configured_oauth_providers() if is_oauth_enabled() else []
+        ),
     }
 
 
 def create_jwt(data: User) -> str:
-    to_encode = data.to_dict()  # type: Dict[str, Any]
+    to_encode: Dict[str, Any] = data.to_dict()
     to_encode.update(
         {
             "exp": datetime.utcnow() + timedelta(minutes=60 * 24 * 15),  # 15 days
@@ -69,16 +69,18 @@ async def authenticate_user(token: str = Depends(reuseable_oauth)):
         )
         del dict["exp"]
         user = User(**dict)
-    except Exception as e:
+    except Exception:
         raise HTTPException(status_code=401, detail="Invalid authentication token")
     if data_layer := get_data_layer():
         try:
             persisted_user = await data_layer.get_user(user.identifier)
-            if persisted_user == None:
+            if persisted_user is None:
                 persisted_user = await data_layer.create_user(user)
-        except Exception as e:
+        except Exception:
             return user
 
+        if user and user.display_name:
+            persisted_user.display_name = user.display_name
         return persisted_user
     else:
         return user

chainlit/cli/__init__.py

@@ -5,8 +5,10 @@ import click
 import nest_asyncio
 import uvicorn
 
+# Not sure if it is necessary to call nest_asyncio.apply() before the other imports
 nest_asyncio.apply()
 
+# ruff: noqa: E402
 from chainlit.cache import init_lc_cache
 from chainlit.config import (
     BACKEND_ROOT,
@@ -183,7 +185,7 @@ def chainlit_run(
         no_cache = True
         # 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 authenticationt tests
+        # This is required for authentication tests
         os.environ["CHAINLIT_AUTH_SECRET"] = "SUPER_SECRET"
     else:
         trace_event("chainlit run")

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