extensions/python/tool_execute_after/_10_mask_secrets.py

from helpers.extension import Extension
from helpers.secrets import get_secrets_manager
from helpers.tool import Response


class MaskToolSecrets(Extension):

    async def execute(self, response: Response | None = None, **kwargs):
        if not self.agent:
            return

        if not response:
            return
        secrets_mgr = get_secrets_manager(self.agent.context)
        response.message = secrets_mgr.mask_values(response.message)