U h@sHdZddlmZmZmZddlmZmZddlm Z GdddeZ dS)zUtil that calls WolframAlpha.)AnyDictOptional) BaseModelroot_validator)get_from_dict_or_envc@sbeZdZUdZeed<dZeeed<GdddZ e dde e d d d Z eed d dZ dS)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 wolfram_clientNwolfram_alpha_appidc@seZdZdZdS)zWolframAlphaAPIWrapper.ConfigZforbidN)__name__ __module__ __qualname__extrarrO/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/utilities/wolfram_alpha.pyConfigsrT)pre)valuesreturncCsTt|dd}||d<z ddl}Wntk r<tdYnX||}||d<|S)z?Validate that api key and python package exists in environment.r ZWOLFRAM_ALPHA_APPIDrNzPwolframalpha is not installed. Please install it with `pip install wolframalpha`r )r wolframalpha ImportErrorZClient)clsrr rclientrrrvalidate_environments   z+WolframAlphaAPIWrapper.validate_environment)queryrcCsh|j|}zt|jj}t|jj}Wntk r>YdSX|dksP|dkrTdSd|d|SdS)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 rnextZpodstextresults StopIteration)selfrresZ assumptionZanswerrrrrun0s  zWolframAlphaAPIWrapper.run)r r r __doc__r__annotations__r rstrrrrrr"rrrrr s  rN) r#typingrrrZlangchain_core.pydantic_v1rrZlangchain_core.utilsrrrrrrs