a bg @sdZddlmZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZmZmZddlmZd d iZGd d d e Zd S)z!zLLMRequestsChain.T)default_factoryexcluder requests_wrapperi@int text_lengthZrequests_resultstr requests_keyurl input_keyoutput output_keyZforbid)Zarbitrary_types_allowedextraz 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: )r r$rrr output_keys6szLLMRequestsChain.output_keysbefore)moderr)valuesr"cCs0zddlm}Wnty*tdYn0|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 Urz*LLMRequestsChain._call..z html.parser callbacks)r-r,r Zget_noop_manageritemsrrgetZget_textrrrZpredictZ get_childr ) r%r1r2r,Z _run_managerZ other_keysrresZsoupresultrr$r_callLs      zLLMRequestsChain._callcCsdS)NZllm_requests_chainrr$rrr _chain_type`szLLMRequestsChain._chain_type)N)__name__ __module__ __qualname____doc____annotations__r rrrrr r Z model_configpropertyr&r'r classmethodr0r<r=rrrrrs2      rN)rA __future__rtypingrrrrZlangchain.chainsrZlangchain.chains.baserZlangchain_core.callbacksr Zpydanticr r r Z&langchain_community.utilities.requestsr rrrrrrs