@ai-setting/roy-agent-core 1.3.10 → 1.3.14
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/dist/config/index.js +1647 -0
- package/dist/index.js +12579 -89691
- package/package.json +19 -56
- package/src/config/config-component.test.ts +0 -627
- package/src/config/config-component.ts +0 -906
- package/src/config/config-parser.test.ts +0 -319
- package/src/config/config-parser.ts +0 -203
- package/src/config/decentralized-config.test.ts +0 -740
- package/src/config/env-key.ts +0 -210
- package/src/config/env-source.test.ts +0 -252
- package/src/config/env-source.ts +0 -301
- package/src/config/file-source.test.ts +0 -357
- package/src/config/file-source.ts +0 -421
- package/src/config/index.ts +0 -24
- package/src/config/protocol-resolver.test.ts +0 -217
- package/src/config/protocol-resolver.ts +0 -228
- package/src/env/agent/agent-component.abort.test.ts +0 -511
- package/src/env/agent/agent-component.record-session.test.ts +0 -349
- package/src/env/agent/agent-component.test.ts +0 -1389
- package/src/env/agent/agent-component.tool-error.test.ts +0 -327
- package/src/env/agent/agent-component.ts +0 -1711
- package/src/env/agent/agent-config-registration.test.ts +0 -226
- package/src/env/agent/agent-config-registration.ts +0 -46
- package/src/env/agent/agent-reminder-plugin.integration.test.ts +0 -243
- package/src/env/agent/index.ts +0 -10
- package/src/env/agent/summary-agent.parse-hint.test.ts +0 -360
- package/src/env/agent/summary-agent.ts +0 -508
- package/src/env/agent/types.ts +0 -536
- package/src/env/commands/commands-component.test.ts +0 -364
- package/src/env/commands/commands-component.ts +0 -604
- package/src/env/commands/commands-config-registration.test.ts +0 -198
- package/src/env/commands/commands-config-registration.ts +0 -38
- package/src/env/commands/index.ts +0 -21
- package/src/env/commands/parser.test.ts +0 -203
- package/src/env/commands/parser.ts +0 -115
- package/src/env/commands/types.ts +0 -184
- package/src/env/commands-prompt-integration.test.ts +0 -243
- package/src/env/component-env.test.ts +0 -119
- package/src/env/component.ts +0 -335
- package/src/env/constants.test.ts +0 -72
- package/src/env/constants.ts +0 -123
- package/src/env/debug/debug-component.test.ts +0 -114
- package/src/env/debug/debug-component.ts +0 -547
- package/src/env/debug/formatters/index.ts +0 -9
- package/src/env/debug/formatters/repl-formatter.test.ts +0 -139
- package/src/env/debug/formatters/repl-formatter.ts +0 -358
- package/src/env/debug/formatters/trace-formatter.test.ts +0 -119
- package/src/env/debug/formatters/trace-formatter.ts +0 -191
- package/src/env/debug/formatters/tree-formatter.test.ts +0 -107
- package/src/env/debug/formatters/tree-formatter.ts +0 -325
- package/src/env/debug/index.ts +0 -38
- package/src/env/debug/parser/regex-parser.test.ts +0 -201
- package/src/env/debug/parser/regex-parser.ts +0 -196
- package/src/env/debug/parser/span-builder.test.ts +0 -241
- package/src/env/debug/parser/span-builder.ts +0 -386
- package/src/env/debug/reader/log-reader.test.ts +0 -170
- package/src/env/debug/reader/log-reader.ts +0 -186
- package/src/env/debug/reader/span-db-reader.test.ts +0 -118
- package/src/env/debug/reader/span-db-reader.ts +0 -201
- package/src/env/debug/types.test.ts +0 -187
- package/src/env/debug/types.ts +0 -171
- package/src/env/environment-init.test.ts +0 -183
- package/src/env/environment-lifecycle.test.ts +0 -516
- package/src/env/environment-service.test.ts +0 -332
- package/src/env/environment.handle-query.test.ts +0 -96
- package/src/env/environment.test.ts +0 -232
- package/src/env/environment.ts +0 -708
- package/src/env/errors.test.ts +0 -165
- package/src/env/errors.ts +0 -157
- package/src/env/event-source/event-source-agent-handler.test.ts +0 -193
- package/src/env/event-source/event-source-agent-handler.ts +0 -111
- package/src/env/event-source/event-source-component.process-cleanup.test.ts +0 -236
- package/src/env/event-source/event-source-component.stop.test.ts +0 -346
- package/src/env/event-source/event-source-component.test.ts +0 -1207
- package/src/env/event-source/event-source-component.ts +0 -1379
- package/src/env/event-source/event-source-config-registration.test.ts +0 -242
- package/src/env/event-source/event-source-config-registration.ts +0 -37
- package/src/env/event-source/event-source-integration.test.ts +0 -320
- package/src/env/event-source/event-source-platform.test.ts +0 -630
- package/src/env/event-source/types.ts +0 -298
- package/src/env/hook/global-hook-manager.ts +0 -162
- package/src/env/hook/hook-manager.test.ts +0 -374
- package/src/env/hook/hook-manager.ts +0 -309
- package/src/env/hook/index.ts +0 -38
- package/src/env/hook/types.ts +0 -138
- package/src/env/index.ts +0 -144
- package/src/env/interface.ts +0 -203
- package/src/env/llm/hooks.test.ts +0 -293
- package/src/env/llm/hooks.ts +0 -316
- package/src/env/llm/index.ts +0 -61
- package/src/env/llm/invoke-threshold-check.test.ts +0 -88
- package/src/env/llm/invoke-timeout.test.ts +0 -54
- package/src/env/llm/invoke.test.ts +0 -71
- package/src/env/llm/invoke.ts +0 -1039
- package/src/env/llm/llm-config.test.ts +0 -523
- package/src/env/llm/llm.test.ts +0 -233
- package/src/env/llm/llm.ts +0 -568
- package/src/env/llm/provider.test.ts +0 -182
- package/src/env/llm/provider.ts +0 -108
- package/src/env/llm/transform.test.ts +0 -251
- package/src/env/llm/transform.ts +0 -286
- package/src/env/llm/types.test.ts +0 -580
- package/src/env/llm/types.ts +0 -424
- package/src/env/log-trace/decorator-otel.test.ts +0 -182
- package/src/env/log-trace/decorator.ts +0 -230
- package/src/env/log-trace/index.ts +0 -79
- package/src/env/log-trace/log-trace-component.test.ts +0 -242
- package/src/env/log-trace/log-trace-component.ts +0 -497
- package/src/env/log-trace/log-trace-config-registration.test.ts +0 -348
- package/src/env/log-trace/log-trace-config-registration.ts +0 -45
- package/src/env/log-trace/logger.test.ts +0 -149
- package/src/env/log-trace/logger.ts +0 -522
- package/src/env/log-trace/opentelemetry/cli-propagation.test.ts +0 -147
- package/src/env/log-trace/opentelemetry/cli-propagation.ts +0 -194
- package/src/env/log-trace/opentelemetry/integration.test.ts +0 -668
- package/src/env/log-trace/opentelemetry/mod.ts +0 -25
- package/src/env/log-trace/opentelemetry/propagation-env.test.ts +0 -181
- package/src/env/log-trace/opentelemetry/propagation-env.ts +0 -136
- package/src/env/log-trace/opentelemetry/propagation.test.ts +0 -259
- package/src/env/log-trace/opentelemetry/propagation.ts +0 -215
- package/src/env/log-trace/opentelemetry/tracer-provider-context.test.ts +0 -166
- package/src/env/log-trace/opentelemetry/tracer-provider.test.ts +0 -379
- package/src/env/log-trace/opentelemetry/tracer-provider.ts +0 -612
- package/src/env/log-trace/span-storage.test.ts +0 -145
- package/src/env/log-trace/span-storage.ts +0 -230
- package/src/env/log-trace/trace-context.test.ts +0 -187
- package/src/env/log-trace/trace-context.ts +0 -162
- package/src/env/log-trace/types.test.ts +0 -63
- package/src/env/log-trace/types.ts +0 -172
- package/src/env/mcp/README.md +0 -244
- package/src/env/mcp/__integration__/mcp-component.integration.test.ts +0 -373
- package/src/env/mcp/config.test.ts +0 -74
- package/src/env/mcp/config.ts +0 -116
- package/src/env/mcp/index.ts +0 -41
- package/src/env/mcp/loader.test.ts +0 -161
- package/src/env/mcp/loader.ts +0 -209
- package/src/env/mcp/mcp-component.test.ts +0 -111
- package/src/env/mcp/mcp-component.ts +0 -358
- package/src/env/mcp/mcp-config-registration.test.ts +0 -304
- package/src/env/mcp/mcp-config-registration.ts +0 -50
- package/src/env/mcp/scanner.test.ts +0 -170
- package/src/env/mcp/scanner.ts +0 -246
- package/src/env/mcp/tool/adapter.test.ts +0 -520
- package/src/env/mcp/tool/adapter.ts +0 -521
- package/src/env/mcp/tool/index.ts +0 -5
- package/src/env/mcp/types.test.ts +0 -171
- package/src/env/mcp/types.ts +0 -79
- package/src/env/memory/README.md +0 -177
- package/src/env/memory/built-in/index.ts +0 -59
- package/src/env/memory/built-in/recall-memory.ts +0 -103
- package/src/env/memory/built-in/record-memory.ts +0 -148
- package/src/env/memory/index.ts +0 -20
- package/src/env/memory/memory-component.test.ts +0 -239
- package/src/env/memory/memory-component.ts +0 -503
- package/src/env/memory/memory-config-registration.test.ts +0 -67
- package/src/env/memory/memory-config-registration.ts +0 -48
- package/src/env/memory/memory-config.ts +0 -45
- package/src/env/memory/memory-file.test.ts +0 -268
- package/src/env/memory/plugin/index.ts +0 -48
- package/src/env/memory/plugin/memory-agent.test.ts +0 -249
- package/src/env/memory/plugin/memory-agent.ts +0 -365
- package/src/env/memory/plugin/memory-manager.ts +0 -198
- package/src/env/memory/plugin/memory-plugin-agent.test.ts +0 -145
- package/src/env/memory/plugin/memory-plugin.ts +0 -210
- package/src/env/memory/plugin/plugin-simplified.test.ts +0 -51
- package/src/env/memory/plugin/recall-memory.test.ts +0 -106
- package/src/env/memory/plugin/recall-memory.ts +0 -53
- package/src/env/memory/plugin/types.ts +0 -101
- package/src/env/memory/tools/memory-agent-tools.ts +0 -228
- package/src/env/memory/types.ts +0 -85
- package/src/env/paths.ts +0 -118
- package/src/env/prompt/index.ts +0 -18
- package/src/env/prompt/memory-prompts.test.ts +0 -91
- package/src/env/prompt/prompt-component.test.ts +0 -491
- package/src/env/prompt/prompt-component.ts +0 -619
- package/src/env/prompt/prompt-config-registration.test.ts +0 -213
- package/src/env/prompt/prompt-config-registration.ts +0 -39
- package/src/env/prompt/prompts-index.ts +0 -504
- package/src/env/prompt/renderer.ts +0 -67
- package/src/env/prompt/types.ts +0 -136
- package/src/env/session/hooks.ts +0 -18
- package/src/env/session/index.ts +0 -37
- package/src/env/session/search-query-parser.test.ts +0 -425
- package/src/env/session/search-query-parser.ts +0 -171
- package/src/env/session/session-checkpoint.test.ts +0 -523
- package/src/env/session/session-component.extract-recent-messages.test.ts +0 -209
- package/src/env/session/session-component.test.ts +0 -132
- package/src/env/session/session-component.ts +0 -1249
- package/src/env/session/session-config-registration.test.ts +0 -138
- package/src/env/session/session-config-registration.ts +0 -52
- package/src/env/session/session-message-converter.test.ts +0 -763
- package/src/env/session/session-message-converter.ts +0 -415
- package/src/env/session/session-message-e2e.test.ts +0 -448
- package/src/env/session/session-search.test.ts +0 -391
- package/src/env/session/session-store.test.ts +0 -362
- package/src/env/session/session-store.ts +0 -141
- package/src/env/session/storage/index.ts +0 -6
- package/src/env/session/storage/memory.ts +0 -502
- package/src/env/session/storage/sqlite.ts +0 -794
- package/src/env/session/types.ts +0 -742
- package/src/env/skill/config.ts +0 -39
- package/src/env/skill/index.ts +0 -6
- package/src/env/skill/parser.test.ts +0 -116
- package/src/env/skill/parser.ts +0 -77
- package/src/env/skill/scanner.test.ts +0 -211
- package/src/env/skill/scanner.ts +0 -119
- package/src/env/skill/skill-component.test.ts +0 -234
- package/src/env/skill/skill-component.ts +0 -352
- package/src/env/skill/skill-config-registration.test.ts +0 -60
- package/src/env/skill/skill-config-registration.ts +0 -43
- package/src/env/skill/tool/index.ts +0 -1
- package/src/env/skill/tool/skill-tool.test.ts +0 -100
- package/src/env/skill/tool/skill-tool.ts +0 -72
- package/src/env/skill/types.ts +0 -64
- package/src/env/task/delegate/delegate-tool.test.ts +0 -498
- package/src/env/task/delegate/delegate-tool.ts +0 -1014
- package/src/env/task/delegate/index.ts +0 -18
- package/src/env/task/delegate/stop-tool.test.ts +0 -140
- package/src/env/task/delegate/stop-tool.ts +0 -119
- package/src/env/task/delegate/task-events.test.ts +0 -178
- package/src/env/task/delegate/task-events.ts +0 -143
- package/src/env/task/hooks/contexts.test.ts +0 -92
- package/src/env/task/hooks/contexts.ts +0 -192
- package/src/env/task/hooks/index.ts +0 -23
- package/src/env/task/hooks/task-hook-points.test.ts +0 -32
- package/src/env/task/hooks/task-hook-points.ts +0 -54
- package/src/env/task/index.ts +0 -7
- package/src/env/task/plugins/index.ts +0 -13
- package/src/env/task/plugins/task-plugin.test.ts +0 -74
- package/src/env/task/plugins/task-plugin.ts +0 -89
- package/src/env/task/plugins/task-tag-plugin.test.ts +0 -377
- package/src/env/task/plugins/task-tag-plugin.ts +0 -319
- package/src/env/task/plugins/task-workflow-extractor.integration.test.ts +0 -226
- package/src/env/task/plugins/workflow-extractor-agent.test.ts +0 -107
- package/src/env/task/plugins/workflow-extractor-agent.ts +0 -225
- package/src/env/task/storage/index.ts +0 -6
- package/src/env/task/storage/sqlite-task-store.test.ts +0 -283
- package/src/env/task/storage/sqlite-task-store.ts +0 -903
- package/src/env/task/storage/task-search.test.ts +0 -291
- package/src/env/task/tag-service.test.ts +0 -198
- package/src/env/task/tag-service.ts +0 -264
- package/src/env/task/task-component.test.ts +0 -193
- package/src/env/task/task-component.ts +0 -658
- package/src/env/task/task-config-registration.test.ts +0 -57
- package/src/env/task/task-config-registration.ts +0 -37
- package/src/env/task/task-types.test.ts +0 -137
- package/src/env/task/tools/complete-tool.ts +0 -44
- package/src/env/task/tools/create-tool.ts +0 -49
- package/src/env/task/tools/delete-tool.ts +0 -43
- package/src/env/task/tools/get-tool.ts +0 -59
- package/src/env/task/tools/index.ts +0 -10
- package/src/env/task/tools/list-tool.ts +0 -40
- package/src/env/task/tools/operation/create-tool.ts +0 -48
- package/src/env/task/tools/operation/delete-tool.ts +0 -43
- package/src/env/task/tools/operation/get-tool.ts +0 -43
- package/src/env/task/tools/operation/index.ts +0 -9
- package/src/env/task/tools/operation/list-tool.ts +0 -40
- package/src/env/task/tools/operation/operation-tools.test.ts +0 -274
- package/src/env/task/tools/operation/operation-types.ts +0 -75
- package/src/env/task/tools/operation/update-tool.ts +0 -47
- package/src/env/task/tools/task-tools.test.ts +0 -203
- package/src/env/task/tools/task-types.test.ts +0 -75
- package/src/env/task/tools/task-types.ts +0 -68
- package/src/env/task/tools/update-tool.ts +0 -70
- package/src/env/task/types.ts +0 -160
- package/src/env/tool/built-in/bash.ts +0 -201
- package/src/env/tool/built-in/echo.ts +0 -29
- package/src/env/tool/built-in/edit-file.test.ts +0 -136
- package/src/env/tool/built-in/edit-file.ts +0 -92
- package/src/env/tool/built-in/glob.test.ts +0 -94
- package/src/env/tool/built-in/glob.ts +0 -65
- package/src/env/tool/built-in/grep.test.ts +0 -122
- package/src/env/tool/built-in/grep.ts +0 -108
- package/src/env/tool/built-in/index.ts +0 -44
- package/src/env/tool/built-in/read-file.test.ts +0 -84
- package/src/env/tool/built-in/read-file.ts +0 -75
- package/src/env/tool/built-in/write-file.test.ts +0 -119
- package/src/env/tool/built-in/write-file.ts +0 -68
- package/src/env/tool/index.ts +0 -24
- package/src/env/tool/registry.test.ts +0 -257
- package/src/env/tool/registry.ts +0 -167
- package/src/env/tool/tool-component.test.ts +0 -559
- package/src/env/tool/tool-component.ts +0 -563
- package/src/env/tool/tool-config-registration.test.ts +0 -249
- package/src/env/tool/tool-config-registration.ts +0 -46
- package/src/env/tool/types.ts +0 -267
- package/src/env/tool/validator.test.ts +0 -143
- package/src/env/tool/validator.ts +0 -44
- package/src/env/types.ts +0 -180
- package/src/env/workflow/ask-user-tool-registration.test.ts +0 -216
- package/src/env/workflow/complex-workflow.integration.test.ts +0 -1900
- package/src/env/workflow/decorators/decorator-node.ts +0 -229
- package/src/env/workflow/decorators/decorator.test.ts +0 -196
- package/src/env/workflow/decorators/edge.ts +0 -82
- package/src/env/workflow/decorators/index.ts +0 -31
- package/src/env/workflow/decorators/node-as.ts +0 -98
- package/src/env/workflow/decorators/workflow.ts +0 -54
- package/src/env/workflow/engine/dag-manager.test.ts +0 -570
- package/src/env/workflow/engine/dag-manager.ts +0 -594
- package/src/env/workflow/engine/engine.ts +0 -1422
- package/src/env/workflow/engine/event-bus.test.ts +0 -359
- package/src/env/workflow/engine/event-bus.ts +0 -156
- package/src/env/workflow/engine/executor-agent-session.test.ts +0 -84
- package/src/env/workflow/engine/executor.test.ts +0 -619
- package/src/env/workflow/engine/executor.ts +0 -593
- package/src/env/workflow/engine/index.ts +0 -24
- package/src/env/workflow/engine/node-registry.test.ts +0 -560
- package/src/env/workflow/engine/node-registry.ts +0 -289
- package/src/env/workflow/engine/resume-removed.test.ts +0 -22
- package/src/env/workflow/engine/scheduler.test.ts +0 -715
- package/src/env/workflow/engine/scheduler.ts +0 -318
- package/src/env/workflow/engine/workflow-engine.test.ts +0 -815
- package/src/env/workflow/extractor/workflow-converter.ts +0 -306
- package/src/env/workflow/fixtures.ts +0 -380
- package/src/env/workflow/index.ts +0 -38
- package/src/env/workflow/integration/run-resume-unified.test.ts +0 -186
- package/src/env/workflow/integration/service-integration.test.ts +0 -267
- package/src/env/workflow/metadata/keys.ts +0 -12
- package/src/env/workflow/nodes/agent-component-adapter.test.ts +0 -318
- package/src/env/workflow/nodes/agent-component-adapter.ts +0 -448
- package/src/env/workflow/nodes/agent-node.test.ts +0 -371
- package/src/env/workflow/nodes/agent-node.ts +0 -598
- package/src/env/workflow/nodes/ask-user-node.ts +0 -113
- package/src/env/workflow/nodes/condition-node.ts +0 -200
- package/src/env/workflow/nodes/index.ts +0 -9
- package/src/env/workflow/nodes/merge-node.ts +0 -141
- package/src/env/workflow/nodes/skill-node.test.ts +0 -253
- package/src/env/workflow/nodes/skill-node.ts +0 -393
- package/src/env/workflow/nodes/tool-node.test.ts +0 -251
- package/src/env/workflow/nodes/tool-node.ts +0 -493
- package/src/env/workflow/nodes/workflow-llm-history.test.ts +0 -455
- package/src/env/workflow/nodes/workflow-node.test.ts +0 -315
- package/src/env/workflow/nodes/workflow-node.ts +0 -311
- package/src/env/workflow/service/index.ts +0 -27
- package/src/env/workflow/service/registry.test.ts +0 -133
- package/src/env/workflow/service/registry.ts +0 -71
- package/src/env/workflow/service/workflow-service.test.ts +0 -310
- package/src/env/workflow/service/workflow-service.ts +0 -393
- package/src/env/workflow/storage/index.ts +0 -28
- package/src/env/workflow/storage/mock-repositories.ts +0 -385
- package/src/env/workflow/storage/sqlite.test.ts +0 -179
- package/src/env/workflow/storage/sqlite.ts +0 -163
- package/src/env/workflow/storage/workflow-repo.test.ts +0 -780
- package/src/env/workflow/storage/workflow-repo.ts +0 -342
- package/src/env/workflow/tools/ask-user-tool.ts +0 -82
- package/src/env/workflow/tools/index.ts +0 -26
- package/src/env/workflow/tools/run-workflow.test.ts +0 -352
- package/src/env/workflow/tools/run-workflow.ts +0 -214
- package/src/env/workflow/types/context.ts +0 -18
- package/src/env/workflow/types/decorators-types.ts +0 -198
- package/src/env/workflow/types/event.test.ts +0 -515
- package/src/env/workflow/types/event.ts +0 -193
- package/src/env/workflow/types/index.ts +0 -49
- package/src/env/workflow/types/run.test.ts +0 -437
- package/src/env/workflow/types/run.ts +0 -173
- package/src/env/workflow/types/workflow-hil.ts +0 -114
- package/src/env/workflow/types/workflow-message.test.ts +0 -138
- package/src/env/workflow/types/workflow-message.ts +0 -196
- package/src/env/workflow/types/workflow-session.test.ts +0 -95
- package/src/env/workflow/types/workflow-session.ts +0 -59
- package/src/env/workflow/types/workflow.test.ts +0 -495
- package/src/env/workflow/types/workflow.ts +0 -195
- package/src/env/workflow/types_compat.ts +0 -51
- package/src/env/workflow/utils/create-workflow.ts +0 -47
- package/src/env/workflow/utils/execution-state.ts +0 -245
- package/src/env/workflow/utils/index.ts +0 -18
- package/src/env/workflow/utils/node-registry-helper.ts +0 -58
- package/src/env/workflow/utils/recovery-validator.test.ts +0 -460
- package/src/env/workflow/utils/recovery-validator.ts +0 -377
- package/src/env/workflow/utils/session-parser.test.ts +0 -111
- package/src/env/workflow/utils/session-parser.ts +0 -94
- package/src/env/workflow/utils/session-recovery.test.ts +0 -334
- package/src/env/workflow/utils/session-recovery.ts +0 -188
- package/src/env/workflow/utils/template-resolver.test.ts +0 -258
- package/src/env/workflow/utils/template-resolver.ts +0 -436
- package/src/env/workflow/utils/validation-rules.ts +0 -149
- package/src/env/workflow/workflow-component.ts +0 -544
- package/src/index.ts +0 -422
- package/src/utils/id.ts +0 -21
|
@@ -1,365 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @fileoverview Memory Agent 核心逻辑 (Simplified)
|
|
3
|
-
*
|
|
4
|
-
* 简化设计:
|
|
5
|
-
* - 移除 DraftManager(草稿管理已废弃)
|
|
6
|
-
* - 移除 AccessLog(访问日志已废弃)
|
|
7
|
-
* - 只保留核心的 Agent 类型和格式化逻辑
|
|
8
|
-
*/
|
|
9
|
-
|
|
10
|
-
import type { ModelMessage } from "ai";
|
|
11
|
-
|
|
12
|
-
// ============================================================================
|
|
13
|
-
// Type Definitions
|
|
14
|
-
// ============================================================================
|
|
15
|
-
|
|
16
|
-
/**
|
|
17
|
-
* Memory 作用域类型
|
|
18
|
-
*/
|
|
19
|
-
export type MemoryScope = "project" | "global";
|
|
20
|
-
|
|
21
|
-
/**
|
|
22
|
-
* 项目摘要信息
|
|
23
|
-
*/
|
|
24
|
-
export interface ProjectSummary {
|
|
25
|
-
/** 项目根目录路径 */
|
|
26
|
-
projectPath: string;
|
|
27
|
-
/** 项目名称 */
|
|
28
|
-
projectName: string;
|
|
29
|
-
/** 最后更新时间戳 */
|
|
30
|
-
lastUpdated: number;
|
|
31
|
-
/** 项目 memory.md 摘要(一段话) */
|
|
32
|
-
summary: string;
|
|
33
|
-
/** 简短摘要(几个词) */
|
|
34
|
-
shortSummary: string;
|
|
35
|
-
/** 项目 memory.md 路径 */
|
|
36
|
-
memoryMdPath: string;
|
|
37
|
-
}
|
|
38
|
-
|
|
39
|
-
/**
|
|
40
|
-
* Memory Agent 输入
|
|
41
|
-
*/
|
|
42
|
-
export interface MemoryAgentInput {
|
|
43
|
-
/** 作用域 */
|
|
44
|
-
scope: MemoryScope;
|
|
45
|
-
/** 基础目录(project: .roy/memory, global: ~/.config/roy-agent/memory) */
|
|
46
|
-
baseDir: string;
|
|
47
|
-
/** 当前 memory.md 内容 */
|
|
48
|
-
currentMemoryMd: string;
|
|
49
|
-
/** 当前对话消息 */
|
|
50
|
-
messages: ModelMessage[];
|
|
51
|
-
/** 项目摘要列表(仅 global scope 使用) */
|
|
52
|
-
projectSummaries?: ProjectSummary[];
|
|
53
|
-
}
|
|
54
|
-
|
|
55
|
-
/**
|
|
56
|
-
* Memory Agent 输出结果
|
|
57
|
-
*/
|
|
58
|
-
export interface MemoryAgentResult {
|
|
59
|
-
/** 作用域 */
|
|
60
|
-
scope: MemoryScope;
|
|
61
|
-
/** 新的 memory.md 内容(全文) */
|
|
62
|
-
memoryMdContent: string;
|
|
63
|
-
/** 项目摘要(仅 project scope 输出) */
|
|
64
|
-
projectSummary?: ProjectSummary;
|
|
65
|
-
/** 是否有更新 */
|
|
66
|
-
hasUpdates: boolean;
|
|
67
|
-
}
|
|
68
|
-
|
|
69
|
-
/**
|
|
70
|
-
* Memory Agent 配置
|
|
71
|
-
*/
|
|
72
|
-
export interface MemoryAgentConfig {
|
|
73
|
-
/** Agent 名称 */
|
|
74
|
-
name: string;
|
|
75
|
-
/** System prompt */
|
|
76
|
-
systemPrompt: string;
|
|
77
|
-
/** 最大迭代次数 */
|
|
78
|
-
maxIterations: number;
|
|
79
|
-
}
|
|
80
|
-
|
|
81
|
-
// ============================================================================
|
|
82
|
-
// System Prompts (Simplified)
|
|
83
|
-
// ============================================================================
|
|
84
|
-
|
|
85
|
-
/**
|
|
86
|
-
* Project Memory Agent System Prompt
|
|
87
|
-
*/
|
|
88
|
-
export const PROJECT_MEMORY_AGENT_PROMPT = `你是项目级 Memory Agent,负责整理和更新当前项目的记忆。
|
|
89
|
-
|
|
90
|
-
## 你的职责
|
|
91
|
-
|
|
92
|
-
1. **信息收集**:在提取记忆前,先收集项目上下文信息
|
|
93
|
-
2. **从对话中提取记忆**:分析 messages,提取与项目相关的知识、决策、解决方案
|
|
94
|
-
3. **生成 memory.md**:更新项目记忆索引
|
|
95
|
-
|
|
96
|
-
## 信息收集步骤(重要!)
|
|
97
|
-
|
|
98
|
-
在提取记忆之前,请先使用可用工具收集项目上下文:
|
|
99
|
-
|
|
100
|
-
1. **阅读项目 README**:了解项目概述、功能、使用方法
|
|
101
|
-
- 路径示例:README.md, README.zh-CN.md, docs/README.md
|
|
102
|
-
|
|
103
|
-
2. **检查设计文档**:了解项目架构和设计决策
|
|
104
|
-
- 路径示例:docs/design.md, docs/ARCHITECTURE.md, docs/superpowers/specs/*.md
|
|
105
|
-
|
|
106
|
-
3. **查看代码实现**:了解关键模块和实现细节
|
|
107
|
-
- 路径示例:src/**/*.ts, packages/*/src/**/*.ts
|
|
108
|
-
|
|
109
|
-
4. **回顾现有 memory**:了解已有的项目记忆
|
|
110
|
-
- 路径:当前目录下的 .roy/memory/memory.md
|
|
111
|
-
|
|
112
|
-
## 输入信息
|
|
113
|
-
|
|
114
|
-
- 当前项目的 memory.md 内容
|
|
115
|
-
- 当前对话消息
|
|
116
|
-
|
|
117
|
-
## 输出要求
|
|
118
|
-
|
|
119
|
-
请以 JSON 格式输出结果,包含以下字段:
|
|
120
|
-
- memoryMdContent: 新的 memory.md 内容
|
|
121
|
-
- projectSummary: 项目摘要(包含 summary, shortSummary 等)
|
|
122
|
-
- hasUpdates: 是否有更新
|
|
123
|
-
|
|
124
|
-
## 记忆组织原则
|
|
125
|
-
|
|
126
|
-
- 优先记录具体、可操作的记忆(代码位置、配置项、命令)
|
|
127
|
-
- 避免过于抽象或通用的描述
|
|
128
|
-
- 记录设计决策及其理由`;
|
|
129
|
-
|
|
130
|
-
/**
|
|
131
|
-
* Global Memory Agent System Prompt
|
|
132
|
-
*/
|
|
133
|
-
export const GLOBAL_MEMORY_AGENT_PROMPT = `你是全局 Memory Agent,负责整理跨项目的通用记忆。
|
|
134
|
-
|
|
135
|
-
## 你的职责
|
|
136
|
-
|
|
137
|
-
1. **聚合项目摘要**:分析 projectSummaries,提取共性知识
|
|
138
|
-
2. **维护全局记忆**:更新全局 memory.md
|
|
139
|
-
3. **维护项目索引**:在 memory.md 中维护项目列表
|
|
140
|
-
4. **提取共性经验**:将项目经验提炼为通用最佳实践
|
|
141
|
-
|
|
142
|
-
## 输入信息
|
|
143
|
-
|
|
144
|
-
- 全局 memory.md 内容
|
|
145
|
-
- 当前对话消息
|
|
146
|
-
- 项目摘要列表(projectSummaries)
|
|
147
|
-
|
|
148
|
-
## 输出要求
|
|
149
|
-
|
|
150
|
-
请以 JSON 格式输出结果,包含以下字段:
|
|
151
|
-
- memoryMdContent: 新的 memory.md 内容
|
|
152
|
-
- hasUpdates: 是否有更新
|
|
153
|
-
|
|
154
|
-
## 记忆组织原则
|
|
155
|
-
|
|
156
|
-
- 关注跨项目共性(工作流、规范、偏好)
|
|
157
|
-
- 项目索引只记录摘要和路径,不复制内容
|
|
158
|
-
- 用户个人偏好(饮食、工作习惯)优先记录`;
|
|
159
|
-
|
|
160
|
-
// ============================================================================
|
|
161
|
-
// Core Functions
|
|
162
|
-
// ============================================================================
|
|
163
|
-
|
|
164
|
-
/**
|
|
165
|
-
* 创建 Memory Agent 配置
|
|
166
|
-
*/
|
|
167
|
-
export function createMemoryAgent(scope: MemoryScope): MemoryAgentConfig {
|
|
168
|
-
const name = `memory-agent-${scope}-${Date.now()}`;
|
|
169
|
-
const systemPrompt = scope === "project"
|
|
170
|
-
? PROJECT_MEMORY_AGENT_PROMPT
|
|
171
|
-
: GLOBAL_MEMORY_AGENT_PROMPT;
|
|
172
|
-
|
|
173
|
-
return {
|
|
174
|
-
name,
|
|
175
|
-
systemPrompt,
|
|
176
|
-
maxIterations: 5,
|
|
177
|
-
};
|
|
178
|
-
}
|
|
179
|
-
|
|
180
|
-
/**
|
|
181
|
-
* 格式化 Memory Agent 输入为字符串
|
|
182
|
-
*/
|
|
183
|
-
export function formatMemoryInput(input: MemoryAgentInput): string {
|
|
184
|
-
const lines: string[] = [];
|
|
185
|
-
|
|
186
|
-
// Scope header
|
|
187
|
-
lines.push(`# Memory Agent Input (${input.scope} scope)\n`);
|
|
188
|
-
|
|
189
|
-
// Current memory.md
|
|
190
|
-
lines.push("## Current memory.md");
|
|
191
|
-
lines.push(input.currentMemoryMd || "_No existing memory_");
|
|
192
|
-
lines.push("");
|
|
193
|
-
|
|
194
|
-
// Messages
|
|
195
|
-
lines.push("## Recent Messages");
|
|
196
|
-
if (input.messages.length === 0) {
|
|
197
|
-
lines.push("_No messages_");
|
|
198
|
-
} else {
|
|
199
|
-
for (const msg of input.messages.slice(-10)) {
|
|
200
|
-
const content = typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content);
|
|
201
|
-
lines.push(`**${msg.role}**: ${content.substring(0, 500)}`);
|
|
202
|
-
}
|
|
203
|
-
}
|
|
204
|
-
lines.push("");
|
|
205
|
-
|
|
206
|
-
// Project Summaries (for global scope)
|
|
207
|
-
if (input.scope === "global" && input.projectSummaries && input.projectSummaries.length > 0) {
|
|
208
|
-
lines.push("## Project Summaries");
|
|
209
|
-
for (const ps of input.projectSummaries) {
|
|
210
|
-
lines.push(`### ${ps.projectName}`);
|
|
211
|
-
lines.push(`Path: ${ps.projectPath}`);
|
|
212
|
-
lines.push(`Last Updated: ${new Date(ps.lastUpdated).toLocaleString()}`);
|
|
213
|
-
lines.push(`Summary: ${ps.summary}`);
|
|
214
|
-
lines.push(`Memory: ${ps.memoryMdPath}`);
|
|
215
|
-
lines.push("---");
|
|
216
|
-
}
|
|
217
|
-
lines.push("");
|
|
218
|
-
}
|
|
219
|
-
|
|
220
|
-
lines.push("## Task");
|
|
221
|
-
lines.push("请分析以上信息,更新记忆并以 JSON 格式输出结果。");
|
|
222
|
-
|
|
223
|
-
return lines.join("\n");
|
|
224
|
-
}
|
|
225
|
-
|
|
226
|
-
/**
|
|
227
|
-
* 解析 Memory Agent 输出
|
|
228
|
-
*/
|
|
229
|
-
export function parseMemoryAgentOutput(output: string, scope: MemoryScope): MemoryAgentResult {
|
|
230
|
-
// Try to extract JSON from output
|
|
231
|
-
let jsonStr = output;
|
|
232
|
-
|
|
233
|
-
// Try to find JSON in the output
|
|
234
|
-
const jsonMatch = output.match(/```json\s*([\s\S]*?)\s*```/) ||
|
|
235
|
-
output.match(/```\s*([\s\S]*?)\s*```/) ||
|
|
236
|
-
output.match(/(\{[\s\S]*\})/);
|
|
237
|
-
|
|
238
|
-
if (jsonMatch) {
|
|
239
|
-
jsonStr = jsonMatch[1] || jsonMatch[0];
|
|
240
|
-
}
|
|
241
|
-
|
|
242
|
-
try {
|
|
243
|
-
const parsed = JSON.parse(jsonStr);
|
|
244
|
-
|
|
245
|
-
return {
|
|
246
|
-
scope,
|
|
247
|
-
memoryMdContent: parsed.memoryMdContent || "",
|
|
248
|
-
projectSummary: parsed.projectSummary,
|
|
249
|
-
hasUpdates: parsed.hasUpdates ?? false,
|
|
250
|
-
};
|
|
251
|
-
} catch {
|
|
252
|
-
// Return empty result on parse failure
|
|
253
|
-
return {
|
|
254
|
-
scope,
|
|
255
|
-
memoryMdContent: output, // Use raw output as fallback
|
|
256
|
-
hasUpdates: false,
|
|
257
|
-
};
|
|
258
|
-
}
|
|
259
|
-
}
|
|
260
|
-
|
|
261
|
-
/**
|
|
262
|
-
* 创建默认的 MemoryAgentResult
|
|
263
|
-
*/
|
|
264
|
-
export function createEmptyResult(scope: MemoryScope): MemoryAgentResult {
|
|
265
|
-
return {
|
|
266
|
-
scope,
|
|
267
|
-
memoryMdContent: "",
|
|
268
|
-
hasUpdates: false,
|
|
269
|
-
};
|
|
270
|
-
}
|
|
271
|
-
|
|
272
|
-
// ============================================================================
|
|
273
|
-
// Input Builders
|
|
274
|
-
// ============================================================================
|
|
275
|
-
|
|
276
|
-
/**
|
|
277
|
-
* 构建 Project Memory Agent 输入
|
|
278
|
-
*/
|
|
279
|
-
export async function buildProjectInput(params: {
|
|
280
|
-
projectPath: string;
|
|
281
|
-
currentMemoryMd: string;
|
|
282
|
-
messages: ModelMessage[];
|
|
283
|
-
}): Promise<MemoryAgentInput> {
|
|
284
|
-
const baseDir = params.projectPath + "/.roy/memory";
|
|
285
|
-
|
|
286
|
-
return {
|
|
287
|
-
scope: "project",
|
|
288
|
-
baseDir,
|
|
289
|
-
currentMemoryMd: params.currentMemoryMd,
|
|
290
|
-
messages: params.messages,
|
|
291
|
-
};
|
|
292
|
-
}
|
|
293
|
-
|
|
294
|
-
/**
|
|
295
|
-
* 构建 Global Memory Agent 输入
|
|
296
|
-
*/
|
|
297
|
-
export function buildGlobalInput(params: {
|
|
298
|
-
baseDir: string;
|
|
299
|
-
currentMemoryMd: string;
|
|
300
|
-
messages: ModelMessage[];
|
|
301
|
-
projectSummaries?: ProjectSummary[];
|
|
302
|
-
}): MemoryAgentInput {
|
|
303
|
-
return {
|
|
304
|
-
scope: "global",
|
|
305
|
-
baseDir: params.baseDir,
|
|
306
|
-
currentMemoryMd: params.currentMemoryMd,
|
|
307
|
-
messages: params.messages,
|
|
308
|
-
projectSummaries: params.projectSummaries,
|
|
309
|
-
};
|
|
310
|
-
}
|
|
311
|
-
|
|
312
|
-
// ============================================================================
|
|
313
|
-
// Result Application
|
|
314
|
-
// ============================================================================
|
|
315
|
-
|
|
316
|
-
/**
|
|
317
|
-
* 应用 Memory Agent 结果到文件系统
|
|
318
|
-
*
|
|
319
|
-
* @param result MemoryAgent 结果
|
|
320
|
-
* @param baseDir 基础目录
|
|
321
|
-
*/
|
|
322
|
-
export async function applyMemoryResult(
|
|
323
|
-
result: MemoryAgentResult,
|
|
324
|
-
baseDir: string
|
|
325
|
-
): Promise<void> {
|
|
326
|
-
const fs = await import("fs/promises");
|
|
327
|
-
const path = await import("path");
|
|
328
|
-
|
|
329
|
-
// Write memory.md
|
|
330
|
-
const memoryMdPath = path.join(baseDir, "memory.md");
|
|
331
|
-
await fs.mkdir(path.dirname(memoryMdPath), { recursive: true });
|
|
332
|
-
await fs.writeFile(memoryMdPath, result.memoryMdContent, "utf-8");
|
|
333
|
-
}
|
|
334
|
-
|
|
335
|
-
// ============================================================================
|
|
336
|
-
// Project Summary Generation
|
|
337
|
-
// ============================================================================
|
|
338
|
-
|
|
339
|
-
/**
|
|
340
|
-
* 生成项目摘要
|
|
341
|
-
*/
|
|
342
|
-
export function generateProjectSummary(params: {
|
|
343
|
-
projectPath: string;
|
|
344
|
-
projectName: string;
|
|
345
|
-
memoryMdContent: string;
|
|
346
|
-
memoryMdPath: string;
|
|
347
|
-
}): ProjectSummary {
|
|
348
|
-
// Extract first paragraph as summary
|
|
349
|
-
const paragraphs = params.memoryMdContent.split(/\n\n+/);
|
|
350
|
-
const firstParagraph = paragraphs[0] || "";
|
|
351
|
-
const summary = firstParagraph.replace(/^#+\s*/g, "").trim() || "No summary available";
|
|
352
|
-
|
|
353
|
-
// Generate short summary (first sentence, max 50 chars)
|
|
354
|
-
const firstSentence = summary.split(/[.!?]/)[0] || summary;
|
|
355
|
-
const shortSummary = firstSentence.substring(0, 50) + (firstSentence.length > 50 ? "..." : "");
|
|
356
|
-
|
|
357
|
-
return {
|
|
358
|
-
projectPath: params.projectPath,
|
|
359
|
-
projectName: params.projectName,
|
|
360
|
-
lastUpdated: Date.now(),
|
|
361
|
-
summary,
|
|
362
|
-
shortSummary,
|
|
363
|
-
memoryMdPath: params.memoryMdPath,
|
|
364
|
-
};
|
|
365
|
-
}
|
|
@@ -1,198 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @fileoverview MemoryManager V3 - Simplified
|
|
3
|
-
*
|
|
4
|
-
* 后台异步任务队列处理
|
|
5
|
-
*
|
|
6
|
-
* 简化设计:
|
|
7
|
-
* - 移除 DraftManager(草稿管理已废弃)
|
|
8
|
-
* - 移除 MemoryOrganizer(整理机制已废弃)
|
|
9
|
-
* - 移除 AccessLog(访问日志已废弃)
|
|
10
|
-
* - 只保留核心的任务队列处理
|
|
11
|
-
*/
|
|
12
|
-
|
|
13
|
-
import type { MemoryTask, ExtractMemoryPayload } from "./types";
|
|
14
|
-
import { createLogger } from "../../log-trace/logger";
|
|
15
|
-
|
|
16
|
-
const logger = createLogger("MemoryManagerV3");
|
|
17
|
-
|
|
18
|
-
/**
|
|
19
|
-
* MemoryManager 配置
|
|
20
|
-
*/
|
|
21
|
-
export interface MemoryManagerConfig {
|
|
22
|
-
/** 回调:当需要提取记忆时调用 */
|
|
23
|
-
onExtractMemory?: (payload: ExtractMemoryPayload) => Promise<void>;
|
|
24
|
-
}
|
|
25
|
-
|
|
26
|
-
/**
|
|
27
|
-
* 生成唯一任务 ID
|
|
28
|
-
*/
|
|
29
|
-
function generateTaskId(): string {
|
|
30
|
-
return `task_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
|
|
31
|
-
}
|
|
32
|
-
|
|
33
|
-
/**
|
|
34
|
-
* MemoryManager V3 - Simplified
|
|
35
|
-
*
|
|
36
|
-
* 简化的后台异步内存处理,只负责任务队列
|
|
37
|
-
*/
|
|
38
|
-
export class MemoryManager {
|
|
39
|
-
/** 队列最大容量,防止内存溢出 */
|
|
40
|
-
private static readonly MAX_QUEUE_SIZE = 100;
|
|
41
|
-
|
|
42
|
-
private queue: MemoryTask[] = [];
|
|
43
|
-
private processing: boolean = false;
|
|
44
|
-
private shuttingDown: boolean = false;
|
|
45
|
-
private config: MemoryManagerConfig;
|
|
46
|
-
private processingPromise: Promise<void> | null = null;
|
|
47
|
-
private resolveProcessing: (() => void) | null = null;
|
|
48
|
-
|
|
49
|
-
constructor(config: MemoryManagerConfig = {}) {
|
|
50
|
-
this.config = config;
|
|
51
|
-
}
|
|
52
|
-
|
|
53
|
-
/**
|
|
54
|
-
* 更新配置
|
|
55
|
-
*/
|
|
56
|
-
updateConfig(config: Partial<MemoryManagerConfig>): void {
|
|
57
|
-
this.config = { ...this.config, ...config };
|
|
58
|
-
}
|
|
59
|
-
|
|
60
|
-
/**
|
|
61
|
-
* 入队(非阻塞)
|
|
62
|
-
*/
|
|
63
|
-
enqueue(task: Omit<MemoryTask, "id">): void {
|
|
64
|
-
if (this.shuttingDown) {
|
|
65
|
-
logger.debug("MemoryManager is shutting down, ignoring new task");
|
|
66
|
-
return;
|
|
67
|
-
}
|
|
68
|
-
|
|
69
|
-
const fullTask: MemoryTask = {
|
|
70
|
-
...task,
|
|
71
|
-
id: generateTaskId(),
|
|
72
|
-
};
|
|
73
|
-
|
|
74
|
-
// 队列大小限制,防止内存溢出
|
|
75
|
-
if (this.queue.length >= MemoryManager.MAX_QUEUE_SIZE) {
|
|
76
|
-
const dropped = this.queue.shift();
|
|
77
|
-
logger.warn(`Queue full (${MemoryManager.MAX_QUEUE_SIZE}), dropping oldest task: ${dropped?.id}`);
|
|
78
|
-
}
|
|
79
|
-
|
|
80
|
-
this.queue.push(fullTask);
|
|
81
|
-
logger.debug(`Task enqueued: ${task.type}, queue size: ${this.queue.length}`);
|
|
82
|
-
|
|
83
|
-
// 如果没有正在处理,启动处理循环
|
|
84
|
-
if (!this.processing) {
|
|
85
|
-
this.startProcessingLoop();
|
|
86
|
-
}
|
|
87
|
-
}
|
|
88
|
-
|
|
89
|
-
/**
|
|
90
|
-
* 启动处理循环
|
|
91
|
-
*/
|
|
92
|
-
private startProcessingLoop(): void {
|
|
93
|
-
const runLoop = async (): Promise<void> => {
|
|
94
|
-
while (this.queue.length > 0 && !this.shuttingDown) {
|
|
95
|
-
const task = this.queue.shift()!;
|
|
96
|
-
try {
|
|
97
|
-
await this.processTask(task);
|
|
98
|
-
} catch (error) {
|
|
99
|
-
logger.error(`Task ${task.id} failed:`, error);
|
|
100
|
-
}
|
|
101
|
-
}
|
|
102
|
-
// 处理完成
|
|
103
|
-
this.processing = false;
|
|
104
|
-
if (this.resolveProcessing) {
|
|
105
|
-
this.resolveProcessing();
|
|
106
|
-
this.resolveProcessing = null;
|
|
107
|
-
this.processingPromise = null;
|
|
108
|
-
}
|
|
109
|
-
};
|
|
110
|
-
|
|
111
|
-
this.processing = true;
|
|
112
|
-
this.processingPromise = new Promise((resolve) => {
|
|
113
|
-
this.resolveProcessing = resolve;
|
|
114
|
-
});
|
|
115
|
-
|
|
116
|
-
setImmediate(() => {
|
|
117
|
-
runLoop().catch((error) => {
|
|
118
|
-
logger.error("Processing loop error:", error);
|
|
119
|
-
this.processing = false;
|
|
120
|
-
});
|
|
121
|
-
});
|
|
122
|
-
}
|
|
123
|
-
|
|
124
|
-
/**
|
|
125
|
-
* 检查是否有待处理任务
|
|
126
|
-
*/
|
|
127
|
-
hasPendingTasks(): boolean {
|
|
128
|
-
return this.queue.length > 0 || this.processing;
|
|
129
|
-
}
|
|
130
|
-
|
|
131
|
-
/**
|
|
132
|
-
* 等待处理完成
|
|
133
|
-
*/
|
|
134
|
-
async waitForCompletion(): Promise<void> {
|
|
135
|
-
// 如果队列为空且没有正在处理,快速路径
|
|
136
|
-
if (this.queue.length === 0 && !this.processing) {
|
|
137
|
-
return;
|
|
138
|
-
}
|
|
139
|
-
|
|
140
|
-
// 如果有待处理的 Promise,等待它
|
|
141
|
-
if (this.processingPromise) {
|
|
142
|
-
await this.processingPromise;
|
|
143
|
-
}
|
|
144
|
-
}
|
|
145
|
-
|
|
146
|
-
/**
|
|
147
|
-
* 准备关闭
|
|
148
|
-
*/
|
|
149
|
-
prepareShutdown(): void {
|
|
150
|
-
this.shuttingDown = true;
|
|
151
|
-
logger.info("MemoryManager preparing for shutdown");
|
|
152
|
-
}
|
|
153
|
-
|
|
154
|
-
/**
|
|
155
|
-
* 释放资源
|
|
156
|
-
*/
|
|
157
|
-
async dispose(): Promise<void> {
|
|
158
|
-
// 等待当前任务完成
|
|
159
|
-
await this.waitForCompletion();
|
|
160
|
-
|
|
161
|
-
// 清空队列
|
|
162
|
-
this.queue = [];
|
|
163
|
-
|
|
164
|
-
logger.info("MemoryManager disposed");
|
|
165
|
-
}
|
|
166
|
-
|
|
167
|
-
/**
|
|
168
|
-
* 处理单个任务
|
|
169
|
-
*/
|
|
170
|
-
private async processTask(task: MemoryTask): Promise<void> {
|
|
171
|
-
logger.debug(`Processing task: ${task.type}`);
|
|
172
|
-
|
|
173
|
-
switch (task.type) {
|
|
174
|
-
case "extract_memory":
|
|
175
|
-
await this.handleExtractMemory(task.payload as ExtractMemoryPayload);
|
|
176
|
-
break;
|
|
177
|
-
default:
|
|
178
|
-
logger.warn(`Unknown task type: ${task.type}`);
|
|
179
|
-
}
|
|
180
|
-
}
|
|
181
|
-
|
|
182
|
-
/**
|
|
183
|
-
* 处理记忆提取
|
|
184
|
-
*/
|
|
185
|
-
private async handleExtractMemory(payload: ExtractMemoryPayload): Promise<void> {
|
|
186
|
-
const messageCount = payload.messages?.length || 0;
|
|
187
|
-
logger.info(`Extracting memory from ${messageCount} messages`);
|
|
188
|
-
|
|
189
|
-
// 调用回调
|
|
190
|
-
if (this.config.onExtractMemory) {
|
|
191
|
-
try {
|
|
192
|
-
await this.config.onExtractMemory(payload);
|
|
193
|
-
} catch (error) {
|
|
194
|
-
logger.error("Memory extraction failed", { error });
|
|
195
|
-
}
|
|
196
|
-
}
|
|
197
|
-
}
|
|
198
|
-
}
|
|
@@ -1,145 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @fileoverview Memory Plugin Agent 集成测试 (Simplified)
|
|
3
|
-
*
|
|
4
|
-
* 简化后的 Memory Agent 集成测试
|
|
5
|
-
*/
|
|
6
|
-
|
|
7
|
-
import { describe, it, expect, beforeEach, afterEach } from "bun:test";
|
|
8
|
-
import path from "path";
|
|
9
|
-
import fs from "fs/promises";
|
|
10
|
-
import os from "os";
|
|
11
|
-
import type {
|
|
12
|
-
MemoryAgentInput,
|
|
13
|
-
MemoryAgentResult,
|
|
14
|
-
ProjectSummary,
|
|
15
|
-
} from "./memory-agent";
|
|
16
|
-
import {
|
|
17
|
-
buildProjectInput,
|
|
18
|
-
buildGlobalInput,
|
|
19
|
-
applyMemoryResult,
|
|
20
|
-
generateProjectSummary,
|
|
21
|
-
} from "./memory-agent";
|
|
22
|
-
|
|
23
|
-
describe("buildProjectInput", () => {
|
|
24
|
-
let testDir: string;
|
|
25
|
-
let projectDir: string;
|
|
26
|
-
|
|
27
|
-
beforeEach(async () => {
|
|
28
|
-
testDir = path.join(os.tmpdir(), `memory-agent-test-${Date.now()}`);
|
|
29
|
-
projectDir = path.join(testDir, "project-a");
|
|
30
|
-
await fs.mkdir(projectDir, { recursive: true });
|
|
31
|
-
});
|
|
32
|
-
|
|
33
|
-
afterEach(async () => {
|
|
34
|
-
await fs.rm(testDir, { recursive: true, force: true }).catch(() => {});
|
|
35
|
-
});
|
|
36
|
-
|
|
37
|
-
it("should build input for project scope", async () => {
|
|
38
|
-
const input = await buildProjectInput({
|
|
39
|
-
projectPath: projectDir,
|
|
40
|
-
currentMemoryMd: "# Project Memory\n\nTest content",
|
|
41
|
-
messages: [
|
|
42
|
-
{ role: "user", content: "Hello" } as any,
|
|
43
|
-
],
|
|
44
|
-
});
|
|
45
|
-
|
|
46
|
-
expect(input.scope).toBe("project");
|
|
47
|
-
expect(input.baseDir).toContain(".roy/memory");
|
|
48
|
-
expect(input.currentMemoryMd).toContain("Project Memory");
|
|
49
|
-
expect(input.messages.length).toBe(1);
|
|
50
|
-
});
|
|
51
|
-
});
|
|
52
|
-
|
|
53
|
-
describe("buildGlobalInput", () => {
|
|
54
|
-
let testDir: string;
|
|
55
|
-
|
|
56
|
-
beforeEach(async () => {
|
|
57
|
-
testDir = path.join(os.tmpdir(), `memory-agent-test-${Date.now()}`);
|
|
58
|
-
await fs.mkdir(path.join(testDir, "memory"), { recursive: true });
|
|
59
|
-
});
|
|
60
|
-
|
|
61
|
-
afterEach(async () => {
|
|
62
|
-
await fs.rm(testDir, { recursive: true, force: true }).catch(() => {});
|
|
63
|
-
});
|
|
64
|
-
|
|
65
|
-
it("should build input for global scope with project summaries", async () => {
|
|
66
|
-
const projectSummaries: ProjectSummary[] = [
|
|
67
|
-
{
|
|
68
|
-
projectPath: "/path/to/project-a",
|
|
69
|
-
projectName: "project-a",
|
|
70
|
-
lastUpdated: Date.now(),
|
|
71
|
-
summary: "Project A summary",
|
|
72
|
-
shortSummary: "A",
|
|
73
|
-
memoryMdPath: "/path/to/project-a/.roy/memory/memory.md",
|
|
74
|
-
},
|
|
75
|
-
];
|
|
76
|
-
|
|
77
|
-
const input = buildGlobalInput({
|
|
78
|
-
baseDir: testDir,
|
|
79
|
-
currentMemoryMd: "# Global Memory",
|
|
80
|
-
messages: [],
|
|
81
|
-
projectSummaries,
|
|
82
|
-
});
|
|
83
|
-
|
|
84
|
-
expect(input.scope).toBe("global");
|
|
85
|
-
expect(input.baseDir).toBe(testDir);
|
|
86
|
-
expect(input.projectSummaries).toHaveLength(1);
|
|
87
|
-
expect(input.projectSummaries![0].projectName).toBe("project-a");
|
|
88
|
-
});
|
|
89
|
-
});
|
|
90
|
-
|
|
91
|
-
describe("applyMemoryResult", () => {
|
|
92
|
-
let testDir: string;
|
|
93
|
-
|
|
94
|
-
beforeEach(async () => {
|
|
95
|
-
testDir = path.join(os.tmpdir(), `memory-agent-test-${Date.now()}`);
|
|
96
|
-
});
|
|
97
|
-
|
|
98
|
-
afterEach(async () => {
|
|
99
|
-
await fs.rm(testDir, { recursive: true, force: true }).catch(() => {});
|
|
100
|
-
});
|
|
101
|
-
|
|
102
|
-
it("should write memory.md content", async () => {
|
|
103
|
-
const result: MemoryAgentResult = {
|
|
104
|
-
scope: "project",
|
|
105
|
-
memoryMdContent: "# Updated Memory\n\nNew content",
|
|
106
|
-
hasUpdates: true,
|
|
107
|
-
};
|
|
108
|
-
|
|
109
|
-
await applyMemoryResult(result, testDir);
|
|
110
|
-
|
|
111
|
-
const memoryPath = path.join(testDir, "memory.md");
|
|
112
|
-
const content = await fs.readFile(memoryPath, "utf-8");
|
|
113
|
-
expect(content).toContain("Updated Memory");
|
|
114
|
-
});
|
|
115
|
-
});
|
|
116
|
-
|
|
117
|
-
describe("generateProjectSummary", () => {
|
|
118
|
-
it("should generate summary with required fields", () => {
|
|
119
|
-
const summary = generateProjectSummary({
|
|
120
|
-
projectPath: "/path/to/project",
|
|
121
|
-
projectName: "test-project",
|
|
122
|
-
memoryMdContent: "# Test Project\n\nSome important content here.",
|
|
123
|
-
memoryMdPath: "/path/to/project/.roy/memory/memory.md",
|
|
124
|
-
});
|
|
125
|
-
|
|
126
|
-
expect(summary.projectPath).toBe("/path/to/project");
|
|
127
|
-
expect(summary.projectName).toBe("test-project");
|
|
128
|
-
expect(summary.lastUpdated).toBeGreaterThan(0);
|
|
129
|
-
expect(summary.summary).toBeTruthy();
|
|
130
|
-
expect(summary.shortSummary).toBeTruthy();
|
|
131
|
-
expect(summary.memoryMdPath).toBe("/path/to/project/.roy/memory/memory.md");
|
|
132
|
-
});
|
|
133
|
-
|
|
134
|
-
it("should generate short summary from memory content", () => {
|
|
135
|
-
const summary = generateProjectSummary({
|
|
136
|
-
projectPath: "/path/to/project",
|
|
137
|
-
projectName: "test",
|
|
138
|
-
memoryMdContent: "# Test Project\n\nThis is a long description of what the project does with many more details.",
|
|
139
|
-
memoryMdPath: "/path/memory.md",
|
|
140
|
-
});
|
|
141
|
-
|
|
142
|
-
// Short summary should be shorter or equal to full summary
|
|
143
|
-
expect(summary.shortSummary.length).toBeLessThanOrEqual(summary.summary.length);
|
|
144
|
-
});
|
|
145
|
-
});
|