PyPI - chainlit - Versions diffs - 1.0.506__py3-none-any.whl → 1.1.0__py3-none-any.whl - Mend

chainlit 1.0.506py3-none-any.whl → 1.1.0py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of chainlit might be problematic. Click here for more details.

Files changed (30) hide show

chainlit/__init__.py +34 -1
chainlit/config.py +43 -21
chainlit/context.py +19 -7
chainlit/copilot/dist/index.js +650 -528
chainlit/data/__init__.py +4 -2
chainlit/data/acl.py +4 -1
chainlit/data/sql_alchemy.py +39 -31
chainlit/discord/__init__.py +6 -0
chainlit/discord/app.py +304 -0
chainlit/element.py +9 -3
chainlit/emitter.py +11 -2
chainlit/frontend/dist/assets/{index-d4233b49.js → index-0a52365d.js} +189 -185
chainlit/frontend/dist/assets/react-plotly-509d26a7.js +3602 -0
chainlit/frontend/dist/index.html +1 -1
chainlit/llama_index/callbacks.py +7 -6
chainlit/message.py +3 -3
chainlit/server.py +31 -4
chainlit/session.py +83 -62
chainlit/slack/__init__.py +6 -0
chainlit/slack/app.py +368 -0
chainlit/socket.py +91 -33
chainlit/step.py +25 -1
chainlit/types.py +21 -1
chainlit/user_session.py +6 -2
chainlit/utils.py +2 -1
{chainlit-1.0.506.dist-info → chainlit-1.1.0.dist-info}/METADATA +4 -3
{chainlit-1.0.506.dist-info → chainlit-1.1.0.dist-info}/RECORD +29 -25
chainlit/frontend/dist/assets/react-plotly-2b7fa4f9.js +0 -3484
{chainlit-1.0.506.dist-info → chainlit-1.1.0.dist-info}/WHEEL +0 -0
{chainlit-1.0.506.dist-info → chainlit-1.1.0.dist-info}/entry_points.txt +0 -0

chainlit/frontend/dist/index.html CHANGED Viewed

@@ -22,7 +22,7 @@
     <script>
       const global = globalThis;
     </script>
-    <script type="module" crossorigin src="/assets/index-d4233b49.js"></script>
+    <script type="module" crossorigin src="/assets/index-0a52365d.js"></script>
     <link rel="stylesheet" href="/assets/index-d088547c.css">
   </head>
   <body>

chainlit/llama_index/callbacks.py CHANGED Viewed

@@ -70,7 +70,7 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
     ) -> str:
         """Run when an event starts and return id of event."""
         self._restore_context()
         step_type: StepType = "undefined"
         if event_type == CBEventType.RETRIEVE:
             step_type = "retrieval"
@@ -104,7 +104,6 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
         """Run when an event ends."""
         step = self.steps.get(event_id, None)
         if payload is None or step is None:
             return
@@ -117,11 +116,13 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
             source_nodes = getattr(response, "source_nodes", None)
             if source_nodes:
                 source_refs = ", ".join(
-                    [f"Source {idx}" for idx, _ in enumerate(source_nodes)])
+                    [f"Source {idx}" for idx, _ in enumerate(source_nodes)]
+                )
                 step.elements = [
                     Text(
                         name=f"Source {idx}",
                         content=source.text or "Empty node",
+                        display="side",
                     )
                     for idx, source in enumerate(source_nodes)
                 ]
@@ -137,6 +138,7 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
                 step.elements = [
                     Text(
                         name=f"Source {idx}",
+                        display="side",
                         content=source.node.get_text() or "Empty node",
                     )
                     for idx, source in enumerate(sources)
@@ -173,7 +175,7 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
             token_count = self.total_llm_token_count or None
             raw_response = response.raw if response else None
             model = raw_response.get("model", None) if raw_response else None
             if messages and isinstance(response, ChatResponse):
                 msg: ChatMessage = response.message
                 step.generation = ChatGeneration(
@@ -198,7 +200,7 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
         else:
             step.output = payload
             self.context.loop.create_task(step.update())
         self.steps.pop(event_id, None)
     def _noop(self, *args, **kwargs):
@@ -206,4 +208,3 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
     start_trace = _noop
     end_trace = _noop

chainlit/message.py CHANGED Viewed

@@ -166,7 +166,7 @@ class MessageBase(ABC):
         step_dict = await self._create()
         await context.emitter.send_step(step_dict)
-        return self.id
+        return self
     async def stream_token(self, token: str, is_sequence=False):
         """
@@ -251,7 +251,7 @@ class Message(MessageBase):
         super().__post_init__()
-    async def send(self) -> str:
+    async def send(self):
         """
         Send the message to the UI and persist it in the cloud if a project ID is configured.
         Return the ID of the message.
@@ -268,7 +268,7 @@ class Message(MessageBase):
         # Run all tasks concurrently
         await asyncio.gather(*tasks)
-        return self.id
+        return self
     async def update(self):
         """

chainlit/server.py CHANGED Viewed

@@ -119,16 +119,27 @@ async def lifespan(app: FastAPI):
         watch_task = asyncio.create_task(watch_files_for_changes())
+    discord_task = None
+    if discord_bot_token := os.environ.get("DISCORD_BOT_TOKEN"):
+        from chainlit.discord.app import client
+        discord_task = asyncio.create_task(client.start(discord_bot_token))
     try:
         yield
     finally:
-        if watch_task:
-            try:
+        try:
+            if watch_task:
                 stop_event.set()
                 watch_task.cancel()
                 await watch_task
-            except asyncio.exceptions.CancelledError:
-                pass
+            if discord_task:
+                discord_task.cancel()
+                await discord_task
+        except asyncio.exceptions.CancelledError:
+            pass
         if FILES_DIRECTORY.is_dir():
             shutil.rmtree(FILES_DIRECTORY)
@@ -195,6 +206,18 @@ socket = SocketManager(
 )
+# -------------------------------------------------------------------------------
+#                               SLACK HANDLER
+# -------------------------------------------------------------------------------
+if os.environ.get("SLACK_BOT_TOKEN") and os.environ.get("SLACK_SIGNING_SECRET"):
+    from chainlit.slack.app import slack_app_handler
+    @app.post("/slack/events")
+    async def endpoint(req: Request):
+        return await slack_app_handler.handle(req)
 # -------------------------------------------------------------------------------
 #                               HTTP HANDLERS
 # -------------------------------------------------------------------------------
@@ -536,6 +559,10 @@ async def project_settings(
         chat_profiles = await config.code.set_chat_profiles(current_user)
         if chat_profiles:
             profiles = [p.to_dict() for p in chat_profiles]
+    if config.code.on_audio_chunk:
+        config.features.audio.enabled = True
     return JSONResponse(
         content={
             "ui": config.ui.to_dict(),

chainlit/session.py CHANGED Viewed

@@ -1,3 +1,4 @@
+import asyncio
 import json
 import mimetypes
 import shutil
@@ -23,7 +24,7 @@ if TYPE_CHECKING:
     from chainlit.types import FileDict, FileReference
     from chainlit.user import PersistedUser, User
-ClientType = Literal["app", "copilot", "teams", "slack"]
+ClientType = Literal["webapp", "copilot", "teams", "slack", "discord"]
 class JSONEncoderIgnoreNonSerializable(json.JSONEncoder):
@@ -34,17 +35,28 @@ class JSONEncoderIgnoreNonSerializable(json.JSONEncoder):
             return None
-def clean_metadata(metadata: Dict):
-    return json.loads(
+def clean_metadata(metadata: Dict, max_size: int = 1048576):
+    cleaned_metadata = json.loads(
         json.dumps(metadata, cls=JSONEncoderIgnoreNonSerializable, ensure_ascii=False)
     )
+    metadata_size = len(json.dumps(cleaned_metadata).encode('utf-8'))
+    if metadata_size > max_size:
+        # Redact the metadata if it exceeds the maximum size
+        cleaned_metadata = {
+            'message': f'Metadata size exceeds the limit of {max_size} bytes. Redacted.'
+        }
+    return cleaned_metadata
 class BaseSession:
     """Base object."""
     thread_id_to_resume: Optional[str] = None
     client_type: ClientType
+    current_task: Optional[asyncio.Task] = None
     def __init__(
         self,
@@ -63,6 +75,8 @@ class BaseSession:
         root_message: Optional["Message"] = None,
         # Chat profile selected before the session was created
         chat_profile: Optional[str] = None,
+        # Origin of the request
+        http_referer: Optional[str] = None,
     ):
         if thread_id:
             self.thread_id_to_resume = thread_id
@@ -74,19 +88,68 @@ class BaseSession:
         self.has_first_interaction = False
         self.user_env = user_env or {}
         self.chat_profile = chat_profile
+        self.http_referer = http_referer
+        self.files = {}  # type: Dict[str, "FileDict"]
         self.id = id
         self.chat_settings: Dict[str, Any] = {}
+    @property
+    def files_dir(self):
+        from chainlit.config import FILES_DIRECTORY
+        return FILES_DIRECTORY / self.id
     async def persist_file(
         self,
         name: str,
         mime: str,
         path: Optional[str] = None,
         content: Optional[Union[bytes, str]] = None,
-    ):
-        return None
+    ) -> "FileReference":
+        if not path and not content:
+            raise ValueError(
+                "Either path or content must be provided to persist a file"
+            )
+        self.files_dir.mkdir(exist_ok=True)
+        file_id = str(uuid.uuid4())
+        file_path = self.files_dir / file_id
+        file_extension = mimetypes.guess_extension(mime)
+        if file_extension:
+            file_path = file_path.with_suffix(file_extension)
+        if path:
+            # Copy the file from the given path
+            async with aiofiles.open(path, "rb") as src, aiofiles.open(
+                file_path, "wb"
+            ) as dst:
+                await dst.write(await src.read())
+        elif content:
+            # Write the provided content to the file
+            async with aiofiles.open(file_path, "wb") as buffer:
+                if isinstance(content, str):
+                    content = content.encode("utf-8")
+                await buffer.write(content)
+        # Get the file size
+        file_size = file_path.stat().st_size
+        # Store the file content in memory
+        self.files[file_id] = {
+            "id": file_id,
+            "path": file_path,
+            "name": name,
+            "type": mime,
+            "size": file_size,
+        }
+        return {"id": file_id}
     def to_persistable(self) -> Dict:
         from chainlit.user_session import user_sessions
@@ -94,6 +157,8 @@ class BaseSession:
         user_session = user_sessions.get(self.id) or {}  # type: Dict
         user_session["chat_settings"] = self.chat_settings
         user_session["chat_profile"] = self.chat_profile
+        user_session["http_referer"] = self.http_referer
+        user_session["client_type"] = self.client_type
         metadata = clean_metadata(user_session)
         return metadata
@@ -115,7 +180,8 @@ class HTTPSession(BaseSession):
         user_env: Optional[Dict[str, str]] = None,
         # Last message at the root of the chat
         root_message: Optional["Message"] = None,
-        # User specific environment variables. Empty if no user environment variables are required.
+        # Origin of the request
+        http_referer: Optional[str] = None,
     ):
         super().__init__(
             id=id,
@@ -125,8 +191,14 @@ class HTTPSession(BaseSession):
             client_type=client_type,
             user_env=user_env,
             root_message=root_message,
+            http_referer=http_referer,
         )
+    def delete(self):
+        """Delete the session."""
+        if self.files_dir.is_dir():
+            shutil.rmtree(self.files_dir)
 class WebsocketSession(BaseSession):
     """Internal web socket session object.
@@ -140,6 +212,8 @@ class WebsocketSession(BaseSession):
     socket id for convenience.
     """
+    to_clear: bool = False
     def __init__(
         self,
         # Id from the session cookie
@@ -165,6 +239,8 @@ class WebsocketSession(BaseSession):
         chat_profile: Optional[str] = None,
         # Languages of the user's browser
         languages: Optional[str] = None,
+        # Origin of the request
+        http_referer: Optional[str] = None,
     ):
         super().__init__(
             id=id,
@@ -175,77 +251,22 @@ class WebsocketSession(BaseSession):
             client_type=client_type,
             root_message=root_message,
             chat_profile=chat_profile,
+            http_referer=http_referer,
         )
         self.socket_id = socket_id
         self.emit_call = emit_call
         self.emit = emit
-        self.should_stop = False
         self.restored = False
         self.thread_queues = {}  # type: Dict[str, Deque[Callable]]
-        self.files = {}  # type: Dict[str, "FileDict"]
         ws_sessions_id[self.id] = self
         ws_sessions_sid[socket_id] = self
         self.languages = languages
-    @property
-    def files_dir(self):
-        from chainlit.config import FILES_DIRECTORY
-        return FILES_DIRECTORY / self.id
-    async def persist_file(
-        self,
-        name: str,
-        mime: str,
-        path: Optional[str] = None,
-        content: Optional[Union[bytes, str]] = None,
-    ) -> "FileReference":
-        if not path and not content:
-            raise ValueError(
-                "Either path or content must be provided to persist a file"
-            )
-        self.files_dir.mkdir(exist_ok=True)
-        file_id = str(uuid.uuid4())
-        file_path = self.files_dir / file_id
-        file_extension = mimetypes.guess_extension(mime)
-        if file_extension:
-            file_path = file_path.with_suffix(file_extension)
-        if path:
-            # Copy the file from the given path
-            async with aiofiles.open(path, "rb") as src, aiofiles.open(
-                file_path, "wb"
-            ) as dst:
-                await dst.write(await src.read())
-        elif content:
-            # Write the provided content to the file
-            async with aiofiles.open(file_path, "wb") as buffer:
-                if isinstance(content, str):
-                    content = content.encode("utf-8")
-                await buffer.write(content)
-        # Get the file size
-        file_size = file_path.stat().st_size
-        # Store the file content in memory
-        self.files[file_id] = {
-            "id": file_id,
-            "path": file_path,
-            "name": name,
-            "type": mime,
-            "size": file_size,
-        }
-        return {"id": file_id}
     def restore(self, new_socket_id: str):
         """Associate a new socket id to the session."""
         ws_sessions_sid.pop(self.socket_id, None)

chainlit/slack/__init__.py ADDED Viewed

@@ -0,0 +1,6 @@
+try:
+    import slack_bolt
+except ModuleNotFoundError:
+    raise ValueError(
+        "The slack_bolt package is required to integrate Chainlit with a Slack app. Run `pip install slack_bolt --upgrade`"
+    )

chainlit 1.0.506__py3-none-any.whl → 1.1.0__py3-none-any.whl

Potentially problematic release.

chainlit 1.0.506py3-none-any.whl → 1.1.0py3-none-any.whl