chainlit 1.1.306__py3-none-any.whl → 1.1.400rc0__py3-none-any.whl

chainlit/__init__.py

@@ -1,3 +1,4 @@
+import inspect
 import os
 
 from dotenv import load_dotenv
@@ -156,7 +157,15 @@ def on_message(func: Callable) -> Callable:
         Callable[[str], Any]: The decorated on_message function.
     """
 
-    config.code.on_message = wrap_user_function(func)
+    async def with_parent_id(message: Message):
+        async with Step(name="on_message", type="run", parent_id=message.id) as s:
+            s.input = message.content
+            if len(inspect.signature(func).parameters) > 0:
+                await func(message)
+            else:
+                await func()
+
+    config.code.on_message = wrap_user_function(with_parent_id)
     return func
 
 
@@ -172,7 +181,9 @@ def on_chat_start(func: Callable) -> Callable:
         Callable[], Any]: The decorated hook.
     """
 
-    config.code.on_chat_start = wrap_user_function(func, with_task=True)
+    config.code.on_chat_start = wrap_user_function(
+        step(func, name="on_chat_start", type="run"), with_task=True
+    )
     return func
 
 
@@ -270,7 +281,9 @@ def on_audio_end(func: Callable) -> Callable:
         Callable[], Any]: The decorated hook.
     """
 
-    config.code.on_audio_end = wrap_user_function(func, with_task=True)
+    config.code.on_audio_end = wrap_user_function(
+        step(func, name="on_audio_end", type="run"), with_task=True
+    )
     return func
 
 

chainlit/config.py

@@ -103,8 +103,8 @@ name = "Assistant"
 # Large size content are by default collapsed for a cleaner ui
 default_collapse_content = true
 
-# Hide the chain of thought details from the user in the UI.
-hide_cot = false
+# Chain of Thought (CoT) display mode. Can be "hidden", "tool_call" or "full".
+cot = "full"
 
 # Link to your github repo. This will add a github button in the UI's header.
 # github = ""
@@ -240,14 +240,14 @@ class FeaturesSettings(DataClassJsonMixin):
     latex: bool = False
     unsafe_allow_html: bool = False
     auto_tag_thread: bool = True
-    edit_message:bool = True
+    edit_message: bool = True
 
 
 @dataclass()
 class UISettings(DataClassJsonMixin):
     name: str
     description: str = ""
-    hide_cot: bool = False
+    cot: Literal["hidden", "tool_call", "full"] = "full"
     # Large size content are by default collapsed for a cleaner ui
     default_collapse_content: bool = True
     github: Optional[str] = None

chainlit/context.py

@@ -11,6 +11,8 @@ if TYPE_CHECKING:
     from chainlit.step import Step
     from chainlit.user import PersistedUser, User
 
+CL_RUN_NAMES = ["on_chat_start", "on_message", "on_audio_end"]
+
 
 class ChainlitContextException(Exception):
     def __init__(self, msg="Chainlit context not found", *args, **kwargs):
@@ -28,6 +30,13 @@ class ChainlitContext:
         if self.active_steps:
             return self.active_steps[-1]
 
+    @property
+    def current_run(self):
+        if self.active_steps:
+            return next(
+                (step for step in self.active_steps if step.name in CL_RUN_NAMES), None
+            )
+
     def __init__(
         self,
         session: Union["HTTPSession", "WebsocketSession"],