a
    bg                     @   sd   d dl mZmZ d dlmZ d dlmZ d dlmZ d dl	m
Z
 G dd deZG dd	 d	eZd
S )    )OptionalType)CallbackManagerForToolRun)BaseTool)	BaseModel)PolygonAPIWrapperc                   @   s   e Zd ZU dZeed< dS )Inputsz#Inputs for Polygon's Financials APIqueryN)__name__
__module____qualname____doc__str__annotations__ r   r   z/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/tools/polygon/financials.pyr   
   s   
r   c                   @   sf   e Zd ZU dZdZeed< dZeed< dZeed< e	Z
ee ed< eed	< deee edddZd
S )PolygonFinancialsz6Tool that gets the financials of a ticker from PolygonZget_financialsmodeZpolygon_financialsnamea  A wrapper around Polygon's Stock Financials API. This tool is useful for fetching fundamental financials from balance sheets, income statements, and cash flow statements for a stock ticker. The input should be the ticker that you want to get the latest fundamental financial data for.descriptionargs_schemaapi_wrapperN)r	   run_managerreturnc                 C   s   | j j| j|dS )zUse the Polygon API tool.)Zticker)r   runr   )selfr	   r   r   r   r   _run    s    zPolygonFinancials._run)N)r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      s   

 r   N)typingr   r   Zlangchain_core.callbacksr   Zlangchain_core.toolsr   Zpydanticr   Z%langchain_community.utilities.polygonr   r   r   r   r   r   r   <module>   s   