diff --git a/libs/experimental/langchain_experimental/cpal/models.py b/libs/experimental/langchain_experimental/cpal/models.py index 6be95bc70f4..6e2d83cdc8c 100644 --- a/libs/experimental/langchain_experimental/cpal/models.py +++ b/libs/experimental/langchain_experimental/cpal/models.py @@ -135,9 +135,9 @@ class ResultModel(BaseModel): class StoryModel(BaseModel): """Story data.""" - causal_operations: Any = Field(required=True) - intervention: Any = Field(required=True) - query: Any = Field(required=True) + causal_operations: Any = Field() + intervention: Any = Field() + query: Any = Field() _outcome_table: Any = PrivateAttr(default=None) _networkx_wrapper: Any = PrivateAttr(default=None) diff --git a/libs/experimental/tests/integration_tests/chains/test_cpal.py b/libs/experimental/tests/integration_tests/chains/test_cpal.py index c3ad368e5b0..ebb65839d12 100644 --- a/libs/experimental/tests/integration_tests/chains/test_cpal.py +++ b/libs/experimental/tests/integration_tests/chains/test_cpal.py @@ -261,7 +261,7 @@ class TestUnitCPALChain_MathWordProblems(unittest.TestCase): expected_output = { "chain_answer": None, "chain_data": QueryModel( - narrative_input="how many pets does jan have? ", + question="how many pets does jan have? ", llm_error_msg="", expression="SELECT name, value FROM df WHERE name = 'jan'", ),