@ai-zen/cli 0.1.0-alpha.2
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/README.md +319 -0
- package/README.zh.md +319 -0
- package/dist/agent-creator.d.ts +17 -0
- package/dist/agent-creator.d.ts.map +1 -0
- package/dist/agent-creator.js +112 -0
- package/dist/agent-creator.js.map +1 -0
- package/dist/config-wizard.d.ts +6 -0
- package/dist/config-wizard.d.ts.map +1 -0
- package/dist/config-wizard.js +46 -0
- package/dist/config-wizard.js.map +1 -0
- package/dist/config.d.ts +68 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +111 -0
- package/dist/config.js.map +1 -0
- package/dist/conversation-commands/back.d.ts +3 -0
- package/dist/conversation-commands/back.d.ts.map +1 -0
- package/dist/conversation-commands/back.js +144 -0
- package/dist/conversation-commands/back.js.map +1 -0
- package/dist/conversation-commands/clear.d.ts +3 -0
- package/dist/conversation-commands/clear.d.ts.map +1 -0
- package/dist/conversation-commands/clear.js +8 -0
- package/dist/conversation-commands/clear.js.map +1 -0
- package/dist/conversation-commands/editor.d.ts +3 -0
- package/dist/conversation-commands/editor.d.ts.map +1 -0
- package/dist/conversation-commands/editor.js +15 -0
- package/dist/conversation-commands/editor.js.map +1 -0
- package/dist/conversation-commands/exit.d.ts +3 -0
- package/dist/conversation-commands/exit.d.ts.map +1 -0
- package/dist/conversation-commands/exit.js +39 -0
- package/dist/conversation-commands/exit.js.map +1 -0
- package/dist/conversation-commands/help.d.ts +3 -0
- package/dist/conversation-commands/help.d.ts.map +1 -0
- package/dist/conversation-commands/help.js +12 -0
- package/dist/conversation-commands/help.js.map +1 -0
- package/dist/conversation-commands/index.d.ts +15 -0
- package/dist/conversation-commands/index.d.ts.map +1 -0
- package/dist/conversation-commands/index.js +63 -0
- package/dist/conversation-commands/index.js.map +1 -0
- package/dist/conversation-commands/new.d.ts +3 -0
- package/dist/conversation-commands/new.d.ts.map +1 -0
- package/dist/conversation-commands/new.js +47 -0
- package/dist/conversation-commands/new.js.map +1 -0
- package/dist/conversation-commands/save.d.ts +3 -0
- package/dist/conversation-commands/save.d.ts.map +1 -0
- package/dist/conversation-commands/save.js +34 -0
- package/dist/conversation-commands/save.js.map +1 -0
- package/dist/conversation-runner.d.ts +17 -0
- package/dist/conversation-runner.d.ts.map +1 -0
- package/dist/conversation-runner.js +212 -0
- package/dist/conversation-runner.js.map +1 -0
- package/dist/delta-renderer.d.ts +33 -0
- package/dist/delta-renderer.d.ts.map +1 -0
- package/dist/delta-renderer.js +117 -0
- package/dist/delta-renderer.js.map +1 -0
- package/dist/format-time.d.ts +38 -0
- package/dist/format-time.d.ts.map +1 -0
- package/dist/format-time.js +66 -0
- package/dist/format-time.js.map +1 -0
- package/dist/hook.d.ts +3 -0
- package/dist/hook.d.ts.map +1 -0
- package/dist/hook.js +83 -0
- package/dist/hook.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +47 -0
- package/dist/index.js.map +1 -0
- package/dist/menus/agents.d.ts +5 -0
- package/dist/menus/agents.d.ts.map +1 -0
- package/dist/menus/agents.js +219 -0
- package/dist/menus/agents.js.map +1 -0
- package/dist/menus/common.d.ts +41 -0
- package/dist/menus/common.d.ts.map +1 -0
- package/dist/menus/common.js +104 -0
- package/dist/menus/common.js.map +1 -0
- package/dist/menus/config-display.d.ts +2 -0
- package/dist/menus/config-display.d.ts.map +1 -0
- package/dist/menus/config-display.js +69 -0
- package/dist/menus/config-display.js.map +1 -0
- package/dist/menus/config.d.ts +8 -0
- package/dist/menus/config.d.ts.map +1 -0
- package/dist/menus/config.js +407 -0
- package/dist/menus/config.js.map +1 -0
- package/dist/menus/conversations.d.ts +20 -0
- package/dist/menus/conversations.d.ts.map +1 -0
- package/dist/menus/conversations.js +74 -0
- package/dist/menus/conversations.js.map +1 -0
- package/dist/menus/index.d.ts +5 -0
- package/dist/menus/index.d.ts.map +1 -0
- package/dist/menus/index.js +63 -0
- package/dist/menus/index.js.map +1 -0
- package/dist/menus/start-conversation.d.ts +16 -0
- package/dist/menus/start-conversation.d.ts.map +1 -0
- package/dist/menus/start-conversation.js +176 -0
- package/dist/menus/start-conversation.js.map +1 -0
- package/dist/types.d.ts +20 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +8 -0
- package/dist/types.js.map +1 -0
- package/package.json +53 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"exit.js","sourceRoot":"","sources":["../../src/conversation-commands/exit.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,QAAQ,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAC7E,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAG7D,MAAM,gBAAgB,GAAG,IAAI,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;AACvE,MAAM,SAAS,GAAG,IAAI,eAAe,CAAC,UAAU,CAAC,CAAC;AAElD,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,GAAwB;IACvD,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;IACxB,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9B,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC;YAC/C;gBACE,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EAAE,cAAc;gBACvB,OAAO,EAAE,IAAI;aACd;SACF,CAAC,CAAC;QAEH,IAAI,cAAc,EAAE,CAAC;YACnB,IAAI,CAAC;gBACH,MAAM,EAAE,GAAG,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,WAAW,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC;gBAC1E,gBAAgB,CAAC,KAAK,CAAC;oBACrB,EAAE;oBACF,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,SAAS;oBACjC,OAAO,EAAE,GAAG,CAAC,OAAO;oBACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC,CAAC,CAAC;gBACH,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,cAAc,GAAG,CAAC,WAAW,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;gBACxE,SAAS,CAAC,MAAM,EAAE,CAAC;YACrB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,KAAK,IAAI,CAAC,CAAC,CAAC;YACnD,CAAC;QACH,CAAC;IACH,CAAC;IACD,GAAG,CAAC,OAAO,GAAG,KAAK,CAAC;AACtB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"help.d.ts","sourceRoot":"","sources":["../../src/conversation-commands/help.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAElD,wBAAsB,UAAU,CAAC,GAAG,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC,CASxE"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import chalk from "chalk";
|
|
2
|
+
export async function handleHelp(ctx) {
|
|
3
|
+
console.log(chalk.blue.bold("\n📖 可用命令:\n"));
|
|
4
|
+
console.log(` ${chalk.cyan("/exit")} ${chalk.gray("/quit")} 退出对话`);
|
|
5
|
+
console.log(` ${chalk.cyan("/save")} 保存当前对话`);
|
|
6
|
+
console.log(` ${chalk.cyan("/new")} 重置会话(清空历史)`);
|
|
7
|
+
console.log(` ${chalk.cyan("/back")} 撤回消息(可修改后重发)`);
|
|
8
|
+
console.log(` ${chalk.cyan("/editor")} 使用系统编辑器输入长消息`);
|
|
9
|
+
console.log(` ${chalk.cyan("/clear")} 清屏`);
|
|
10
|
+
console.log(` ${chalk.cyan("/help")} 显示此帮助\n`);
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=help.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"help.js","sourceRoot":"","sources":["../../src/conversation-commands/help.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAG1B,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,GAAwB;IACvD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACxE,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACzD,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;IAC/D,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;IAC/D,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;IACrD,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;AAC5D,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import { ConversationContext } from "../types.js";
|
|
2
|
+
/**
|
|
3
|
+
* 判断输入是否为命令(以 / 开头)
|
|
4
|
+
*/
|
|
5
|
+
export declare function isCommand(input: string): boolean;
|
|
6
|
+
/**
|
|
7
|
+
* 分发命令
|
|
8
|
+
* @returns true 表示是命令且已处理,false 表示不是命令
|
|
9
|
+
*/
|
|
10
|
+
export declare function dispatchCommand(ctx: ConversationContext): Promise<boolean>;
|
|
11
|
+
/**
|
|
12
|
+
* 获取所有命令名称列表(用于提示信息)
|
|
13
|
+
*/
|
|
14
|
+
export declare function getCommandNames(): string[];
|
|
15
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/conversation-commands/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAkB,MAAM,aAAa,CAAC;AAwBlE;;GAEG;AACH,wBAAgB,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAEhD;AAED;;;GAGG;AACH,wBAAsB,eAAe,CACnC,GAAG,EAAE,mBAAmB,GACvB,OAAO,CAAC,OAAO,CAAC,CAiBlB;AAED;;GAEG;AACH,wBAAgB,eAAe,IAAI,MAAM,EAAE,CAY1C"}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
import { handleExit } from "./exit.js";
|
|
2
|
+
import { handleSave } from "./save.js";
|
|
3
|
+
import { handleNew } from "./new.js";
|
|
4
|
+
import { handleBack } from "./back.js";
|
|
5
|
+
import { handleEditor } from "./editor.js";
|
|
6
|
+
import { handleClear } from "./clear.js";
|
|
7
|
+
import { handleHelp } from "./help.js";
|
|
8
|
+
/**
|
|
9
|
+
* 命令注册表
|
|
10
|
+
* key 为命令名称(不含 / 前缀),value 为处理函数
|
|
11
|
+
*/
|
|
12
|
+
const commands = {
|
|
13
|
+
exit: handleExit,
|
|
14
|
+
quit: handleExit,
|
|
15
|
+
save: handleSave,
|
|
16
|
+
new: handleNew,
|
|
17
|
+
back: handleBack,
|
|
18
|
+
editor: handleEditor,
|
|
19
|
+
clear: handleClear,
|
|
20
|
+
help: handleHelp,
|
|
21
|
+
};
|
|
22
|
+
/**
|
|
23
|
+
* 判断输入是否为命令(以 / 开头)
|
|
24
|
+
*/
|
|
25
|
+
export function isCommand(input) {
|
|
26
|
+
return input.startsWith("/");
|
|
27
|
+
}
|
|
28
|
+
/**
|
|
29
|
+
* 分发命令
|
|
30
|
+
* @returns true 表示是命令且已处理,false 表示不是命令
|
|
31
|
+
*/
|
|
32
|
+
export async function dispatchCommand(ctx) {
|
|
33
|
+
const input = ctx.input;
|
|
34
|
+
if (!isCommand(input))
|
|
35
|
+
return false;
|
|
36
|
+
// 去掉 / 前缀,取命令名
|
|
37
|
+
const cmdName = input.slice(1).toLowerCase().trim();
|
|
38
|
+
const handler = commands[cmdName];
|
|
39
|
+
if (!handler) {
|
|
40
|
+
console.log(`\n❌ 未知命令: ${input} (输入 /help 查看可用命令)\n`);
|
|
41
|
+
ctx.input = "";
|
|
42
|
+
return true;
|
|
43
|
+
}
|
|
44
|
+
await handler(ctx);
|
|
45
|
+
return true;
|
|
46
|
+
}
|
|
47
|
+
/**
|
|
48
|
+
* 获取所有命令名称列表(用于提示信息)
|
|
49
|
+
*/
|
|
50
|
+
export function getCommandNames() {
|
|
51
|
+
// 去重(exit 和 quit 指向同一处理函数)
|
|
52
|
+
const names = new Set();
|
|
53
|
+
names.add("exit");
|
|
54
|
+
names.add("quit");
|
|
55
|
+
names.add("save");
|
|
56
|
+
names.add("new");
|
|
57
|
+
names.add("back");
|
|
58
|
+
names.add("editor");
|
|
59
|
+
names.add("clear");
|
|
60
|
+
names.add("help");
|
|
61
|
+
return Array.from(names).sort();
|
|
62
|
+
}
|
|
63
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/conversation-commands/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AACrC,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AAEvC;;;GAGG;AACH,MAAM,QAAQ,GAAmC;IAC/C,IAAI,EAAE,UAAU;IAChB,IAAI,EAAE,UAAU;IAChB,IAAI,EAAE,UAAU;IAChB,GAAG,EAAE,SAAS;IACd,IAAI,EAAE,UAAU;IAChB,MAAM,EAAE,YAAY;IACpB,KAAK,EAAE,WAAW;IAClB,IAAI,EAAE,UAAU;CACjB,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,SAAS,CAAC,KAAa;IACrC,OAAO,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;AAC/B,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,GAAwB;IAExB,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;IAExB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAEpC,eAAe;IACf,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;IACpD,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;IAElC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,CAAC,GAAG,CAAC,aAAa,KAAK,uBAAuB,CAAC,CAAC;QACvD,GAAG,CAAC,KAAK,GAAG,EAAE,CAAC;QACf,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC;IACnB,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe;IAC7B,2BAA2B;IAC3B,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;IAChC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAClB,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAClB,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAClB,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACjB,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAClB,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACpB,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACnB,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAClB,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;AAClC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"new.d.ts","sourceRoot":"","sources":["../../src/conversation-commands/new.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAKlD,wBAAsB,SAAS,CAAC,GAAG,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC,CA0CvE"}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
import chalk from "chalk";
|
|
2
|
+
import inquirer from "inquirer";
|
|
3
|
+
import { Message } from "@ai-zen/agents-core";
|
|
4
|
+
import { ConversationRepository, DraftRepository } from "@ai-zen/agents-sdk";
|
|
5
|
+
import { formatShortTime } from "../format-time.js";
|
|
6
|
+
import { CONVERSATIONS_DIR, DRAFTS_DIR } from "../config.js";
|
|
7
|
+
const conversationRepo = new ConversationRepository(CONVERSATIONS_DIR);
|
|
8
|
+
const draftRepo = new DraftRepository(DRAFTS_DIR);
|
|
9
|
+
export async function handleNew(ctx) {
|
|
10
|
+
const agent = ctx.agent;
|
|
11
|
+
const { confirmNew } = await inquirer.prompt([
|
|
12
|
+
{
|
|
13
|
+
type: "confirm",
|
|
14
|
+
name: "confirmNew",
|
|
15
|
+
message: "确定要开始新对话吗?当前对话将丢失。",
|
|
16
|
+
default: false,
|
|
17
|
+
},
|
|
18
|
+
]);
|
|
19
|
+
if (!confirmNew) {
|
|
20
|
+
console.log(chalk.gray("已取消\n"));
|
|
21
|
+
return;
|
|
22
|
+
}
|
|
23
|
+
// 将当前草稿存档(如果有),避免丢失
|
|
24
|
+
const draft = draftRepo.read();
|
|
25
|
+
if (draft && draft.messages.length > 1) {
|
|
26
|
+
const name = `草稿-${formatShortTime(draft.updatedAt)}`;
|
|
27
|
+
const id = name.replace(/[\\/:*?"<>|]/g, "_");
|
|
28
|
+
conversationRepo.write({
|
|
29
|
+
id,
|
|
30
|
+
agentId: draft.agentId || "default",
|
|
31
|
+
modelId: draft.modelId,
|
|
32
|
+
messages: draft.messages,
|
|
33
|
+
createdAt: new Date().toISOString(),
|
|
34
|
+
updatedAt: new Date().toISOString(),
|
|
35
|
+
});
|
|
36
|
+
console.log(chalk.gray(`📦 当前对话已存档: ${name}\n`));
|
|
37
|
+
draftRepo.delete();
|
|
38
|
+
}
|
|
39
|
+
// 重置为 Agent 定义的初始消息(system prompt + few-shot 示例等)
|
|
40
|
+
agent.messages = agent.definition.messages.map((m) => new Message(m));
|
|
41
|
+
// 更新上下文状态
|
|
42
|
+
ctx.currentName = `对话_${formatShortTime(new Date().toISOString())}`;
|
|
43
|
+
ctx.currentId = undefined;
|
|
44
|
+
ctx.input = "";
|
|
45
|
+
console.log(chalk.blue.bold("\n🆕 新会话已开始\n"));
|
|
46
|
+
}
|
|
47
|
+
//# sourceMappingURL=new.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"new.js","sourceRoot":"","sources":["../../src/conversation-commands/new.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,QAAQ,MAAM,UAAU,CAAC;AAChC,OAAO,EAAW,OAAO,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,sBAAsB,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAC7E,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAG7D,MAAM,gBAAgB,GAAG,IAAI,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;AACvE,MAAM,SAAS,GAAG,IAAI,eAAe,CAAC,UAAU,CAAC,CAAC;AAElD,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,GAAwB;IACtD,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;IACxB,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC;QAC3C;YACE,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,YAAY;YAClB,OAAO,EAAE,oBAAoB;YAC7B,OAAO,EAAE,KAAK;SACf;KACF,CAAC,CAAC;IAEH,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;QACjC,OAAO;IACT,CAAC;IAED,oBAAoB;IACpB,MAAM,KAAK,GAAG,SAAS,CAAC,IAAI,EAAE,CAAC;IAC/B,IAAI,KAAK,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvC,MAAM,IAAI,GAAG,MAAM,eAAe,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;QACtD,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC;QAC9C,gBAAgB,CAAC,KAAK,CAAC;YACrB,EAAE;YACF,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,SAAS;YACnC,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,CAAC,CAAC;QACjD,SAAS,CAAC,MAAM,EAAE,CAAC;IACrB,CAAC;IAED,kDAAkD;IAClD,KAAK,CAAC,QAAQ,GAAI,KAAK,CAAC,UAAU,CAAC,QAA8B,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAE7F,UAAU;IACV,GAAG,CAAC,WAAW,GAAG,MAAM,eAAe,CAAC,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;IACpE,GAAG,CAAC,SAAS,GAAG,SAAS,CAAC;IAC1B,GAAG,CAAC,KAAK,GAAG,EAAE,CAAC;IAEf,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC;AAChD,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"save.d.ts","sourceRoot":"","sources":["../../src/conversation-commands/save.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAIvD,wBAAsB,UAAU,CAAC,GAAG,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC,CA2BxE"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
import chalk from "chalk";
|
|
2
|
+
import inquirer from "inquirer";
|
|
3
|
+
import { ConversationRepository } from "@ai-zen/agents-sdk";
|
|
4
|
+
import { CONVERSATIONS_DIR } from "../config.js";
|
|
5
|
+
const conversationRepo = new ConversationRepository(CONVERSATIONS_DIR);
|
|
6
|
+
export async function handleSave(ctx) {
|
|
7
|
+
const agent = ctx.agent;
|
|
8
|
+
const { name } = await inquirer.prompt([
|
|
9
|
+
{
|
|
10
|
+
type: "input",
|
|
11
|
+
name: "name",
|
|
12
|
+
message: "对话名称:",
|
|
13
|
+
default: ctx.currentName,
|
|
14
|
+
},
|
|
15
|
+
]);
|
|
16
|
+
try {
|
|
17
|
+
const id = ctx.currentId || name.replace(/[\\/:*?"<>|]/g, "_");
|
|
18
|
+
conversationRepo.write({
|
|
19
|
+
id,
|
|
20
|
+
agentId: ctx.agentId || "default",
|
|
21
|
+
modelId: ctx.modelId,
|
|
22
|
+
messages: agent.messages,
|
|
23
|
+
createdAt: new Date().toISOString(),
|
|
24
|
+
updatedAt: new Date().toISOString(),
|
|
25
|
+
});
|
|
26
|
+
console.log(chalk.green(`\n✅ 对话已保存: ${name} (ID: ${id})\n`));
|
|
27
|
+
ctx.currentName = name;
|
|
28
|
+
ctx.currentId = id;
|
|
29
|
+
}
|
|
30
|
+
catch (error) {
|
|
31
|
+
console.error(chalk.red(`\n❌ 保存失败: ${error}\n`));
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
//# sourceMappingURL=save.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"save.js","sourceRoot":"","sources":["../../src/conversation-commands/save.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,QAAQ,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAC5D,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAGjD,MAAM,gBAAgB,GAAG,IAAI,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;AAEvE,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,GAAwB;IACvD,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;IACxB,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC;QACrC;YACE,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,OAAO;YAChB,OAAO,EAAE,GAAG,CAAC,WAAW;SACzB;KACF,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,MAAM,EAAE,GAAG,GAAG,CAAC,SAAS,IAAI,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC;QAC/D,gBAAgB,CAAC,KAAK,CAAC;YACrB,EAAE;YACF,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,SAAS;YACjC,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,cAAc,IAAI,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;QAC7D,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC;QACvB,GAAG,CAAC,SAAS,GAAG,EAAE,CAAC;IACrB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,KAAK,IAAI,CAAC,CAAC,CAAC;IACnD,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 对话运行器 — CLI 对话主循环
|
|
3
|
+
*
|
|
4
|
+
* 使用 SDK 的 SdkAgent + 插件驱动对话。
|
|
5
|
+
* CLI 层只负责:流式渲染、命令处理、UI 交互。
|
|
6
|
+
*/
|
|
7
|
+
import { AgentNS } from "@ai-zen/agents-core";
|
|
8
|
+
export interface RunConversationOptions {
|
|
9
|
+
modelId: string;
|
|
10
|
+
agentId?: string;
|
|
11
|
+
messages?: AgentNS.Message[];
|
|
12
|
+
conversationId?: string;
|
|
13
|
+
conversationName?: string;
|
|
14
|
+
query?: string;
|
|
15
|
+
}
|
|
16
|
+
export declare function runConversation(options: RunConversationOptions): Promise<void>;
|
|
17
|
+
//# sourceMappingURL=conversation-runner.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"conversation-runner.d.ts","sourceRoot":"","sources":["../src/conversation-runner.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AA8E9C,MAAM,WAAW,sBAAsB;IACrC,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;IAC7B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,wBAAsB,eAAe,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,IAAI,CAAC,CA8KpF"}
|
|
@@ -0,0 +1,212 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 对话运行器 — CLI 对话主循环
|
|
3
|
+
*
|
|
4
|
+
* 使用 SDK 的 SdkAgent + 插件驱动对话。
|
|
5
|
+
* CLI 层只负责:流式渲染、命令处理、UI 交互。
|
|
6
|
+
*/
|
|
7
|
+
import chalk from "chalk";
|
|
8
|
+
import inquirer from "inquirer";
|
|
9
|
+
import { AgentNS } from "@ai-zen/agents-core";
|
|
10
|
+
import { AutoMigratePlugin, AutoDraftPlugin, AutoRefreshToolsPlugin, ConversationRepository, } from "@ai-zen/agents-sdk";
|
|
11
|
+
import { DeltaRenderer } from "./delta-renderer.js";
|
|
12
|
+
import { createAgent, createMigrationAgent } from "./agent-creator.js";
|
|
13
|
+
import { readConfig, CONVERSATIONS_DIR, DRAFTS_DIR } from "./config.js";
|
|
14
|
+
import { ensureEndpointConfig } from "./config-wizard.js";
|
|
15
|
+
import { formatShortTime } from "./format-time.js";
|
|
16
|
+
import { dispatchCommand, getCommandNames } from "./conversation-commands/index.js";
|
|
17
|
+
const conversationRepo = new ConversationRepository(CONVERSATIONS_DIR);
|
|
18
|
+
/** 保存当前对话到 conversations 目录 */
|
|
19
|
+
function saveCurrentConversation(name, messages, modelId, existingId, agentId) {
|
|
20
|
+
const id = existingId || name.replace(/[\\/:*?"<>|]/g, "_");
|
|
21
|
+
conversationRepo.write({
|
|
22
|
+
id,
|
|
23
|
+
agentId: agentId || "default",
|
|
24
|
+
modelId,
|
|
25
|
+
messages,
|
|
26
|
+
createdAt: new Date().toISOString(),
|
|
27
|
+
updatedAt: new Date().toISOString(),
|
|
28
|
+
});
|
|
29
|
+
return id;
|
|
30
|
+
}
|
|
31
|
+
// ==================== 发送消息 ====================
|
|
32
|
+
async function sendAndStream(ctx) {
|
|
33
|
+
console.log(chalk.green.bold("\n🤖 AI:"));
|
|
34
|
+
try {
|
|
35
|
+
const messages = await ctx.agent.send(ctx.input);
|
|
36
|
+
process.stdout.write("\n\n");
|
|
37
|
+
const lastMessage = messages.at(-1);
|
|
38
|
+
if (lastMessage?.status === "error") {
|
|
39
|
+
console.error(chalk.red(`\n❌ 发生错误: ${JSON.stringify(lastMessage)}\n`));
|
|
40
|
+
try {
|
|
41
|
+
saveCurrentConversation(ctx.currentName, ctx.agent.messages, ctx.modelId, ctx.currentId, ctx.agentId);
|
|
42
|
+
console.log(chalk.yellow(`💾 错误时对话已自动保存: ${ctx.currentName}\n`));
|
|
43
|
+
}
|
|
44
|
+
catch (saveError) {
|
|
45
|
+
console.error(chalk.red(`❌ 自动保存失败: ${saveError}\n`));
|
|
46
|
+
}
|
|
47
|
+
return;
|
|
48
|
+
}
|
|
49
|
+
if (lastMessage?.role === AgentNS.Role.Assistant && Array.isArray(lastMessage.content)) {
|
|
50
|
+
for (const section of lastMessage.content) {
|
|
51
|
+
if (section.type === "image_url")
|
|
52
|
+
console.log(chalk.yellow(`[图片: ${section.image_url.url}]`));
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
console.log();
|
|
56
|
+
}
|
|
57
|
+
catch (error) {
|
|
58
|
+
process.stdout.write(chalk.red(`\n❌ 请求错误: ${error?.message || error}\n`));
|
|
59
|
+
if (error.message?.includes("API Key") || error.message?.includes("401") || error.message?.includes("403")) {
|
|
60
|
+
console.log(chalk.yellow("💡 提示: 请使用 'aiz config set-key' 设置正确的 API Key\n"));
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
export async function runConversation(options) {
|
|
65
|
+
const { modelId, agentId, messages, conversationId, conversationName, query } = options;
|
|
66
|
+
// 确保端点配置完整(如 API Key 已设置)
|
|
67
|
+
await ensureEndpointConfig(modelId);
|
|
68
|
+
// 在 runConversation 内部创建 agent
|
|
69
|
+
const agent = createAgent({ messages, agentId });
|
|
70
|
+
// ============ 插件注册 ============
|
|
71
|
+
// 1. autoRefreshTools — 每次 send 前刷新文件系统工具
|
|
72
|
+
agent.use(new AutoRefreshToolsPlugin());
|
|
73
|
+
// 2. autoDraft — 每次 send 后自动保存草稿
|
|
74
|
+
agent.use(new AutoDraftPlugin({
|
|
75
|
+
draftsDir: DRAFTS_DIR,
|
|
76
|
+
agentId: agentId || "default",
|
|
77
|
+
modelId,
|
|
78
|
+
conversationId,
|
|
79
|
+
cwd: process.cwd(),
|
|
80
|
+
}));
|
|
81
|
+
// 3. autoMigrate — 检测 token 超限时自动迁移
|
|
82
|
+
const migrationAgent = createMigrationAgent(modelId);
|
|
83
|
+
const config = readConfig();
|
|
84
|
+
const modelConfig = config.models.find((m) => m.id === modelId);
|
|
85
|
+
const maxTokens = modelConfig?.maxContextTokens ?? (modelConfig?.maxContextChars ? Math.floor(modelConfig.maxContextChars / 4) : undefined);
|
|
86
|
+
if (maxTokens && maxTokens > 0) {
|
|
87
|
+
agent.use(new AutoMigratePlugin({
|
|
88
|
+
maxTokens,
|
|
89
|
+
migrationAgent,
|
|
90
|
+
onHandoff: (handoffDoc, oldAgent, newAgent) => {
|
|
91
|
+
saveCurrentConversation(ctx.currentName, oldAgent.messages, ctx.modelId, ctx.currentId, ctx.agentId);
|
|
92
|
+
console.log(chalk.gray(` ✅ 原对话已保存: ${ctx.currentName}`));
|
|
93
|
+
ctx.agent = newAgent;
|
|
94
|
+
ctx.currentName = `对话_${formatShortTime(new Date().toISOString())}`;
|
|
95
|
+
ctx.currentId = undefined;
|
|
96
|
+
console.log(chalk.green.bold("\n🚀 任务迁移完成!新会话已就绪,交接文档已作为上下文注入。\n"));
|
|
97
|
+
console.log(chalk.gray("💡 你可以继续提问,新助手已经了解之前的全部工作。\n"));
|
|
98
|
+
},
|
|
99
|
+
}));
|
|
100
|
+
}
|
|
101
|
+
// 初始化所有插件
|
|
102
|
+
await agent.init();
|
|
103
|
+
const cmdList = getCommandNames().map((c) => `/${c}`).join(", ");
|
|
104
|
+
console.log(chalk.blue.bold(`💬 对话已开始 (输入 ${cmdList} 查看和操作)\n`));
|
|
105
|
+
const ctx = {
|
|
106
|
+
agent,
|
|
107
|
+
input: "",
|
|
108
|
+
currentName: conversationName || `对话_${formatShortTime(new Date().toISOString())}`,
|
|
109
|
+
currentId: conversationId,
|
|
110
|
+
modelId,
|
|
111
|
+
agentId,
|
|
112
|
+
running: true,
|
|
113
|
+
};
|
|
114
|
+
// ============ 流式渲染器 ============
|
|
115
|
+
const mainRenderer = new DeltaRenderer({
|
|
116
|
+
reasoningHeader: "\n\n💭 思考中...\n",
|
|
117
|
+
contentHeader: "\n\n💭 回答中...\n",
|
|
118
|
+
reasoningStyle: chalk.blue,
|
|
119
|
+
contentStyle: chalk.white,
|
|
120
|
+
});
|
|
121
|
+
const onRun = () => { mainRenderer.reset(); };
|
|
122
|
+
const onChunk = (chunk) => {
|
|
123
|
+
if (!chunk?.choices?.[0]?.delta)
|
|
124
|
+
return;
|
|
125
|
+
const delta = chunk.choices[0].delta;
|
|
126
|
+
const fr = chunk.choices[0].finish_reason ?? null;
|
|
127
|
+
mainRenderer.render(delta, fr);
|
|
128
|
+
};
|
|
129
|
+
const onError = (error) => {
|
|
130
|
+
process.stdout.write(chalk.red(`\n❌ 错误: ${error?.message || error}\n`));
|
|
131
|
+
};
|
|
132
|
+
// ============ 子 Agent 渲染 ============
|
|
133
|
+
const onSubAgent = (event) => {
|
|
134
|
+
const subAgent = event.agent;
|
|
135
|
+
const subCtx = event.ctx;
|
|
136
|
+
const toolName = subCtx.function_call?.name || "子任务";
|
|
137
|
+
const renderer = new DeltaRenderer({
|
|
138
|
+
reasoningHeader: "💭 ",
|
|
139
|
+
contentHeader: "",
|
|
140
|
+
reasoningStyle: chalk.blue,
|
|
141
|
+
contentStyle: chalk.white,
|
|
142
|
+
indent: " ",
|
|
143
|
+
});
|
|
144
|
+
let namePrinted = false;
|
|
145
|
+
subAgent.events.on("run", () => renderer.reset());
|
|
146
|
+
subAgent.events.on("chunk", (chunk) => {
|
|
147
|
+
if (!namePrinted) {
|
|
148
|
+
process.stdout.write(chalk.yellow.bold(`\n 🧩 ${toolName}:\n`));
|
|
149
|
+
namePrinted = true;
|
|
150
|
+
}
|
|
151
|
+
const delta = chunk?.choices?.[0]?.delta;
|
|
152
|
+
if (!delta)
|
|
153
|
+
return;
|
|
154
|
+
const fr = chunk?.choices?.[0]?.finish_reason ?? null;
|
|
155
|
+
renderer.render(delta, fr);
|
|
156
|
+
});
|
|
157
|
+
subAgent.events.on("error", (error) => {
|
|
158
|
+
renderer.reset();
|
|
159
|
+
process.stdout.write(chalk.red(`\n ❌ ${toolName} 错误: ${error?.message || error}\n`));
|
|
160
|
+
});
|
|
161
|
+
};
|
|
162
|
+
const onSubAgentEnd = ({ ctx: subCtx }) => {
|
|
163
|
+
const toolName = subCtx.function_call?.name || "子任务";
|
|
164
|
+
process.stdout.write(chalk.gray(`\n ✅ ${toolName} 完成\n`));
|
|
165
|
+
};
|
|
166
|
+
// ============ 注册事件 ============
|
|
167
|
+
ctx.agent.events.on("run", onRun);
|
|
168
|
+
ctx.agent.events.on("chunk", onChunk);
|
|
169
|
+
ctx.agent.events.on("error", onError);
|
|
170
|
+
ctx.agent.events.on("sub-agent", onSubAgent);
|
|
171
|
+
ctx.agent.events.on("sub-agent-end", onSubAgentEnd);
|
|
172
|
+
// ============ 初始消息 ============
|
|
173
|
+
if (query) {
|
|
174
|
+
ctx.input = query;
|
|
175
|
+
console.log(chalk.cyan("💬 你: ") + query + "\n");
|
|
176
|
+
await sendAndStream(ctx);
|
|
177
|
+
}
|
|
178
|
+
// ============ 主循环 ============
|
|
179
|
+
while (ctx.running) {
|
|
180
|
+
const { question } = await inquirer.prompt([
|
|
181
|
+
{
|
|
182
|
+
type: "input",
|
|
183
|
+
name: "question",
|
|
184
|
+
message: chalk.cyan("你:"),
|
|
185
|
+
prefix: "💬",
|
|
186
|
+
},
|
|
187
|
+
]);
|
|
188
|
+
ctx.input = question.trim();
|
|
189
|
+
if (!ctx.input)
|
|
190
|
+
continue;
|
|
191
|
+
const handled = await dispatchCommand(ctx);
|
|
192
|
+
if (handled) {
|
|
193
|
+
if (ctx.shouldSend) {
|
|
194
|
+
ctx.shouldSend = false;
|
|
195
|
+
console.log(chalk.cyan("💬 你: ") + ctx.input + "\n");
|
|
196
|
+
}
|
|
197
|
+
else {
|
|
198
|
+
continue;
|
|
199
|
+
}
|
|
200
|
+
}
|
|
201
|
+
if (!ctx.input)
|
|
202
|
+
continue;
|
|
203
|
+
await sendAndStream(ctx);
|
|
204
|
+
}
|
|
205
|
+
ctx.agent.events.off("run", onRun);
|
|
206
|
+
ctx.agent.events.off("chunk", onChunk);
|
|
207
|
+
ctx.agent.events.off("error", onError);
|
|
208
|
+
ctx.agent.events.off("sub-agent", onSubAgent);
|
|
209
|
+
ctx.agent.events.off("sub-agent-end", onSubAgentEnd);
|
|
210
|
+
console.log(chalk.blue.bold("\n👋 再见!\n"));
|
|
211
|
+
}
|
|
212
|
+
//# sourceMappingURL=conversation-runner.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"conversation-runner.js","sourceRoot":"","sources":["../src/conversation-runner.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,QAAQ,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAE9C,OAAO,EACL,iBAAiB,EACjB,eAAe,EACf,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AACvE,OAAO,EAAE,UAAU,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACxE,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAE1D,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAC;AAEpF,MAAM,gBAAgB,GAAG,IAAI,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;AAEvE,+BAA+B;AAC/B,SAAS,uBAAuB,CAC9B,IAAY,EACZ,QAA2B,EAC3B,OAAe,EACf,UAAmB,EACnB,OAAgB;IAEhB,MAAM,EAAE,GAAG,UAAU,IAAI,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC;IAC5D,gBAAgB,CAAC,KAAK,CAAC;QACrB,EAAE;QACF,OAAO,EAAE,OAAO,IAAI,SAAS;QAC7B,OAAO;QACP,QAAQ;QACR,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC,CAAC,CAAC;IACH,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,iDAAiD;AAEjD,KAAK,UAAU,aAAa,CAAC,GAAwB;IACnD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;IAE1C,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAE7B,MAAM,WAAW,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAEpC,IAAI,WAAW,EAAE,MAAM,KAAK,OAAO,EAAE,CAAC;YACpC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;YACvE,IAAI,CAAC;gBACH,uBAAuB,CAAC,GAAG,CAAC,WAAW,EAAE,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;gBACtG,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,kBAAkB,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;YACnE,CAAC;YAAC,OAAO,SAAS,EAAE,CAAC;gBACnB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,SAAS,IAAI,CAAC,CAAC,CAAC;YACvD,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,WAAW,EAAE,IAAI,KAAK,OAAO,CAAC,IAAI,CAAC,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC;YACvF,KAAK,MAAM,OAAO,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;gBAC1C,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;oBAC9B,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,OAAO,CAAC,SAAS,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QAED,OAAO,CAAC,GAAG,EAAE,CAAC;IAChB,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,KAAK,EAAE,OAAO,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC;QAC1E,IAAI,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3G,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,iDAAiD,CAAC,CAAC,CAAC;QAC/E,CAAC;IACH,CAAC;AACH,CAAC;AAaD,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,OAA+B;IACnE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,gBAAgB,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;IAExF,0BAA0B;IAC1B,MAAM,oBAAoB,CAAC,OAAO,CAAC,CAAC;IAEpC,+BAA+B;IAC/B,MAAM,KAAK,GAAG,WAAW,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;IAEjD,iCAAiC;IAEjC,0CAA0C;IAC1C,KAAK,CAAC,GAAG,CAAC,IAAI,sBAAsB,EAAE,CAAC,CAAC;IAExC,iCAAiC;IACjC,KAAK,CAAC,GAAG,CAAC,IAAI,eAAe,CAAC;QAC5B,SAAS,EAAE,UAAU;QACrB,OAAO,EAAE,OAAO,IAAI,SAAS;QAC7B,OAAO;QACP,cAAc;QACd,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE;KACnB,CAAC,CAAC,CAAC;IAEJ,oCAAoC;IACpC,MAAM,cAAc,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAC;IACrD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC;IAChE,MAAM,SAAS,GAAG,WAAW,EAAE,gBAAgB,IAAI,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,eAAe,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAC5I,IAAI,SAAS,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;QAC/B,KAAK,CAAC,GAAG,CAAC,IAAI,iBAAiB,CAAC;YAC9B,SAAS;YACT,cAAc;YACd,SAAS,EAAE,CAAC,UAAkB,EAAE,QAAkB,EAAE,QAAkB,EAAE,EAAE;gBACxE,uBAAuB,CAAC,GAAG,CAAC,WAAW,EAAE,QAAQ,CAAC,QAAQ,EAAE,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;gBACrG,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,eAAe,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;gBAE1D,GAAG,CAAC,KAAK,GAAG,QAAQ,CAAC;gBACrB,GAAG,CAAC,WAAW,GAAG,MAAM,eAAe,CAAC,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;gBACpE,GAAG,CAAC,SAAS,GAAG,SAAS,CAAC;gBAE1B,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC,CAAC;gBACpE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC,CAAC;YAC1D,CAAC;SACF,CAAC,CAAC,CAAC;IACN,CAAC;IAED,UAAU;IACV,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;IAEnB,MAAM,OAAO,GAAG,eAAe,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,OAAO,WAAW,CAAC,CAAC,CAAC;IAEjE,MAAM,GAAG,GAAwB;QAC/B,KAAK;QACL,KAAK,EAAE,EAAE;QACT,WAAW,EAAE,gBAAgB,IAAI,MAAM,eAAe,CAAC,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,EAAE;QAClF,SAAS,EAAE,cAAc;QACzB,OAAO;QACP,OAAO;QACP,OAAO,EAAE,IAAI;KACd,CAAC;IAEF,kCAAkC;IAElC,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC;QACrC,eAAe,EAAE,iBAAiB;QAClC,aAAa,EAAE,iBAAiB;QAChC,cAAc,EAAE,KAAK,CAAC,IAAI;QAC1B,YAAY,EAAE,KAAK,CAAC,KAAK;KAC1B,CAAC,CAAC;IAEH,MAAM,KAAK,GAAG,GAAG,EAAE,GAAG,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;IAE9C,MAAM,OAAO,GAAG,CAAC,KAAiC,EAAE,EAAE;QACpD,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK;YAAE,OAAO;QACxC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QACrC,MAAM,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,IAAI,IAAI,CAAC;QAClD,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACjC,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,CAAC,KAAU,EAAE,EAAE;QAC7B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,KAAK,EAAE,OAAO,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC;IAC1E,CAAC,CAAC;IAEF,uCAAuC;IAEvC,MAAM,UAAU,GAAG,CAAC,KAA+B,EAAE,EAAE;QACrD,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC;QAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC;QACzB,MAAM,QAAQ,GAAG,MAAM,CAAC,aAAa,EAAE,IAAI,IAAI,KAAK,CAAC;QACrD,MAAM,QAAQ,GAAG,IAAI,aAAa,CAAC;YACjC,eAAe,EAAE,KAAK;YACtB,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,KAAK,CAAC,IAAI;YAC1B,YAAY,EAAE,KAAK,CAAC,KAAK;YACzB,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;QACH,IAAI,WAAW,GAAG,KAAK,CAAC;QAExB,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;QAClD,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAiC,EAAE,EAAE;YAChE,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,QAAQ,KAAK,CAAC,CAAC,CAAC;gBACjE,WAAW,GAAG,IAAI,CAAC;YACrB,CAAC;YACD,MAAM,KAAK,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;YACzC,IAAI,CAAC,KAAK;gBAAE,OAAO;YACnB,MAAM,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,IAAI,CAAC;YACtD,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QACH,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAU,EAAE,EAAE;YACzC,QAAQ,CAAC,KAAK,EAAE,CAAC;YACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,QAAQ,QAAQ,KAAK,EAAE,OAAO,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC;QAC1F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,aAAa,GAAG,CAAC,EAAE,GAAG,EAAE,MAAM,EAA4B,EAAE,EAAE;QAClE,MAAM,QAAQ,GAAG,MAAM,CAAC,aAAa,EAAE,IAAI,IAAI,KAAK,CAAC;QACrD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,QAAQ,OAAO,CAAC,CAAC,CAAC;IAC/D,CAAC,CAAC;IAEF,iCAAiC;IAEjC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAClC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACtC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACtC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;IAC7C,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,eAAe,EAAE,aAAa,CAAC,CAAC;IAEpD,iCAAiC;IAEjC,IAAI,KAAK,EAAE,CAAC;QACV,GAAG,CAAC,KAAK,GAAG,KAAK,CAAC;QAClB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,KAAK,GAAG,IAAI,CAAC,CAAC;QACjD,MAAM,aAAa,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAED,gCAAgC;IAEhC,OAAO,GAAG,CAAC,OAAO,EAAE,CAAC;QACnB,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC;YACzC;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;gBACzB,MAAM,EAAE,IAAI;aACb;SACF,CAAC,CAAC;QAEH,GAAG,CAAC,KAAK,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC5B,IAAI,CAAC,GAAG,CAAC,KAAK;YAAE,SAAS;QAEzB,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC,GAAG,CAAC,CAAC;QAC3C,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;gBACnB,GAAG,CAAC,UAAU,GAAG,KAAK,CAAC;gBACvB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC;YACvD,CAAC;iBAAM,CAAC;gBACN,SAAS;YACX,CAAC;QACH,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,KAAK;YAAE,SAAS;QAEzB,MAAM,aAAa,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAED,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACnC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACvC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACvC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;IAC9C,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,eAAe,EAAE,aAAa,CAAC,CAAC;IAErD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;AAC7C,CAAC"}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
import { AgentNS } from "@ai-zen/agents-core";
|
|
2
|
+
export interface DeltaRendererOptions {
|
|
3
|
+
/** 首次 reasoning 时打印的头部 */
|
|
4
|
+
reasoningHeader?: string;
|
|
5
|
+
/** 首次 content 时打印的头部 */
|
|
6
|
+
contentHeader?: string;
|
|
7
|
+
/** reasoning 文本的样式 */
|
|
8
|
+
reasoningStyle?: any;
|
|
9
|
+
/** content 文本的样式 */
|
|
10
|
+
contentStyle?: any;
|
|
11
|
+
/** 每行前的缩进 */
|
|
12
|
+
indent?: string;
|
|
13
|
+
}
|
|
14
|
+
/**
|
|
15
|
+
* 流式 Delta 渲染器,管理 reasoning/content/tool_calls 的状态和输出
|
|
16
|
+
*/
|
|
17
|
+
export declare class DeltaRenderer {
|
|
18
|
+
private reasoningPrinted;
|
|
19
|
+
private contentPrinted;
|
|
20
|
+
private toolPrints;
|
|
21
|
+
private opts;
|
|
22
|
+
constructor(opts?: DeltaRendererOptions);
|
|
23
|
+
/**
|
|
24
|
+
* 渲染一个 delta,同时处理 content、reasoning_content 和 tool_calls
|
|
25
|
+
* @param delta - 流式 delta
|
|
26
|
+
* @param finishReason - 当前 choice 的 finish_reason(用于 tool_calls 完成标记)
|
|
27
|
+
*/
|
|
28
|
+
render(delta: AgentNS.Delta, finishReason?: AgentNS.FinishReason | null): void;
|
|
29
|
+
/** 重置状态(新一轮对话时调用) */
|
|
30
|
+
reset(): void;
|
|
31
|
+
private renderToolCalls;
|
|
32
|
+
}
|
|
33
|
+
//# sourceMappingURL=delta-renderer.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"delta-renderer.d.ts","sourceRoot":"","sources":["../src/delta-renderer.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAc9C,MAAM,WAAW,oBAAoB;IACnC,0BAA0B;IAC1B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,wBAAwB;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,sBAAsB;IACtB,cAAc,CAAC,EAAE,GAAG,CAAC;IACrB,oBAAoB;IACpB,YAAY,CAAC,EAAE,GAAG,CAAC;IACnB,aAAa;IACb,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAID;;GAEG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,UAAU,CAAqC;IACvD,OAAO,CAAC,IAAI,CAAiC;gBAEjC,IAAI,GAAE,oBAAyB;IAU3C;;;;OAIG;IACH,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,YAAY,GAAE,OAAO,CAAC,YAAY,GAAG,IAAW,GAAG,IAAI;IAiCpF,qBAAqB;IACrB,KAAK,IAAI,IAAI;IAQb,OAAO,CAAC,eAAe;CA0DxB"}
|
|
@@ -0,0 +1,117 @@
|
|
|
1
|
+
import chalk from "chalk";
|
|
2
|
+
import { AgentNS } from "@ai-zen/agents-core";
|
|
3
|
+
// ==================== 流式渲染器 ====================
|
|
4
|
+
/**
|
|
5
|
+
* 流式 Delta 渲染器,管理 reasoning/content/tool_calls 的状态和输出
|
|
6
|
+
*/
|
|
7
|
+
export class DeltaRenderer {
|
|
8
|
+
reasoningPrinted = false;
|
|
9
|
+
contentPrinted = false;
|
|
10
|
+
toolPrints = {};
|
|
11
|
+
opts;
|
|
12
|
+
constructor(opts = {}) {
|
|
13
|
+
this.opts = {
|
|
14
|
+
reasoningHeader: opts.reasoningHeader ?? "",
|
|
15
|
+
contentHeader: opts.contentHeader ?? "",
|
|
16
|
+
reasoningStyle: opts.reasoningStyle ?? chalk.blue,
|
|
17
|
+
contentStyle: opts.contentStyle ?? chalk.white,
|
|
18
|
+
indent: opts.indent ?? "",
|
|
19
|
+
};
|
|
20
|
+
}
|
|
21
|
+
/**
|
|
22
|
+
* 渲染一个 delta,同时处理 content、reasoning_content 和 tool_calls
|
|
23
|
+
* @param delta - 流式 delta
|
|
24
|
+
* @param finishReason - 当前 choice 的 finish_reason(用于 tool_calls 完成标记)
|
|
25
|
+
*/
|
|
26
|
+
render(delta, finishReason = null) {
|
|
27
|
+
// 1. 渲染工具调用
|
|
28
|
+
if (delta.tool_calls) {
|
|
29
|
+
this.renderToolCalls(delta, finishReason);
|
|
30
|
+
}
|
|
31
|
+
// 2. 渲染 reasoning
|
|
32
|
+
if (delta.reasoning_content) {
|
|
33
|
+
if (!this.reasoningPrinted && this.opts.reasoningHeader) {
|
|
34
|
+
process.stdout.write(this.opts.indent + this.opts.reasoningHeader);
|
|
35
|
+
this.reasoningPrinted = true;
|
|
36
|
+
}
|
|
37
|
+
process.stdout.write(this.opts.reasoningStyle(delta.reasoning_content));
|
|
38
|
+
}
|
|
39
|
+
// 3. 渲染 content
|
|
40
|
+
if (delta.content) {
|
|
41
|
+
if (!this.contentPrinted && this.opts.contentHeader) {
|
|
42
|
+
process.stdout.write(this.opts.indent + this.opts.contentHeader);
|
|
43
|
+
this.contentPrinted = true;
|
|
44
|
+
}
|
|
45
|
+
if (typeof delta.content === "string") {
|
|
46
|
+
process.stdout.write(this.opts.contentStyle(delta.content));
|
|
47
|
+
}
|
|
48
|
+
else if (Array.isArray(delta.content)) {
|
|
49
|
+
for (const s of delta.content) {
|
|
50
|
+
if (s.type === "text" && s.text) {
|
|
51
|
+
process.stdout.write(this.opts.contentStyle(s.text));
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
/** 重置状态(新一轮对话时调用) */
|
|
58
|
+
reset() {
|
|
59
|
+
this.reasoningPrinted = false;
|
|
60
|
+
this.contentPrinted = false;
|
|
61
|
+
for (const k in this.toolPrints)
|
|
62
|
+
delete this.toolPrints[k];
|
|
63
|
+
}
|
|
64
|
+
// ==================== 工具调用渲染 ====================
|
|
65
|
+
renderToolCalls(delta, finishReason) {
|
|
66
|
+
if (!delta.tool_calls || delta.tool_calls.length === 0)
|
|
67
|
+
return;
|
|
68
|
+
const isFirstToolCall = Object.keys(this.toolPrints).length === 0 &&
|
|
69
|
+
delta.tool_calls.some((tc) => tc.function?.name || tc.function?.arguments);
|
|
70
|
+
if (isFirstToolCall)
|
|
71
|
+
process.stdout.write(chalk.blue.bold("\n\n💭 工具调用中..."));
|
|
72
|
+
for (const tc of delta.tool_calls) {
|
|
73
|
+
const index = tc.index ?? 0;
|
|
74
|
+
const func = tc.function;
|
|
75
|
+
if (!this.toolPrints[index]) {
|
|
76
|
+
this.toolPrints[index] = {
|
|
77
|
+
name: "",
|
|
78
|
+
arguments: "",
|
|
79
|
+
namePrinted: false,
|
|
80
|
+
argsPrinted: false,
|
|
81
|
+
completed: false,
|
|
82
|
+
};
|
|
83
|
+
}
|
|
84
|
+
const p = this.toolPrints[index];
|
|
85
|
+
if (func?.name)
|
|
86
|
+
p.name += func.name;
|
|
87
|
+
if (func?.arguments)
|
|
88
|
+
p.arguments += func.arguments;
|
|
89
|
+
if (p.name && !p.namePrinted) {
|
|
90
|
+
process.stdout.write(chalk.magenta.bold(`\n🔧 ${index} ${p.name}\n`));
|
|
91
|
+
p.namePrinted = true;
|
|
92
|
+
}
|
|
93
|
+
if (p.arguments && !p.argsPrinted && p.namePrinted)
|
|
94
|
+
p.argsPrinted = true;
|
|
95
|
+
if (func?.arguments && p.argsPrinted)
|
|
96
|
+
process.stdout.write(chalk.gray(func.arguments));
|
|
97
|
+
if (finishReason === AgentNS.FinishReason.ToolCalls)
|
|
98
|
+
p.completed = true;
|
|
99
|
+
}
|
|
100
|
+
if (finishReason === AgentNS.FinishReason.ToolCalls) {
|
|
101
|
+
for (const idx of Object.keys(this.toolPrints).map(Number)) {
|
|
102
|
+
const p = this.toolPrints[idx];
|
|
103
|
+
if (p.completed && p.arguments) {
|
|
104
|
+
process.stdout.write("\n");
|
|
105
|
+
try {
|
|
106
|
+
const parsed = JSON.parse(p.arguments);
|
|
107
|
+
process.stdout.write(chalk.gray(` ${JSON.stringify(parsed, null, 4)}\n`));
|
|
108
|
+
}
|
|
109
|
+
catch {
|
|
110
|
+
process.stdout.write(chalk.gray(` ${p.arguments}\n`));
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
//# sourceMappingURL=delta-renderer.js.map
|