issue #14: update tests

ai-cfia · Apr 3, 2024 · 67bb3d0 · 67bb3d0
1 parent bb68ba0
commit 67bb3d0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/finesse/accuracy_functions.py b/finesse/accuracy_functions.py
@@ -230,7 +230,8 @@ def update_dict_bing_data(test_data: dict):
     load_dotenv()
     endpoint = os.getenv("BING_ENDPOINT")
     subscription_key = os.getenv("BING_SEARCH_KEY")
-    search_engine = BingSearch(endpoint, subscription_key, "finesse/cache/")
+    cache_path = os.getenv("CACHE_PATH", "finesse/cache/")
+    search_engine = BingSearch(endpoint, subscription_key, cache_path)
     count = 1
     for key, value in copy_data.items():
         question = value.get("question")

diff --git a/tests/test_bing_search.py b/tests/test_bing_search.py
@@ -7,7 +7,8 @@ def test_search_urls(self):
         load_dotenv()
         endpoint = os.getenv("BING_ENDPOINT")
         subscription_key = os.getenv("BING_SEARCH_KEY")
-        bing_search = BingSearch(endpoint, subscription_key)
+        cache_path = os.getenv("CACHE_PATH", "finesse/cache/")
+        bing_search = BingSearch(endpoint, subscription_key, cache_path)
 
         query = "Canadian Food Inspection Agency"
         num_results = 100