esarunkumar commited on
Commit
e5817bb
·
verified ·
1 Parent(s): b273384

Upload agent

Browse files
agent.json CHANGED
@@ -1,6 +1,11 @@
1
  {
2
  "class": "CodeAgent",
3
  "tools": [
 
 
 
 
 
4
  "final_answer"
5
  ],
6
  "model": {
@@ -26,13 +31,16 @@
26
  "post_messages": "Based on the above, please provide an answer to the following user task:\n{{task}}"
27
  }
28
  },
29
- "max_steps": 20,
30
- "verbosity_level": 1,
31
  "grammar": null,
32
  "planning_interval": null,
33
  "name": null,
34
  "description": null,
35
  "requirements": [
 
 
 
36
  "smolagents"
37
  ],
38
  "authorized_imports": [
 
1
  {
2
  "class": "CodeAgent",
3
  "tools": [
4
+ "web_search",
5
+ "visit_webpage",
6
+ "suggest_menu",
7
+ "catering_service_tool",
8
+ "superhero_party_theme_generator",
9
  "final_answer"
10
  ],
11
  "model": {
 
31
  "post_messages": "Based on the above, please provide an answer to the following user task:\n{{task}}"
32
  }
33
  },
34
+ "max_steps": 10,
35
+ "verbosity_level": 2,
36
  "grammar": null,
37
  "planning_interval": null,
38
  "name": null,
39
  "description": null,
40
  "requirements": [
41
+ "duckduckgo_search",
42
+ "markdownify",
43
+ "requests",
44
  "smolagents"
45
  ],
46
  "authorized_imports": [
app.py CHANGED
@@ -5,6 +5,11 @@ from smolagents import GradioUI, CodeAgent, InferenceClientModel
5
  # Get current directory path
6
  CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
7
 
 
 
 
 
 
8
  from tools.final_answer import FinalAnswerTool as FinalAnswer
9
 
10
 
@@ -13,6 +18,11 @@ model = InferenceClientModel(
13
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
14
  )
15
 
 
 
 
 
 
16
  final_answer = FinalAnswer()
17
 
18
 
@@ -21,11 +31,11 @@ with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
21
 
22
  agent = CodeAgent(
23
  model=model,
24
- tools=[],
25
  managed_agents=[],
26
  class='CodeAgent',
27
- max_steps=20,
28
- verbosity_level=1,
29
  grammar=None,
30
  planning_interval=None,
31
  name=None,
 
5
  # Get current directory path
6
  CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
7
 
8
+ from tools.web_search import DuckDuckGoSearchTool as WebSearch
9
+ from tools.visit_webpage import VisitWebpageTool as VisitWebpage
10
+ from tools.suggest_menu import SimpleTool as SuggestMenu
11
+ from tools.catering_service_tool import SimpleTool as CateringServiceTool
12
+ from tools.superhero_party_theme_generator import SuperheroPartyThemeTool as SuperheroPartyThemeGenerator
13
  from tools.final_answer import FinalAnswerTool as FinalAnswer
14
 
15
 
 
18
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
19
  )
20
 
21
+ web_search = WebSearch()
22
+ visit_webpage = VisitWebpage()
23
+ suggest_menu = SuggestMenu()
24
+ catering_service_tool = CateringServiceTool()
25
+ superhero_party_theme_generator = SuperheroPartyThemeGenerator()
26
  final_answer = FinalAnswer()
27
 
28
 
 
31
 
32
  agent = CodeAgent(
33
  model=model,
34
+ tools=[web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator],
35
  managed_agents=[],
36
  class='CodeAgent',
37
+ max_steps=10,
38
+ verbosity_level=2,
39
  grammar=None,
40
  planning_interval=None,
41
  name=None,
requirements.txt CHANGED
@@ -1 +1,4 @@
 
 
 
1
  smolagents
 
1
+ duckduckgo_search
2
+ markdownify
3
+ requests
4
  smolagents
tools/catering_service_tool.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from smolagents import Tool
2
+ from typing import Any, Optional
3
+
4
+ class SimpleTool(Tool):
5
+ name = "catering_service_tool"
6
+ description = "This tool returns the highest-rated catering service in Gotham City."
7
+ inputs = {'query': {'type': 'string', 'description': 'A search term for finding catering services.'}}
8
+ output_type = "string"
9
+
10
+ def forward(self, query: str) -> str:
11
+ """
12
+ This tool returns the highest-rated catering service in Gotham City.
13
+
14
+ Args:
15
+ query: A search term for finding catering services.
16
+ """
17
+ # Example list of catering services and their ratings
18
+ services = {
19
+ "Gotham Catering Co.": 4.9,
20
+ "Wayne Manor Catering": 4.8,
21
+ "Gotham City Events": 4.7,
22
+ }
23
+
24
+ # Find the highest rated catering service (simulating search query filtering)
25
+ best_service = max(services, key=services.get)
26
+
27
+ return best_service
tools/suggest_menu.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from smolagents import Tool
2
+ from typing import Any, Optional
3
+
4
+ class SimpleTool(Tool):
5
+ name = "suggest_menu"
6
+ description = "Suggests a menu based on the occasion."
7
+ inputs = {'occasion': {'type': 'string', 'description': 'The type of occasion for the party.'}}
8
+ output_type = "string"
9
+
10
+ def forward(self, occasion: str) -> str:
11
+ """
12
+ Suggests a menu based on the occasion.
13
+ Args:
14
+ occasion: The type of occasion for the party.
15
+ """
16
+ if occasion == "casual":
17
+ return "Pizza, snacks, and drinks."
18
+ elif occasion == "formal":
19
+ return "3-course dinner with wine and dessert."
20
+ elif occasion == "superhero":
21
+ return "Buffet with high-energy and healthy food."
22
+ else:
23
+ return "Custom menu for the butler."
tools/superhero_party_theme_generator.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Any, Optional
2
+ from smolagents.tools import Tool
3
+
4
+ class SuperheroPartyThemeTool(Tool):
5
+ name = "superhero_party_theme_generator"
6
+ description = """
7
+ This tool suggests creative superhero-themed party ideas based on a category.
8
+ It returns a unique party theme idea."""
9
+ inputs = {'category': {'type': 'string', 'description': "The type of superhero party (e.g., 'classic heroes', 'villain masquerade', 'futuristic gotham')."}}
10
+ output_type = "string"
11
+
12
+ def forward(self, category: str):
13
+ themes = {
14
+ "classic heroes": "Justice League Gala: Guests come dressed as their favorite DC heroes with themed cocktails like 'The Kryptonite Punch'.",
15
+ "villain masquerade": "Gotham Rogues' Ball: A mysterious masquerade where guests dress as classic Batman villains.",
16
+ "futuristic gotham": "Neo-Gotham Night: A cyberpunk-style party inspired by Batman Beyond, with neon decorations and futuristic gadgets."
17
+ }
18
+
19
+ return themes.get(category.lower(), "Themed party idea not found. Try 'classic heroes', 'villain masquerade', or 'futuristic gotham'.")
20
+
21
+ def __init__(self, *args, **kwargs):
22
+ self.is_initialized = False
tools/visit_webpage.py ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Any, Optional
2
+ from smolagents.tools import Tool
3
+ import requests
4
+ import re
5
+ import markdownify
6
+
7
+ class VisitWebpageTool(Tool):
8
+ name = "visit_webpage"
9
+ description = "Visits a webpage at the given url and reads its content as a markdown string. Use this to browse webpages."
10
+ inputs = {'url': {'type': 'string', 'description': 'The url of the webpage to visit.'}}
11
+ output_type = "string"
12
+
13
+ def __init__(self, max_output_length: int = 40000):
14
+ super().__init__()
15
+ self.max_output_length = max_output_length
16
+
17
+ def _truncate_content(self, content: str, max_length: int) -> str:
18
+ if len(content) <= max_length:
19
+ return content
20
+ return (
21
+ content[: max_length // 2]
22
+ + f"\n..._This content has been truncated to stay below {max_length} characters_...\n"
23
+ + content[-max_length // 2 :]
24
+ )
25
+
26
+ def forward(self, url: str) -> str:
27
+ try:
28
+ import re
29
+
30
+ import requests
31
+ from markdownify import markdownify
32
+ from requests.exceptions import RequestException
33
+ except ImportError as e:
34
+ raise ImportError(
35
+ "You must install packages `markdownify` and `requests` to run this tool: for instance run `pip install markdownify requests`."
36
+ ) from e
37
+ try:
38
+ # Send a GET request to the URL with a 20-second timeout
39
+ response = requests.get(url, timeout=20)
40
+ response.raise_for_status() # Raise an exception for bad status codes
41
+
42
+ # Convert the HTML content to Markdown
43
+ markdown_content = markdownify(response.text).strip()
44
+
45
+ # Remove multiple line breaks
46
+ markdown_content = re.sub(r"\n{3,}", "\n\n", markdown_content)
47
+
48
+ return self._truncate_content(markdown_content, self.max_output_length)
49
+
50
+ except requests.exceptions.Timeout:
51
+ return "The request timed out. Please try again later or check the URL."
52
+ except RequestException as e:
53
+ return f"Error fetching the webpage: {str(e)}"
54
+ except Exception as e:
55
+ return f"An unexpected error occurred: {str(e)}"
tools/web_search.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Any, Optional
2
+ from smolagents.tools import Tool
3
+ import duckduckgo_search
4
+
5
+ class DuckDuckGoSearchTool(Tool):
6
+ name = "web_search"
7
+ description = "Performs a duckduckgo web search based on your query (think a Google search) then returns the top search results."
8
+ inputs = {'query': {'type': 'string', 'description': 'The search query to perform.'}}
9
+ output_type = "string"
10
+
11
+ def __init__(self, max_results=10, **kwargs):
12
+ super().__init__()
13
+ self.max_results = max_results
14
+ try:
15
+ from duckduckgo_search import DDGS
16
+ except ImportError as e:
17
+ raise ImportError(
18
+ "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."
19
+ ) from e
20
+ self.ddgs = DDGS(**kwargs)
21
+
22
+ def forward(self, query: str) -> str:
23
+ results = self.ddgs.text(query, max_results=self.max_results)
24
+ if len(results) == 0:
25
+ raise Exception("No results found! Try a less restrictive/shorter query.")
26
+ postprocessed_results = [f"[{result['title']}]({result['href']})\n{result['body']}" for result in results]
27
+ return "## Search Results\n\n" + "\n\n".join(postprocessed_results)