mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-12 21:11:43 +00:00
core: Fix IndexError
when trim_messages
invoked with empty list (#26896)
This prevents `trim_messages` from raising an `IndexError` when invoked with `include_system=True`, `strategy="last"`, and an empty message list. Fixes #26895 Dependencies: none
This commit is contained in:
@@ -332,6 +332,19 @@ def test_trim_messages_allow_partial_text_splitter() -> None:
|
||||
assert _MESSAGES_TO_TRIM == _MESSAGES_TO_TRIM_COPY
|
||||
|
||||
|
||||
def test_trim_messages_include_system_strategy_last_empty_messages() -> None:
|
||||
expected: list[BaseMessage] = []
|
||||
|
||||
actual = trim_messages(
|
||||
max_tokens=10,
|
||||
token_counter=dummy_token_counter,
|
||||
strategy="last",
|
||||
include_system=True,
|
||||
).invoke([])
|
||||
|
||||
assert actual == expected
|
||||
|
||||
|
||||
def test_trim_messages_invoke() -> None:
|
||||
actual = trim_messages(max_tokens=10, token_counter=dummy_token_counter).invoke(
|
||||
_MESSAGES_TO_TRIM
|
||||
|
Reference in New Issue
Block a user