Skip to content
Original file line number Diff line number Diff line change
Expand Up @@ -75,6 +75,7 @@ def __init__(
max_refinements: int = 3,
llm_factory: Callable[[Agent], AugmentedLLM] | None = None,
context: Optional["Context"] = None,
share_memory: bool = False,
):
"""
Initialize the evaluator-optimizer workflow.
Expand All @@ -89,6 +90,7 @@ def __init__(
min_rating: Minimum acceptable quality rating
max_refinements: Maximum refinement iterations
llm_factory: Optional factory to create LLMs from agents
share_memory: Whether to share the memory between the optimizer and evaluator
"""
super().__init__(
name=name,
Expand Down Expand Up @@ -147,7 +149,8 @@ def __init__(
)
else:
raise ValueError(f"Unsupported evaluator type: {type(evaluator)}")

if share_memory:
self.evaluator_llm.history = self.optimizer_llm.history
self.min_rating = min_rating
self.max_refinements = max_refinements

Expand Down
Loading