diff --git a/src/khoj/routers/api_chat.py b/src/khoj/routers/api_chat.py index 76cf6d12..9af00053 100644 --- a/src/khoj/routers/api_chat.py +++ b/src/khoj/routers/api_chat.py @@ -76,6 +76,7 @@ def chat_history( request: Request, common: CommonQueryParams, conversation_id: Optional[int] = None, + n: Optional[int] = None, ): user = request.user.object validate_conversation_config() @@ -109,6 +110,13 @@ def chat_history( } ) + # Get latest N messages if N > 0 + if n > 0: + meta_log["chat"] = meta_log["chat"][-n:] + # Else return all messages except latest N + else: + meta_log["chat"] = meta_log["chat"][:n] + update_telemetry_state( request=request, telemetry_type="api",