From 31dfbfd7cfd17f3748a3349c6b8221be8e83b42f Mon Sep 17 00:00:00 2001 From: Mark Schroering Date: Mon, 22 May 2023 11:38:56 -0400 Subject: [PATCH] fix: Correct recursion error on load in JupyterHub (#178) * fix: Correct recursion error on load in JupyterHub * fix: Renamed get_current_user to get_chat_user --- packages/jupyter-ai/jupyter_ai/handlers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/packages/jupyter-ai/jupyter_ai/handlers.py b/packages/jupyter-ai/jupyter_ai/handlers.py index 24db73b2c..0a83a7058 100644 --- a/packages/jupyter-ai/jupyter_ai/handlers.py +++ b/packages/jupyter-ai/jupyter_ai/handlers.py @@ -161,7 +161,7 @@ async def get(self, *args, **kwargs): res = super().get(*args, **kwargs) await res - def get_current_user(self) -> ChatUser: + def get_chat_user(self) -> ChatUser: """Retrieves the current user. If collaborative mode is disabled, one is synthesized from the login.""" collaborative = self.config.get("LabApp", {}).get("collaborative", False) @@ -189,7 +189,7 @@ def open(self): """Handles opening of a WebSocket connection. Client ID can be retrieved from `self.client_id`.""" - current_user = self.get_current_user().dict() + current_user = self.get_chat_user().dict() client_id = self.generate_client_id() self.chat_handlers[client_id] = self