7|hddlZddlmZmZmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZmZddlmZej&eZGd d e Zy) N)AnyDictListOptional)CallbackManagerForLLMRun)BaseLanguageModel)LLM) AIMessage)get_from_dict_or_envpre_init) ConfigDictc eZdZUdZeed< edZede de fdZ dd e d e e e d e ed ede f d Zede fdZy) 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_llmforbid)extravaluesreturnc ddl}|j tdt |ddd}|s td |S#t$r tdwxYw) 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.opaqueprompts_api_keyOPAQUEPROMPTS_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 )clsropapi_keys e/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/llms/opaqueprompts.pyvalidate_environmentz"OpaquePrompts.validate_environment(s  & >> !5  ' +-Db 5   + F  s ;ANpromptstop run_managerkwargsc \ddl}|xstj}|j|g}|jd}|j j |j|} t| tr | j} |j| |j} | jS)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)rrget_noop_managersanitizesanitized_textsrbindinvoke isinstancer content desanitizer(desanitized_text) selfr#r$r%r&r _run_managersanitize_responsesanitized_prompt_value_str llm_responsedesanitize_responses r!_callzOpaquePrompts._callDs( #"Q&>&O&O&Q 24fX1F%6%F%Fq%I"}}))t)4;; &  lI .'//L68]] ,;;6C6 #333cy)zSReturn type of LLM. This is an override of the base class method. r)r2s r! _llm_typezOpaquePrompts._llm_typeos r9)NN)__name__ __module__ __qualname____doc__r__annotations__r model_configr rr"strrrrrr8propertyr<r;r9r!rrs$ L$4<%):> )4)4tCy!)467 )4  )4 )4V3r9r)loggingtypingrrrrlangchain_core.callbacksrlangchain_core.language_modelsr#langchain_core.language_models.llmsr langchain_core.messagesr langchain_core.utilsr r pydanticr getLoggerr=loggerrr;r9r!rOs@,,=<3-?   8 $gCgr9