seclaw-agent 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +668 -0
- package/SECURITY.md +253 -0
- package/assets/logo.png +0 -0
- package/dist/agent/context.d.ts +37 -0
- package/dist/agent/context.d.ts.map +1 -0
- package/dist/agent/context.js +211 -0
- package/dist/agent/context.js.map +1 -0
- package/dist/agent/docker_sandbox.d.ts +41 -0
- package/dist/agent/docker_sandbox.d.ts.map +1 -0
- package/dist/agent/docker_sandbox.js +239 -0
- package/dist/agent/docker_sandbox.js.map +1 -0
- package/dist/agent/loop.d.ts +86 -0
- package/dist/agent/loop.d.ts.map +1 -0
- package/dist/agent/loop.js +858 -0
- package/dist/agent/loop.js.map +1 -0
- package/dist/agent/memory.d.ts +21 -0
- package/dist/agent/memory.d.ts.map +1 -0
- package/dist/agent/memory.js +128 -0
- package/dist/agent/memory.js.map +1 -0
- package/dist/agent/security/execution_audit.d.ts +17 -0
- package/dist/agent/security/execution_audit.d.ts.map +1 -0
- package/dist/agent/security/execution_audit.js +126 -0
- package/dist/agent/security/execution_audit.js.map +1 -0
- package/dist/agent/security/input_validation/entity.d.ts +57 -0
- package/dist/agent/security/input_validation/entity.d.ts.map +1 -0
- package/dist/agent/security/input_validation/entity.js +121 -0
- package/dist/agent/security/input_validation/entity.js.map +1 -0
- package/dist/agent/security/input_validation/index.d.ts +114 -0
- package/dist/agent/security/input_validation/index.d.ts.map +1 -0
- package/dist/agent/security/input_validation/index.js +971 -0
- package/dist/agent/security/input_validation/index.js.map +1 -0
- package/dist/agent/security/input_validation/lattice.d.ts +33 -0
- package/dist/agent/security/input_validation/lattice.d.ts.map +1 -0
- package/dist/agent/security/input_validation/lattice.js +61 -0
- package/dist/agent/security/input_validation/lattice.js.map +1 -0
- package/dist/agent/security/input_validation/program_graph.d.ts +51 -0
- package/dist/agent/security/input_validation/program_graph.d.ts.map +1 -0
- package/dist/agent/security/input_validation/program_graph.js +285 -0
- package/dist/agent/security/input_validation/program_graph.js.map +1 -0
- package/dist/agent/security/input_validation/security_policy.d.ts +29 -0
- package/dist/agent/security/input_validation/security_policy.d.ts.map +1 -0
- package/dist/agent/security/input_validation/security_policy.js +256 -0
- package/dist/agent/security/input_validation/security_policy.js.map +1 -0
- package/dist/agent/security/memory_audit.d.ts +14 -0
- package/dist/agent/security/memory_audit.d.ts.map +1 -0
- package/dist/agent/security/memory_audit.js +126 -0
- package/dist/agent/security/memory_audit.js.map +1 -0
- package/dist/agent/security/skill_audit.d.ts +15 -0
- package/dist/agent/security/skill_audit.d.ts.map +1 -0
- package/dist/agent/security/skill_audit.js +112 -0
- package/dist/agent/security/skill_audit.js.map +1 -0
- package/dist/agent/security/snapshot_and_rollback/base.d.ts +10 -0
- package/dist/agent/security/snapshot_and_rollback/base.d.ts.map +1 -0
- package/dist/agent/security/snapshot_and_rollback/base.js +10 -0
- package/dist/agent/security/snapshot_and_rollback/base.js.map +1 -0
- package/dist/agent/security/snapshot_and_rollback/docker_snapshot.d.ts +52 -0
- package/dist/agent/security/snapshot_and_rollback/docker_snapshot.d.ts.map +1 -0
- package/dist/agent/security/snapshot_and_rollback/docker_snapshot.js +358 -0
- package/dist/agent/security/snapshot_and_rollback/docker_snapshot.js.map +1 -0
- package/dist/agent/security/snapshot_and_rollback/index.d.ts +7 -0
- package/dist/agent/security/snapshot_and_rollback/index.d.ts.map +1 -0
- package/dist/agent/security/snapshot_and_rollback/index.js +450 -0
- package/dist/agent/security/snapshot_and_rollback/index.js.map +1 -0
- package/dist/agent/skills.d.ts +35 -0
- package/dist/agent/skills.d.ts.map +1 -0
- package/dist/agent/skills.js +235 -0
- package/dist/agent/skills.js.map +1 -0
- package/dist/agent/subagent.d.ts +39 -0
- package/dist/agent/subagent.d.ts.map +1 -0
- package/dist/agent/subagent.js +151 -0
- package/dist/agent/subagent.js.map +1 -0
- package/dist/agent/tools/base.d.ts +32 -0
- package/dist/agent/tools/base.d.ts.map +1 -0
- package/dist/agent/tools/base.js +91 -0
- package/dist/agent/tools/base.js.map +1 -0
- package/dist/agent/tools/cron.d.ts +46 -0
- package/dist/agent/tools/cron.d.ts.map +1 -0
- package/dist/agent/tools/cron.js +95 -0
- package/dist/agent/tools/cron.js.map +1 -0
- package/dist/agent/tools/filesystem.d.ts +102 -0
- package/dist/agent/tools/filesystem.d.ts.map +1 -0
- package/dist/agent/tools/filesystem.js +257 -0
- package/dist/agent/tools/filesystem.js.map +1 -0
- package/dist/agent/tools/message.d.ts +40 -0
- package/dist/agent/tools/message.d.ts.map +1 -0
- package/dist/agent/tools/message.js +55 -0
- package/dist/agent/tools/message.js.map +1 -0
- package/dist/agent/tools/registry.d.ts +16 -0
- package/dist/agent/tools/registry.d.ts.map +1 -0
- package/dist/agent/tools/registry.js +47 -0
- package/dist/agent/tools/registry.js.map +1 -0
- package/dist/agent/tools/shell.d.ts +40 -0
- package/dist/agent/tools/shell.d.ts.map +1 -0
- package/dist/agent/tools/shell.js +166 -0
- package/dist/agent/tools/shell.js.map +1 -0
- package/dist/agent/tools/spawn.d.ts +30 -0
- package/dist/agent/tools/spawn.d.ts.map +1 -0
- package/dist/agent/tools/spawn.js +50 -0
- package/dist/agent/tools/spawn.js.map +1 -0
- package/dist/agent/tools/web.d.ts +59 -0
- package/dist/agent/tools/web.d.ts.map +1 -0
- package/dist/agent/tools/web.js +167 -0
- package/dist/agent/tools/web.js.map +1 -0
- package/dist/bus/events.d.ts +31 -0
- package/dist/bus/events.d.ts.map +1 -0
- package/dist/bus/events.js +28 -0
- package/dist/bus/events.js.map +1 -0
- package/dist/bus/queue.d.ts +32 -0
- package/dist/bus/queue.d.ts.map +1 -0
- package/dist/bus/queue.js +104 -0
- package/dist/bus/queue.js.map +1 -0
- package/dist/channels/base.d.ts +25 -0
- package/dist/channels/base.d.ts.map +1 -0
- package/dist/channels/base.js +54 -0
- package/dist/channels/base.js.map +1 -0
- package/dist/channels/dingtalk.d.ts +31 -0
- package/dist/channels/dingtalk.d.ts.map +1 -0
- package/dist/channels/dingtalk.js +177 -0
- package/dist/channels/dingtalk.js.map +1 -0
- package/dist/channels/discord.d.ts +30 -0
- package/dist/channels/discord.d.ts.map +1 -0
- package/dist/channels/discord.js +197 -0
- package/dist/channels/discord.js.map +1 -0
- package/dist/channels/email.d.ts +41 -0
- package/dist/channels/email.d.ts.map +1 -0
- package/dist/channels/email.js +210 -0
- package/dist/channels/email.js.map +1 -0
- package/dist/channels/feishu.d.ts +32 -0
- package/dist/channels/feishu.d.ts.map +1 -0
- package/dist/channels/feishu.js +109 -0
- package/dist/channels/feishu.js.map +1 -0
- package/dist/channels/manager.d.ts +24 -0
- package/dist/channels/manager.d.ts.map +1 -0
- package/dist/channels/manager.js +205 -0
- package/dist/channels/manager.js.map +1 -0
- package/dist/channels/mochat.d.ts +38 -0
- package/dist/channels/mochat.d.ts.map +1 -0
- package/dist/channels/mochat.js +201 -0
- package/dist/channels/mochat.js.map +1 -0
- package/dist/channels/qq.d.ts +40 -0
- package/dist/channels/qq.d.ts.map +1 -0
- package/dist/channels/qq.js +280 -0
- package/dist/channels/qq.js.map +1 -0
- package/dist/channels/slack.d.ts +27 -0
- package/dist/channels/slack.d.ts.map +1 -0
- package/dist/channels/slack.js +118 -0
- package/dist/channels/slack.js.map +1 -0
- package/dist/channels/telegram.d.ts +31 -0
- package/dist/channels/telegram.d.ts.map +1 -0
- package/dist/channels/telegram.js +218 -0
- package/dist/channels/telegram.js.map +1 -0
- package/dist/channels/whatsapp.d.ts +29 -0
- package/dist/channels/whatsapp.d.ts.map +1 -0
- package/dist/channels/whatsapp.js +117 -0
- package/dist/channels/whatsapp.js.map +1 -0
- package/dist/cli/commands.d.ts +8 -0
- package/dist/cli/commands.d.ts.map +1 -0
- package/dist/cli/commands.js +537 -0
- package/dist/cli/commands.js.map +1 -0
- package/dist/config/loader.d.ts +24 -0
- package/dist/config/loader.d.ts.map +1 -0
- package/dist/config/loader.js +182 -0
- package/dist/config/loader.js.map +1 -0
- package/dist/config/schema.d.ts +2921 -0
- package/dist/config/schema.d.ts.map +1 -0
- package/dist/config/schema.js +257 -0
- package/dist/config/schema.js.map +1 -0
- package/dist/cron/service.d.ts +38 -0
- package/dist/cron/service.d.ts.map +1 -0
- package/dist/cron/service.js +336 -0
- package/dist/cron/service.js.map +1 -0
- package/dist/cron/types.d.ts +46 -0
- package/dist/cron/types.d.ts.map +1 -0
- package/dist/cron/types.js +6 -0
- package/dist/cron/types.js.map +1 -0
- package/dist/heartbeat/service.d.ts +26 -0
- package/dist/heartbeat/service.d.ts.map +1 -0
- package/dist/heartbeat/service.js +142 -0
- package/dist/heartbeat/service.js.map +1 -0
- package/dist/index.d.ts +7 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +14 -0
- package/dist/index.js.map +1 -0
- package/dist/providers/base.d.ts +38 -0
- package/dist/providers/base.d.ts.map +1 -0
- package/dist/providers/base.js +21 -0
- package/dist/providers/base.js.map +1 -0
- package/dist/providers/litellm_provider.d.ts +35 -0
- package/dist/providers/litellm_provider.d.ts.map +1 -0
- package/dist/providers/litellm_provider.js +205 -0
- package/dist/providers/litellm_provider.js.map +1 -0
- package/dist/providers/registry.d.ts +44 -0
- package/dist/providers/registry.d.ts.map +1 -0
- package/dist/providers/registry.js +252 -0
- package/dist/providers/registry.js.map +1 -0
- package/dist/providers/transcription.d.ts +10 -0
- package/dist/providers/transcription.d.ts.map +1 -0
- package/dist/providers/transcription.js +83 -0
- package/dist/providers/transcription.js.map +1 -0
- package/dist/session/manager.d.ts +35 -0
- package/dist/session/manager.d.ts.map +1 -0
- package/dist/session/manager.js +193 -0
- package/dist/session/manager.js.map +1 -0
- package/dist/utils/helpers.d.ts +15 -0
- package/dist/utils/helpers.d.ts.map +1 -0
- package/dist/utils/helpers.js +100 -0
- package/dist/utils/helpers.js.map +1 -0
- package/dist/utils/logger.d.ts +7 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +25 -0
- package/dist/utils/logger.js.map +1 -0
- package/package.json +58 -0
- package/templates/AGENTS.md +51 -0
- package/templates/HEARTBEAT.md +16 -0
- package/templates/SOUL.md +36 -0
- package/templates/TOOLS.md +150 -0
- package/templates/USER.md +17 -0
- package/templates/memory/MEMORY.md +23 -0
|
@@ -0,0 +1,235 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Skills loader
|
|
4
|
+
*/
|
|
5
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
6
|
+
if (k2 === undefined) k2 = k;
|
|
7
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
8
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
9
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
10
|
+
}
|
|
11
|
+
Object.defineProperty(o, k2, desc);
|
|
12
|
+
}) : (function(o, m, k, k2) {
|
|
13
|
+
if (k2 === undefined) k2 = k;
|
|
14
|
+
o[k2] = m[k];
|
|
15
|
+
}));
|
|
16
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
17
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
18
|
+
}) : function(o, v) {
|
|
19
|
+
o["default"] = v;
|
|
20
|
+
});
|
|
21
|
+
var __importStar = (this && this.__importStar) || (function () {
|
|
22
|
+
var ownKeys = function(o) {
|
|
23
|
+
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
24
|
+
var ar = [];
|
|
25
|
+
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
26
|
+
return ar;
|
|
27
|
+
};
|
|
28
|
+
return ownKeys(o);
|
|
29
|
+
};
|
|
30
|
+
return function (mod) {
|
|
31
|
+
if (mod && mod.__esModule) return mod;
|
|
32
|
+
var result = {};
|
|
33
|
+
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
34
|
+
__setModuleDefault(result, mod);
|
|
35
|
+
return result;
|
|
36
|
+
};
|
|
37
|
+
})();
|
|
38
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
39
|
+
exports.SkillsLoader = void 0;
|
|
40
|
+
const fs = __importStar(require("fs"));
|
|
41
|
+
const path = __importStar(require("path"));
|
|
42
|
+
const child_process_1 = require("child_process");
|
|
43
|
+
// Default builtin skills directory
|
|
44
|
+
const BUILTIN_SKILLS_DIR = path.resolve(__dirname, "../../skills");
|
|
45
|
+
function commandExists(cmd) {
|
|
46
|
+
try {
|
|
47
|
+
(0, child_process_1.execSync)(`which ${cmd}`, { stdio: "ignore" });
|
|
48
|
+
return true;
|
|
49
|
+
}
|
|
50
|
+
catch {
|
|
51
|
+
return false;
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
function escapeXml(s) {
|
|
55
|
+
return s.replace(/&/g, "&").replace(/</g, "<").replace(/>/g, ">");
|
|
56
|
+
}
|
|
57
|
+
class SkillsLoader {
|
|
58
|
+
workspace;
|
|
59
|
+
workspaceSkills;
|
|
60
|
+
builtinSkills;
|
|
61
|
+
containerWorkspace;
|
|
62
|
+
pathTranslator;
|
|
63
|
+
constructor(opts) {
|
|
64
|
+
this.workspace = opts.workspace;
|
|
65
|
+
this.workspaceSkills = path.join(opts.workspace, "skills");
|
|
66
|
+
this.builtinSkills = opts.builtinSkillsDir ?? BUILTIN_SKILLS_DIR;
|
|
67
|
+
this.containerWorkspace = opts.containerWorkspace;
|
|
68
|
+
this.pathTranslator = opts.pathTranslator;
|
|
69
|
+
}
|
|
70
|
+
toContainerPath(hostPath) {
|
|
71
|
+
if (this.pathTranslator)
|
|
72
|
+
return this.pathTranslator(hostPath);
|
|
73
|
+
if (!this.containerWorkspace)
|
|
74
|
+
return hostPath;
|
|
75
|
+
const hostWs = path.resolve(this.workspace);
|
|
76
|
+
const resolved = path.resolve(hostPath);
|
|
77
|
+
if (resolved === hostWs)
|
|
78
|
+
return this.containerWorkspace;
|
|
79
|
+
if (resolved.startsWith(hostWs + "/"))
|
|
80
|
+
return this.containerWorkspace + resolved.slice(hostWs.length);
|
|
81
|
+
return hostPath;
|
|
82
|
+
}
|
|
83
|
+
listSkills(filterUnavailable = true) {
|
|
84
|
+
const skills = [];
|
|
85
|
+
if (fs.existsSync(this.workspaceSkills)) {
|
|
86
|
+
for (const entry of fs.readdirSync(this.workspaceSkills)) {
|
|
87
|
+
const skillDir = path.join(this.workspaceSkills, entry);
|
|
88
|
+
const skillFile = path.join(skillDir, "SKILL.md");
|
|
89
|
+
if (fs.statSync(skillDir).isDirectory() && fs.existsSync(skillFile)) {
|
|
90
|
+
skills.push({ name: entry, path: skillFile, source: "workspace" });
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
if (this.builtinSkills && fs.existsSync(this.builtinSkills)) {
|
|
95
|
+
for (const entry of fs.readdirSync(this.builtinSkills)) {
|
|
96
|
+
const skillDir = path.join(this.builtinSkills, entry);
|
|
97
|
+
const skillFile = path.join(skillDir, "SKILL.md");
|
|
98
|
+
if (fs.statSync(skillDir).isDirectory() &&
|
|
99
|
+
fs.existsSync(skillFile) &&
|
|
100
|
+
!skills.some((s) => s.name === entry)) {
|
|
101
|
+
skills.push({ name: entry, path: skillFile, source: "builtin" });
|
|
102
|
+
}
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
if (filterUnavailable) {
|
|
106
|
+
return skills.filter((s) => this.checkRequirements(this.getSkillMeta(s.name)));
|
|
107
|
+
}
|
|
108
|
+
return skills;
|
|
109
|
+
}
|
|
110
|
+
loadSkill(name) {
|
|
111
|
+
const workspaceSkill = path.join(this.workspaceSkills, name, "SKILL.md");
|
|
112
|
+
if (fs.existsSync(workspaceSkill))
|
|
113
|
+
return fs.readFileSync(workspaceSkill, "utf-8");
|
|
114
|
+
if (this.builtinSkills) {
|
|
115
|
+
const builtinSkill = path.join(this.builtinSkills, name, "SKILL.md");
|
|
116
|
+
if (fs.existsSync(builtinSkill))
|
|
117
|
+
return fs.readFileSync(builtinSkill, "utf-8");
|
|
118
|
+
}
|
|
119
|
+
return null;
|
|
120
|
+
}
|
|
121
|
+
loadSkillsForContext(skillNames) {
|
|
122
|
+
const parts = [];
|
|
123
|
+
for (const name of skillNames) {
|
|
124
|
+
const content = this.loadSkill(name);
|
|
125
|
+
if (content) {
|
|
126
|
+
const stripped = this.stripFrontmatter(content);
|
|
127
|
+
parts.push(`### Skill: ${name}\n\n${stripped}`);
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
return parts.join("\n\n---\n\n");
|
|
131
|
+
}
|
|
132
|
+
buildSkillsSummary() {
|
|
133
|
+
const allSkills = this.listSkills(false);
|
|
134
|
+
if (allSkills.length === 0)
|
|
135
|
+
return "";
|
|
136
|
+
const lines = ["<skills>"];
|
|
137
|
+
for (const s of allSkills) {
|
|
138
|
+
const meta = this.getSkillMeta(s.name);
|
|
139
|
+
const available = this.checkRequirements(meta);
|
|
140
|
+
const description = escapeXml(this.getSkillDescription(s.name));
|
|
141
|
+
lines.push(` <skill available="${available}">`);
|
|
142
|
+
lines.push(` <name>${escapeXml(s.name)}</name>`);
|
|
143
|
+
lines.push(` <description>${description}</description>`);
|
|
144
|
+
lines.push(` <location>${this.toContainerPath(s.path)}</location>`);
|
|
145
|
+
if (!available) {
|
|
146
|
+
const missing = this.getMissingRequirements(meta);
|
|
147
|
+
if (missing)
|
|
148
|
+
lines.push(` <requires>${escapeXml(missing)}</requires>`);
|
|
149
|
+
}
|
|
150
|
+
lines.push(` </skill>`);
|
|
151
|
+
}
|
|
152
|
+
lines.push("</skills>");
|
|
153
|
+
return lines.join("\n");
|
|
154
|
+
}
|
|
155
|
+
getAlwaysSkills() {
|
|
156
|
+
return this.listSkills(true)
|
|
157
|
+
.filter((s) => {
|
|
158
|
+
const meta = this.getSkillMetadata(s.name);
|
|
159
|
+
const skillMeta = this.parseNanobotMetadata(meta?.["metadata"] ?? "");
|
|
160
|
+
return skillMeta["always"] || meta?.["always"];
|
|
161
|
+
})
|
|
162
|
+
.map((s) => s.name);
|
|
163
|
+
}
|
|
164
|
+
getSkillMetadata(name) {
|
|
165
|
+
const content = this.loadSkill(name);
|
|
166
|
+
if (!content)
|
|
167
|
+
return null;
|
|
168
|
+
if (content.startsWith("---")) {
|
|
169
|
+
const match = content.match(/^---\n([\s\S]*?)\n---/);
|
|
170
|
+
if (match) {
|
|
171
|
+
const metadata = {};
|
|
172
|
+
for (const line of match[1].split("\n")) {
|
|
173
|
+
const colon = line.indexOf(":");
|
|
174
|
+
if (colon > 0) {
|
|
175
|
+
metadata[line.slice(0, colon).trim()] = line.slice(colon + 1).trim().replace(/^['"]|['"]$/g, "");
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
return metadata;
|
|
179
|
+
}
|
|
180
|
+
}
|
|
181
|
+
return null;
|
|
182
|
+
}
|
|
183
|
+
getSkillMeta(name) {
|
|
184
|
+
const meta = this.getSkillMetadata(name) ?? {};
|
|
185
|
+
return this.parseNanobotMetadata(meta["metadata"] ?? "");
|
|
186
|
+
}
|
|
187
|
+
parseNanobotMetadata(raw) {
|
|
188
|
+
try {
|
|
189
|
+
const data = JSON.parse(raw);
|
|
190
|
+
return typeof data === "object" && data !== null ? (data["nanobot"] ?? {}) : {};
|
|
191
|
+
}
|
|
192
|
+
catch {
|
|
193
|
+
return {};
|
|
194
|
+
}
|
|
195
|
+
}
|
|
196
|
+
checkRequirements(skillMeta) {
|
|
197
|
+
const requires = (skillMeta["requires"] ?? {});
|
|
198
|
+
for (const bin of requires["bins"] ?? []) {
|
|
199
|
+
if (!commandExists(bin))
|
|
200
|
+
return false;
|
|
201
|
+
}
|
|
202
|
+
for (const env of requires["env"] ?? []) {
|
|
203
|
+
if (!process.env[env])
|
|
204
|
+
return false;
|
|
205
|
+
}
|
|
206
|
+
return true;
|
|
207
|
+
}
|
|
208
|
+
getMissingRequirements(skillMeta) {
|
|
209
|
+
const missing = [];
|
|
210
|
+
const requires = (skillMeta["requires"] ?? {});
|
|
211
|
+
for (const bin of requires["bins"] ?? []) {
|
|
212
|
+
if (!commandExists(bin))
|
|
213
|
+
missing.push(`CLI: ${bin}`);
|
|
214
|
+
}
|
|
215
|
+
for (const env of requires["env"] ?? []) {
|
|
216
|
+
if (!process.env[env])
|
|
217
|
+
missing.push(`ENV: ${env}`);
|
|
218
|
+
}
|
|
219
|
+
return missing.join(", ");
|
|
220
|
+
}
|
|
221
|
+
getSkillDescription(name) {
|
|
222
|
+
const meta = this.getSkillMetadata(name);
|
|
223
|
+
return meta?.["description"] ?? name;
|
|
224
|
+
}
|
|
225
|
+
stripFrontmatter(content) {
|
|
226
|
+
if (content.startsWith("---")) {
|
|
227
|
+
const match = content.match(/^---\n[\s\S]*?\n---\n/);
|
|
228
|
+
if (match)
|
|
229
|
+
return content.slice(match[0].length).trim();
|
|
230
|
+
}
|
|
231
|
+
return content;
|
|
232
|
+
}
|
|
233
|
+
}
|
|
234
|
+
exports.SkillsLoader = SkillsLoader;
|
|
235
|
+
//# sourceMappingURL=skills.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"skills.js","sourceRoot":"","sources":["../../src/agent/skills.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,uCAAyB;AACzB,2CAA6B;AAC7B,iDAAyC;AAEzC,mCAAmC;AACnC,MAAM,kBAAkB,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;AAEnE,SAAS,aAAa,CAAC,GAAW;IAChC,IAAI,CAAC;QACH,IAAA,wBAAQ,EAAC,SAAS,GAAG,EAAE,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC9C,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,CAAS;IAC1B,OAAO,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AAC9E,CAAC;AAQD,MAAa,YAAY;IACvB,SAAS,CAAS;IAClB,eAAe,CAAS;IACxB,aAAa,CAAS;IACtB,kBAAkB,CAAU;IACpB,cAAc,CAAyB;IAE/C,YAAY,IAKX;QACC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC3D,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,gBAAgB,IAAI,kBAAkB,CAAC;QACjE,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC;QAClD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC;IAC5C,CAAC;IAEO,eAAe,CAAC,QAAgB;QACtC,IAAI,IAAI,CAAC,cAAc;YAAE,OAAO,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAC9D,IAAI,CAAC,IAAI,CAAC,kBAAkB;YAAE,OAAO,QAAQ,CAAC;QAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,QAAQ,KAAK,MAAM;YAAE,OAAO,IAAI,CAAC,kBAAkB,CAAC;QACxD,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM,GAAG,GAAG,CAAC;YAAE,OAAO,IAAI,CAAC,kBAAkB,GAAG,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACtG,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,UAAU,CAAC,iBAAiB,GAAG,IAAI;QACjC,MAAM,MAAM,GAAgB,EAAE,CAAC;QAE/B,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;YACxC,KAAK,MAAM,KAAK,IAAI,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;gBACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;gBACxD,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;gBAClD,IAAI,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;oBACpE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;gBACrE,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC;YAC5D,KAAK,MAAM,KAAK,IAAI,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC;gBACvD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;gBACtD,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;gBAClD,IACE,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE;oBACnC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC;oBACxB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,EACrC,CAAC;oBACD,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;gBACnE,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,iBAAiB,EAAE,CAAC;YACtB,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjF,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,SAAS,CAAC,IAAY;QACpB,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;QACzE,IAAI,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC;YAAE,OAAO,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QACnF,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;YACrE,IAAI,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC;gBAAE,OAAO,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QACjF,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,oBAAoB,CAAC,UAAoB;QACvC,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;YAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACrC,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;gBAChD,KAAK,CAAC,IAAI,CAAC,cAAc,IAAI,OAAO,QAAQ,EAAE,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACnC,CAAC;IAED,kBAAkB;QAChB,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACzC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEtC,MAAM,KAAK,GAAa,CAAC,UAAU,CAAC,CAAC;QACrC,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;YAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACvC,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC/C,MAAM,WAAW,GAAG,SAAS,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YAChE,KAAK,CAAC,IAAI,CAAC,uBAAuB,SAAS,IAAI,CAAC,CAAC;YACjD,KAAK,CAAC,IAAI,CAAC,aAAa,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACpD,KAAK,CAAC,IAAI,CAAC,oBAAoB,WAAW,gBAAgB,CAAC,CAAC;YAC5D,KAAK,CAAC,IAAI,CAAC,iBAAiB,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACvE,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,OAAO,GAAG,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC;gBAClD,IAAI,OAAO;oBAAE,KAAK,CAAC,IAAI,CAAC,iBAAiB,SAAS,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAC5E,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC3B,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACxB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;aACzB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;YACZ,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,SAAS,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;YACtE,OAAO,SAAS,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC,QAAQ,CAAC,CAAC;QACjD,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAED,gBAAgB,CAAC,IAAY;QAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACrC,IAAI,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAC1B,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;YACrD,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,QAAQ,GAA2B,EAAE,CAAC;gBAC5C,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;oBACxC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;oBAChC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;wBACd,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;oBACnG,CAAC;gBACH,CAAC;gBACD,OAAO,QAAQ,CAAC;YAClB,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,YAAY,CAAC,IAAY;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAC/C,OAAO,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;IAC3D,CAAC;IAEO,oBAAoB,CAAC,GAAW;QACtC,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC7B,OAAO,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAClF,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,iBAAiB,CAAC,SAAkC;QAC1D,MAAM,QAAQ,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,CAA4B,CAAC;QAC1E,KAAK,MAAM,GAAG,IAAK,QAAQ,CAAC,MAAM,CAA0B,IAAI,EAAE,EAAE,CAAC;YACnE,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC;gBAAE,OAAO,KAAK,CAAC;QACxC,CAAC;QACD,KAAK,MAAM,GAAG,IAAK,QAAQ,CAAC,KAAK,CAA0B,IAAI,EAAE,EAAE,CAAC;YAClE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC;gBAAE,OAAO,KAAK,CAAC;QACtC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,sBAAsB,CAAC,SAAkC;QAC/D,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,CAA4B,CAAC;QAC1E,KAAK,MAAM,GAAG,IAAK,QAAQ,CAAC,MAAM,CAA0B,IAAI,EAAE,EAAE,CAAC;YACnE,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC;gBAAE,OAAO,CAAC,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC,CAAC;QACvD,CAAC;QACD,KAAK,MAAM,GAAG,IAAK,QAAQ,CAAC,KAAK,CAA0B,IAAI,EAAE,EAAE,CAAC;YAClE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC;gBAAE,OAAO,CAAC,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAEO,mBAAmB,CAAC,IAAY;QACtC,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACzC,OAAO,IAAI,EAAE,CAAC,aAAa,CAAC,IAAI,IAAI,CAAC;IACvC,CAAC;IAEO,gBAAgB,CAAC,OAAe;QACtC,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;YACrD,IAAI,KAAK;gBAAE,OAAO,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;QAC1D,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AA1LD,oCA0LC"}
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Subagent manager
|
|
3
|
+
*/
|
|
4
|
+
import { MessageBus } from "../bus/queue";
|
|
5
|
+
import type { LLMProvider } from "../providers/base";
|
|
6
|
+
export interface ExecToolConfig {
|
|
7
|
+
timeout?: number;
|
|
8
|
+
}
|
|
9
|
+
export interface SpawnOptions {
|
|
10
|
+
task: string;
|
|
11
|
+
label?: string;
|
|
12
|
+
originChannel?: string;
|
|
13
|
+
originChatId?: string;
|
|
14
|
+
}
|
|
15
|
+
export declare class SubagentManager {
|
|
16
|
+
private provider;
|
|
17
|
+
private workspace;
|
|
18
|
+
private bus;
|
|
19
|
+
private model;
|
|
20
|
+
private braveApiKey?;
|
|
21
|
+
private execConfig;
|
|
22
|
+
private restrictToWorkspace;
|
|
23
|
+
private runningTasks;
|
|
24
|
+
constructor(opts: {
|
|
25
|
+
provider: LLMProvider;
|
|
26
|
+
workspace: string;
|
|
27
|
+
bus: MessageBus;
|
|
28
|
+
model?: string;
|
|
29
|
+
braveApiKey?: string;
|
|
30
|
+
execConfig?: ExecToolConfig;
|
|
31
|
+
restrictToWorkspace?: boolean;
|
|
32
|
+
});
|
|
33
|
+
spawn(opts: SpawnOptions): Promise<string>;
|
|
34
|
+
private _runSubagent;
|
|
35
|
+
private _announceResult;
|
|
36
|
+
private _buildSubagentPrompt;
|
|
37
|
+
getRunningCount(): number;
|
|
38
|
+
}
|
|
39
|
+
//# sourceMappingURL=subagent.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"subagent.d.ts","sourceRoot":"","sources":["../../src/agent/subagent.ts"],"names":[],"mappings":"AAAA;;GAEG;AAKH,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAE1C,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAMrD,MAAM,WAAW,cAAc;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,eAAe;IAC1B,OAAO,CAAC,QAAQ,CAAc;IAC9B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,GAAG,CAAa;IACxB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,WAAW,CAAC,CAAS;IAC7B,OAAO,CAAC,UAAU,CAAiB;IACnC,OAAO,CAAC,mBAAmB,CAAU;IACrC,OAAO,CAAC,YAAY,CAAyC;gBAEjD,IAAI,EAAE;QAChB,QAAQ,EAAE,WAAW,CAAC;QACtB,SAAS,EAAE,MAAM,CAAC;QAClB,GAAG,EAAE,UAAU,CAAC;QAChB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,UAAU,CAAC,EAAE,cAAc,CAAC;QAC5B,mBAAmB,CAAC,EAAE,OAAO,CAAC;KAC/B;IAUK,KAAK,CAAC,IAAI,EAAE,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC;YAgBlC,YAAY;YAyEZ,eAAe;IAqB7B,OAAO,CAAC,oBAAoB;IA+B5B,eAAe,IAAI,MAAM;CAG1B"}
|
|
@@ -0,0 +1,151 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Subagent manager
|
|
4
|
+
*/
|
|
5
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
6
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
7
|
+
};
|
|
8
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
9
|
+
exports.SubagentManager = void 0;
|
|
10
|
+
const uuid_1 = require("uuid");
|
|
11
|
+
const logger_1 = __importDefault(require("../utils/logger"));
|
|
12
|
+
const events_1 = require("../bus/events");
|
|
13
|
+
const registry_1 = require("./tools/registry");
|
|
14
|
+
const filesystem_1 = require("./tools/filesystem");
|
|
15
|
+
const shell_1 = require("./tools/shell");
|
|
16
|
+
const web_1 = require("./tools/web");
|
|
17
|
+
class SubagentManager {
|
|
18
|
+
provider;
|
|
19
|
+
workspace;
|
|
20
|
+
bus;
|
|
21
|
+
model;
|
|
22
|
+
braveApiKey;
|
|
23
|
+
execConfig;
|
|
24
|
+
restrictToWorkspace;
|
|
25
|
+
runningTasks = new Map();
|
|
26
|
+
constructor(opts) {
|
|
27
|
+
this.provider = opts.provider;
|
|
28
|
+
this.workspace = opts.workspace;
|
|
29
|
+
this.bus = opts.bus;
|
|
30
|
+
this.model = opts.model ?? opts.provider.getDefaultModel();
|
|
31
|
+
this.braveApiKey = opts.braveApiKey;
|
|
32
|
+
this.execConfig = opts.execConfig ?? {};
|
|
33
|
+
this.restrictToWorkspace = opts.restrictToWorkspace ?? false;
|
|
34
|
+
}
|
|
35
|
+
async spawn(opts) {
|
|
36
|
+
const taskId = (0, uuid_1.v4)().slice(0, 8);
|
|
37
|
+
const { task, label, originChannel = "cli", originChatId = "direct" } = opts;
|
|
38
|
+
const displayLabel = label ?? (task.length > 30 ? task.slice(0, 30) + "..." : task);
|
|
39
|
+
const origin = { channel: originChannel, chatId: originChatId };
|
|
40
|
+
const taskPromise = this._runSubagent(taskId, task, displayLabel, origin).then(() => {
|
|
41
|
+
this.runningTasks.delete(taskId);
|
|
42
|
+
});
|
|
43
|
+
this.runningTasks.set(taskId, taskPromise);
|
|
44
|
+
logger_1.default.info(`Spawned subagent [${taskId}]: ${displayLabel}`);
|
|
45
|
+
return `Subagent [${displayLabel}] started (id: ${taskId}). I'll notify you when it completes.`;
|
|
46
|
+
}
|
|
47
|
+
async _runSubagent(taskId, task, label, origin) {
|
|
48
|
+
logger_1.default.info(`Subagent [${taskId}] starting task: ${label}`);
|
|
49
|
+
try {
|
|
50
|
+
const tools = new registry_1.ToolRegistry();
|
|
51
|
+
const allowedDir = this.restrictToWorkspace ? this.workspace : undefined;
|
|
52
|
+
tools.register(new filesystem_1.ReadFileTool({ allowedDir }));
|
|
53
|
+
tools.register(new filesystem_1.WriteFileTool({ allowedDir }));
|
|
54
|
+
tools.register(new filesystem_1.ListDirTool({ allowedDir }));
|
|
55
|
+
tools.register(new shell_1.ExecTool({
|
|
56
|
+
workingDir: this.workspace,
|
|
57
|
+
timeout: this.execConfig.timeout,
|
|
58
|
+
restrictToWorkspace: this.restrictToWorkspace,
|
|
59
|
+
}));
|
|
60
|
+
tools.register(new web_1.WebSearchTool({ apiKey: this.braveApiKey }));
|
|
61
|
+
tools.register(new web_1.WebFetchTool());
|
|
62
|
+
const systemPrompt = this._buildSubagentPrompt(task);
|
|
63
|
+
const messages = [
|
|
64
|
+
{ role: "system", content: systemPrompt },
|
|
65
|
+
{ role: "user", content: task },
|
|
66
|
+
];
|
|
67
|
+
let finalResult = null;
|
|
68
|
+
const maxIterations = 15;
|
|
69
|
+
for (let i = 0; i < maxIterations; i++) {
|
|
70
|
+
const response = await this.provider.chat(messages, { tools: tools.getDefinitions(), model: this.model });
|
|
71
|
+
if (response.toolCalls.length > 0) {
|
|
72
|
+
const toolCallDicts = response.toolCalls.map((tc) => ({
|
|
73
|
+
id: tc.id,
|
|
74
|
+
type: "function",
|
|
75
|
+
function: { name: tc.name, arguments: JSON.stringify(tc.arguments) },
|
|
76
|
+
}));
|
|
77
|
+
messages.push({
|
|
78
|
+
role: "assistant",
|
|
79
|
+
content: response.content ?? "",
|
|
80
|
+
tool_calls: toolCallDicts,
|
|
81
|
+
});
|
|
82
|
+
for (const tc of response.toolCalls) {
|
|
83
|
+
logger_1.default.debug(`Subagent [${taskId}] executing: ${tc.name}`);
|
|
84
|
+
const result = await tools.execute(tc.name, tc.arguments);
|
|
85
|
+
messages.push({ role: "tool", tool_call_id: tc.id, name: tc.name, content: result });
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
else {
|
|
89
|
+
finalResult = response.content ?? null;
|
|
90
|
+
break;
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
if (finalResult === null)
|
|
94
|
+
finalResult = "Task completed but no final response was generated.";
|
|
95
|
+
logger_1.default.info(`Subagent [${taskId}] completed successfully`);
|
|
96
|
+
await this._announceResult(taskId, label, task, finalResult, origin, "ok");
|
|
97
|
+
}
|
|
98
|
+
catch (e) {
|
|
99
|
+
const errMsg = `Error: ${String(e)}`;
|
|
100
|
+
logger_1.default.error(`Subagent [${taskId}] failed: ${e}`);
|
|
101
|
+
await this._announceResult(taskId, label, task, errMsg, origin, "error");
|
|
102
|
+
}
|
|
103
|
+
}
|
|
104
|
+
async _announceResult(taskId, label, task, result, origin, status) {
|
|
105
|
+
const statusText = status === "ok" ? "completed successfully" : "failed";
|
|
106
|
+
const content = `[Subagent '${label}' ${statusText}]\n\nTask: ${task}\n\nResult:\n${result}\n\nSummarize this naturally for the user. Keep it brief (1-2 sentences). Do not mention technical details like "subagent" or task IDs.`;
|
|
107
|
+
const msg = (0, events_1.makeInboundMessage)({
|
|
108
|
+
channel: "system",
|
|
109
|
+
senderId: "subagent",
|
|
110
|
+
chatId: `${origin.channel}:${origin.chatId}`,
|
|
111
|
+
content,
|
|
112
|
+
});
|
|
113
|
+
await this.bus.publishInbound(msg);
|
|
114
|
+
logger_1.default.debug(`Subagent [${taskId}] announced result to ${origin.channel}:${origin.chatId}`);
|
|
115
|
+
}
|
|
116
|
+
_buildSubagentPrompt(task) {
|
|
117
|
+
return `# Subagent
|
|
118
|
+
|
|
119
|
+
You are a subagent spawned by the main agent to complete a specific task.
|
|
120
|
+
|
|
121
|
+
## Your Task
|
|
122
|
+
${task}
|
|
123
|
+
|
|
124
|
+
## Rules
|
|
125
|
+
1. Stay focused - complete only the assigned task, nothing else
|
|
126
|
+
2. Your final response will be reported back to the main agent
|
|
127
|
+
3. Do not initiate conversations or take on side tasks
|
|
128
|
+
4. Be concise but informative in your findings
|
|
129
|
+
|
|
130
|
+
## What You Can Do
|
|
131
|
+
- Read and write files in the workspace
|
|
132
|
+
- Execute shell commands
|
|
133
|
+
- Search the web and fetch web pages
|
|
134
|
+
- Complete the task thoroughly
|
|
135
|
+
|
|
136
|
+
## What You Cannot Do
|
|
137
|
+
- Send messages directly to users (no message tool available)
|
|
138
|
+
- Spawn other subagents
|
|
139
|
+
- Access the main agent's conversation history
|
|
140
|
+
|
|
141
|
+
## Workspace
|
|
142
|
+
Your workspace is at: ${this.workspace}
|
|
143
|
+
|
|
144
|
+
When you have completed the task, provide a clear summary of your findings or actions.`;
|
|
145
|
+
}
|
|
146
|
+
getRunningCount() {
|
|
147
|
+
return this.runningTasks.size;
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
exports.SubagentManager = SubagentManager;
|
|
151
|
+
//# sourceMappingURL=subagent.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"subagent.js","sourceRoot":"","sources":["../../src/agent/subagent.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;;AAGH,+BAAoC;AACpC,6DAAqC;AAErC,0CAAmD;AAEnD,+CAAgD;AAChD,mDAA8E;AAC9E,yCAAyC;AACzC,qCAA0D;AAa1D,MAAa,eAAe;IAClB,QAAQ,CAAc;IACtB,SAAS,CAAS;IAClB,GAAG,CAAa;IAChB,KAAK,CAAS;IACd,WAAW,CAAU;IACrB,UAAU,CAAiB;IAC3B,mBAAmB,CAAU;IAC7B,YAAY,GAA+B,IAAI,GAAG,EAAE,CAAC;IAE7D,YAAY,IAQX;QACC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;QACpB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC;QAC3D,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACpC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;QACxC,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,mBAAmB,IAAI,KAAK,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAkB;QAC5B,MAAM,MAAM,GAAG,IAAA,SAAM,GAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACpC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa,GAAG,KAAK,EAAE,YAAY,GAAG,QAAQ,EAAE,GAAG,IAAI,CAAC;QAC7E,MAAM,YAAY,GAAG,KAAK,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAEpF,MAAM,MAAM,GAAG,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC;QAEhE,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE;YAClF,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAE3C,gBAAM,CAAC,IAAI,CAAC,qBAAqB,MAAM,MAAM,YAAY,EAAE,CAAC,CAAC;QAC7D,OAAO,aAAa,YAAY,kBAAkB,MAAM,uCAAuC,CAAC;IAClG,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,MAAc,EACd,IAAY,EACZ,KAAa,EACb,MAA2C;QAE3C,gBAAM,CAAC,IAAI,CAAC,aAAa,MAAM,oBAAoB,KAAK,EAAE,CAAC,CAAC;QAE5D,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,uBAAY,EAAE,CAAC;YACjC,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;YAEzE,KAAK,CAAC,QAAQ,CAAC,IAAI,yBAAY,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;YACjD,KAAK,CAAC,QAAQ,CAAC,IAAI,0BAAa,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;YAClD,KAAK,CAAC,QAAQ,CAAC,IAAI,wBAAW,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;YAChD,KAAK,CAAC,QAAQ,CACZ,IAAI,gBAAQ,CAAC;gBACX,UAAU,EAAE,IAAI,CAAC,SAAS;gBAC1B,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO;gBAChC,mBAAmB,EAAE,IAAI,CAAC,mBAAmB;aAC9C,CAAC,CACH,CAAC;YACF,KAAK,CAAC,QAAQ,CAAC,IAAI,mBAAa,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;YAChE,KAAK,CAAC,QAAQ,CAAC,IAAI,kBAAY,EAAE,CAAC,CAAC;YAEnC,MAAM,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;YACrD,MAAM,QAAQ,GAA8B;gBAC1C,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE;gBACzC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE;aAChC,CAAC;YAEF,IAAI,WAAW,GAAkB,IAAI,CAAC;YACtC,MAAM,aAAa,GAAG,EAAE,CAAC;YAEzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CACvC,QAAiD,EACjD,EAAE,KAAK,EAAE,KAAK,CAAC,cAAc,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CACrD,CAAC;gBAEF,IAAI,QAAQ,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAClC,MAAM,aAAa,GAAG,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;wBACpD,EAAE,EAAE,EAAE,CAAC,EAAE;wBACT,IAAI,EAAE,UAAU;wBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE;qBACrE,CAAC,CAAC,CAAC;oBACJ,QAAQ,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,EAAE;wBAC/B,UAAU,EAAE,aAAa;qBAC1B,CAAC,CAAC;oBAEH,KAAK,MAAM,EAAE,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;wBACpC,gBAAM,CAAC,KAAK,CAAC,aAAa,MAAM,gBAAgB,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;wBAC3D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC;wBAC1D,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;oBACvF,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,WAAW,GAAG,QAAQ,CAAC,OAAO,IAAI,IAAI,CAAC;oBACvC,MAAM;gBACR,CAAC;YACH,CAAC;YAED,IAAI,WAAW,KAAK,IAAI;gBAAE,WAAW,GAAG,qDAAqD,CAAC;YAC9F,gBAAM,CAAC,IAAI,CAAC,aAAa,MAAM,0BAA0B,CAAC,CAAC;YAC3D,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QAC7E,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,MAAM,GAAG,UAAU,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;YACrC,gBAAM,CAAC,KAAK,CAAC,aAAa,MAAM,aAAa,CAAC,EAAE,CAAC,CAAC;YAClD,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,MAAc,EACd,KAAa,EACb,IAAY,EACZ,MAAc,EACd,MAA2C,EAC3C,MAAc;QAEd,MAAM,UAAU,GAAG,MAAM,KAAK,IAAI,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC,CAAC,QAAQ,CAAC;QACzE,MAAM,OAAO,GAAG,cAAc,KAAK,KAAK,UAAU,cAAc,IAAI,gBAAgB,MAAM,yIAAyI,CAAC;QAEpO,MAAM,GAAG,GAAG,IAAA,2BAAkB,EAAC;YAC7B,OAAO,EAAE,QAAQ;YACjB,QAAQ,EAAE,UAAU;YACpB,MAAM,EAAE,GAAG,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,EAAE;YAC5C,OAAO;SACR,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QACnC,gBAAM,CAAC,KAAK,CAAC,aAAa,MAAM,yBAAyB,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IAC9F,CAAC;IAEO,oBAAoB,CAAC,IAAY;QACvC,OAAO;;;;;EAKT,IAAI;;;;;;;;;;;;;;;;;;;;wBAoBkB,IAAI,CAAC,SAAS;;uFAEiD,CAAC;IACtF,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;IAChC,CAAC;CACF;AA5KD,0CA4KC"}
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Base class for agent tools
|
|
3
|
+
*/
|
|
4
|
+
export type JSONSchemaType = "string" | "integer" | "number" | "boolean" | "array" | "object";
|
|
5
|
+
export interface JSONSchema {
|
|
6
|
+
type?: JSONSchemaType;
|
|
7
|
+
properties?: Record<string, JSONSchema>;
|
|
8
|
+
required?: string[];
|
|
9
|
+
items?: JSONSchema;
|
|
10
|
+
enum?: unknown[];
|
|
11
|
+
minimum?: number;
|
|
12
|
+
maximum?: number;
|
|
13
|
+
minLength?: number;
|
|
14
|
+
maxLength?: number;
|
|
15
|
+
description?: string;
|
|
16
|
+
default?: unknown;
|
|
17
|
+
[key: string]: unknown;
|
|
18
|
+
}
|
|
19
|
+
/**
|
|
20
|
+
* Abstract base class for agent tools.
|
|
21
|
+
*/
|
|
22
|
+
export declare abstract class Tool {
|
|
23
|
+
abstract get name(): string;
|
|
24
|
+
abstract get description(): string;
|
|
25
|
+
abstract get parameters(): JSONSchema;
|
|
26
|
+
abstract execute(params: Record<string, unknown>): Promise<string>;
|
|
27
|
+
validateParams(params: Record<string, unknown>): string[];
|
|
28
|
+
private _validate;
|
|
29
|
+
private _checkType;
|
|
30
|
+
toSchema(): Record<string, unknown>;
|
|
31
|
+
}
|
|
32
|
+
//# sourceMappingURL=base.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/agent/tools/base.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,MAAM,MAAM,cAAc,GAAG,QAAQ,GAAG,SAAS,GAAG,QAAQ,GAAG,SAAS,GAAG,OAAO,GAAG,QAAQ,CAAC;AAE9F,MAAM,WAAW,UAAU;IACzB,IAAI,CAAC,EAAE,cAAc,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACxC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,UAAU,CAAC;IACnB,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED;;GAEG;AACH,8BAAsB,IAAI;IACxB,QAAQ,KAAK,IAAI,IAAI,MAAM,CAAC;IAC5B,QAAQ,KAAK,WAAW,IAAI,MAAM,CAAC;IACnC,QAAQ,KAAK,UAAU,IAAI,UAAU,CAAC;IAEtC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAElE,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,EAAE;IAKzD,OAAO,CAAC,SAAS;IAwDjB,OAAO,CAAC,UAAU;IAiBlB,QAAQ,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;CAUpC"}
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Base class for agent tools
|
|
4
|
+
*/
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.Tool = void 0;
|
|
7
|
+
/**
|
|
8
|
+
* Abstract base class for agent tools.
|
|
9
|
+
*/
|
|
10
|
+
class Tool {
|
|
11
|
+
validateParams(params) {
|
|
12
|
+
const schema = this.parameters ?? {};
|
|
13
|
+
return this._validate(params, { ...schema, type: "object" }, "");
|
|
14
|
+
}
|
|
15
|
+
_validate(val, schema, path) {
|
|
16
|
+
const label = path || "parameter";
|
|
17
|
+
const errors = [];
|
|
18
|
+
if (schema.type) {
|
|
19
|
+
const typeOk = this._checkType(val, schema.type);
|
|
20
|
+
if (!typeOk) {
|
|
21
|
+
errors.push(`${label} should be ${schema.type}`);
|
|
22
|
+
return errors;
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
if (schema.enum !== undefined && !schema.enum.includes(val)) {
|
|
26
|
+
errors.push(`${label} must be one of ${JSON.stringify(schema.enum)}`);
|
|
27
|
+
}
|
|
28
|
+
if (schema.type === "integer" || schema.type === "number") {
|
|
29
|
+
const n = val;
|
|
30
|
+
if (schema.minimum !== undefined && n < schema.minimum)
|
|
31
|
+
errors.push(`${label} must be >= ${schema.minimum}`);
|
|
32
|
+
if (schema.maximum !== undefined && n > schema.maximum)
|
|
33
|
+
errors.push(`${label} must be <= ${schema.maximum}`);
|
|
34
|
+
}
|
|
35
|
+
if (schema.type === "string") {
|
|
36
|
+
const s = val;
|
|
37
|
+
if (schema.minLength !== undefined && s.length < schema.minLength)
|
|
38
|
+
errors.push(`${label} must be at least ${schema.minLength} chars`);
|
|
39
|
+
if (schema.maxLength !== undefined && s.length > schema.maxLength)
|
|
40
|
+
errors.push(`${label} must be at most ${schema.maxLength} chars`);
|
|
41
|
+
}
|
|
42
|
+
if (schema.type === "object" && typeof val === "object" && val !== null) {
|
|
43
|
+
const obj = val;
|
|
44
|
+
const props = schema.properties ?? {};
|
|
45
|
+
for (const k of schema.required ?? []) {
|
|
46
|
+
if (!(k in obj)) {
|
|
47
|
+
errors.push(`missing required ${path ? path + "." + k : k}`);
|
|
48
|
+
}
|
|
49
|
+
}
|
|
50
|
+
for (const [k, v] of Object.entries(obj)) {
|
|
51
|
+
if (k in props) {
|
|
52
|
+
errors.push(...this._validate(v, props[k], path ? `${path}.${k}` : k));
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
if (schema.type === "array" && schema.items && Array.isArray(val)) {
|
|
57
|
+
for (let i = 0; i < val.length; i++) {
|
|
58
|
+
errors.push(...this._validate(val[i], schema.items, path ? `${path}[${i}]` : `[${i}]`));
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
return errors;
|
|
62
|
+
}
|
|
63
|
+
_checkType(val, type) {
|
|
64
|
+
switch (type) {
|
|
65
|
+
case "string":
|
|
66
|
+
return typeof val === "string";
|
|
67
|
+
case "integer":
|
|
68
|
+
return typeof val === "number" && Number.isInteger(val);
|
|
69
|
+
case "number":
|
|
70
|
+
return typeof val === "number";
|
|
71
|
+
case "boolean":
|
|
72
|
+
return typeof val === "boolean";
|
|
73
|
+
case "array":
|
|
74
|
+
return Array.isArray(val);
|
|
75
|
+
case "object":
|
|
76
|
+
return typeof val === "object" && val !== null && !Array.isArray(val);
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
toSchema() {
|
|
80
|
+
return {
|
|
81
|
+
type: "function",
|
|
82
|
+
function: {
|
|
83
|
+
name: this.name,
|
|
84
|
+
description: this.description,
|
|
85
|
+
parameters: this.parameters,
|
|
86
|
+
},
|
|
87
|
+
};
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
exports.Tool = Tool;
|
|
91
|
+
//# sourceMappingURL=base.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../src/agent/tools/base.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAmBH;;GAEG;AACH,MAAsB,IAAI;IAOxB,cAAc,CAAC,MAA+B;QAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;QACrC,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,QAA0B,EAAE,EAAE,EAAE,CAAC,CAAC;IACrF,CAAC;IAEO,SAAS,CAAC,GAAY,EAAE,MAAkB,EAAE,IAAY;QAC9D,MAAM,KAAK,GAAG,IAAI,IAAI,WAAW,CAAC;QAClC,MAAM,MAAM,GAAa,EAAE,CAAC;QAE5B,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAChB,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YACjD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,cAAc,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;gBACjD,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAC5D,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,mBAAmB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxE,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1D,MAAM,CAAC,GAAG,GAAa,CAAC;YACxB,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,GAAG,MAAM,CAAC,OAAO;gBACpD,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,eAAe,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;YACvD,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,GAAG,MAAM,CAAC,OAAO;gBACpD,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,eAAe,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QACzD,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,MAAM,CAAC,GAAG,GAAa,CAAC;YACxB,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,GAAG,MAAM,CAAC,SAAS;gBAC/D,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,qBAAqB,MAAM,CAAC,SAAS,QAAQ,CAAC,CAAC;YACrE,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,GAAG,MAAM,CAAC,SAAS;gBAC/D,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,oBAAoB,MAAM,CAAC,SAAS,QAAQ,CAAC,CAAC;QACtE,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;YACxE,MAAM,GAAG,GAAG,GAA8B,CAAC;YAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC;YACtC,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,QAAQ,IAAI,EAAE,EAAE,CAAC;gBACtC,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;oBAChB,MAAM,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBAC/D,CAAC;YACH,CAAC;YACD,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;gBACzC,IAAI,CAAC,IAAI,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAE,EAAE,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC1E,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,IAAI,MAAM,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YAClE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpC,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1F,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,UAAU,CAAC,GAAY,EAAE,IAAoB;QACnD,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,QAAQ;gBACX,OAAO,OAAO,GAAG,KAAK,QAAQ,CAAC;YACjC,KAAK,SAAS;gBACZ,OAAO,OAAO,GAAG,KAAK,QAAQ,IAAI,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YAC1D,KAAK,QAAQ;gBACX,OAAO,OAAO,GAAG,KAAK,QAAQ,CAAC;YACjC,KAAK,SAAS;gBACZ,OAAO,OAAO,GAAG,KAAK,SAAS,CAAC;YAClC,KAAK,OAAO;gBACV,OAAO,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAC5B,KAAK,QAAQ;gBACX,OAAO,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;IAED,QAAQ;QACN,OAAO;YACL,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACR,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;aAC5B;SACF,CAAC;IACJ,CAAC;CACF;AA/FD,oBA+FC"}
|
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Cron tool
|
|
3
|
+
*/
|
|
4
|
+
import { Tool } from "./base";
|
|
5
|
+
import { CronService } from "../../cron/service";
|
|
6
|
+
export declare class CronTool extends Tool {
|
|
7
|
+
private cron;
|
|
8
|
+
private channel;
|
|
9
|
+
private chatId;
|
|
10
|
+
constructor(cronService: CronService);
|
|
11
|
+
setContext(channel: string, chatId: string): void;
|
|
12
|
+
get name(): string;
|
|
13
|
+
get description(): string;
|
|
14
|
+
get parameters(): {
|
|
15
|
+
type: "object";
|
|
16
|
+
properties: {
|
|
17
|
+
action: {
|
|
18
|
+
type: "string";
|
|
19
|
+
enum: string[];
|
|
20
|
+
description: string;
|
|
21
|
+
};
|
|
22
|
+
message: {
|
|
23
|
+
type: "string";
|
|
24
|
+
description: string;
|
|
25
|
+
};
|
|
26
|
+
every_seconds: {
|
|
27
|
+
type: "integer";
|
|
28
|
+
description: string;
|
|
29
|
+
};
|
|
30
|
+
cron_expr: {
|
|
31
|
+
type: "string";
|
|
32
|
+
description: string;
|
|
33
|
+
};
|
|
34
|
+
job_id: {
|
|
35
|
+
type: "string";
|
|
36
|
+
description: string;
|
|
37
|
+
};
|
|
38
|
+
};
|
|
39
|
+
required: string[];
|
|
40
|
+
};
|
|
41
|
+
execute(params: Record<string, unknown>): Promise<string>;
|
|
42
|
+
private _addJob;
|
|
43
|
+
private _listJobs;
|
|
44
|
+
private _removeJob;
|
|
45
|
+
}
|
|
46
|
+
//# sourceMappingURL=cron.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cron.d.ts","sourceRoot":"","sources":["../../../src/agent/tools/cron.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAGjD,qBAAa,QAAS,SAAQ,IAAI;IAChC,OAAO,CAAC,IAAI,CAAc;IAC1B,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,MAAM,CAAS;gBAEX,WAAW,EAAE,WAAW;IAOpC,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IAKjD,IAAI,IAAI,IAAI,MAAM,CAAmB;IACrC,IAAI,WAAW,IAAI,MAAM,CAAkF;IAC3G,IAAI,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;;MAgBb;IAEK,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAQ/D,OAAO,CAAC,OAAO;IA4Bf,OAAO,CAAC,SAAS;IAOjB,OAAO,CAAC,UAAU;CAKnB"}
|