fix part of broken tests

This commit is contained in:
Reinier van der Leer
2026-02-11 14:54:21 +01:00
parent ee2805d14c
commit 7083dcf226
3 changed files with 8 additions and 8 deletions

View File

@@ -76,12 +76,12 @@ class TestFindBlockFiltering:
}.get(block_id)
with patch(
"backend.api.features.chat.tools.find_block.unified_hybrid_search",
"backend.copilot.tools.find_block.unified_hybrid_search",
new_callable=AsyncMock,
return_value=(search_results, 2),
):
with patch(
"backend.api.features.chat.tools.find_block.get_block",
"backend.copilot.tools.find_block.get_block",
side_effect=mock_get_block,
):
tool = FindBlockTool()
@@ -120,12 +120,12 @@ class TestFindBlockFiltering:
}.get(block_id)
with patch(
"backend.api.features.chat.tools.find_block.unified_hybrid_search",
"backend.copilot.tools.find_block.unified_hybrid_search",
new_callable=AsyncMock,
return_value=(search_results, 2),
):
with patch(
"backend.api.features.chat.tools.find_block.get_block",
"backend.copilot.tools.find_block.get_block",
side_effect=mock_get_block,
):
tool = FindBlockTool()

View File

@@ -39,7 +39,7 @@ class TestRunBlockFiltering:
input_block = make_mock_block("input-block-id", "Input Block", BlockType.INPUT)
with patch(
"backend.api.features.chat.tools.run_block.get_block",
"backend.copilot.tools.run_block.get_block",
return_value=input_block,
):
tool = RunBlockTool()
@@ -65,7 +65,7 @@ class TestRunBlockFiltering:
)
with patch(
"backend.api.features.chat.tools.run_block.get_block",
"backend.copilot.tools.run_block.get_block",
return_value=smart_block,
):
tool = RunBlockTool()
@@ -89,7 +89,7 @@ class TestRunBlockFiltering:
)
with patch(
"backend.api.features.chat.tools.run_block.get_block",
"backend.copilot.tools.run_block.get_block",
return_value=standard_block,
):
tool = RunBlockTool()

View File

@@ -151,7 +151,7 @@ class TestSearchMarketplaceAgentsForGeneration:
return_value=mock_response,
) as mock_search,
patch(
"backend.api.features.chat.tools.agent_generator.core.get_store_listed_graphs",
"backend.copilot.tools.agent_generator.core.get_store_listed_graphs",
new_callable=AsyncMock,
return_value={"graph-123": mock_graph},
),