tben
08d02e6033
A skill is a runbook: a name, a unique slug, a one-line description of when it applies, and the markdown body. A run only ever sees the slug and description (one load_skill tool carries them), and reads the whole body only when it calls the tool, so a long runbook costs nothing until needed. Skills are the third extension kind beside MCP servers and custom tools, on the same botExtensionsSql path, so per-bot, per-workspace and all-bots grants work the same way, and attachDefaultExtensions hands an every-bot skill to new bots. load_skill is a reserved tool name, so a custom tool cannot shadow it, and a built-in wins a clash. Managed on Settings > Skills; ExtensionRow links each kind to its own page and a skill never reports a failure, since nothing connects or runs.
67 lines
2.2 KiB
TypeScript
67 lines
2.2 KiB
TypeScript
import BunSQLite from "@moduletrace/bun-sqlite";
|
|
import attachToAllBots from "./attach-to-all-bots";
|
|
import type { BUN_SQLITE_COBBLE_SKILLS, BunSQLiteTables } from "@/db/types/db";
|
|
import type { SaveSkillRequest } from "@/src/schemas/extension-schema";
|
|
import AppError from "@/src/utils/app-error";
|
|
import grabSkill from "./grab-skill";
|
|
|
|
type Params = {
|
|
/**
|
|
* The super admin saving it (checked by the caller).
|
|
*/
|
|
user_id: number | string;
|
|
data: SaveSkillRequest;
|
|
};
|
|
|
|
/**
|
|
* Adds or edits a skill and returns its id. Slugs are unique: a bot loads a
|
|
* skill by its slug, so two skills can't answer to the same one.
|
|
*/
|
|
export default async function saveSkill({ user_id, data }: Params) {
|
|
const clash = (
|
|
await BunSQLite.sql<{ id: number }>({
|
|
sql: `SELECT id FROM skills WHERE slug = ? AND id != ?`,
|
|
values: [data.slug, data.skill_id || 0],
|
|
})
|
|
).payload?.[0];
|
|
|
|
if (clash) throw new AppError(`Another skill already uses the slug ${data.slug}.`);
|
|
|
|
const fields = {
|
|
name: data.name,
|
|
slug: data.slug,
|
|
description: data.description,
|
|
body: data.body,
|
|
enabled: data.enabled ? 1 : 0,
|
|
all_bots: data.all_bots ? 1 : 0,
|
|
} as const;
|
|
|
|
if (data.skill_id) {
|
|
const skill = await grabSkill({ skill_id: data.skill_id });
|
|
const res = await BunSQLite.update<BUN_SQLITE_COBBLE_SKILLS, (typeof BunSQLiteTables)[number]>({
|
|
table: "skills",
|
|
targetId: skill.id,
|
|
data: fields,
|
|
});
|
|
|
|
if (!res.success) throw new Error(`Couldn't save the skill: ${res.msg}`);
|
|
|
|
if (data.all_bots && !skill.all_bots) await attachToAllBots({ kind: "skill", id: skill.id });
|
|
|
|
return skill.id;
|
|
}
|
|
|
|
const res = await BunSQLite.insert<BUN_SQLITE_COBBLE_SKILLS, (typeof BunSQLiteTables)[number]>({
|
|
table: "skills",
|
|
data: [{ ...fields, user_id: Number(user_id) }],
|
|
});
|
|
|
|
const id = res.singleRes?.id || res.postInsertReturn?.insertId;
|
|
|
|
if (!res.success || !id) throw new Error(`Couldn't add the skill: ${res.msg}`);
|
|
|
|
if (data.all_bots) await attachToAllBots({ kind: "skill", id: Number(id) });
|
|
|
|
return Number(id);
|
|
}
|