PyPI - chainlit - Versions diffs - 1.1.202__py3-none-any.whl → 1.1.300__py3-none-any.whl - Mend

chainlit 1.1.202py3-none-any.whl → 1.1.300py3-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 (60) hide show

chainlit/__init__.py +22 -4
chainlit/cli/__init__.py +53 -6
chainlit/config.py +25 -18
chainlit/context.py +9 -0
chainlit/copilot/dist/index.js +443 -410
chainlit/data/__init__.py +19 -5
chainlit/data/dynamodb.py +586 -0
chainlit/data/sql_alchemy.py +47 -28
chainlit/discord/app.py +4 -2
chainlit/element.py +36 -20
chainlit/emitter.py +8 -7
chainlit/frontend/dist/assets/{DailyMotion-53376209.js → DailyMotion-578b63e6.js} +1 -1
chainlit/frontend/dist/assets/{Facebook-aee41f5b.js → Facebook-b825e5bb.js} +1 -1
chainlit/frontend/dist/assets/{FilePlayer-b2cdb30f.js → FilePlayer-bcba3b4e.js} +1 -1
chainlit/frontend/dist/assets/{Kaltura-51db0377.js → Kaltura-fc1c9497.js} +1 -1
chainlit/frontend/dist/assets/{Mixcloud-cb900886.js → Mixcloud-4cfb2724.js} +1 -1
chainlit/frontend/dist/assets/{Mux-79ac59e6.js → Mux-aa92055c.js} +1 -1
chainlit/frontend/dist/assets/{Preview-cfe7584c.js → Preview-9f55905a.js} +1 -1
chainlit/frontend/dist/assets/{SoundCloud-a985707c.js → SoundCloud-f991fe03.js} +1 -1
chainlit/frontend/dist/assets/{Streamable-3d89aab5.js → Streamable-53128f49.js} +1 -1
chainlit/frontend/dist/assets/{Twitch-bf016588.js → Twitch-fce8b9f5.js} +1 -1
chainlit/frontend/dist/assets/{Vidyard-1891ecd7.js → Vidyard-e35c6102.js} +1 -1
chainlit/frontend/dist/assets/{Vimeo-0645662c.js → Vimeo-fff35f8e.js} +1 -1
chainlit/frontend/dist/assets/{Wistia-3b449fe2.js → Wistia-ec07dc64.js} +1 -1
chainlit/frontend/dist/assets/{YouTube-5ea2381e.js → YouTube-ad068e2a.js} +1 -1
chainlit/frontend/dist/assets/index-aaf974a9.css +1 -0
chainlit/frontend/dist/assets/index-d40d41cc.js +727 -0
chainlit/frontend/dist/assets/{react-plotly-2ff19c9f.js → react-plotly-b2c6442b.js} +1 -1
chainlit/frontend/dist/index.html +2 -3
chainlit/langchain/callbacks.py +4 -2
chainlit/llama_index/callbacks.py +2 -2
chainlit/message.py +30 -25
chainlit/oauth_providers.py +118 -0
chainlit/server.py +208 -83
chainlit/slack/app.py +2 -3
chainlit/socket.py +27 -23
chainlit/step.py +44 -30
chainlit/teams/__init__.py +6 -0
chainlit/teams/app.py +332 -0
chainlit/translations/en-US.json +2 -4
chainlit/types.py +17 -17
chainlit/user.py +9 -1
chainlit/utils.py +47 -3
{chainlit-1.1.202.dist-info → chainlit-1.1.300.dist-info}/METADATA +22 -14
chainlit-1.1.300.dist-info/RECORD +79 -0
chainlit/cli/utils.py +0 -24
chainlit/frontend/dist/assets/index-a0c5a67e.js +0 -698
chainlit/frontend/dist/assets/index-d088547c.css +0 -1
chainlit/playground/__init__.py +0 -2
chainlit/playground/config.py +0 -36
chainlit/playground/provider.py +0 -108
chainlit/playground/providers/__init__.py +0 -11
chainlit/playground/providers/anthropic.py +0 -118
chainlit/playground/providers/huggingface.py +0 -75
chainlit/playground/providers/langchain.py +0 -89
chainlit/playground/providers/openai.py +0 -386
chainlit/playground/providers/vertexai.py +0 -171
chainlit-1.1.202.dist-info/RECORD +0 -86
{chainlit-1.1.202.dist-info → chainlit-1.1.300.dist-info}/WHEEL +0 -0
{chainlit-1.1.202.dist-info → chainlit-1.1.300.dist-info}/entry_points.txt +0 -0

chainlit/frontend/dist/index.html CHANGED Viewed

@@ -4,7 +4,6 @@
     <meta charset="UTF-8" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <!-- TAG INJECTION PLACEHOLDER -->
-    <link rel="icon" href="/favicon" />
     <link rel="preconnect" href="https://fonts.googleapis.com" />
     <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin />
     <!-- FONT START -->
@@ -22,8 +21,8 @@
     <script>
       const global = globalThis;
     </script>
-    <script type="module" crossorigin src="/assets/index-a0c5a67e.js"></script>
-    <link rel="stylesheet" href="/assets/index-d088547c.css">
+    <script type="module" crossorigin src="/assets/index-d40d41cc.js"></script>
+    <link rel="stylesheet" href="/assets/index-aaf974a9.css">
   </head>
   <body>
     <div id="root"></div>

chainlit/langchain/callbacks.py CHANGED Viewed

@@ -456,7 +456,7 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
         elif run.run_type == "llm":
             step_type = "llm"
         elif run.run_type == "retriever":
-            step_type = "retrieval"
+            step_type = "tool"
         elif run.run_type == "tool":
             step_type = "tool"
         elif run.run_type == "embedding":
@@ -533,7 +533,9 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
                     break
                 current_step.language = "json"
-                current_step.output = json.dumps(message_completion, indent=4, ensure_ascii=False)
+                current_step.output = json.dumps(
+                    message_completion, indent=4, ensure_ascii=False
+                )
             else:
                 completion_start = self.completion_generations[str(run.id)]
                 completion = generation.get("text", "")

chainlit/llama_index/callbacks.py CHANGED Viewed

@@ -73,9 +73,9 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
         step_type: StepType = "undefined"
         if event_type == CBEventType.RETRIEVE:
-            step_type = "retrieval"
+            step_type = "tool"
         elif event_type == CBEventType.QUERY:
-            step_type = "retrieval"
+            step_type = "tool"
         elif event_type == CBEventType.LLM:
             step_type = "llm"
         else:

chainlit/message.py CHANGED Viewed

@@ -7,7 +7,7 @@ from typing import Dict, List, Optional, Union, cast
 from chainlit.action import Action
 from chainlit.config import config
-from chainlit.context import context
+from chainlit.context import context, local_steps
 from chainlit.data import get_data_layer
 from chainlit.element import ElementBased
 from chainlit.logger import logger
@@ -21,7 +21,6 @@ from chainlit.types import (
     AskSpec,
     FileDict,
 )
-from literalai import BaseGeneration
 from literalai.helper import utc_now
 from literalai.step import MessageStepType
@@ -38,17 +37,22 @@ class MessageBase(ABC):
     fail_on_persist_error: bool = False
     persisted = False
     is_error = False
+    parent_id: Optional[str] = None
     language: Optional[str] = None
     metadata: Optional[Dict] = None
     tags: Optional[List[str]] = None
     wait_for_answer = False
     indent: Optional[int] = None
-    generation: Optional[BaseGeneration] = None
     def __post_init__(self) -> None:
         trace_event(f"init {self.__class__.__name__}")
         self.thread_id = context.session.thread_id
+        previous_steps = local_steps.get() or []
+        parent_step = previous_steps[-1] if previous_steps else None
+        if parent_step:
+            self.parent_id = parent_step.id
         if not getattr(self, "id", None):
             self.id = str(uuid.uuid4())
@@ -57,6 +61,7 @@ class MessageBase(ABC):
         type = _dict.get("type", "assistant_message")
         message = Message(
             id=_dict["id"],
+            parent_id=_dict.get("parentId"),
             created_at=_dict["createdAt"],
             content=_dict["output"],
             author=_dict.get("name", config.ui.name),
@@ -71,6 +76,7 @@ class MessageBase(ABC):
         _dict: StepDict = {
             "id": self.id,
             "threadId": self.thread_id,
+            "parentId": self.parent_id,
             "createdAt": self.created_at,
             "start": self.created_at,
             "end": self.created_at,
@@ -84,7 +90,6 @@ class MessageBase(ABC):
             "isError": self.is_error,
             "waitForAnswer": self.wait_for_answer,
             "indent": self.indent,
-            "generation": self.generation.to_dict() if self.generation else None,
             "metadata": self.metadata or {},
             "tags": self.tags,
         }
@@ -173,21 +178,21 @@ class MessageBase(ABC):
         Sends a token to the UI. This is useful for streaming messages.
         Once all tokens have been streamed, call .send() to end the stream and persist the message if persistence is enabled.
         """
-        if not self.streaming:
-            self.streaming = True
-            step_dict = self.to_dict()
-            await context.emitter.stream_start(step_dict)
         if is_sequence:
             self.content = token
         else:
             self.content += token
         assert self.id
-        await context.emitter.send_token(
-            id=self.id, token=token, is_sequence=is_sequence
-        )
+        if not self.streaming:
+            self.streaming = True
+            step_dict = self.to_dict()
+            await context.emitter.stream_start(step_dict)
+        else:
+            await context.emitter.send_token(
+                id=self.id, token=token, is_sequence=is_sequence
+            )
 class Message(MessageBase):
@@ -196,7 +201,7 @@ class Message(MessageBase):
     Args:
         content (Union[str, Dict]): The content of the message.
-        author (str, optional): The author of the message, this will be used in the UI. Defaults to the chatbot name (see config).
+        author (str, optional): The author of the message, this will be used in the UI. Defaults to the assistant name (see config).
         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.
@@ -212,15 +217,14 @@ class Message(MessageBase):
         elements: Optional[List[ElementBased]] = None,
         disable_feedback: bool = False,
         type: MessageStepType = "assistant_message",
-        generation: Optional[BaseGeneration] = None,
         metadata: Optional[Dict] = None,
         tags: Optional[List[str]] = None,
         id: Optional[str] = None,
+        parent_id: Optional[str] = None,
         created_at: Union[str, None] = None,
     ):
         time.sleep(0.001)
         self.language = language
-        self.generation = generation
         if isinstance(content, dict):
             try:
                 self.content = json.dumps(content, indent=4, ensure_ascii=False)
@@ -237,6 +241,9 @@ class Message(MessageBase):
         if id:
             self.id = str(id)
+        if parent_id:
+            self.parent_id = str(parent_id)
         if created_at:
             self.created_at = created_at
@@ -303,9 +310,7 @@ class ErrorMessage(MessageBase):
     Args:
         content (str): Text displayed above the upload button.
-        author (str, optional): The author of the message, this will be used in the UI. Defaults to the chatbot name (see config).
-        parent_id (str, optional): If provided, the message will be nested inside the parent in the UI.
-        indent (int, optional): If positive, the message will be nested in the UI.
+        author (str, optional): The author of the message, this will be used in the UI. Defaults to the assistant name (see config).
     """
     def __init__(
@@ -316,7 +321,7 @@ class ErrorMessage(MessageBase):
     ):
         self.content = content
         self.author = author
-        self.type = "system_message"
+        self.type = "assistant_message"
         self.is_error = True
         self.fail_on_persist_error = fail_on_persist_error
@@ -346,7 +351,7 @@ 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 chatbot name (see config).
+        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.
@@ -357,7 +362,7 @@ class AskUserMessage(AskMessageBase):
         content: str,
         author: str = config.ui.name,
         type: MessageStepType = "assistant_message",
-        disable_feedback: bool = False,
+        disable_feedback: bool = True,
         timeout: int = 60,
         raise_on_timeout: bool = False,
     ):
@@ -411,7 +416,7 @@ class AskFileMessage(AskMessageBase):
         accept (Union[List[str], Dict[str, List[str]]]): List of mime type to accept like ["text/csv", "application/pdf"] or a dict like {"text/plain": [".txt", ".py"]}.
         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 chatbot name (see config).
+        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.
@@ -425,7 +430,7 @@ class AskFileMessage(AskMessageBase):
         max_files=1,
         author=config.ui.name,
         type: MessageStepType = "assistant_message",
-        disable_feedback: bool = False,
+        disable_feedback: bool = True,
         timeout=90,
         raise_on_timeout=False,
     ):
@@ -501,7 +506,7 @@ class AskActionMessage(AskMessageBase):
         content: str,
         actions: List[Action],
         author=config.ui.name,
-        disable_feedback=False,
+        disable_feedback=True,
         timeout=90,
         raise_on_timeout=False,
     ):

chainlit/oauth_providers.py CHANGED Viewed

@@ -4,6 +4,7 @@ import urllib.parse
 from typing import Dict, List, Optional, Tuple
 import httpx
+from chainlit.secret import random_secret
 from chainlit.user import User
 from fastapi import HTTPException
@@ -186,6 +187,60 @@ class AzureADOAuthProvider(OAuthProvider):
                 )
             return token
+class AzureADHybridOAuthProvider(OAuthProvider):
+    id = "azure-ad-hybrid"
+    env = [
+        "OAUTH_AZURE_AD_HYBRID_CLIENT_ID",
+        "OAUTH_AZURE_AD_HYBRID_CLIENT_SECRET",
+        "OAUTH_AZURE_AD_HYBRID_TENANT_ID",
+    ]
+    authorize_url = (
+        f"https://login.microsoftonline.com/{os.environ.get('OAUTH_AZURE_AD_HYBRID_TENANT_ID', '')}/oauth2/v2.0/authorize"
+        if os.environ.get("OAUTH_AZURE_AD_HYBRID_ENABLE_SINGLE_TENANT")
+        else "https://login.microsoftonline.com/common/oauth2/v2.0/authorize"
+    )
+    token_url = (
+        f"https://login.microsoftonline.com/{os.environ.get('OAUTH_AZURE_AD_HYBRID_TENANT_ID', '')}/oauth2/v2.0/token"
+        if os.environ.get("OAUTH_AZURE_AD_HYBRID_ENABLE_SINGLE_TENANT")
+        else "https://login.microsoftonline.com/common/oauth2/v2.0/token"
+    )
+    def __init__(self):
+        self.client_id = os.environ.get("OAUTH_AZURE_AD_HYBRID_CLIENT_ID")
+        self.client_secret = os.environ.get("OAUTH_AZURE_AD_HYBRID_CLIENT_SECRET")
+        nonce = random_secret(16)
+        self.authorize_params = {
+            "tenant": os.environ.get("OAUTH_AZURE_AD_HYBRID_TENANT_ID"),
+            "response_type": "code id_token",
+            "scope": "https://graph.microsoft.com/User.Read https://graph.microsoft.com/openid",
+            "response_mode": "form_post",
+            "nonce": nonce,
+        }
+    async def get_token(self, code: str, url: str):
+        payload = {
+            "client_id": self.client_id,
+            "client_secret": self.client_secret,
+            "code": code,
+            "grant_type": "authorization_code",
+            "redirect_uri": url,
+        }
+        async with httpx.AsyncClient() as client:
+            response = await client.post(
+                self.token_url,
+                data=payload,
+            )
+            response.raise_for_status()
+            json = response.json()
+            token = json["access_token"]
+            if not token:
+                raise HTTPException(
+                    status_code=400, detail="Failed to get the access token"
+                )
+            return token
     async def get_user_info(self, token: str):
         async with httpx.AsyncClient() as client:
             response = await client.get(
@@ -475,14 +530,77 @@ class AWSCognitoOAuthProvider(OAuthProvider):
             return (cognito_user, user)
+class GitlabOAuthProvider(OAuthProvider):
+    id = "gitlab"
+    env = [
+        "OAUTH_GITLAB_CLIENT_ID",
+        "OAUTH_GITLAB_CLIENT_SECRET",
+        "OAUTH_GITLAB_DOMAIN",
+    ]
+    def __init__(self):
+        self.client_id = os.environ.get("OAUTH_GITLAB_CLIENT_ID")
+        self.client_secret = os.environ.get("OAUTH_GITLAB_CLIENT_SECRET")
+        # Ensure that the domain does not have a trailing slash
+        self.domain = f"https://{os.environ.get('OAUTH_GITLAB_DOMAIN', '').rstrip('/')}"
+        self.authorize_url = f"{self.domain}/oauth/authorize"
+        self.authorize_params = {
+            "scope": "openid profile email",
+            "response_type": "code",
+        }
+    async def get_token(self, code: str, url: str):
+        payload = {
+            "client_id": self.client_id,
+            "client_secret": self.client_secret,
+            "code": code,
+            "grant_type": "authorization_code",
+            "redirect_uri": url,
+        }
+        async with httpx.AsyncClient() as client:
+            response = await client.post(
+                f"{self.domain}/oauth/token",
+                data=payload,
+            )
+            response.raise_for_status()
+            json_content = response.json()
+            token = json_content.get("access_token")
+            if not token:
+                raise HTTPException(
+                    status_code=400, detail="Failed to get the access token"
+                )
+            return token
+    async def get_user_info(self, token: str):
+        async with httpx.AsyncClient() as client:
+            response = await client.get(
+                f"{self.domain}/oauth/userinfo",
+                headers={"Authorization": f"Bearer {token}"},
+            )
+            response.raise_for_status()
+            gitlab_user = response.json()
+            user = User(
+                identifier=gitlab_user.get("email"),
+                metadata={
+                    "image": gitlab_user.get("picture", ""),
+                    "provider": "gitlab",
+                },
+            )
+            return (gitlab_user, user)
 providers = [
     GithubOAuthProvider(),
     GoogleOAuthProvider(),
     AzureADOAuthProvider(),
+    AzureADHybridOAuthProvider(),
     OktaOAuthProvider(),
     Auth0OAuthProvider(),
     DescopeOAuthProvider(),
     AWSCognitoOAuthProvider(),
+    GitlabOAuthProvider(),
 ]

chainlit 1.1.202__py3-none-any.whl → 1.1.300__py3-none-any.whl

Potentially problematic release.

chainlit 1.1.202py3-none-any.whl → 1.1.300py3-none-any.whl