o ȳg@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 de de fdd Z d edee fd d Zd edee fd dZd edee fddZd ede dee fddZded ede defddZdS)PolygonAPIWrapperzWrapper for Polygon API.Npolygon_api_keybefore)modevaluesreturncCst|dd}||d<|S)z%Validate that api key in environment.r POLYGON_API_KEYr)clsr r ra/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/polygon.pyvalidate_environments z&PolygonAPIWrapper.validate_environmenttickercCVtd|d|j}t|}|}|dd}|dvr%td||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=statusNOK STOCKBUSINESSSTOCKSBUSINESS API Error: resultsPOLYGON_BASE_URLr requestsgetjson ValueErrorselfrurlresponsedatarrrrget_financials    z PolygonAPIWrapper.get_financialscCr)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_quotecCr)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_newsEr*z!PolygonAPIWrapper.get_ticker_newskwargsc Ks|dd}|dd}|dd}|dd}|dd }|d d }td |d |d|d|d|d|jd|d|} t| } | } | dd} | dvr[td| | 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} timespandaytimespan_multiplier from_dateNto_dateadjustedTsortasczv2/aggs/ticker/z/range//r+z &adjusted=z&sort=rrrr)r!rr r r"r#) r%rr.r/ multiplierr3r4r5r6r&r'r(rrrrget_aggregatesZs<         z PolygonAPIWrapper.get_aggregatesr cKsx|dkr t||S|dkrt||S|dkr$t||S|dkr4t|j|fi|Std|d)Nr)r,r-r:z Invalid mode z for Polygon API.)r"dumpsr)r,r-r:r#)r%r rr.rrrrun|szPolygonAPIWrapper.run)__name__ __module__ __qualname____doc__r rstr__annotations__r classmethodrrrdictr)r,r-r:r<rrrrr s  "r )r@r"typingrrrr langchain_core.utilsrpydanticrrrr rrrrs