From 8058562b75001e9d9978c143b80a60ffa58886d1 Mon Sep 17 00:00:00 2001 From: Ted Sanders Date: Fri, 16 Jun 2023 16:34:58 -0700 Subject: [PATCH] changes search date to June 2023 --- ...y_generation_and_embeddings_reranking.ipynb | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/examples/Search_augmented_by_query_generation_and_embeddings_reranking.ipynb b/examples/Search_augmented_by_query_generation_and_embeddings_reranking.ipynb index f64bdf7..91989a5 100644 --- a/examples/Search_augmented_by_query_generation_and_embeddings_reranking.ipynb +++ b/examples/Search_augmented_by_query_generation_and_embeddings_reranking.ipynb @@ -192,18 +192,22 @@ } ], "source": [ - "def search_news(query: str):\n", - " # get date 1 week ago\n", - " one_week_ago = (date.today() - timedelta(weeks=1)).strftime(\"%Y-%m-%d\")\n", - "\n", + "def search_news(\n", + " query: str,\n", + " news_api_key: str = news_api_key,\n", + " num_articles: int = 50,\n", + " from_datetime: str = \"2023-06-01\", # the 2023 NBA finals were played in June 2023\n", + " to_datetime: str = \"2023-06-30\",\n", + ") -> dict:\n", " response = requests.get(\n", " \"https://newsapi.org/v2/everything\",\n", " params={\n", " \"q\": query,\n", " \"apiKey\": news_api_key,\n", - " \"pageSize\": 50,\n", + " \"pageSize\": num_articles,\n", " \"sortBy\": \"relevancy\",\n", - " \"from\": one_week_ago,\n", + " \"from\": from_datetime,\n", + " \"to\": to_datetime,\n", " },\n", " )\n", "\n", @@ -217,7 +221,7 @@ " if result[\"status\"] == \"ok\":\n", " articles = articles + result[\"articles\"]\n", " else:\n", - " raise Exception(result[\"message\"])" + " raise Exception(result[\"message\"])\n" ] }, {