diff --git a/cognee/api/v1/cognify/routers/get_cognify_router.py b/cognee/api/v1/cognify/routers/get_cognify_router.py index 1882a3fb4..0382e31e0 100644 --- a/cognee/api/v1/cognify/routers/get_cognify_router.py +++ b/cognee/api/v1/cognify/routers/get_cognify_router.py @@ -163,6 +163,14 @@ def get_cognify_router() -> APIRouter: user_manager=user_manager, bearer=None, ) + logger.info( + f"WebSocket user authenticated for pipeline {pipeline_run_id}", + extra={ + "user_id": user.id, + "user_email": user.email, + "pipeline_run_id": str(pipeline_run_id), + }, + ) except Exception as error: logger.error(f"Authentication failed: {str(error)}") await websocket.close(code=WS_1008_POLICY_VIOLATION, reason="Unauthorized") diff --git a/cognee/api/v1/search/routers/get_search_router.py b/cognee/api/v1/search/routers/get_search_router.py index 37123dd3d..57a9203b1 100644 --- a/cognee/api/v1/search/routers/get_search_router.py +++ b/cognee/api/v1/search/routers/get_search_router.py @@ -9,8 +9,11 @@ from cognee.modules.users.methods import get_authenticated_user from cognee.modules.users.models import User from cognee.modules.users.exceptions import PermissionDeniedError from cognee.modules.data.methods import get_history +from cognee.shared.logging_utils import get_logger from cognee.exceptions import UnsupportedSearchTypeError, InvalidQueryError, NoDataToProcessError +logger = get_logger() + class SearchPayloadDTO(InDTO): search_type: SearchType @@ -86,6 +89,14 @@ def get_search_router() -> APIRouter: # Check if search type is supported try: search_type = payload.search_type + logger.info( + f"Search type validated: {search_type.value}", + extra={ + "search_type": search_type.value, + "user_id": user.id, + "query_length": len(payload.query), + }, + ) except ValueError: raise UnsupportedSearchTypeError( search_type=str(payload.search_type), supported_types=[t.value for t in SearchType]