fix optional k arg in YChatHistory

dlqqq · Dec 24, 2024 · 37fa257 · 37fa257
1 parent 2f131eb
commit 37fa257
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/jupyter-ai/jupyter_ai/history.py b/packages/jupyter-ai/jupyter_ai/history.py
@@ -17,7 +17,7 @@ class YChatHistory(BaseChatMessageHistory):
     TODO: Consider just defining `k` as the number of messages and default to 4.
     """
 
-    def __init__(self, ychat: YChat, k: Optional[int]):
+    def __init__(self, ychat: YChat, k: Optional[int] = None):
         self.ychat = ychat
         self.k = k