PyPI - chainlit - Versions diffs - 2.0.0__py3-none-any.whl → 2.0.dev0__py3-none-any.whl - Mend

chainlit 2.0.0py3-none-any.whl → 2.0.dev0py3-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 (98) hide show

chainlit/__init__.py +57 -56
chainlit/action.py +10 -12
chainlit/{auth/__init__.py → auth.py} +34 -26
chainlit/cache.py +6 -4
chainlit/callbacks.py +7 -52
chainlit/chat_context.py +2 -2
chainlit/chat_settings.py +1 -3
chainlit/cli/__init__.py +2 -15
chainlit/config.py +70 -41
chainlit/context.py +9 -8
chainlit/copilot/dist/index.js +874 -8533
chainlit/data/__init__.py +8 -96
chainlit/data/acl.py +2 -3
chainlit/data/base.py +15 -1
chainlit/data/dynamodb.py +4 -7
chainlit/data/literalai.py +6 -4
chainlit/data/sql_alchemy.py +9 -10
chainlit/data/{storage_clients/azure.py → storage_clients.py} +33 -2
chainlit/discord/__init__.py +4 -4
chainlit/discord/app.py +1 -2
chainlit/element.py +9 -41
chainlit/emitter.py +21 -17
chainlit/frontend/dist/assets/DailyMotion-b4b7af47.js +1 -0
chainlit/frontend/dist/assets/Facebook-572972a0.js +1 -0
chainlit/frontend/dist/assets/FilePlayer-85c69ca8.js +1 -0
chainlit/frontend/dist/assets/Kaltura-dfc24672.js +1 -0
chainlit/frontend/dist/assets/Mixcloud-705011f4.js +1 -0
chainlit/frontend/dist/assets/Mux-4201a9e6.js +1 -0
chainlit/frontend/dist/assets/Preview-23ba40a6.js +1 -0
chainlit/frontend/dist/assets/SoundCloud-1a582d51.js +1 -0
chainlit/frontend/dist/assets/Streamable-5017c4ba.js +1 -0
chainlit/frontend/dist/assets/Twitch-bb2de2fa.js +1 -0
chainlit/frontend/dist/assets/Vidyard-54e269b1.js +1 -0
chainlit/frontend/dist/assets/Vimeo-d92c37dd.js +1 -0
chainlit/frontend/dist/assets/Wistia-25a1363b.js +1 -0
chainlit/frontend/dist/assets/YouTube-616e8cb7.js +1 -0
chainlit/frontend/dist/assets/index-aaf974a9.css +1 -0
chainlit/frontend/dist/assets/index-f5df2072.js +1027 -0
chainlit/frontend/dist/assets/{react-plotly-BpxUS-ab.js → react-plotly-f0315f86.js} +94 -94
chainlit/frontend/dist/index.html +3 -2
chainlit/haystack/callbacks.py +4 -5
chainlit/input_widget.py +4 -6
chainlit/langchain/callbacks.py +47 -56
chainlit/langflow/__init__.py +0 -1
chainlit/llama_index/callbacks.py +7 -7
chainlit/message.py +10 -8
chainlit/mistralai/__init__.py +2 -3
chainlit/oauth_providers.py +12 -113
chainlit/openai/__init__.py +7 -6
chainlit/secret.py +1 -1
chainlit/server.py +181 -491
chainlit/session.py +5 -7
chainlit/slack/__init__.py +3 -3
chainlit/slack/app.py +2 -3
chainlit/socket.py +103 -78
chainlit/step.py +29 -21
chainlit/sync.py +1 -2
chainlit/teams/__init__.py +3 -3
chainlit/teams/app.py +0 -1
chainlit/types.py +4 -20
chainlit/user.py +1 -2
chainlit/utils.py +2 -3
chainlit/version.py +2 -3
{chainlit-2.0.0.dist-info → chainlit-2.0.dev0.dist-info}/METADATA +39 -27
chainlit-2.0.dev0.dist-info/RECORD +96 -0
chainlit/auth/cookie.py +0 -123
chainlit/auth/jwt.py +0 -37
chainlit/data/chainlit_data_layer.py +0 -584
chainlit/data/storage_clients/__init__.py +0 -0
chainlit/data/storage_clients/azure_blob.py +0 -80
chainlit/data/storage_clients/base.py +0 -22
chainlit/data/storage_clients/gcs.py +0 -78
chainlit/data/storage_clients/s3.py +0 -49
chainlit/frontend/dist/assets/DailyMotion-DgRzV5GZ.js +0 -1
chainlit/frontend/dist/assets/Dataframe-DVgwSMU2.js +0 -22
chainlit/frontend/dist/assets/Facebook-C0vx6HWv.js +0 -1
chainlit/frontend/dist/assets/FilePlayer-CdhzeHPP.js +0 -1
chainlit/frontend/dist/assets/Kaltura-5iVmeUct.js +0 -1
chainlit/frontend/dist/assets/Mixcloud-C2zi77Ex.js +0 -1
chainlit/frontend/dist/assets/Mux-Vkebogdf.js +0 -1
chainlit/frontend/dist/assets/Preview-DwY_sEIl.js +0 -1
chainlit/frontend/dist/assets/SoundCloud-CREBXAWo.js +0 -1
chainlit/frontend/dist/assets/Streamable-B5Lu25uy.js +0 -1
chainlit/frontend/dist/assets/Twitch-y9iKCcM1.js +0 -1
chainlit/frontend/dist/assets/Vidyard-ClYvcuEu.js +0 -1
chainlit/frontend/dist/assets/Vimeo-D6HvM2jt.js +0 -1
chainlit/frontend/dist/assets/Wistia-Cu4zZ2Ci.js +0 -1
chainlit/frontend/dist/assets/YouTube-D10tR6CJ.js +0 -1
chainlit/frontend/dist/assets/index-CI4qFOt5.js +0 -8665
chainlit/frontend/dist/assets/index-CrrqM0nZ.css +0 -1
chainlit/translations/nl-NL.json +0 -229
chainlit-2.0.0.dist-info/RECORD +0 -106
/chainlit/copilot/dist/assets/{logo_dark-IkGJ_IwC.svg → logo_dark-2a3cf740.svg} +0 -0
/chainlit/copilot/dist/assets/{logo_light-Bb_IPh6r.svg → logo_light-b078e7bc.svg} +0 -0
/chainlit/frontend/dist/assets/{logo_dark-IkGJ_IwC.svg → logo_dark-2a3cf740.svg} +0 -0
/chainlit/frontend/dist/assets/{logo_light-Bb_IPh6r.svg → logo_light-b078e7bc.svg} +0 -0
{chainlit-2.0.0.dist-info → chainlit-2.0.dev0.dist-info}/WHEEL +0 -0
{chainlit-2.0.0.dist-info → chainlit-2.0.dev0.dist-info}/entry_points.txt +0 -0

chainlit/frontend/dist/index.html CHANGED Viewed

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

chainlit/haystack/callbacks.py CHANGED Viewed

@@ -1,13 +1,12 @@
 import re
 from typing import Any, Generic, List, Optional, TypeVar
-from haystack.agents import Agent, Tool
-from haystack.agents.agent_step import AgentStep
-from literalai.helper import utc_now
 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
 T = TypeVar("T")
@@ -132,7 +131,7 @@ class HaystackAgentCallbackHandler:
         tool_result: str,
         tool_name: Optional[str] = None,
         tool_input: Optional[str] = None,
-        **kwargs: Any,
+        **kwargs: Any
     ) -> None:
         # Tool finished, send step with tool_result
         tool_step = self.stack.pop()

chainlit/input_widget.py CHANGED Viewed

@@ -2,10 +2,8 @@ from abc import abstractmethod
 from collections import defaultdict
 from typing import Any, Dict, List, Optional
-from pydantic import Field
-from pydantic.dataclasses import dataclass
 from chainlit.types import InputWidgetType
+from pydantic.dataclasses import Field, dataclass
 @dataclass
@@ -77,7 +75,7 @@ class Select(InputWidget):
     initial: Optional[str] = None
     initial_index: Optional[int] = None
     initial_value: Optional[str] = None
-    values: List[str] = Field(default_factory=list)
+    values: List[str] = Field(default_factory=lambda: [])
     items: Dict[str, str] = Field(default_factory=lambda: defaultdict(dict))
     def __post_init__(
@@ -169,8 +167,8 @@ class Tags(InputWidget):
     """Useful to create an input for an array of strings."""
     type: InputWidgetType = "tags"
-    initial: List[str] = Field(default_factory=list)
-    values: List[str] = Field(default_factory=list)
+    initial: List[str] = Field(default_factory=lambda: [])
+    values: List[str] = Field(default_factory=lambda: [])
     def to_dict(self) -> Dict[str, Any]:
         return {

chainlit/langchain/callbacks.py CHANGED Viewed

@@ -3,19 +3,17 @@ import time
 from typing import Any, Dict, List, Optional, Sequence, Tuple, TypedDict, Union
 from uuid import UUID
-import pydantic
+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
 from langchain_core.outputs import ChatGenerationChunk, GenerationChunk
-from langchain_core.tracers.base import AsyncBaseTracer
 from literalai import ChatGeneration, CompletionGeneration, GenerationMessage
 from literalai.helper import utc_now
 from literalai.observability.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", ":"]
@@ -124,14 +122,6 @@ class GenerationHelper:
                 key: self.ensure_values_serializable(value)
                 for key, value in data.items()
             }
-        elif isinstance(data, pydantic.BaseModel):
-            # Fallback to support pydantic v1
-            # https://docs.pydantic.dev/latest/migration/#changes-to-pydanticbasemodel
-            if pydantic.VERSION.startswith("1"):
-                return data.dict()
-            # pydantic v2
-            return data.model_dump()  # pyright: ignore reportAttributeAccessIssue
         elif isinstance(data, list):
             return [self.ensure_values_serializable(item) for item in data]
         elif isinstance(data, (str, int, float, bool, type(None))):
@@ -259,7 +249,7 @@ DEFAULT_TO_IGNORE = [
 DEFAULT_TO_KEEP = ["retriever", "llm", "agent", "chain", "tool"]
-class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
+class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
     steps: Dict[str, Step]
     parent_id_map: Dict[str, str]
     ignored_runs: set
@@ -278,7 +268,7 @@ class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
         to_keep: Optional[List[str]] = None,
         **kwargs: Any,
     ) -> None:
-        AsyncBaseTracer.__init__(self, **kwargs)
+        BaseTracer.__init__(self, **kwargs)
         GenerationHelper.__init__(self)
         FinalStreamHelper.__init__(
             self,
@@ -306,7 +296,7 @@ class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
         else:
             self.to_keep = to_keep
-    async def on_chat_model_start(
+    def on_chat_model_start(
         self,
         serialized: Dict[str, Any],
         messages: List[List[BaseMessage]],
@@ -315,9 +305,8 @@ class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
         parent_run_id: Optional["UUID"] = None,
         tags: Optional[List[str]] = None,
         metadata: Optional[Dict[str, Any]] = None,
-        name: Optional[str] = None,
         **kwargs: Any,
-    ) -> Run:
+    ) -> Any:
         lc_messages = messages[0]
         self.chat_generations[str(run_id)] = {
             "input_messages": lc_messages,
@@ -326,48 +315,46 @@ class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
             "tt_first_token": None,
         }
-        return await super().on_chat_model_start(
+        return super().on_chat_model_start(
             serialized,
             messages,
             run_id=run_id,
             parent_run_id=parent_run_id,
             tags=tags,
             metadata=metadata,
-            name=name,
             **kwargs,
         )
-    async def on_llm_start(
+    def on_llm_start(
         self,
         serialized: Dict[str, Any],
         prompts: List[str],
         *,
         run_id: "UUID",
-        parent_run_id: Optional[UUID] = None,
         tags: Optional[List[str]] = None,
+        parent_run_id: Optional["UUID"] = None,
         metadata: Optional[Dict[str, Any]] = None,
+        name: Optional[str] = None,
         **kwargs: Any,
-    ) -> None:
-        await super().on_llm_start(
+    ) -> Run:
+        self.completion_generations[str(run_id)] = {
+            "prompt": prompts[0],
+            "start": time.time(),
+            "token_count": 0,
+            "tt_first_token": None,
+        }
+        return super().on_llm_start(
             serialized,
             prompts,
             run_id=run_id,
             parent_run_id=parent_run_id,
             tags=tags,
             metadata=metadata,
+            name=name,
             **kwargs,
         )
-        self.completion_generations[str(run_id)] = {
-            "prompt": prompts[0],
-            "start": time.time(),
-            "token_count": 0,
-            "tt_first_token": None,
-        }
-        return None
-    async def on_llm_new_token(
+    def on_llm_new_token(
         self,
         token: str,
         *,
@@ -375,14 +362,7 @@ class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
         run_id: "UUID",
         parent_run_id: Optional["UUID"] = None,
         **kwargs: Any,
-    ) -> None:
-        await super().on_llm_new_token(
-            token=token,
-            chunk=chunk,
-            run_id=run_id,
-            parent_run_id=parent_run_id,
-            **kwargs,
-        )
+    ) -> Run:
         if isinstance(chunk, ChatGenerationChunk):
             start = self.chat_generations[str(run_id)]
         else:
@@ -397,13 +377,24 @@ class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
             if self.answer_reached:
                 if not self.final_stream:
                     self.final_stream = Message(content="")
-                    await self.final_stream.send()
-                await self.final_stream.stream_token(token)
+                    self._run_sync(self.final_stream.send())
+                self._run_sync(self.final_stream.stream_token(token))
                 self.has_streamed_final_answer = True
             else:
                 self.answer_reached = self._check_if_answer_reached()
-    async def _persist_run(self, run: Run) -> None:
+        return super().on_llm_new_token(
+            token,
+            chunk=chunk,
+            run_id=run_id,
+            parent_run_id=parent_run_id,
+        )
+    def _run_sync(self, co):  # TODO: WHAT TO DO WITH THIS?
+        context_var.set(self.context)
+        self.context.loop.create_task(co)
+    def _persist_run(self, run: Run) -> None:
         pass
     def _get_run_parent_id(self, run: Run):
@@ -454,8 +445,8 @@ class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
                 self.ignored_runs.add(str(run.id))
             return ignore, parent_id
-    async def _start_trace(self, run: Run) -> None:
-        await super()._start_trace(run)
+    def _start_trace(self, run: Run) -> None:
+        super()._start_trace(run)
         context_var.set(self.context)
         ignore, parent_id = self._should_ignore_run(run)
@@ -468,7 +459,7 @@ class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
         if ignore:
             return
-        step_type: TrueStepType = "undefined"
+        step_type: "TrueStepType" = "undefined"
         if run.run_type == "agent":
             step_type = "run"
         elif run.run_type == "chain":
@@ -498,9 +489,9 @@ class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
         self.steps[str(run.id)] = step
-        await step.send()
+        self._run_sync(step.send())
-    async def _on_run_update(self, run: Run) -> None:
+    def _on_run_update(self, run: Run) -> None:
         """Process a run upon update."""
         context_var.set(self.context)
@@ -585,10 +576,10 @@ class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
             if current_step:
                 current_step.end = utc_now()
-                await current_step.update()
+                self._run_sync(current_step.update())
             if self.final_stream and self.has_streamed_final_answer:
-                await self.final_stream.update()
+                self._run_sync(self.final_stream.update())
             return
@@ -608,16 +599,16 @@ class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
                 else output
             )
             current_step.end = utc_now()
-            await current_step.update()
+            self._run_sync(current_step.update())
-    async def _on_error(self, error: BaseException, *, run_id: UUID, **kwargs: Any):
+    def _on_error(self, error: BaseException, *, run_id: UUID, **kwargs: Any):
         context_var.set(self.context)
         if current_step := self.steps.get(str(run_id), None):
             current_step.is_error = True
             current_step.output = str(error)
             current_step.end = utc_now()
-            await current_step.update()
+            self._run_sync(current_step.update())
     on_llm_error = _on_error
     on_chain_error = _on_error

chainlit/langflow/__init__.py CHANGED Viewed

@@ -8,7 +8,6 @@ if not check_module_version("langflow", "0.1.4"):
 from typing import Dict, Optional, Union
 import httpx
 from chainlit.telemetry import trace_event

chainlit/llama_index/callbacks.py CHANGED Viewed

@@ -1,5 +1,8 @@
 from typing import Any, Dict, List, Optional
+from chainlit.context import context_var
+from chainlit.element import Text
+from chainlit.step import Step, StepType
 from literalai import ChatGeneration, CompletionGeneration, GenerationMessage
 from literalai.helper import utc_now
 from llama_index.core.callbacks import TokenCountingHandler
@@ -7,10 +10,6 @@ from llama_index.core.callbacks.schema import CBEventType, EventPayload
 from llama_index.core.llms import ChatMessage, ChatResponse, CompletionResponse
 from llama_index.core.tools.types import ToolMetadata
-from chainlit.context import context_var
-from chainlit.element import Text
-from chainlit.step import Step, StepType
 DEFAULT_IGNORE = [
     CBEventType.CHUNKING,
     CBEventType.SYNTHESIZE,
@@ -144,15 +143,16 @@ class LlamaIndexCallbackHandler(TokenCountingHandler):
             context_var.get().loop.create_task(step.update())
         elif event_type == CBEventType.LLM:
-            formatted_messages = payload.get(EventPayload.MESSAGES)  # type: Optional[List[ChatMessage]]
+            formatted_messages = payload.get(
+                EventPayload.MESSAGES
+            )  # type: Optional[List[ChatMessage]]
             formatted_prompt = payload.get(EventPayload.PROMPT)
             response = payload.get(EventPayload.RESPONSE)
             if formatted_messages:
                 messages = [
                     GenerationMessage(
-                        role=m.role.value,  # type: ignore
-                        content=m.content or "",
+                        role=m.role.value, content=m.content or ""  # type: ignore
                     )
                     for m in formatted_messages
                 ]

chainlit/message.py CHANGED Viewed

@@ -5,9 +5,6 @@ import uuid
 from abc import ABC
 from typing import Dict, List, Optional, Union, cast
-from literalai.helper import utc_now
-from literalai.observability.step import MessageStepType
 from chainlit.action import Action
 from chainlit.chat_context import chat_context
 from chainlit.config import config
@@ -25,6 +22,8 @@ from chainlit.types import (
     AskSpec,
     FileDict,
 )
+from literalai.helper import utc_now
+from literalai.observability.step import MessageStepType
 class MessageBase(ABC):
@@ -43,6 +42,7 @@ class MessageBase(ABC):
     metadata: Optional[Dict] = None
     tags: Optional[List[str]] = None
     wait_for_answer = False
+    indent: Optional[int] = None
     def __post_init__(self) -> None:
         trace_event(f"init {self.__class__.__name__}")
@@ -59,7 +59,7 @@ class MessageBase(ABC):
     @classmethod
     def from_dict(self, _dict: StepDict):
         type = _dict.get("type", "assistant_message")
-        return Message(
+        message = Message(
             id=_dict["id"],
             parent_id=_dict.get("parentId"),
             created_at=_dict["createdAt"],
@@ -67,9 +67,10 @@ class MessageBase(ABC):
             author=_dict.get("name", config.ui.name),
             type=type,  # type: ignore
             language=_dict.get("language"),
-            metadata=_dict.get("metadata", {}),
         )
+        return message
     def to_dict(self) -> StepDict:
         _dict: StepDict = {
             "id": self.id,
@@ -85,6 +86,7 @@ class MessageBase(ABC):
             "streaming": self.streaming,
             "isError": self.is_error,
             "waitForAnswer": self.wait_for_answer,
+            "indent": self.indent,
             "metadata": self.metadata or {},
             "tags": self.tags,
         }
@@ -112,7 +114,7 @@ class MessageBase(ABC):
             except Exception as e:
                 if self.fail_on_persist_error:
                     raise e
-                logger.error(f"Failed to persist message update: {e!s}")
+                logger.error(f"Failed to persist message update: {str(e)}")
         await context.emitter.update_step(step_dict)
@@ -132,7 +134,7 @@ class MessageBase(ABC):
             except Exception as e:
                 if self.fail_on_persist_error:
                     raise e
-                logger.error(f"Failed to persist message deletion: {e!s}")
+                logger.error(f"Failed to persist message deletion: {str(e)}")
         await context.emitter.delete_step(step_dict)
@@ -148,7 +150,7 @@ class MessageBase(ABC):
             except Exception as e:
                 if self.fail_on_persist_error:
                     raise e
-                logger.error(f"Failed to persist message creation: {e!s}")
+                logger.error(f"Failed to persist message creation: {str(e)}")
         return step_dict

chainlit/mistralai/__init__.py CHANGED Viewed

@@ -1,11 +1,10 @@
 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 literalai import ChatGeneration, CompletionGeneration
+from literalai.helper import timestamp_utc
 def instrument_mistralai():

chainlit 2.0.0__py3-none-any.whl → 2.0.dev0__py3-none-any.whl

Potentially problematic release.

chainlit 2.0.0py3-none-any.whl → 2.0.dev0py3-none-any.whl