a bg@sLdZddlmZmZmZddlmZddlmZm Z m Z GdddeZ dS)zUtil that calls WolframAlpha.)AnyDictOptional)get_from_dict_or_env) BaseModel ConfigDictmodel_validatorc@sfeZdZUdZdZeed<dZee ed<e ddZ e dde eed d d Ze e d d dZdS)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_appidZforbid)extrabefore)mode)valuesreturncCsRt|dd}||d<z ddl}Wnty:tdYn0||}||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 rclientry/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/utilities/wolfram_alpha.pyvalidate_environments    z+WolframAlphaAPIWrapper.validate_environment)queryrcCsf|j|}zt|jj}t|jj}Wnty<YdS0|dusN|dkrRdSd|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 assumptionZanswerrrrrun2s   zWolframAlphaAPIWrapper.run)__name__ __module__ __qualname____doc__r r__annotations__r rstrrZ model_configr classmethodrrr rrrrr s  r N) r$typingrrrZlangchain_core.utilsrZpydanticrrrr rrrrs