CoderBurntt commited on
Commit
d26a26e
·
verified ·
1 Parent(s): f459fb0

Update tools/visit_webpage.py

Browse files
Files changed (1) hide show
  1. tools/visit_webpage.py +6 -8
tools/visit_webpage.py CHANGED
@@ -3,6 +3,7 @@ from smolagents.tools import Tool
3
  import requests
4
  import markdownify
5
  import smolagents
 
6
 
7
  class VisitWebpageTool(Tool):
8
  name = "visit_webpage"
@@ -10,6 +11,10 @@ class VisitWebpageTool(Tool):
10
  inputs = {'url': {'type': 'string', 'description': 'The url of the webpage to visit.'}}
11
  output_type = "string"
12
 
 
 
 
 
13
  def forward(self, url: str) -> str:
14
  try:
15
  import requests
@@ -35,11 +40,4 @@ class VisitWebpageTool(Tool):
35
  return truncate_content(markdown_content, 10000)
36
 
37
  except requests.exceptions.Timeout:
38
- return "The request timed out. Please try again later or check the URL."
39
- except RequestException as e:
40
- return f"Error fetching the webpage: {str(e)}"
41
- except Exception as e:
42
- return f"An unexpected error occurred: {str(e)}"
43
-
44
- def __init__(self, *args, **kwargs):
45
- self.is_initialized = False
 
3
  import requests
4
  import markdownify
5
  import smolagents
6
+ import re # Import the re module for regular expressions
7
 
8
  class VisitWebpageTool(Tool):
9
  name = "visit_webpage"
 
11
  inputs = {'url': {'type': 'string', 'description': 'The url of the webpage to visit.'}}
12
  output_type = "string"
13
 
14
+ def __init__(self, *args, **kwargs):
15
+ super().__init__(*args, **kwargs)
16
+ self.is_initialized = True # Example initialization logic
17
+
18
  def forward(self, url: str) -> str:
19
  try:
20
  import requests
 
40
  return truncate_content(markdown_content, 10000)
41
 
42
  except requests.exceptions.Timeout:
43
+ return