o Zh @sdZddlmZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZmZmZddlmZd d iZGd d d e Zd S)z!s zLLMRequestsChain.T)default_factoryexcluder requests_wrapperi@int text_lengthZrequests_resultstr requests_keyurl input_keyoutput output_keyZforbid)Zarbitrary_types_allowedextrareturn List[str]cC|jgS)zJWill be whatever keys the prompt expects. :meta private: rselfrrr input_keys.zLLMRequestsChain.input_keyscCr#)z=Will always return text key. :meta private: )rr%rrr output_keys6r(zLLMRequestsChain.output_keysbefore)modevaluesrrcCs*z ddlm}W|Stytdw)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_environmentNinputsDict[str, Any] run_manager$Optional[CallbackManagerForChainRun]c sddlm}|p t}fdd|D}|j}j|}||d}|dj |j <j j dd| i|} j| iS)Nrr-cs i|] \}}|jkr||qSrr$).0kvr%rr Us z*LLMRequestsChain._call..z html.parser callbacksr)r/r.r Zget_noop_manageritemsrrgetZget_textrrrZpredictZ get_childr) r&r3r5r.Z _run_managerZ other_keysrresZsoupresultrr%r_callLs       zLLMRequestsChain._callcCsdS)NZllm_requests_chainrr%rrr _chain_type`szLLMRequestsChain._chain_type)r!r")r,rr!r)N)r3r4r5r6r!r4)r!r)__name__ __module__ __qualname____doc____annotations__r rrrrrr Z model_configpropertyr'r)r classmethodr2r@rArrrrrs4         rN)rE __future__rtypingrrrrZlangchain.chainsrZlangchain.chains.baserZlangchain_core.callbacksr Zpydanticr r r Z&langchain_community.utilities.requestsr rrrrrrs