U hW @sDddlZddlmZmZmZddlmZmZmZGdddeZ dS)N)AnyDictLiteral) BaseModelFieldroot_validatorc@seZdZUdZeed<dZeed<dZe ded<e e dZ e eefed <d Zeed <ed d e e dddZeedddZdS)StackExchangeAPIWrapperzWrapper for Stack Exchange API.client max_resultsall)r titlebody query_type)default_factory fetch_paramsz result_separatorT)pre)valuesreturncCs>zddlm}|d|d<Wntk r8tdYnX|S)z1Validate that the required Python package exists.r)StackAPIZ stackoverflowr z^The 'stackapi' Python package is not installed. Please install it with `pip install stackapi`.)Zstackapir ImportError)clsrrrO/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/utilities/stackexchange.pyvalidate_environments  z,StackExchangeAPIWrapper.validate_environment)queryrc s|jdkrdn|j}|jjd||i|j}t|ddkrHd|dSdd |dDd |j}d d |dD}g}|D]zd d dd}fdd |D}dd |D} |r| r| dn|d} t| d} |d| 7}||q~|j |S)z6Run query through StackExchange API and parse results.r qsearch/excerptsitemszNo relevant results found for 'z' on Stack Overflow.cSsg|]}|ddkr|qS) item_typequestionr.0itemrrr -s z/StackExchangeAPIWrapper.run..NcSsg|]}|ddkr|qS)r!answerrr#rrrr&0s z Question: r  excerptcs g|]}|ddkr|qS)Z question_idrr$r'r"rrr&4scSsg|]}|dr|qS)Z is_acceptedrr*rrrr&9srz Answer: )r) rr fetchrlenr htmlunescapeappendrjoin) selfrZ query_keyoutputZ questionsZanswersresultsZres_textZrelevant_answersZaccepted_answersZ top_answerr)rr+rrun$s>   zStackExchangeAPIWrapper.runN)__name__ __module__ __qualname____doc__r__annotations__r intrrrdictrrstrrrrr5rrrrrs    r) r.typingrrrZlangchain_core.pydantic_v1rrrrrrrrs