From cfb628f2ee8cdceab0779b456207f4bde0b59de9 Mon Sep 17 00:00:00 2001 From: Bohdan Petryshyn Date: Tue, 29 Apr 2025 07:52:22 +0800 Subject: [PATCH] fix: preserve pydantic model in _replace_sensitive_data Signed-off-by: Bohdan Petryshyn --- browser_use/controller/registry/service.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/browser_use/controller/registry/service.py b/browser_use/controller/registry/service.py index 89fc4bc15..d6aad0194 100644 --- a/browser_use/controller/registry/service.py +++ b/browser_use/controller/registry/service.py @@ -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]: