昺hddlZddlmZmZmZmZddlmZddlm Z ddl m Z ddl m Z mZddlmZejeZGdd e ZdS) N)AnyDictListOptional)CallbackManagerForLLMRun)BaseLanguageModel)LLM)Extraroot_validator)get_from_dict_or_envc eZdZUdZeed< GddZededefdZ dd e d e e e d e e d ede f d Zede fdZdS) OpaquePromptsauAn LLM wrapper 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_llmc eZdZdZejZdS)OpaquePrompts.Configz'Configuration for this pydantic object.N)__name__ __module__ __qualname____doc__r forbidextran/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/llms/opaqueprompts.pyConfigr#s55 rrvaluesreturnc ddl}n#t$rtdwxYw|jtdt |ddd}|std |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.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 rvalidate_environmentz"OpaquePrompts.validate_environment(s  & & & & &   F   > !5  ' +-Db    5   s!Npromptstop run_managerkwargsc ddl}|ptj}||g}|jd}|j||} || |j} | j S)a<Call 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("Tell me a joke.") rN)r,)secure_context) r#rget_noop_managersanitizesanitized_textsrpredict desanitizer0desanitized_text) selfr+r,r-r.r( _run_managersanitize_responsesanitized_prompt_value_str llm_responsedesanitize_responses r_callzOpaquePrompts._callDs( #""""Q&>&O&Q&Q 24fX1F1F%6%Fq%I"},, &-  68]] ,;6C6 6 #33rcdS)zSReturn type of LLM. This is an override of the base class method. r#r)r7s r _llm_typezOpaquePrompts._llm_typens r)NN)rrrrr__annotations__rr rr*strrrrrr=propertyr?rrrrr s$  ^$4<%):> (4(4(4tCy!(467 (4  (4 (4(4(4(4T3Xrr)loggingtypingrrrrlangchain_core.callbacksrlangchain_core.language_modelsr#langchain_core.language_models.llmsr langchain_core.pydantic_v1r r langchain_core.utilsr getLoggerrloggerrrrrrLs,,,,,,,,,,,,======<<<<<<333333<<<<<<<<555555  8 $ $gggggCgggggr