fix: remove extra params

Author: srini047

integrations/github/src/haystack_integrations/tools/github/repo_forker_tool.py

@@ -24,8 +24,6 @@ def __init__(
         description: Optional[str] = REPO_FORKER_PROMPT,
         parameters: Optional[Dict[str, Any]] = REPO_FORKER_SCHEMA,
         github_token: Optional[Secret] = None,
-        repo: Optional[str] = None,
-        branch: str = "main",
         raise_on_failure: bool = True,
         outputs_to_string: Optional[Dict[str, Union[str, Callable[[Any], str]]]] = None,
         inputs_from_state: Optional[Dict[str, str]] = None,
@@ -38,8 +36,6 @@ def __init__(
         :param description: Optional description.
         :param parameters: Optional JSON schema defining the parameters expected by the Tool.
         :param github_token: GitHub personal access token for API authentication
-        :param repo: Default repository in owner/repo format
-        :param branch: Default branch to work with
         :param raise_on_failure: If True, raises exceptions on API errors
         :param outputs_to_string:
             Optional dictionary defining how a tool outputs should be converted into a string.
@@ -66,8 +62,6 @@ def __init__(
         self.description = description
         self.parameters = parameters
         self.github_token = github_token
-        self.repo = repo
-        self.branch = branch
         self.raise_on_failure = raise_on_failure
         self.outputs_to_string = outputs_to_string
         self.inputs_from_state = inputs_from_state
@@ -100,8 +94,6 @@ def to_dict(self) -> Dict[str, Any]:
             "description": self.description,
             "parameters": self.parameters,
             "github_token": self.github_token.to_dict() if self.github_token else None,
-            "repo": self.repo,
-            "branch": self.branch,
             "raise_on_failure": self.raise_on_failure,
             "outputs_to_string": self.outputs_to_string,
             "inputs_from_state": self.inputs_from_state,

integrations/github/tests/test_repo_forker_tool.py

@@ -15,8 +15,6 @@ def test_init(self, monkeypatch):
         assert tool.description == REPO_FORKER_PROMPT
         assert tool.parameters == REPO_FORKER_SCHEMA
         assert tool.github_token is None
-        assert tool.repo is None
-        assert tool.branch == "main"
         assert tool.raise_on_failure is True
         assert tool.outputs_to_string is None
         assert tool.inputs_from_state is None
@@ -31,8 +29,6 @@ def test_from_dict(self, monkeypatch):
                 "description": REPO_FORKER_PROMPT,
                 "parameters": REPO_FORKER_SCHEMA,
                 "github_token": None,
-                "repo": None,
-                "branch": "main",
                 "raise_on_failure": True,
                 "outputs_to_string": None,
                 "inputs_from_state": None,
@@ -44,8 +40,6 @@ def test_from_dict(self, monkeypatch):
         assert tool.description == REPO_FORKER_PROMPT
         assert tool.parameters == REPO_FORKER_SCHEMA
         assert tool.github_token is None
-        assert tool.repo is None
-        assert tool.branch == "main"
         assert tool.raise_on_failure is True
         assert tool.outputs_to_string is None
         assert tool.inputs_from_state is None
@@ -60,8 +54,6 @@ def test_to_dict(self, monkeypatch):
         assert tool_dict["data"]["description"] == REPO_FORKER_PROMPT
         assert tool_dict["data"]["parameters"] == REPO_FORKER_SCHEMA
         assert tool_dict["data"]["github_token"] is None
-        assert tool_dict["data"]["repo"] is None
-        assert tool_dict["data"]["branch"] == "main"
         assert tool_dict["data"]["raise_on_failure"] is True
         assert tool_dict["data"]["outputs_to_string"] is None
         assert tool_dict["data"]["inputs_from_state"] is None
@@ -71,8 +63,6 @@ def test_to_dict_with_extra_params(self, monkeypatch):
         monkeypatch.setenv("GITHUB_TOKEN", "test-token")
         tool = GitHubRepoForkerTool(
             github_token=None,
-            repo="owner/repo",
-            branch="dev",
             raise_on_failure=False,
             outputs_to_string={"source": "docs", "handler": message_handler},
             inputs_from_state={"repository": "repo"},
@@ -84,8 +74,6 @@ def test_to_dict_with_extra_params(self, monkeypatch):
         assert tool_dict["data"]["description"] == REPO_FORKER_PROMPT
         assert tool_dict["data"]["parameters"] == REPO_FORKER_SCHEMA
         assert tool_dict["data"]["github_token"] is None
-        assert tool_dict["data"]["repo"] == "owner/repo"
-        assert tool_dict["data"]["branch"] == "dev"
         assert tool_dict["data"]["raise_on_failure"] is False
         assert (
             tool_dict["data"]["outputs_to_string"]["handler"]
@@ -107,8 +95,6 @@ def test_from_dict_with_extra_params(self, monkeypatch):
                 "description": REPO_FORKER_PROMPT,
                 "parameters": REPO_FORKER_SCHEMA,
                 "github_token": None,
-                "repo": "owner/repo",
-                "branch": "dev",
                 "raise_on_failure": False,
                 "outputs_to_string": {"handler": "haystack_integrations.tools.github.utils.message_handler"},
                 "inputs_from_state": {"repository": "repo"},
@@ -125,8 +111,6 @@ def test_from_dict_with_extra_params(self, monkeypatch):
         assert tool.description == REPO_FORKER_PROMPT
         assert tool.parameters == REPO_FORKER_SCHEMA
         assert tool.github_token is None
-        assert tool.repo == "owner/repo"
-        assert tool.branch == "dev"
         assert tool.raise_on_failure is False
         assert tool.outputs_to_string["handler"] == message_handler
         assert tool.inputs_from_state == {"repository": "repo"}