diff --git a/agent/context_compressor.py b/agent/context_compressor.py index 93d26d00ec..a2b6cd370b 100644 --- a/agent/context_compressor.py +++ b/agent/context_compressor.py @@ -17,8 +17,6 @@ from typing import Any, Dict, List, Optional, Sequence, Tuple from agent.image_eviction_policy import outbound_image_retire_count from agent.compression_marker import ( ELISION_MARKER_MAX_LEN, - _COMPRESSION_MARKER_PREFIX, - _COMPRESSION_MARKER_TEMPLATE, _elision_marker, elide, elide_middle, diff --git a/tests/agent/test_context_compressor.py b/tests/agent/test_context_compressor.py index 00f4de6871..b4d470fa89 100644 --- a/tests/agent/test_context_compressor.py +++ b/tests/agent/test_context_compressor.py @@ -18,6 +18,7 @@ from agent.context_compressor import ( _summarize_tool_result, _is_summary_access_or_quota_error, ) +from agent.compression_marker import _COMPRESSION_MARKER_PREFIX from hermes_state import SessionDB from agent.auxiliary_client import CODEX_STREAM_STALL_MARKER