chainlit 2.0.0__tar.gz → 2.0.dev0__tar.gz

{chainlit-2.0.0 → chainlit-2.0.dev0}/PKG-INFO

@@ -1,30 +1,23 @@
 Metadata-Version: 2.1
 Name: chainlit
-Version: 2.0.0
+Version: 2.0.dev0
 Summary: Build Conversational AI.
-Home-page: https://chainlit.io/
-License: Apache-2.0
+Home-page: https://github.com/Chainlit/chainlit
+License: Apache-2.0 license
 Keywords: LLM,Agents,gen ai,chat ui,chatbot ui,openai,copilot,langchain,conversational ai
-Author: Willy Douhard
+Author: Chainlit
 Requires-Python: >=3.9,<4.0.0
-Classifier: Environment :: Web Environment
-Classifier: Framework :: FastAPI
-Classifier: License :: OSI Approved :: Apache Software License
-Classifier: Programming Language :: JavaScript
+Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Topic :: Communications :: Chat
-Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Classifier: Topic :: Software Development :: User Interfaces
 Requires-Dist: aiofiles (>=23.1.0,<24.0.0)
 Requires-Dist: asyncer (>=0.0.7,<0.0.8)
 Requires-Dist: click (>=8.1.3,<9.0.0)
 Requires-Dist: dataclasses_json (>=0.6.7,<0.7.0)
-Requires-Dist: fastapi (>=0.115.3,<0.116)
+Requires-Dist: fastapi (>=0.110.1,<0.113)
 Requires-Dist: filetype (>=1.2.0,<2.0.0)
 Requires-Dist: httpx (>=0.23.0)
 Requires-Dist: lazify (>=0.4.0,<0.5.0)
@@ -35,15 +28,14 @@ Requires-Dist: packaging (>=23.1,<24.0)
 Requires-Dist: pydantic (>=1,<3)
 Requires-Dist: pyjwt (>=2.8.0,<3.0.0)
 Requires-Dist: python-dotenv (>=1.0.0,<2.0.0)
-Requires-Dist: python-multipart (>=0.0.18,<0.0.19)
+Requires-Dist: python-multipart (>=0.0.9,<0.0.10)
 Requires-Dist: python-socketio (>=5.11.0,<6.0.0)
-Requires-Dist: starlette (>=0.41.2,<0.42.0)
+Requires-Dist: starlette (>=0.37.2,<0.38.0)
 Requires-Dist: syncer (>=2.0.3,<3.0.0)
 Requires-Dist: tomli (>=2.0.1,<3.0.0)
 Requires-Dist: uptrace (>=1.22.0,<2.0.0)
 Requires-Dist: uvicorn (>=0.25.0,<0.26.0)
 Requires-Dist: watchfiles (>=0.20.0,<0.21.0)
-Project-URL: Documentation, https://docs.chainlit.io/
 Project-URL: Repository, https://github.com/Chainlit/chainlit
 Description-Content-Type: text/markdown
 
@@ -59,10 +51,22 @@ Description-Content-Type: text/markdown
 
 Chainlit is an open-source async Python framework which allows developers to build scalable Conversational AI or agentic applications.
 
+- ✅ ChatGPT-like application
+- ✅ Embedded Chatbot & Software Copilot
+- ✅ Slack & Discord
+- ✅ Custom frontend (build your own agentic experience)
+- ✅ API Endpoint
+
 Full documentation is available [here](https://docs.chainlit.io). You can ask Chainlit related questions to [Chainlit Help](https://help.chainlit.io/), an app built using Chainlit!
 
-https://github.com/user-attachments/assets/b3738aba-55c0-42fa-ac00-6efd1ee0d148
+> [!NOTE]  
+> Check out [Literal AI](https://literalai.com), our product to monitor and evaluate LLM applications! It works with any Python or TypeScript applications and [seamlessly](https://docs.chainlit.io/data-persistence/overview) with Chainlit by adding a `LITERAL_API_KEY` in your project.
+> 
+> Chainlit is developed and maintained by the Literal AI team, which is currently focused on expanding the capabilities of Literal AI. While we continue to support and maintain Chainlit, we are also committed to enabling the community to contribute, particularly in areas like integrations and data layers.
 
+<p align="center">
+    <img src="https://github.com/Chainlit/chainlit/assets/13104895/0c2cc7a9-766c-41d3-aae2-117a2d0eb8ed" alt="Chainlit user interface" width="80%"></img>
+</p>
 
 ## Installation
 
@@ -75,16 +79,6 @@ chainlit hello
 
 If this opens the `hello app` in your browser, you're all set!
 
-### Development version
-
-The latest in-development version can be installed straight from GitHub with:
-
-```sh
-pip install git+https://github.com/Chainlit/chainlit.git#subdirectory=backend/
-```
-
-(Requires Node and pnpm installed on the system.)
-
 ## 🚀 Quickstart
 
 ### 🐍 Pure Python
@@ -130,6 +124,24 @@ chainlit run demo.py -w
 
 <img src="/images/quick-start.png" alt="Quick Start"></img>
 
+## 🎉 Key Features and Integrations
+
+Full documentation is available [here](https://docs.chainlit.io). Key features:
+
+- [💬 Multi Modal chats](https://docs.chainlit.io/advanced-features/multi-modal)
+- [💭 Chain of Thought visualization](https://docs.chainlit.io/concepts/step)
+- [💾 Data persistence + human feedback](https://docs.chainlit.io/data-persistence/overview)
+- [🐛 Debug Mode](https://docs.chainlit.io/data-persistence/enterprise#debug-mode)
+- [👤 Authentication](https://docs.chainlit.io/authentication/overview)
+
+Chainlit is compatible with all Python programs and libraries. That being said, it comes with integrations for:
+
+- [LangChain](https://docs.chainlit.io/integrations/langchain)
+- [Llama Index](https://docs.chainlit.io/integrations/llama-index)
+- [Autogen](https://github.com/Chainlit/cookbook/tree/main/pyautogen)
+- [OpenAI Assistant](https://github.com/Chainlit/cookbook/tree/main/openai-assistant)
+- [Haystack](https://docs.chainlit.io/integrations/haystack)
+
 ## 📚 More Examples - Cookbook
 
 You can find various examples of Chainlit apps [here](https://github.com/Chainlit/cookbook) that leverage tools and services such as OpenAI, Anthropiс, LangChain, LlamaIndex, ChromaDB, Pinecone and more.

{chainlit-2.0.0 → chainlit-2.0.dev0}/README.md

@@ -10,10 +10,22 @@
 
 Chainlit is an open-source async Python framework which allows developers to build scalable Conversational AI or agentic applications.
 
+- ✅ ChatGPT-like application
+- ✅ Embedded Chatbot & Software Copilot
+- ✅ Slack & Discord
+- ✅ Custom frontend (build your own agentic experience)
+- ✅ API Endpoint
+
 Full documentation is available [here](https://docs.chainlit.io). You can ask Chainlit related questions to [Chainlit Help](https://help.chainlit.io/), an app built using Chainlit!
 
-https://github.com/user-attachments/assets/b3738aba-55c0-42fa-ac00-6efd1ee0d148
+> [!NOTE]  
+> Check out [Literal AI](https://literalai.com), our product to monitor and evaluate LLM applications! It works with any Python or TypeScript applications and [seamlessly](https://docs.chainlit.io/data-persistence/overview) with Chainlit by adding a `LITERAL_API_KEY` in your project.
+> 
+> Chainlit is developed and maintained by the Literal AI team, which is currently focused on expanding the capabilities of Literal AI. While we continue to support and maintain Chainlit, we are also committed to enabling the community to contribute, particularly in areas like integrations and data layers.
 
+<p align="center">
+    <img src="https://github.com/Chainlit/chainlit/assets/13104895/0c2cc7a9-766c-41d3-aae2-117a2d0eb8ed" alt="Chainlit user interface" width="80%"></img>
+</p>
 
 ## Installation
 
@@ -26,16 +38,6 @@ chainlit hello
 
 If this opens the `hello app` in your browser, you're all set!
 
-### Development version
-
-The latest in-development version can be installed straight from GitHub with:
-
-```sh
-pip install git+https://github.com/Chainlit/chainlit.git#subdirectory=backend/
-```
-
-(Requires Node and pnpm installed on the system.)
-
 ## 🚀 Quickstart
 
 ### 🐍 Pure Python
@@ -81,6 +83,24 @@ chainlit run demo.py -w
 
 <img src="/images/quick-start.png" alt="Quick Start"></img>
 
+## 🎉 Key Features and Integrations
+
+Full documentation is available [here](https://docs.chainlit.io). Key features:
+
+- [💬 Multi Modal chats](https://docs.chainlit.io/advanced-features/multi-modal)
+- [💭 Chain of Thought visualization](https://docs.chainlit.io/concepts/step)
+- [💾 Data persistence + human feedback](https://docs.chainlit.io/data-persistence/overview)
+- [🐛 Debug Mode](https://docs.chainlit.io/data-persistence/enterprise#debug-mode)
+- [👤 Authentication](https://docs.chainlit.io/authentication/overview)
+
+Chainlit is compatible with all Python programs and libraries. That being said, it comes with integrations for:
+
+- [LangChain](https://docs.chainlit.io/integrations/langchain)
+- [Llama Index](https://docs.chainlit.io/integrations/llama-index)
+- [Autogen](https://github.com/Chainlit/cookbook/tree/main/pyautogen)
+- [OpenAI Assistant](https://github.com/Chainlit/cookbook/tree/main/openai-assistant)
+- [Haystack](https://docs.chainlit.io/integrations/haystack)
+
 ## 📚 More Examples - Cookbook
 
 You can find various examples of Chainlit apps [here](https://github.com/Chainlit/cookbook) that leverage tools and services such as OpenAI, Anthropiс, LangChain, LlamaIndex, ChromaDB, Pinecone and more.

{chainlit-2.0.0 → chainlit-2.0.dev0}/chainlit/__init__.py

@@ -14,9 +14,6 @@ if env_found:
 import asyncio
 from typing import TYPE_CHECKING, Any, Dict
 
-from literalai import ChatGeneration, CompletionGeneration, GenerationMessage
-from pydantic.dataclasses import dataclass
-
 import chainlit.input_widget as input_widget
 from chainlit.action import Action
 from chainlit.cache import cache
@@ -25,8 +22,7 @@ from chainlit.chat_settings import ChatSettings
 from chainlit.context import context
 from chainlit.element import (
     Audio,
-    CustomElement,
-    Dataframe,
+    Component,
     File,
     Image,
     Pdf,
@@ -47,21 +43,22 @@ from chainlit.message import (
 )
 from chainlit.step import Step, step
 from chainlit.sync import make_async, run_sync
-from chainlit.types import ChatProfile, InputAudioChunk, OutputAudioChunk, Starter
+from chainlit.types import InputAudioChunk, OutputAudioChunk, ChatProfile, Starter
 from chainlit.user import PersistedUser, User
 from chainlit.user_session import user_session
 from chainlit.utils import make_module_getattr
 from chainlit.version import __version__
+from literalai import ChatGeneration, CompletionGeneration, GenerationMessage
+from pydantic.dataclasses import dataclass
 
 from .callbacks import (
     action_callback,
     author_rename,
-    data_layer,
     header_auth_callback,
     oauth_callback,
+    on_audio_start,
     on_audio_chunk,
     on_audio_end,
-    on_audio_start,
     on_chat_end,
     on_chat_resume,
     on_chat_start,
@@ -69,9 +66,7 @@ from .callbacks import (
     on_message,
     on_settings_update,
     on_stop,
-    on_window_message,
     password_auth_callback,
-    send_window_message,
     set_chat_profiles,
     set_starters,
 )
@@ -117,73 +112,79 @@ __getattr__ = make_module_getattr(
 )
 
 __all__ = [
-    "Action",
-    "AskActionMessage",
-    "AskFileMessage",
-    "AskUserMessage",
-    "AsyncLangchainCallbackHandler",
-    "Audio",
-    "ChatGeneration",
+    "__version__",
     "ChatProfile",
-    "ChatSettings",
-    "CompletionGeneration",
+    "Starter",
+    "user_session",
+    "chat_context",
     "CopilotFunction",
-    "CustomElement",
-    "Dataframe",
-    "ErrorMessage",
-    "File",
-    "GenerationMessage",
-    "HaystackAgentCallbackHandler",
-    "Image",
     "InputAudioChunk",
-    "LangchainCallbackHandler",
-    "LlamaIndexCallbackHandler",
-    "Message",
     "OutputAudioChunk",
-    "Pdf",
+    "Action",
+    "User",
     "PersistedUser",
+    "Audio",
+    "Pdf",
     "Plotly",
+    "Image",
+    "Text",
+    "Component",
     "Pyplot",
-    "Starter",
-    "Step",
+    "File",
     "Task",
     "TaskList",
     "TaskStatus",
-    "Text",
-    "User",
     "Video",
-    "__version__",
+    "ChatSettings",
+    "input_widget",
+    "Message",
+    "ErrorMessage",
+    "AskUserMessage",
+    "AskActionMessage",
+    "AskFileMessage",
+    "Step",
+    "step",
+    "ChatGeneration",
+    "CompletionGeneration",
+    "GenerationMessage",
+    "on_logout",
+    "on_chat_start",
+    "on_chat_end",
+    "on_chat_resume",
+    "on_stop",
     "action_callback",
     "author_rename",
+    "on_settings_update",
+    "password_auth_callback",
+    "header_auth_callback",
+    "sleep",
+    "run_sync",
+    "make_async",
     "cache",
-    "chat_context",
     "context",
-    "data_layer",
-    "header_auth_callback",
-    "input_widget",
-    "instrument_mistralai",
+    "LangchainCallbackHandler",
+    "AsyncLangchainCallbackHandler",
+    "LlamaIndexCallbackHandler",
+    "HaystackAgentCallbackHandler",
     "instrument_openai",
-    "make_async",
+    "instrument_mistralai",
+    "password_auth_callback",
+    "header_auth_callback",
     "oauth_callback",
-    "on_audio_chunk",
-    "on_audio_end",
-    "on_audio_start",
-    "on_chat_end",
-    "on_chat_resume",
-    "on_chat_start",
     "on_logout",
     "on_message",
-    "on_settings_update",
-    "on_stop",
-    "on_window_message",
-    "password_auth_callback",
-    "run_sync",
-    "send_window_message",
+    "on_chat_start",
+    "on_chat_resume",
     "set_chat_profiles",
     "set_starters",
-    "sleep",
-    "step",
-    "user_session",
+    "on_chat_end",
+    "on_audio_start",
+    "on_audio_chunk",
+    "on_audio_end",
+    "author_rename",
+    "on_stop",
+    "action_callback",
+    "on_settings_update",
 ]
 
 

{chainlit-2.0.0 → chainlit-2.0.dev0}/chainlit/action.py

@@ -1,30 +1,28 @@
 import uuid
-from typing import Dict, Optional
-
-from dataclasses_json import DataClassJsonMixin
-from pydantic import Field
-from pydantic.dataclasses import dataclass
+from typing import Optional
 
 from chainlit.context import context
 from chainlit.telemetry import trace_event
+from dataclasses_json import DataClassJsonMixin
+from pydantic.dataclasses import Field, dataclass
 
 
 @dataclass
 class Action(DataClassJsonMixin):
     # Name of the action, this should be used in the action_callback
     name: str
-    # The parameters to call this action with.
-    payload: Dict
-    # The label of the action. This is what the user will see.
+    # The value associated with the action. This is useful to differentiate between multiple actions with the same name.
+    value: str
+    # The label of the action. This is what the user will see. If not provided the name will be used.
     label: str = ""
-    # The tooltip of the action button. This is what the user will see when they hover the action.
-    tooltip: str = ""
-    # The lucid icon name for this action.
-    icon: Optional[str] = None
+    # The description of the action. This is what the user will see when they hover the action.
+    description: str = ""
     # This should not be set manually, only used internally.
     forId: Optional[str] = None
     # The ID of the action
     id: str = Field(default_factory=lambda: str(uuid.uuid4()))
+    # Show the action in a drawer menu
+    collapsed: bool = False
 
     def __post_init__(self) -> None:
         trace_event(f"init {self.__class__.__name__}")

chainlit-2.0.0/chainlit/auth/__init__.py → chainlit-2.0.dev0/chainlit/auth.py

@@ -1,16 +1,20 @@
 import os
+from datetime import datetime, timedelta
+from typing import Any, Dict
 
-from fastapi import Depends, HTTPException
-
+import jwt
 from chainlit.config import config
 from chainlit.data import get_data_layer
-from chainlit.logger import logger
 from chainlit.oauth_providers import get_configured_oauth_providers
+from chainlit.user import User
+from fastapi import Depends, HTTPException
+from fastapi.security import OAuth2PasswordBearer
+
+reuseable_oauth = OAuth2PasswordBearer(tokenUrl="/login", auto_error=False)
 
-from .cookie import OAuth2PasswordBearerWithCookie
-from .jwt import create_jwt, decode_jwt, get_jwt_secret
 
-reuseable_oauth = OAuth2PasswordBearerWithCookie(tokenUrl="/login", auto_error=False)
+def get_jwt_secret():
+    return os.environ.get("CHAINLIT_AUTH_SECRET")
 
 
 def ensure_jwt_secret():
@@ -41,36 +45,43 @@ def get_configuration():
         "oauthProviders": (
             get_configured_oauth_providers() if is_oauth_enabled() else []
         ),
-        "default_theme": config.ui.default_theme,
     }
 
 
+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
+        }
+    )
+    encoded_jwt = jwt.encode(to_encode, get_jwt_secret(), algorithm="HS256")
+    return encoded_jwt
+
+
 async def authenticate_user(token: str = Depends(reuseable_oauth)):
     try:
-        user = decode_jwt(token)
-    except Exception as e:
-        raise HTTPException(
-            status_code=401, detail="Invalid authentication token"
-        ) from e
-
+        dict = jwt.decode(
+            token,
+            get_jwt_secret(),
+            algorithms=["HS256"],
+            options={"verify_signature": True},
+        )
+        del dict["exp"]
+        user = User(**dict)
+    except Exception:
+        raise HTTPException(status_code=401, detail="Invalid authentication token")
     if data_layer := get_data_layer():
-        # Get or create persistent user if we've a data layer available.
         try:
             persisted_user = await data_layer.get_user(user.identifier)
             if persisted_user is None:
                 persisted_user = await data_layer.create_user(user)
-                assert persisted_user
-        except Exception as e:
-            logger.exception("Unable to get persisted_user from data layer: %s", e)
+        except Exception:
             return user
 
-        if user and user.display_name:
-            # Copy ephemeral display_name from authenticated user to persistent user.
-            persisted_user.display_name = user.display_name
-
         return persisted_user
-
-    return user
+    else:
+        return user
 
 
 async def get_current_user(token: str = Depends(reuseable_oauth)):
@@ -78,6 +89,3 @@ async def get_current_user(token: str = Depends(reuseable_oauth)):
         return None
 
     return await authenticate_user(token)
-
-
-__all__ = ["create_jwt", "get_configuration", "get_current_user"]

{chainlit-2.0.0 → chainlit-2.0.dev0}/chainlit/cache.py

@@ -1,7 +1,5 @@
-import importlib.util
 import os
 import threading
-from typing import Any
 
 from chainlit.config import config
 from chainlit.logger import logger
@@ -10,7 +8,11 @@ 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 and importlib.util.find_spec("langchain") is not None:
+    if use_cache:
+        try:
+            import langchain
+        except ImportError:
+            return
         from langchain.cache import SQLiteCache
         from langchain.globals import set_llm_cache
 
@@ -23,7 +25,7 @@ def init_lc_cache():
                 )
 
 
-_cache: dict[tuple, Any] = {}
+_cache = {}
 _cache_lock = threading.Lock()
 
 

{chainlit-2.0.0 → chainlit-2.0.dev0}/chainlit/callbacks.py

@@ -1,13 +1,8 @@
 import inspect
 from typing import Any, Awaitable, Callable, Dict, List, Optional
 
-from fastapi import Request, Response
-from starlette.datastructures import Headers
-
 from chainlit.action import Action
 from chainlit.config import config
-from chainlit.context import context
-from chainlit.data.base import BaseDataLayer
 from chainlit.message import Message
 from chainlit.oauth_providers import get_configured_oauth_providers
 from chainlit.step import Step, step
@@ -15,11 +10,13 @@ from chainlit.telemetry import trace
 from chainlit.types import ChatProfile, Starter, ThreadDict
 from chainlit.user import User
 from chainlit.utils import wrap_user_function
+from fastapi import Request, Response
+from starlette.datastructures import Headers
 
 
 @trace
 def password_auth_callback(
-    func: Callable[[str, str], Awaitable[Optional[User]]],
+    func: Callable[[str, str], Awaitable[Optional[User]]]
 ) -> Callable:
     """
     Framework agnostic decorator to authenticate the user.
@@ -41,7 +38,7 @@ def password_auth_callback(
 
 @trace
 def header_auth_callback(
-    func: Callable[[Headers], Awaitable[Optional[User]]],
+    func: Callable[[Headers], Awaitable[Optional[User]]]
 ) -> Callable:
     """
     Framework agnostic decorator to authenticate the user via a header
@@ -126,33 +123,6 @@ def on_message(func: Callable) -> Callable:
     return func
 
 
-@trace
-async def send_window_message(data: Any):
-    """
-    Send custom data to the host window via a window.postMessage event.
-
-    Args:
-        data (Any): The data to send with the event.
-    """
-    await context.emitter.send_window_message(data)
-
-
-@trace
-def on_window_message(func: Callable[[str], Any]) -> Callable:
-    """
-    Hook to react to javascript postMessage events coming from the UI.
-
-    Args:
-        func (Callable[[str], Any]): The function to be called when a window message is received.
-                                     Takes the message content as a string parameter.
-
-    Returns:
-        Callable[[str], Any]: The decorated on_window_message function.
-    """
-    config.code.on_window_message = wrap_user_function(func)
-    return func
-
-
 @trace
 def on_chat_start(func: Callable) -> Callable:
     """
@@ -207,7 +177,7 @@ def set_chat_profiles(
 
 @trace
 def set_starters(
-    func: Callable[[Optional["User"]], Awaitable[List["Starter"]]],
+    func: Callable[[Optional["User"]], Awaitable[List["Starter"]]]
 ) -> Callable:
     """
     Programmatic declaration of the available starter (can depend on the User from the session if authentication is setup).
@@ -251,7 +221,6 @@ def on_audio_start(func: Callable) -> Callable:
     config.code.on_audio_start = wrap_user_function(func, with_task=False)
     return func
 
-
 @trace
 def on_audio_chunk(func: Callable) -> Callable:
     """
@@ -285,7 +254,7 @@ def on_audio_end(func: Callable) -> Callable:
 
 @trace
 def author_rename(
-    func: Callable[[str], Awaitable[str]],
+    func: Callable[[str], Awaitable[str]]
 ) -> Callable[[str], Awaitable[str]]:
     """
     Useful to rename the author of message to display more friendly author names in the UI.
@@ -325,7 +294,7 @@ def action_callback(name: str) -> Callable:
     """
 
     def decorator(func: Callable[[Action], Any]):
-        config.code.action_callbacks[name] = wrap_user_function(func, with_task=False)
+        config.code.action_callbacks[name] = wrap_user_function(func, with_task=True)
         return func
 
     return decorator
@@ -346,17 +315,3 @@ def on_settings_update(
 
     config.code.on_settings_update = wrap_user_function(func, with_task=True)
     return func
-
-
-def data_layer(
-    func: Callable[[], BaseDataLayer],
-) -> Callable[[], BaseDataLayer]:
-    """
-    Hook to configure custom data layer.
-    """
-
-    # We don't use wrap_user_function here because:
-    # 1. We don't need to support async here and;
-    # 2. We don't want to change the API for get_data_layer() to be async, everywhere (at this point).
-    config.code.data_layer = func
-    return func

{chainlit-2.0.0 → chainlit-2.0.dev0}/chainlit/chat_context.py

@@ -25,10 +25,10 @@ class ChatContext:
 
         if context.session.id not in chat_contexts:
             chat_contexts[context.session.id] = []
-
+            
         if message not in chat_contexts[context.session.id]:
             chat_contexts[context.session.id].append(message)
-
+        
         return message
 
     def remove(self, message: "Message") -> bool:

{chainlit-2.0.0 → chainlit-2.0.dev0}/chainlit/chat_settings.py

@@ -1,10 +1,8 @@
 from typing import List
 
-from pydantic import Field
-from pydantic.dataclasses import dataclass
-
 from chainlit.context import context
 from chainlit.input_widget import InputWidget
+from pydantic.dataclasses import Field, dataclass
 
 
 @dataclass