chainlit 1.0.504__py3-none-any.whl → 1.0.505__py3-none-any.whl

chainlit/data/__init__.py

@@ -2,16 +2,35 @@ import functools
 import json
 import os
 from collections import deque
-from typing import TYPE_CHECKING, Dict, List, Literal, Optional, Union, cast, Protocol, Any
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Dict,
+    List,
+    Literal,
+    Optional,
+    Protocol,
+    Union,
+    cast,
+)
 
 import aiofiles
 from chainlit.config import config
 from chainlit.context import context
 from chainlit.logger import logger
 from chainlit.session import WebsocketSession
-from chainlit.types import Feedback, Pagination, ThreadDict, ThreadFilter, PageInfo, PaginatedResponse
+from chainlit.types import (
+    Feedback,
+    PageInfo,
+    PaginatedResponse,
+    Pagination,
+    ThreadDict,
+    ThreadFilter,
+)
 from chainlit.user import PersistedUser, User
-from literalai import Attachment, PaginatedResponse as LiteralPaginatedResponse, Score as LiteralScore, Step as LiteralStep
+from literalai import Attachment
+from literalai import Score as LiteralScore
+from literalai import Step as LiteralStep
 from literalai.filter import threads_filters as LiteralThreadsFilters
 from literalai.step import StepDict as LiteralStepDict
 
@@ -226,7 +245,7 @@ class ChainlitDataLayer(BaseDataLayer):
         return PersistedUser(
             id=_user.id or "",
             identifier=_user.identifier or "",
-            metadata=_user.metadata,
+            metadata=user.metadata,
             createdAt=_user.created_at or "",
         )
 
@@ -411,7 +430,7 @@ class ChainlitDataLayer(BaseDataLayer):
                 }
             )
 
-        literal_response: LiteralPaginatedResponse = await self.client.api.list_threads(
+        literal_response = await self.client.api.list_threads(
             first=pagination.first,
             after=pagination.cursor,
             filters=literal_filters,
@@ -421,8 +440,8 @@ class ChainlitDataLayer(BaseDataLayer):
             pageInfo=PageInfo(
                 hasNextPage=literal_response.pageInfo.hasNextPage,
                 startCursor=literal_response.pageInfo.startCursor,
-                endCursor=literal_response.pageInfo.endCursor
-                ),
+                endCursor=literal_response.pageInfo.endCursor,
+            ),
             data=literal_response.data,
         )
 
@@ -470,11 +489,20 @@ class ChainlitDataLayer(BaseDataLayer):
             tags=tags,
         )
 
+
 class BaseStorageClient(Protocol):
     """Base class for non-text data persistence like Azure Data Lake, S3, Google Storage, etc."""
-    async def upload_file(self, object_key: str, data: Union[bytes, str], mime: str = 'application/octet-stream', overwrite: bool = True) -> Dict[str, Any]:
+
+    async def upload_file(
+        self,
+        object_key: str,
+        data: Union[bytes, str],
+        mime: str = "application/octet-stream",
+        overwrite: bool = True,
+    ) -> Dict[str, Any]:
         pass
 
+
 if api_key := os.environ.get("LITERAL_API_KEY"):
     # support legacy LITERAL_SERVER variable as fallback
     server = os.environ.get("LITERAL_API_URL") or os.environ.get("LITERAL_SERVER")