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

chainlit/__init__.py

@@ -19,10 +19,12 @@ if TYPE_CHECKING:
     )
     from chainlit.llama_index.callbacks import LlamaIndexCallbackHandler
     from chainlit.openai import instrument_openai
+    from chainlit.mistralai import instrument_mistralai
 
 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
@@ -360,6 +362,7 @@ __getattr__ = make_module_getattr(
         "LlamaIndexCallbackHandler": "chainlit.llama_index.callbacks",
         "HaystackAgentCallbackHandler": "chainlit.haystack.callbacks",
         "instrument_openai": "chainlit.openai",
+        "instrument_mistralai": "chainlit.mistralai",
     }
 )
 
@@ -367,6 +370,7 @@ __all__ = [
     "ChatProfile",
     "Starter",
     "user_session",
+    "chat_context",
     "CopilotFunction",
     "AudioChunk",
     "Action",
@@ -416,6 +420,7 @@ __all__ = [
     "LlamaIndexCallbackHandler",
     "HaystackAgentCallbackHandler",
     "instrument_openai",
+    "instrument_mistralai",
 ]
 
 

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()