7|h^dZddlZddlmZmZmZddlZddlmZddl m Z dZ Gdde Z y) zp Util that calls several of financial datasets stock market REST APIs. Docs: https://docs.financialdatasets.ai/ N)AnyListOptional)get_from_dict_or_env) BaseModelz!https://api.financialdatasets.ai/c eZdZUdZdZeeed<deffd Z e defdZ ded ed ee dee fd Zded ed ee dee fd Zded ed ee dee fd ZdedededefdZxZS)FinancialDatasetsAPIWrapperz#Wrapper for financial datasets API.Nfinancial_datasets_api_keydatac Ht|di|t|dd|_y)Nr FINANCIAL_DATASETS_API_KEY)super__init__rr )selfr __class__s o/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/utilities/financial_datasets.pyrz$FinancialDatasetsAPIWrapper.__init__s)  4 *> .0L+ 'returncH|j 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)rs r_api_keyz$FinancialDatasetsAPIWrapper._api_keys/  * * 2W ...rtickerperiodlimitctd|d|d|r|nd}d|ji}tj||}|j }|jddS) 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-KEYheadersincome_statementsNFINANCIAL_DATASETS_BASE_URLrrequestsgetjsonrrrrurlr"responser s rget_income_statementsz1FinancialDatasetsAPIWrapper.get_income_statements'sp ++,hvhue"- /  .<<W5}}xx+T22rctd|d|d|r|nd}d|ji}tj||}|j }|jddS) 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=rrrr r!balance_sheetsNr$r)s rget_balance_sheetsz.FinancialDatasetsAPIWrapper.get_balance_sheetsFsp ++,hvhue"- /  .<<W5}}xx($//rctd|d|d|r|nd}d|ji}tj||}|j }|jddS) 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=rrrr r!cash_flow_statementsNr$r)s rget_cash_flow_statementsz4FinancialDatasetsAPIWrapper.get_cash_flow_statementsesp"++,hvhue"- /  .<<W5}}xx.55rmodekwargsc |dk(rJ|jdd}|jdd}tj|j|||S|dk(rJ|jdd}|jdd}tj|j |||S|dk(rJ|jdd}|jdd}tj|j |||St d|d ) Nr,rannualrrr/r2z Invalid mode z for financial datasets API.)r'r(dumpsr,r/r2r)rr3rr4rrs rrunzFinancialDatasetsAPIWrapper.runs * *ZZ(3FJJw+E::d88OP P ) )ZZ(3FJJw+E::d55ffeLM M / /ZZ(3FJJw+E::d;;FFERS S}TF2NOP Pr)__name__ __module__ __qualname____doc__r rstr__annotations__rrpropertyrintdictr,rr/r2r8 __classcell__)rs@rr r s-04 4 s  /# / /333} 3 $ 3>000} 0 d 0>666} 6 d 6@QQSQCQCQrr ) r<r(typingrrrr&langchain_core.utilsrpydanticrr%r rrrrFs4  &&5ACQ)CQr