mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-04-25 00:36:55 +02:00
fix skill manager
This commit is contained in:
parent
ee98f41131
commit
4e61062a5e
4 changed files with 7 additions and 7 deletions
|
|
@ -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):
|
||||
"""
|
||||
|
|
|
|||
|
|
@ -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"
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue