chainlit 1.1.305__py3-none-any.whl → 1.1.306__py3-none-any.whl

chainlit/__init__.py

@@ -21,11 +21,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 +59,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")
@@ -370,6 +370,7 @@ __all__ = [
     "ChatProfile",
     "Starter",
     "user_session",
+    "chat_context",
     "CopilotFunction",
     "AudioChunk",
     "Action",

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/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"
@@ -238,6 +240,7 @@ class FeaturesSettings(DataClassJsonMixin):
     latex: bool = False
     unsafe_allow_html: bool = False
     auto_tag_thread: bool = True
+    edit_message:bool = True
 
 
 @dataclass()