feat: 移植6个缺失工具定义和handler到AIAgentService

- memory_introspect — 查看记忆状态统计 - memory_archive — 归档旧记忆 - memory_cleanup — 清理已删除数据 - memory_query_archived — 查询已归档记忆 - context_rewrite — 压缩工具调用上下文 - persona_remove — 删除单条人设属性同时在GraphMemoryService添加： - queryArchived() 方法 - personaRemove() 方法更新系统提示词，加入工具调用规则和context_rewrite使用说明
2026-04-30 16:44:27 +08:00
parent c8571f9d5f
commit 1a4a8dce98
2 changed files with 143 additions and 0 deletions
--- a/entry/src/main/ets/services/AIAgentService.ets
+++ b/entry/src/main/ets/services/AIAgentService.ets
@ -232,7 +232,13 @@ ${personaContext || '你是一个帮助用户记录和回忆信息的助手。
 - memory_recall(queryIntent, seedEntities?, depth?, timeRange?, sessionFilter?): 检索记忆
 - memory_commit(triplets, entityTypes?, sessionId?, turnId?): 写入记忆
 - memory_purge(criteria, mode, newRelation?): 删除/修正记忆
+- memory_introspect(sessionId?): 查看记忆状态统计
+- memory_archive(days?): 归档旧记忆
+- memory_cleanup(dryRun?): 清理已删除数据
+- memory_query_archived(days?, keyword?): 查询已归档记忆
+- context_rewrite(summary): 压缩工具调用上下文
 - persona_update(tone?, style?, personality?, catchphrase?, background?): 更新人设
+- persona_remove(attribute): 删除单条人设属性
 - persona_clear(): 清除人设
 - task_create(taskId, description, infoNodes?): 创建任务
 - task_set_state(taskId, state): 设置任务状态
@ -241,6 +247,11 @@ ${personaContext || '你是一个帮助用户记录和回忆信息的助手。
 - task_archive(taskId, summary?): 归档任务
 - task_query(limit?, stateFilter?): 查询任务列表

+## 工具调用规则
+1. 每轮对话必须按顺序执行：步骤1查询人设 → 步骤2查询工作记忆链 → 步骤3处理请求
+2. context_rewrite 必须单独调用，不能和其他工具在同一轮一起调！
+3. 工具调用 ≥5 次后应使用 context_rewrite 压缩上下文
+
 ## 写入规则
 用户明确表达以下信息时必须写入记忆：
 - 偏好、兴趣
@ -324,12 +335,37 @@ const queryProps: ToolPropertiesDefinition = {
    limit: makeIntegerProp('返回数量，默认10'),
    stateFilter: makeStringProp('状态过滤: 进行中/已完成/已暂停/已取消/archived')
 };
+const introspectProps: ToolPropertiesDefinition = {
+    sessionId: makeStringProp('会话ID（可选）')
+};
+const archiveProps2: ToolPropertiesDefinition = {
+    days: makeIntegerProp('归档天数，默认30')
+};
+const cleanupProps: ToolPropertiesDefinition = {
+    dryRun: makeBoolProp('仅预览不删除')
+};
+const queryArchivedProps: ToolPropertiesDefinition = {
+    days: makeIntegerProp('最近N天内的归档记录'),
+    keyword: makeStringProp('关键词过滤')
+};
+const contextRewriteProps: ToolPropertiesDefinition = {
+    summary: makeStringProp('压缩后的摘要文本，必须包含工具调用元信息')
+};
+const personaRemoveProps: ToolPropertiesDefinition = {
+    attribute: makeStringProp('要删除的属性名（如：扮演角色、说话风格）')
+};

 const TOOLS_DEFINITION: ToolFunctionDef[] = [
    makeToolDef('memory_recall', '检索记忆。支持关键词、种子实体、深度扩展。返回相关实体和关系。', recallProps, ['queryIntent']),
    makeToolDef('memory_commit', '写入记忆。将三元组写入图数据库。', commitProps, ['triplets']),
    makeToolDef('memory_purge', '删除或修正记忆。支持条件删除和纠错替代。', purgeProps, ['criteria', 'mode']),
+    makeToolDef('memory_introspect', '查看记忆状态。返回实体数量、关系数量、热点实体。', introspectProps),
+    makeToolDef('memory_archive', '归档旧记忆。将N天前的非活跃关系标记为归档状态。', archiveProps2, ['days']),
+    makeToolDef('memory_cleanup', '清理无效数据。物理删除已删除状态超过90天的关系和孤立节点。', cleanupProps),
+    makeToolDef('memory_query_archived', '查询已归档的记忆。返回所有 status=archived 的关系记录。', queryArchivedProps),
+    makeToolDef('context_rewrite', '压缩本轮对话的工具调用上下文。将冗长的JSON工具结果提炼为简洁摘要。', contextRewriteProps, ['summary']),
    makeToolDef('persona_update', '更新AI人设属性（语气、风格、性格等）。', personaProps),
+    makeToolDef('persona_remove', '删除单条人设属性。保留其他人设不变。', personaRemoveProps, ['attribute']),
    makeToolDef('persona_clear', '清除所有人设信息。', EMPTY_PROPS),
    makeToolDef('task_create', '创建新的工作记忆任务节点。', createProps, ['taskId', 'description']),
    makeToolDef('task_set_state', '设置任务状态。', setStateProps, ['taskId', 'state']),
@ -353,7 +389,13 @@ type ToolHandlerName =
    | 'memoryRecal'
    | 'memoryCommit'
    | 'memoryPurge'
+    | 'memoryIntrospect'
+    | 'memoryArchive'
+    | 'memoryCleanup'
+    | 'memoryQueryArchived'
+    | 'contextRewrite'
    | 'personaUpdate'
+    | 'personaRemove'
    | 'personaClear'
    | 'taskCreate'
    | 'taskSetState'
@ -366,7 +408,13 @@ const TOOL_HANDLER_MAP: Record<string, ToolHandlerName> = {
    'memory_recall': 'memoryRecal',
    'memory_commit': 'memoryCommit',
    'memory_purge': 'memoryPurge',
+    'memory_introspect': 'memoryIntrospect',
+    'memory_archive': 'memoryArchive',
+    'memory_cleanup': 'memoryCleanup',
+    'memory_query_archived': 'memoryQueryArchived',
+    'context_rewrite': 'contextRewrite',
    'persona_update': 'personaUpdate',
+    'persona_remove': 'personaRemove',
    'persona_clear': 'personaClear',
    'task_create': 'taskCreate',
    'task_set_state': 'taskSetState',
@ -599,6 +647,56 @@ export class AIAgentService {
                    return result;
                }

+                case 'memoryIntrospect': {
+                    const introspectResult = await this.memoryService.memoryIntrospect(args.sessionId as string);
+                    const result: ToolCallResult = {
+                        name: 'memory_introspect',
+                        success: true,
+                        message: `实体: ${introspectResult.entityCount}, 关系: ${introspectResult.relationCount}, 热点: ${introspectResult.hotNodes.length}`
+                    };
+                    return result;
+                }
+
+                case 'memoryArchive': {
+                    const archiveResult = await this.memoryService.archive(args.days as number);
+                    const result: ToolCallResult = {
+                        name: 'memory_archive',
+                        success: true,
+                        message: `已归档 ${archiveResult.archived} 条关系`
+                    };
+                    return result;
+                }
+
+                case 'memoryCleanup': {
+                    const cleanupResult = await this.memoryService.cleanup((args.dryRun as boolean) !== false);
+                    const result: ToolCallResult = {
+                        name: 'memory_cleanup',
+                        success: true,
+                        message: `清理: ${cleanupResult.cleaned} 条关系, ${cleanupResult.deletedOrphans} 个孤儿节点` + (cleanupResult.dryRun ? ' (预览模式)' : '')
+                    };
+                    return result;
+                }
+
+                case 'memoryQueryArchived': {
+                    const qaResult = await this.memoryService.queryArchived(args.days as number, args.keyword as string);
+                    const result: ToolCallResult = {
+                        name: 'memory_query_archived',
+                        success: true,
+                        message: `找到 ${qaResult.length} 条归档记录`
+                    };
+                    return result;
+                }
+
+                case 'contextRewrite': {
+                    const summary = args.summary as string;
+                    const result: ToolCallResult = {
+                        name: 'context_rewrite',
+                        success: summary.includes('[工具调用总结'),
+                        message: summary.includes('[工具调用总结') ? '上下文已压缩' : '格式错误：必须包含[工具调用总结]标记'
+                    };
+                    return result;
+                }
+
                case 'personaUpdate': {
                    const personaParams: PersonaUpdateParams = {
                        tone: args.tone as string,
@ -616,6 +714,16 @@ export class AIAgentService {
                    return result;
                }

+                case 'personaRemove': {
+                    const prResult = await this.memoryService.personaRemove(args.attribute as string);
+                    const result: ToolCallResult = {
+                        name: 'persona_remove',
+                        success: prResult.success,
+                        message: prResult.message
+                    };
+                    return result;
+                }
+
                case 'personaClear': {
                    const pcResult = await this.memoryService.personaClear();
                    const result: ToolCallResult = {
--- a/entry/src/main/ets/services/GraphMemoryService.ets
+++ b/entry/src/main/ets/services/GraphMemoryService.ets
@ -429,6 +429,20 @@ export class GraphMemoryService {
        return cleanupResult;
    }

+    /**
+     * 查询已归档的记忆
+     * 对应 tools.memory_query_archived
+     */
+    async queryArchived(days?: number, keyword?: string): Promise<RelationQueryResult[]> {
+        try {
+            const result = await this.db.queryArchived(days, keyword);
+            return result;
+        } catch (e) {
+            console.error('queryArchived error: ' + JSON.stringify(e));
+            return [];
+        }
+    }
+
    // ========= 人设管理 =========

    /**
@ -521,6 +535,27 @@ export class GraphMemoryService {
        }
    }

+    /**
+     * 删除单条人设属性
+     * 对应 tools.persona_remove
+     */
+    async personaRemove(attribute: string): Promise<PersonaResult> {
+        try {
+            // 将 attribute 转为关系名格式
+            const relationName = 'HAS_PERSONA_' + attribute.toUpperCase();
+            const result = await this.db.purge({ subject: PERSONA_NODE_NAME, relationType: relationName }, 'hard');
+            return {
+                success: result.deleted_count > 0,
+                message: result.deleted_count > 0 ? `已删除人设属性: ${attribute}` : `未找到人设属性: ${attribute}`
+            };
+        } catch (e) {
+            return {
+                success: false,
+                message: `删除人设属性失败: ${(e as Error).message || ''}`
+            };
+        }
+    }
+
    /**
     * 查询当前人设
     */