diff --git a/metagpt/actions/write_prd.py b/metagpt/actions/write_prd.py index 1cb857a62..8e4229991 100644 --- a/metagpt/actions/write_prd.py +++ b/metagpt/actions/write_prd.py @@ -66,7 +66,7 @@ NEW_REQ_TEMPLATE = """ class WritePRD(Action): - name: str = "" + name: str = "WritePRD" content: Optional[str] = None llm: BaseLLM = Field(default_factory=LLM) diff --git a/metagpt/management/skill_manager.py b/metagpt/management/skill_manager.py index 5ab6273fb..2ddf98ee3 100644 --- a/metagpt/management/skill_manager.py +++ b/metagpt/management/skill_manager.py @@ -28,7 +28,7 @@ class SkillManager: :return: """ self._skills[skill.name] = skill - self._store.add(skill.desc, {}, skill.name) + self._store.add(skill.desc, {"name": skill.name, "desc": skill.desc}, skill.name) def del_skill(self, skill_name: str): """ diff --git a/requirements.txt b/requirements.txt index 81d81ba9c..cab719f24 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,13 +1,13 @@ aiohttp==3.8.4 #azure_storage==0.37.0 channels==4.0.0 -# chromadb==0.3.22 +chromadb==0.4.21 # Django==4.1.5 # docx==0.2.4 #faiss==1.5.3 faiss_cpu==1.7.4 fire==0.4.0 -typer +typer==0.9.0 # godot==0.1.1 # google_api_python_client==2.93.0 # Used by search_engine.py lancedb==0.4.0 diff --git a/tests/metagpt/management/test_skill_manager.py b/tests/metagpt/management/test_skill_manager.py index 27bed8f64..489aea82b 100644 --- a/tests/metagpt/management/test_skill_manager.py +++ b/tests/metagpt/management/test_skill_manager.py @@ -14,9 +14,9 @@ def test_skill_manager(): manager = SkillManager() logger.info(manager._store) - write_prd = WritePRD() + write_prd = WritePRD(name="WritePRD") write_prd.desc = "基于老板或其他人的需求进行PRD的撰写,包括用户故事、需求分解等" - write_test = WriteTest() + write_test = WriteTest(name="WriteTest") write_test.desc = "进行测试用例的撰写" manager.add_skill(write_prd) manager.add_skill(write_test) @@ -24,7 +24,7 @@ def test_skill_manager(): skill = manager.get_skill("WriteTest") logger.info(skill) - rsp = manager.retrieve_skill("写PRD") + rsp = manager.retrieve_skill("WritePRD") logger.info(rsp) assert rsp[0] == "WritePRD"