a bŠÝgôã@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/cs²eZdZUdZdZeeed<edœ‡fdd„ Z e edœdd „ƒZ eeee ee d œd d „Zeeee ee d œd d„Zeeee ee d œdd„Zeeeedœdd„Z‡ZS)ÚFinancialDatasetsAPIWrapperz#Wrapper for financial datasets API.NÚfinancial_datasets_api_key)Údatac s$tƒjfi|¤Žt|ddƒ|_dS)NrZFINANCIAL_DATASETS_API_KEY)ÚsuperÚ__init__rr)Úselfr ©Ú __class__©ú~/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/utilities/financial_datasets.pyr sÿz$FinancialDatasetsAPIWrapper.__init__)ÚreturncCs|jdurtdƒ‚|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 rrrÚ_api_keys  ÿz$FinancialDatasetsAPIWrapper._api_key)ÚtickerÚperiodÚlimitrcCsNt›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 rrrÚget_income_statements'sÿþ ýÿ z1FinancialDatasetsAPIWrapper.get_income_statementscCsNt›d|›d|›d|r|nd›}d|ji}tj||d}| ¡}| dd¡S) 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"rrrÚget_balance_sheetsFsÿþ ýÿ z.FinancialDatasetsAPIWrapper.get_balance_sheetscCsNt›d|›d|›d|r|nd›}d|ji}tj||d}| ¡}| dd¡S) 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"rrrÚget_cash_flow_statementsesÿþ ýÿ z4FinancialDatasetsAPIWrapper.get_cash_flow_statements)ÚmoderÚkwargsrcKs°|dkr4| dd¡}| dd¡}t | |||¡¡S|dkrh| dd¡}| dd¡}t | |||¡¡S|dkrœ| dd¡}| dd¡}t | |||¡¡Std|›d ƒ‚dS) Nr%rZannualrrr&r'z Invalid mode z for financial datasets API.)r r!Údumpsr%r&r'r)r r(rr)rrrrrÚrun…s      zFinancialDatasetsAPIWrapper.run)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrÚstrÚ__annotations__rr ÚpropertyrÚintÚdictr%rr&r'r+Ú __classcell__rrr rrs*  û !û !û r) r/r!ÚtypingrrrrZlangchain_core.utilsrZpydanticrrrrrrrÚs