fix: preserve pydantic model in _replace_sensitive_data

Signed-off-by: Bohdan Petryshyn <b.y.petryshyn@gmail.com>
This commit is contained in:
Bohdan Petryshyn
2025-04-29 07:52:22 +08:00
parent 88fb9a1ff5
commit cfb628f2ee

View File

@@ -170,9 +170,9 @@ class Registry(Generic[Context]):
return [replace_secrets(v) for v in value]
return value
for key, value in params.model_dump().items():
params.__dict__[key] = replace_secrets(value)
return params
params_dump = params.model_dump()
processed_params = replace_secrets(params_dump)
return type(params).model_validate(processed_params)
# @time_execution_sync('--create_action_model')
def create_action_model(self, include_actions: Optional[list[str]] = None, page=None) -> Type[ActionModel]: