andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-28 09:38:17 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix flows lint

Browse Source
This commit is contained in:
Brandon Hancock
2024-11-01 16:40:16 -04:00
parent cf90bd3105
commit d2db938d50
1 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/crewai/flow/flow.py
View File

@@ -217,12 +217,20 @@ class Flow(Generic[T], metaclass=FlowMeta):
if isinstance(self._state, BaseModel): if isinstance(self._state, BaseModel):
# Structured state management # Structured state management
try: try:
M = self._state.__class__ # Define a function to create the dynamic class
def create_model_with_extra_forbid(
base_model: Type[BaseModel],
) -> Type[BaseModel]:
class ModelWithExtraForbid(base_model): # type: ignore
model_config = base_model.model_config.copy()
model_config["extra"] = "forbid"
# Dynamically create a new model class with 'extra' set to 'forbid' return ModelWithExtraForbid
class ModelWithExtraForbid(M):
model_config = M.model_config.copy() # Create the dynamic class
model_config["extra"] = "forbid" ModelWithExtraForbid = create_model_with_extra_forbid(
self._state.__class__
)
# Create a new instance using the combined state and inputs # Create a new instance using the combined state and inputs
self._state = cast( self._state = cast(