mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-25 21:36:52 +08:00
25d9cf2890
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: Graham Neubig <neubig@gmail.com> Co-authored-by: Engel Nyst <enyst@users.noreply.github.com>
40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
from openhands.core.config.condenser_config import ObservationMaskingCondenserConfig
|
|
from openhands.events.event import Event
|
|
from openhands.events.observation import Observation
|
|
from openhands.events.observation.agent import AgentCondensationObservation
|
|
from openhands.llm.llm_registry import LLMRegistry
|
|
from openhands.memory.condenser.condenser import Condensation, Condenser, View
|
|
|
|
|
|
class ObservationMaskingCondenser(Condenser):
|
|
"""A condenser that masks the values of observations outside of a recent attention window."""
|
|
|
|
def __init__(self, attention_window: int = 5):
|
|
self.attention_window = attention_window
|
|
|
|
super().__init__()
|
|
|
|
def condense(self, view: View) -> View | Condensation:
|
|
"""Replace the content of observations outside of the attention window with a placeholder."""
|
|
results: list[Event] = []
|
|
for i, event in enumerate(view):
|
|
if isinstance(event, Observation) and i < len(view) - self.attention_window:
|
|
results.append(AgentCondensationObservation('<MASKED>'))
|
|
else:
|
|
results.append(event)
|
|
|
|
return View(events=results)
|
|
|
|
@classmethod
|
|
def from_config(
|
|
cls,
|
|
config: ObservationMaskingCondenserConfig,
|
|
llm_registry: LLMRegistry,
|
|
) -> ObservationMaskingCondenser:
|
|
return ObservationMaskingCondenser(**config.model_dump(exclude={'type'}))
|
|
|
|
|
|
ObservationMaskingCondenser.register_config(ObservationMaskingCondenserConfig)
|