U hz @sdZddlmZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZmZddlmZd d iZGd d d e Zd S)z!zLLMRequestsChain.T)default_factoryexcluder requests_wrapperi@int text_lengthZrequests_resultstr requests_keyurl input_keyoutput output_keyc@seZdZdZdZdS)zLLMRequestsChain.ConfigTZforbidN)__name__ __module__ __qualname__Zarbitrary_types_allowedextrarrrrConfig)sr$z List[str])returncCs|jgS)zJWill be whatever keys the prompt expects. :meta private: rselfrrr input_keys-szLLMRequestsChain.input_keyscCs|jgS)z=Will always return text key. :meta private: )rr'rrr output_keys5szLLMRequestsChain.output_keys)prer)valuesr%cCs2zddlm}Wntk r,tdYnX|S)z?Validate that api key and python package exists in environment.r BeautifulSoupzNCould not import bs4 python package. Please install it with `pip install bs4`.)bs4r. ImportError)clsr,r.rrrvalidate_environment=s z%LLMRequestsChain.validate_environmentNzDict[str, Any]z$Optional[CallbackManagerForChainRun])inputs run_managerr%c sddlm}|pt}fdd|D}|j}j|}||d}|dj |j <j j fd| i|} j| iS)Nrr-cs i|]\}}|jkr||qSrr&).0kvr'rr Ss z*LLMRequestsChain._call..z html.parser callbacks)r/r.r Zget_noop_manageritemsrrgetZget_textrrrZpredictZ get_childr) r(r3r4r.Z _run_managerZ other_keysrresZsoupresultrr'r_callJs      zLLMRequestsChain._callcCsdS)NZllm_requests_chainrr'rrr _chain_type^szLLMRequestsChain._chain_type)N)r r!r"__doc____annotations__r rrrrrr$propertyr)r*r r2r>r?rrrrr s*      r N)r@ __future__rtypingrrrrZlangchain.chainsrZlangchain.chains.baserZlangchain_core.callbacksr Zlangchain_core.pydantic_v1r r Z&langchain_community.utilities.requestsr rr rrrrs