PyPI - chainlit - Versions diffs - 1.2.0rc0__py3-none-any.whl → 1.3.0__py3-none-any.whl - Mend

chainlit 1.2.0rc0py3-none-any.whl → 1.3.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 (72) hide show

chainlit/__init__.py +0 -10
chainlit/auth.py +9 -7
chainlit/callbacks.py +0 -32
chainlit/cli/__init__.py +2 -0
chainlit/config.py +7 -11
chainlit/context.py +4 -6
chainlit/copilot/dist/index.js +298 -307
chainlit/data/__init__.py +11 -6
chainlit/data/dynamodb.py +7 -3
chainlit/data/literalai.py +168 -59
chainlit/data/sql_alchemy.py +89 -15
chainlit/element.py +2 -1
chainlit/emitter.py +0 -7
chainlit/frontend/dist/assets/DailyMotion-CwoOhIL8.js +1 -0
chainlit/frontend/dist/assets/Facebook-BhnGXlzq.js +1 -0
chainlit/frontend/dist/assets/FilePlayer-CPSVT6fz.js +1 -0
chainlit/frontend/dist/assets/Kaltura-COYaLzsL.js +1 -0
chainlit/frontend/dist/assets/Mixcloud-JdadNiQ5.js +1 -0
chainlit/frontend/dist/assets/Mux-CBN7RO2u.js +1 -0
chainlit/frontend/dist/assets/Preview-CxAFvvjV.js +1 -0
chainlit/frontend/dist/assets/SoundCloud-JlgmASWm.js +1 -0
chainlit/frontend/dist/assets/Streamable-CUWgr6Zw.js +1 -0
chainlit/frontend/dist/assets/Twitch-BiN1HEDM.js +1 -0
chainlit/frontend/dist/assets/Vidyard-qhPmrhDm.js +1 -0
chainlit/frontend/dist/assets/Vimeo-CrZVSCaT.js +1 -0
chainlit/frontend/dist/assets/Wistia-C891KrBP.js +1 -0
chainlit/frontend/dist/assets/YouTube-DKjw5Hbn.js +1 -0
chainlit/frontend/dist/assets/index-DLRdQOIx.js +723 -0
chainlit/frontend/dist/assets/{react-plotly-55648373.js → react-plotly-Dpmqg5Sy.js} +94 -94
chainlit/frontend/dist/index.html +2 -3
chainlit/haystack/callbacks.py +1 -3
chainlit/input_widget.py +0 -22
chainlit/langchain/callbacks.py +1 -2
chainlit/message.py +1 -1
chainlit/mistralai/__init__.py +0 -1
chainlit/oauth_providers.py +46 -3
chainlit/openai/__init__.py +3 -5
chainlit/server.py +20 -32
chainlit/session.py +5 -36
chainlit/socket.py +2 -51
chainlit/step.py +13 -20
chainlit/translations/en-US.json +0 -6
chainlit/types.py +1 -1
chainlit/user_session.py +0 -4
{chainlit-1.2.0rc0.dist-info → chainlit-1.3.0.dist-info}/METADATA +17 -9
chainlit-1.3.0.dist-info/RECORD +96 -0
chainlit/assistant.py +0 -16
chainlit/assistant_settings.py +0 -35
chainlit/frontend/dist/assets/DailyMotion-aa368b7e.js +0 -1
chainlit/frontend/dist/assets/Facebook-0335db46.js +0 -1
chainlit/frontend/dist/assets/FilePlayer-8d04256c.js +0 -1
chainlit/frontend/dist/assets/Kaltura-67c9dd31.js +0 -1
chainlit/frontend/dist/assets/Mixcloud-6bbaccf5.js +0 -1
chainlit/frontend/dist/assets/Mux-c2bcb757.js +0 -1
chainlit/frontend/dist/assets/Preview-210f3955.js +0 -1
chainlit/frontend/dist/assets/SoundCloud-a0276b84.js +0 -1
chainlit/frontend/dist/assets/Streamable-a007323d.js +0 -1
chainlit/frontend/dist/assets/Twitch-e6a88aa3.js +0 -1
chainlit/frontend/dist/assets/Vidyard-dfb88a35.js +0 -1
chainlit/frontend/dist/assets/Vimeo-3baa13d9.js +0 -1
chainlit/frontend/dist/assets/Wistia-e52f7bef.js +0 -1
chainlit/frontend/dist/assets/YouTube-1715f22b.js +0 -1
chainlit/frontend/dist/assets/index-bfdd8585.js +0 -729
chainlit/translations/fr-FR.json +0 -236
chainlit-1.2.0rc0.dist-info/RECORD +0 -99
/chainlit/copilot/dist/assets/{logo_dark-2a3cf740.svg → logo_dark-IkGJ_IwC.svg} +0 -0
/chainlit/copilot/dist/assets/{logo_light-b078e7bc.svg → logo_light-Bb_IPh6r.svg} +0 -0
/chainlit/frontend/dist/assets/{index-aaf974a9.css → index-CwmincdQ.css} +0 -0
/chainlit/frontend/dist/assets/{logo_dark-2a3cf740.svg → logo_dark-IkGJ_IwC.svg} +0 -0
/chainlit/frontend/dist/assets/{logo_light-b078e7bc.svg → logo_light-Bb_IPh6r.svg} +0 -0
{chainlit-1.2.0rc0.dist-info → chainlit-1.3.0.dist-info}/WHEEL +0 -0
{chainlit-1.2.0rc0.dist-info → chainlit-1.3.0.dist-info}/entry_points.txt +0 -0

chainlit/frontend/dist/index.html CHANGED Viewed

@@ -21,11 +21,10 @@
     <script>
       const global = globalThis;
     </script>
-    <script type="module" crossorigin src="/assets/index-bfdd8585.js"></script>
-    <link rel="stylesheet" href="/assets/index-aaf974a9.css">
+    <script type="module" crossorigin src="/assets/index-DLRdQOIx.js"></script>
+    <link rel="stylesheet" crossorigin href="/assets/index-CwmincdQ.css">
   </head>
   <body>
     <div id="root"></div>
   </body>
 </html>

chainlit/haystack/callbacks.py CHANGED Viewed

@@ -1,15 +1,13 @@
 import re
 from typing import Any, Generic, List, Optional, TypeVar
-from chainlit.context import context
+from chainlit import Message
 from chainlit.step import Step
 from chainlit.sync import run_sync
 from haystack.agents import Agent, Tool
 from haystack.agents.agent_step import AgentStep
 from literalai.helper import utc_now
-from chainlit import Message
 T = TypeVar("T")

chainlit/input_widget.py CHANGED Viewed

@@ -161,28 +161,6 @@ class NumberInput(InputWidget):
             "description": self.description,
         }
-@dataclass
-class FileUploadInput(InputWidget):
-    """Useful to create a file upload input."""
-    type: InputWidgetType = "fileupload"
-    initial: Optional[str] = None
-    placeholder: Optional[str] = None
-    accept: List[str] = Field(default_factory=lambda: [])
-    max_size_mb: Optional[int] = None
-    max_files: Optional[int] = None
-    def to_dict(self) -> Dict[str, Any]:
-        return {
-            "type": self.type,
-            "id": self.id,
-            "label": self.label,
-            "initial": self.initial,
-            "placeholder": self.placeholder,
-            "tooltip": self.tooltip,
-            "description": self.description,
-        }
 @dataclass
 class Tags(InputWidget):

chainlit/langchain/callbacks.py CHANGED Viewed

@@ -9,11 +9,10 @@ from chainlit.step import Step
 from langchain.callbacks.tracers.base import BaseTracer
 from langchain.callbacks.tracers.schemas import Run
 from langchain.schema import BaseMessage
-from langchain.schema.output import ChatGenerationChunk, GenerationChunk
 from langchain_core.outputs import ChatGenerationChunk, GenerationChunk
 from literalai import ChatGeneration, CompletionGeneration, GenerationMessage
 from literalai.helper import utc_now
-from literalai.step import TrueStepType
+from literalai.observability.step import TrueStepType
 DEFAULT_ANSWER_PREFIX_TOKENS = ["Final", "Answer", ":"]

chainlit/message.py CHANGED Viewed

@@ -23,7 +23,7 @@ from chainlit.types import (
     FileDict,
 )
 from literalai.helper import utc_now
-from literalai.step import MessageStepType
+from literalai.observability.step import MessageStepType
 class MessageBase(ABC):

chainlit/mistralai/__init__.py CHANGED Viewed

@@ -3,7 +3,6 @@ from typing import Union
 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

chainlit/oauth_providers.py CHANGED Viewed

@@ -16,6 +16,7 @@ class OAuthProvider:
     client_secret: str
     authorize_url: str
     authorize_params: Dict[str, str]
+    default_prompt: Optional[str] = None
     def is_configured(self):
         return all([os.environ.get(env) for env in self.env])
@@ -26,6 +27,21 @@ class OAuthProvider:
     async def get_user_info(self, token: str) -> Tuple[Dict[str, str], User]:
         raise NotImplementedError()
+    def get_env_prefix(self) -> str:
+        """Return environment prefix, like AZURE_AD."""
+        return self.id.replace("-", "_").upper()
+    def get_prompt(self) -> Optional[str]:
+        """Return OAuth prompt param."""
+        if prompt := os.environ.get(f"OAUTH_{self.get_env_prefix()}_PROMPT"):
+            return prompt
+        if prompt := os.environ.get("OAUTH_PROMPT"):
+            return prompt
+        return self.default_prompt
 class GithubOAuthProvider(OAuthProvider):
     id = "github"
@@ -39,6 +55,9 @@ class GithubOAuthProvider(OAuthProvider):
             "scope": "user:email",
         }
+        if prompt := self.get_prompt():
+            self.authorize_params["prompt"] = prompt
     async def get_token(self, code: str, url: str):
         payload = {
             "client_id": self.client_id,
@@ -97,6 +116,9 @@ class GoogleOAuthProvider(OAuthProvider):
             "access_type": "offline",
         }
+        if prompt := self.get_prompt():
+            self.authorize_params["prompt"] = prompt
     async def get_token(self, code: str, url: str):
         payload = {
             "client_id": self.client_id,
@@ -164,6 +186,9 @@ class AzureADOAuthProvider(OAuthProvider):
             "response_mode": "query",
         }
+        if prompt := self.get_prompt():
+            self.authorize_params["prompt"] = prompt
     async def get_token(self, code: str, url: str):
         payload = {
             "client_id": self.client_id,
@@ -207,7 +232,7 @@ class AzureADOAuthProvider(OAuthProvider):
                 azure_user["image"] = (
                     f"data:{photo_response.headers['Content-Type']};base64,{base64_image.decode('utf-8')}"
                 )
-            except Exception as e:
+            except Exception:
                 # Ignore errors getting the photo
                 pass
@@ -248,6 +273,9 @@ class AzureADHybridOAuthProvider(OAuthProvider):
             "nonce": nonce,
         }
+        if prompt := self.get_prompt():
+            self.authorize_params["prompt"] = prompt
     async def get_token(self, code: str, url: str):
         payload = {
             "client_id": self.client_id,
@@ -291,7 +319,7 @@ class AzureADHybridOAuthProvider(OAuthProvider):
                 azure_user["image"] = (
                     f"data:{photo_response.headers['Content-Type']};base64,{base64_image.decode('utf-8')}"
                 )
-            except Exception as e:
+            except Exception:
                 # Ignore errors getting the photo
                 pass
@@ -327,6 +355,9 @@ class OktaOAuthProvider(OAuthProvider):
             "response_mode": "query",
         }
+        if prompt := self.get_prompt():
+            self.authorize_params["prompt"] = prompt
     def get_authorization_server_path(self):
         if not self.authorization_server_id:
             return "/default"
@@ -398,6 +429,9 @@ class Auth0OAuthProvider(OAuthProvider):
             "audience": f"{self.original_domain}/userinfo",
         }
+        if prompt := self.get_prompt():
+            self.authorize_params["prompt"] = prompt
     async def get_token(self, code: str, url: str):
         payload = {
             "client_id": self.client_id,
@@ -442,7 +476,7 @@ class DescopeOAuthProvider(OAuthProvider):
     id = "descope"
     env = ["OAUTH_DESCOPE_CLIENT_ID", "OAUTH_DESCOPE_CLIENT_SECRET"]
     # Ensure that the domain does not have a trailing slash
-    domain = f"https://api.descope.com/oauth2/v1"
+    domain = "https://api.descope.com/oauth2/v1"
     authorize_url = f"{domain}/authorize"
@@ -455,6 +489,9 @@ class DescopeOAuthProvider(OAuthProvider):
             "audience": f"{self.domain}/userinfo",
         }
+        if prompt := self.get_prompt():
+            self.authorize_params["prompt"] = prompt
     async def get_token(self, code: str, url: str):
         payload = {
             "client_id": self.client_id,
@@ -513,6 +550,9 @@ class AWSCognitoOAuthProvider(OAuthProvider):
             "scope": "openid profile email",
         }
+        if prompt := self.get_prompt():
+            self.authorize_params["prompt"] = prompt
     async def get_token(self, code: str, url: str):
         payload = {
             "client_id": self.client_id,
@@ -581,6 +621,9 @@ class GitlabOAuthProvider(OAuthProvider):
             "response_type": "code",
         }
+        if prompt := self.get_prompt():
+            self.authorize_params["prompt"] = prompt
     async def get_token(self, code: str, url: str):
         payload = {
             "client_id": self.client_id,

chainlit/openai/__init__.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import asyncio
 from typing import Union
-from chainlit.context import get_context
+from chainlit.context import local_steps
 from chainlit.step import Step
 from chainlit.utils import check_module_version
 from literalai import ChatGeneration, CompletionGeneration
@@ -19,11 +19,9 @@ def instrument_openai():
     def on_new_generation(
         generation: Union["ChatGeneration", "CompletionGeneration"], timing
     ):
-        context = get_context()
+        previous_steps = local_steps.get()
-        parent_id = None
-        if context.current_step:
-            parent_id = context.current_step.id
+        parent_id = previous_steps[-1].id if previous_steps else None
         step = Step(
             name=generation.model if generation.model else generation.provider,

chainlit/server.py CHANGED Viewed

@@ -41,6 +41,7 @@ from fastapi import (
     APIRouter,
     Depends,
     FastAPI,
+    File,
     Form,
     HTTPException,
     Query,
@@ -228,6 +229,7 @@ app.mount(
     name="copilot",
 )
 # -------------------------------------------------------------------------------
 #                               SLACK HANDLER
 # -------------------------------------------------------------------------------
@@ -644,7 +646,7 @@ async def oauth_azure_hf_callback(
 _language_pattern = (
-    "^[a-zA-Z]{2,3}(-[a-zA-Z]{2,3})?(-[a-zA-Z]{2,8})?(-x-[a-zA-Z0-9]{1,8})?$"
+    "^[a-zA-Z]{2,3}(-[a-zA-Z0-9]{2,3})?(-[a-zA-Z0-9]{2,8})?(-x-[a-zA-Z0-9]{1,8})?$"
 )
@@ -838,11 +840,9 @@ async def delete_thread(
 @router.post("/project/file")
 async def upload_file(
+    current_user: Annotated[Union[User, PersistedUser], Depends(get_current_user)],
     session_id: str,
     file: UploadFile,
-    current_user: Annotated[
-        Union[None, User, PersistedUser], Depends(get_current_user)
-    ],
 ):
     """Upload a file to the session files directory."""
@@ -867,17 +867,21 @@ async def upload_file(
     content = await file.read()
+    assert file.filename, "No filename for uploaded file"
+    assert file.content_type, "No content type for uploaded file"
     file_response = await session.persist_file(
         name=file.filename, content=content, mime=file.content_type
     )
-    return JSONResponse(file_response)
+    return JSONResponse(content=file_response)
 @router.get("/project/file/{file_id}")
 async def get_file(
     file_id: str,
-    session_id: Optional[str] = None,
+    session_id: str,
+    current_user: Annotated[Union[User, PersistedUser], Depends(get_current_user)],
 ):
     """Get a file from the session files directory."""
@@ -887,10 +891,17 @@ async def get_file(
     if not session:
         raise HTTPException(
-            status_code=404,
-            detail="Session not found",
+            status_code=401,
+            detail="Unauthorized",
         )
+    if current_user:
+        if not session.user or session.user.identifier != current_user.identifier:
+            raise HTTPException(
+                status_code=401,
+                detail="You are not authorized to download files from this session",
+            )
     if file_id in session.files:
         file = session.files[file_id]
         return FileResponse(file["path"], media_type=file["type"])
@@ -960,7 +971,7 @@ async def get_logo(theme: Optional[Theme] = Query(Theme.light)):
 @router.get("/avatars/{avatar_id:str}")
 async def get_avatar(avatar_id: str):
     """Get the avatar for the user based on the avatar_id."""
-    if not re.match(r"^[a-zA-Z0-9_-]+$", avatar_id):
+    if not re.match(r"^[a-zA-Z0-9_ -]+$", avatar_id):
         raise HTTPException(status_code=400, detail="Invalid avatar_id")
     if avatar_id == "default":
@@ -984,29 +995,6 @@ async def get_avatar(avatar_id: str):
     return await get_favicon()
-# post avatar/{avatar_id} (only for authenticated users)
-@router.post("/avatars/{avatar_id}")
-async def upload_avatar(
-    avatar_id: str,
-    file: UploadFile,
-    current_user: Annotated[
-        Union[None, User, PersistedUser], Depends(get_current_user)
-    ],
-):
-    try:
-        avatar_path = os.path.join(APP_ROOT, "public", "avatars", avatar_id)
-        # Ensure the avatars directory exists
-        os.makedirs(os.path.dirname(avatar_path), exist_ok=True)
-        with open(avatar_path, "wb") as f:
-            f.write(await file.read())
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-    return {"id": avatar_id}
 @router.head("/")
 def status_check():
     """Check if the site is operational."""

chainlit/session.py CHANGED Viewed

@@ -3,26 +3,14 @@ import json
 import mimetypes
 import shutil
 import uuid
-from typing import (
-    TYPE_CHECKING,
-    Any,
-    Callable,
-    Deque,
-    Dict,
-    List,
-    Literal,
-    Optional,
-    Union,
-)
+from typing import TYPE_CHECKING, Any, Callable, Deque, Dict, Literal, Optional, Union
 import aiofiles
-from chainlit.assistant import Assistant
 from chainlit.logger import logger
+from chainlit.types import FileReference
 if TYPE_CHECKING:
-    from chainlit.message import Message
-    from chainlit.step import Step
-    from chainlit.types import FileDict, FileReference
+    from chainlit.types import FileDict
     from chainlit.user import PersistedUser, User
 ClientType = Literal["webapp", "copilot", "teams", "slack", "discord"]
@@ -73,12 +61,8 @@ class BaseSession:
         user_env: Optional[Dict[str, str]],
         # Chat profile selected before the session was created
         chat_profile: Optional[str] = None,
-        # Selected assistant
-        selected_assistant: Optional[Assistant] = None,
         # Origin of the request
         http_referer: Optional[str] = None,
-        # assistant settings
-        assistant_settings: Optional[Dict[str, Any]] = None,
     ):
         if thread_id:
             self.thread_id_to_resume = thread_id
@@ -91,13 +75,11 @@ class BaseSession:
         self.chat_profile = chat_profile
         self.http_referer = http_referer
-        self.files = {}  # type: Dict[str, "FileDict"]
+        self.files: Dict[str, FileDict] = {}
         self.id = id
-        self.assistant_settings = assistant_settings
         self.chat_settings: Dict[str, Any] = {}
-        self.selected_assistant = selected_assistant
     @property
     def files_dir(self):
@@ -111,7 +93,7 @@ class BaseSession:
         mime: str,
         path: Optional[str] = None,
         content: Optional[Union[bytes, str]] = None,
-    ) -> "FileReference":
+    ) -> FileReference:
         if not path and not content:
             raise ValueError(
                 "Either path or content must be provided to persist a file"
@@ -160,7 +142,6 @@ 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["selected_assistant"] = self.selected_assistant
         user_session["http_referer"] = self.http_referer
         user_session["client_type"] = self.client_type
         metadata = clean_metadata(user_session)
@@ -184,10 +165,6 @@ class HTTPSession(BaseSession):
         user_env: Optional[Dict[str, str]] = None,
         # Origin of the request
         http_referer: Optional[str] = None,
-        # assistant settings
-        assistant_settings: Optional[Dict[str, Any]] = None,
-        # selected assistant
-        selected_assistant: Optional[Assistant] = None,
     ):
         super().__init__(
             id=id,
@@ -197,8 +174,6 @@ class HTTPSession(BaseSession):
             client_type=client_type,
             user_env=user_env,
             http_referer=http_referer,
-            assistant_settings=assistant_settings,
-            selected_assistant=selected_assistant,
         )
     def delete(self):
@@ -245,14 +220,10 @@ class WebsocketSession(BaseSession):
         token: Optional[str] = None,
         # Chat profile selected before the session was created
         chat_profile: Optional[str] = None,
-        # Selected assistant
-        selected_assistant: Optional[Assistant] = None,
         # Languages of the user's browser
         languages: Optional[str] = None,
         # Origin of the request
         http_referer: Optional[str] = None,
-        # chat settings
-        assistant_settings: Optional[Dict[str, Any]] = None,
     ):
         super().__init__(
             id=id,
@@ -262,9 +233,7 @@ class WebsocketSession(BaseSession):
             user_env=user_env,
             client_type=client_type,
             chat_profile=chat_profile,
-            selected_assistant=selected_assistant,
             http_referer=http_referer,
-            assistant_settings=assistant_settings,
         )
         self.socket_id = socket_id

chainlit/socket.py CHANGED Viewed

@@ -6,7 +6,6 @@ from typing import Any, Dict, Literal
 from urllib.parse import unquote
 from chainlit.action import Action
-from chainlit.assistant import Assistant
 from chainlit.auth import get_current_user, require_login
 from chainlit.chat_context import chat_context
 from chainlit.config import config
@@ -120,7 +119,7 @@ async def connect(sid, environ):
             authorization_header = environ.get("HTTP_AUTHORIZATION")
             token = authorization_header.split(" ")[1] if authorization_header else None
             user = await get_current_user(token=token)
-    except Exception as e:
+    except Exception:
         logger.info("Authentication failed")
         return False
@@ -146,7 +145,7 @@ async def connect(sid, environ):
         unquote(url_encoded_chat_profile) if url_encoded_chat_profile else None
     )
-    ws_session = WebsocketSession(
+    WebsocketSession(
         id=session_id,
         socket_id=sid,
         emit=emit_fn,
@@ -407,51 +406,3 @@ async def change_settings(sid, settings: Dict[str, Any]):
     if config.code.on_settings_update:
         await config.code.on_settings_update(settings)
-@sio.on("on_create_assistant")
-async def on_create_assistant(sid, options):
-    context = init_ws_context(sid)
-    logger.info(f"Received request to create assistant {options['settings_values']}")
-    if not options["settings_values"].get("id"):
-        options["settings_values"]["id"] = str(uuid.uuid4())
-    if not options["settings_values"].get("created_by"):
-        options["settings_values"]["created_by"] = context.session.user.identifier
-    logger.info(
-        f"Added id: {options['settings_values']['id']} and created_by: {options['settings_values']['created_by']} to the assistant options"
-    )
-    if config.code.on_create_assistant:
-        new_assistant = Assistant(
-            input_widgets=options["input_widgets"],
-            settings_values=options["settings_values"],
-        )
-        await config.code.on_create_assistant(context.session.user, new_assistant)
-        return new_assistant.to_dict()
-    logger.info("Assistant creation process completed")
-@sio.on("on_list_assistants")
-async def on_list_assistants(sid):
-    context = init_ws_context(sid)
-    if config.code.on_list_assistants:
-        assistants = await config.code.on_list_assistants(context.session.user)
-        return [assistant.to_dict() for assistant in assistants]
-@sio.on("select_assistant")
-async def select_assistant(sid, selected_assistant: Dict):
-    session = WebsocketSession.require(sid)
-    new_assistant = Assistant(
-        input_widgets=selected_assistant.get("input_widgets", []),
-        settings_values=selected_assistant.get("settings_values", {}),
-    )
-    session.selected_assistant = new_assistant
-    user_identifier = session.user.identifier if session.user else "Anonymous user"
-    logger.info(
-        f"{user_identifier} selected assistant {new_assistant.settings_values.get('name', 'Unknown')} for session with id {session.id}"
-    )
-    return new_assistant.to_dict()

chainlit/step.py CHANGED Viewed

@@ -16,7 +16,7 @@ from chainlit.telemetry import trace_event
 from chainlit.types import FeedbackDict
 from literalai import BaseGeneration
 from literalai.helper import utc_now
-from literalai.step import StepType, TrueStepType
+from literalai.observability.step import StepType, TrueStepType
 def check_add_step_in_cot(step: "Step"):
@@ -30,7 +30,7 @@ def check_add_step_in_cot(step: "Step"):
     return True
-def stub_step(step: "Step"):
+def stub_step(step: "Step") -> "StepDict":
     return {
         "type": step.type,
         "name": step.name,
@@ -65,7 +65,7 @@ class StepDict(TypedDict, total=False):
     feedback: Optional[FeedbackDict]
-def flatten_args_kwargs(func, *args, **kwargs):
+def flatten_args_kwargs(func, args, kwargs):
     signature = inspect.signature(func)
     bound_arguments = signature.bind(*args, **kwargs)
     bound_arguments.apply_defaults()
@@ -189,12 +189,13 @@ class Step:
         tags: Optional[List[str]] = None,
         language: Optional[str] = None,
         show_input: Union[bool, str] = "json",
+        thread_id: Optional[str] = None,
     ):
         trace_event(f"init {self.__class__.__name__} {type}")
         time.sleep(0.001)
         self._input = ""
         self._output = ""
-        self.thread_id = context.session.thread_id
+        self.thread_id = thread_id or context.session.thread_id
         self.name = name or ""
         self.type = type
         self.id = id or str(uuid.uuid4())
@@ -433,7 +434,6 @@ class Step:
         if not self.parent_id:
             if parent_step:
                 self.parent_id = parent_step.id
-        context.active_steps.append(self)
         local_steps.set(previous_steps + [self])
         await self.send()
         return self
@@ -445,13 +445,10 @@ class Step:
             self.output = str(exc_val)
             self.is_error = True
-        if self in context.active_steps:
-            context.active_steps.remove(self)
-        local_active_steps = local_steps.get()
-        if local_active_steps and self in local_active_steps:
-            local_active_steps.remove(self)
-            local_steps.set(local_active_steps)
+        current_steps = local_steps.get()
+        if current_steps and self in current_steps:
+            current_steps.remove(self)
+            local_steps.set(current_steps)
         await self.update()
@@ -464,7 +461,6 @@ class Step:
         if not self.parent_id:
             if parent_step:
                 self.parent_id = parent_step.id
-        context.active_steps.append(self)
         local_steps.set(previous_steps + [self])
         asyncio.create_task(self.send())
@@ -477,12 +473,9 @@ class Step:
             self.output = str(exc_val)
             self.is_error = True
-        if self in context.active_steps:
-            context.active_steps.remove(self)
-        local_active_steps = local_steps.get()
-        if local_active_steps and self in local_active_steps:
-            local_active_steps.remove(self)
-            local_steps.set(local_active_steps)
+        current_steps = local_steps.get()
+        if current_steps and self in current_steps:
+            current_steps.remove(self)
+            local_steps.set(current_steps)
         asyncio.create_task(self.update())

chainlit/translations/en-US.json CHANGED Viewed

@@ -16,9 +16,6 @@
       "newChatButton": {
         "newChat": "New Chat"
       },
-      "newAssistantButton": {
-        "newAssistant": "New Assistant"
-      },
       "tasklist": {
         "TaskList": {
           "title": "🗒️ Task List",
@@ -112,9 +109,6 @@
       }
     },
     "organisms": {
-      "assistantCreationModal": {
-        "title": "Create new assistant"
-      },
       "chat": {
         "history": {
           "index": {

chainlit 1.2.0rc0__py3-none-any.whl → 1.3.0__py3-none-any.whl

Potentially problematic release.

chainlit 1.2.0rc0py3-none-any.whl → 1.3.0py3-none-any.whl