apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-20 15:38:09 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

rename tool type

Browse source
This commit is contained in:
yzlin 2024-02-05 11:38:07 +08:00
parent a35f5366b8
commit 20393e9d7a
11 changed files with 37 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/metagpt/roles/ci/test_ml_engineer.py
View file

@ -4,7 +4,7 @@ from metagpt.actions.ci.execute_nb_code import ExecuteNbCode
from metagpt.logs import logger
from metagpt.roles.ci.ml_engineer import MLEngineer
from metagpt.schema import Message, Plan, Task
from metagpt.tools.tool_types import ToolTypes
from metagpt.tools.tool_type import ToolType
from tests.metagpt.actions.ci.test_debug_code import CODE, DebugContext, ErrorStr
@ -61,7 +61,7 @@ async def test_mle_update_data_columns(mocker):
mle.planner.plan = MockPlan
# manually update task type to test update
mle.planner.plan.current_task.task_type = ToolTypes.DATA_PREPROCESS.value
mle.planner.plan.current_task.task_type = ToolType.DATA_PREPROCESS.value
result = await mle._update_data_columns()
assert result is not None

4
tests/metagpt/tools/test_tool_registry.py
View file

@ -1,7 +1,7 @@
import pytest
from metagpt.tools.tool_registry import ToolRegistry
from metagpt.tools.tool_types import ToolTypes
from metagpt.tools.tool_type import ToolType
@pytest.fixture
@ -11,7 +11,7 @@ def tool_registry():
@pytest.fixture
def tool_registry_full():
return ToolRegistry(tool_types=ToolTypes)
return ToolRegistry(tool_types=ToolType)
# Test Initialization