initial commit

venv/Lib/site-packages/langchain_community/tools/youtube/search.py

@@ -0,0 +1,53 @@
+"""
+Adapted from https://github.com/venuv/langchain_yt_tools
+
+CustomYTSearchTool searches YouTube videos related to a person
+and returns a specified number of video URLs.
+Input to this tool should be a comma separated list,
+ - the first part contains a person name
+ - and the second(optional) a number that is the
+    maximum number of video results to return
+"""
+
+import json
+from typing import Optional
+
+from langchain_core.callbacks import CallbackManagerForToolRun
+from langchain_core.tools import BaseTool
+
+
+class YouTubeSearchTool(BaseTool):
+    """Tool that queries YouTube."""
+
+    name: str = "youtube_search"
+    description: str = (
+        "search for youtube videos associated with a person. "
+        "the input to this tool should be a comma separated list, "
+        "the first part contains a person name and the second a "
+        "number that is the maximum number of video results "
+        "to return aka num_results. the second part is optional"
+    )
+
+    def _search(self, person: str, num_results: int) -> str:
+        from youtube_search import YoutubeSearch
+
+        results = YoutubeSearch(person, num_results).to_json()
+        data = json.loads(results)
+        url_suffix_list = [
+            "https://www.youtube.com" + video["url_suffix"] for video in data["videos"]
+        ]
+        return str(url_suffix_list)
+
+    def _run(
+        self,
+        query: str,
+        run_manager: Optional[CallbackManagerForToolRun] = None,
+    ) -> str:
+        """Use the tool."""
+        values = query.split(",")
+        person = values[0]
+        if len(values) > 1:
+            num_results = int(values[1])
+        else:
+            num_results = 2
+        return self._search(person, num_results)