Custom Data Layer
Literal provides the simplest way to persist, analyze and monitor your data.
If you’re considering implementing a custom data layer, check out this example here for some inspiration.
Also, we would absolutely love to see a community-led open source data layer implementation and list it here. If you’re interested in contributing, please reach out to us on Discord.
How to use a custom data layer?
You need to import you custom data layer in your chainlit app, and assign it to the data layer variable, like so:
import chainlit.data as cl_data
class CustomDataLayer(cl_data.BaseDataLayer):
# TODO: implement all methods from cl_data.BaseDataLayer
cl_data._data_layer = CustomDataLayer()
SQL alchemy data layer
This custom layer has been tested for PostgreSQL, however it should support more SQL databases thanks to the use of the SQL Alchemy database.
This data layer also supports the BaseStorageClient
that enables you to store your elements into Azure Blob Storage or AWS S3.
Here is the SQL used to create the schema for this data layer:
CREATE TABLE users (
"id" UUID PRIMARY KEY,
"identifier" TEXT NOT NULL UNIQUE,
"metadata" JSONB NOT NULL,
"createdAt" TEXT
);
CREATE TABLE IF NOT EXISTS threads (
"id" UUID PRIMARY KEY,
"createdAt" TEXT,
"name" TEXT,
"userId" UUID,
"userIdentifier" TEXT,
"tags" TEXT[],
"metadata" JSONB,
FOREIGN KEY ("userId") REFERENCES users("id") ON DELETE CASCADE
);
CREATE TABLE IF NOT EXISTS steps (
"id" UUID PRIMARY KEY,
"name" TEXT NOT NULL,
"type" TEXT NOT NULL,
"threadId" UUID NOT NULL,
"parentId" UUID,
"disableFeedback" BOOLEAN NOT NULL,
"streaming" BOOLEAN NOT NULL,
"waitForAnswer" BOOLEAN,
"isError" BOOLEAN,
"metadata" JSONB,
"tags" TEXT[],
"input" TEXT,
"output" TEXT,
"createdAt" TEXT,
"start" TEXT,
"end" TEXT,
"generation" JSONB,
"showInput" TEXT,
"language" TEXT,
"indent" INT
);
CREATE TABLE IF NOT EXISTS elements (
"id" UUID PRIMARY KEY,
"threadId" UUID,
"type" TEXT,
"url" TEXT,
"chainlitKey" TEXT,
"name" TEXT NOT NULL,
"display" TEXT,
"objectKey" TEXT,
"size" TEXT,
"page" INT,
"language" TEXT,
"forId" UUID,
"mime" TEXT
);
CREATE TABLE IF NOT EXISTS feedbacks (
"id" UUID PRIMARY KEY,
"forId" UUID NOT NULL,
"value" INT NOT NULL,
"comment" TEXT
);
Example
Here is an example of setting up this data layer on a PostgreSQL database with an Azure storage client. First install the required dependencies:
pip install asyncpg SQLAlchemy azure-identity azure-storage-file-datalake
Import the custom data layer and storage client, and set the cl_data._data_layer
variable at the beginning of your Chainlit app.
import chainlit.data as cl_data
from chainlit.data.sql_alchemy import SQLAlchemyDataLayer
from chainlit.data.storage_clients import AzureStorageClient
storage_client = AzureStorageClient(account_url="<your_account_url>", container="<your_container>")
cl_data._data_layer = SQLAlchemyDataLayer(conninfo="<your conninfo>", storage_provider=storage_client)
Note that you need to add +asyncpg
to the protocol in the conninfo
string so that it uses the asyncpg library.
How to implement a custom data layer?
Follow the reference for an exhaustive list of the methods your custom data layer needs to implement.
Was this page helpful?