From 6dc0d29d8e54f313089c25c0a1e776f5bbcb66de Mon Sep 17 00:00:00 2001
From: Jiwon Kim <jiwon@openai.com>
Date: Wed, 8 Oct 2025 11:36:40 -0700
Subject: [PATCH] Add data attr to EntitySource so it can be wired up with
 onClick

---
 chatkit/types.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/chatkit/types.py b/chatkit/types.py
index c81dc2a..5cfa024 100644
--- a/chatkit/types.py
+++ b/chatkit/types.py
@@ -813,6 +813,7 @@ class EntitySource(SourceBase):
     id: str
     icon: str | None = None
     preview: Literal["lazy"] | None = None
+    data: dict[str, Any] = Field(default_factory=dict)
 
 
 Source = Annotated[