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

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

chainlit/__init__.py +58 -56
chainlit/action.py +12 -10
chainlit/{auth.py → auth/__init__.py} +24 -34
chainlit/auth/cookie.py +123 -0
chainlit/auth/jwt.py +37 -0
chainlit/cache.py +4 -6
chainlit/callbacks.py +65 -11
chainlit/chat_context.py +2 -2
chainlit/chat_settings.py +3 -1
chainlit/cli/__init__.py +15 -2
chainlit/config.py +46 -90
chainlit/context.py +4 -3
chainlit/copilot/dist/index.js +8608 -642
chainlit/data/__init__.py +96 -8
chainlit/data/acl.py +3 -2
chainlit/data/base.py +1 -15
chainlit/data/chainlit_data_layer.py +584 -0
chainlit/data/dynamodb.py +7 -4
chainlit/data/literalai.py +4 -6
chainlit/data/sql_alchemy.py +9 -8
chainlit/data/storage_clients/__init__.py +0 -0
chainlit/data/{storage_clients.py → storage_clients/azure.py} +2 -33
chainlit/data/storage_clients/azure_blob.py +80 -0
chainlit/data/storage_clients/base.py +22 -0
chainlit/data/storage_clients/gcs.py +78 -0
chainlit/data/storage_clients/s3.py +49 -0
chainlit/discord/__init__.py +4 -4
chainlit/discord/app.py +2 -1
chainlit/element.py +41 -9
chainlit/emitter.py +37 -16
chainlit/frontend/dist/assets/{DailyMotion-CwoOhIL8.js → DailyMotion-DgRzV5GZ.js} +1 -1
chainlit/frontend/dist/assets/Dataframe-DVgwSMU2.js +22 -0
chainlit/frontend/dist/assets/{Facebook-BhnGXlzq.js → Facebook-C0vx6HWv.js} +1 -1
chainlit/frontend/dist/assets/{FilePlayer-CPSVT6fz.js → FilePlayer-CdhzeHPP.js} +1 -1
chainlit/frontend/dist/assets/{Kaltura-COYaLzsL.js → Kaltura-5iVmeUct.js} +1 -1
chainlit/frontend/dist/assets/{Mixcloud-JdadNiQ5.js → Mixcloud-C2zi77Ex.js} +1 -1
chainlit/frontend/dist/assets/{Mux-CBN7RO2u.js → Mux-Vkebogdf.js} +1 -1
chainlit/frontend/dist/assets/{Preview-CxAFvvjV.js → Preview-DwY_sEIl.js} +1 -1
chainlit/frontend/dist/assets/{SoundCloud-JlgmASWm.js → SoundCloud-CREBXAWo.js} +1 -1
chainlit/frontend/dist/assets/{Streamable-CUWgr6Zw.js → Streamable-B5Lu25uy.js} +1 -1
chainlit/frontend/dist/assets/{Twitch-BiN1HEDM.js → Twitch-y9iKCcM1.js} +1 -1
chainlit/frontend/dist/assets/{Vidyard-qhPmrhDm.js → Vidyard-ClYvcuEu.js} +1 -1
chainlit/frontend/dist/assets/{Vimeo-CrZVSCaT.js → Vimeo-D6HvM2jt.js} +1 -1
chainlit/frontend/dist/assets/Wistia-Cu4zZ2Ci.js +1 -0
chainlit/frontend/dist/assets/{YouTube-DKjw5Hbn.js → YouTube-D10tR6CJ.js} +1 -1
chainlit/frontend/dist/assets/index-CI4qFOt5.js +8665 -0
chainlit/frontend/dist/assets/index-CrrqM0nZ.css +1 -0
chainlit/frontend/dist/assets/{react-plotly-Dpmqg5Sy.js → react-plotly-BpxUS-ab.js} +1 -1
chainlit/frontend/dist/index.html +2 -2
chainlit/haystack/callbacks.py +5 -4
chainlit/input_widget.py +6 -4
chainlit/langchain/callbacks.py +56 -47
chainlit/langflow/__init__.py +1 -0
chainlit/llama_index/callbacks.py +7 -7
chainlit/message.py +8 -10
chainlit/mistralai/__init__.py +3 -2
chainlit/oauth_providers.py +70 -3
chainlit/openai/__init__.py +3 -2
chainlit/secret.py +1 -1
chainlit/server.py +481 -182
chainlit/session.py +7 -5
chainlit/slack/__init__.py +3 -3
chainlit/slack/app.py +3 -2
chainlit/socket.py +89 -112
chainlit/step.py +12 -12
chainlit/sync.py +2 -1
chainlit/teams/__init__.py +3 -3
chainlit/teams/app.py +1 -0
chainlit/translations/en-US.json +2 -1
chainlit/translations/nl-NL.json +229 -0
chainlit/types.py +24 -8
chainlit/user.py +2 -1
chainlit/utils.py +3 -2
chainlit/version.py +3 -2
{chainlit-1.3.1.dist-info → chainlit-2.0.0.dist-info}/METADATA +17 -37
chainlit-2.0.0.dist-info/RECORD +106 -0
chainlit/frontend/dist/assets/Wistia-C891KrBP.js +0 -1
chainlit/frontend/dist/assets/index-CwmincdQ.css +0 -1
chainlit/frontend/dist/assets/index-DLRdQOIx.js +0 -723
chainlit-1.3.1.dist-info/RECORD +0 -96
{chainlit-1.3.1.dist-info → chainlit-2.0.0.dist-info}/WHEEL +0 -0
{chainlit-1.3.1.dist-info → chainlit-2.0.0.dist-info}/entry_points.txt +0 -0

chainlit/frontend/dist/index.html CHANGED Viewed

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

chainlit/haystack/callbacks.py CHANGED Viewed

@@ -1,13 +1,14 @@
 import re
 from typing import Any, Generic, List, Optional, TypeVar
-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
+from chainlit.step import Step
+from chainlit.sync import run_sync
 T = TypeVar("T")
@@ -131,7 +132,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,8 +2,10 @@ 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
@@ -75,7 +77,7 @@ class Select(InputWidget):
     initial: Optional[str] = None
     initial_index: Optional[int] = None
     initial_value: Optional[str] = None
-    values: List[str] = Field(default_factory=lambda: [])
+    values: List[str] = Field(default_factory=list)
     items: Dict[str, str] = Field(default_factory=lambda: defaultdict(dict))
     def __post_init__(
@@ -167,8 +169,8 @@ class Tags(InputWidget):
     """Useful to create an input for an array of strings."""
     type: InputWidgetType = "tags"
-    initial: List[str] = Field(default_factory=lambda: [])
-    values: List[str] = Field(default_factory=lambda: [])
+    initial: List[str] = Field(default_factory=list)
+    values: List[str] = Field(default_factory=list)
     def to_dict(self) -> Dict[str, Any]:
         return {

chainlit/langchain/callbacks.py CHANGED Viewed

@@ -3,17 +3,19 @@ import time
 from typing import Any, Dict, List, Optional, Sequence, Tuple, TypedDict, Union
 from uuid import UUID
-from chainlit.context import context_var
-from chainlit.message import Message
-from chainlit.step import Step
-from langchain.callbacks.tracers.base import BaseTracer
+import pydantic
 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", ":"]
@@ -122,6 +124,14 @@ 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))):
@@ -249,7 +259,7 @@ DEFAULT_TO_IGNORE = [
 DEFAULT_TO_KEEP = ["retriever", "llm", "agent", "chain", "tool"]
-class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
+class LangchainTracer(AsyncBaseTracer, GenerationHelper, FinalStreamHelper):
     steps: Dict[str, Step]
     parent_id_map: Dict[str, str]
     ignored_runs: set
@@ -268,7 +278,7 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
         to_keep: Optional[List[str]] = None,
         **kwargs: Any,
     ) -> None:
-        BaseTracer.__init__(self, **kwargs)
+        AsyncBaseTracer.__init__(self, **kwargs)
         GenerationHelper.__init__(self)
         FinalStreamHelper.__init__(
             self,
@@ -296,7 +306,7 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
         else:
             self.to_keep = to_keep
-    def on_chat_model_start(
+    async def on_chat_model_start(
         self,
         serialized: Dict[str, Any],
         messages: List[List[BaseMessage]],
@@ -305,8 +315,9 @@ class LangchainTracer(BaseTracer, 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,
-    ) -> Any:
+    ) -> Run:
         lc_messages = messages[0]
         self.chat_generations[str(run_id)] = {
             "input_messages": lc_messages,
@@ -315,46 +326,48 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
             "tt_first_token": None,
         }
-        return super().on_chat_model_start(
+        return await super().on_chat_model_start(
             serialized,
             messages,
             run_id=run_id,
             parent_run_id=parent_run_id,
             tags=tags,
             metadata=metadata,
+            name=name,
             **kwargs,
         )
-    def on_llm_start(
+    async 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,
-    ) -> 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(
+    ) -> None:
+        await super().on_llm_start(
             serialized,
             prompts,
             run_id=run_id,
             parent_run_id=parent_run_id,
             tags=tags,
             metadata=metadata,
-            name=name,
             **kwargs,
         )
-    def on_llm_new_token(
+        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(
         self,
         token: str,
         *,
@@ -362,7 +375,14 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
         run_id: "UUID",
         parent_run_id: Optional["UUID"] = None,
         **kwargs: Any,
-    ) -> Run:
+    ) -> None:
+        await super().on_llm_new_token(
+            token=token,
+            chunk=chunk,
+            run_id=run_id,
+            parent_run_id=parent_run_id,
+            **kwargs,
+        )
         if isinstance(chunk, ChatGenerationChunk):
             start = self.chat_generations[str(run_id)]
         else:
@@ -377,24 +397,13 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
             if self.answer_reached:
                 if not self.final_stream:
                     self.final_stream = Message(content="")
-                    self._run_sync(self.final_stream.send())
-                self._run_sync(self.final_stream.stream_token(token))
+                    await self.final_stream.send()
+                await self.final_stream.stream_token(token)
                 self.has_streamed_final_answer = True
             else:
                 self.answer_reached = self._check_if_answer_reached()
-        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:
+    async def _persist_run(self, run: Run) -> None:
         pass
     def _get_run_parent_id(self, run: Run):
@@ -445,8 +454,8 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
                 self.ignored_runs.add(str(run.id))
             return ignore, parent_id
-    def _start_trace(self, run: Run) -> None:
-        super()._start_trace(run)
+    async def _start_trace(self, run: Run) -> None:
+        await super()._start_trace(run)
         context_var.set(self.context)
         ignore, parent_id = self._should_ignore_run(run)
@@ -459,7 +468,7 @@ class LangchainTracer(BaseTracer, 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":
@@ -489,9 +498,9 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
         self.steps[str(run.id)] = step
-        self._run_sync(step.send())
+        await step.send()
-    def _on_run_update(self, run: Run) -> None:
+    async def _on_run_update(self, run: Run) -> None:
         """Process a run upon update."""
         context_var.set(self.context)
@@ -576,10 +585,10 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
             if current_step:
                 current_step.end = utc_now()
-                self._run_sync(current_step.update())
+                await current_step.update()
             if self.final_stream and self.has_streamed_final_answer:
-                self._run_sync(self.final_stream.update())
+                await self.final_stream.update()
             return
@@ -599,16 +608,16 @@ class LangchainTracer(BaseTracer, GenerationHelper, FinalStreamHelper):
                 else output
             )
             current_step.end = utc_now()
-            self._run_sync(current_step.update())
+            await current_step.update()
-    def _on_error(self, error: BaseException, *, run_id: UUID, **kwargs: Any):
+    async 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()
-            self._run_sync(current_step.update())
+            await current_step.update()
     on_llm_error = _on_error
     on_chain_error = _on_error

chainlit/langflow/__init__.py CHANGED Viewed

@@ -8,6 +8,7 @@ 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,8 +1,5 @@
 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
@@ -10,6 +7,10 @@ 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,
@@ -143,16 +144,15 @@ 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, content=m.content or ""  # type: ignore
+                        role=m.role.value,  # type: ignore
+                        content=m.content or "",
                     )
                     for m in formatted_messages
                 ]

chainlit/message.py CHANGED Viewed

@@ -5,6 +5,9 @@ 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
@@ -22,8 +25,6 @@ from chainlit.types import (
     AskSpec,
     FileDict,
 )
-from literalai.helper import utc_now
-from literalai.observability.step import MessageStepType
 class MessageBase(ABC):
@@ -42,7 +43,6 @@ 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")
-        message = Message(
+        return Message(
             id=_dict["id"],
             parent_id=_dict.get("parentId"),
             created_at=_dict["createdAt"],
@@ -67,10 +67,9 @@ 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,
@@ -86,7 +85,6 @@ 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,
         }
@@ -114,7 +112,7 @@ class MessageBase(ABC):
             except Exception as e:
                 if self.fail_on_persist_error:
                     raise e
-                logger.error(f"Failed to persist message update: {str(e)}")
+                logger.error(f"Failed to persist message update: {e!s}")
         await context.emitter.update_step(step_dict)
@@ -134,7 +132,7 @@ class MessageBase(ABC):
             except Exception as e:
                 if self.fail_on_persist_error:
                     raise e
-                logger.error(f"Failed to persist message deletion: {str(e)}")
+                logger.error(f"Failed to persist message deletion: {e!s}")
         await context.emitter.delete_step(step_dict)
@@ -150,7 +148,7 @@ class MessageBase(ABC):
             except Exception as e:
                 if self.fail_on_persist_error:
                     raise e
-                logger.error(f"Failed to persist message creation: {str(e)}")
+                logger.error(f"Failed to persist message creation: {e!s}")
         return step_dict

chainlit/mistralai/__init__.py CHANGED Viewed

@@ -1,11 +1,12 @@
 import asyncio
 from typing import Union
-from chainlit.context import get_context
-from chainlit.step import Step
 from literalai import ChatGeneration, CompletionGeneration
 from literalai.helper import timestamp_utc
+from chainlit.context import get_context
+from chainlit.step import Step
 def instrument_mistralai():
     from literalai.instrumentation.mistralai import instrument_mistralai

chainlit/oauth_providers.py CHANGED Viewed

@@ -4,9 +4,10 @@ import urllib.parse
 from typing import Dict, List, Optional, Tuple
 import httpx
+from fastapi import HTTPException
 from chainlit.secret import random_secret
 from chainlit.user import User
-from fastapi import HTTPException
 class OAuthProvider:
@@ -22,10 +23,10 @@ class OAuthProvider:
         return all([os.environ.get(env) for env in self.env])
     async def get_token(self, code: str, url: str) -> str:
-        raise NotImplementedError()
+        raise NotImplementedError
     async def get_user_info(self, token: str) -> Tuple[Dict[str, str], User]:
-        raise NotImplementedError()
+        raise NotImplementedError
     def get_env_prefix(self) -> str:
         """Return environment prefix, like AZURE_AD."""
@@ -664,6 +665,71 @@ class GitlabOAuthProvider(OAuthProvider):
             return (gitlab_user, user)
+class KeycloakOAuthProvider(OAuthProvider):
+    env = [
+        "OAUTH_KEYCLOAK_CLIENT_ID",
+        "OAUTH_KEYCLOAK_CLIENT_SECRET",
+        "OAUTH_KEYCLOAK_REALM",
+        "OAUTH_KEYCLOAK_BASE_URL",
+    ]
+    id = os.environ.get("OAUTH_KEYCLOAK_NAME", "keycloak")
+    def __init__(self):
+        self.client_id = os.environ.get("OAUTH_KEYCLOAK_CLIENT_ID")
+        self.client_secret = os.environ.get("OAUTH_KEYCLOAK_CLIENT_SECRET")
+        self.realm = os.environ.get("OAUTH_KEYCLOAK_REALM")
+        self.base_url = os.environ.get("OAUTH_KEYCLOAK_BASE_URL")
+        self.authorize_url = (
+            f"{self.base_url}/realms/{self.realm}/protocol/openid-connect/auth"
+        )
+        self.authorize_params = {
+            "scope": "profile email openid",
+            "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,
+            "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.base_url}/realms/{self.realm}/protocol/openid-connect/token",
+                data=payload,
+            )
+            response.raise_for_status()
+            json = response.json()
+            token = json.get("access_token")
+            if not token:
+                raise httpx.HTTPStatusError(
+                    "Failed to get the access token",
+                    request=response.request,
+                    response=response,
+                )
+            return token
+    async def get_user_info(self, token: str):
+        async with httpx.AsyncClient() as client:
+            response = await client.get(
+                f"{self.base_url}/realms/{self.realm}/protocol/openid-connect/userinfo",
+                headers={"Authorization": f"Bearer {token}"},
+            )
+            response.raise_for_status()
+            kc_user = response.json()
+            user = User(
+                identifier=kc_user["email"],
+                metadata={"provider": "keycloak"},
+            )
+            return (kc_user, user)
 providers = [
     GithubOAuthProvider(),
     GoogleOAuthProvider(),
@@ -674,6 +740,7 @@ providers = [
     DescopeOAuthProvider(),
     AWSCognitoOAuthProvider(),
     GitlabOAuthProvider(),
+    KeycloakOAuthProvider(),
 ]

chainlit/openai/__init__.py CHANGED Viewed

@@ -1,11 +1,12 @@
 import asyncio
 from typing import Union
+from literalai import ChatGeneration, CompletionGeneration
+from literalai.helper import timestamp_utc
 from chainlit.context import local_steps
 from chainlit.step import Step
 from chainlit.utils import check_module_version
-from literalai import ChatGeneration, CompletionGeneration
-from literalai.helper import timestamp_utc
 def instrument_openai():

chainlit/secret.py CHANGED Viewed

@@ -6,4 +6,4 @@ chars = string.ascii_letters + string.digits + "$%*,-./:=>?@^_~"
 def random_secret(length: int = 64):
-    return "".join((secrets.choice(chars) for i in range(length)))
+    return "".join(secrets.choice(chars) for i in range(length))

chainlit 1.3.1__py3-none-any.whl → 2.0.0__py3-none-any.whl

Potentially problematic release.

chainlit 1.3.1py3-none-any.whl → 2.0.0py3-none-any.whl