@zhin.js/core 1.0.25 → 1.0.27
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/CHANGELOG.md +19 -0
- package/README.md +84 -342
- package/lib/adapter.d.ts +17 -0
- package/lib/adapter.d.ts.map +1 -1
- package/lib/adapter.js +84 -2
- package/lib/adapter.js.map +1 -1
- package/lib/ai/agent.d.ts +126 -0
- package/lib/ai/agent.d.ts.map +1 -0
- package/lib/ai/agent.js +645 -0
- package/lib/ai/agent.js.map +1 -0
- package/lib/ai/context-manager.d.ts +213 -0
- package/lib/ai/context-manager.d.ts.map +1 -0
- package/lib/ai/context-manager.js +313 -0
- package/lib/ai/context-manager.js.map +1 -0
- package/lib/ai/conversation-memory.d.ts +181 -0
- package/lib/ai/conversation-memory.d.ts.map +1 -0
- package/lib/ai/conversation-memory.js +581 -0
- package/lib/ai/conversation-memory.js.map +1 -0
- package/lib/ai/follow-up.d.ts +131 -0
- package/lib/ai/follow-up.d.ts.map +1 -0
- package/lib/ai/follow-up.js +265 -0
- package/lib/ai/follow-up.js.map +1 -0
- package/lib/ai/index.d.ts +29 -0
- package/lib/ai/index.d.ts.map +1 -0
- package/lib/ai/index.js +34 -0
- package/lib/ai/index.js.map +1 -0
- package/lib/ai/init.d.ts +30 -0
- package/lib/ai/init.d.ts.map +1 -0
- package/lib/ai/init.js +424 -0
- package/lib/ai/init.js.map +1 -0
- package/lib/ai/output.d.ts +93 -0
- package/lib/ai/output.d.ts.map +1 -0
- package/lib/ai/output.js +176 -0
- package/lib/ai/output.js.map +1 -0
- package/lib/ai/providers/anthropic.d.ts +23 -0
- package/lib/ai/providers/anthropic.d.ts.map +1 -0
- package/lib/ai/providers/anthropic.js +322 -0
- package/lib/ai/providers/anthropic.js.map +1 -0
- package/lib/ai/providers/base.d.ts +43 -0
- package/lib/ai/providers/base.d.ts.map +1 -0
- package/lib/ai/providers/base.js +135 -0
- package/lib/ai/providers/base.js.map +1 -0
- package/lib/ai/providers/index.d.ts +12 -0
- package/lib/ai/providers/index.d.ts.map +1 -0
- package/lib/ai/providers/index.js +9 -0
- package/lib/ai/providers/index.js.map +1 -0
- package/lib/ai/providers/ollama.d.ts +25 -0
- package/lib/ai/providers/ollama.d.ts.map +1 -0
- package/lib/ai/providers/ollama.js +243 -0
- package/lib/ai/providers/ollama.js.map +1 -0
- package/lib/ai/providers/openai.d.ts +46 -0
- package/lib/ai/providers/openai.d.ts.map +1 -0
- package/lib/ai/providers/openai.js +132 -0
- package/lib/ai/providers/openai.js.map +1 -0
- package/lib/ai/rate-limiter.d.ts +38 -0
- package/lib/ai/rate-limiter.d.ts.map +1 -0
- package/lib/ai/rate-limiter.js +86 -0
- package/lib/ai/rate-limiter.js.map +1 -0
- package/lib/ai/service.d.ts +81 -0
- package/lib/ai/service.d.ts.map +1 -0
- package/lib/ai/service.js +274 -0
- package/lib/ai/service.js.map +1 -0
- package/lib/ai/session.d.ts +186 -0
- package/lib/ai/session.d.ts.map +1 -0
- package/lib/ai/session.js +443 -0
- package/lib/ai/session.js.map +1 -0
- package/lib/ai/tone-detector.d.ts +19 -0
- package/lib/ai/tone-detector.d.ts.map +1 -0
- package/lib/ai/tone-detector.js +72 -0
- package/lib/ai/tone-detector.js.map +1 -0
- package/lib/ai/tools.d.ts +45 -0
- package/lib/ai/tools.d.ts.map +1 -0
- package/lib/ai/tools.js +206 -0
- package/lib/ai/tools.js.map +1 -0
- package/lib/ai/types.d.ts +264 -0
- package/lib/ai/types.d.ts.map +1 -0
- package/lib/ai/types.js +6 -0
- package/lib/ai/types.js.map +1 -0
- package/lib/ai/user-profile.d.ts +56 -0
- package/lib/ai/user-profile.d.ts.map +1 -0
- package/lib/ai/user-profile.js +130 -0
- package/lib/ai/user-profile.js.map +1 -0
- package/lib/ai/zhin-agent.d.ts +165 -0
- package/lib/ai/zhin-agent.d.ts.map +1 -0
- package/lib/ai/zhin-agent.js +707 -0
- package/lib/ai/zhin-agent.js.map +1 -0
- package/lib/built/ai-trigger.d.ts.map +1 -1
- package/lib/built/ai-trigger.js +7 -3
- package/lib/built/ai-trigger.js.map +1 -1
- package/lib/built/command.d.ts +33 -17
- package/lib/built/command.d.ts.map +1 -1
- package/lib/built/command.js +71 -44
- package/lib/built/command.js.map +1 -1
- package/lib/built/component.d.ts +42 -15
- package/lib/built/component.d.ts.map +1 -1
- package/lib/built/component.js +84 -52
- package/lib/built/component.js.map +1 -1
- package/lib/built/config.d.ts +64 -5
- package/lib/built/config.d.ts.map +1 -1
- package/lib/built/config.js +129 -12
- package/lib/built/config.js.map +1 -1
- package/lib/built/cron.d.ts +41 -18
- package/lib/built/cron.d.ts.map +1 -1
- package/lib/built/cron.js +106 -63
- package/lib/built/cron.js.map +1 -1
- package/lib/built/database.d.ts +55 -6
- package/lib/built/database.d.ts.map +1 -1
- package/lib/built/database.js +93 -22
- package/lib/built/database.js.map +1 -1
- package/lib/built/dispatcher.d.ts +118 -0
- package/lib/built/dispatcher.d.ts.map +1 -0
- package/lib/built/dispatcher.js +196 -0
- package/lib/built/dispatcher.js.map +1 -0
- package/lib/built/permission.d.ts +45 -5
- package/lib/built/permission.d.ts.map +1 -1
- package/lib/built/permission.js +56 -11
- package/lib/built/permission.js.map +1 -1
- package/lib/built/skill.d.ts +117 -0
- package/lib/built/skill.d.ts.map +1 -0
- package/lib/built/skill.js +191 -0
- package/lib/built/skill.js.map +1 -0
- package/lib/built/tool.d.ts +71 -164
- package/lib/built/tool.d.ts.map +1 -1
- package/lib/built/tool.js +212 -297
- package/lib/built/tool.js.map +1 -1
- package/lib/feature.d.ts +75 -0
- package/lib/feature.d.ts.map +1 -0
- package/lib/feature.js +69 -0
- package/lib/feature.js.map +1 -0
- package/lib/index.d.ts +4 -0
- package/lib/index.d.ts.map +1 -1
- package/lib/index.js +7 -0
- package/lib/index.js.map +1 -1
- package/lib/plugin.d.ts +25 -17
- package/lib/plugin.d.ts.map +1 -1
- package/lib/plugin.js +180 -20
- package/lib/plugin.js.map +1 -1
- package/lib/types.d.ts +4 -9
- package/lib/types.d.ts.map +1 -1
- package/package.json +6 -6
- package/src/adapter.ts +101 -2
- package/src/ai/agent.ts +772 -0
- package/src/ai/context-manager.ts +440 -0
- package/src/ai/conversation-memory.ts +774 -0
- package/src/ai/follow-up.ts +357 -0
- package/src/ai/index.ts +128 -0
- package/src/ai/init.ts +502 -0
- package/src/ai/output.ts +261 -0
- package/src/ai/providers/anthropic.ts +375 -0
- package/src/ai/providers/base.ts +173 -0
- package/src/ai/providers/index.ts +13 -0
- package/src/ai/providers/ollama.ts +292 -0
- package/src/ai/providers/openai.ts +167 -0
- package/src/ai/rate-limiter.ts +129 -0
- package/src/ai/service.ts +319 -0
- package/src/ai/session.ts +544 -0
- package/src/ai/tone-detector.ts +89 -0
- package/src/ai/tools.ts +218 -0
- package/src/ai/types.ts +296 -0
- package/src/ai/user-profile.ts +181 -0
- package/src/ai/zhin-agent.ts +845 -0
- package/src/built/ai-trigger.ts +6 -3
- package/src/built/command.ts +75 -69
- package/src/built/component.ts +94 -76
- package/src/built/config.ts +288 -128
- package/src/built/cron.ts +117 -101
- package/src/built/database.ts +128 -33
- package/src/built/dispatcher.ts +332 -0
- package/src/built/permission.ts +146 -54
- package/src/built/skill.ts +280 -0
- package/src/built/tool.ts +245 -366
- package/src/feature.ts +113 -0
- package/src/index.ts +7 -0
- package/src/plugin.ts +198 -33
- package/src/types.ts +6 -10
- package/tests/adapter.test.ts +153 -1
- package/tests/ai/agent.test.ts +614 -0
- package/tests/ai/ai-trigger.test.ts +368 -0
- package/tests/ai/context-manager.test.ts +413 -0
- package/tests/ai/conversation-memory.test.ts +128 -0
- package/tests/ai/follow-up.test.ts +175 -0
- package/tests/ai/integration.test.ts +584 -0
- package/tests/ai/output.test.ts +128 -0
- package/tests/ai/providers.integration.test.ts +227 -0
- package/tests/ai/rate-limiter.test.ts +108 -0
- package/tests/ai/session.test.ts +375 -0
- package/tests/ai/setup.ts +308 -0
- package/tests/ai/tone-detector.test.ts +80 -0
- package/tests/ai/tool.test.ts +800 -0
- package/tests/ai/tools-builtin.test.ts +346 -0
- package/tests/ai/user-profile.test.ts +73 -0
- package/tests/ai/zhin-agent.test.ts +177 -0
- package/tests/config.test.ts +46 -0
- package/tests/cron.test.ts +94 -5
- package/tests/dispatcher.test.ts +146 -0
- package/tests/feature.test.ts +145 -0
- package/tests/features-builtin.test.ts +191 -0
- package/tests/plugin.test.ts +88 -14
- package/tests/skill-feature.test.ts +179 -0
- package/tests/tool-feature.test.ts +254 -0
- package/test/minimal-bot.ts +0 -31
- package/test/stress-test.ts +0 -123
|
@@ -0,0 +1,186 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @zhin.js/ai - Session Manager
|
|
3
|
+
* 会话管理器,支持上下文记忆和数据库持久化
|
|
4
|
+
*
|
|
5
|
+
* 特性:
|
|
6
|
+
* - 数据库持久化存储(使用 Zhin 的数据库服务)
|
|
7
|
+
* - 内存缓存加速读取
|
|
8
|
+
* - 自动过期清理
|
|
9
|
+
* - 更长的上下文记忆能力
|
|
10
|
+
*/
|
|
11
|
+
import type { ChatMessage, SessionConfig, Session } from './types.js';
|
|
12
|
+
/**
|
|
13
|
+
* 数据库模型定义
|
|
14
|
+
*/
|
|
15
|
+
export declare const AI_SESSION_MODEL: {
|
|
16
|
+
session_id: {
|
|
17
|
+
type: "text";
|
|
18
|
+
nullable: boolean;
|
|
19
|
+
};
|
|
20
|
+
messages: {
|
|
21
|
+
type: "json";
|
|
22
|
+
default: never[];
|
|
23
|
+
};
|
|
24
|
+
config: {
|
|
25
|
+
type: "json";
|
|
26
|
+
default: {};
|
|
27
|
+
};
|
|
28
|
+
created_at: {
|
|
29
|
+
type: "integer";
|
|
30
|
+
default: number;
|
|
31
|
+
};
|
|
32
|
+
updated_at: {
|
|
33
|
+
type: "integer";
|
|
34
|
+
default: number;
|
|
35
|
+
};
|
|
36
|
+
};
|
|
37
|
+
/**
|
|
38
|
+
* 会话管理器接口
|
|
39
|
+
*/
|
|
40
|
+
export interface ISessionManager {
|
|
41
|
+
get(sessionId: string, config?: SessionConfig): Session | Promise<Session>;
|
|
42
|
+
has(sessionId: string): boolean | Promise<boolean>;
|
|
43
|
+
addMessage(sessionId: string, message: ChatMessage): void | Promise<void>;
|
|
44
|
+
getMessages(sessionId: string): ChatMessage[] | Promise<ChatMessage[]>;
|
|
45
|
+
setSystemPrompt(sessionId: string, prompt: string): void | Promise<void>;
|
|
46
|
+
clear(sessionId: string): boolean | Promise<boolean>;
|
|
47
|
+
reset(sessionId: string): void | Promise<void>;
|
|
48
|
+
listSessions(): string[] | Promise<string[]>;
|
|
49
|
+
getStats(): {
|
|
50
|
+
total: number;
|
|
51
|
+
active: number;
|
|
52
|
+
expired: number;
|
|
53
|
+
} | Promise<{
|
|
54
|
+
total: number;
|
|
55
|
+
active: number;
|
|
56
|
+
expired: number;
|
|
57
|
+
}>;
|
|
58
|
+
cleanup(): number | Promise<number>;
|
|
59
|
+
dispose(): void | Promise<void>;
|
|
60
|
+
}
|
|
61
|
+
/**
|
|
62
|
+
* 内存会话管理器(回退方案)
|
|
63
|
+
*/
|
|
64
|
+
export declare class MemorySessionManager implements ISessionManager {
|
|
65
|
+
private sessions;
|
|
66
|
+
private config;
|
|
67
|
+
private cleanupTimer?;
|
|
68
|
+
constructor(config?: {
|
|
69
|
+
maxHistory?: number;
|
|
70
|
+
expireMs?: number;
|
|
71
|
+
});
|
|
72
|
+
get(sessionId: string, config?: SessionConfig): Session;
|
|
73
|
+
has(sessionId: string): boolean;
|
|
74
|
+
addMessage(sessionId: string, message: ChatMessage): void;
|
|
75
|
+
private trimMessages;
|
|
76
|
+
getMessages(sessionId: string): ChatMessage[];
|
|
77
|
+
setSystemPrompt(sessionId: string, prompt: string): void;
|
|
78
|
+
clear(sessionId: string): boolean;
|
|
79
|
+
reset(sessionId: string): void;
|
|
80
|
+
listSessions(): string[];
|
|
81
|
+
getStats(): {
|
|
82
|
+
total: number;
|
|
83
|
+
active: number;
|
|
84
|
+
expired: number;
|
|
85
|
+
};
|
|
86
|
+
cleanup(): number;
|
|
87
|
+
dispose(): void;
|
|
88
|
+
}
|
|
89
|
+
/**
|
|
90
|
+
* 数据库会话管理器
|
|
91
|
+
* 使用 Zhin 的数据库服务进行持久化存储
|
|
92
|
+
*/
|
|
93
|
+
export declare class DatabaseSessionManager implements ISessionManager {
|
|
94
|
+
private cache;
|
|
95
|
+
private config;
|
|
96
|
+
private cleanupTimer?;
|
|
97
|
+
private saveQueue;
|
|
98
|
+
private saveTimer?;
|
|
99
|
+
private model;
|
|
100
|
+
constructor(model: any, config?: {
|
|
101
|
+
maxHistory?: number;
|
|
102
|
+
expireMs?: number;
|
|
103
|
+
});
|
|
104
|
+
/**
|
|
105
|
+
* 从数据库加载会话
|
|
106
|
+
*/
|
|
107
|
+
private loadSession;
|
|
108
|
+
/**
|
|
109
|
+
* 保存会话到数据库(防抖)
|
|
110
|
+
*/
|
|
111
|
+
private schedulesSave;
|
|
112
|
+
/**
|
|
113
|
+
* 批量保存队列中的会话
|
|
114
|
+
*/
|
|
115
|
+
private flushSaveQueue;
|
|
116
|
+
get(sessionId: string, config?: SessionConfig): Promise<Session>;
|
|
117
|
+
has(sessionId: string): Promise<boolean>;
|
|
118
|
+
addMessage(sessionId: string, message: ChatMessage): Promise<void>;
|
|
119
|
+
private trimMessages;
|
|
120
|
+
getMessages(sessionId: string): Promise<ChatMessage[]>;
|
|
121
|
+
setSystemPrompt(sessionId: string, prompt: string): Promise<void>;
|
|
122
|
+
clear(sessionId: string): Promise<boolean>;
|
|
123
|
+
reset(sessionId: string): Promise<void>;
|
|
124
|
+
listSessions(): Promise<string[]>;
|
|
125
|
+
getStats(): Promise<{
|
|
126
|
+
total: number;
|
|
127
|
+
active: number;
|
|
128
|
+
expired: number;
|
|
129
|
+
}>;
|
|
130
|
+
cleanup(): Promise<number>;
|
|
131
|
+
dispose(): Promise<void>;
|
|
132
|
+
}
|
|
133
|
+
/**
|
|
134
|
+
* 会话管理器包装器
|
|
135
|
+
* 支持同步和异步接口的统一使用
|
|
136
|
+
*/
|
|
137
|
+
export declare class SessionManager implements ISessionManager {
|
|
138
|
+
private manager;
|
|
139
|
+
constructor(manager: ISessionManager);
|
|
140
|
+
/**
|
|
141
|
+
* 生成会话 ID
|
|
142
|
+
*/
|
|
143
|
+
static generateId(platform: string, userId: string, channelId?: string): string;
|
|
144
|
+
get(sessionId: string, config?: SessionConfig): Session | Promise<Session>;
|
|
145
|
+
has(sessionId: string): boolean | Promise<boolean>;
|
|
146
|
+
addMessage(sessionId: string, message: ChatMessage): void | Promise<void>;
|
|
147
|
+
getMessages(sessionId: string): ChatMessage[] | Promise<ChatMessage[]>;
|
|
148
|
+
setSystemPrompt(sessionId: string, prompt: string): void | Promise<void>;
|
|
149
|
+
clear(sessionId: string): boolean | Promise<boolean>;
|
|
150
|
+
reset(sessionId: string): void | Promise<void>;
|
|
151
|
+
listSessions(): string[] | Promise<string[]>;
|
|
152
|
+
getStats(): {
|
|
153
|
+
total: number;
|
|
154
|
+
active: number;
|
|
155
|
+
expired: number;
|
|
156
|
+
} | Promise<{
|
|
157
|
+
total: number;
|
|
158
|
+
active: number;
|
|
159
|
+
expired: number;
|
|
160
|
+
}>;
|
|
161
|
+
cleanup(): number | Promise<number>;
|
|
162
|
+
dispose(): void | Promise<void>;
|
|
163
|
+
}
|
|
164
|
+
/**
|
|
165
|
+
* 创建内存会话管理器(回退方案)
|
|
166
|
+
*/
|
|
167
|
+
export declare function createMemorySessionManager(config?: {
|
|
168
|
+
maxHistory?: number;
|
|
169
|
+
expireMs?: number;
|
|
170
|
+
}): SessionManager;
|
|
171
|
+
/**
|
|
172
|
+
* 创建数据库会话管理器
|
|
173
|
+
*/
|
|
174
|
+
export declare function createDatabaseSessionManager(model: any, config?: {
|
|
175
|
+
maxHistory?: number;
|
|
176
|
+
expireMs?: number;
|
|
177
|
+
}): SessionManager;
|
|
178
|
+
/**
|
|
179
|
+
* 创建会话管理器(向后兼容)
|
|
180
|
+
* @deprecated 使用 createMemorySessionManager 或 createDatabaseSessionManager
|
|
181
|
+
*/
|
|
182
|
+
export declare function createSessionManager(config?: {
|
|
183
|
+
maxHistory?: number;
|
|
184
|
+
expireMs?: number;
|
|
185
|
+
}): SessionManager;
|
|
186
|
+
//# sourceMappingURL=session.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../src/ai/session.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAGH,OAAO,KAAK,EAAE,WAAW,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAItE;;GAEG;AACH,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;;;;;CAM5B,CAAC;AAcF;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,GAAG,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,aAAa,GAAG,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3E,GAAG,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACnD,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1E,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,WAAW,EAAE,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;IACvE,eAAe,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzE,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACrD,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/C,YAAY,IAAI,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC7C,QAAQ,IAAI;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC7H,OAAO,IAAI,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IACpC,OAAO,IAAI,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACjC;AAED;;GAEG;AACH,qBAAa,oBAAqB,YAAW,eAAe;IAC1D,OAAO,CAAC,QAAQ,CAAmC;IACnD,OAAO,CAAC,MAAM,CAA2D;IACzE,OAAO,CAAC,YAAY,CAAC,CAAiC;gBAE1C,MAAM,GAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAO;IAUnE,GAAG,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,aAAa,GAAG,OAAO;IAmBvD,GAAG,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAI/B,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,GAAG,IAAI;IAOzD,OAAO,CAAC,YAAY;IAUpB,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,WAAW,EAAE;IAI7C,eAAe,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IAOxD,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAIjC,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAS9B,YAAY,IAAI,MAAM,EAAE;IAIxB,QAAQ,IAAI;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE;IAgB9D,OAAO,IAAI,MAAM;IAejB,OAAO,IAAI,IAAI;CAOhB;AAED;;;GAGG;AACH,qBAAa,sBAAuB,YAAW,eAAe;IAC5D,OAAO,CAAC,KAAK,CAAmC;IAChD,OAAO,CAAC,MAAM,CAA2D;IACzE,OAAO,CAAC,YAAY,CAAC,CAAiC;IACtD,OAAO,CAAC,SAAS,CAAmC;IACpD,OAAO,CAAC,SAAS,CAAC,CAAgC;IAClD,OAAO,CAAC,KAAK,CAAM;gBAGjB,KAAK,EAAE,GAAG,EACV,MAAM,GAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAO;IAYzD;;OAEG;YACW,WAAW;IA0BzB;;OAEG;IACH,OAAO,CAAC,aAAa;IAQrB;;OAEG;YACW,cAAc;IA4BtB,GAAG,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC;IA4BhE,GAAG,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAaxC,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IAQxE,OAAO,CAAC,YAAY;IAUd,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAKtD,eAAe,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQjE,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAa1C,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQvC,YAAY,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAUjC,QAAQ,IAAI,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;IAkCvE,OAAO,IAAI,OAAO,CAAC,MAAM,CAAC;IAsB1B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAe/B;AAED;;;GAGG;AACH,qBAAa,cAAe,YAAW,eAAe;IACpD,OAAO,CAAC,OAAO,CAAkB;gBAErB,OAAO,EAAE,eAAe;IAIpC;;OAEG;IACH,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM;IAM/E,GAAG,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,aAAa,GAAG,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IAI1E,GAAG,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IAIlD,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAIzE,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,WAAW,EAAE,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAItE,eAAe,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAIxE,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IAIpD,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAI9C,YAAY,IAAI,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAI5C,QAAQ,IAAI;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;IAI5H,OAAO,IAAI,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAInC,OAAO,IAAI,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;CAGhC;AAED;;GAEG;AACH,wBAAgB,0BAA0B,CAAC,MAAM,CAAC,EAAE;IAAE,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,cAAc,CAE9G;AAED;;GAEG;AACH,wBAAgB,4BAA4B,CAC1C,KAAK,EAAE,GAAG,EACV,MAAM,CAAC,EAAE;IAAE,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;CAAE,GAClD,cAAc,CAEhB;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,CAAC,MAAM,CAAC,EAAE;IAAE,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,cAAc,CAExG"}
|
|
@@ -0,0 +1,443 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @zhin.js/ai - Session Manager
|
|
3
|
+
* 会话管理器,支持上下文记忆和数据库持久化
|
|
4
|
+
*
|
|
5
|
+
* 特性:
|
|
6
|
+
* - 数据库持久化存储(使用 Zhin 的数据库服务)
|
|
7
|
+
* - 内存缓存加速读取
|
|
8
|
+
* - 自动过期清理
|
|
9
|
+
* - 更长的上下文记忆能力
|
|
10
|
+
*/
|
|
11
|
+
import { Logger } from '@zhin.js/logger';
|
|
12
|
+
const logger = new Logger(null, 'AI-Session');
|
|
13
|
+
/**
|
|
14
|
+
* 数据库模型定义
|
|
15
|
+
*/
|
|
16
|
+
export const AI_SESSION_MODEL = {
|
|
17
|
+
session_id: { type: 'text', nullable: false },
|
|
18
|
+
messages: { type: 'json', default: [] },
|
|
19
|
+
config: { type: 'json', default: {} },
|
|
20
|
+
created_at: { type: 'integer', default: 0 },
|
|
21
|
+
updated_at: { type: 'integer', default: 0 },
|
|
22
|
+
};
|
|
23
|
+
/**
|
|
24
|
+
* 内存会话管理器(回退方案)
|
|
25
|
+
*/
|
|
26
|
+
export class MemorySessionManager {
|
|
27
|
+
sessions = new Map();
|
|
28
|
+
config;
|
|
29
|
+
cleanupTimer;
|
|
30
|
+
constructor(config = {}) {
|
|
31
|
+
this.config = {
|
|
32
|
+
maxHistory: config.maxHistory ?? 100,
|
|
33
|
+
expireMs: config.expireMs ?? 24 * 60 * 60 * 1000, // 24 小时
|
|
34
|
+
};
|
|
35
|
+
// 定期清理过期会话
|
|
36
|
+
this.cleanupTimer = setInterval(() => this.cleanup(), 5 * 60 * 1000);
|
|
37
|
+
}
|
|
38
|
+
get(sessionId, config) {
|
|
39
|
+
let session = this.sessions.get(sessionId);
|
|
40
|
+
if (!session) {
|
|
41
|
+
session = {
|
|
42
|
+
id: sessionId,
|
|
43
|
+
config: config || { provider: 'openai' },
|
|
44
|
+
messages: [],
|
|
45
|
+
createdAt: Date.now(),
|
|
46
|
+
updatedAt: Date.now(),
|
|
47
|
+
};
|
|
48
|
+
this.sessions.set(sessionId, session);
|
|
49
|
+
}
|
|
50
|
+
else {
|
|
51
|
+
session.updatedAt = Date.now();
|
|
52
|
+
}
|
|
53
|
+
return session;
|
|
54
|
+
}
|
|
55
|
+
has(sessionId) {
|
|
56
|
+
return this.sessions.has(sessionId);
|
|
57
|
+
}
|
|
58
|
+
addMessage(sessionId, message) {
|
|
59
|
+
const session = this.get(sessionId);
|
|
60
|
+
session.messages.push(message);
|
|
61
|
+
session.updatedAt = Date.now();
|
|
62
|
+
this.trimMessages(session);
|
|
63
|
+
}
|
|
64
|
+
trimMessages(session) {
|
|
65
|
+
const maxHistory = session.config.maxHistory ?? this.config.maxHistory;
|
|
66
|
+
if (session.messages.length > maxHistory) {
|
|
67
|
+
const systemMessages = session.messages.filter(m => m.role === 'system');
|
|
68
|
+
const otherMessages = session.messages.filter(m => m.role !== 'system');
|
|
69
|
+
const keepCount = maxHistory - systemMessages.length;
|
|
70
|
+
session.messages = [...systemMessages, ...otherMessages.slice(-keepCount)];
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
getMessages(sessionId) {
|
|
74
|
+
return this.sessions.get(sessionId)?.messages || [];
|
|
75
|
+
}
|
|
76
|
+
setSystemPrompt(sessionId, prompt) {
|
|
77
|
+
const session = this.get(sessionId);
|
|
78
|
+
session.messages = session.messages.filter(m => m.role !== 'system');
|
|
79
|
+
session.messages.unshift({ role: 'system', content: prompt });
|
|
80
|
+
session.updatedAt = Date.now();
|
|
81
|
+
}
|
|
82
|
+
clear(sessionId) {
|
|
83
|
+
return this.sessions.delete(sessionId);
|
|
84
|
+
}
|
|
85
|
+
reset(sessionId) {
|
|
86
|
+
const session = this.sessions.get(sessionId);
|
|
87
|
+
if (session) {
|
|
88
|
+
const systemMessages = session.messages.filter(m => m.role === 'system');
|
|
89
|
+
session.messages = systemMessages;
|
|
90
|
+
session.updatedAt = Date.now();
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
listSessions() {
|
|
94
|
+
return Array.from(this.sessions.keys());
|
|
95
|
+
}
|
|
96
|
+
getStats() {
|
|
97
|
+
const now = Date.now();
|
|
98
|
+
let active = 0;
|
|
99
|
+
let expired = 0;
|
|
100
|
+
for (const session of this.sessions.values()) {
|
|
101
|
+
if (now - session.updatedAt > this.config.expireMs) {
|
|
102
|
+
expired++;
|
|
103
|
+
}
|
|
104
|
+
else {
|
|
105
|
+
active++;
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
return { total: this.sessions.size, active, expired };
|
|
109
|
+
}
|
|
110
|
+
cleanup() {
|
|
111
|
+
const now = Date.now();
|
|
112
|
+
let cleaned = 0;
|
|
113
|
+
for (const [id, session] of this.sessions) {
|
|
114
|
+
const expireMs = session.config.expireMs ?? this.config.expireMs;
|
|
115
|
+
if (now - session.updatedAt > expireMs) {
|
|
116
|
+
this.sessions.delete(id);
|
|
117
|
+
cleaned++;
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
return cleaned;
|
|
121
|
+
}
|
|
122
|
+
dispose() {
|
|
123
|
+
if (this.cleanupTimer) {
|
|
124
|
+
clearInterval(this.cleanupTimer);
|
|
125
|
+
this.cleanupTimer = undefined;
|
|
126
|
+
}
|
|
127
|
+
this.sessions.clear();
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
/**
|
|
131
|
+
* 数据库会话管理器
|
|
132
|
+
* 使用 Zhin 的数据库服务进行持久化存储
|
|
133
|
+
*/
|
|
134
|
+
export class DatabaseSessionManager {
|
|
135
|
+
cache = new Map();
|
|
136
|
+
config;
|
|
137
|
+
cleanupTimer;
|
|
138
|
+
saveQueue = new Map();
|
|
139
|
+
saveTimer;
|
|
140
|
+
model; // 数据库模型
|
|
141
|
+
constructor(model, config = {}) {
|
|
142
|
+
this.model = model;
|
|
143
|
+
this.config = {
|
|
144
|
+
maxHistory: config.maxHistory ?? 200, // 数据库支持更长的历史
|
|
145
|
+
expireMs: config.expireMs ?? 7 * 24 * 60 * 60 * 1000, // 7 天过期
|
|
146
|
+
};
|
|
147
|
+
// 定期清理过期会话(每小时)
|
|
148
|
+
this.cleanupTimer = setInterval(() => this.cleanup(), 60 * 60 * 1000);
|
|
149
|
+
}
|
|
150
|
+
/**
|
|
151
|
+
* 从数据库加载会话
|
|
152
|
+
*/
|
|
153
|
+
async loadSession(sessionId) {
|
|
154
|
+
try {
|
|
155
|
+
const records = await this.model.select().where({ session_id: sessionId });
|
|
156
|
+
if (records && records.length > 0) {
|
|
157
|
+
const record = records[0];
|
|
158
|
+
// SQLite 中 json 类型存储为 TEXT,读回时需要解析
|
|
159
|
+
const messages = typeof record.messages === 'string'
|
|
160
|
+
? JSON.parse(record.messages)
|
|
161
|
+
: (record.messages || []);
|
|
162
|
+
const config = typeof record.config === 'string'
|
|
163
|
+
? JSON.parse(record.config)
|
|
164
|
+
: (record.config || { provider: 'openai' });
|
|
165
|
+
return {
|
|
166
|
+
id: record.session_id,
|
|
167
|
+
config: Array.isArray(config) ? { provider: 'openai' } : config,
|
|
168
|
+
messages: Array.isArray(messages) ? messages : [],
|
|
169
|
+
createdAt: record.created_at,
|
|
170
|
+
updatedAt: record.updated_at,
|
|
171
|
+
};
|
|
172
|
+
}
|
|
173
|
+
}
|
|
174
|
+
catch (error) {
|
|
175
|
+
logger.debug('加载会话失败:', error);
|
|
176
|
+
}
|
|
177
|
+
return null;
|
|
178
|
+
}
|
|
179
|
+
/**
|
|
180
|
+
* 保存会话到数据库(防抖)
|
|
181
|
+
*/
|
|
182
|
+
schedulesSave(session) {
|
|
183
|
+
this.saveQueue.set(session.id, session);
|
|
184
|
+
if (!this.saveTimer) {
|
|
185
|
+
this.saveTimer = setTimeout(() => this.flushSaveQueue(), 1000);
|
|
186
|
+
}
|
|
187
|
+
}
|
|
188
|
+
/**
|
|
189
|
+
* 批量保存队列中的会话
|
|
190
|
+
*/
|
|
191
|
+
async flushSaveQueue() {
|
|
192
|
+
this.saveTimer = undefined;
|
|
193
|
+
const sessions = Array.from(this.saveQueue.values());
|
|
194
|
+
this.saveQueue.clear();
|
|
195
|
+
for (const session of sessions) {
|
|
196
|
+
try {
|
|
197
|
+
const existing = await this.model.select().where({ session_id: session.id });
|
|
198
|
+
const record = {
|
|
199
|
+
session_id: session.id,
|
|
200
|
+
messages: session.messages,
|
|
201
|
+
config: session.config,
|
|
202
|
+
updated_at: session.updatedAt,
|
|
203
|
+
};
|
|
204
|
+
if (existing && existing.length > 0) {
|
|
205
|
+
await this.model.update(record).where({ session_id: session.id });
|
|
206
|
+
}
|
|
207
|
+
else {
|
|
208
|
+
record.created_at = session.createdAt;
|
|
209
|
+
await this.model.create(record);
|
|
210
|
+
}
|
|
211
|
+
}
|
|
212
|
+
catch (error) {
|
|
213
|
+
logger.debug(`保存会话 ${session.id} 失败:`, error);
|
|
214
|
+
}
|
|
215
|
+
}
|
|
216
|
+
}
|
|
217
|
+
async get(sessionId, config) {
|
|
218
|
+
// 先检查缓存
|
|
219
|
+
let session = this.cache.get(sessionId);
|
|
220
|
+
if (!session) {
|
|
221
|
+
// 从数据库加载
|
|
222
|
+
session = await this.loadSession(sessionId) ?? undefined;
|
|
223
|
+
if (!session) {
|
|
224
|
+
// 创建新会话
|
|
225
|
+
session = {
|
|
226
|
+
id: sessionId,
|
|
227
|
+
config: config || { provider: 'openai' },
|
|
228
|
+
messages: [],
|
|
229
|
+
createdAt: Date.now(),
|
|
230
|
+
updatedAt: Date.now(),
|
|
231
|
+
};
|
|
232
|
+
}
|
|
233
|
+
this.cache.set(sessionId, session);
|
|
234
|
+
}
|
|
235
|
+
session.updatedAt = Date.now();
|
|
236
|
+
this.schedulesSave(session);
|
|
237
|
+
return session;
|
|
238
|
+
}
|
|
239
|
+
async has(sessionId) {
|
|
240
|
+
if (this.cache.has(sessionId)) {
|
|
241
|
+
return true;
|
|
242
|
+
}
|
|
243
|
+
try {
|
|
244
|
+
const records = await this.model.select().where({ session_id: sessionId });
|
|
245
|
+
return records && records.length > 0;
|
|
246
|
+
}
|
|
247
|
+
catch {
|
|
248
|
+
return false;
|
|
249
|
+
}
|
|
250
|
+
}
|
|
251
|
+
async addMessage(sessionId, message) {
|
|
252
|
+
const session = await this.get(sessionId);
|
|
253
|
+
session.messages.push(message);
|
|
254
|
+
session.updatedAt = Date.now();
|
|
255
|
+
this.trimMessages(session);
|
|
256
|
+
this.schedulesSave(session);
|
|
257
|
+
}
|
|
258
|
+
trimMessages(session) {
|
|
259
|
+
const maxHistory = session.config.maxHistory ?? this.config.maxHistory;
|
|
260
|
+
if (session.messages.length > maxHistory) {
|
|
261
|
+
const systemMessages = session.messages.filter(m => m.role === 'system');
|
|
262
|
+
const otherMessages = session.messages.filter(m => m.role !== 'system');
|
|
263
|
+
const keepCount = maxHistory - systemMessages.length;
|
|
264
|
+
session.messages = [...systemMessages, ...otherMessages.slice(-keepCount)];
|
|
265
|
+
}
|
|
266
|
+
}
|
|
267
|
+
async getMessages(sessionId) {
|
|
268
|
+
const session = await this.get(sessionId);
|
|
269
|
+
return session.messages;
|
|
270
|
+
}
|
|
271
|
+
async setSystemPrompt(sessionId, prompt) {
|
|
272
|
+
const session = await this.get(sessionId);
|
|
273
|
+
session.messages = session.messages.filter(m => m.role !== 'system');
|
|
274
|
+
session.messages.unshift({ role: 'system', content: prompt });
|
|
275
|
+
session.updatedAt = Date.now();
|
|
276
|
+
this.schedulesSave(session);
|
|
277
|
+
}
|
|
278
|
+
async clear(sessionId) {
|
|
279
|
+
this.cache.delete(sessionId);
|
|
280
|
+
this.saveQueue.delete(sessionId);
|
|
281
|
+
try {
|
|
282
|
+
await this.model.delete({ session_id: sessionId });
|
|
283
|
+
return true;
|
|
284
|
+
}
|
|
285
|
+
catch (error) {
|
|
286
|
+
logger.debug(`删除会话 ${sessionId} 失败:`, error);
|
|
287
|
+
return false;
|
|
288
|
+
}
|
|
289
|
+
}
|
|
290
|
+
async reset(sessionId) {
|
|
291
|
+
const session = await this.get(sessionId);
|
|
292
|
+
const systemMessages = session.messages.filter(m => m.role === 'system');
|
|
293
|
+
session.messages = systemMessages;
|
|
294
|
+
session.updatedAt = Date.now();
|
|
295
|
+
this.schedulesSave(session);
|
|
296
|
+
}
|
|
297
|
+
async listSessions() {
|
|
298
|
+
try {
|
|
299
|
+
const records = await this.model.select();
|
|
300
|
+
return records.map(r => r.session_id);
|
|
301
|
+
}
|
|
302
|
+
catch (error) {
|
|
303
|
+
logger.debug('列出会话失败:', error);
|
|
304
|
+
return Array.from(this.cache.keys());
|
|
305
|
+
}
|
|
306
|
+
}
|
|
307
|
+
async getStats() {
|
|
308
|
+
const now = Date.now();
|
|
309
|
+
let total = 0;
|
|
310
|
+
let active = 0;
|
|
311
|
+
let expired = 0;
|
|
312
|
+
try {
|
|
313
|
+
const records = await this.model.select();
|
|
314
|
+
total = records.length;
|
|
315
|
+
for (const record of records) {
|
|
316
|
+
const expireMs = record.config?.expireMs ?? this.config.expireMs;
|
|
317
|
+
if (now - record.updated_at > expireMs) {
|
|
318
|
+
expired++;
|
|
319
|
+
}
|
|
320
|
+
else {
|
|
321
|
+
active++;
|
|
322
|
+
}
|
|
323
|
+
}
|
|
324
|
+
}
|
|
325
|
+
catch (error) {
|
|
326
|
+
logger.debug('获取统计失败:', error);
|
|
327
|
+
// 回退到缓存统计
|
|
328
|
+
total = this.cache.size;
|
|
329
|
+
for (const session of this.cache.values()) {
|
|
330
|
+
if (now - session.updatedAt > this.config.expireMs) {
|
|
331
|
+
expired++;
|
|
332
|
+
}
|
|
333
|
+
else {
|
|
334
|
+
active++;
|
|
335
|
+
}
|
|
336
|
+
}
|
|
337
|
+
}
|
|
338
|
+
return { total, active, expired };
|
|
339
|
+
}
|
|
340
|
+
async cleanup() {
|
|
341
|
+
const now = Date.now();
|
|
342
|
+
let cleaned = 0;
|
|
343
|
+
try {
|
|
344
|
+
const records = await this.model.select();
|
|
345
|
+
for (const record of records) {
|
|
346
|
+
const expireMs = record.config?.expireMs ?? this.config.expireMs;
|
|
347
|
+
if (now - record.updated_at > expireMs) {
|
|
348
|
+
await this.model.delete({ session_id: record.session_id });
|
|
349
|
+
this.cache.delete(record.session_id);
|
|
350
|
+
cleaned++;
|
|
351
|
+
}
|
|
352
|
+
}
|
|
353
|
+
}
|
|
354
|
+
catch (error) {
|
|
355
|
+
logger.debug('清理会话失败:', error);
|
|
356
|
+
}
|
|
357
|
+
return cleaned;
|
|
358
|
+
}
|
|
359
|
+
async dispose() {
|
|
360
|
+
// 保存所有待保存的会话
|
|
361
|
+
if (this.saveTimer) {
|
|
362
|
+
clearTimeout(this.saveTimer);
|
|
363
|
+
this.saveTimer = undefined;
|
|
364
|
+
}
|
|
365
|
+
await this.flushSaveQueue();
|
|
366
|
+
if (this.cleanupTimer) {
|
|
367
|
+
clearInterval(this.cleanupTimer);
|
|
368
|
+
this.cleanupTimer = undefined;
|
|
369
|
+
}
|
|
370
|
+
this.cache.clear();
|
|
371
|
+
}
|
|
372
|
+
}
|
|
373
|
+
/**
|
|
374
|
+
* 会话管理器包装器
|
|
375
|
+
* 支持同步和异步接口的统一使用
|
|
376
|
+
*/
|
|
377
|
+
export class SessionManager {
|
|
378
|
+
manager;
|
|
379
|
+
constructor(manager) {
|
|
380
|
+
this.manager = manager;
|
|
381
|
+
}
|
|
382
|
+
/**
|
|
383
|
+
* 生成会话 ID
|
|
384
|
+
*/
|
|
385
|
+
static generateId(platform, userId, channelId) {
|
|
386
|
+
return channelId
|
|
387
|
+
? `${platform}:${channelId}:${userId}`
|
|
388
|
+
: `${platform}:${userId}`;
|
|
389
|
+
}
|
|
390
|
+
get(sessionId, config) {
|
|
391
|
+
return this.manager.get(sessionId, config);
|
|
392
|
+
}
|
|
393
|
+
has(sessionId) {
|
|
394
|
+
return this.manager.has(sessionId);
|
|
395
|
+
}
|
|
396
|
+
addMessage(sessionId, message) {
|
|
397
|
+
return this.manager.addMessage(sessionId, message);
|
|
398
|
+
}
|
|
399
|
+
getMessages(sessionId) {
|
|
400
|
+
return this.manager.getMessages(sessionId);
|
|
401
|
+
}
|
|
402
|
+
setSystemPrompt(sessionId, prompt) {
|
|
403
|
+
return this.manager.setSystemPrompt(sessionId, prompt);
|
|
404
|
+
}
|
|
405
|
+
clear(sessionId) {
|
|
406
|
+
return this.manager.clear(sessionId);
|
|
407
|
+
}
|
|
408
|
+
reset(sessionId) {
|
|
409
|
+
return this.manager.reset(sessionId);
|
|
410
|
+
}
|
|
411
|
+
listSessions() {
|
|
412
|
+
return this.manager.listSessions();
|
|
413
|
+
}
|
|
414
|
+
getStats() {
|
|
415
|
+
return this.manager.getStats();
|
|
416
|
+
}
|
|
417
|
+
cleanup() {
|
|
418
|
+
return this.manager.cleanup();
|
|
419
|
+
}
|
|
420
|
+
dispose() {
|
|
421
|
+
return this.manager.dispose();
|
|
422
|
+
}
|
|
423
|
+
}
|
|
424
|
+
/**
|
|
425
|
+
* 创建内存会话管理器(回退方案)
|
|
426
|
+
*/
|
|
427
|
+
export function createMemorySessionManager(config) {
|
|
428
|
+
return new SessionManager(new MemorySessionManager(config));
|
|
429
|
+
}
|
|
430
|
+
/**
|
|
431
|
+
* 创建数据库会话管理器
|
|
432
|
+
*/
|
|
433
|
+
export function createDatabaseSessionManager(model, config) {
|
|
434
|
+
return new SessionManager(new DatabaseSessionManager(model, config));
|
|
435
|
+
}
|
|
436
|
+
/**
|
|
437
|
+
* 创建会话管理器(向后兼容)
|
|
438
|
+
* @deprecated 使用 createMemorySessionManager 或 createDatabaseSessionManager
|
|
439
|
+
*/
|
|
440
|
+
export function createSessionManager(config) {
|
|
441
|
+
return createMemorySessionManager(config);
|
|
442
|
+
}
|
|
443
|
+
//# sourceMappingURL=session.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.js","sourceRoot":"","sources":["../../src/ai/session.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAGzC,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;AAE9C;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B,UAAU,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,QAAQ,EAAE,KAAK,EAAE;IACtD,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,EAAE,EAAE;IAChD,MAAM,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,EAAE,EAAE;IAC9C,UAAU,EAAE,EAAE,IAAI,EAAE,SAAkB,EAAE,OAAO,EAAE,CAAC,EAAE;IACpD,UAAU,EAAE,EAAE,IAAI,EAAE,SAAkB,EAAE,OAAO,EAAE,CAAC,EAAE;CACrD,CAAC;AA+BF;;GAEG;AACH,MAAM,OAAO,oBAAoB;IACvB,QAAQ,GAAyB,IAAI,GAAG,EAAE,CAAC;IAC3C,MAAM,CAA2D;IACjE,YAAY,CAAkC;IAEtD,YAAY,SAAqD,EAAE;QACjE,IAAI,CAAC,MAAM,GAAG;YACZ,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,GAAG;YACpC,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,QAAQ;SAC3D,CAAC;QAEF,WAAW;QACX,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;IACvE,CAAC;IAED,GAAG,CAAC,SAAiB,EAAE,MAAsB;QAC3C,IAAI,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAE3C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG;gBACR,EAAE,EAAE,SAAS;gBACb,MAAM,EAAE,MAAM,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE;gBACxC,QAAQ,EAAE,EAAE;gBACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;YACF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,GAAG,CAAC,SAAiB;QACnB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACtC,CAAC;IAED,UAAU,CAAC,SAAiB,EAAE,OAAoB;QAChD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACpC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/B,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IAC7B,CAAC;IAEO,YAAY,CAAC,OAAgB;QACnC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QACvE,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,UAAU,EAAE,CAAC;YACzC,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YACzE,MAAM,aAAa,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YACxE,MAAM,SAAS,GAAG,UAAU,GAAG,cAAc,CAAC,MAAM,CAAC;YACrD,OAAO,CAAC,QAAQ,GAAG,CAAC,GAAG,cAAc,EAAE,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QAC7E,CAAC;IACH,CAAC;IAED,WAAW,CAAC,SAAiB;QAC3B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;IACtD,CAAC;IAED,eAAe,CAAC,SAAiB,EAAE,MAAc;QAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACpC,OAAO,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACrE,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAC9D,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,SAAiB;QACrB,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,SAAiB;QACrB,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YACzE,OAAO,CAAC,QAAQ,GAAG,cAAc,CAAC;YAClC,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,CAAC;IACH,CAAC;IAED,YAAY;QACV,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,QAAQ;QACN,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,IAAI,GAAG,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACnD,OAAO,EAAE,CAAC;YACZ,CAAC;iBAAM,CAAC;gBACN,MAAM,EAAE,CAAC;YACX,CAAC;QACH,CAAC;QAED,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;IACxD,CAAC;IAED,OAAO;QACL,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC1C,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;YACjE,IAAI,GAAG,GAAG,OAAO,CAAC,SAAS,GAAG,QAAQ,EAAE,CAAC;gBACvC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACzB,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,OAAO;QACL,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACjC,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,sBAAsB;IACzB,KAAK,GAAyB,IAAI,GAAG,EAAE,CAAC;IACxC,MAAM,CAA2D;IACjE,YAAY,CAAkC;IAC9C,SAAS,GAAyB,IAAI,GAAG,EAAE,CAAC;IAC5C,SAAS,CAAiC;IAC1C,KAAK,CAAM,CAAC,QAAQ;IAE5B,YACE,KAAU,EACV,SAAqD,EAAE;QAEvD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG;YACZ,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,GAAG,EAAE,aAAa;YACnD,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,QAAQ;SAC/D,CAAC;QAEF,gBAAgB;QAChB,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;IACxE,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW,CAAC,SAAiB;QACzC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;YAC3E,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClC,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAkB,CAAC;gBAC3C,mCAAmC;gBACnC,MAAM,QAAQ,GAAG,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ;oBAClD,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC;oBAC7B,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;gBAC5B,MAAM,MAAM,GAAG,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ;oBAC9C,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC;oBAC3B,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;gBAC9C,OAAO;oBACL,EAAE,EAAE,MAAM,CAAC,UAAU;oBACrB,MAAM,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,MAAM;oBAC/D,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE;oBACjD,SAAS,EAAE,MAAM,CAAC,UAAU;oBAC5B,SAAS,EAAE,MAAM,CAAC,UAAU;iBAC7B,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACjC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,OAAgB;QACpC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAExC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,IAAI,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc;QAC1B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAE3B,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;QACrD,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QAEvB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC7E,MAAM,MAAM,GAA2B;oBACrC,UAAU,EAAE,OAAO,CAAC,EAAE;oBACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ;oBAC1B,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,UAAU,EAAE,OAAO,CAAC,SAAS;iBAC9B,CAAC;gBAEF,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACpC,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC;gBACpE,CAAC;qBAAM,CAAC;oBACN,MAAM,CAAC,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC;oBACtC,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,KAAK,CAAC,QAAQ,OAAO,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,SAAiB,EAAE,MAAsB;QACjD,QAAQ;QACR,IAAI,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAExC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,SAAS;YACT,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC;YAEzD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,QAAQ;gBACR,OAAO,GAAG;oBACR,EAAE,EAAE,SAAS;oBACb,MAAM,EAAE,MAAM,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE;oBACxC,QAAQ,EAAE,EAAE;oBACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;oBACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iBACtB,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAE5B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,SAAiB;QACzB,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;YAC3E,OAAO,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;QACvC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,SAAiB,EAAE,OAAoB;QACtD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1C,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/B,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC3B,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IAEO,YAAY,CAAC,OAAgB;QACnC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QACvE,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,UAAU,EAAE,CAAC;YACzC,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YACzE,MAAM,aAAa,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YACxE,MAAM,SAAS,GAAG,UAAU,GAAG,cAAc,CAAC,MAAM,CAAC;YACrD,OAAO,CAAC,QAAQ,GAAG,CAAC,GAAG,cAAc,EAAE,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QAC7E,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,SAAiB;QACjC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1C,OAAO,OAAO,CAAC,QAAQ,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,SAAiB,EAAE,MAAc;QACrD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1C,OAAO,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACrE,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAC9D,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,SAAiB;QAC3B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC7B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAEjC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;YACnD,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,QAAQ,SAAS,MAAM,EAAE,KAAK,CAAC,CAAC;YAC7C,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,SAAiB;QAC3B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACzE,OAAO,CAAC,QAAQ,GAAG,cAAc,CAAC;QAClC,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YAC1C,OAAQ,OAA2B,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QAC7D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YAC/B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,EAAqB,CAAC;YAC7D,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC;YAEvB,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,EAAE,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;gBACjE,IAAI,GAAG,GAAG,MAAM,CAAC,UAAU,GAAG,QAAQ,EAAE,CAAC;oBACvC,OAAO,EAAE,CAAC;gBACZ,CAAC;qBAAM,CAAC;oBACN,MAAM,EAAE,CAAC;gBACX,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YAC/B,UAAU;YACV,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;YACxB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC1C,IAAI,GAAG,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;oBACnD,OAAO,EAAE,CAAC;gBACZ,CAAC;qBAAM,CAAC;oBACN,MAAM,EAAE,CAAC;gBACX,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,OAAO,GAAG,CAAC,CAAC;QAEhB,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,EAAqB,CAAC;YAE7D,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,EAAE,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;gBACjE,IAAI,GAAG,GAAG,MAAM,CAAC,UAAU,GAAG,QAAQ,EAAE,CAAC;oBACvC,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC;oBAC3D,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;oBACrC,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACjC,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,OAAO;QACX,aAAa;QACb,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC7B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC7B,CAAC;QACD,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAE5B,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACjC,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;QAChC,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,cAAc;IACjB,OAAO,CAAkB;IAEjC,YAAY,OAAwB;QAClC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,UAAU,CAAC,QAAgB,EAAE,MAAc,EAAE,SAAkB;QACpE,OAAO,SAAS;YACd,CAAC,CAAC,GAAG,QAAQ,IAAI,SAAS,IAAI,MAAM,EAAE;YACtC,CAAC,CAAC,GAAG,QAAQ,IAAI,MAAM,EAAE,CAAC;IAC9B,CAAC;IAED,GAAG,CAAC,SAAiB,EAAE,MAAsB;QAC3C,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAC7C,CAAC;IAED,GAAG,CAAC,SAAiB;QACnB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACrC,CAAC;IAED,UAAU,CAAC,SAAiB,EAAE,OAAoB;QAChD,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IACrD,CAAC;IAED,WAAW,CAAC,SAAiB;QAC3B,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IAC7C,CAAC;IAED,eAAe,CAAC,SAAiB,EAAE,MAAc;QAC/C,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,SAAiB;QACrB,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,SAAiB;QACrB,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;IACrC,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;IACjC,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IAChC,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IAChC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,0BAA0B,CAAC,MAAmD;IAC5F,OAAO,IAAI,cAAc,CAAC,IAAI,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAC;AAC9D,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,4BAA4B,CAC1C,KAAU,EACV,MAAmD;IAEnD,OAAO,IAAI,cAAc,CAAC,IAAI,sBAAsB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;AACvE,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAAC,MAAmD;IACtF,OAAO,0BAA0B,CAAC,MAAM,CAAC,CAAC;AAC5C,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ToneDetector — 轻量级情绪/语气检测
|
|
3
|
+
*
|
|
4
|
+
* 通过标点符号、emoji 密度、关键词分析用户语气,
|
|
5
|
+
* 生成一条 hint 注入 system prompt,让 AI 的回复匹配用户情绪。
|
|
6
|
+
*
|
|
7
|
+
* 零 LLM 开销,纯正则/统计分析。
|
|
8
|
+
*/
|
|
9
|
+
export type Tone = 'neutral' | 'frustrated' | 'excited' | 'questioning' | 'sad' | 'urgent';
|
|
10
|
+
interface ToneResult {
|
|
11
|
+
tone: Tone;
|
|
12
|
+
hint: string;
|
|
13
|
+
}
|
|
14
|
+
/**
|
|
15
|
+
* 检测用户消息的情绪语气
|
|
16
|
+
*/
|
|
17
|
+
export declare function detectTone(message: string): ToneResult;
|
|
18
|
+
export {};
|
|
19
|
+
//# sourceMappingURL=tone-detector.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tone-detector.d.ts","sourceRoot":"","sources":["../../src/ai/tone-detector.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,MAAM,MAAM,IAAI,GAAG,SAAS,GAAG,YAAY,GAAG,SAAS,GAAG,aAAa,GAAG,KAAK,GAAG,QAAQ,CAAC;AAE3F,UAAU,UAAU;IAClB,IAAI,EAAE,IAAI,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;CACd;AAQD;;GAEG;AACH,wBAAgB,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,UAAU,CA+DtD"}
|