tidy(nodes): remove no-op model_config

Because we now customize the JSON Schema creation for GraphExecutionState, the model_config did nothing.
This commit is contained in:
psychedelicious 2024-02-17 20:02:37 +11:00 committed by Brandon Rising
parent b711c46fa4
commit 0977a5e4aa

View File

@ -7,7 +7,6 @@ from typing import Annotated, Any, Optional, TypeVar, Union, get_args, get_origi
import networkx as nx
from pydantic import (
BaseModel,
ConfigDict,
GetJsonSchemaHandler,
field_validator,
)
@ -804,22 +803,6 @@ class GraphExecutionState(BaseModel):
json_schema = handler.resolve_ref_schema(json_schema)
return json_schema
model_config = ConfigDict(
json_schema_extra={
"required": [
"id",
"graph",
"execution_graph",
"executed",
"executed_history",
"results",
"errors",
"prepared_source_mapping",
"source_prepared_mapping",
]
}
)
def next(self) -> Optional[BaseInvocation]:
"""Gets the next node ready to execute."""