diff --git a/langchain/experimental/plan_and_execute/schema.py b/langchain/experimental/plan_and_execute/schema.py index 1cc040120f..cb5b4cea3c 100644 --- a/langchain/experimental/plan_and_execute/schema.py +++ b/langchain/experimental/plan_and_execute/schema.py @@ -28,7 +28,7 @@ class BaseStepContainer(BaseModel): """Return the final response based on steps taken.""" -class ListStepContainer(BaseModel): +class ListStepContainer(BaseStepContainer): steps: List[Tuple[Step, StepResponse]] = Field(default_factory=list) def add_step(self, step: Step, step_response: StepResponse) -> None: