mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-10-17 11:58:18 +00:00
11d72b7b42
Render available profiles as one line per profile ID and title, using delegation context with a description fallback. Preserve project-scoped discovery and normalize whitespace without changing profile identifiers. Document the catalog format and test scoped routing, fallback descriptions, and empty catalogs. All six catalog and example checks pass against the staged snapshot.
77 lines
3.1 KiB
Python
77 lines
3.1 KiB
Python
import json
|
|
import re
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from helpers import files, projects, skills, subagents
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
@pytest.mark.parametrize("project", [None, "demo"])
|
|
def test_profile_catalog_keeps_ids_titles_scope_and_routing(monkeypatch, project):
|
|
profiles = {
|
|
"custom-id": SimpleNamespace(
|
|
title="Custom Title", context="Review\ncode", description="Fallback"
|
|
),
|
|
"fallback": SimpleNamespace(title="Fallback", context=" \n", description="Find facts"),
|
|
}
|
|
scopes = []
|
|
monkeypatch.setattr(projects, "get_context_project_name", lambda _: project)
|
|
monkeypatch.setattr(
|
|
subagents, "get_available_agents_dict", lambda scope: scopes.append(scope) or profiles
|
|
)
|
|
agent = SimpleNamespace(context=object())
|
|
prompt = files.read_prompt_file(
|
|
"agent.system.tool.call_sub.md", _directories=[str(ROOT / "prompts")], _agent=agent
|
|
)
|
|
assert scopes == [project]
|
|
assert "available profiles:\n- custom-id (Custom Title): Review code\n- fallback (Fallback): Find facts" in prompt
|
|
profiles.clear()
|
|
prompt = files.read_prompt_file(
|
|
"agent.system.tool.call_sub.md", _directories=[str(ROOT / "prompts")], _agent=agent
|
|
)
|
|
assert "available profiles:" not in prompt
|
|
|
|
|
|
@pytest.mark.parametrize("path", [
|
|
"plugins/_code_execution/prompts/agent.system.tool.code_exe.md",
|
|
"plugins/_code_execution/prompts/agent.system.tool.input.md",
|
|
"agents/agent0/prompts/agent.system.tool.response.md",
|
|
])
|
|
def test_tool_examples_are_complete_valid_json(path):
|
|
examples = re.findall(r"~~~json\n(.*?)\n~~~", (ROOT / path).read_text(), re.S)
|
|
assert examples
|
|
for example in examples:
|
|
request = json.loads(example)
|
|
assert {"thoughts", "headline", "tool_name", "tool_args"} <= request.keys()
|
|
assert isinstance(request["tool_args"], dict)
|
|
|
|
|
|
def test_default_skill_descriptions_fit_preview_and_remain_searchable(monkeypatch):
|
|
paths = [*ROOT.glob("skills/*/SKILL.md"), *ROOT.glob("plugins/*/skills/*/SKILL.md")]
|
|
candidates = []
|
|
for path in paths:
|
|
if "_a0_connector" in path.parts: # Host-only skills are outside the default catalog.
|
|
continue
|
|
skill = skills.skill_from_markdown(path)
|
|
assert skill is not None, path
|
|
assert 0 < len(skill.description) <= 100, path
|
|
candidates.append(skill)
|
|
monkeypatch.setattr(skills, "list_skills", lambda *args, **kwargs: candidates)
|
|
for query, name in [
|
|
("publish plugin", "a0-contribute-plugin"),
|
|
("plugin not loading", "a0-debug-plugin"),
|
|
("create agent", "a0-create-agent"),
|
|
("form validation", "browser-form-workflows"),
|
|
("Chrome Web Store", "browser-extension-control"),
|
|
("extension permissions", "browser-extension-control"),
|
|
("connect local files", "setup-a0-cli"),
|
|
("Word document", "writer-documents"),
|
|
("Docker desktop terminal", "linux-desktop"),
|
|
("framework development", "a0-development"),
|
|
]:
|
|
assert name in [skill.name for skill in skills.search_skills(query, limit=3)], query
|