From 15482c54b5031e35a252acb93d9d92dad3d22a8f Mon Sep 17 00:00:00 2001 From: Debanjum Date: Thu, 13 Nov 2025 19:22:21 -0800 Subject: [PATCH] Fix type of count total tokens system_message argument --- src/khoj/processor/conversation/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/khoj/processor/conversation/utils.py b/src/khoj/processor/conversation/utils.py index e85da56c..009709fd 100644 --- a/src/khoj/processor/conversation/utils.py +++ b/src/khoj/processor/conversation/utils.py @@ -790,7 +790,9 @@ def count_tokens( return len(encoder.encode(json.dumps(message_content))) -def count_total_tokens(messages: list[ChatMessage], encoder, system_message: Optional[ChatMessage]) -> Tuple[int, int]: +def count_total_tokens( + messages: list[ChatMessage], encoder, system_message: Optional[list[ChatMessage]] = None +) -> Tuple[int, int]: """Count total tokens in messages including system message""" system_message_tokens = ( sum([count_tokens(message.content, encoder) for message in system_message]) if system_message else 0