Abraham E. Tavarez commited on
Commit
788d320
·
1 Parent(s): 8a43ec3

regex import and code cleanup

Browse files
Files changed (2) hide show
  1. agent.py +2 -2
  2. tools/visit_website.py +1 -0
agent.py CHANGED
@@ -36,8 +36,8 @@ codeAgent = CodeAgent(
36
 
37
  codeAgent.logger.console.width=66
38
 
39
- res = codeAgent.run(question)
40
- print(res)
41
 
42
 
43
  # Tool Calling Agent
 
36
 
37
  codeAgent.logger.console.width=66
38
 
39
+ # res = codeAgent.run(question)
40
+ # print(res)
41
 
42
 
43
  # Tool Calling Agent
tools/visit_website.py CHANGED
@@ -15,6 +15,7 @@ class VisitWebpageTool(Tool):
15
  import requests
16
  from markdownify import markdownify
17
  from requests.exceptions import RequestException
 
18
 
19
  from smolagents.utils import truncate_content
20
  except ImportError as e:
 
15
  import requests
16
  from markdownify import markdownify
17
  from requests.exceptions import RequestException
18
+ import re
19
 
20
  from smolagents.utils import truncate_content
21
  except ImportError as e: