Skip to content

Commit c80589f

Browse files
committed
fix tool recommend
1 parent 222a44d commit c80589f

File tree

3 files changed

+4
-7
lines changed

3 files changed

+4
-7
lines changed

metagpt/roles/role_zero.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@
2929
)
3030
from metagpt.core.roles import BaseRoleZero
3131
from metagpt.core.schema import AIMessage, Message, UserMessage
32-
from metagpt.core.tools.tool_recommend import BM25ToolRecommender
3332
from metagpt.core.tools.tool_registry import register_tool
3433
from metagpt.core.utils.common import CodeParser, any_to_str, extract_and_encode_images
3534
from metagpt.core.utils.repair_llm_raw_output import (
@@ -41,6 +40,7 @@
4140
from metagpt.strategy.planner import Planner
4241
from metagpt.tools.libs.browser import Browser
4342
from metagpt.tools.libs.editor import Editor
43+
from metagpt.tools.tool_recommend import BM25ToolRecommender
4444

4545

4646
@register_tool(include_functions=["ask_human", "reply_to_human"])

tests/metagpt/roles/di/test_data_analyst.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55
from metagpt.actions.execute_nb_code import ExecuteNbCode
66
from metagpt.actions.write_analysis_code import WriteAnalysisCode
77
from metagpt.core.logs import logger
8-
from metagpt.core.tools.tool_recommend import BM25ToolRecommender
98
from metagpt.roles.data_analyst import DataAnalyst
9+
from metagpt.tools.tool_recommend import BM25ToolRecommender
1010

1111

1212
class TestDataAnalyst:

tests/metagpt/tools/test_tool_recommend.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,8 @@
22

33
from metagpt.core.schema import Plan, Task
44
from metagpt.core.tools import TOOL_REGISTRY
5-
from metagpt.core.tools.tool_recommend import (
6-
BM25ToolRecommender,
7-
ToolRecommender,
8-
TypeMatchToolRecommender,
9-
)
5+
from metagpt.core.tools.tool_recommend_base import ToolRecommender
6+
from metagpt.tools.tool_recommend import BM25ToolRecommender, TypeMatchToolRecommender
107

118

129
@pytest.fixture

0 commit comments

Comments
 (0)