chainlit 1.1.305__tar.gz → 1.1.400__tar.gz

{chainlit-1.1.305 → chainlit-1.1.400}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: chainlit
-Version: 1.1.305
+Version: 1.1.400
 Summary: Build Conversational AI.
 Home-page: https://github.com/Chainlit/chainlit
 License: Apache-2.0 license

{chainlit-1.1.305 → chainlit-1.1.400}/chainlit/__init__.py

@@ -1,3 +1,4 @@
+import inspect
 import os
 
 from dotenv import load_dotenv
@@ -21,11 +22,10 @@ if TYPE_CHECKING:
     from chainlit.openai import instrument_openai
     from chainlit.mistralai import instrument_mistralai
 
-from literalai import ChatGeneration, CompletionGeneration, GenerationMessage
-
 import chainlit.input_widget as input_widget
 from chainlit.action import Action
 from chainlit.cache import cache
+from chainlit.chat_context import chat_context
 from chainlit.chat_settings import ChatSettings
 from chainlit.config import config
 from chainlit.context import context
@@ -60,6 +60,7 @@ from chainlit.user import PersistedUser, User
 from chainlit.user_session import user_session
 from chainlit.utils import make_module_getattr, wrap_user_function
 from chainlit.version import __version__
+from literalai import ChatGeneration, CompletionGeneration, GenerationMessage
 
 if env_found:
     logger.info("Loaded .env file")
@@ -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
 
 
@@ -370,6 +383,7 @@ __all__ = [
     "ChatProfile",
     "Starter",
     "user_session",
+    "chat_context",
     "CopilotFunction",
     "AudioChunk",
     "Action",

chainlit-1.1.400/chainlit/chat_context.py

@@ -0,0 +1,64 @@
+from typing import TYPE_CHECKING, Dict, List
+
+from chainlit.context import context
+
+if TYPE_CHECKING:
+    from chainlit.message import Message
+
+chat_contexts: Dict[str, List["Message"]] = {}
+
+
+class ChatContext:
+    def get(self) -> List["Message"]:
+        if not context.session:
+            return []
+
+        if context.session.id not in chat_contexts:
+            # Create a new chat context
+            chat_contexts[context.session.id] = []
+
+        return chat_contexts[context.session.id].copy()
+
+    def add(self, message: "Message"):
+        if not context.session:
+            return
+
+        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:
+        if not context.session:
+            return False
+
+        if context.session.id not in chat_contexts:
+            return False
+
+        if message in chat_contexts[context.session.id]:
+            chat_contexts[context.session.id].remove(message)
+            return True
+
+        return False
+
+    def clear(self) -> None:
+        if context.session and context.session.id in chat_contexts:
+            chat_contexts[context.session.id] = []
+
+    def to_openai(self):
+        messages = []
+        for message in self.get():
+            if message.type == "assistant_message":
+                messages.append({"role": "assistant", "content": message.content})
+            elif message.type == "user_message":
+                messages.append({"role": "user", "content": message.content})
+            else:
+                messages.append({"role": "system", "content": message.content})
+
+        return messages
+
+
+chat_context = ChatContext()

{chainlit-1.1.305 → chainlit-1.1.400}/chainlit/config.py

@@ -91,6 +91,8 @@ auto_tag_thread = true
     # Sample rate of the audio
     sample_rate = 44100
 
+edit_message = true
+
 [UI]
 # Name of the assistant.
 name = "Assistant"
@@ -101,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 = ""
@@ -238,13 +240,14 @@ class FeaturesSettings(DataClassJsonMixin):
     latex: bool = False
     unsafe_allow_html: bool = False
     auto_tag_thread: 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-1.1.305 → chainlit-1.1.400}/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"],