a
    bgF                     @   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 Ticker News APIqueryN)__name__
__module____qualname____doc__str__annotations__ r   r   {/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/tools/polygon/ticker_news.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 )PolygonTickerNewsz>Tool that gets the latest news for a given ticker from PolygonZget_ticker_newsmodeZpolygon_ticker_newsnamezA wrapper around Polygon's Ticker News API. This tool is useful for fetching the latest news for a stock. Input should be the ticker that you want to get the latest news 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PolygonTickerNews._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   