2023-10-23 13:22:10 +08:00
|
|
|
# Langchain 自带的 YouTube 搜索工具封装
|
|
|
|
|
from langchain.tools import YouTubeSearchTool
|
2023-10-27 22:53:43 +08:00
|
|
|
from pydantic import BaseModel, Field
|
2023-10-23 13:22:10 +08:00
|
|
|
def youtube_search(query: str):
|
|
|
|
|
tool = YouTubeSearchTool()
|
2023-10-27 22:53:43 +08:00
|
|
|
return tool.run(tool_input=query)
|
|
|
|
|
|
|
|
|
|
class YoutubeInput(BaseModel):
|
|
|
|
|
location: str = Field(description="要搜索视频关键字")
|
