Adjusted get_num_tokens_from_messages method

langchain-ai · Feb 7, 2024 · 9f9715b · 9f9715b
1 parent c37ca45
commit 9f9715b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/core/langchain_core/language_models/base.py b/libs/core/langchain_core/language_models/base.py
@@ -283,7 +283,7 @@ def get_num_tokens_from_messages(self, messages: List[BaseMessage]) -> int:
         Returns:
             The sum of the number of tokens across the messages.
         """
-        return sum([self.get_num_tokens(get_buffer_string([m])) for m in messages])
+        return self.get_num_tokens(get_buffer_string(messages))
 
     @classmethod
     def _all_required_field_names(cls) -> Set: