diff --git a/autogen/agentchat/contrib/capabilities/transforms.py b/autogen/agentchat/contrib/capabilities/transforms.py index d9ad365b91b3..7893f76e5409 100644 --- a/autogen/agentchat/contrib/capabilities/transforms.py +++ b/autogen/agentchat/contrib/capabilities/transforms.py @@ -319,7 +319,7 @@ def __init__( text_compressor: Optional[TextCompressor] = None, min_tokens: Optional[int] = None, compression_params: Dict = dict(), - cache: Optional[AbstractCache] = Cache.disk(), + cache: Optional[AbstractCache] = None, filter_dict: Optional[Dict] = None, exclude_filter: bool = True, ): @@ -391,6 +391,7 @@ def apply_transform(self, messages: List[Dict]) -> List[Dict]: cache_key = transforms_util.cache_key(message["content"], self._min_tokens) cached_content = transforms_util.cache_content_get(self._cache, cache_key) + if cached_content is not None: message["content"], savings = cached_content else: