a bg@s\dZddlZddlmZmZmZddlZddlmZddl m Z m Z dZ Gddde Z dS) zs Util that calls several of Polygon's stock market REST APIs. Docs: https://polygon.io/docs/stocks/getting-started N)AnyDictOptionalget_from_dict_or_env) BaseModelmodel_validatorzhttps://api.polygon.io/c@seZdZUdZdZeeed<edde e e dddZ eee d d d Zeee d d d Zeee d ddZee ee dddZeee edddZdS)PolygonAPIWrapperzWrapper for Polygon API.Npolygon_api_keybefore)mode)valuesreturncCst|dd}||d<|S)z%Validate that api key in environment.r ZPOLYGON_API_KEYr)clsr r rs/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/utilities/polygon.pyvalidate_environments z&PolygonAPIWrapper.validate_environment)tickerrcCsVtd|d|j}t|}|}|dd}|dvrJtd||ddS)z Get fundamental financial data, which is found in balance sheets, income statements, and cash flow statements for a given ticker. /vX/reference/financials zvX/reference/financials?ticker=&apiKey=statusNOKZ STOCKBUSINESSZSTOCKSBUSINESS API Error: resultsPOLYGON_BASE_URLr requestsgetjson ValueErrorselfrurlresponsedatarrrrget_financials s  z PolygonAPIWrapper.get_financialscCsVtd|d|j}t|}|}|dd}|dvrJtd||ddS)zw Get the most recent National Best Bid and Offer (Quote) for a ticker. /v2/last/nbbo/{ticker} z v2/last/nbbo/?apiKey=rNrrrrr rrrget_last_quote5s  z PolygonAPIWrapper.get_last_quotecCsVtd|d|j}t|}|}|dd}|dvrJtd||ddS)z Get the most recent news articles relating to a stock ticker symbol, including a summary of the article and a link to the original source. /v2/reference/news zv2/reference/news?ticker=rrNrrrrr rrrget_ticker_newsEs  z!PolygonAPIWrapper.get_ticker_news)rkwargsrc Ks|dd}|dd}|dd}|dd}|dd }|d d }td |d |d|d|d|d|jd|d|} t| } | } | dd} | dvrtd| | ddS)z Get aggregate bars for a stock over a given date range in custom time window sizes. /v2/aggs/ticker/{ticker}/range/{multiplier}/{timespan}/{from_date}/{to_date} timespandayZtimespan_multiplier from_dateNto_dateadjustedTsortasczv2/aggs/ticker/z/range//r&z &adjusted=z&sort=rrrr)rrr rrr) r!rr)r* multiplierr-r.r/r0r"r#r$rrrrget_aggregatesZs<        z PolygonAPIWrapper.get_aggregates)r rr)rcKs||dkrt||S|dkr0t||S|dkrHt||S|dkrht|j|fi|Std|ddS)Nr%r'r(r4z Invalid mode z for Polygon API.)rdumpsr%r'r(r4r)r!r rr)rrrrun|szPolygonAPIWrapper.run)__name__ __module__ __qualname____doc__r rstr__annotations__r classmethodrrrdictr%r'r(r4r6rrrrr s  "r )r:rtypingrrrrZlangchain_core.utilsrZpydanticrrrr rrrrs