diff --git a/integrations/github/tests/test_issue_commenter_tool.py b/integrations/github/tests/test_issue_commenter_tool.py index b43872fdba..eb9545fd14 100644 --- a/integrations/github/tests/test_issue_commenter_tool.py +++ b/integrations/github/tests/test_issue_commenter_tool.py @@ -76,7 +76,7 @@ def test_to_dict_with_extra_params(self, monkeypatch): raise_on_failure=False, retry_attempts=3, outputs_to_string={"handler": message_handler}, - inputs_from_state={"repository": "repo"}, + inputs_from_state={"repository": "url"}, outputs_to_state={"documents": {"source": "success", "handler": message_handler}}, ) tool_dict = tool.to_dict() @@ -91,7 +91,7 @@ def test_to_dict_with_extra_params(self, monkeypatch): tool_dict["data"]["outputs_to_string"]["handler"] == "haystack_integrations.tools.github.utils.message_handler" ) - assert tool_dict["data"]["inputs_from_state"] == {"repository": "repo"} + assert tool_dict["data"]["inputs_from_state"] == {"repository": "url"} assert tool_dict["data"]["outputs_to_state"]["documents"]["source"] == "success" assert ( tool_dict["data"]["outputs_to_state"]["documents"]["handler"] @@ -110,7 +110,7 @@ def test_from_dict_with_extra_params(self, monkeypatch): "raise_on_failure": False, "retry_attempts": 3, "outputs_to_string": {"handler": "haystack_integrations.tools.github.utils.message_handler"}, - "inputs_from_state": {"repository": "repo"}, + "inputs_from_state": {"repository": "url"}, "outputs_to_state": { "documents": { "source": "success", @@ -127,6 +127,6 @@ def test_from_dict_with_extra_params(self, monkeypatch): assert tool.raise_on_failure is False assert tool.retry_attempts == 3 assert tool.outputs_to_string["handler"] == message_handler - assert tool.inputs_from_state == {"repository": "repo"} + assert tool.inputs_from_state == {"repository": "url"} assert tool.outputs_to_state["documents"]["source"] == "success" assert tool.outputs_to_state["documents"]["handler"] == message_handler diff --git a/integrations/github/tests/test_issue_viewer_tool.py b/integrations/github/tests/test_issue_viewer_tool.py index cd0e482bfb..36158ce1aa 100644 --- a/integrations/github/tests/test_issue_viewer_tool.py +++ b/integrations/github/tests/test_issue_viewer_tool.py @@ -72,7 +72,7 @@ def test_to_dict_with_extra_params(self, monkeypatch): raise_on_failure=False, retry_attempts=3, outputs_to_string={"handler": message_handler}, - inputs_from_state={"repository": "repo"}, + inputs_from_state={"repository": "url"}, outputs_to_state={"documents": {"source": "documents", "handler": message_handler}}, ) tool_dict = tool.to_dict() @@ -87,7 +87,7 @@ def test_to_dict_with_extra_params(self, monkeypatch): tool_dict["data"]["outputs_to_string"]["handler"] == "haystack_integrations.tools.github.utils.message_handler" ) - assert tool_dict["data"]["inputs_from_state"] == {"repository": "repo"} + assert tool_dict["data"]["inputs_from_state"] == {"repository": "url"} assert tool_dict["data"]["outputs_to_state"]["documents"]["source"] == "documents" assert ( tool_dict["data"]["outputs_to_state"]["documents"]["handler"] @@ -106,7 +106,7 @@ def test_from_dict_with_extra_params(self, monkeypatch): "raise_on_failure": False, "retry_attempts": 3, "outputs_to_string": {"handler": "haystack_integrations.tools.github.utils.message_handler"}, - "inputs_from_state": {"repository": "repo"}, + "inputs_from_state": {"repository": "url"}, "outputs_to_state": { "documents": { "source": "documents", @@ -123,6 +123,6 @@ def test_from_dict_with_extra_params(self, monkeypatch): assert tool.raise_on_failure is False assert tool.retry_attempts == 3 assert tool.outputs_to_string["handler"] == message_handler - assert tool.inputs_from_state == {"repository": "repo"} + assert tool.inputs_from_state == {"repository": "url"} assert tool.outputs_to_state["documents"]["source"] == "documents" assert tool.outputs_to_state["documents"]["handler"] == message_handler diff --git a/integrations/github/tests/test_pr_creator_tool.py b/integrations/github/tests/test_pr_creator_tool.py index d58354e368..0cffb5d49f 100644 --- a/integrations/github/tests/test_pr_creator_tool.py +++ b/integrations/github/tests/test_pr_creator_tool.py @@ -73,7 +73,7 @@ def test_to_dict_with_extra_params(self, monkeypatch): github_token=Secret.from_env_var("GITHUB_TOKEN"), raise_on_failure=False, outputs_to_string={"handler": message_handler}, - inputs_from_state={"repository": "repo"}, + inputs_from_state={"repository": "issue_url"}, outputs_to_state={"documents": {"source": "result", "handler": message_handler}}, ) tool_dict = tool.to_dict() @@ -91,7 +91,7 @@ def test_to_dict_with_extra_params(self, monkeypatch): tool_dict["data"]["outputs_to_string"]["handler"] == "haystack_integrations.tools.github.utils.message_handler" ) - assert tool_dict["data"]["inputs_from_state"] == {"repository": "repo"} + assert tool_dict["data"]["inputs_from_state"] == {"repository": "issue_url"} assert tool_dict["data"]["outputs_to_state"]["documents"]["source"] == "result" assert ( tool_dict["data"]["outputs_to_state"]["documents"]["handler"] @@ -109,7 +109,7 @@ def test_from_dict_with_extra_params(self, monkeypatch): "github_token": {"env_vars": ["GITHUB_TOKEN"], "strict": True, "type": "env_var"}, "raise_on_failure": False, "outputs_to_string": {"handler": "haystack_integrations.tools.github.utils.message_handler"}, - "inputs_from_state": {"repository": "repo"}, + "inputs_from_state": {"repository": "issue_url"}, "outputs_to_state": { "documents": { "source": "result", @@ -125,6 +125,6 @@ def test_from_dict_with_extra_params(self, monkeypatch): assert tool.github_token == Secret.from_env_var("GITHUB_TOKEN") assert tool.raise_on_failure is False assert tool.outputs_to_string["handler"] == message_handler - assert tool.inputs_from_state == {"repository": "repo"} + assert tool.inputs_from_state == {"repository": "issue_url"} assert tool.outputs_to_state["documents"]["source"] == "result" assert tool.outputs_to_state["documents"]["handler"] == message_handler diff --git a/integrations/github/tests/test_repo_forker_tool.py b/integrations/github/tests/test_repo_forker_tool.py index 316ca7e918..70090b4be9 100644 --- a/integrations/github/tests/test_repo_forker_tool.py +++ b/integrations/github/tests/test_repo_forker_tool.py @@ -71,7 +71,7 @@ def test_to_dict_with_extra_params(self, monkeypatch): github_token=Secret.from_env_var("GITHUB_TOKEN"), raise_on_failure=False, outputs_to_string={"source": "repo", "handler": message_handler}, - inputs_from_state={"repository": "repo"}, + inputs_from_state={"repository": "url"}, outputs_to_state={"documents": {"source": "repo", "handler": message_handler}}, ) tool_dict = tool.to_dict() @@ -89,7 +89,7 @@ def test_to_dict_with_extra_params(self, monkeypatch): tool_dict["data"]["outputs_to_string"]["handler"] == "haystack_integrations.tools.github.utils.message_handler" ) - assert tool_dict["data"]["inputs_from_state"] == {"repository": "repo"} + assert tool_dict["data"]["inputs_from_state"] == {"repository": "url"} assert tool_dict["data"]["outputs_to_state"]["documents"]["source"] == "repo" assert ( tool_dict["data"]["outputs_to_state"]["documents"]["handler"] @@ -107,7 +107,7 @@ def test_from_dict_with_extra_params(self, monkeypatch): "github_token": {"env_vars": ["GITHUB_TOKEN"], "strict": True, "type": "env_var"}, "raise_on_failure": False, "outputs_to_string": {"handler": "haystack_integrations.tools.github.utils.message_handler"}, - "inputs_from_state": {"repository": "repo"}, + "inputs_from_state": {"repository": "url"}, "outputs_to_state": { "documents": { "source": "repo", @@ -123,6 +123,6 @@ def test_from_dict_with_extra_params(self, monkeypatch): assert tool.github_token == Secret.from_env_var("GITHUB_TOKEN") assert tool.raise_on_failure is False assert tool.outputs_to_string["handler"] == message_handler - assert tool.inputs_from_state == {"repository": "repo"} + assert tool.inputs_from_state == {"repository": "url"} assert tool.outputs_to_state["documents"]["source"] == "repo" assert tool.outputs_to_state["documents"]["handler"] == message_handler