mriusero commited on
Commit
dc645d9
·
1 Parent(s): ea9236b

fix: Tool instance

Browse files
Files changed (2) hide show
  1. tools/analyze_repo.py +5 -4
  2. tools/visit_webpage.py +1 -0
tools/analyze_repo.py CHANGED
@@ -1,7 +1,8 @@
1
  from typing import Any, Dict
2
  import requests
 
3
 
4
- class GitHubCodeQualityTool:
5
  name = "github_code_quality"
6
  description = "Analyzes the code quality of a public GitHub repository."
7
  inputs = {
@@ -9,9 +10,6 @@ class GitHubCodeQualityTool:
9
  }
10
  output_type = "dict"
11
 
12
- def __init__(self):
13
- pass
14
-
15
  def forward(self, github_url: str) -> Dict[str, Any]:
16
  """
17
  Analyzes the code quality of a public GitHub repository.
@@ -58,3 +56,6 @@ class GitHubCodeQualityTool:
58
  def extract_repo_info(self, github_url: str) -> str:
59
  """Extracts repository information from the GitHub URL."""
60
  return "/".join(github_url.split("/")[-2:])
 
 
 
 
1
  from typing import Any, Dict
2
  import requests
3
+ from smolagents.tools import Tool
4
 
5
+ class GitHubCodeQualityTool(Tool):
6
  name = "github_code_quality"
7
  description = "Analyzes the code quality of a public GitHub repository."
8
  inputs = {
 
10
  }
11
  output_type = "dict"
12
 
 
 
 
13
  def forward(self, github_url: str) -> Dict[str, Any]:
14
  """
15
  Analyzes the code quality of a public GitHub repository.
 
56
  def extract_repo_info(self, github_url: str) -> str:
57
  """Extracts repository information from the GitHub URL."""
58
  return "/".join(github_url.split("/")[-2:])
59
+
60
+ def __init__(self, *args, **kwargs):
61
+ self.is_initialized = False
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"
 
3
  import requests
4
  import markdownify
5
  import smolagents
6
+ import re
7
 
8
  class VisitWebpageTool(Tool):
9
  name = "visit_webpage"