Update tools/social_media_search.py

tools/social_media_search.py

@@ -1,14 +1,16 @@
 from duckduckgo_search import DDGS
+from smolagents.tools import Tool
 from typing import List, Dict
 import re
 
 
-class SocialMediaSearchTool:
+class SocialMediaSearchTool(Tool):
     name = "social_media_search"
-    description = "Visits a each and every social accounts web page of a particular person at the given url and reads its content as a markdown string. Use this to browse webpages."
-    inputs = {'url': {'type': 'string', 'description': 'The url of the social account webpage to visit.'}}
+    description = "Performs a duckduckgo web search based on your query (think a Google search) then returns the top search results."
+    inputs = {'query': {'type': 'string', 'description': 'The url of the social account webpage to visit.'}}
     output_type = "string"
-    def __init__(self):
+    def __init__(self, **kwargs):
+        super().__init__()
         self.ddgs = DDGS(**kwargs)
         self.platforms = {
             "LinkedIn": r"linkedin\.com/in/",