PyPI - chainlit - Versions diffs - 1.0.401__py3-none-any.whl → 1.0.501__py3-none-any.whl - Mend

chainlit 1.0.401py3-none-any.whl → 1.0.501py3-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 (28) hide show

chainlit/cli/__init__.py +9 -0
chainlit/config.py +62 -7
chainlit/copilot/dist/index.js +531 -676
chainlit/data/__init__.py +111 -66
chainlit/frontend/dist/assets/index-e306c2e5.js +698 -0
chainlit/frontend/dist/assets/react-plotly-cc656f1c.js +3484 -0
chainlit/frontend/dist/index.html +1 -1
chainlit/langchain/callbacks.py +1 -1
chainlit/llama_index/callbacks.py +35 -6
chainlit/markdown.py +12 -2
chainlit/message.py +10 -1
chainlit/openai/__init__.py +3 -4
chainlit/playground/config.py +0 -4
chainlit/playground/providers/__init__.py +0 -2
chainlit/playground/providers/openai.py +5 -27
chainlit/server.py +58 -6
chainlit/socket.py +1 -1
chainlit/step.py +12 -3
chainlit/translations/en-US.json +77 -1
chainlit/translations.py +60 -0
chainlit/types.py +12 -8
{chainlit-1.0.401.dist-info → chainlit-1.0.501.dist-info}/METADATA +4 -4
{chainlit-1.0.401.dist-info → chainlit-1.0.501.dist-info}/RECORD +25 -25
chainlit/frontend/dist/assets/index-9711593e.js +0 -723
chainlit/frontend/dist/assets/react-plotly-d8762cc2.js +0 -3602
chainlit/translations/pt-BR.json +0 -155
{chainlit-1.0.401.dist-info → chainlit-1.0.501.dist-info}/WHEEL +0 -0
{chainlit-1.0.401.dist-info → chainlit-1.0.501.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-9711593e.js"></script>
+    <script type="module" crossorigin src="/assets/index-e306c2e5.js"></script>
     <link rel="stylesheet" href="/assets/index-d088547c.css">
   </head>
   <body>

chainlit/langchain/callbacks.py CHANGED Viewed

@@ -533,7 +533,7 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
                     break
                 current_step.language = "json"
-                current_step.output = json.dumps(message_completion)
+                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

@@ -14,7 +14,6 @@ DEFAULT_IGNORE = [
     CBEventType.SYNTHESIZE,
     CBEventType.EMBEDDING,
     CBEventType.NODE_PARSING,
-    CBEventType.QUERY,
     CBEventType.TREE,
 ]
@@ -71,9 +70,12 @@ 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"
+        elif event_type == CBEventType.QUERY:
+            step_type = "retrieval"
         elif event_type == CBEventType.LLM:
             step_type = "llm"
         else:
@@ -84,7 +86,7 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
             type=step_type,
             parent_id=self._get_parent_id(parent_id),
             id=event_id,
-            disable_feedback=False,
+            disable_feedback=True,
         )
         self.steps[event_id] = step
         step.start = utc_now()
@@ -102,6 +104,7 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
         """Run when an event ends."""
         step = self.steps.get(event_id, None)
         if payload is None or step is None:
             return
@@ -109,10 +112,26 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
         step.end = utc_now()
-        if event_type == CBEventType.RETRIEVE:
+        if event_type == CBEventType.QUERY:
+            response = payload.get(EventPayload.RESPONSE)
+            source_nodes = getattr(response, "source_nodes", None)
+            if source_nodes:
+                source_refs = ", ".join(
+                    [f"Source {idx}" for idx, _ in enumerate(source_nodes)])
+                step.elements = [
+                    Text(
+                        name=f"Source {idx}",
+                        content=source.text or "Empty node",
+                    )
+                    for idx, source in enumerate(source_nodes)
+                ]
+                step.output = f"Retrieved the following sources: {source_refs}"
+                self.context.loop.create_task(step.update())
+        elif event_type == CBEventType.RETRIEVE:
             sources = payload.get(EventPayload.NODES)
             if sources:
-                source_refs = "\, ".join(
+                source_refs = ", ".join(
                     [f"Source {idx}" for idx, _ in enumerate(sources)]
                 )
                 step.elements = [
@@ -125,7 +144,7 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
                 step.output = f"Retrieved the following sources: {source_refs}"
             self.context.loop.create_task(step.update())
-        if event_type == CBEventType.LLM:
+        elif event_type == CBEventType.LLM:
             formatted_messages = payload.get(
                 EventPayload.MESSAGES
             )  # type: Optional[List[ChatMessage]]
@@ -152,10 +171,13 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
             step.output = content
             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(
+                    model=model,
                     messages=messages,
                     message_completion=GenerationMessage(
                         role=msg.role.value,  # type: ignore
@@ -165,6 +187,7 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
                 )
             elif formatted_prompt:
                 step.generation = CompletionGeneration(
+                    model=model,
                     prompt=formatted_prompt,
                     completion=content,
                     token_count=token_count,
@@ -172,6 +195,11 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
             self.context.loop.create_task(step.update())
+        else:
+            step.output = payload.get
+            self.context.loop.create_task(step.update())
+            return
         self.steps.pop(event_id, None)
     def _noop(self, *args, **kwargs):
@@ -179,3 +207,4 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
     start_trace = _noop
     end_trace = _noop

chainlit/markdown.py CHANGED Viewed

@@ -30,9 +30,19 @@ def init_markdown(root: str):
             logger.info(f"Created default chainlit markdown file at {chainlit_md_file}")
-def get_markdown_str(root: str):
+def get_markdown_str(root: str, language: str):
     """Get the chainlit.md file as a string."""
-    chainlit_md_path = os.path.join(root, "chainlit.md")
+    translated_chainlit_md_path = os.path.join(root, f"chainlit_{language}.md")
+    default_chainlit_md_path = os.path.join(root, "chainlit.md")
+    if os.path.exists(translated_chainlit_md_path):
+        chainlit_md_path = translated_chainlit_md_path
+    else:
+        chainlit_md_path = default_chainlit_md_path
+        logger.warning(
+            f"Translated markdown file for {language} not found. Defaulting to chainlit.md."
+        )
     if os.path.exists(chainlit_md_path):
         with open(chainlit_md_path, "r", encoding="utf-8") as f:
             chainlit_md = f.read()

chainlit/message.py CHANGED Viewed

@@ -39,6 +39,8 @@ class MessageBase(ABC):
     persisted = False
     is_error = False
     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
@@ -83,6 +85,8 @@ class MessageBase(ABC):
             "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,
         }
         return _dict
@@ -209,6 +213,8 @@ class Message(MessageBase):
         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,
         created_at: Union[str, None] = None,
     ):
@@ -234,6 +240,9 @@ class Message(MessageBase):
         if created_at:
             self.created_at = created_at
+        self.metadata = metadata
+        self.tags = tags
         self.author = author
         self.type = type
         self.actions = actions if actions is not None else []
@@ -542,7 +551,7 @@ class AskActionMessage(AskMessageBase):
         if res is None:
             self.content = "Timed out: no action was taken"
         else:
-            self.content = f'**Selected action:** {res["label"]}'
+            self.content = f'**Selected:** {res["label"]}'
         self.wait_for_answer = False

chainlit/openai/__init__.py CHANGED Viewed

@@ -1,8 +1,7 @@
 from typing import Union
+import asyncio
 from chainlit.context import get_context
 from chainlit.step import Step
-from chainlit.sync import run_sync
 from chainlit.utils import check_module_version
 from literalai import ChatGeneration, CompletionGeneration
 from literalai.helper import timestamp_utc
@@ -57,6 +56,6 @@ def instrument_openai():
     def on_new_generation_sync(
         generation: Union["ChatGeneration", "CompletionGeneration"], timing
     ):
-        run_sync(on_new_generation(generation, timing))
+        asyncio.create_task(on_new_generation(generation, timing))
     instrument_openai(None, on_new_generation_sync)

chainlit/playground/config.py CHANGED Viewed

@@ -4,9 +4,7 @@ from chainlit.playground.provider import BaseProvider
 from chainlit.playground.providers import (
     Anthropic,
     AzureChatOpenAI,
-    AzureOpenAI,
     ChatOpenAI,
-    OpenAI,
     ChatVertexAI,
     GenerationVertexAI,
     Gemini,
@@ -14,9 +12,7 @@ from chainlit.playground.providers import (
 providers = {
     AzureChatOpenAI.id: AzureChatOpenAI,
-    AzureOpenAI.id: AzureOpenAI,
     ChatOpenAI.id: ChatOpenAI,
-    OpenAI.id: OpenAI,
     Anthropic.id: Anthropic,
     ChatVertexAI.id: ChatVertexAI,
     GenerationVertexAI.id: GenerationVertexAI,

chainlit/playground/providers/__init__.py CHANGED Viewed

@@ -2,9 +2,7 @@ from .anthropic import Anthropic
 from .huggingface import HFFlanT5
 from .openai import (
     AzureChatOpenAI,
-    AzureOpenAI,
     ChatOpenAI,
-    OpenAI,
 )
 from .vertexai import (
     ChatVertexAI,

chainlit/playground/providers/openai.py CHANGED Viewed

@@ -168,8 +168,8 @@ class ChatOpenAIProvider(BaseProvider):
             async def create_event_stream():
                 message = response.choices[0].message
-                if function_call := message.function_call:
-                    yield stringify_function_call(function_call)
+                if tool_calls := message.tool_calls:
+                    yield json.dumps([tc.model_dump() for tc in tool_calls], indent=4, ensure_ascii=False)
                 else:
                     yield message.content or ""
@@ -335,8 +335,8 @@ class AzureChatOpenAIProvider(BaseProvider):
             async def create_event_stream():
                 message = response.choices[0].message
-                if function_call := message.function_call:
-                    yield stringify_function_call(function_call)
+                if tool_calls := message.tool_calls:
+                    yield json.dumps([tc.model_dump() for tc in tool_calls], indent=4, ensure_ascii=False)
                 else:
                     yield message.content or ""
@@ -374,21 +374,7 @@ ChatOpenAI = ChatOpenAIProvider(
     is_chat=True,
 )
-OpenAI = OpenAIProvider(
-    id="openai",
-    name="OpenAI",
-    env_vars=openai_env_vars,
-    inputs=[
-        Select(
-            id="model",
-            label="Model",
-            values=["text-davinci-003", "text-davinci-002"],
-            initial_value="text-davinci-003",
-        ),
-        *openai_common_inputs,
-    ],
-    is_chat=False,
-)
 AzureChatOpenAI = AzureChatOpenAIProvider(
@@ -398,11 +384,3 @@ AzureChatOpenAI = AzureChatOpenAIProvider(
     inputs=openai_common_inputs,
     is_chat=True,
 )
-AzureOpenAI = AzureOpenAIProvider(
-    id="azure",
-    name="AzureOpenAI",
-    env_vars=azure_openai_env_vars,
-    inputs=openai_common_inputs,
-    is_chat=False,
-)

chainlit/server.py CHANGED Viewed

@@ -36,6 +36,7 @@ from chainlit.markdown import get_markdown_str
 from chainlit.playground.config import get_llm_providers
 from chainlit.telemetry import trace_event
 from chainlit.types import (
+    DeleteFeedbackRequest,
     DeleteThreadRequest,
     GenerationRequest,
     GetThreadsRequest,
@@ -53,6 +54,7 @@ from fastapi import (
     UploadFile,
     status,
 )
+from fastapi.middleware.gzip import GZipMiddleware
 from fastapi.responses import FileResponse, HTMLResponse, JSONResponse, RedirectResponse
 from fastapi.security import OAuth2PasswordRequestForm
 from fastapi.staticfiles import StaticFiles
@@ -139,7 +141,12 @@ async def lifespan(app: FastAPI):
 def get_build_dir(local_target: str, packaged_target: str):
     local_build_dir = os.path.join(PACKAGE_ROOT, local_target, "dist")
     packaged_build_dir = os.path.join(BACKEND_ROOT, packaged_target, "dist")
-    if os.path.exists(local_build_dir):
+    if config.ui.custom_build and os.path.exists(
+        os.path.join(APP_ROOT, config.ui.custom_build, packaged_target, "dist")
+    ):
+        return os.path.join(APP_ROOT, config.ui.custom_build, packaged_target, "dist")
+    elif os.path.exists(local_build_dir):
         return local_build_dir
     elif os.path.exists(packaged_build_dir):
         return packaged_build_dir
@@ -181,10 +188,13 @@ app.add_middleware(
     allow_headers=["*"],
 )
+app.add_middleware(GZipMiddleware)
 socket = SocketManager(
     app,
     cors_allowed_origins=[],
     async_mode="asgi",
+    socketio_path="/ws/socket.io",
 )
@@ -498,6 +508,22 @@ async def get_providers(
     return JSONResponse(content={"providers": providers})
+@app.get("/project/translations")
+async def project_translations(
+    language: str = Query(default="en-US", description="Language code"),
+):
+    """Return project translations."""
+    # Load translation based on the provided language
+    translation = config.load_translation(language)
+    return JSONResponse(
+        content={
+            "translation": translation,
+        }
+    )
 @app.get("/project/settings")
 async def project_settings(
     current_user: Annotated[Union[User, PersistedUser], Depends(get_current_user)],
@@ -505,8 +531,8 @@ async def project_settings(
 ):
     """Return project settings. This is called by the UI before the establishing the websocket connection."""
-    # Load translation based on the provided language
-    translation = config.load_translation(language)
+    # Load the markdown file based on the provided language
+    markdown = get_markdown_str(config.root, language)
     profiles = []
     if config.code.set_chat_profiles:
@@ -520,9 +546,8 @@ async def project_settings(
             "userEnv": config.project.user_env,
             "dataPersistence": get_data_layer() is not None,
             "threadResumable": bool(config.code.on_chat_resume),
-            "markdown": get_markdown_str(config.root),
+            "markdown": markdown,
             "chatProfiles": profiles,
-            "translation": translation,
         }
     )
@@ -546,6 +571,25 @@ async def update_feedback(
     return JSONResponse(content={"success": True, "feedbackId": feedback_id})
+@app.delete("/feedback")
+async def delete_feedback(
+    request: Request,
+    payload: DeleteFeedbackRequest,
+    current_user: Annotated[Union[User, PersistedUser], Depends(get_current_user)],
+):
+    """Delete a feedback."""
+    data_layer = get_data_layer()
+    if not data_layer:
+        raise HTTPException(status_code=400, detail="Data persistence is not enabled")
+    feedback_id = payload.feedbackId
+    await data_layer.delete_feedback(feedback_id)
+    return JSONResponse(content={"success": True})
 @app.post("/project/threads")
 async def get_user_threads(
     request: Request,
@@ -560,7 +604,10 @@ async def get_user_threads(
     if not data_layer:
         raise HTTPException(status_code=400, detail="Data persistence is not enabled")
-    payload.filter.userIdentifier = current_user.identifier
+    if not isinstance(current_user, PersistedUser):
+        raise HTTPException(status_code=400, detail="User not persisted")
+    payload.filter.userId = current_user.id
     res = await data_layer.list_threads(payload.pagination, payload.filter)
     return JSONResponse(content=res.to_dict())
@@ -737,6 +784,11 @@ async def get_logo(theme: Optional[Theme] = Query(Theme.light)):
     return FileResponse(logo_path, media_type=media_type)
+@app.head("/")
+def status_check():
+    return {"message": "Site is operational"}
 def register_wildcard_route_handler():
     @app.get("/{path:path}")
     async def serve(request: Request, path: str):

chainlit/socket.py CHANGED Viewed

@@ -42,7 +42,7 @@ async def resume_thread(session: WebsocketSession):
     if not thread:
         return
-    author = thread.get("user").get("identifier") if thread["user"] else None
+    author = thread.get("userIdentifier")
     user_is_author = author == session.user.identifier
     if user_is_author:

chainlit/step.py CHANGED Viewed

@@ -29,6 +29,7 @@ class StepDict(TypedDict, total=False):
     waitForAnswer: Optional[bool]
     isError: Optional[bool]
     metadata: Dict
+    tags: Optional[List[str]]
     input: str
     output: str
     createdAt: Optional[str]
@@ -47,6 +48,7 @@ def step(
     name: Optional[str] = "",
     type: TrueStepType = "undefined",
     id: Optional[str] = None,
+    tags: Optional[List[str]] = None,
     disable_feedback: bool = True,
     root: bool = False,
     language: Optional[str] = None,
@@ -71,6 +73,7 @@ def step(
                     id=id,
                     disable_feedback=disable_feedback,
                     root=root,
+                    tags=tags,
                     language=language,
                     show_input=show_input,
                 ) as step:
@@ -97,6 +100,7 @@ def step(
                     id=id,
                     disable_feedback=disable_feedback,
                     root=root,
+                    tags=tags,
                     language=language,
                     show_input=show_input,
                 ) as step:
@@ -137,6 +141,7 @@ class Step:
     is_error: Optional[bool]
     metadata: Dict
+    tags: Optional[List[str]]
     thread_id: str
     created_at: Union[str, None]
     start: Union[str, None]
@@ -153,6 +158,8 @@ class Step:
         id: Optional[str] = None,
         parent_id: Optional[str] = None,
         elements: Optional[List[Element]] = None,
+        metadata: Optional[Dict] = None,
+        tags: Optional[List[str]] = None,
         disable_feedback: bool = True,
         root: bool = False,
         language: Optional[str] = None,
@@ -167,7 +174,8 @@ class Step:
         self.type = type
         self.id = id or str(uuid.uuid4())
         self.disable_feedback = disable_feedback
-        self.metadata = {}
+        self.metadata = metadata or {}
+        self.tags = tags
         self.is_error = False
         self.show_input = show_input
         self.parent_id = parent_id
@@ -194,13 +202,13 @@ class Step:
                 if set_language:
                     self.language = "json"
             except TypeError:
-                processed_content = str(content)
+                processed_content = str(content).replace("\\n", "\n")
                 if set_language:
                     self.language = "text"
         elif isinstance(content, str):
             processed_content = content
         else:
-            processed_content = str(content)
+            processed_content = str(content).replace("\\n", "\n")
             if set_language:
                 self.language = "text"
         return processed_content
@@ -231,6 +239,7 @@ class Step:
             "disableFeedback": self.disable_feedback,
             "streaming": self.streaming,
             "metadata": self.metadata,
+            "tags": self.tags,
             "input": self.input,
             "isError": self.is_error,
             "output": self.output,

chainlit/translations/en-US.json CHANGED Viewed

@@ -34,15 +34,87 @@
         "confirm": "Confirm"
       },
       "settingsModal": {
+        "settings": "Settings",
         "expandMessages": "Expand Messages",
         "hideChainOfThought": "Hide Chain of Thought",
         "darkMode": "Dark Mode"
+      },
+      "detailsButton": {
+        "using": "Using",
+        "running": "Running",
+        "took_one": "Took {{count}} step",
+        "took_other": "Took {{count}} steps"
+      },
+      "auth": {
+        "authLogin": {
+          "title": "Login to access the app.",
+          "form": {
+            "email": "Email address",
+            "password": "Password",
+            "noAccount": "Don't have an account?",
+            "alreadyHaveAccount": "Already have an account?",
+            "signup": "Sign Up",
+            "signin": "Sign In",
+            "or": "OR",
+            "continue": "Continue",
+            "forgotPassword": "Forgot password?",
+            "passwordMustContain": "Your password must contain:",
+            "emailRequired": "email is a required field",
+            "passwordRequired": "password is a required field"
+          },
+          "error": {
+            "default": "Unable to sign in.",
+            "signin": "Try signing in with a different account.",
+            "oauthsignin": "Try signing in with a different account.",
+            "redirect_uri_mismatch": "The redirect URI is not matching the oauth app configuration.",
+            "oauthcallbackerror": "Try signing in with a different account.",
+            "oauthcreateaccount": "Try signing in with a different account.",
+            "emailcreateaccount": "Try signing in with a different account.",
+            "callback": "Try signing in with a different account.",
+            "oauthaccountnotlinked": "To confirm your identity, sign in with the same account you used originally.",
+            "emailsignin": "The e-mail could not be sent.",
+            "emailverify": "Please verify your email, a new email has been sent.",
+            "credentialssignin": "Sign in failed. Check the details you provided are correct.",
+            "sessionrequired": "Please sign in to access this page."
+          }
+        },
+        "authVerifyEmail": {
+          "almostThere": "You're almost there! We've sent an email to ",
+          "verifyEmailLink": "Please click on the link in that email to complete your signup.",
+          "didNotReceive": "Can't find the email?",
+          "resendEmail": "Resend email",
+          "goBack": "Go Back",
+          "emailSent": "Email sent successfully.",
+          "verifyEmail": "Verify your email address"
+        },
+        "providerButton": {
+          "continue": "Continue with {{provider}}",
+          "signup": "Sign up with {{provider}}"
+        },
+        "authResetPassword": {
+          "newPasswordRequired": "New password is a required field",
+          "passwordsMustMatch": "Passwords must match",
+          "confirmPasswordRequired": "Confirm password is a required field",
+          "newPassword": "New password",
+          "confirmPassword": "Confirm password",
+          "resetPassword": "Reset Password"
+        },
+        "authForgotPassword": {
+          "email": "Email address",
+          "emailRequired": "email is a required field",
+          "emailSent": "Please check the email address {{email}} for instructions to reset your password.",
+          "enterEmail": "Enter your email address and we will send you instructions to reset your password.",
+          "resendEmail": "Resend email",
+          "continue": "Continue",
+          "goBack": "Go Back"
+        }
       }
     },
     "organisms": {
       "chat": {
         "history": {
           "index": {
+            "showHistory": "Show history",
             "lastInputs": "Last Inputs",
             "noInputs": "Such empty...",
             "loading": "Loading..."
@@ -115,7 +187,11 @@
             "pastChats": "Past Chats"
           },
           "ThreadList": {
-            "empty": "Empty..."
+            "empty": "Empty...",
+            "today": "Today",
+            "yesterday": "Yesterday",
+            "previous7days": "Previous 7 days",
+            "previous30days": "Previous 30 days"
           },
           "TriggerButton": {
             "closeSidebar": "Close sidebar",

chainlit 1.0.401__py3-none-any.whl → 1.0.501__py3-none-any.whl

Potentially problematic release.

chainlit 1.0.401py3-none-any.whl → 1.0.501py3-none-any.whl