bravecode-cli 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 +387 -0
- package/dist/autoModel.d.ts +39 -0
- package/dist/autoModel.d.ts.map +1 -0
- package/dist/autoModel.js +127 -0
- package/dist/autoModel.js.map +1 -0
- package/dist/cli.cjs +41727 -0
- package/dist/cli.cjs.map +7 -0
- package/dist/cli.d.ts +2 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +459 -0
- package/dist/cli.js.map +1 -0
- package/dist/cli.mjs +41000 -0
- package/dist/cli.mjs.map +7 -0
- package/dist/core/agent/index.d.ts +31 -0
- package/dist/core/agent/index.d.ts.map +1 -0
- package/dist/core/agent/index.js +339 -0
- package/dist/core/agent/index.js.map +1 -0
- package/dist/core/config/index.d.ts +54 -0
- package/dist/core/config/index.d.ts.map +1 -0
- package/dist/core/config/index.js +138 -0
- package/dist/core/config/index.js.map +1 -0
- package/dist/core/context/index.d.ts +20 -0
- package/dist/core/context/index.d.ts.map +1 -0
- package/dist/core/context/index.js +70 -0
- package/dist/core/context/index.js.map +1 -0
- package/dist/core/mcp/index.d.ts +32 -0
- package/dist/core/mcp/index.d.ts.map +1 -0
- package/dist/core/mcp/index.js +160 -0
- package/dist/core/mcp/index.js.map +1 -0
- package/dist/core/permission/index.d.ts +24 -0
- package/dist/core/permission/index.d.ts.map +1 -0
- package/dist/core/permission/index.js +96 -0
- package/dist/core/permission/index.js.map +1 -0
- package/dist/core/plugin/index.d.ts +44 -0
- package/dist/core/plugin/index.d.ts.map +1 -0
- package/dist/core/plugin/index.js +139 -0
- package/dist/core/plugin/index.js.map +1 -0
- package/dist/core/session/index.d.ts +23 -0
- package/dist/core/session/index.d.ts.map +1 -0
- package/dist/core/session/index.js +51 -0
- package/dist/core/session/index.js.map +1 -0
- package/dist/core/share/index.d.ts +28 -0
- package/dist/core/share/index.d.ts.map +1 -0
- package/dist/core/share/index.js +100 -0
- package/dist/core/share/index.js.map +1 -0
- package/dist/core/snapshot/index.d.ts +30 -0
- package/dist/core/snapshot/index.d.ts.map +1 -0
- package/dist/core/snapshot/index.js +141 -0
- package/dist/core/snapshot/index.js.map +1 -0
- package/dist/core/tool/bash.d.ts +8 -0
- package/dist/core/tool/bash.d.ts.map +1 -0
- package/dist/core/tool/bash.js +29 -0
- package/dist/core/tool/bash.js.map +1 -0
- package/dist/core/tool/browser.d.ts +5 -0
- package/dist/core/tool/browser.d.ts.map +1 -0
- package/dist/core/tool/browser.js +119 -0
- package/dist/core/tool/browser.js.map +1 -0
- package/dist/core/tool/file.d.ts +5 -0
- package/dist/core/tool/file.d.ts.map +1 -0
- package/dist/core/tool/file.js +79 -0
- package/dist/core/tool/file.js.map +1 -0
- package/dist/core/tool/index.d.ts +21 -0
- package/dist/core/tool/index.d.ts.map +1 -0
- package/dist/core/tool/index.js +90 -0
- package/dist/core/tool/index.js.map +1 -0
- package/dist/core/tool/search.d.ts +4 -0
- package/dist/core/tool/search.d.ts.map +1 -0
- package/dist/core/tool/search.js +107 -0
- package/dist/core/tool/search.js.map +1 -0
- package/dist/core/tool/task.d.ts +9 -0
- package/dist/core/tool/task.d.ts.map +1 -0
- package/dist/core/tool/task.js +54 -0
- package/dist/core/tool/task.js.map +1 -0
- package/dist/core/tool/todowrite.d.ts +3 -0
- package/dist/core/tool/todowrite.d.ts.map +1 -0
- package/dist/core/tool/todowrite.js +75 -0
- package/dist/core/tool/todowrite.js.map +1 -0
- package/dist/core/tool/websearch.d.ts +3 -0
- package/dist/core/tool/websearch.d.ts.map +1 -0
- package/dist/core/tool/websearch.js +46 -0
- package/dist/core/tool/websearch.js.map +1 -0
- package/dist/core/tui/ChatTUI.d.ts +17 -0
- package/dist/core/tui/ChatTUI.d.ts.map +1 -0
- package/dist/core/tui/ChatTUI.js +64 -0
- package/dist/core/tui/ChatTUI.js.map +1 -0
- package/dist/core/tui/index.d.ts +15 -0
- package/dist/core/tui/index.d.ts.map +1 -0
- package/dist/core/tui/index.js +41 -0
- package/dist/core/tui/index.js.map +1 -0
- package/dist/core/update/index.d.ts +23 -0
- package/dist/core/update/index.d.ts.map +1 -0
- package/dist/core/update/index.js +115 -0
- package/dist/core/update/index.js.map +1 -0
- package/dist/index.d.ts +18 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +11 -0
- package/dist/index.js.map +1 -0
- package/dist/models.json +545 -0
- package/dist/provider.d.ts +43 -0
- package/dist/provider.d.ts.map +1 -0
- package/dist/provider.js +294 -0
- package/dist/provider.js.map +1 -0
- package/package.json +88 -0
|
@@ -0,0 +1,70 @@
|
|
|
1
|
+
export class ContextCompactor {
|
|
2
|
+
maxTokens;
|
|
3
|
+
summaryRatio;
|
|
4
|
+
constructor(maxTokens = 8000, summaryRatio = 0.3) {
|
|
5
|
+
this.maxTokens = maxTokens;
|
|
6
|
+
this.summaryRatio = summaryRatio;
|
|
7
|
+
}
|
|
8
|
+
estimateTokens(messages) {
|
|
9
|
+
let tokens = 0;
|
|
10
|
+
for (const msg of messages) {
|
|
11
|
+
const content = typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content);
|
|
12
|
+
tokens += Math.ceil(content.length / 4);
|
|
13
|
+
}
|
|
14
|
+
return tokens;
|
|
15
|
+
}
|
|
16
|
+
shouldCompact(messages) {
|
|
17
|
+
const estimated = this.estimateTokens(messages);
|
|
18
|
+
return estimated > this.maxTokens;
|
|
19
|
+
}
|
|
20
|
+
async compact(messages, provider) {
|
|
21
|
+
if (!this.shouldCompact(messages)) {
|
|
22
|
+
return {
|
|
23
|
+
summary: "",
|
|
24
|
+
originalTokens: this.estimateTokens(messages),
|
|
25
|
+
compactedTokens: this.estimateTokens(messages),
|
|
26
|
+
};
|
|
27
|
+
}
|
|
28
|
+
const originalTokens = this.estimateTokens(messages);
|
|
29
|
+
const systemMessages = messages.filter((m) => m.role === "system");
|
|
30
|
+
const conversationMessages = messages.filter((m) => m.role !== "system");
|
|
31
|
+
const cutoffIndex = Math.floor(conversationMessages.length * (1 - this.summaryRatio));
|
|
32
|
+
const toSummarize = conversationMessages.slice(0, cutoffIndex);
|
|
33
|
+
const toKeep = conversationMessages.slice(cutoffIndex);
|
|
34
|
+
const summaryPrompt = [
|
|
35
|
+
{ role: "user", content: `Summarize this conversation concisely, preserving key decisions and context:\n\n${toSummarize.map((m) => `${m.role}: ${typeof m.content === "string" ? m.content : "[tool result]"}`).join("\n\n")}` }
|
|
36
|
+
];
|
|
37
|
+
let summary;
|
|
38
|
+
try {
|
|
39
|
+
const result = await provider.chat({
|
|
40
|
+
model: "codestral-latest",
|
|
41
|
+
messages: summaryPrompt,
|
|
42
|
+
maxTokens: 500,
|
|
43
|
+
});
|
|
44
|
+
summary = result.text || "Conversation summary unavailable.";
|
|
45
|
+
}
|
|
46
|
+
catch {
|
|
47
|
+
summary = "Conversation summary unavailable.";
|
|
48
|
+
}
|
|
49
|
+
const compacted = [
|
|
50
|
+
...systemMessages,
|
|
51
|
+
{ role: "user", content: `[Previous conversation summary]: ${summary}` },
|
|
52
|
+
...toKeep,
|
|
53
|
+
];
|
|
54
|
+
const compactedTokens = this.estimateTokens(compacted);
|
|
55
|
+
return {
|
|
56
|
+
summary,
|
|
57
|
+
originalTokens,
|
|
58
|
+
compactedTokens,
|
|
59
|
+
};
|
|
60
|
+
}
|
|
61
|
+
getStatus(messages) {
|
|
62
|
+
const tokens = this.estimateTokens(messages);
|
|
63
|
+
return {
|
|
64
|
+
tokens,
|
|
65
|
+
shouldCompact: tokens > this.maxTokens,
|
|
66
|
+
percentage: Math.round((tokens / this.maxTokens) * 100),
|
|
67
|
+
};
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/context/index.ts"],"names":[],"mappings":"AAQA,MAAM,OAAO,gBAAgB;IACnB,SAAS,CAAS;IAClB,YAAY,CAAS;IAE7B,YAAY,YAAoB,IAAI,EAAE,eAAuB,GAAG;QAC9D,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;IAEO,cAAc,CAAC,QAAmB;QACxC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,MAAM,OAAO,GAAG,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC5F,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC1C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,aAAa,CAAC,QAAmB;QACvC,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAChD,OAAO,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAAmB,EAAE,QAAa;QAC9C,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC;YAClC,OAAO;gBACL,OAAO,EAAE,EAAE;gBACX,cAAc,EAAE,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC;gBAC7C,eAAe,EAAE,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC;aAC/C,CAAC;QACJ,CAAC;QAED,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAErD,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACnE,MAAM,oBAAoB,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAEzE,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QACtF,MAAM,WAAW,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,oBAAoB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAEvD,MAAM,aAAa,GAAG;YACpB,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,mFAAmF,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE;SAC1O,CAAC;QAEF,IAAI,OAAe,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC;gBACjC,KAAK,EAAE,kBAAkB;gBACzB,QAAQ,EAAE,aAAa;gBACvB,SAAS,EAAE,GAAG;aACf,CAAC,CAAC;YACH,OAAO,GAAG,MAAM,CAAC,IAAI,IAAI,mCAAmC,CAAC;QAC/D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,GAAG,mCAAmC,CAAC;QAChD,CAAC;QAED,MAAM,SAAS,GAAc;YAC3B,GAAG,cAAc;YACjB,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,oCAAoC,OAAO,EAAE,EAAE;YACjF,GAAG,MAAM;SACV,CAAC;QAEF,MAAM,eAAe,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAEvD,OAAO;YACL,OAAO;YACP,cAAc;YACd,eAAe;SAChB,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,QAAmB;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAC7C,OAAO;YACL,MAAM;YACN,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC,SAAS;YACtC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,GAAG,CAAC;SACxD,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
import type { Tool } from "../tool";
|
|
2
|
+
export interface MCPServerConfig {
|
|
3
|
+
id: string;
|
|
4
|
+
name: string;
|
|
5
|
+
command: string;
|
|
6
|
+
args?: string[];
|
|
7
|
+
env?: Record<string, string>;
|
|
8
|
+
enabled: boolean;
|
|
9
|
+
}
|
|
10
|
+
export interface MCPTool {
|
|
11
|
+
name: string;
|
|
12
|
+
description: string;
|
|
13
|
+
inputSchema: any;
|
|
14
|
+
}
|
|
15
|
+
export declare class MCPServer {
|
|
16
|
+
private config;
|
|
17
|
+
private process;
|
|
18
|
+
private tools;
|
|
19
|
+
private requestId;
|
|
20
|
+
private pendingRequests;
|
|
21
|
+
private initialized;
|
|
22
|
+
constructor(config: MCPServerConfig);
|
|
23
|
+
start(): Promise<void>;
|
|
24
|
+
private initialize;
|
|
25
|
+
private sendRequest;
|
|
26
|
+
private handleMessage;
|
|
27
|
+
callTool(toolName: string, args: any): Promise<string>;
|
|
28
|
+
getTools(): Tool[];
|
|
29
|
+
isRunning(): boolean;
|
|
30
|
+
stop(): Promise<void>;
|
|
31
|
+
}
|
|
32
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/core/mcp/index.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAEpC,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7B,OAAO,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,GAAG,CAAC;CAClB;AAED,qBAAa,SAAS;IACpB,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,OAAO,CAA6B;IAC5C,OAAO,CAAC,KAAK,CAAiB;IAC9B,OAAO,CAAC,SAAS,CAAK;IACtB,OAAO,CAAC,eAAe,CAAmE;IAC1F,OAAO,CAAC,WAAW,CAAS;gBAEhB,MAAM,EAAE,eAAe;IAI7B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;YAgDd,UAAU;IA0BxB,OAAO,CAAC,WAAW;IA4BnB,OAAO,CAAC,aAAa;IAqBf,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;IAwB5D,QAAQ,IAAI,IAAI,EAAE;IASlB,SAAS,IAAI,OAAO;IAId,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;CAO5B"}
|
|
@@ -0,0 +1,160 @@
|
|
|
1
|
+
import { spawn, ChildProcess } from "child_process";
|
|
2
|
+
import { existsSync } from "fs";
|
|
3
|
+
export class MCPServer {
|
|
4
|
+
config;
|
|
5
|
+
process = null;
|
|
6
|
+
tools = [];
|
|
7
|
+
requestId = 0;
|
|
8
|
+
pendingRequests = new Map();
|
|
9
|
+
initialized = false;
|
|
10
|
+
constructor(config) {
|
|
11
|
+
this.config = config;
|
|
12
|
+
}
|
|
13
|
+
async start() {
|
|
14
|
+
if (this.process) {
|
|
15
|
+
return;
|
|
16
|
+
}
|
|
17
|
+
const command = this.config.command;
|
|
18
|
+
const args = this.config.args || [];
|
|
19
|
+
if (!existsSync(command) && !command.includes("/") && !command.includes("\\")) {
|
|
20
|
+
try {
|
|
21
|
+
this.process = spawn("npx", ["-y", command, ...args], {
|
|
22
|
+
env: { ...process.env, ...this.config.env },
|
|
23
|
+
stdio: ["pipe", "pipe", "pipe"],
|
|
24
|
+
});
|
|
25
|
+
}
|
|
26
|
+
catch {
|
|
27
|
+
throw new Error(`Failed to start MCP server: ${command}`);
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
else {
|
|
31
|
+
this.process = spawn(command, args, {
|
|
32
|
+
env: { ...process.env, ...this.config.env },
|
|
33
|
+
stdio: ["pipe", "pipe", "pipe"],
|
|
34
|
+
});
|
|
35
|
+
}
|
|
36
|
+
this.process.on("error", (error) => {
|
|
37
|
+
console.error(`MCP server ${this.config.id} error:`, error.message);
|
|
38
|
+
});
|
|
39
|
+
this.process.on("exit", (code) => {
|
|
40
|
+
console.log(`MCP server ${this.config.id} exited with code ${code}`);
|
|
41
|
+
this.process = null;
|
|
42
|
+
this.initialized = false;
|
|
43
|
+
});
|
|
44
|
+
this.process.stdout?.on("data", (data) => {
|
|
45
|
+
this.handleMessage(data.toString());
|
|
46
|
+
});
|
|
47
|
+
this.process.stderr?.on("data", (data) => {
|
|
48
|
+
const msg = data.toString().trim();
|
|
49
|
+
if (msg) {
|
|
50
|
+
console.error(`MCP server ${this.config.id} stderr:`, msg);
|
|
51
|
+
}
|
|
52
|
+
});
|
|
53
|
+
await this.initialize();
|
|
54
|
+
}
|
|
55
|
+
async initialize() {
|
|
56
|
+
try {
|
|
57
|
+
const response = await this.sendRequest("initialize", {
|
|
58
|
+
protocolVersion: "2024-11-05",
|
|
59
|
+
capabilities: {
|
|
60
|
+
tools: {},
|
|
61
|
+
},
|
|
62
|
+
clientInfo: {
|
|
63
|
+
name: "bravecode-cli",
|
|
64
|
+
version: "0.1.0",
|
|
65
|
+
},
|
|
66
|
+
});
|
|
67
|
+
await this.sendRequest("notifications/initialized", {});
|
|
68
|
+
this.initialized = true;
|
|
69
|
+
const toolsResponse = await this.sendRequest("tools/list", {});
|
|
70
|
+
if (toolsResponse && toolsResponse.tools) {
|
|
71
|
+
this.tools = toolsResponse.tools;
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
catch (error) {
|
|
75
|
+
throw new Error(`Failed to initialize MCP server: ${error.message}`);
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
sendRequest(method, params) {
|
|
79
|
+
return new Promise((resolve, reject) => {
|
|
80
|
+
if (!this.process?.stdin) {
|
|
81
|
+
reject(new Error("MCP server not running"));
|
|
82
|
+
return;
|
|
83
|
+
}
|
|
84
|
+
const id = ++this.requestId;
|
|
85
|
+
this.pendingRequests.set(id, { resolve, reject });
|
|
86
|
+
const message = JSON.stringify({
|
|
87
|
+
jsonrpc: "2.0",
|
|
88
|
+
id,
|
|
89
|
+
method,
|
|
90
|
+
params,
|
|
91
|
+
});
|
|
92
|
+
this.process.stdin.write(`${message}\n`);
|
|
93
|
+
setTimeout(() => {
|
|
94
|
+
if (this.pendingRequests.has(id)) {
|
|
95
|
+
this.pendingRequests.delete(id);
|
|
96
|
+
reject(new Error(`Request timeout: ${method}`));
|
|
97
|
+
}
|
|
98
|
+
}, 30000);
|
|
99
|
+
});
|
|
100
|
+
}
|
|
101
|
+
handleMessage(data) {
|
|
102
|
+
const lines = data.split("\n").filter(Boolean);
|
|
103
|
+
for (const line of lines) {
|
|
104
|
+
try {
|
|
105
|
+
const message = JSON.parse(line);
|
|
106
|
+
if (message.id && this.pendingRequests.has(message.id)) {
|
|
107
|
+
const { resolve, reject } = this.pendingRequests.get(message.id);
|
|
108
|
+
this.pendingRequests.delete(message.id);
|
|
109
|
+
if (message.error) {
|
|
110
|
+
reject(new Error(message.error.message || "Unknown error"));
|
|
111
|
+
}
|
|
112
|
+
else {
|
|
113
|
+
resolve(message.result);
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
catch { }
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
async callTool(toolName, args) {
|
|
121
|
+
if (!this.initialized) {
|
|
122
|
+
throw new Error(`MCP server ${this.config.id} not initialized`);
|
|
123
|
+
}
|
|
124
|
+
try {
|
|
125
|
+
const response = await this.sendRequest("tools/call", {
|
|
126
|
+
name: toolName,
|
|
127
|
+
arguments: args,
|
|
128
|
+
});
|
|
129
|
+
if (response && response.content) {
|
|
130
|
+
return response.content
|
|
131
|
+
.filter((c) => c.type === "text")
|
|
132
|
+
.map((c) => c.text)
|
|
133
|
+
.join("\n");
|
|
134
|
+
}
|
|
135
|
+
return JSON.stringify(response);
|
|
136
|
+
}
|
|
137
|
+
catch (error) {
|
|
138
|
+
throw new Error(`MCP tool error: ${error.message}`);
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
getTools() {
|
|
142
|
+
return this.tools.map((mcpTool) => ({
|
|
143
|
+
name: `mcp_${this.config.id}_${mcpTool.name}`,
|
|
144
|
+
description: `[MCP: ${this.config.name}] ${mcpTool.description}`,
|
|
145
|
+
parameters: mcpTool.inputSchema,
|
|
146
|
+
execute: async (args) => this.callTool(mcpTool.name, args),
|
|
147
|
+
}));
|
|
148
|
+
}
|
|
149
|
+
isRunning() {
|
|
150
|
+
return this.process !== null && this.initialized;
|
|
151
|
+
}
|
|
152
|
+
async stop() {
|
|
153
|
+
if (this.process) {
|
|
154
|
+
this.process.kill();
|
|
155
|
+
this.process = null;
|
|
156
|
+
this.initialized = false;
|
|
157
|
+
}
|
|
158
|
+
}
|
|
159
|
+
}
|
|
160
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/mcp/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAkBhC,MAAM,OAAO,SAAS;IACZ,MAAM,CAAkB;IACxB,OAAO,GAAwB,IAAI,CAAC;IACpC,KAAK,GAAc,EAAE,CAAC;IACtB,SAAS,GAAG,CAAC,CAAC;IACd,eAAe,GAAyD,IAAI,GAAG,EAAE,CAAC;IAClF,WAAW,GAAG,KAAK,CAAC;IAE5B,YAAY,MAAuB;QACjC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;QACpC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;QAEpC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9E,IAAI,CAAC;gBACH,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,EAAE;oBACpD,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE;oBAC3C,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;iBAChC,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,IAAI,KAAK,CAAC,+BAA+B,OAAO,EAAE,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE;gBAClC,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE;gBAC3C,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;aAChC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YACjC,OAAO,CAAC,KAAK,CAAC,cAAc,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YAC/B,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,CAAC,MAAM,CAAC,EAAE,qBAAqB,IAAI,EAAE,CAAC,CAAC;YACrE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YACvC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YACvC,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;YACnC,IAAI,GAAG,EAAE,CAAC;gBACR,OAAO,CAAC,KAAK,CAAC,cAAc,IAAI,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE;gBACpD,eAAe,EAAE,YAAY;gBAC7B,YAAY,EAAE;oBACZ,KAAK,EAAE,EAAE;iBACV;gBACD,UAAU,EAAE;oBACV,IAAI,EAAE,eAAe;oBACrB,OAAO,EAAE,OAAO;iBACjB;aACF,CAAC,CAAC;YAEH,MAAM,IAAI,CAAC,WAAW,CAAC,2BAA2B,EAAE,EAAE,CAAC,CAAC;YAExD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YAExB,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;YAC/D,IAAI,aAAa,IAAI,aAAa,CAAC,KAAK,EAAE,CAAC;gBACzC,IAAI,CAAC,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC;YACnC,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,oCAAoC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,MAAc,EAAE,MAAW;QAC7C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;gBACzB,MAAM,CAAC,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC,CAAC;gBAC5C,OAAO;YACT,CAAC;YAED,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC;YAC5B,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAElD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;gBAC7B,OAAO,EAAE,KAAK;gBACd,EAAE;gBACF,MAAM;gBACN,MAAM;aACP,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,OAAO,IAAI,CAAC,CAAC;YAEzC,UAAU,CAAC,GAAG,EAAE;gBACd,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;oBACjC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBAChC,MAAM,CAAC,IAAI,KAAK,CAAC,oBAAoB,MAAM,EAAE,CAAC,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC,EAAE,KAAK,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,aAAa,CAAC,IAAY;QAChC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAE/C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAEjC,IAAI,OAAO,CAAC,EAAE,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;oBACvD,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAE,CAAC;oBAClE,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBAExC,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;wBAClB,MAAM,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,IAAI,eAAe,CAAC,CAAC,CAAC;oBAC9D,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;oBAC1B,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACZ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,IAAS;QACxC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,cAAc,IAAI,CAAC,MAAM,CAAC,EAAE,kBAAkB,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE;gBACpD,IAAI,EAAE,QAAQ;gBACd,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;YAEH,IAAI,QAAQ,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACjC,OAAO,QAAQ,CAAC,OAAO;qBACpB,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;qBACrC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;qBACvB,IAAI,CAAC,IAAI,CAAC,CAAC;YAChB,CAAC;YAED,OAAO,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,mBAAmB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YAClC,IAAI,EAAE,OAAO,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,OAAO,CAAC,IAAI,EAAE;YAC7C,WAAW,EAAE,SAAS,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,OAAO,CAAC,WAAW,EAAE;YAChE,UAAU,EAAE,OAAO,CAAC,WAAW;YAC/B,OAAO,EAAE,KAAK,EAAE,IAAS,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC;SAChE,CAAC,CAAC,CAAC;IACN,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,OAAO,KAAK,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YACpB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QAC3B,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
export type PermissionAction = "allow" | "ask" | "deny";
|
|
2
|
+
export interface PermissionConfig {
|
|
3
|
+
[key: string]: PermissionAction | {
|
|
4
|
+
[pattern: string]: PermissionAction;
|
|
5
|
+
};
|
|
6
|
+
}
|
|
7
|
+
export interface PermissionResult {
|
|
8
|
+
action: PermissionAction;
|
|
9
|
+
reason?: string;
|
|
10
|
+
}
|
|
11
|
+
export declare class PermissionSystem {
|
|
12
|
+
private globalPermissions;
|
|
13
|
+
private agentPermissions;
|
|
14
|
+
private sessionApprovals;
|
|
15
|
+
constructor(globalPermissions?: PermissionConfig);
|
|
16
|
+
setAgentPermissions(agentId: string, permissions: PermissionConfig): void;
|
|
17
|
+
checkPermission(toolName: string, input: any, agentId?: string): Promise<PermissionResult>;
|
|
18
|
+
private checkPatternMatch;
|
|
19
|
+
approveOnce(toolName: string, input: any): void;
|
|
20
|
+
approveAlways(toolName: string): void;
|
|
21
|
+
clearApprovals(): void;
|
|
22
|
+
}
|
|
23
|
+
export declare function createDefaultPermissions(): PermissionConfig;
|
|
24
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/core/permission/index.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,gBAAgB,GAAG,OAAO,GAAG,KAAK,GAAG,MAAM,CAAC;AAExD,MAAM,WAAW,gBAAgB;IAC/B,CAAC,GAAG,EAAE,MAAM,GAAG,gBAAgB,GAAG;QAAE,CAAC,OAAO,EAAE,MAAM,GAAG,gBAAgB,CAAA;KAAE,CAAC;CAC3E;AAED,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,gBAAgB,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,iBAAiB,CAAmB;IAC5C,OAAO,CAAC,gBAAgB,CAA4C;IACpE,OAAO,CAAC,gBAAgB,CAA6C;gBAEzD,iBAAiB,GAAE,gBAAqB;IAepD,mBAAmB,CAAC,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,gBAAgB,GAAG,IAAI;IAInE,eAAe,CACnB,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,GAAG,EACV,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,gBAAgB,CAAC;IA8B5B,OAAO,CAAC,iBAAiB;IAsBzB,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,IAAI;IAK/C,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAKrC,cAAc,IAAI,IAAI;CAGvB;AAED,wBAAgB,wBAAwB,IAAI,gBAAgB,CAyB3D"}
|
|
@@ -0,0 +1,96 @@
|
|
|
1
|
+
import { minimatch } from "minimatch";
|
|
2
|
+
export class PermissionSystem {
|
|
3
|
+
globalPermissions;
|
|
4
|
+
agentPermissions = new Map();
|
|
5
|
+
sessionApprovals = new Map();
|
|
6
|
+
constructor(globalPermissions = {}) {
|
|
7
|
+
this.globalPermissions = {
|
|
8
|
+
read: "allow",
|
|
9
|
+
glob: "allow",
|
|
10
|
+
grep: "allow",
|
|
11
|
+
edit: "ask",
|
|
12
|
+
bash: "ask",
|
|
13
|
+
write: "ask",
|
|
14
|
+
browser: "allow",
|
|
15
|
+
webfetch: "allow",
|
|
16
|
+
curl: "allow",
|
|
17
|
+
...globalPermissions,
|
|
18
|
+
};
|
|
19
|
+
}
|
|
20
|
+
setAgentPermissions(agentId, permissions) {
|
|
21
|
+
this.agentPermissions.set(agentId, permissions);
|
|
22
|
+
}
|
|
23
|
+
async checkPermission(toolName, input, agentId) {
|
|
24
|
+
const sessionKey = `${toolName}:${JSON.stringify(input)}`;
|
|
25
|
+
if (this.sessionApprovals.has(sessionKey)) {
|
|
26
|
+
return { action: "allow" };
|
|
27
|
+
}
|
|
28
|
+
let permissions = this.globalPermissions;
|
|
29
|
+
if (agentId && this.agentPermissions.has(agentId)) {
|
|
30
|
+
permissions = { ...permissions, ...this.agentPermissions.get(agentId) };
|
|
31
|
+
}
|
|
32
|
+
const toolPermission = permissions[toolName];
|
|
33
|
+
if (!toolPermission) {
|
|
34
|
+
return { action: "ask", reason: `No permission defined for ${toolName}` };
|
|
35
|
+
}
|
|
36
|
+
if (typeof toolPermission === "string") {
|
|
37
|
+
return { action: toolPermission };
|
|
38
|
+
}
|
|
39
|
+
if (typeof toolPermission === "object") {
|
|
40
|
+
return this.checkPatternMatch(toolPermission, toolName, input);
|
|
41
|
+
}
|
|
42
|
+
return { action: "ask" };
|
|
43
|
+
}
|
|
44
|
+
checkPatternMatch(patterns, toolName, input) {
|
|
45
|
+
const inputStr = JSON.stringify(input);
|
|
46
|
+
let lastMatch = null;
|
|
47
|
+
for (const [pattern, action] of Object.entries(patterns)) {
|
|
48
|
+
if (pattern === "*") {
|
|
49
|
+
lastMatch = action;
|
|
50
|
+
continue;
|
|
51
|
+
}
|
|
52
|
+
if (minimatch(inputStr, pattern) || minimatch(inputStr, `*${pattern}*`)) {
|
|
53
|
+
lastMatch = action;
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
return { action: lastMatch || "ask" };
|
|
57
|
+
}
|
|
58
|
+
approveOnce(toolName, input) {
|
|
59
|
+
const sessionKey = `${toolName}:${JSON.stringify(input)}`;
|
|
60
|
+
this.sessionApprovals.set(sessionKey, "once");
|
|
61
|
+
}
|
|
62
|
+
approveAlways(toolName) {
|
|
63
|
+
const pattern = `${toolName}:*`;
|
|
64
|
+
this.sessionApprovals.set(pattern, "always");
|
|
65
|
+
}
|
|
66
|
+
clearApprovals() {
|
|
67
|
+
this.sessionApprovals.clear();
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
export function createDefaultPermissions() {
|
|
71
|
+
return {
|
|
72
|
+
read: "allow",
|
|
73
|
+
glob: "allow",
|
|
74
|
+
grep: "allow",
|
|
75
|
+
webfetch: "allow",
|
|
76
|
+
curl: "allow",
|
|
77
|
+
browser: "allow",
|
|
78
|
+
edit: "ask",
|
|
79
|
+
write: "ask",
|
|
80
|
+
bash: {
|
|
81
|
+
"*": "ask",
|
|
82
|
+
"git status*": "allow",
|
|
83
|
+
"git log*": "allow",
|
|
84
|
+
"ls *": "allow",
|
|
85
|
+
"pwd": "allow",
|
|
86
|
+
"echo *": "allow",
|
|
87
|
+
"cat *": "allow",
|
|
88
|
+
"npm test*": "allow",
|
|
89
|
+
"npm run *": "allow",
|
|
90
|
+
"rm *": "deny",
|
|
91
|
+
"sudo *": "deny",
|
|
92
|
+
"chmod 777 *": "deny",
|
|
93
|
+
},
|
|
94
|
+
};
|
|
95
|
+
}
|
|
96
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/permission/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAatC,MAAM,OAAO,gBAAgB;IACnB,iBAAiB,CAAmB;IACpC,gBAAgB,GAAkC,IAAI,GAAG,EAAE,CAAC;IAC5D,gBAAgB,GAAmC,IAAI,GAAG,EAAE,CAAC;IAErE,YAAY,oBAAsC,EAAE;QAClD,IAAI,CAAC,iBAAiB,GAAG;YACvB,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,KAAK;YACZ,OAAO,EAAE,OAAO;YAChB,QAAQ,EAAE,OAAO;YACjB,IAAI,EAAE,OAAO;YACb,GAAG,iBAAiB;SACrB,CAAC;IACJ,CAAC;IAED,mBAAmB,CAAC,OAAe,EAAE,WAA6B;QAChE,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,QAAgB,EAChB,KAAU,EACV,OAAgB;QAEhB,MAAM,UAAU,GAAG,GAAG,QAAQ,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;QAE1D,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YAC1C,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;QAC7B,CAAC;QAED,IAAI,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC;QAEzC,IAAI,OAAO,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YAClD,WAAW,GAAG,EAAE,GAAG,WAAW,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QAC1E,CAAC;QAED,MAAM,cAAc,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;QAE7C,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,6BAA6B,QAAQ,EAAE,EAAE,CAAC;QAC5E,CAAC;QAED,IAAI,OAAO,cAAc,KAAK,QAAQ,EAAE,CAAC;YACvC,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;QACpC,CAAC;QAED,IAAI,OAAO,cAAc,KAAK,QAAQ,EAAE,CAAC;YACvC,OAAO,IAAI,CAAC,iBAAiB,CAAC,cAAc,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;QACjE,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;IAC3B,CAAC;IAEO,iBAAiB,CACvB,QAAiD,EACjD,QAAgB,EAChB,KAAU;QAEV,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,SAAS,GAA4B,IAAI,CAAC;QAE9C,KAAK,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YACzD,IAAI,OAAO,KAAK,GAAG,EAAE,CAAC;gBACpB,SAAS,GAAG,MAAM,CAAC;gBACnB,SAAS;YACX,CAAC;YAED,IAAI,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,IAAI,SAAS,CAAC,QAAQ,EAAE,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;gBACxE,SAAS,GAAG,MAAM,CAAC;YACrB,CAAC;QACH,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,SAAS,IAAI,KAAK,EAAE,CAAC;IACxC,CAAC;IAED,WAAW,CAAC,QAAgB,EAAE,KAAU;QACtC,MAAM,UAAU,GAAG,GAAG,QAAQ,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1D,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAChD,CAAC;IAED,aAAa,CAAC,QAAgB;QAC5B,MAAM,OAAO,GAAG,GAAG,QAAQ,IAAI,CAAC;QAChC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAED,cAAc;QACZ,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;IAChC,CAAC;CACF;AAED,MAAM,UAAU,wBAAwB;IACtC,OAAO;QACL,IAAI,EAAE,OAAO;QACb,IAAI,EAAE,OAAO;QACb,IAAI,EAAE,OAAO;QACb,QAAQ,EAAE,OAAO;QACjB,IAAI,EAAE,OAAO;QACb,OAAO,EAAE,OAAO;QAChB,IAAI,EAAE,KAAK;QACX,KAAK,EAAE,KAAK;QACZ,IAAI,EAAE;YACJ,GAAG,EAAE,KAAK;YACV,aAAa,EAAE,OAAO;YACtB,UAAU,EAAE,OAAO;YACnB,MAAM,EAAE,OAAO;YACf,KAAK,EAAE,OAAO;YACd,QAAQ,EAAE,OAAO;YACjB,OAAO,EAAE,OAAO;YAChB,WAAW,EAAE,OAAO;YACpB,WAAW,EAAE,OAAO;YACpB,MAAM,EAAE,MAAM;YACd,QAAQ,EAAE,MAAM;YAChB,aAAa,EAAE,MAAM;SACtB;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
import type { Tool } from "../tool";
|
|
2
|
+
export interface PluginConfig {
|
|
3
|
+
name: string;
|
|
4
|
+
version: string;
|
|
5
|
+
description: string;
|
|
6
|
+
author?: string;
|
|
7
|
+
tools?: PluginTool[];
|
|
8
|
+
hooks?: PluginHooks;
|
|
9
|
+
}
|
|
10
|
+
export interface PluginTool {
|
|
11
|
+
name: string;
|
|
12
|
+
description: string;
|
|
13
|
+
parameters: any;
|
|
14
|
+
execute: (args: any) => Promise<string>;
|
|
15
|
+
}
|
|
16
|
+
export interface PluginHooks {
|
|
17
|
+
beforeToolCall?: (toolName: string, args: any) => Promise<any>;
|
|
18
|
+
afterToolCall?: (toolName: string, args: any, result: string) => Promise<string>;
|
|
19
|
+
beforeAgentRun?: (agentId: string, message: string) => Promise<string>;
|
|
20
|
+
afterAgentRun?: (agentId: string, message: string, result: string) => Promise<string>;
|
|
21
|
+
}
|
|
22
|
+
export declare class Plugin {
|
|
23
|
+
config: PluginConfig;
|
|
24
|
+
private loaded;
|
|
25
|
+
constructor(config: PluginConfig);
|
|
26
|
+
load(): Promise<void>;
|
|
27
|
+
getTools(): Tool[];
|
|
28
|
+
getHooks(): PluginHooks;
|
|
29
|
+
isLoaded(): boolean;
|
|
30
|
+
}
|
|
31
|
+
export declare class PluginManager {
|
|
32
|
+
private plugins;
|
|
33
|
+
private pluginDirs;
|
|
34
|
+
constructor(pluginDirs?: string[]);
|
|
35
|
+
loadPlugins(): Promise<void>;
|
|
36
|
+
private loadPluginFromDir;
|
|
37
|
+
registerPlugin(plugin: Plugin): void;
|
|
38
|
+
getPlugin(name: string): Plugin | undefined;
|
|
39
|
+
getAllPlugins(): Plugin[];
|
|
40
|
+
getAllTools(): Tool[];
|
|
41
|
+
getAllHooks(): PluginHooks;
|
|
42
|
+
uninstallPlugin(name: string): Promise<void>;
|
|
43
|
+
}
|
|
44
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/core/plugin/index.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAEpC,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,UAAU,EAAE,CAAC;IACrB,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,GAAG,CAAC;IAChB,OAAO,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;CACzC;AAED,MAAM,WAAW,WAAW;IAC1B,cAAc,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAC/D,aAAa,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;IACjF,cAAc,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;IACvE,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;CACvF;AAED,qBAAa,MAAM;IACV,MAAM,EAAE,YAAY,CAAC;IAC5B,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,EAAE,YAAY;IAI1B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAK3B,QAAQ,IAAI,IAAI,EAAE;IAWlB,QAAQ,IAAI,WAAW;IAIvB,QAAQ,IAAI,OAAO;CAGpB;AAED,qBAAa,aAAa;IACxB,OAAO,CAAC,OAAO,CAAkC;IACjD,OAAO,CAAC,UAAU,CAAW;gBAEjB,UAAU,CAAC,EAAE,MAAM,EAAE;IAO3B,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;YAiBpB,iBAAiB;IAwB/B,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAIpC,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAI3C,aAAa,IAAI,MAAM,EAAE;IAIzB,WAAW,IAAI,IAAI,EAAE;IAQrB,WAAW,IAAI,WAAW;IA0CpB,eAAe,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAMnD"}
|
|
@@ -0,0 +1,139 @@
|
|
|
1
|
+
import { existsSync, readFileSync, readdirSync } from "fs";
|
|
2
|
+
import { join } from "path";
|
|
3
|
+
export class Plugin {
|
|
4
|
+
config;
|
|
5
|
+
loaded = false;
|
|
6
|
+
constructor(config) {
|
|
7
|
+
this.config = config;
|
|
8
|
+
}
|
|
9
|
+
async load() {
|
|
10
|
+
if (this.loaded)
|
|
11
|
+
return;
|
|
12
|
+
this.loaded = true;
|
|
13
|
+
}
|
|
14
|
+
getTools() {
|
|
15
|
+
if (!this.config.tools)
|
|
16
|
+
return [];
|
|
17
|
+
return this.config.tools.map((pluginTool) => ({
|
|
18
|
+
name: `plugin_${this.config.name}_${pluginTool.name}`,
|
|
19
|
+
description: `[Plugin: ${this.config.name}] ${pluginTool.description}`,
|
|
20
|
+
parameters: pluginTool.parameters,
|
|
21
|
+
execute: pluginTool.execute,
|
|
22
|
+
}));
|
|
23
|
+
}
|
|
24
|
+
getHooks() {
|
|
25
|
+
return this.config.hooks || {};
|
|
26
|
+
}
|
|
27
|
+
isLoaded() {
|
|
28
|
+
return this.loaded;
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
export class PluginManager {
|
|
32
|
+
plugins = new Map();
|
|
33
|
+
pluginDirs;
|
|
34
|
+
constructor(pluginDirs) {
|
|
35
|
+
this.pluginDirs = pluginDirs || [
|
|
36
|
+
join(process.cwd(), ".bravecode", "plugins"),
|
|
37
|
+
join(process.env.HOME || "~", ".bravecode", "plugins"),
|
|
38
|
+
];
|
|
39
|
+
}
|
|
40
|
+
async loadPlugins() {
|
|
41
|
+
for (const dir of this.pluginDirs) {
|
|
42
|
+
if (!existsSync(dir))
|
|
43
|
+
continue;
|
|
44
|
+
try {
|
|
45
|
+
const entries = readdirSync(dir, { withFileTypes: true });
|
|
46
|
+
for (const entry of entries) {
|
|
47
|
+
if (!entry.isDirectory())
|
|
48
|
+
continue;
|
|
49
|
+
const pluginPath = join(dir, entry.name);
|
|
50
|
+
await this.loadPluginFromDir(pluginPath);
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
catch { }
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
async loadPluginFromDir(pluginPath) {
|
|
57
|
+
const manifestPath = join(pluginPath, "plugin.json");
|
|
58
|
+
if (!existsSync(manifestPath))
|
|
59
|
+
return;
|
|
60
|
+
try {
|
|
61
|
+
const manifest = JSON.parse(readFileSync(manifestPath, "utf-8"));
|
|
62
|
+
const pluginFile = join(pluginPath, "index.js");
|
|
63
|
+
if (existsSync(pluginFile)) {
|
|
64
|
+
const pluginModule = await import(pluginFile);
|
|
65
|
+
const pluginConfig = {
|
|
66
|
+
...manifest,
|
|
67
|
+
tools: pluginModule.tools || manifest.tools,
|
|
68
|
+
hooks: pluginModule.hooks || manifest.hooks,
|
|
69
|
+
};
|
|
70
|
+
const plugin = new Plugin(pluginConfig);
|
|
71
|
+
await plugin.load();
|
|
72
|
+
this.plugins.set(manifest.name, plugin);
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
catch { }
|
|
76
|
+
}
|
|
77
|
+
registerPlugin(plugin) {
|
|
78
|
+
this.plugins.set(plugin.config.name, plugin);
|
|
79
|
+
}
|
|
80
|
+
getPlugin(name) {
|
|
81
|
+
return this.plugins.get(name);
|
|
82
|
+
}
|
|
83
|
+
getAllPlugins() {
|
|
84
|
+
return Array.from(this.plugins.values());
|
|
85
|
+
}
|
|
86
|
+
getAllTools() {
|
|
87
|
+
const tools = [];
|
|
88
|
+
for (const plugin of this.plugins.values()) {
|
|
89
|
+
tools.push(...plugin.getTools());
|
|
90
|
+
}
|
|
91
|
+
return tools;
|
|
92
|
+
}
|
|
93
|
+
getAllHooks() {
|
|
94
|
+
const combined = {};
|
|
95
|
+
for (const plugin of this.plugins.values()) {
|
|
96
|
+
const hooks = plugin.getHooks();
|
|
97
|
+
if (hooks.beforeToolCall) {
|
|
98
|
+
const existing = combined.beforeToolCall;
|
|
99
|
+
combined.beforeToolCall = async (toolName, args) => {
|
|
100
|
+
if (existing)
|
|
101
|
+
args = await existing(toolName, args);
|
|
102
|
+
return hooks.beforeToolCall(toolName, args);
|
|
103
|
+
};
|
|
104
|
+
}
|
|
105
|
+
if (hooks.afterToolCall) {
|
|
106
|
+
const existing = combined.afterToolCall;
|
|
107
|
+
combined.afterToolCall = async (toolName, args, result) => {
|
|
108
|
+
if (existing)
|
|
109
|
+
result = await existing(toolName, args, result);
|
|
110
|
+
return hooks.afterToolCall(toolName, args, result);
|
|
111
|
+
};
|
|
112
|
+
}
|
|
113
|
+
if (hooks.beforeAgentRun) {
|
|
114
|
+
const existing = combined.beforeAgentRun;
|
|
115
|
+
combined.beforeAgentRun = async (agentId, message) => {
|
|
116
|
+
if (existing)
|
|
117
|
+
message = await existing(agentId, message);
|
|
118
|
+
return hooks.beforeAgentRun(agentId, message);
|
|
119
|
+
};
|
|
120
|
+
}
|
|
121
|
+
if (hooks.afterAgentRun) {
|
|
122
|
+
const existing = combined.afterAgentRun;
|
|
123
|
+
combined.afterAgentRun = async (agentId, message, result) => {
|
|
124
|
+
if (existing)
|
|
125
|
+
result = await existing(agentId, message, result);
|
|
126
|
+
return hooks.afterAgentRun(agentId, message, result);
|
|
127
|
+
};
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
return combined;
|
|
131
|
+
}
|
|
132
|
+
async uninstallPlugin(name) {
|
|
133
|
+
const plugin = this.plugins.get(name);
|
|
134
|
+
if (plugin) {
|
|
135
|
+
this.plugins.delete(name);
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
}
|
|
139
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/plugin/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,IAAI,CAAC;AAC3D,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AA0B5B,MAAM,OAAO,MAAM;IACV,MAAM,CAAe;IACpB,MAAM,GAAG,KAAK,CAAC;IAEvB,YAAY,MAAoB;QAC9B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAED,QAAQ;QACN,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK;YAAE,OAAO,EAAE,CAAC;QAElC,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YAC5C,IAAI,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,UAAU,CAAC,IAAI,EAAE;YACrD,WAAW,EAAE,YAAY,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,UAAU,CAAC,WAAW,EAAE;YACtE,UAAU,EAAE,UAAU,CAAC,UAAU;YACjC,OAAO,EAAE,UAAU,CAAC,OAAO;SAC5B,CAAC,CAAC,CAAC;IACN,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;IACjC,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;CACF;AAED,MAAM,OAAO,aAAa;IAChB,OAAO,GAAwB,IAAI,GAAG,EAAE,CAAC;IACzC,UAAU,CAAW;IAE7B,YAAY,UAAqB;QAC/B,IAAI,CAAC,UAAU,GAAG,UAAU,IAAI;YAC9B,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,YAAY,EAAE,SAAS,CAAC;YAC5C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,GAAG,EAAE,YAAY,EAAE,SAAS,CAAC;SACvD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW;QACf,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAClC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS;YAE/B,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;gBAE1D,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;oBAC5B,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE;wBAAE,SAAS;oBAEnC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;oBACzC,MAAM,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;gBAC3C,CAAC;YACH,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,UAAkB;QAChD,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;QAErD,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC;YAAE,OAAO;QAEtC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC,CAAC;YAEjE,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YAChD,IAAI,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC3B,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;gBAC9C,MAAM,YAAY,GAAiB;oBACjC,GAAG,QAAQ;oBACX,KAAK,EAAE,YAAY,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK;oBAC3C,KAAK,EAAE,YAAY,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK;iBAC5C,CAAC;gBAEF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,YAAY,CAAC,CAAC;gBACxC,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBACpB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACZ,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC/C,CAAC;IAED,SAAS,CAAC,IAAY;QACpB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED,aAAa;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,WAAW;QACT,MAAM,KAAK,GAAW,EAAE,CAAC;QACzB,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QACnC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,WAAW;QACT,MAAM,QAAQ,GAAgB,EAAE,CAAC;QAEjC,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;YAEhC,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;gBACzB,MAAM,QAAQ,GAAG,QAAQ,CAAC,cAAc,CAAC;gBACzC,QAAQ,CAAC,cAAc,GAAG,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE;oBACjD,IAAI,QAAQ;wBAAE,IAAI,GAAG,MAAM,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;oBACpD,OAAO,KAAK,CAAC,cAAe,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;gBAC/C,CAAC,CAAC;YACJ,CAAC;YAED,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;gBACxB,MAAM,QAAQ,GAAG,QAAQ,CAAC,aAAa,CAAC;gBACxC,QAAQ,CAAC,aAAa,GAAG,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;oBACxD,IAAI,QAAQ;wBAAE,MAAM,GAAG,MAAM,QAAQ,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;oBAC9D,OAAO,KAAK,CAAC,aAAc,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;gBACtD,CAAC,CAAC;YACJ,CAAC;YAED,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;gBACzB,MAAM,QAAQ,GAAG,QAAQ,CAAC,cAAc,CAAC;gBACzC,QAAQ,CAAC,cAAc,GAAG,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE;oBACnD,IAAI,QAAQ;wBAAE,OAAO,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;oBACzD,OAAO,KAAK,CAAC,cAAe,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBACjD,CAAC,CAAC;YACJ,CAAC;YAED,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;gBACxB,MAAM,QAAQ,GAAG,QAAQ,CAAC,aAAa,CAAC;gBACxC,QAAQ,CAAC,aAAa,GAAG,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;oBAC1D,IAAI,QAAQ;wBAAE,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;oBAChE,OAAO,KAAK,CAAC,aAAc,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;gBACxD,CAAC,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,IAAY;QAChC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import { Agent } from "../agent";
|
|
2
|
+
import { LLMProviderManager } from "../../provider";
|
|
3
|
+
import { ToolRegistry } from "../tool";
|
|
4
|
+
export interface SessionConfig {
|
|
5
|
+
agentId?: string;
|
|
6
|
+
model?: string;
|
|
7
|
+
}
|
|
8
|
+
export declare class Session {
|
|
9
|
+
id: string;
|
|
10
|
+
title: string;
|
|
11
|
+
agentId: string;
|
|
12
|
+
createdAt: Date;
|
|
13
|
+
updatedAt: Date;
|
|
14
|
+
private agent;
|
|
15
|
+
private provider;
|
|
16
|
+
private tools;
|
|
17
|
+
constructor(provider: LLMProviderManager, tools: ToolRegistry, config?: SessionConfig);
|
|
18
|
+
prompt(message: string, onChunk?: (text: string) => void): Promise<string>;
|
|
19
|
+
getAgent(): Agent;
|
|
20
|
+
switchAgent(agentId: string): void;
|
|
21
|
+
clear(): void;
|
|
22
|
+
}
|
|
23
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/core/session/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC;AAGjC,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAEvC,MAAM,WAAW,aAAa;IAC5B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,qBAAa,OAAO;IACX,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;IACvB,OAAO,CAAC,KAAK,CAAQ;IACrB,OAAO,CAAC,QAAQ,CAAqB;IACrC,OAAO,CAAC,KAAK,CAAe;gBAG1B,QAAQ,EAAE,kBAAkB,EAC5B,KAAK,EAAE,YAAY,EACnB,MAAM,GAAE,aAAkB;IAiBtB,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC;IAWhF,QAAQ,IAAI,KAAK;IAIjB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAQlC,KAAK,IAAI,IAAI;CAGd"}
|