Spaces:
Build error
Build error
File size: 1,890 Bytes
51ff9e5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
from __future__ import annotations
from contextlib import contextmanager
from openhands.controller.state.state import State
from openhands.core.config.condenser_config import CondenserPipelineConfig
from openhands.memory.condenser.condenser import Condensation, Condenser
from openhands.memory.view import View
class CondenserPipeline(Condenser):
"""Combines multiple condensers into a single condenser.
This is useful for creating a pipeline of condensers that can be chained together to achieve very specific condensation aims. Each condenser is run in sequence, passing the output view of one to the next, until we reach the end or a `CondensationAction` is returned instead.
"""
def __init__(self, *condenser: Condenser) -> None:
self.condensers = list(condenser)
super().__init__()
@contextmanager
def metadata_batch(self, state: State):
try:
yield
finally:
# The parent class assumes the metadata is stored in the "calling
# condenser" -- since we're not threading a State through to each
# step in the pipeline, we need to walk back through the pipeline
# and manually collect the relevant metadata.
for condenser in self.condensers:
condenser.write_metadata(state)
def condense(self, view: View) -> View | Condensation:
result: View | Condensation = view
for condenser in self.condensers:
result = condenser.condense(result)
if isinstance(result, Condensation):
break
return result
@classmethod
def from_config(cls, config: CondenserPipelineConfig) -> CondenserPipeline:
condensers = [Condenser.from_config(c) for c in config.condensers]
return CondenserPipeline(*condensers)
CondenserPipeline.register_config(CondenserPipelineConfig)
|