o Zh@sddlZddlmZmZmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZmZddlmZeeZGd d d e ZdS) N)AnyDictListOptional)CallbackManagerForLLMRun)BaseLanguageModel)LLM) AIMessage)get_from_dict_or_envpre_init) ConfigDictc @s~eZdZUdZeed< eddZede de fddZ dd e d e e e d e ed ede f ddZede fddZd S) OpaquePromptsajLLM that uses OpaquePrompts to sanitize prompts. Wraps another LLM and sanitizes prompts before passing it to the LLM, then de-sanitizes the response. To use, you should have the ``opaqueprompts`` python package installed, and the environment variable ``OPAQUEPROMPTS_API_KEY`` set with your API key, or pass it as a named parameter to the constructor. Example: .. code-block:: python from langchain_community.llms import OpaquePrompts from langchain_community.chat_models import ChatOpenAI op_llm = OpaquePrompts(base_llm=ChatOpenAI()) base_llmZforbid)extravaluesreturncCsVzddl}Wn tytdw|jdurtdt|dddd}|s)td |S) zFValidates that the OpaquePrompts API key and the Python package exist.rNzhCould not import the `opaqueprompts` Python package, please install it with `pip install opaqueprompts`.zMCould not properly import `opaqueprompts`, opaqueprompts.__package__ is None.Zopaqueprompts_api_keyZOPAQUEPROMPTS_API_KEY)defaultzCould not find OPAQUEPROMPTS_API_KEY in the environment. Please set it to your OpaquePrompts API key.You can get it by creating an account on the OpaquePrompts website: https://opaqueprompts.opaque.co/ .) opaqueprompts ImportError __package__ ValueErrorr )clsropZapi_keyr]/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/opaqueprompts.pyvalidate_environment(s&   z"OpaquePrompts.validate_environmentNpromptstop run_managerkwargsc Ksdddl}|p t}||g}|jd}|jj|d|} t| t r'| j } |j | |j d} | j S)aCCall base LLM with sanitization before and de-sanitization after. Args: prompt: The prompt to pass into the model. Returns: The string generated by the model. Example: .. code-block:: python response = op_llm.invoke("Tell me a joke.") rN)r)secure_context)rrZget_noop_managersanitizeZsanitized_textsrbindZinvoke isinstancer contentZ desanitizer!Zdesanitized_text) selfrrrr rZ _run_managerZsanitize_responseZsanitized_prompt_value_strZ llm_responseZdesanitize_responserrr_callDs    zOpaquePrompts._callcCsdS)zSReturn type of LLM. This is an override of the base class method. rr)r&rrr _llm_typeoszOpaquePrompts._llm_type)NN)__name__ __module__ __qualname____doc__r__annotations__r Z model_configr rrstrrrrrr'propertyr(rrrrr s0   +r )loggingtypingrrrrZlangchain_core.callbacksrZlangchain_core.language_modelsrZ#langchain_core.language_models.llmsrZlangchain_core.messagesr Zlangchain_core.utilsr r Zpydanticr getLoggerr)loggerr rrrrs