U ™ÀÂhûã@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)Ú BaseModel)Úget_from_dict_or_envz!https://api.financialdatasets.ai/cs eZdZUdZdZeeed<edœ‡fdd„ Z eeee ee dœdd „Z eeee e e dœd d „Zeeee e e dœd d „Zeeeedœdd„Z‡ZS)ÚFinancialDatasetsAPIWrapperz#Wrapper for financial datasets API.NÚfinancial_datasets_api_key)Údatac s tƒjf|Žt|ddƒ|_dS)NrZFINANCIAL_DATASETS_API_KEY)ÚsuperÚ__init__rr)Úselfr ©Ú __class__©úT/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/utilities/financial_datasets.pyr s ÿz$FinancialDatasetsAPIWrapper.__init__)ÚtickerÚperiodÚlimitÚreturncCsNt›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_statementss  ÿ 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_sheets:s  ÿ 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_statementsYs  ÿ 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.)rrÚdumpsr#r$r%Ú ValueError)r r&rr'rrrrrÚrunys      zFinancialDatasetsAPIWrapper.run)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrÚstrÚ__annotations__rr ÚintÚdictr#rr$r%r*Ú __classcell__rrr rrs& û !û !û r) r.rÚtypingrrrrZlangchain_core.pydantic_v1rZlangchain_core.utilsrrrrrrrÚs