Skip to content

Commit 6e173b4

Browse files
committed
fix test
1 parent cf6f70b commit 6e173b4

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/test_ai_message_utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def sample_messages():
3636
@pytest.fixture
3737
def large_messages():
3838
"""Messages that will definitely exceed size limits"""
39-
large_content = "This is a very long message. " * 1000
39+
large_content = "This is a very long message. " * 100
4040
return [
4141
{"role": "system", "content": large_content},
4242
{"role": "user", "content": large_content},
@@ -56,7 +56,7 @@ def test_no_truncation_needed(self, sample_messages):
5656

5757
def test_truncation_removes_oldest_first(self, large_messages):
5858
"""Test that oldest messages are removed first during truncation"""
59-
small_limit = MAX_GEN_AI_MESSAGE_BYTES // 100
59+
small_limit = 3000
6060
result = truncate_messages_by_size(large_messages, max_bytes=small_limit)
6161
assert len(result) < len(large_messages)
6262

@@ -246,7 +246,7 @@ class MockScope:
246246
def __init__(self):
247247
self._gen_ai_messages_truncated = {}
248248

249-
small_limit = 1000
249+
small_limit = 3000
250250
span = MockSpan()
251251
scope = MockScope()
252252
original_count = len(large_messages)

0 commit comments

Comments
 (0)