fixbug: skill_yaml_file_name

This commit is contained in:
莘权 马 2023-08-26 22:24:25 +08:00
parent 2574ecaecf
commit f33af9dbc9

View file

@ -32,9 +32,10 @@ class SkillsDeclaration(BaseModel):
class SkillLoader:
def __init__(self):
skill_file_name = Path(__file__).parent.parent.parent / ".well-known/skills.yaml"
with open(str(skill_file_name), 'r') as file:
def __init__(self, skill_yaml_file_name: Path = None):
if not skill_yaml_file_name:
skill_yaml_file_name = Path(__file__).parent.parent.parent / ".well-known/skills.yaml"
with open(str(skill_yaml_file_name), 'r') as file:
skills = yaml.safe_load(file)
self._skills = SkillsDeclaration(**skills)