o ȳg @sdddlmZmZdeeeeeffdeeeeeeeffffddZdededefdd Zd S) )DictUnioninputreturnc Cszddl}Wn tytdwt|tr&||g}|jd|jdSt|tr\t}|D] }| ||q0||}|j}d}t}|D] }||||<|d7}qI||jdSt dt |)a Sanitize input string or dict of strings by replacing sensitive data with placeholders. It returns the sanitized input string or dict of strings and the secure context as a dict following the format: { "sanitized_input": , "secure_context": } The secure context is a bytes object that is needed to de-sanitize the response from the LLM. Args: input: Input string or dict of strings. Returns: Sanitized input string or dict of strings and the secure context as a dict following the format: { "sanitized_input": , "secure_context": } The `secure_context` needs to be passed to the `desanitize` function. Raises: ValueError: If the input is not a string or dict of strings. ImportError: If the `opaqueprompts` Python package is not installed. rNhCould not import the `opaqueprompts` Python package, please install it with `pip install opaqueprompts`.)sanitized_inputsecure_contextzUnexpected input type ) opaqueprompts ImportError isinstancestrsanitizesanitized_textsrdictlistappend ValueErrortype) ropsanitize_responsevalueskeysanitize_values_responsesanitized_input_valuesidxrrg/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/opaqueprompts.pyrs6"        rsanitized_textrcCs6zddl}Wn tytdw|||}|jS)z Restore the original sensitive data from the sanitized text. Args: sanitized_text: Sanitized text. secure_context: Secure context returned by the `sanitize` function. Returns: De-sanitized text. rNr)r r desanitizedesanitized_text)rrrdesanitize_responserrrrQs  rN)typingrrr rbytesrrrrrs M