updates on new FinNLP

amura · Apr 14, 2023 · 43dfbfc · 43dfbfc
1 parent 03d433f
commit 43dfbfc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/demos/chatgpt-robo-advisor-v1/ChatGPT_Robo_Advisor.ipynb b/demos/chatgpt-robo-advisor-v1/ChatGPT_Robo_Advisor.ipynb
@@ -22,7 +22,7 @@
     "import pandas as pd\n",
     "from tqdm.notebook import tqdm\n",
     "from meta.data_processors.yahoofinance import Yahoofinance\n",
-    "from finnlp.data_sources.news.finnhub import Finnhub_News\n",
+    "from finnlp.data_sources.news.finnhub_date_range import Finnhub_Date_Range\n",
     "from finnlp.large_language_models.openai.openai_chat_agent import Openai_Chat_Agent"
    ]
   },
@@ -108,7 +108,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "news_downloader = Finnhub_News({\"token\":\"Your Finnhub Token\"})"
+    "news_downloader = Finnhub_Date_Range({\"token\":\"Your Finnhub Token\"})"
    ]
   },
   {
@@ -132,7 +132,7 @@
     }
    ],
    "source": [
-    "news_downloader.download_news(start_date = start_date,end_date = end_date, stock = \"AAPL\")"
+    "news_downloader.download_date_range_stock(start_date = start_date,end_date = end_date, stock = \"AAPL\")"
    ]
   },
   {

diff --git a/demos/chatgpt-robo-advisor-v2/ChatGPT_Robo_Advisor_v2.ipynb b/demos/chatgpt-robo-advisor-v2/ChatGPT_Robo_Advisor_v2.ipynb
@@ -17,7 +17,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from finnlp.data_sources.company_announcement.juchao import Juchao_Annoumcement_Downloader\n",
+    "from finnlp.data_sources.company_announcement.juchao import Juchao_Annoumcement\n",
     "from finnlp.large_language_models.openai.openai_chat_agent import Openai_Chat_Agent\n",
     "\n",
     "from meta.data_processors.akshare import Akshare"
@@ -73,7 +73,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "downloader = Juchao_Annoumcement_Downloader()"
+    "downloader = Juchao_Annoumcement()"
    ]
   },
   {
@@ -111,7 +111,7 @@
     }
    ],
    "source": [
-    "downloader.download(\n",
+    "downloader.download_date_range_stock(\n",
     "    start_date,\n",
     "    end_date,\n",
     "    stock,\n",