添加一个代码

This commit is contained in:
ziming 2023-09-05 22:27:06 +08:00
commit 552547e662
58 changed files with 632 additions and 327 deletions

View file

@ -15,7 +15,7 @@ Skill = Action
class SkillManager:
"""用来管理所有技能"""
"""Used to manage all skills"""
def __init__(self):
self._llm=LLM.DEFAULT_LLM
@ -24,8 +24,8 @@ class SkillManager:
def add_skill(self, skill: Skill):
"""
增加技能将技能加入到技能池与可检索的存储中
:param skill: 技能
Add a skill, add the skill to the skill pool and searchable storage
:param skill: Skill
:return:
"""
self._skills[skill.name] = skill
@ -33,8 +33,8 @@ class SkillManager:
def del_skill(self, skill_name: str):
"""
删除技能将技能从技能池与可检索的存储中移除
:param skill_name: 技能名
Delete a skill, remove the skill from the skill pool and searchable storage
:param skill_name: Skill name
:return:
"""
self._skills.pop(skill_name)
@ -42,31 +42,31 @@ class SkillManager:
def get_skill(self, skill_name: str) -> Skill:
"""
通过技能名获得精确的技能
:param skill_name: 技能名
:return: 技能
Obtain a specific skill by skill name
:param skill_name: Skill name
:return: Skill
"""
return self._skills.get(skill_name)
def retrieve_skill(self, desc: str, n_results: int = 2) -> list[Skill]:
"""
通过检索引擎获得技能
:param desc: 技能描述
:return: 技能多个
Obtain skills through the search engine
:param desc: Skill description
:return: Multiple skills
"""
return self._store.search(desc, n_results=n_results)['ids'][0]
def retrieve_skill_scored(self, desc: str, n_results: int = 2) -> dict:
"""
通过检索引擎获得技能
:param desc: 技能描述
:return: 技能与分数组成的字典
Obtain skills through the search engine
:param desc: Skill description
:return: Dictionary consisting of skills and scores
"""
return self._store.search(desc, n_results=n_results)
def generate_skill_desc(self, skill: Skill) -> str:
"""
为每个技能生成对应的描述性文本
Generate descriptive text for each skill
:param skill:
:return:
"""