Add 5 pi extensions: pi-subagents, pi-crew, rpiv-pi, pi-interactive-shell, pi-intercom
This commit is contained in:
79
extensions/pi-subagents/src/skill-loader.ts
Normal file
79
extensions/pi-subagents/src/skill-loader.ts
Normal file
@@ -0,0 +1,79 @@
|
||||
/**
|
||||
* skill-loader.ts — Preload specific skill files and inject their content into the system prompt.
|
||||
*
|
||||
* When skills is a string[], reads each named skill from .pi/skills/ or ~/.pi/skills/
|
||||
* and returns their content for injection into the agent's system prompt.
|
||||
*/
|
||||
|
||||
import { homedir } from "node:os";
|
||||
import { join } from "node:path";
|
||||
import { isUnsafeName, safeReadFile } from "./memory.js";
|
||||
|
||||
export interface PreloadedSkill {
|
||||
name: string;
|
||||
content: string;
|
||||
}
|
||||
|
||||
/**
|
||||
* Attempt to load named skills from project and global skill directories.
|
||||
* Looks for: <dir>/<name>.md, <dir>/<name>.txt, <dir>/<name>
|
||||
*
|
||||
* @param skillNames List of skill names to preload.
|
||||
* @param cwd Working directory for project-level skills.
|
||||
* @returns Array of loaded skills (missing skills are skipped with a warning comment).
|
||||
*/
|
||||
export function preloadSkills(skillNames: string[], cwd: string): PreloadedSkill[] {
|
||||
const results: PreloadedSkill[] = [];
|
||||
|
||||
for (const name of skillNames) {
|
||||
// Unlike memory (which throws on unsafe names because it's part of agent setup),
|
||||
// skills are optional — skip gracefully to avoid blocking agent startup.
|
||||
if (isUnsafeName(name)) {
|
||||
results.push({ name, content: `(Skill "${name}" skipped: name contains path traversal characters)` });
|
||||
continue;
|
||||
}
|
||||
const content = findAndReadSkill(name, cwd);
|
||||
if (content !== undefined) {
|
||||
results.push({ name, content });
|
||||
} else {
|
||||
// Include a note about missing skills so the agent knows it was requested but not found
|
||||
results.push({ name, content: `(Skill "${name}" not found in .pi/skills/ or ~/.pi/skills/)` });
|
||||
}
|
||||
}
|
||||
|
||||
return results;
|
||||
}
|
||||
|
||||
/**
|
||||
* Search for a skill file in project and global directories.
|
||||
* Project-level takes priority over global.
|
||||
*/
|
||||
function findAndReadSkill(name: string, cwd: string): string | undefined {
|
||||
const projectDir = join(cwd, ".pi", "skills");
|
||||
const globalDir = join(homedir(), ".pi", "skills");
|
||||
|
||||
// Try project first, then global
|
||||
for (const dir of [projectDir, globalDir]) {
|
||||
const content = tryReadSkillFile(dir, name);
|
||||
if (content !== undefined) return content;
|
||||
}
|
||||
|
||||
return undefined;
|
||||
}
|
||||
|
||||
/**
|
||||
* Try to read a skill file from a directory.
|
||||
* Tries extensions in order: .md, .txt, (no extension)
|
||||
*/
|
||||
function tryReadSkillFile(dir: string, name: string): string | undefined {
|
||||
const extensions = [".md", ".txt", ""];
|
||||
|
||||
for (const ext of extensions) {
|
||||
const path = join(dir, name + ext);
|
||||
// safeReadFile rejects symlinks to prevent reading arbitrary files
|
||||
const content = safeReadFile(path);
|
||||
if (content !== undefined) return content.trim();
|
||||
}
|
||||
|
||||
return undefined;
|
||||
}
|
||||
Reference in New Issue
Block a user