a bg @s^ddlmZmZeeeeeffeeeeeeefffdddZeeedddZdS) )DictUnion)inputreturnc Csz ddl}Wnty&tdYn0t|trP||g}|jd|jdSt|trt}|D]}| ||qd||}|j}d}t}|D]}||||<|d7}q||jdSt dt |dS)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 isinstancestrsanitizeZsanitized_textsrdictlistappend ValueErrortype) ropZsanitize_responsevalueskeyZsanitize_values_responseZsanitized_input_valuesidxrry/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/utilities/opaqueprompts.pyrs4"         r)sanitized_textrrcCs:z ddl}Wnty&tdYn0|||}|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 desanitizeZdesanitized_text)rrrZdesanitize_responserrrrQs    rN)typingrrr rbytesrrrrrs M