Add metadata getter and setter in the ychat

jupyterlab · Nov 8, 2024 · 90aab6d · 90aab6d
1 parent 72598e2
commit 90aab6d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/packages/jupyterlab-collaborative-chat/src/ychat.ts b/packages/jupyterlab-collaborative-chat/src/ychat.ts
@@ -137,6 +137,16 @@ export class YChat extends YDocument<IChatChanges> {
     });
   }
 
+  getMetadata(key: string): IMetadata | undefined {
+    return this._metadata.get(key);
+  }
+
+  setMetadata(key: string, value: PartialJSONValue) {
+    this.transact(() => {
+      this._metadata.set(key, value);
+    });
+  }
+
   private _usersObserver = (event: Y.YMapEvent<IUser>): void => {
     const userChange = new Array<UserChange>();
     event.keysChanged.forEach(key => {

diff --git a/python/jupyterlab-collaborative-chat/jupyterlab_collaborative_chat/ychat.py b/python/jupyterlab-collaborative-chat/jupyterlab_collaborative_chat/ychat.py
@@ -149,7 +149,7 @@ def get_metadata(self) -> dict[str, dict]:
         """
         return self._ymetadata.to_py()
 
-    def set_metadata(self, name: str, metadata: dict):
+    def set_metadata(self, name: str, metadata: Any):
         """
         Adds or modifies a metadata of the document.
         """