o ȳg@sLdZddlmZmZmZddlmZddlmZm Z m Z GdddeZ dS)zUtil that calls WolframAlpha.)AnyDictOptional)get_from_dict_or_env) BaseModel ConfigDictmodel_validatorc@sjeZdZUdZdZeed<dZee ed<e ddZ e dde d ed efd d Zd e d e fddZdS)WolframAlphaAPIWrapperzWrapper for Wolfram Alpha. Docs for using: 1. Go to wolfram alpha and sign up for a developer account 2. Create an app and get your APP ID 3. Save your APP ID into WOLFRAM_ALPHA_APPID env variable 4. pip install wolframalpha Nwolfram_clientwolfram_alpha_appidforbid)extrabefore)modevaluesreturncCsNt|dd}||d<zddl}Wn tytdw||}||d<|S)z?Validate that api key and python package exists in environment.r WOLFRAM_ALPHA_APPIDrNzPwolframalpha is not installed. Please install it with `pip install wolframalpha`r )r wolframalpha ImportErrorClient)clsrr rclientrg/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/wolfram_alpha.pyvalidate_environments   z+WolframAlphaAPIWrapper.validate_environmentquerycCsb|j|}zt|jj}t|jj}Wn tyYdSw|dus'|dkr)dSd|d|S)z0Run query through WolframAlpha and parse result.z&Wolfram Alpha wasn't able to answer itNz&No good Wolfram Alpha Result was foundz Assumption: z Answer: )r rnextpodstextresults StopIteration)selfrres assumptionanswerrrrrun2s   zWolframAlphaAPIWrapper.run)__name__ __module__ __qualname____doc__r r__annotations__r rstrr model_configr classmethodrrr&rrrrr s  r N) r*typingrrrlangchain_core.utilsrpydanticrrrr rrrrs