Revert "Data layer refactor/cleanup (#1277)" (#1287)

This reverts commit 88f7679.
Chainlit · Sep 3, 2024 · 37148c2 · 37148c2
1 parent 88f7679
commit 37148c2
Show file tree

Hide file tree

Showing 9 changed files with 557 additions and 686 deletions.
diff --git a/backend/chainlit/data/__init__.py b/backend/chainlit/data/__init__.py
diff --git a/backend/chainlit/data/base.py b/backend/chainlit/data/base.py
diff --git a/backend/chainlit/data/dynamodb.py b/backend/chainlit/data/dynamodb.py
@@ -12,8 +12,7 @@
 import boto3  # type: ignore
 from boto3.dynamodb.types import TypeDeserializer, TypeSerializer
 from chainlit.context import context
-from chainlit.data.base import BaseDataLayer, BaseStorageClient
-from chainlit.data.utils import queue_until_user_message
+from chainlit.data import BaseDataLayer, BaseStorageClient, queue_until_user_message
 from chainlit.element import ElementDict
 from chainlit.logger import logger
 from chainlit.step import StepDict
@@ -37,6 +36,7 @@
 
 
 class DynamoDBDataLayer(BaseDataLayer):
+
     def __init__(
         self,
         table_name: str,
@@ -579,5 +579,8 @@ async def update_thread(
             updates=item,
         )
 
+    async def delete_user_session(self, id: str) -> bool:
+        return True  # Not sure why documentation wants this
+
     async def build_debug_url(self) -> str:
         return ""