Merge pull request assafelovic#685 from nikkag/set-default-header-tavily

Make header an optional argument in TavilyClient
ElishaKay · Jul 18, 2024 · 701d1d6 · 701d1d6
2 parents fb45321 + 5752aad
commit 701d1d6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gpt_researcher/retrievers/tavily/tavily_search.py b/gpt_researcher/retrievers/tavily/tavily_search.py
@@ -10,14 +10,14 @@ class TavilySearch():
     """
     Tavily API Retriever
     """
-    def __init__(self, query, headers, topic="general"):
+    def __init__(self, query, headers=None, topic="general"):
         """
         Initializes the TavilySearch object
         Args:
             query:
         """
         self.query = query
-        self.headers = headers
+        self.headers = headers or {}
         self.api_key = self.get_api_key()
         self.client = TavilyClient(self.api_key)
         self.topic = topic