feat(pkg-py): Expose id in .ui, .sidebar and .server methods

Author: gadenbuie

pkg-py/src/querychat/_querychat.py

@@ -188,6 +188,7 @@ def sidebar(
         width: int = 400,
         height: str = "100%",
         fillable: bool = True,
+        id: Optional[str] = None,
         **kwargs,
     ) -> ui.Sidebar:
         """
@@ -201,6 +202,9 @@ def sidebar(
             Height of the sidebar.
         fillable
             Whether the sidebar should be fillable. Default is `True`.
+        id
+            Optional ID for the QueryChat instance. If not provided,
+            will use the ID provided at initialization.
         **kwargs
             Additional arguments passed to `shiny.ui.sidebar()`.
 
@@ -211,20 +215,23 @@ def sidebar(
 
         """
         return ui.sidebar(
-            self.ui(),
+            self.ui(id=id),
             width=width,
             height=height,
             fillable=fillable,
             class_="querychat-sidebar",
             **kwargs,
         )
 
-    def ui(self, **kwargs):
+    def ui(self, *, id: Optional[str] = None, **kwargs):
         """
         Create the UI for the querychat component.
 
         Parameters
         ----------
+        id
+            Optional ID for the QueryChat instance. If not provided,
+            will use the ID provided at initialization.
         **kwargs
             Additional arguments to pass to `shinychat.chat_ui()`.
 
@@ -234,7 +241,7 @@ def ui(self, **kwargs):
             A UI component.
 
         """
-        return mod_ui(self.id, **kwargs)
+        return mod_ui(id or self.id, **kwargs)
 
     def generate_greeting(self, *, echo: Literal["none", "output"] = "none"):
         """
@@ -561,7 +568,9 @@ class QueryChat(QueryChatBase):
 
     """
 
-    def server(self, *, enable_bookmarking: bool = False) -> ServerValues:
+    def server(
+        self, *, enable_bookmarking: bool = False, id: Optional[str] = None
+    ) -> ServerValues:
         """
         Initialize Shiny server logic.
 
@@ -574,6 +583,10 @@ def server(self, *, enable_bookmarking: bool = False) -> ServerValues:
         ----------
         enable_bookmarking
             Whether to enable bookmarking for the querychat module.
+        id
+            Optional module ID for the QueryChat instance. If not provided,
+            will use the ID provided at initialization. This must match the ID
+            used in the `.ui()` or `.sidebar()` methods.
 
         Examples
         --------
@@ -629,7 +642,7 @@ def title():
             )
 
         return mod_server(
-            self.id,
+            id or self.id,
             data_source=self._data_source,
             greeting=self.greeting,
             client=self.client,