o
    ȳg9                     @   sX   d Z ddlmZmZmZmZ ddlmZmZ ddl	m
Z
mZmZmZ G dd de
ZdS )z&Util that calls Google Finance Search.    )AnyDictOptionalcast)convert_to_secret_strget_from_dict_or_env)	BaseModel
ConfigDict	SecretStrmodel_validatorc                   @   sj   e Zd ZU dZdZeed< dZee	 ed< e
ddZedded	ed
efddZded
efddZdS )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')
    Nserp_search_engineserp_api_keyforbid)extrabefore)modevaluesreturnc                 C   sL   t t|dd|d< zddlm} W n ty   tdw |}||d< |S )z?Validate that api key and python package exists in environment.r   SERPAPI_API_KEYr   )SerpApiClientzigoogle-search-results is not installed. Please install it with `pip install google-search-results>=2.4.2`r   )r   r   serpapir   ImportError)clsr   r   r    r   h/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/google_finance.pyvalidate_environment    s   
z,GoogleFinanceAPIWrapper.validate_environmentqueryc                 C   sJ  t t| j}d| |d}i }| |}| }|sd| S |di }d| d }d|v rf|dg d }|d	 }	|d
 }
|d }|d }|d }|d|	 dd|
 d d| d d| d 7 }n|d7 }|D ]6}|dksz|dksz|dkr||7 }|d7 }|t|| d d
 7 }|d7 }||| d d d 7 }|d7 }ql|S )z-Run query through Google Finance with Serpapigoogle_finance)engineapi_keyqz"Nothing was found from the query: marketsz
Query: 
futures_chainr   stockpriceprice_movement
percentagemovementzstock: zprice: zpercentage: z
movement: zNo summary information
usasiaeuropez
: price = z, movement = )r   r
   r   get_secret_valuer   get_dictgetstr)selfr   serpapi_api_keyparamstotal_resultsclientr"   resr$   r%   r&   tempr(   r)   keyr   r   r   run6   sN   




zGoogleFinanceAPIWrapper.run)__name__
__module____qualname____doc__r   r   __annotations__r   r   r
   r	   model_configr   classmethodr   r   r0   r9   r   r   r   r   r   	   s   
 r   N)r=   typingr   r   r   r   langchain_core.utilsr   r   pydanticr   r	   r
   r   r   r   r   r   r   <module>   s
    