mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-19 10:14:46 +00:00
fix: mongo in unit tests
This commit is contained in:
1 parent
a3b08a5b44
commit
727495c553
1 file changed
+6
-2
@@ -729,7 +729,9 @@ class TestResearchAgentToolSetup:
|
|||||||
**agent_base_params,
|
**agent_base_params,
|
||||||
)
|
)
|
||||||
|
|
||||||
with patch(
|
with patch.object(
|
||||||
|
agent.tool_executor, "get_tools", return_value={}
|
||||||
|
), patch(
|
||||||
"application.agents.research_agent.add_internal_search_tool"
|
"application.agents.research_agent.add_internal_search_tool"
|
||||||
) as mock_add:
|
) as mock_add:
|
||||||
tools = agent._setup_tools()
|
tools = agent._setup_tools()
|
||||||
@@ -741,7 +743,9 @@ class TestResearchAgentToolSetup:
|
|||||||
):
|
):
|
||||||
agent = ResearchAgent(**agent_base_params)
|
agent = ResearchAgent(**agent_base_params)
|
||||||
|
|
||||||
with patch(
|
with patch.object(
|
||||||
|
agent.tool_executor, "get_tools", return_value={}
|
||||||
|
), patch(
|
||||||
"application.agents.research_agent.add_internal_search_tool"
|
"application.agents.research_agent.add_internal_search_tool"
|
||||||
) as mock_add:
|
) as mock_add:
|
||||||
tools = agent._setup_tools()
|
tools = agent._setup_tools()
|
||||||
|
|||||||
Reference in new issue
Block a user