GoReed commited on
Commit
dd0a1a8
·
verified ·
1 Parent(s): 12c8c7d

Added youtube search tool

Browse files
Files changed (1) hide show
  1. agent.py +2 -0
agent.py CHANGED
@@ -5,6 +5,7 @@ import json
5
  import pandas as pd
6
  from smolagents import (
7
  CodeAgent,
 
8
  LiteLLMModel,
9
  DuckDuckGoSearchTool,
10
  FinalAnswerTool,
@@ -105,6 +106,7 @@ agent = CodeAgent(
105
  WikipediaSearchTool(),
106
  arxiv_search,
107
  FinalAnswerTool(),
 
108
  extract_text_from_image
109
  ],
110
  planning_interval=3,
 
5
  import pandas as pd
6
  from smolagents import (
7
  CodeAgent,
8
+ YouTubeSearchTool
9
  LiteLLMModel,
10
  DuckDuckGoSearchTool,
11
  FinalAnswerTool,
 
106
  WikipediaSearchTool(),
107
  arxiv_search,
108
  FinalAnswerTool(),
109
+ YouTubeSearchTool(),
110
  extract_text_from_image
111
  ],
112
  planning_interval=3,