U h @sTdZddlmZmZmZmZddlmZmZm Z ddl m Z m Z GdddeZ dS)z&Util that calls Google Finance Search.)AnyDictOptionalcast) BaseModel SecretStrroot_validator)convert_to_secret_strget_from_dict_or_envc@sbeZdZUdZeed<dZeeed<GdddZ e dde e d d d Z e e d d dZdS)GoogleFinanceAPIWrappera`Wrapper for SerpApi's Google Finance API You can create SerpApi.com key by signing up at: https://serpapi.com/users/sign_up. The wrapper uses the SerpApi.com python package: https://serpapi.com/integrations/python To use, you should have the environment variable ``SERPAPI_API_KEY`` set with your API key, or pass `serp_api_key` as a named parameter to the constructor. Example: .. code-block:: python from langchain_community.utilities import GoogleFinanceAPIWrapper google_Finance = GoogleFinanceAPIWrapper() google_Finance.run('langchain') serp_search_engineN serp_api_keyc@seZdZdZdS)zGoogleFinanceAPIWrapper.ConfigZforbidN)__name__ __module__ __qualname__extrarrP/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/utilities/google_finance.pyConfigsrT)pre)valuesreturncCsRtt|dd|d<zddlm}Wntk r@tdYnX|}||d<|S)z?Validate that api key and python package exists in environment.r ZSERPAPI_API_KEYr) SerpApiClientzigoogle-search-results is not installed. Please install it with `pip install google-search-results>=2.4.2`r )r r Zserpapir ImportError)clsrrr rrrvalidate_environments  z,GoogleFinanceAPIWrapper.validate_environment)queryrcCsJtt|j}d||d}i}||}|}|s>d|S|di}d|d}d|kr|dgd}|d } |d } |d } | d } | d } |d| dd| dd| dd| d7}n|d7}|D]l}|dks|dks|dkr||7}|d7}|t||dd 7}|d7}|||dd d 7}|d7}q|S)z-Run query through Google Finance with SerpapiZgoogle_finance)ZengineZapi_keyqz"Nothing was found from the query: marketsz Query:  futures_chainrstockpriceZprice_movement percentagemovementzstock: zprice: z percentage: z movement: zNo summary information usZasiaZeuropez : price = z , movement = )rrr Zget_secret_valuer get_dictgetstr)selfrZserpapi_api_keyparamsZ total_resultsclientrresr r!r"tempr#r$keyrrrrun4sL         zGoogleFinanceAPIWrapper.run)rrr__doc__r__annotations__r rrrrrrr(r/rrrrr s r N)r0typingrrrrZlangchain_core.pydantic_v1rrrZlangchain_core.utilsr r r rrrrs