mirror of
https://github.com/microsoft/autogen.git
synced 2025-10-30 17:29:47 +00:00
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
import os
|
|
|
|
from autogenstudio.datamodel import Skill
|
|
from autogenstudio.utils import utils
|
|
|
|
|
|
class TestUtilGetSkillsPrompt:
|
|
|
|
def test_get_skills_prompt(self):
|
|
|
|
skill_clazz = Skill(
|
|
name="skill_clazz",
|
|
description="skill_clazz",
|
|
user_id="guestuser@gmail.com",
|
|
libraries=["lib1.0", "lib1.1"],
|
|
content="I am the skill clazz content",
|
|
secrets=[{"secret": "secret_1", "value": "value_1"}],
|
|
agents=[],
|
|
)
|
|
|
|
skill_dict = Skill(
|
|
name="skill_dict",
|
|
description="skill_dict",
|
|
user_id="guestuser@gmail.com",
|
|
libraries=["lib2.0", "lib2.1"],
|
|
content="I am the skill dict content",
|
|
secrets=[{"secret": "secret_2", "value": "value_2"}],
|
|
agents=[],
|
|
)
|
|
|
|
skills = [skill_dict.__dict__, skill_clazz]
|
|
|
|
prompt = utils.get_skills_prompt(skills, work_dir="work_dir")
|
|
|
|
# test that prompt contains contents of skills class and dict
|
|
assert prompt.find(skill_clazz.content) > 0
|
|
assert prompt.find(skill_dict.content) > 0
|
|
|
|
# test that secrets are set in environ
|
|
assert os.getenv("secret_1") == "value_1"
|
|
assert os.getenv("secret_2") == "value_2"
|
|
|
|
# cleanup test work_dir
|
|
try:
|
|
os.system("rm -rf work_dir")
|
|
except Exception:
|
|
pass
|