mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-02-23 08:23:44 +08:00
243 lines
8.9 KiB
Python
243 lines
8.9 KiB
Python
"""
|
|
Skill loader for discovering and loading skills from directories.
|
|
"""
|
|
|
|
import os
|
|
from pathlib import Path
|
|
from typing import List, Optional, Dict
|
|
from common.log import logger
|
|
from agent.skills.types import Skill, SkillEntry, LoadSkillsResult, SkillMetadata
|
|
from agent.skills.frontmatter import parse_frontmatter, parse_metadata, parse_boolean_value, get_frontmatter_value
|
|
|
|
|
|
class SkillLoader:
|
|
"""Loads skills from various directories."""
|
|
|
|
def __init__(self, workspace_dir: Optional[str] = None):
|
|
"""
|
|
Initialize the skill loader.
|
|
|
|
:param workspace_dir: Agent workspace directory (for workspace-specific skills)
|
|
"""
|
|
self.workspace_dir = workspace_dir
|
|
|
|
def load_skills_from_dir(self, dir_path: str, source: str) -> LoadSkillsResult:
|
|
"""
|
|
Load skills from a directory.
|
|
|
|
Discovery rules:
|
|
- Direct .md files in the root directory
|
|
- Recursive SKILL.md files under subdirectories
|
|
|
|
:param dir_path: Directory path to scan
|
|
:param source: Source identifier (e.g., 'managed', 'workspace', 'bundled')
|
|
:return: LoadSkillsResult with skills and diagnostics
|
|
"""
|
|
skills = []
|
|
diagnostics = []
|
|
|
|
if not os.path.exists(dir_path):
|
|
diagnostics.append(f"Directory does not exist: {dir_path}")
|
|
return LoadSkillsResult(skills=skills, diagnostics=diagnostics)
|
|
|
|
if not os.path.isdir(dir_path):
|
|
diagnostics.append(f"Path is not a directory: {dir_path}")
|
|
return LoadSkillsResult(skills=skills, diagnostics=diagnostics)
|
|
|
|
# Load skills from root-level .md files and subdirectories
|
|
result = self._load_skills_recursive(dir_path, source, include_root_files=True)
|
|
|
|
return result
|
|
|
|
def _load_skills_recursive(
|
|
self,
|
|
dir_path: str,
|
|
source: str,
|
|
include_root_files: bool = False
|
|
) -> LoadSkillsResult:
|
|
"""
|
|
Recursively load skills from a directory.
|
|
|
|
:param dir_path: Directory to scan
|
|
:param source: Source identifier
|
|
:param include_root_files: Whether to include root-level .md files
|
|
:return: LoadSkillsResult
|
|
"""
|
|
skills = []
|
|
diagnostics = []
|
|
|
|
try:
|
|
entries = os.listdir(dir_path)
|
|
except Exception as e:
|
|
diagnostics.append(f"Failed to list directory {dir_path}: {e}")
|
|
return LoadSkillsResult(skills=skills, diagnostics=diagnostics)
|
|
|
|
for entry in entries:
|
|
# Skip hidden files and directories
|
|
if entry.startswith('.'):
|
|
continue
|
|
|
|
# Skip common non-skill directories
|
|
if entry in ('node_modules', '__pycache__', 'venv', '.git'):
|
|
continue
|
|
|
|
full_path = os.path.join(dir_path, entry)
|
|
|
|
# Handle directories
|
|
if os.path.isdir(full_path):
|
|
# Recursively scan subdirectories
|
|
sub_result = self._load_skills_recursive(full_path, source, include_root_files=False)
|
|
skills.extend(sub_result.skills)
|
|
diagnostics.extend(sub_result.diagnostics)
|
|
continue
|
|
|
|
# Handle files
|
|
if not os.path.isfile(full_path):
|
|
continue
|
|
|
|
# Check if this is a skill file
|
|
is_root_md = include_root_files and entry.endswith('.md')
|
|
is_skill_md = not include_root_files and entry == 'SKILL.md'
|
|
|
|
if not (is_root_md or is_skill_md):
|
|
continue
|
|
|
|
# Load the skill
|
|
skill_result = self._load_skill_from_file(full_path, source)
|
|
if skill_result.skills:
|
|
skills.extend(skill_result.skills)
|
|
diagnostics.extend(skill_result.diagnostics)
|
|
|
|
return LoadSkillsResult(skills=skills, diagnostics=diagnostics)
|
|
|
|
def _load_skill_from_file(self, file_path: str, source: str) -> LoadSkillsResult:
|
|
"""
|
|
Load a single skill from a markdown file.
|
|
|
|
:param file_path: Path to the skill markdown file
|
|
:param source: Source identifier
|
|
:return: LoadSkillsResult
|
|
"""
|
|
diagnostics = []
|
|
|
|
try:
|
|
with open(file_path, 'r', encoding='utf-8') as f:
|
|
content = f.read()
|
|
except Exception as e:
|
|
diagnostics.append(f"Failed to read skill file {file_path}: {e}")
|
|
return LoadSkillsResult(skills=[], diagnostics=diagnostics)
|
|
|
|
# Parse frontmatter
|
|
frontmatter = parse_frontmatter(content)
|
|
|
|
# Get skill name and description
|
|
skill_dir = os.path.dirname(file_path)
|
|
parent_dir_name = os.path.basename(skill_dir)
|
|
|
|
name = frontmatter.get('name', parent_dir_name)
|
|
description = frontmatter.get('description', '')
|
|
|
|
if not description or not description.strip():
|
|
diagnostics.append(f"Skill {name} has no description: {file_path}")
|
|
return LoadSkillsResult(skills=[], diagnostics=diagnostics)
|
|
|
|
# Parse disable-model-invocation flag
|
|
disable_model_invocation = parse_boolean_value(
|
|
get_frontmatter_value(frontmatter, 'disable-model-invocation'),
|
|
default=False
|
|
)
|
|
|
|
# Create skill object
|
|
skill = Skill(
|
|
name=name,
|
|
description=description,
|
|
file_path=file_path,
|
|
base_dir=skill_dir,
|
|
source=source,
|
|
content=content,
|
|
disable_model_invocation=disable_model_invocation,
|
|
frontmatter=frontmatter,
|
|
)
|
|
|
|
return LoadSkillsResult(skills=[skill], diagnostics=diagnostics)
|
|
|
|
def load_all_skills(
|
|
self,
|
|
managed_dir: Optional[str] = None,
|
|
workspace_skills_dir: Optional[str] = None,
|
|
extra_dirs: Optional[List[str]] = None,
|
|
) -> Dict[str, SkillEntry]:
|
|
"""
|
|
Load skills from all configured locations with precedence.
|
|
|
|
Precedence (lowest to highest):
|
|
1. Extra directories
|
|
2. Managed skills directory
|
|
3. Workspace skills directory
|
|
|
|
:param managed_dir: Managed skills directory (e.g., ~/.cow/skills)
|
|
:param workspace_skills_dir: Workspace skills directory (e.g., workspace/skills)
|
|
:param extra_dirs: Additional directories to load skills from
|
|
:return: Dictionary mapping skill name to SkillEntry
|
|
"""
|
|
skill_map: Dict[str, SkillEntry] = {}
|
|
all_diagnostics = []
|
|
|
|
# Load from extra directories (lowest precedence)
|
|
if extra_dirs:
|
|
for extra_dir in extra_dirs:
|
|
if not os.path.exists(extra_dir):
|
|
continue
|
|
result = self.load_skills_from_dir(extra_dir, source='extra')
|
|
all_diagnostics.extend(result.diagnostics)
|
|
for skill in result.skills:
|
|
entry = self._create_skill_entry(skill)
|
|
skill_map[skill.name] = entry
|
|
|
|
# Load from managed directory
|
|
if managed_dir and os.path.exists(managed_dir):
|
|
result = self.load_skills_from_dir(managed_dir, source='managed')
|
|
all_diagnostics.extend(result.diagnostics)
|
|
for skill in result.skills:
|
|
entry = self._create_skill_entry(skill)
|
|
skill_map[skill.name] = entry
|
|
|
|
# Load from workspace directory (highest precedence)
|
|
if workspace_skills_dir and os.path.exists(workspace_skills_dir):
|
|
result = self.load_skills_from_dir(workspace_skills_dir, source='workspace')
|
|
all_diagnostics.extend(result.diagnostics)
|
|
for skill in result.skills:
|
|
entry = self._create_skill_entry(skill)
|
|
skill_map[skill.name] = entry
|
|
|
|
# Log diagnostics
|
|
if all_diagnostics:
|
|
logger.debug(f"Skill loading diagnostics: {len(all_diagnostics)} issues")
|
|
for diag in all_diagnostics[:5]: # Log first 5
|
|
logger.debug(f" - {diag}")
|
|
|
|
logger.info(f"Loaded {len(skill_map)} skills from all sources")
|
|
|
|
return skill_map
|
|
|
|
def _create_skill_entry(self, skill: Skill) -> SkillEntry:
|
|
"""
|
|
Create a SkillEntry from a Skill with parsed metadata.
|
|
|
|
:param skill: The skill to create an entry for
|
|
:return: SkillEntry with metadata
|
|
"""
|
|
metadata = parse_metadata(skill.frontmatter)
|
|
|
|
# Parse user-invocable flag
|
|
user_invocable = parse_boolean_value(
|
|
get_frontmatter_value(skill.frontmatter, 'user-invocable'),
|
|
default=True
|
|
)
|
|
|
|
return SkillEntry(
|
|
skill=skill,
|
|
metadata=metadata,
|
|
user_invocable=user_invocable,
|
|
)
|