PyPI - chainlit - Versions diffs - 1.1.305__py3-none-any.whl → 1.1.400__py3-none-any.whl - Mend

chainlit 1.1.305py3-none-any.whl → 1.1.400py3-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 (48) hide show

chainlit/__init__.py +19 -5
chainlit/chat_context.py +64 -0
chainlit/config.py +6 -3
chainlit/context.py +9 -0
chainlit/copilot/dist/index.js +150 -147
chainlit/data/__init__.py +8 -12
chainlit/data/sql_alchemy.py +0 -4
chainlit/discord/app.py +44 -24
chainlit/emitter.py +9 -11
chainlit/frontend/dist/assets/{DailyMotion-4f715d15.js → DailyMotion-e2c926d0.js} +1 -1
chainlit/frontend/dist/assets/{Facebook-25f45c09.js → Facebook-88f80aba.js} +1 -1
chainlit/frontend/dist/assets/{FilePlayer-04482650.js → FilePlayer-febd1b0d.js} +1 -1
chainlit/frontend/dist/assets/{Kaltura-37152a96.js → Kaltura-606fe0bd.js} +1 -1
chainlit/frontend/dist/assets/{Mixcloud-914b75ee.js → Mixcloud-b3db090a.js} +1 -1
chainlit/frontend/dist/assets/{Mux-fb751398.js → Mux-ca847a7d.js} +1 -1
chainlit/frontend/dist/assets/{Preview-85fbb8da.js → Preview-7a3747b5.js} +1 -1
chainlit/frontend/dist/assets/{SoundCloud-8afad6c0.js → SoundCloud-0da01fb1.js} +1 -1
chainlit/frontend/dist/assets/{Streamable-08844d93.js → Streamable-4a6ab048.js} +1 -1
chainlit/frontend/dist/assets/{Twitch-1b95f5c8.js → Twitch-3e260619.js} +1 -1
chainlit/frontend/dist/assets/{Vidyard-5028fa2f.js → Vidyard-73692980.js} +1 -1
chainlit/frontend/dist/assets/{Vimeo-ca732959.js → Vimeo-6627c7a8.js} +1 -1
chainlit/frontend/dist/assets/{Wistia-74e58d71.js → Wistia-27df9c66.js} +1 -1
chainlit/frontend/dist/assets/{YouTube-bdf4ca10.js → YouTube-a11d419d.js} +1 -1
chainlit/frontend/dist/assets/index-d66d1800.js +730 -0
chainlit/frontend/dist/assets/{react-plotly-cf9b99cc.js → react-plotly-012ed517.js} +1 -1
chainlit/frontend/dist/index.html +1 -1
chainlit/haystack/callbacks.py +1 -3
chainlit/langchain/callbacks.py +39 -20
chainlit/llama_index/callbacks.py +8 -28
chainlit/message.py +4 -17
chainlit/mistralai/__init__.py +2 -5
chainlit/openai/__init__.py +0 -2
chainlit/server.py +1 -1
chainlit/session.py +2 -12
chainlit/slack/app.py +38 -31
chainlit/socket.py +37 -1
chainlit/step.py +37 -32
chainlit/teams/app.py +23 -18
chainlit/translations/zh-CN.json +229 -0
chainlit/types.py +0 -1
chainlit/user_session.py +0 -3
chainlit/utils.py +1 -0
{chainlit-1.1.305.dist-info → chainlit-1.1.400.dist-info}/METADATA +1 -1
chainlit-1.1.400.dist-info/RECORD +82 -0
chainlit/frontend/dist/assets/index-621140f9.js +0 -727
chainlit-1.1.305.dist-info/RECORD +0 -80
{chainlit-1.1.305.dist-info → chainlit-1.1.400.dist-info}/WHEEL +0 -0
{chainlit-1.1.305.dist-info → chainlit-1.1.400.dist-info}/entry_points.txt +0 -0

chainlit/frontend/dist/index.html CHANGED Viewed

@@ -21,7 +21,7 @@
     <script>
       const global = globalThis;
     </script>
-    <script type="module" crossorigin src="/assets/index-621140f9.js"></script>
+    <script type="module" crossorigin src="/assets/index-d66d1800.js"></script>
     <link rel="stylesheet" href="/assets/index-aaf974a9.css">
   </head>
   <body>

chainlit/haystack/callbacks.py CHANGED Viewed

@@ -68,9 +68,7 @@ class HaystackAgentCallbackHandler:
             self.last_tokens: List[str] = []
             self.answer_reached = False
-        root_message = context.session.root_message
-        parent_id = root_message.id if root_message else None
-        run_step = Step(name=self.agent_name, type="run", parent_id=parent_id)
+        run_step = Step(name=self.agent_name, type="run")
         run_step.start = utc_now()
         run_step.input = kwargs

chainlit/langchain/callbacks.py CHANGED Viewed

@@ -1,11 +1,8 @@
 import json
 import time
-from typing import Any, Dict, List, Optional, TypedDict, Union
+from typing import Any, Dict, List, Optional, Tuple, TypedDict, Union
 from uuid import UUID
-from chainlit.context import context_var
-from chainlit.message import Message
-from chainlit.step import Step
 from langchain.callbacks.tracers.base import BaseTracer
 from langchain.callbacks.tracers.schemas import Run
 from langchain.schema import BaseMessage
@@ -15,6 +12,10 @@ from literalai import ChatGeneration, CompletionGeneration, GenerationMessage
 from literalai.helper import utc_now
 from literalai.step import TrueStepType
+from chainlit.context import context_var
+from chainlit.message import Message
+from chainlit.step import Step
 DEFAULT_ANSWER_PREFIX_TOKENS = ["Final", "Answer", ":"]
@@ -229,7 +230,24 @@ class GenerationHelper:
         return provider, model, tools, settings
-DEFAULT_TO_IGNORE = ["Runnable", "<lambda>"]
+def process_content(content: Any) -> Tuple[Dict, Optional[str]]:
+    if content is None:
+        return {}, None
+    if isinstance(content, dict):
+        return content, "json"
+    elif isinstance(content, str):
+        return {"content": content}, "text"
+    else:
+        return {"content": str(content)}, "text"
+DEFAULT_TO_IGNORE = [
+    "RunnableSequence",
+    "RunnableParallel",
+    "RunnableAssign",
+    "RunnableLambda",
+    "<lambda>",
+]
 DEFAULT_TO_KEEP = ["retriever", "llm", "agent", "chain", "tool"]
@@ -267,8 +285,6 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
         if self.context.current_step:
             self.root_parent_id = self.context.current_step.id
-        elif self.context.session.root_message:
-            self.root_parent_id = self.context.session.root_message.id
         else:
             self.root_parent_id = None
@@ -431,9 +447,6 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
                 self.ignored_runs.add(str(run.id))
             return ignore, parent_id
-    def _is_annotable(self, run: Run):
-        return run.run_type in ["retriever", "llm"]
     def _start_trace(self, run: Run) -> None:
         super()._start_trace(run)
         context_var.set(self.context)
@@ -452,7 +465,8 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
         if run.run_type == "agent":
             step_type = "run"
         elif run.run_type == "chain":
-            pass
+            if not self.steps:
+                step_type = "run"
         elif run.run_type == "llm":
             step_type = "llm"
         elif run.run_type == "retriever":
@@ -462,20 +476,18 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
         elif run.run_type == "embedding":
             step_type = "embedding"
-        if not self.steps and step_type != "llm":
-            step_type = "run"
-        disable_feedback = not self._is_annotable(run)
         step = Step(
             id=str(run.id),
             name=run.name,
             type=step_type,
             parent_id=parent_id,
-            disable_feedback=disable_feedback,
         )
         step.start = utc_now()
-        step.input = run.inputs
+        step.input, language = process_content(run.inputs)
+        if language is not None:
+            if step.metadata is None:
+                step.metadata = {}
+            step.metadata["language"] = language
         self.steps[str(run.id)] = step
@@ -499,6 +511,9 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
             generations = (run.outputs or {}).get("generations", [])
             generation = generations[0][0]
             variables = self.generation_inputs.get(str(run.parent_run_id), {})
+            variables = {
+                k: process_content(v) for k, v in variables.items() if v is not None
+            }
             if message := generation.get("message"):
                 chat_start = self.chat_generations[str(run.id)]
                 duration = time.time() - chat_start["start"]
@@ -529,7 +544,11 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
                             "prompt_id"
                         ]
                         if custom_variables := m.additional_kwargs.get("variables"):
-                            current_step.generation.variables = custom_variables
+                            current_step.generation.variables = {
+                                k: process_content(v)
+                                for k, v in custom_variables.items()
+                                if v is not None
+                            }
                     break
                 current_step.language = "json"
@@ -573,7 +592,7 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
             output = outputs.get(output_keys[0], outputs)
         if current_step:
-            current_step.output = output
+            current_step.output, _ = output
             current_step.end = utc_now()
             self._run_sync(current_step.update())

chainlit/llama_index/callbacks.py CHANGED Viewed

@@ -33,33 +33,17 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
             event_starts_to_ignore=event_starts_to_ignore,
             event_ends_to_ignore=event_ends_to_ignore,
         )
-        self.context = context_var.get()
         self.steps = {}
     def _get_parent_id(self, event_parent_id: Optional[str] = None) -> Optional[str]:
         if event_parent_id and event_parent_id in self.steps:
             return event_parent_id
-        elif self.context.current_step:
-            return self.context.current_step.id
-        elif self.context.session.root_message:
-            return self.context.session.root_message.id
+        elif context_var.get().current_step:
+            return context_var.get().current_step.id
         else:
             return None
-    def _restore_context(self) -> None:
-        """Restore Chainlit context in the current thread
-        Chainlit context is local to the main thread, and LlamaIndex
-        runs the callbacks in its own threads, so they don't have a
-        Chainlit context by default.
-        This method restores the context in which the callback handler
-        has been created (it's always created in the main thread), so
-        that we can actually send messages.
-        """
-        context_var.set(self.context)
     def on_event_start(
         self,
         event_type: CBEventType,
@@ -69,8 +53,6 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
         **kwargs: Any,
     ) -> 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 = "tool"
@@ -86,12 +68,12 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
             type=step_type,
             parent_id=self._get_parent_id(parent_id),
             id=event_id,
-            disable_feedback=True,
         )
         self.steps[event_id] = step
         step.start = utc_now()
         step.input = payload or {}
-        self.context.loop.create_task(step.send())
+        context_var.get().loop.create_task(step.send())
         return event_id
     def on_event_end(
@@ -107,8 +89,6 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
         if payload is None or step is None:
             return
-        self._restore_context()
         step.end = utc_now()
         if event_type == CBEventType.QUERY:
@@ -127,7 +107,7 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
                     for idx, source in enumerate(source_nodes)
                 ]
                 step.output = f"Retrieved the following sources: {source_refs}"
-                self.context.loop.create_task(step.update())
+                context_var.get().loop.create_task(step.update())
         elif event_type == CBEventType.RETRIEVE:
             sources = payload.get(EventPayload.NODES)
@@ -144,7 +124,7 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
                     for idx, source in enumerate(sources)
                 ]
                 step.output = f"Retrieved the following sources: {source_refs}"
-            self.context.loop.create_task(step.update())
+            context_var.get().loop.create_task(step.update())
         elif event_type == CBEventType.LLM:
             formatted_messages = payload.get(
@@ -195,11 +175,11 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
                     token_count=token_count,
                 )
-            self.context.loop.create_task(step.update())
+            context_var.get().loop.create_task(step.update())
         else:
             step.output = payload
-            self.context.loop.create_task(step.update())
+            context_var.get().loop.create_task(step.update())
         self.steps.pop(event_id, None)

chainlit/message.py CHANGED Viewed

@@ -6,6 +6,7 @@ from abc import ABC
 from typing import Dict, List, Optional, Union, cast
 from chainlit.action import Action
+from chainlit.chat_context import chat_context
 from chainlit.config import config
 from chainlit.context import context, local_steps
 from chainlit.data import get_data_layer
@@ -31,7 +32,6 @@ class MessageBase(ABC):
     author: str
     content: str = ""
     type: MessageStepType = "assistant_message"
-    disable_feedback = False
     streaming = False
     created_at: Union[str, None] = None
     fail_on_persist_error: bool = False
@@ -66,7 +66,6 @@ class MessageBase(ABC):
             content=_dict["output"],
             author=_dict.get("name", config.ui.name),
             type=type,  # type: ignore
-            disable_feedback=_dict.get("disableFeedback", False),
             language=_dict.get("language"),
         )
@@ -86,7 +85,6 @@ class MessageBase(ABC):
             "createdAt": self.created_at,
             "language": self.language,
             "streaming": self.streaming,
-            "disableFeedback": self.disable_feedback,
             "isError": self.is_error,
             "waitForAnswer": self.wait_for_answer,
             "indent": self.indent,
@@ -108,6 +106,7 @@ class MessageBase(ABC):
             self.streaming = False
         step_dict = self.to_dict()
+        chat_context.add(self)
         data_layer = get_data_layer()
         if data_layer:
@@ -127,7 +126,7 @@ class MessageBase(ABC):
         Remove a message already sent to the UI.
         """
         trace_event("remove_message")
+        chat_context.remove(self)
         step_dict = self.to_dict()
         data_layer = get_data_layer()
         if data_layer:
@@ -169,6 +168,7 @@ class MessageBase(ABC):
             self.streaming = False
         step_dict = await self._create()
+        chat_context.add(self)
         await context.emitter.send_step(step_dict)
         return self
@@ -205,7 +205,6 @@ class Message(MessageBase):
         language (str, optional): Language of the code is the content is code. See https://react-code-blocks-rajinwonderland.vercel.app/?path=/story/codeblock--supported-languages for a list of supported languages.
         actions (List[Action], optional): A list of actions to send with the message.
         elements (List[ElementBased], optional): A list of elements to send with the message.
-        disable_feedback (bool, optional): Hide the feedback buttons for this specific message
     """
     def __init__(
@@ -215,7 +214,6 @@ class Message(MessageBase):
         language: Optional[str] = None,
         actions: Optional[List[Action]] = None,
         elements: Optional[List[ElementBased]] = None,
-        disable_feedback: bool = False,
         type: MessageStepType = "assistant_message",
         metadata: Optional[Dict] = None,
         tags: Optional[List[str]] = None,
@@ -254,7 +252,6 @@ class Message(MessageBase):
         self.type = type
         self.actions = actions if actions is not None else []
         self.elements = elements if elements is not None else []
-        self.disable_feedback = disable_feedback
         super().__post_init__()
@@ -266,8 +263,6 @@ class Message(MessageBase):
         trace_event("send_message")
         await super().send()
-        context.session.root_message = self
         # Create tasks for all actions and elements
         tasks = [action.send(for_id=self.id) for action in self.actions]
         tasks.extend(element.send(for_id=self.id) for element in self.elements)
@@ -352,7 +347,6 @@ class AskUserMessage(AskMessageBase):
     Args:
         content (str): The content of the prompt.
         author (str, optional): The author of the message, this will be used in the UI. Defaults to the assistant name (see config).
-        disable_feedback (bool, optional): Hide the feedback buttons for this specific message
         timeout (int, optional): The number of seconds to wait for an answer before raising a TimeoutError.
         raise_on_timeout (bool, optional): Whether to raise a socketio TimeoutError if the user does not answer in time.
     """
@@ -362,7 +356,6 @@ class AskUserMessage(AskMessageBase):
         content: str,
         author: str = config.ui.name,
         type: MessageStepType = "assistant_message",
-        disable_feedback: bool = True,
         timeout: int = 60,
         raise_on_timeout: bool = False,
     ):
@@ -370,7 +363,6 @@ class AskUserMessage(AskMessageBase):
         self.author = author
         self.timeout = timeout
         self.type = type
-        self.disable_feedback = disable_feedback
         self.raise_on_timeout = raise_on_timeout
         super().__post_init__()
@@ -417,7 +409,6 @@ class AskFileMessage(AskMessageBase):
         max_size_mb (int, optional): Maximum size per file in MB. Maximum value is 100.
         max_files (int, optional): Maximum number of files to upload. Maximum value is 10.
         author (str, optional): The author of the message, this will be used in the UI. Defaults to the assistant name (see config).
-        disable_feedback (bool, optional): Hide the feedback buttons for this specific message
         timeout (int, optional): The number of seconds to wait for an answer before raising a TimeoutError.
         raise_on_timeout (bool, optional): Whether to raise a socketio TimeoutError if the user does not answer in time.
     """
@@ -430,7 +421,6 @@ class AskFileMessage(AskMessageBase):
         max_files=1,
         author=config.ui.name,
         type: MessageStepType = "assistant_message",
-        disable_feedback: bool = True,
         timeout=90,
         raise_on_timeout=False,
     ):
@@ -442,7 +432,6 @@ class AskFileMessage(AskMessageBase):
         self.author = author
         self.timeout = timeout
         self.raise_on_timeout = raise_on_timeout
-        self.disable_feedback = disable_feedback
         super().__post_init__()
@@ -506,14 +495,12 @@ class AskActionMessage(AskMessageBase):
         content: str,
         actions: List[Action],
         author=config.ui.name,
-        disable_feedback=True,
         timeout=90,
         raise_on_timeout=False,
     ):
         self.content = content
         self.actions = actions
         self.author = author
-        self.disable_feedback = disable_feedback
         self.timeout = timeout
         self.raise_on_timeout = raise_on_timeout

chainlit/mistralai/__init__.py CHANGED Viewed

@@ -1,12 +1,11 @@
 import asyncio
 from typing import Union
-from literalai import ChatGeneration, CompletionGeneration
-from literalai.helper import timestamp_utc
 from chainlit.context import get_context
 from chainlit.step import Step
 from chainlit.utils import check_module_version
+from literalai import ChatGeneration, CompletionGeneration
+from literalai.helper import timestamp_utc
 def instrument_mistralai():
@@ -20,8 +19,6 @@ def instrument_mistralai():
         parent_id = None
         if context.current_step:
             parent_id = context.current_step.id
-        elif context.session.root_message:
-            parent_id = context.session.root_message.id
         step = Step(
             name=generation.model if generation.model else generation.provider,

chainlit/openai/__init__.py CHANGED Viewed

@@ -24,8 +24,6 @@ def instrument_openai():
         parent_id = None
         if context.current_step:
             parent_id = context.current_step.id
-        elif context.session.root_message:
-            parent_id = context.session.root_message.id
         step = Step(
             name=generation.model if generation.model else generation.provider,

chainlit/server.py CHANGED Viewed

@@ -266,7 +266,7 @@ def get_html_template():
     )
     url = config.ui.github or default_url
     meta_image_url = config.ui.custom_meta_image_url or default_meta_image_url
-    favicon_path = ROOT_PATH + "/favicon" if ROOT_PATH else "/favicon"
+    favicon_path = "/favicon"
     tags = f"""<title>{config.ui.name}</title>
     <link rel="icon" href="{favicon_path}" />

chainlit/session.py CHANGED Viewed

@@ -35,17 +35,16 @@ class JSONEncoderIgnoreNonSerializable(json.JSONEncoder):
             return None
 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'))
+    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.'
+            "message": f"Metadata size exceeds the limit of {max_size} bytes. Redacted."
         }
     return cleaned_metadata
@@ -71,8 +70,6 @@ class BaseSession:
         token: Optional[str],
         # User specific environment variables. Empty if no user environment variables are required.
         user_env: Optional[Dict[str, str]],
-        # Last message at the root of the chat
-        root_message: Optional["Message"] = None,
         # Chat profile selected before the session was created
         chat_profile: Optional[str] = None,
         # Origin of the request
@@ -84,7 +81,6 @@ class BaseSession:
         self.user = user
         self.client_type = client_type
         self.token = token
-        self.root_message = root_message
         self.has_first_interaction = False
         self.user_env = user_env or {}
         self.chat_profile = chat_profile
@@ -178,8 +174,6 @@ class HTTPSession(BaseSession):
         # Logged-in user token
         token: Optional[str] = None,
         user_env: Optional[Dict[str, str]] = None,
-        # Last message at the root of the chat
-        root_message: Optional["Message"] = None,
         # Origin of the request
         http_referer: Optional[str] = None,
     ):
@@ -190,7 +184,6 @@ class HTTPSession(BaseSession):
             token=token,
             client_type=client_type,
             user_env=user_env,
-            root_message=root_message,
             http_referer=http_referer,
         )
@@ -233,8 +226,6 @@ class WebsocketSession(BaseSession):
         user: Optional[Union["User", "PersistedUser"]] = None,
         # Logged-in user token
         token: Optional[str] = None,
-        # Last message at the root of the chat
-        root_message: Optional["Message"] = None,
         # Chat profile selected before the session was created
         chat_profile: Optional[str] = None,
         # Languages of the user's browser
@@ -249,7 +240,6 @@ class WebsocketSession(BaseSession):
             token=token,
             user_env=user_env,
             client_type=client_type,
-            root_message=root_message,
             chat_profile=chat_profile,
             http_referer=http_referer,
         )

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

Potentially problematic release.

chainlit 1.1.305py3-none-any.whl → 1.1.400py3-none-any.whl