diff --git a/agent/component/yahoofinance.py b/agent/component/yahoofinance.py index 5bb13a422..aa2cd25b4 100644 --- a/agent/component/yahoofinance.py +++ b/agent/component/yahoofinance.py @@ -62,15 +62,8 @@ class YahooFinance(ComponentBase, ABC): yohoo_res.append({"content": "info:\n" + pd.Series(msft.info).to_markdown() + "\n"}) if self._param.history: yohoo_res.append({"content": "history:\n" + msft.history().to_markdown() + "\n"}) - if self._param.count: - yohoo_res.append({"content": "count:\n" + msft.get_shares_full().to_markdown() + "\n"}) if self._param.financials: - yohoo_res.append({"content": "calendar:\n" + pd.DataFrame(msft.calendar).to_markdown + "\n"}) - yohoo_res.append({"content": "sec_filings:\n" + pd.DataFrame(msft.sec_filings).to_markdown() + "\n"}) - if self._param.income_stmt: - yohoo_res.append({"content": "income statement:\n" + msft.income_stmt.to_markdown() + "\n"}) - yohoo_res.append( - {"content": "quarterly income statement:\n" + msft.quarterly_income_stmt.to_markdown() + "\n"}) + yohoo_res.append({"content": "calendar:\n" + pd.DataFrame(msft.calendar).to_markdown() + "\n"}) if self._param.balance_sheet: yohoo_res.append({"content": "balance sheet:\n" + msft.balance_sheet.to_markdown() + "\n"}) yohoo_res.append( diff --git a/requirements.txt b/requirements.txt index 2a8f57721..15a889f9d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -101,5 +101,5 @@ wikipedia==1.4.0 word2number==1.1 xgboost==2.1.0 xpinyin==0.7.6 -yfinance==0.2.43 +yfinance==0.1.96 zhipuai==2.0.1 diff --git a/requirements_arm.txt b/requirements_arm.txt index 90b0de1f0..b70c13b03 100644 --- a/requirements_arm.txt +++ b/requirements_arm.txt @@ -168,7 +168,7 @@ deepl==1.18.0 psycopg2-binary==2.9.9 tabulate==0.9.0 vertexai==1.64.0 -yfinance==0.2.43 +yfinance==0.1.96 pywencai==0.12.2 akshare==1.14.72 ranx==0.3.20