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

chainlit/__init__.py

@@ -16,8 +16,6 @@ from typing import TYPE_CHECKING, Any, Dict
 
 import chainlit.input_widget as input_widget
 from chainlit.action import Action
-from chainlit.assistant import Assistant
-from chainlit.assistant_settings import AssistantSettings
 from chainlit.cache import cache
 from chainlit.chat_context import chat_context
 from chainlit.chat_settings import ChatSettings
@@ -63,8 +61,6 @@ from .callbacks import (
     on_chat_end,
     on_chat_resume,
     on_chat_start,
-    on_create_assistant,
-    on_list_assistants,
     on_logout,
     on_message,
     on_settings_update,
@@ -138,10 +134,6 @@ __all__ = [
     "TaskStatus",
     "Video",
     "ChatSettings",
-    # Experimental
-    "AssistantSettings",
-    # Experimental
-    "Assistant",
     "input_widget",
     "Message",
     "ErrorMessage",
@@ -161,8 +153,6 @@ __all__ = [
     "action_callback",
     "author_rename",
     "on_settings_update",
-    "on_create_assistant",
-    "on_list_assistants",
     "password_auth_callback",
     "header_auth_callback",
     "sleep",

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,14 +69,14 @@ 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
 
         return persisted_user

chainlit/callbacks.py

@@ -1,10 +1,7 @@
-import functools
 import inspect
 from typing import Any, Awaitable, Callable, Dict, List, Optional
 
 from chainlit.action import Action
-from chainlit.assistant import Assistant
-from chainlit.assistant_settings import AssistantSettings
 from chainlit.config import config
 from chainlit.message import Message
 from chainlit.oauth_providers import get_configured_oauth_providers
@@ -17,15 +14,6 @@ from fastapi import Request, Response
 from starlette.datastructures import Headers
 
 
-def experimental(func):
-    @functools.wraps(func)
-    def wrapper(*args, **kwargs):
-        print(f"\033[1;33mexperimental feature: {func.__name__}\033[0m")
-        return func(*args, **kwargs)
-
-    return wrapper
-
-
 @trace
 def password_auth_callback(
     func: Callable[[str, str], Awaitable[Optional[User]]]
@@ -318,23 +306,3 @@ def on_settings_update(
 
     config.code.on_settings_update = wrap_user_function(func, with_task=True)
     return func
-
-
-# Experimental
-@trace
-@experimental
-def on_create_assistant(
-    func: Callable[[Optional[User], AssistantSettings], Any]
-) -> Callable[[Optional[User], AssistantSettings], Any]:
-    config.code.on_create_assistant = wrap_user_function(func)
-    return func
-
-
-# Experimental
-@trace
-@experimental
-def on_list_assistants(
-    func: Callable[[Optional[User]], List[Assistant]]
-) -> Callable[[Optional[User]], List[Assistant]]:
-    config.code.on_list_assistants = wrap_user_function(func)
-    return func

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,

chainlit/config.py

@@ -28,13 +28,13 @@ from ._utils import is_path_inside
 
 if TYPE_CHECKING:
     from chainlit.action import Action
-    from chainlit.assistant import Assistant
     from chainlit.element import ElementBased
     from chainlit.message import Message
     from chainlit.types import AudioChunk, ChatProfile, Starter, ThreadDict
     from chainlit.user import User
     from fastapi import Request, Response
 
+
 BACKEND_ROOT = os.path.dirname(__file__)
 PACKAGE_ROOT = os.path.dirname(os.path.dirname(BACKEND_ROOT))
 TRANSLATIONS_DIR = os.path.join(BACKEND_ROOT, "translations")
@@ -285,9 +285,9 @@ class CodeSettings:
     password_auth_callback: Optional[
         Callable[[str, str], Awaitable[Optional["User"]]]
     ] = None
-    header_auth_callback: Optional[Callable[[Headers], Awaitable[Optional["User"]]]] = (
-        None
-    )
+    header_auth_callback: Optional[
+        Callable[[Headers], Awaitable[Optional["User"]]]
+    ] = None
     oauth_callback: Optional[
         Callable[[str, str, Dict[str, str], "User"], Awaitable[Optional["User"]]]
     ] = None
@@ -305,13 +305,9 @@ class CodeSettings:
     set_chat_profiles: Optional[
         Callable[[Optional["User"]], Awaitable[List["ChatProfile"]]]
     ] = None
-    set_starters: Optional[Callable[[Optional["User"]], Awaitable[List["Starter"]]]] = (
-        None
-    )
-
-    # assistant-related callback function
-    on_create_assistant: Optional[Callable[[Optional["User"], Any], Any]] = None
-    on_list_assistants: Optional[Callable[[Optional["User"]], List["Assistant"]]] = None
+    set_starters: Optional[
+        Callable[[Optional["User"]], Awaitable[List["Starter"]]]
+    ] = None
 
 
 @dataclass()