fix(bug): refactor Tavily class to prioritize metadata over config fo…

…r API key
superagent-ai · Apr 10, 2024 · b0ad699 · b0ad699
1 parent 101626d
commit b0ad699
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/superagent/app/tools/tavily.py b/libs/superagent/app/tools/tavily.py
@@ -12,7 +12,7 @@ class Tavily(BaseTool):
 
     def _run(self, query: str) -> str:
         tavily = TavilyClient(
-            api_key=config("TAVILY_API_KEY") or self.metadata.get("apiKey")
+            api_key=self.metadata.get("apiKey") or config("TAVILY_API_KEY")
         )
         response = tavily.search(query=query, search_depth="advanced")
         context = [
@@ -22,7 +22,7 @@ def _run(self, query: str) -> str:
 
     async def _arun(self, query: str) -> str:
         tavily = TavilyClient(
-            api_key=config("TAVILY_API_KEY") or self.metadata.get("apiKey")
+            api_key=self.metadata.get("apiKey") or config("TAVILY_API_KEY")
         )
         loop = asyncio.get_event_loop()
         response = await loop.run_in_executor(None, tavily.search, query, "advanced")