diff --git a/navi/api/websocket.py b/navi/api/websocket.py index 82a40cd..ed04f25 100644 --- a/navi/api/websocket.py +++ b/navi/api/websocket.py @@ -25,7 +25,7 @@ from typing import Annotated from navi.api.deps import get_session_store -from navi.auth.deps import get_current_user_ws +from navi.auth.deps import get_current_user, get_current_user_ws from navi.auth import User from navi.auth.deps import check_session_access from navi.core import Agent, SessionStore @@ -174,7 +174,7 @@ async def stop_session( session_id: str, store: Annotated[SessionStore, Depends(get_session_store)], - user: Annotated[User | None, Depends(get_current_user_ws)] = None, + user: Annotated[User | None, Depends(get_current_user)] = None, ) -> dict: """Signal the running agent for this session to stop cooperatively.""" if user is not None: