Create WebSearchTool.py

tools/WebSearchTool.py

@@ -0,0 +1,17 @@
+import requests
+import os
+
+TAVILY_API_KEY = "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")
+
+class WebSearchTool(Tool):
+    name = "web_search"
+    description = "Search the web and return relevant results."
+    inputs = {"query": {"type": "string"}}
+    output_type = "string"
+
+    def forward(self, query: str) -> str:
+        url = "https://api.tavily.com/search"
+        payload = {"api_key": TAVILY_API_KEY, "query": query}
+        r = requests.post(url, json=payload)
+        data = r.json()
+        return data["answer"]