mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-09-21 12:50:18 +08:00
fix skill manager
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
@@ -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):
|
||||
"""
|
||||
|
||||
+2
-2
@@ -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
|
||||
|
||||
@@ -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"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user