From 88f8c8ed86c5a8db5eb82f9ca710dda423a49388 Mon Sep 17 00:00:00 2001 From: lidp <43509927+guoyuhao2330@users.noreply.github.com> Date: Thu, 12 Sep 2024 19:00:35 +0800 Subject: [PATCH] Fix volcengine yfinance confliction (#2386) ### What problem does this PR solve? #2379 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- agent/component/yahoofinance.py | 9 +-------- requirements.txt | 2 +- requirements_arm.txt | 2 +- 3 files changed, 3 insertions(+), 10 deletions(-) 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