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

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

chainlit/__init__.py +3 -1
chainlit/cli/__init__.py +53 -6
chainlit/config.py +4 -0
chainlit/context.py +9 -0
chainlit/copilot/dist/index.js +180 -180
chainlit/data/__init__.py +6 -3
chainlit/data/sql_alchemy.py +3 -3
chainlit/element.py +33 -9
chainlit/emitter.py +4 -4
chainlit/frontend/dist/assets/{DailyMotion-1a2b7d60.js → DailyMotion-578b63e6.js} +1 -1
chainlit/frontend/dist/assets/{Facebook-8422f48c.js → Facebook-b825e5bb.js} +1 -1
chainlit/frontend/dist/assets/{FilePlayer-a0a41349.js → FilePlayer-bcba3b4e.js} +1 -1
chainlit/frontend/dist/assets/{Kaltura-aa7990f2.js → Kaltura-fc1c9497.js} +1 -1
chainlit/frontend/dist/assets/{Mixcloud-1647b3e4.js → Mixcloud-4cfb2724.js} +1 -1
chainlit/frontend/dist/assets/{Mux-7e57be81.js → Mux-aa92055c.js} +1 -1
chainlit/frontend/dist/assets/{Preview-cb89b2c6.js → Preview-9f55905a.js} +1 -1
chainlit/frontend/dist/assets/{SoundCloud-c0d86d55.js → SoundCloud-f991fe03.js} +1 -1
chainlit/frontend/dist/assets/{Streamable-57c43c18.js → Streamable-53128f49.js} +1 -1
chainlit/frontend/dist/assets/{Twitch-bed3f21d.js → Twitch-fce8b9f5.js} +1 -1
chainlit/frontend/dist/assets/{Vidyard-4dd76e44.js → Vidyard-e35c6102.js} +1 -1
chainlit/frontend/dist/assets/{Vimeo-93bb5ae2.js → Vimeo-fff35f8e.js} +1 -1
chainlit/frontend/dist/assets/{Wistia-97199246.js → Wistia-ec07dc64.js} +1 -1
chainlit/frontend/dist/assets/{YouTube-fe1a7afe.js → YouTube-ad068e2a.js} +1 -1
chainlit/frontend/dist/assets/index-d40d41cc.js +727 -0
chainlit/frontend/dist/assets/{react-plotly-b416b8f9.js → react-plotly-b2c6442b.js} +1 -1
chainlit/frontend/dist/index.html +1 -2
chainlit/message.py +13 -8
chainlit/oauth_providers.py +96 -4
chainlit/server.py +182 -57
chainlit/slack/app.py +2 -2
chainlit/socket.py +24 -21
chainlit/step.py +12 -3
chainlit/teams/__init__.py +6 -0
chainlit/teams/app.py +332 -0
chainlit/translations/en-US.json +1 -1
chainlit/types.py +7 -17
chainlit/user.py +9 -1
chainlit/utils.py +43 -0
{chainlit-1.1.300rc4.dist-info → chainlit-1.1.301.dist-info}/METADATA +2 -2
chainlit-1.1.301.dist-info/RECORD +79 -0
chainlit/cli/utils.py +0 -24
chainlit/frontend/dist/assets/index-919bea8f.js +0 -727
chainlit-1.1.300rc4.dist-info/RECORD +0 -78
{chainlit-1.1.300rc4.dist-info → chainlit-1.1.301.dist-info}/WHEEL +0 -0
{chainlit-1.1.300rc4.dist-info → chainlit-1.1.301.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,7 +21,7 @@
     <script>
       const global = globalThis;
     </script>
-    <script type="module" crossorigin src="/assets/index-919bea8f.js"></script>
+    <script type="module" crossorigin src="/assets/index-d40d41cc.js"></script>
     <link rel="stylesheet" href="/assets/index-aaf974a9.css">
   </head>
   <body>

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,
         }
@@ -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
@@ -304,8 +311,6 @@ 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 assistant 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.
     """
     def __init__(

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
@@ -203,9 +204,93 @@ class AzureADOAuthProvider(OAuthProvider):
                 )
                 photo_data = await photo_response.aread()
                 base64_image = base64.b64encode(photo_data)
-                azure_user[
-                    "image"
-                ] = f"data:{photo_response.headers['Content-Type']};base64,{base64_image.decode('utf-8')}"
+                azure_user["image"] = (
+                    f"data:{photo_response.headers['Content-Type']};base64,{base64_image.decode('utf-8')}"
+                )
+            except Exception as e:
+                # Ignore errors getting the photo
+                pass
+            user = User(
+                identifier=azure_user["userPrincipalName"],
+                metadata={"image": azure_user.get("image"), "provider": "azure-ad"},
+            )
+            return (azure_user, user)
+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(
+                "https://graph.microsoft.com/v1.0/me",
+                headers={"Authorization": f"Bearer {token}"},
+            )
+            response.raise_for_status()
+            azure_user = response.json()
+            try:
+                photo_response = await client.get(
+                    "https://graph.microsoft.com/v1.0/me/photos/48x48/$value",
+                    headers={"Authorization": f"Bearer {token}"},
+                )
+                photo_data = await photo_response.aread()
+                base64_image = base64.b64encode(photo_data)
+                azure_user["image"] = (
+                    f"data:{photo_response.headers['Content-Type']};base64,{base64_image.decode('utf-8')}"
+                )
             except Exception as e:
                 # Ignore errors getting the photo
                 pass
@@ -474,9 +559,14 @@ class AWSCognitoOAuthProvider(OAuthProvider):
             )
             return (cognito_user, user)
 class GitlabOAuthProvider(OAuthProvider):
     id = "gitlab"
-    env = ["OAUTH_GITLAB_CLIENT_ID", "OAUTH_GITLAB_CLIENT_SECRET", "OAUTH_GITLAB_DOMAIN"]
+    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")
@@ -530,10 +620,12 @@ class GitlabOAuthProvider(OAuthProvider):
             )
             return (gitlab_user, user)
 providers = [
     GithubOAuthProvider(),
     GoogleOAuthProvider(),
     AzureADOAuthProvider(),
+    AzureADHybridOAuthProvider(),
     OktaOAuthProvider(),
     Auth0OAuthProvider(),
     DescopeOAuthProvider(),

chainlit 1.1.300rc4__py3-none-any.whl → 1.1.301__py3-none-any.whl

Potentially problematic release.

chainlit 1.1.300rc4py3-none-any.whl → 1.1.301py3-none-any.whl