o ȳg@sXdZddlZddlmZmZmZddlZddlmZddl m Z dZ Gddde Z dS) 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 seZdZUdZdZeeed<deffdd Z e defdd Z d ed ed ee dee fd dZd ed ed ee dee fddZd ed ed ee dee fddZded ededefddZZS)FinancialDatasetsAPIWrapperz#Wrapper for financial datasets API.Nfinancial_datasets_api_keydatac s$tjdi|t|dd|_dS)NrFINANCIAL_DATASETS_API_KEY)super__init__rr)selfr  __class__r l/var/www/html/chatdoc2/venv/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)rr r r_api_keys z$FinancialDatasetsAPIWrapper._api_keytickerperiodlimitcCNtd|d|d|r |nd}d|ji}tj||d}|}|ddS) 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_URLrrequestsgetjsonrrrrurlrresponser r r rget_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=rrrrrbalance_sheetsNr r%r r rget_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=rrrrrcash_flow_statementsNr r%r r rget_cash_flow_statementses   z4FinancialDatasetsAPIWrapper.get_cash_flow_statementsmodekwargscKs|dkr|dd}|dd}t||||S|dkr4|dd}|dd}t||||S|dkrN|dd}|dd}t||||Std|d ) Nr(rannualrrr+r-z Invalid mode z for financial datasets API.)r#r$dumpsr(r+r-r)rr.rr/rrr r rruns      zFinancialDatasetsAPIWrapper.run)__name__ __module__ __qualname____doc__rrstr__annotations__rr propertyrintdictr(rr+r-r2 __classcell__r r rrrsD     " r) r6r$typingrrrr"langchain_core.utilsrpydanticrr!rr r r rs