From af3f4010155a882b8b1021b6e0de130c628dab2c Mon Sep 17 00:00:00 2001 From: lucasiscovici Date: Thu, 13 Jul 2023 09:03:02 +0200 Subject: [PATCH] update base class of ListStepContainer to BaseStepContainer (#6232) update base class of ListStepContainer to BaseStepContainer Fixes #6231 --- langchain/experimental/plan_and_execute/schema.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: