update base class of ListStepContainer to BaseStepContainer (#6232)

update base class of ListStepContainer to BaseStepContainer Fixes #6231
1 year ago · af3f401015
parent 98e1bbfbbd
commit af3f401015
1 changed files with 1 additions and 1 deletions
--- 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: