chainlit 1.3.0rc1__py3-none-any.whl → 1.3.1__py3-none-any.whl

chainlit/__init__.py

@@ -43,7 +43,7 @@ from chainlit.message import (
 )
 from chainlit.step import Step, step
 from chainlit.sync import make_async, run_sync
-from chainlit.types import InputAudioChunk, OutputAudioChunk, ChatProfile, Starter
+from chainlit.types import AudioChunk, ChatProfile, Starter
 from chainlit.user import PersistedUser, User
 from chainlit.user_session import user_session
 from chainlit.utils import make_module_getattr
@@ -56,7 +56,6 @@ from .callbacks import (
     author_rename,
     header_auth_callback,
     oauth_callback,
-    on_audio_start,
     on_audio_chunk,
     on_audio_end,
     on_chat_end,
@@ -118,8 +117,7 @@ __all__ = [
     "user_session",
     "chat_context",
     "CopilotFunction",
-    "InputAudioChunk",
-    "OutputAudioChunk",
+    "AudioChunk",
     "Action",
     "User",
     "PersistedUser",
@@ -178,7 +176,6 @@ __all__ = [
     "set_chat_profiles",
     "set_starters",
     "on_chat_end",
-    "on_audio_start",
     "on_audio_chunk",
     "on_audio_end",
     "author_rename",

chainlit/auth.py

@@ -79,6 +79,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/callbacks.py

@@ -209,25 +209,13 @@ def on_chat_end(func: Callable) -> Callable:
     return func
 
 
-@trace
-def on_audio_start(func: Callable) -> Callable:
-    """
-    Hook to react to the user initiating audio.
-
-    Returns:
-        Callable[], Any]: The decorated hook.
-    """
-
-    config.code.on_audio_start = wrap_user_function(func, with_task=False)
-    return func
-
 @trace
 def on_audio_chunk(func: Callable) -> Callable:
     """
     Hook to react to the audio chunks being sent.
 
     Args:
-        chunk (InputAudioChunk): The audio chunk being sent.
+        chunk (AudioChunk): The audio chunk being sent.
 
     Returns:
         Callable[], Any]: The decorated hook.
@@ -242,6 +230,9 @@ def on_audio_end(func: Callable) -> Callable:
     """
     Hook to react to the audio stream ending. This is called after the last audio chunk is sent.
 
+    Args:
+    elements ([List[Element]): The files that were uploaded before starting the audio stream (if any).
+
     Returns:
         Callable[], Any]: The decorated hook.
     """

chainlit/config.py

@@ -28,8 +28,9 @@ from ._utils import is_path_inside
 
 if TYPE_CHECKING:
     from chainlit.action import Action
+    from chainlit.element import ElementBased
     from chainlit.message import Message
-    from chainlit.types import InputAudioChunk, ChatProfile, Starter, ThreadDict
+    from chainlit.types import AudioChunk, ChatProfile, Starter, ThreadDict
     from chainlit.user import User
     from fastapi import Request, Response
 
@@ -92,8 +93,18 @@ edit_message = true
     max_size_mb = 500
 
 [features.audio]
+    # Threshold for audio recording
+    min_decibels = -45
+    # Delay for the user to start speaking in MS
+    initial_silence_timeout = 3000
+    # Delay for the user to continue speaking in MS. If the user stops speaking for this duration, the recording will stop.
+    silence_timeout = 1500
+    # Above this duration (MS), the recording will forcefully stop.
+    max_duration = 15000
+    # Duration of the audio chunks in MS
+    chunk_duration = 1000
     # Sample rate of the audio
-    sample_rate = 24000
+    sample_rate = 44100
 
 [UI]
 # Name of the assistant.
@@ -226,7 +237,12 @@ class SpontaneousFileUploadFeature(DataClassJsonMixin):
 
 @dataclass
 class AudioFeature(DataClassJsonMixin):
-    sample_rate: int = 24000
+    min_decibels: int = -45
+    initial_silence_timeout: int = 2000
+    silence_timeout: int = 1500
+    chunk_duration: int = 1000
+    max_duration: int = 15000
+    sample_rate: int = 44100
     enabled: bool = False
 
 
@@ -281,9 +297,8 @@ class CodeSettings:
     on_chat_end: Optional[Callable[[], Any]] = None
     on_chat_resume: Optional[Callable[["ThreadDict"], Any]] = None
     on_message: Optional[Callable[["Message"], Any]] = None
-    on_audio_start: Optional[Callable[[], Any]] = None
-    on_audio_chunk: Optional[Callable[["InputAudioChunk"], Any]] = None
-    on_audio_end: Optional[Callable[[], Any]] = None
+    on_audio_chunk: Optional[Callable[["AudioChunk"], Any]] = None
+    on_audio_end: Optional[Callable[[List["ElementBased"]], Any]] = None
 
     author_rename: Optional[Callable[[str], Awaitable[str]]] = None
     on_settings_update: Optional[Callable[[Dict[str, Any]], Any]] = None

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