o ®©Zhôã@sXdZddlZddlmZmZmZddlZddlmZddl m Z dZ Gdd„de ƒZ dS) zp Util that calls several of financial datasets stock market REST APIs. Docs: https://docs.financialdatasets.ai/ éN)ÚAnyÚListÚOptional)Úget_from_dict_or_env)Ú BaseModelz!https://api.financialdatasets.ai/c sÊeZdZUdZdZeeed<def‡fdd„ Z e defdd „ƒZ d ed ed ee dee fd d„Zd ed ed ee dee fdd„Zd ed ed ee dee fdd„Zded ededefdd„Z‡ZS)ÚFinancialDatasetsAPIWrapperz#Wrapper for financial datasets API.NÚfinancial_datasets_api_keyÚdatac s$tƒjdi|¤Žt|ddƒ|_dS)NrZFINANCIAL_DATASETS_API_KEY©)ÚsuperÚ__init__rr)Úselfr ©Ú __class__r úg/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/utilities/financial_datasets.pyr s ÿz$FinancialDatasetsAPIWrapper.__init__ÚreturncCs|jdur tdƒ‚|jS)Na#API key is required for the FinancialDatasetsAPIWrapper. Please provide the API key by either: 1. Manually specifying it when initializing the wrapper: FinancialDatasetsAPIWrapper(financial_datasets_api_key='your_api_key') 2. Setting it as an environment variable: FINANCIAL_DATASETS_API_KEY)rÚ ValueError)r r r rÚ_api_keys ÿz$FinancialDatasetsAPIWrapper._api_keyÚtickerÚperiodÚlimitcCóNt›d|›d|›d|r |nd›}d|ji}tj||d}| ¡}| dd¡S) av Get the income statements for a stock `ticker` over a `period` of time. :param ticker: the stock ticker :param period: the period of time to get the balance sheets for. Possible values are: annual, quarterly, ttm. :param limit: the number of results to return, default is 10 :return: a list of income statements z%financials/income-statements/?ticker=ú&period=ú&limit=é ú X-API-KEY©ÚheadersZincome_statementsN©ÚFINANCIAL_DATASETS_BASE_URLrÚrequestsÚgetÚjson©r rrrÚurlrÚresponser r r rÚget_income_statements'óÿþ ýÿ  z1FinancialDatasetsAPIWrapper.get_income_statementscCr) ap Get the balance sheets for a stock `ticker` over a `period` of time. :param ticker: the stock ticker :param period: the period of time to get the balance sheets for. Possible values are: annual, quarterly, ttm. :param limit: the number of results to return, default is 10 :return: a list of balance sheets z"financials/balance-sheets/?ticker=rrrrrZbalance_sheetsNrr#r r rÚget_balance_sheetsFr'z.FinancialDatasetsAPIWrapper.get_balance_sheetscCr) a| Get the cash flow statements for a stock `ticker` over a `period` of time. :param ticker: the stock ticker :param period: the period of time to get the balance sheets for. Possible values are: annual, quarterly, ttm. :param limit: the number of results to return, default is 10 :return: a list of cash flow statements z(financials/cash-flow-statements/?ticker=rrrrrZcash_flow_statementsNrr#r r rÚget_cash_flow_statementsesÿþ ýÿ  z4FinancialDatasetsAPIWrapper.get_cash_flow_statementsÚmodeÚkwargscKs¬|dkr| dd¡}| dd¡}t | |||¡¡S|dkr4| dd¡}| dd¡}t | |||¡¡S|dkrN| dd¡}| dd¡}t | |||¡¡Std|›d ƒ‚) Nr&rZannualrrr(r)z Invalid mode z for financial datasets API.)r!r"Údumpsr&r(r)r)r r*rr+rrr r rÚrun…s      zFinancialDatasetsAPIWrapper.run)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrÚstrÚ__annotations__rr ÚpropertyrÚintÚdictr&rr(r)r-Ú __classcell__r r rrrsD  þýü ûþýü ûþýü û" r) r1r"Útypingrrrr Zlangchain_core.utilsrZpydanticrrrr r r rÚs