@chendpoc/pi-memory 0.1.13 → 0.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +66 -258
- package/UBIQUITOUS_LANGUAGE.md +81 -0
- package/dist/adapters/embed/factory.d.ts +7 -0
- package/dist/adapters/embed/factory.d.ts.map +1 -0
- package/dist/adapters/embed/factory.js +44 -0
- package/dist/adapters/embed/factory.js.map +1 -0
- package/dist/adapters/embed/hash.d.ts +3 -0
- package/dist/adapters/embed/hash.d.ts.map +1 -0
- package/dist/adapters/embed/hash.js +26 -0
- package/dist/adapters/embed/hash.js.map +1 -0
- package/dist/adapters/embed/normalize.d.ts +3 -0
- package/dist/adapters/embed/normalize.d.ts.map +1 -0
- package/dist/adapters/embed/normalize.js +15 -0
- package/dist/adapters/embed/normalize.js.map +1 -0
- package/dist/adapters/embed/ollama.d.ts +8 -0
- package/dist/adapters/embed/ollama.d.ts.map +1 -0
- package/dist/adapters/embed/ollama.js +31 -0
- package/dist/adapters/embed/ollama.js.map +1 -0
- package/dist/adapters/embed/openai.d.ts +8 -0
- package/dist/adapters/embed/openai.d.ts.map +1 -0
- package/dist/adapters/embed/openai.js +30 -0
- package/dist/adapters/embed/openai.js.map +1 -0
- package/dist/adapters/embed/types.d.ts +9 -0
- package/dist/adapters/embed/types.d.ts.map +1 -0
- package/dist/adapters/embed/types.js.map +1 -0
- package/dist/adapters/http.d.ts +13 -0
- package/dist/adapters/http.d.ts.map +1 -0
- package/dist/adapters/http.js +31 -0
- package/dist/adapters/http.js.map +1 -0
- package/dist/adapters/index.d.ts +4 -0
- package/dist/adapters/index.d.ts.map +1 -0
- package/dist/adapters/index.js +4 -0
- package/dist/adapters/index.js.map +1 -0
- package/dist/adapters/llm/extractText.d.ts +5 -0
- package/dist/adapters/llm/extractText.d.ts.map +1 -0
- package/dist/adapters/llm/extractText.js +7 -0
- package/dist/adapters/llm/extractText.js.map +1 -0
- package/dist/adapters/llm/factory.d.ts +11 -0
- package/dist/adapters/llm/factory.d.ts.map +1 -0
- package/dist/adapters/llm/factory.js +48 -0
- package/dist/adapters/llm/factory.js.map +1 -0
- package/dist/adapters/llm/index.d.ts +8 -0
- package/dist/adapters/llm/index.d.ts.map +1 -0
- package/dist/adapters/llm/index.js +7 -0
- package/dist/adapters/llm/index.js.map +1 -0
- package/dist/adapters/llm/modelSpec.d.ts +10 -0
- package/dist/adapters/llm/modelSpec.d.ts.map +1 -0
- package/dist/adapters/llm/modelSpec.js +21 -0
- package/dist/adapters/llm/modelSpec.js.map +1 -0
- package/dist/adapters/llm/ollama.d.ts +4 -0
- package/dist/adapters/llm/ollama.d.ts.map +1 -0
- package/dist/adapters/llm/ollama.js +32 -0
- package/dist/adapters/llm/ollama.js.map +1 -0
- package/dist/adapters/llm/openai-compat.d.ts +4 -0
- package/dist/adapters/llm/openai-compat.d.ts.map +1 -0
- package/dist/adapters/llm/openai-compat.js +36 -0
- package/dist/adapters/llm/openai-compat.js.map +1 -0
- package/dist/adapters/llm/pi-ai.d.ts +5 -0
- package/dist/adapters/llm/pi-ai.d.ts.map +1 -0
- package/dist/adapters/llm/pi-ai.js +60 -0
- package/dist/adapters/llm/pi-ai.js.map +1 -0
- package/dist/adapters/llm/standalone.d.ts +6 -0
- package/dist/adapters/llm/standalone.d.ts.map +1 -0
- package/dist/adapters/llm/standalone.js +44 -0
- package/dist/adapters/llm/standalone.js.map +1 -0
- package/dist/adapters/llm/types.d.ts +15 -0
- package/dist/adapters/llm/types.d.ts.map +1 -0
- package/dist/adapters/llm/types.js.map +1 -0
- package/dist/cli/init.d.ts +3 -0
- package/dist/cli/init.d.ts.map +1 -0
- package/dist/cli/init.js +13 -0
- package/dist/cli/init.js.map +1 -0
- package/dist/cli/jobs.d.ts +32 -0
- package/dist/cli/jobs.d.ts.map +1 -0
- package/dist/cli/jobs.js +100 -0
- package/dist/cli/jobs.js.map +1 -0
- package/dist/cli/log.d.ts +14 -0
- package/dist/cli/log.d.ts.map +1 -0
- package/dist/cli/log.js +32 -0
- package/dist/cli/log.js.map +1 -0
- package/dist/cli/parseArgs.d.ts +35 -0
- package/dist/cli/parseArgs.d.ts.map +1 -0
- package/dist/cli/parseArgs.js +127 -0
- package/dist/cli/parseArgs.js.map +1 -0
- package/dist/cli/status.d.ts +44 -0
- package/dist/cli/status.d.ts.map +1 -0
- package/dist/cli/status.js +238 -0
- package/dist/cli/status.js.map +1 -0
- package/dist/cli/theme.d.ts +14 -0
- package/dist/cli/theme.d.ts.map +1 -0
- package/dist/cli/theme.js +20 -0
- package/dist/cli/theme.js.map +1 -0
- package/dist/cli.js +70 -356
- package/dist/cli.js.map +1 -1
- package/dist/commands/index.d.ts +6 -0
- package/dist/commands/index.d.ts.map +1 -0
- package/dist/commands/index.js +14 -0
- package/dist/commands/index.js.map +1 -0
- package/dist/commands/parseMemoryStatusArgs.d.ts +3 -0
- package/dist/commands/parseMemoryStatusArgs.d.ts.map +1 -0
- package/dist/commands/parseMemoryStatusArgs.js +15 -0
- package/dist/commands/parseMemoryStatusArgs.js.map +1 -0
- package/dist/commands/parseRememberArgs.d.ts +10 -0
- package/dist/commands/parseRememberArgs.d.ts.map +1 -0
- package/dist/commands/parseRememberArgs.js +46 -0
- package/dist/commands/parseRememberArgs.js.map +1 -0
- package/dist/commands/remember.d.ts +8 -0
- package/dist/commands/remember.d.ts.map +1 -0
- package/dist/commands/remember.js +30 -0
- package/dist/commands/remember.js.map +1 -0
- package/dist/commands/status.d.ts +10 -0
- package/dist/commands/status.d.ts.map +1 -0
- package/dist/commands/status.js +81 -0
- package/dist/commands/status.js.map +1 -0
- package/dist/commands/types.d.ts +10 -0
- package/dist/commands/types.d.ts.map +1 -0
- package/dist/commands/types.js +2 -0
- package/dist/{consolidation → commands}/types.js.map +1 -1
- package/dist/compact/index.d.ts +5 -0
- package/dist/compact/index.d.ts.map +1 -0
- package/dist/compact/index.js +5 -0
- package/dist/compact/index.js.map +1 -0
- package/dist/compact/parseMemoryExport.d.ts +6 -0
- package/dist/compact/parseMemoryExport.d.ts.map +1 -0
- package/dist/compact/parseMemoryExport.js +62 -0
- package/dist/compact/parseMemoryExport.js.map +1 -0
- package/dist/compact/register.d.ts +10 -0
- package/dist/compact/register.d.ts.map +1 -0
- package/dist/compact/register.js +47 -0
- package/dist/compact/register.js.map +1 -0
- package/dist/compact/runSummary.d.ts +6 -0
- package/dist/compact/runSummary.d.ts.map +1 -0
- package/dist/compact/runSummary.js +29 -0
- package/dist/compact/runSummary.js.map +1 -0
- package/dist/compact/subagentDelta.d.ts +6 -0
- package/dist/compact/subagentDelta.d.ts.map +1 -0
- package/dist/compact/subagentDelta.js +13 -0
- package/dist/compact/subagentDelta.js.map +1 -0
- package/dist/compact/summaryPrompt.d.ts +3 -0
- package/dist/compact/summaryPrompt.d.ts.map +1 -0
- package/dist/compact/summaryPrompt.js +41 -0
- package/dist/compact/summaryPrompt.js.map +1 -0
- package/dist/config/agentDir.d.ts +11 -0
- package/dist/config/agentDir.d.ts.map +1 -0
- package/dist/config/agentDir.js +21 -0
- package/dist/config/agentDir.js.map +1 -0
- package/dist/config/chunking.d.ts +7 -0
- package/dist/config/chunking.d.ts.map +1 -0
- package/dist/config/chunking.js +20 -0
- package/dist/config/chunking.js.map +1 -0
- package/dist/config/env.d.ts +28 -0
- package/dist/config/env.d.ts.map +1 -0
- package/dist/config/env.js +43 -0
- package/dist/config/env.js.map +1 -0
- package/dist/config/index.d.ts +7 -0
- package/dist/config/index.d.ts.map +1 -0
- package/dist/config/index.js +7 -0
- package/dist/config/index.js.map +1 -0
- package/dist/config/loadEnv.d.ts +10 -0
- package/dist/config/loadEnv.d.ts.map +1 -0
- package/dist/config/loadEnv.js +25 -0
- package/dist/config/loadEnv.js.map +1 -0
- package/dist/config/preflightBudget.d.ts +9 -0
- package/dist/config/preflightBudget.d.ts.map +1 -0
- package/dist/config/preflightBudget.js +16 -0
- package/dist/config/preflightBudget.js.map +1 -0
- package/dist/config/retrieval.d.ts +10 -0
- package/dist/config/retrieval.d.ts.map +1 -0
- package/dist/config/retrieval.js +31 -0
- package/dist/config/retrieval.js.map +1 -0
- package/dist/consolidate/entryKey.d.ts +6 -0
- package/dist/consolidate/entryKey.d.ts.map +1 -0
- package/dist/consolidate/entryKey.js +5 -0
- package/dist/consolidate/entryKey.js.map +1 -0
- package/dist/consolidate/index.d.ts +5 -0
- package/dist/consolidate/index.d.ts.map +1 -0
- package/dist/consolidate/index.js +5 -0
- package/dist/consolidate/index.js.map +1 -0
- package/dist/consolidate/mergeEntries.d.ts +4 -0
- package/dist/consolidate/mergeEntries.d.ts.map +1 -0
- package/dist/consolidate/mergeEntries.js +17 -0
- package/dist/consolidate/mergeEntries.js.map +1 -0
- package/dist/consolidate/mergePrompt.d.ts +4 -0
- package/dist/consolidate/mergePrompt.d.ts.map +1 -0
- package/dist/consolidate/mergePrompt.js +40 -0
- package/dist/consolidate/mergePrompt.js.map +1 -0
- package/dist/consolidate/mergeWithLlm.d.ts +4 -0
- package/dist/consolidate/mergeWithLlm.d.ts.map +1 -0
- package/dist/consolidate/mergeWithLlm.js +27 -0
- package/dist/consolidate/mergeWithLlm.js.map +1 -0
- package/dist/consolidate/runJob.d.ts +29 -0
- package/dist/consolidate/runJob.d.ts.map +1 -0
- package/dist/consolidate/runJob.js +48 -0
- package/dist/consolidate/runJob.js.map +1 -0
- package/dist/consolidate/scheduler.d.ts +18 -0
- package/dist/consolidate/scheduler.d.ts.map +1 -0
- package/dist/consolidate/scheduler.js +32 -0
- package/dist/consolidate/scheduler.js.map +1 -0
- package/dist/constants/chunking.d.ts +7 -0
- package/dist/constants/chunking.d.ts.map +1 -0
- package/dist/constants/chunking.js +7 -0
- package/dist/constants/chunking.js.map +1 -0
- package/dist/constants/env.d.ts +44 -0
- package/dist/constants/env.d.ts.map +1 -0
- package/dist/constants/env.js +49 -0
- package/dist/constants/env.js.map +1 -0
- package/dist/constants/index.d.ts +8 -0
- package/dist/constants/index.d.ts.map +1 -0
- package/dist/constants/index.js +8 -0
- package/dist/constants/index.js.map +1 -0
- package/dist/constants/memory.d.ts +17 -0
- package/dist/constants/memory.d.ts.map +1 -0
- package/dist/constants/memory.js +16 -0
- package/dist/constants/memory.js.map +1 -0
- package/dist/constants/paths.d.ts +23 -0
- package/dist/constants/paths.d.ts.map +1 -0
- package/dist/constants/paths.js +23 -0
- package/dist/constants/paths.js.map +1 -0
- package/dist/constants/preflight.d.ts +10 -0
- package/dist/constants/preflight.d.ts.map +1 -0
- package/dist/constants/preflight.js +10 -0
- package/dist/constants/preflight.js.map +1 -0
- package/dist/constants/retrieval.d.ts +10 -0
- package/dist/constants/retrieval.d.ts.map +1 -0
- package/dist/constants/retrieval.js +10 -0
- package/dist/constants/retrieval.js.map +1 -0
- package/dist/constants/security.d.ts +5 -0
- package/dist/constants/security.d.ts.map +1 -0
- package/dist/constants/security.js +5 -0
- package/dist/constants/security.js.map +1 -0
- package/dist/constants/timing.d.ts +36 -0
- package/dist/constants/timing.d.ts.map +1 -0
- package/dist/constants/timing.js +36 -0
- package/dist/constants/timing.js.map +1 -0
- package/dist/index.d.ts +11 -29
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +11 -28
- package/dist/index.js.map +1 -1
- package/dist/init/workspace.d.ts +15 -0
- package/dist/init/workspace.d.ts.map +1 -0
- package/dist/init/workspace.js +32 -0
- package/dist/init/workspace.js.map +1 -0
- package/dist/ipc/jsonlFramer.d.ts +9 -0
- package/dist/ipc/jsonlFramer.d.ts.map +1 -0
- package/dist/ipc/jsonlFramer.js +26 -0
- package/dist/ipc/jsonlFramer.js.map +1 -0
- package/dist/pi-extension.d.ts +0 -2
- package/dist/pi-extension.d.ts.map +1 -1
- package/dist/pi-extension.js +161 -520
- package/dist/pi-extension.js.map +1 -1
- package/dist/preflight/episodic.d.ts +20 -0
- package/dist/preflight/episodic.d.ts.map +1 -0
- package/dist/preflight/episodic.js +109 -0
- package/dist/preflight/episodic.js.map +1 -0
- package/dist/preflight/queryCache.d.ts +15 -0
- package/dist/preflight/queryCache.d.ts.map +1 -0
- package/dist/preflight/queryCache.js +46 -0
- package/dist/preflight/queryCache.js.map +1 -0
- package/dist/preflight/queryIntent.d.ts +20 -0
- package/dist/preflight/queryIntent.d.ts.map +1 -0
- package/dist/preflight/queryIntent.js +75 -0
- package/dist/preflight/queryIntent.js.map +1 -0
- package/dist/preflight/render.d.ts +8 -33
- package/dist/preflight/render.d.ts.map +1 -1
- package/dist/preflight/render.js +29 -126
- package/dist/preflight/render.js.map +1 -1
- package/dist/preflight/session.d.ts +3 -0
- package/dist/preflight/session.d.ts.map +1 -0
- package/dist/preflight/session.js +6 -0
- package/dist/preflight/session.js.map +1 -0
- package/dist/preflight/strip.d.ts +0 -8
- package/dist/preflight/strip.d.ts.map +1 -1
- package/dist/preflight/strip.js +2 -11
- package/dist/preflight/strip.js.map +1 -1
- package/dist/shutdown/enqueue.d.ts +12 -0
- package/dist/shutdown/enqueue.d.ts.map +1 -0
- package/dist/shutdown/enqueue.js +14 -0
- package/dist/shutdown/enqueue.js.map +1 -0
- package/dist/shutdown/extractPrompt.d.ts +4 -0
- package/dist/shutdown/extractPrompt.d.ts.map +1 -0
- package/dist/shutdown/extractPrompt.js +29 -0
- package/dist/shutdown/extractPrompt.js.map +1 -0
- package/dist/shutdown/processed.d.ts +9 -0
- package/dist/shutdown/processed.d.ts.map +1 -0
- package/dist/shutdown/processed.js +28 -0
- package/dist/shutdown/processed.js.map +1 -0
- package/dist/shutdown/readQueue.d.ts +4 -0
- package/dist/shutdown/readQueue.d.ts.map +1 -0
- package/dist/shutdown/readQueue.js +26 -0
- package/dist/shutdown/readQueue.js.map +1 -0
- package/dist/shutdown/runDrainJob.d.ts +26 -0
- package/dist/shutdown/runDrainJob.d.ts.map +1 -0
- package/dist/shutdown/runDrainJob.js +143 -0
- package/dist/shutdown/runDrainJob.js.map +1 -0
- package/dist/shutdown/sessionReader.d.ts +6 -0
- package/dist/shutdown/sessionReader.d.ts.map +1 -0
- package/dist/shutdown/sessionReader.js +44 -0
- package/dist/shutdown/sessionReader.js.map +1 -0
- package/dist/sidecar/client.d.ts +17 -16
- package/dist/sidecar/client.d.ts.map +1 -1
- package/dist/sidecar/client.js +64 -100
- package/dist/sidecar/client.js.map +1 -1
- package/dist/sidecar/paths.d.ts +8 -0
- package/dist/sidecar/paths.d.ts.map +1 -0
- package/dist/sidecar/paths.js +23 -0
- package/dist/sidecar/paths.js.map +1 -0
- package/dist/sidecar/protocol.d.ts +54 -0
- package/dist/sidecar/protocol.d.ts.map +1 -0
- package/dist/sidecar/protocol.js +4 -0
- package/dist/sidecar/protocol.js.map +1 -0
- package/dist/sidecar/reindexBridge.d.ts +13 -0
- package/dist/sidecar/reindexBridge.d.ts.map +1 -0
- package/dist/sidecar/reindexBridge.js +37 -0
- package/dist/sidecar/reindexBridge.js.map +1 -0
- package/dist/sidecar/server/process.d.ts +2 -0
- package/dist/sidecar/server/process.d.ts.map +1 -0
- package/dist/sidecar/server/process.js +22 -0
- package/dist/sidecar/server/process.js.map +1 -0
- package/dist/sidecar/server/query.d.ts +8 -0
- package/dist/sidecar/server/query.d.ts.map +1 -0
- package/dist/sidecar/server/query.js +7 -0
- package/dist/sidecar/server/query.js.map +1 -0
- package/dist/sidecar/server/reindex.d.ts +8 -0
- package/dist/sidecar/server/reindex.d.ts.map +1 -0
- package/dist/sidecar/server/reindex.js +12 -0
- package/dist/sidecar/server/reindex.js.map +1 -0
- package/dist/sidecar/server/server.d.ts +11 -0
- package/dist/sidecar/server/server.d.ts.map +1 -0
- package/dist/sidecar/server/server.js +76 -0
- package/dist/sidecar/server/server.js.map +1 -0
- package/dist/sidecar/server/stats.d.ts +8 -0
- package/dist/sidecar/server/stats.d.ts.map +1 -0
- package/dist/sidecar/server/stats.js +18 -0
- package/dist/sidecar/server/stats.js.map +1 -0
- package/dist/sidecar/server/vec/embedder.d.ts +2 -0
- package/dist/sidecar/server/vec/embedder.d.ts.map +1 -0
- package/dist/sidecar/server/vec/embedder.js +2 -0
- package/dist/sidecar/server/vec/embedder.js.map +1 -0
- package/dist/sidecar/server/vec/mmr.d.ts +13 -0
- package/dist/sidecar/server/vec/mmr.d.ts.map +1 -0
- package/dist/sidecar/server/vec/mmr.js +36 -0
- package/dist/sidecar/server/vec/mmr.js.map +1 -0
- package/dist/sidecar/server/vec/store.d.ts +30 -0
- package/dist/sidecar/server/vec/store.d.ts.map +1 -0
- package/dist/sidecar/server/vec/store.js +179 -0
- package/dist/sidecar/server/vec/store.js.map +1 -0
- package/dist/sidecar/sidecarManager.d.ts +10 -0
- package/dist/sidecar/sidecarManager.d.ts.map +1 -0
- package/dist/sidecar/sidecarManager.js +111 -0
- package/dist/sidecar/sidecarManager.js.map +1 -0
- package/dist/sidecar/utils.d.ts +3 -0
- package/dist/sidecar/utils.d.ts.map +1 -0
- package/dist/sidecar/utils.js +27 -0
- package/dist/sidecar/utils.js.map +1 -0
- package/dist/store/backend.d.ts +12 -0
- package/dist/store/backend.d.ts.map +1 -0
- package/dist/store/backend.js +47 -0
- package/dist/store/backend.js.map +1 -0
- package/dist/store/index.d.ts +6 -0
- package/dist/store/index.d.ts.map +1 -0
- package/dist/store/index.js +6 -0
- package/dist/store/index.js.map +1 -0
- package/dist/store/indexChunks.d.ts +9 -0
- package/dist/store/indexChunks.d.ts.map +1 -0
- package/dist/store/indexChunks.js +78 -0
- package/dist/store/indexChunks.js.map +1 -0
- package/dist/store/markdown/format.d.ts +5 -0
- package/dist/store/markdown/format.d.ts.map +1 -0
- package/dist/store/markdown/format.js +14 -0
- package/dist/store/markdown/format.js.map +1 -0
- package/dist/store/markdown/parse.d.ts +5 -0
- package/dist/store/markdown/parse.d.ts.map +1 -0
- package/dist/store/markdown/parse.js +57 -0
- package/dist/store/markdown/parse.js.map +1 -0
- package/dist/store/markdown/template.d.ts +3 -0
- package/dist/store/markdown/template.d.ts.map +1 -0
- package/dist/store/markdown/template.js +22 -0
- package/dist/store/markdown/template.js.map +1 -0
- package/dist/store/memoryStore.d.ts +69 -0
- package/dist/store/memoryStore.d.ts.map +1 -0
- package/dist/store/memoryStore.js +421 -0
- package/dist/store/memoryStore.js.map +1 -0
- package/dist/store/paths.d.ts +10 -0
- package/dist/store/paths.d.ts.map +1 -0
- package/dist/store/paths.js +19 -0
- package/dist/store/paths.js.map +1 -0
- package/dist/store/types.d.ts +38 -0
- package/dist/store/types.d.ts.map +1 -0
- package/dist/store/types.js +3 -0
- package/dist/store/types.js.map +1 -0
- package/dist/ui/memoryStatusWidget.d.ts +20 -0
- package/dist/ui/memoryStatusWidget.d.ts.map +1 -0
- package/dist/ui/memoryStatusWidget.js +51 -0
- package/dist/ui/memoryStatusWidget.js.map +1 -0
- package/dist/utils/debugLog.d.ts +4 -0
- package/dist/utils/debugLog.d.ts.map +1 -0
- package/dist/utils/debugLog.js +14 -0
- package/dist/utils/debugLog.js.map +1 -0
- package/dist/utils/fs.d.ts +26 -0
- package/dist/utils/fs.d.ts.map +1 -0
- package/dist/utils/fs.js +101 -0
- package/dist/utils/fs.js.map +1 -0
- package/dist/utils/index.d.ts +8 -0
- package/dist/utils/index.d.ts.map +1 -0
- package/dist/utils/index.js +8 -0
- package/dist/utils/index.js.map +1 -0
- package/dist/utils/paths.d.ts +22 -0
- package/dist/utils/paths.d.ts.map +1 -0
- package/dist/utils/paths.js +51 -0
- package/dist/utils/paths.js.map +1 -0
- package/dist/utils/platform.d.ts +7 -0
- package/dist/utils/platform.d.ts.map +1 -0
- package/dist/utils/platform.js +18 -0
- package/dist/utils/platform.js.map +1 -0
- package/dist/utils/scheduler.d.ts +33 -0
- package/dist/utils/scheduler.d.ts.map +1 -0
- package/dist/utils/scheduler.js +63 -0
- package/dist/utils/scheduler.js.map +1 -0
- package/dist/utils/socket.d.ts +7 -0
- package/dist/utils/socket.d.ts.map +1 -0
- package/dist/utils/socket.js +35 -0
- package/dist/utils/socket.js.map +1 -0
- package/dist/utils/time.d.ts +22 -0
- package/dist/utils/time.d.ts.map +1 -0
- package/dist/utils/time.js +42 -0
- package/dist/utils/time.js.map +1 -0
- package/package.json +28 -10
- package/scripts/init-memory-workspace.mjs +62 -0
- package/scripts/path-utils.mjs +12 -0
- package/scripts/platform-fs.mjs +65 -0
- package/scripts/postinstall.mjs +17 -0
- package/src/adapters/embed/factory.ts +50 -0
- package/src/adapters/embed/hash.ts +29 -0
- package/src/adapters/embed/normalize.ts +13 -0
- package/src/adapters/embed/ollama.ts +49 -0
- package/src/adapters/embed/openai.ts +46 -0
- package/src/adapters/embed/types.ts +9 -0
- package/src/adapters/http.ts +42 -0
- package/src/adapters/index.ts +16 -0
- package/src/adapters/llm/extractText.ts +6 -0
- package/src/adapters/llm/factory.ts +60 -0
- package/src/adapters/llm/index.ts +7 -0
- package/src/adapters/llm/modelSpec.ts +39 -0
- package/src/adapters/llm/ollama.ts +42 -0
- package/src/adapters/llm/openai-compat.ts +46 -0
- package/src/adapters/llm/pi-ai.ts +75 -0
- package/src/adapters/llm/standalone.ts +60 -0
- package/src/adapters/llm/types.ts +16 -0
- package/src/cli/init.ts +18 -0
- package/src/cli/jobs.ts +144 -0
- package/src/cli/log.ts +46 -0
- package/src/cli/parseArgs.ts +175 -0
- package/src/cli/status.ts +323 -0
- package/src/cli/theme.ts +26 -0
- package/src/cli.ts +80 -374
- package/src/commands/index.ts +20 -0
- package/src/commands/parseMemoryStatusArgs.ts +11 -0
- package/src/commands/parseRememberArgs.ts +55 -0
- package/src/commands/remember.ts +43 -0
- package/src/commands/status.ts +112 -0
- package/src/commands/types.ts +12 -0
- package/src/compact/index.ts +4 -0
- package/src/compact/parseMemoryExport.ts +71 -0
- package/src/compact/register.ts +57 -0
- package/src/compact/runSummary.ts +42 -0
- package/src/compact/subagentDelta.ts +22 -0
- package/src/compact/summaryPrompt.ts +45 -0
- package/src/config/agentDir.ts +24 -0
- package/src/config/chunking.ts +36 -0
- package/src/config/env.ts +97 -0
- package/src/config/index.ts +6 -0
- package/src/config/loadEnv.ts +26 -0
- package/src/config/preflightBudget.ts +33 -0
- package/src/config/retrieval.ts +50 -0
- package/src/consolidate/entryKey.ts +4 -0
- package/src/consolidate/index.ts +4 -0
- package/src/consolidate/mergeEntries.ts +22 -0
- package/src/consolidate/mergePrompt.ts +46 -0
- package/src/consolidate/mergeWithLlm.ts +39 -0
- package/src/consolidate/runJob.ts +81 -0
- package/src/consolidate/scheduler.ts +54 -0
- package/src/constants/chunking.ts +8 -0
- package/src/constants/env.ts +56 -0
- package/src/constants/index.ts +7 -0
- package/src/constants/memory.ts +22 -0
- package/src/constants/paths.ts +26 -0
- package/src/constants/preflight.ts +13 -0
- package/src/constants/retrieval.ts +10 -0
- package/src/constants/security.ts +5 -0
- package/src/constants/timing.ts +46 -0
- package/src/index.ts +11 -247
- package/src/init/workspace.ts +44 -0
- package/src/ipc/jsonlFramer.ts +29 -0
- package/src/pi-extension.ts +177 -606
- package/src/preflight/episodic.ts +159 -0
- package/src/preflight/queryCache.ts +59 -0
- package/src/preflight/queryIntent.ts +92 -0
- package/src/preflight/render.ts +34 -180
- package/src/preflight/session.ts +7 -0
- package/src/preflight/strip.ts +4 -11
- package/src/shutdown/enqueue.ts +29 -0
- package/src/shutdown/extractPrompt.ts +27 -0
- package/src/shutdown/processed.ts +45 -0
- package/src/shutdown/readQueue.ts +26 -0
- package/src/shutdown/runDrainJob.ts +202 -0
- package/src/shutdown/sessionReader.ts +61 -0
- package/src/sidecar/client.ts +98 -150
- package/src/sidecar/paths.ts +31 -0
- package/src/sidecar/protocol.ts +38 -0
- package/src/sidecar/reindexBridge.ts +52 -0
- package/src/sidecar/server/process.ts +27 -0
- package/src/sidecar/server/query.ts +16 -0
- package/src/sidecar/server/reindex.ts +21 -0
- package/src/sidecar/server/server.ts +105 -0
- package/src/sidecar/server/stats.ts +23 -0
- package/src/sidecar/server/vec/embedder.ts +1 -0
- package/src/sidecar/server/vec/mmr.ts +55 -0
- package/src/sidecar/server/vec/store.ts +232 -0
- package/src/sidecar/sidecarManager.ts +134 -0
- package/src/sidecar/utils.ts +28 -0
- package/src/store/backend.ts +66 -0
- package/src/store/index.ts +5 -0
- package/src/store/indexChunks.ts +89 -0
- package/src/store/markdown/format.ts +16 -0
- package/src/store/markdown/parse.ts +63 -0
- package/src/store/markdown/template.ts +24 -0
- package/src/store/memoryStore.ts +518 -0
- package/src/store/paths.ts +33 -0
- package/src/store/proper-lockfile.d.ts +3 -0
- package/src/store/types.ts +46 -0
- package/src/ui/memoryStatusWidget.ts +80 -0
- package/src/utils/debugLog.ts +22 -0
- package/src/utils/fs.ts +115 -0
- package/src/utils/index.ts +68 -0
- package/src/utils/paths.ts +67 -0
- package/src/utils/platform.ts +20 -0
- package/src/utils/scheduler.ts +98 -0
- package/src/utils/socket.ts +33 -0
- package/src/utils/time.ts +51 -0
- package/templates/MEMORY.md.example +22 -0
- package/dist/adapters/ollamaClient.d.ts +0 -11
- package/dist/adapters/ollamaClient.d.ts.map +0 -1
- package/dist/adapters/ollamaClient.js +0 -122
- package/dist/adapters/ollamaClient.js.map +0 -1
- package/dist/adapters/openaiCompatClient.d.ts +0 -11
- package/dist/adapters/openaiCompatClient.d.ts.map +0 -1
- package/dist/adapters/openaiCompatClient.js +0 -118
- package/dist/adapters/openaiCompatClient.js.map +0 -1
- package/dist/adapters/piComplete.d.ts +0 -17
- package/dist/adapters/piComplete.d.ts.map +0 -1
- package/dist/adapters/piComplete.js +0 -169
- package/dist/adapters/piComplete.js.map +0 -1
- package/dist/bundle/install.d.ts +0 -34
- package/dist/bundle/install.d.ts.map +0 -1
- package/dist/bundle/install.js +0 -183
- package/dist/bundle/install.js.map +0 -1
- package/dist/cache/memoryCaches.d.ts +0 -32
- package/dist/cache/memoryCaches.d.ts.map +0 -1
- package/dist/cache/memoryCaches.js +0 -59
- package/dist/cache/memoryCaches.js.map +0 -1
- package/dist/config.d.ts +0 -55
- package/dist/config.d.ts.map +0 -1
- package/dist/config.js +0 -85
- package/dist/config.js.map +0 -1
- package/dist/consolidation/enqueue.d.ts +0 -33
- package/dist/consolidation/enqueue.d.ts.map +0 -1
- package/dist/consolidation/enqueue.js +0 -145
- package/dist/consolidation/enqueue.js.map +0 -1
- package/dist/consolidation/index.d.ts +0 -19
- package/dist/consolidation/index.d.ts.map +0 -1
- package/dist/consolidation/index.js +0 -15
- package/dist/consolidation/index.js.map +0 -1
- package/dist/consolidation/lock.d.ts +0 -5
- package/dist/consolidation/lock.d.ts.map +0 -1
- package/dist/consolidation/lock.js +0 -33
- package/dist/consolidation/lock.js.map +0 -1
- package/dist/consolidation/log.d.ts +0 -8
- package/dist/consolidation/log.d.ts.map +0 -1
- package/dist/consolidation/log.js +0 -32
- package/dist/consolidation/log.js.map +0 -1
- package/dist/consolidation/memoryIndex.d.ts +0 -18
- package/dist/consolidation/memoryIndex.d.ts.map +0 -1
- package/dist/consolidation/memoryIndex.js +0 -100
- package/dist/consolidation/memoryIndex.js.map +0 -1
- package/dist/consolidation/phase2/runPhase2.d.ts +0 -23
- package/dist/consolidation/phase2/runPhase2.d.ts.map +0 -1
- package/dist/consolidation/phase2/runPhase2.js +0 -196
- package/dist/consolidation/phase2/runPhase2.js.map +0 -1
- package/dist/consolidation/scheduler/launchd.d.ts +0 -5
- package/dist/consolidation/scheduler/launchd.d.ts.map +0 -1
- package/dist/consolidation/scheduler/launchd.js +0 -54
- package/dist/consolidation/scheduler/launchd.js.map +0 -1
- package/dist/consolidation/scheduler/runConsolidate.d.ts +0 -26
- package/dist/consolidation/scheduler/runConsolidate.d.ts.map +0 -1
- package/dist/consolidation/scheduler/runConsolidate.js +0 -83
- package/dist/consolidation/scheduler/runConsolidate.js.map +0 -1
- package/dist/consolidation/scheduler/setupSchedule.d.ts +0 -3
- package/dist/consolidation/scheduler/setupSchedule.d.ts.map +0 -1
- package/dist/consolidation/scheduler/setupSchedule.js +0 -79
- package/dist/consolidation/scheduler/setupSchedule.js.map +0 -1
- package/dist/consolidation/scheduler/systemd.d.ts +0 -9
- package/dist/consolidation/scheduler/systemd.d.ts.map +0 -1
- package/dist/consolidation/scheduler/systemd.js +0 -57
- package/dist/consolidation/scheduler/systemd.js.map +0 -1
- package/dist/consolidation/scheduler/types.d.ts +0 -24
- package/dist/consolidation/scheduler/types.d.ts.map +0 -1
- package/dist/consolidation/scheduler/types.js.map +0 -1
- package/dist/consolidation/scope.d.ts +0 -12
- package/dist/consolidation/scope.d.ts.map +0 -1
- package/dist/consolidation/scope.js +0 -51
- package/dist/consolidation/scope.js.map +0 -1
- package/dist/consolidation/stage1/deltaExtract.d.ts +0 -3
- package/dist/consolidation/stage1/deltaExtract.d.ts.map +0 -1
- package/dist/consolidation/stage1/deltaExtract.js +0 -19
- package/dist/consolidation/stage1/deltaExtract.js.map +0 -1
- package/dist/consolidation/stage1/drainQueue.d.ts +0 -21
- package/dist/consolidation/stage1/drainQueue.d.ts.map +0 -1
- package/dist/consolidation/stage1/drainQueue.js +0 -113
- package/dist/consolidation/stage1/drainQueue.js.map +0 -1
- package/dist/consolidation/stage1/extractSession.d.ts +0 -6
- package/dist/consolidation/stage1/extractSession.d.ts.map +0 -1
- package/dist/consolidation/stage1/extractSession.js +0 -47
- package/dist/consolidation/stage1/extractSession.js.map +0 -1
- package/dist/consolidation/stage1/store.d.ts +0 -45
- package/dist/consolidation/stage1/store.d.ts.map +0 -1
- package/dist/consolidation/stage1/store.js +0 -297
- package/dist/consolidation/stage1/store.js.map +0 -1
- package/dist/consolidation/types.d.ts +0 -49
- package/dist/consolidation/types.d.ts.map +0 -1
- package/dist/errclass.d.ts +0 -7
- package/dist/errclass.d.ts.map +0 -1
- package/dist/errclass.js +0 -32
- package/dist/errclass.js.map +0 -1
- package/dist/extension.d.ts +0 -24
- package/dist/extension.d.ts.map +0 -1
- package/dist/extension.js +0 -7
- package/dist/extension.js.map +0 -1
- package/dist/fallback/index.d.ts +0 -11
- package/dist/fallback/index.d.ts.map +0 -1
- package/dist/fallback/index.js +0 -16
- package/dist/fallback/index.js.map +0 -1
- package/dist/fallback/llmRerank.d.ts +0 -19
- package/dist/fallback/llmRerank.d.ts.map +0 -1
- package/dist/fallback/llmRerank.js +0 -68
- package/dist/fallback/llmRerank.js.map +0 -1
- package/dist/fallback/memoryMd.d.ts +0 -6
- package/dist/fallback/memoryMd.d.ts.map +0 -1
- package/dist/fallback/memoryMd.js +0 -35
- package/dist/fallback/memoryMd.js.map +0 -1
- package/dist/fallback/sessionIndex.d.ts +0 -35
- package/dist/fallback/sessionIndex.d.ts.map +0 -1
- package/dist/fallback/sessionIndex.js +0 -239
- package/dist/fallback/sessionIndex.js.map +0 -1
- package/dist/fallback/sessionSearch.d.ts +0 -18
- package/dist/fallback/sessionSearch.d.ts.map +0 -1
- package/dist/fallback/sessionSearch.js +0 -234
- package/dist/fallback/sessionSearch.js.map +0 -1
- package/dist/local/graphQuery.d.ts +0 -31
- package/dist/local/graphQuery.d.ts.map +0 -1
- package/dist/local/graphQuery.js +0 -195
- package/dist/local/graphQuery.js.map +0 -1
- package/dist/paths.d.ts +0 -7
- package/dist/paths.d.ts.map +0 -1
- package/dist/paths.js +0 -26
- package/dist/paths.js.map +0 -1
- package/dist/preflight/detectIntents.d.ts +0 -102
- package/dist/preflight/detectIntents.d.ts.map +0 -1
- package/dist/preflight/detectIntents.js +0 -631
- package/dist/preflight/detectIntents.js.map +0 -1
- package/dist/preflight/hook.d.ts +0 -76
- package/dist/preflight/hook.d.ts.map +0 -1
- package/dist/preflight/hook.js +0 -122
- package/dist/preflight/hook.js.map +0 -1
- package/dist/service.d.ts +0 -72
- package/dist/service.d.ts.map +0 -1
- package/dist/service.js +0 -257
- package/dist/service.js.map +0 -1
- package/dist/session/activeBranch.d.ts +0 -17
- package/dist/session/activeBranch.d.ts.map +0 -1
- package/dist/session/activeBranch.js +0 -251
- package/dist/session/activeBranch.js.map +0 -1
- package/dist/settings.d.ts +0 -41
- package/dist/settings.d.ts.map +0 -1
- package/dist/settings.js +0 -85
- package/dist/settings.js.map +0 -1
- package/dist/sidecar/bundle.d.ts +0 -19
- package/dist/sidecar/bundle.d.ts.map +0 -1
- package/dist/sidecar/bundle.js +0 -39
- package/dist/sidecar/bundle.js.map +0 -1
- package/dist/sidecar/process.d.ts +0 -14
- package/dist/sidecar/process.d.ts.map +0 -1
- package/dist/sidecar/process.js +0 -138
- package/dist/sidecar/process.js.map +0 -1
- package/dist/tools/memoryAppend.d.ts +0 -47
- package/dist/tools/memoryAppend.d.ts.map +0 -1
- package/dist/tools/memoryAppend.js +0 -138
- package/dist/tools/memoryAppend.js.map +0 -1
- package/dist/tools/memoryRecall.d.ts +0 -113
- package/dist/tools/memoryRecall.d.ts.map +0 -1
- package/dist/tools/memoryRecall.js +0 -351
- package/dist/tools/memoryRecall.js.map +0 -1
- package/dist/trainer/bundleBuilder.d.ts +0 -30
- package/dist/trainer/bundleBuilder.d.ts.map +0 -1
- package/dist/trainer/bundleBuilder.js +0 -106
- package/dist/trainer/bundleBuilder.js.map +0 -1
- package/dist/trainer/bundleLoader.d.ts +0 -12
- package/dist/trainer/bundleLoader.d.ts.map +0 -1
- package/dist/trainer/bundleLoader.js +0 -59
- package/dist/trainer/bundleLoader.js.map +0 -1
- package/dist/trainer/deltaMerge.d.ts +0 -38
- package/dist/trainer/deltaMerge.d.ts.map +0 -1
- package/dist/trainer/deltaMerge.js +0 -183
- package/dist/trainer/deltaMerge.js.map +0 -1
- package/dist/trainer/entityResolver.d.ts +0 -27
- package/dist/trainer/entityResolver.d.ts.map +0 -1
- package/dist/trainer/entityResolver.js +0 -92
- package/dist/trainer/entityResolver.js.map +0 -1
- package/dist/trainer/extractFacts.d.ts +0 -67
- package/dist/trainer/extractFacts.d.ts.map +0 -1
- package/dist/trainer/extractFacts.js +0 -213
- package/dist/trainer/extractFacts.js.map +0 -1
- package/dist/trainer/index.d.ts +0 -54
- package/dist/trainer/index.d.ts.map +0 -1
- package/dist/trainer/index.js +0 -82
- package/dist/trainer/index.js.map +0 -1
- package/dist/trainer/llmExtractor.d.ts +0 -16
- package/dist/trainer/llmExtractor.d.ts.map +0 -1
- package/dist/trainer/llmExtractor.js +0 -146
- package/dist/trainer/llmExtractor.js.map +0 -1
- package/dist/trainer/marker.d.ts +0 -10
- package/dist/trainer/marker.d.ts.map +0 -1
- package/dist/trainer/marker.js +0 -28
- package/dist/trainer/marker.js.map +0 -1
- package/dist/trainer/scheduler.d.ts +0 -33
- package/dist/trainer/scheduler.d.ts.map +0 -1
- package/dist/trainer/scheduler.js +0 -73
- package/dist/trainer/scheduler.js.map +0 -1
- package/dist/trainer/sessionLoader.d.ts +0 -26
- package/dist/trainer/sessionLoader.d.ts.map +0 -1
- package/dist/trainer/sessionLoader.js +0 -145
- package/dist/trainer/sessionLoader.js.map +0 -1
- package/dist/types.d.ts +0 -135
- package/dist/types.d.ts.map +0 -1
- package/dist/types.js +0 -8
- package/dist/types.js.map +0 -1
- package/src/adapters/ollamaClient.ts +0 -179
- package/src/adapters/openaiCompatClient.ts +0 -155
- package/src/adapters/piComplete.ts +0 -233
- package/src/bundle/install.ts +0 -206
- package/src/cache/memoryCaches.ts +0 -72
- package/src/config.ts +0 -180
- package/src/consolidation/enqueue.ts +0 -205
- package/src/consolidation/index.ts +0 -90
- package/src/consolidation/lock.ts +0 -41
- package/src/consolidation/log.ts +0 -43
- package/src/consolidation/memoryIndex.ts +0 -127
- package/src/consolidation/phase2/runPhase2.ts +0 -255
- package/src/consolidation/scheduler/launchd.ts +0 -62
- package/src/consolidation/scheduler/runConsolidate.ts +0 -115
- package/src/consolidation/scheduler/setupSchedule.ts +0 -106
- package/src/consolidation/scheduler/systemd.ts +0 -78
- package/src/consolidation/scheduler/types.ts +0 -27
- package/src/consolidation/scope.ts +0 -58
- package/src/consolidation/stage1/deltaExtract.ts +0 -29
- package/src/consolidation/stage1/drainQueue.ts +0 -143
- package/src/consolidation/stage1/extractSession.ts +0 -58
- package/src/consolidation/stage1/store.ts +0 -411
- package/src/consolidation/types.ts +0 -52
- package/src/errclass.ts +0 -37
- package/src/extension.ts +0 -23
- package/src/fallback/index.ts +0 -24
- package/src/fallback/llmRerank.ts +0 -97
- package/src/fallback/memoryMd.ts +0 -36
- package/src/fallback/sessionIndex.ts +0 -290
- package/src/fallback/sessionSearch.ts +0 -261
- package/src/local/graphQuery.ts +0 -252
- package/src/paths.ts +0 -28
- package/src/preflight/detectIntents.ts +0 -660
- package/src/preflight/hook.ts +0 -213
- package/src/service.ts +0 -306
- package/src/session/activeBranch.ts +0 -294
- package/src/settings.ts +0 -161
- package/src/sidecar/bundle.ts +0 -52
- package/src/sidecar/process.ts +0 -160
- package/src/tools/memoryAppend.ts +0 -166
- package/src/tools/memoryRecall.ts +0 -388
- package/src/trainer/bundleBuilder.ts +0 -192
- package/src/trainer/bundleLoader.ts +0 -105
- package/src/trainer/deltaMerge.ts +0 -221
- package/src/trainer/entityResolver.ts +0 -140
- package/src/trainer/extractFacts.ts +0 -312
- package/src/trainer/index.ts +0 -147
- package/src/trainer/llmExtractor.ts +0 -206
- package/src/trainer/marker.ts +0 -30
- package/src/trainer/scheduler.ts +0 -107
- package/src/trainer/sessionLoader.ts +0 -176
- package/src/types.ts +0 -168
- /package/dist/{consolidation/scheduler → adapters/embed}/types.js +0 -0
- /package/dist/{consolidation → adapters/llm}/types.js +0 -0
|
@@ -0,0 +1,112 @@
|
|
|
1
|
+
import type { ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
|
|
2
|
+
|
|
3
|
+
import {
|
|
4
|
+
formatMemoryStatusLines,
|
|
5
|
+
gatherMemoryStatus,
|
|
6
|
+
} from "../cli/status.js";
|
|
7
|
+
import { resolveMemoryAgentDir } from "../config/agentDir.js";
|
|
8
|
+
import { createMemoryStatusWidget, type MemoryStatusWidgetState } from "../ui/memoryStatusWidget.js";
|
|
9
|
+
|
|
10
|
+
import { parseMemoryStatusArgs } from "./parseMemoryStatusArgs.js";
|
|
11
|
+
|
|
12
|
+
export type MemoryStatusCommandDeps = {
|
|
13
|
+
getAgentDir(): string | null;
|
|
14
|
+
};
|
|
15
|
+
|
|
16
|
+
let memoryStatusWidgetState: MemoryStatusWidgetState | null = null;
|
|
17
|
+
|
|
18
|
+
export function resetMemoryStatusWidgetState(): void {
|
|
19
|
+
memoryStatusWidgetState = null;
|
|
20
|
+
}
|
|
21
|
+
|
|
22
|
+
function renderMemoryStatusWidget(
|
|
23
|
+
ctx: ExtensionCommandContext,
|
|
24
|
+
state: MemoryStatusWidgetState,
|
|
25
|
+
): void {
|
|
26
|
+
ctx.ui.setWidget("pi-memory-status", (tui, theme) => createMemoryStatusWidget(tui, theme, state), {
|
|
27
|
+
placement: "aboveEditor",
|
|
28
|
+
});
|
|
29
|
+
}
|
|
30
|
+
|
|
31
|
+
function applyMemoryStatusAction(
|
|
32
|
+
action: ReturnType<typeof parseMemoryStatusArgs>,
|
|
33
|
+
refreshed: boolean,
|
|
34
|
+
): void {
|
|
35
|
+
if (!memoryStatusWidgetState) return;
|
|
36
|
+
|
|
37
|
+
if (action === "expand") {
|
|
38
|
+
memoryStatusWidgetState.expanded = true;
|
|
39
|
+
return;
|
|
40
|
+
}
|
|
41
|
+
if (action === "collapse") {
|
|
42
|
+
memoryStatusWidgetState.expanded = false;
|
|
43
|
+
return;
|
|
44
|
+
}
|
|
45
|
+
if (action === "toggle" && !refreshed) {
|
|
46
|
+
memoryStatusWidgetState.expanded = !memoryStatusWidgetState.expanded;
|
|
47
|
+
return;
|
|
48
|
+
}
|
|
49
|
+
if (action === "toggle" && refreshed) {
|
|
50
|
+
memoryStatusWidgetState.expanded = false;
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
|
|
54
|
+
export function createMemoryStatusCommand(deps: MemoryStatusCommandDeps) {
|
|
55
|
+
return async (args: string | string[], ctx: ExtensionCommandContext): Promise<void> => {
|
|
56
|
+
const action = parseMemoryStatusArgs(args);
|
|
57
|
+
|
|
58
|
+
if (action === "hide") {
|
|
59
|
+
memoryStatusWidgetState = null;
|
|
60
|
+
if (ctx.hasUI) {
|
|
61
|
+
ctx.ui.setWidget("pi-memory-status", undefined);
|
|
62
|
+
}
|
|
63
|
+
return;
|
|
64
|
+
}
|
|
65
|
+
|
|
66
|
+
const agentDir = deps.getAgentDir() ?? resolveMemoryAgentDir();
|
|
67
|
+
|
|
68
|
+
const needsFetch = action === "refresh" || !memoryStatusWidgetState;
|
|
69
|
+
|
|
70
|
+
if (needsFetch) {
|
|
71
|
+
if (ctx.hasUI) {
|
|
72
|
+
ctx.ui.setWorkingMessage("Checking memory…");
|
|
73
|
+
}
|
|
74
|
+
|
|
75
|
+
try {
|
|
76
|
+
const report = await gatherMemoryStatus(agentDir);
|
|
77
|
+
memoryStatusWidgetState = {
|
|
78
|
+
report,
|
|
79
|
+
expanded: memoryStatusWidgetState?.expanded ?? false,
|
|
80
|
+
};
|
|
81
|
+
} catch (error) {
|
|
82
|
+
ctx.ui.notify(
|
|
83
|
+
`Memory status failed: ${error instanceof Error ? error.message : String(error)}`,
|
|
84
|
+
"warning",
|
|
85
|
+
);
|
|
86
|
+
return;
|
|
87
|
+
} finally {
|
|
88
|
+
if (ctx.hasUI) {
|
|
89
|
+
ctx.ui.setWorkingMessage();
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
|
|
94
|
+
applyMemoryStatusAction(action, needsFetch);
|
|
95
|
+
|
|
96
|
+
if (ctx.hasUI) {
|
|
97
|
+
renderMemoryStatusWidget(ctx, memoryStatusWidgetState!);
|
|
98
|
+
return;
|
|
99
|
+
}
|
|
100
|
+
|
|
101
|
+
const lines = ["pi-memory status", ...formatMemoryStatusLines(memoryStatusWidgetState!.report)];
|
|
102
|
+
ctx.ui.notify(lines.join("\n"), "info");
|
|
103
|
+
};
|
|
104
|
+
}
|
|
105
|
+
|
|
106
|
+
export function getMemoryStatusWidgetStateForTest(): MemoryStatusWidgetState | null {
|
|
107
|
+
return memoryStatusWidgetState;
|
|
108
|
+
}
|
|
109
|
+
|
|
110
|
+
export function setMemoryStatusWidgetStateForTest(state: MemoryStatusWidgetState | null): void {
|
|
111
|
+
memoryStatusWidgetState = state;
|
|
112
|
+
}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { MemoryStore } from "../store/memoryStore.js";
|
|
2
|
+
|
|
3
|
+
export type RememberCommandDeps = {
|
|
4
|
+
getMemoryStore(): MemoryStore | null;
|
|
5
|
+
onRemembered?(): Promise<void>;
|
|
6
|
+
};
|
|
7
|
+
|
|
8
|
+
export type MemoryStatusCommandDeps = {
|
|
9
|
+
getAgentDir(): string | null;
|
|
10
|
+
};
|
|
11
|
+
|
|
12
|
+
export type CommandDeps = RememberCommandDeps & MemoryStatusCommandDeps;
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
export { parseMemoryExport, hasMemoryExportSection } from "./parseMemoryExport.js";
|
|
2
|
+
export { buildCompactionSummaryPrompt, DUAL_PURPOSE_SUMMARY_TEMPLATE } from "./summaryPrompt.js";
|
|
3
|
+
export { runDualPurposeCompactionSummary } from "./runSummary.js";
|
|
4
|
+
export { registerCompactHandlers, type CompactHandlerDeps } from "./register.js";
|
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
import { parseMemoryMarkdown } from "../store/markdown/parse.js";
|
|
2
|
+
import { MEMORY_SECTIONS, type MemorySection, type StoreMemoryEntry } from "../store/types.js";
|
|
3
|
+
|
|
4
|
+
const MEMORY_EXPORT_HEADER_RE = /^##\s+Memory Export\s*$/im;
|
|
5
|
+
|
|
6
|
+
function extractMemoryExportBlock(summary: string): string {
|
|
7
|
+
const match = MEMORY_EXPORT_HEADER_RE.exec(summary);
|
|
8
|
+
if (!match) return "";
|
|
9
|
+
|
|
10
|
+
let body = summary.slice(match.index + match[0].length);
|
|
11
|
+
const nextTopLevel = body.search(/^##\s+(?!#)/m);
|
|
12
|
+
if (nextTopLevel >= 0) {
|
|
13
|
+
body = body.slice(0, nextTopLevel);
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
return body.trim();
|
|
17
|
+
}
|
|
18
|
+
|
|
19
|
+
/** Normalize ### Section headers to ## for shared markdown parser. */
|
|
20
|
+
function normalizeExportMarkdown(exportBody: string): string {
|
|
21
|
+
const lines = exportBody.split("\n");
|
|
22
|
+
const normalized: string[] = [];
|
|
23
|
+
|
|
24
|
+
for (const line of lines) {
|
|
25
|
+
const subsection = line.match(/^###\s+(Preferences|Conventions|Findings|Todos)\s*$/);
|
|
26
|
+
if (subsection) {
|
|
27
|
+
normalized.push(`## ${subsection[1]}`);
|
|
28
|
+
continue;
|
|
29
|
+
}
|
|
30
|
+
normalized.push(line);
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
return normalized.join("\n");
|
|
34
|
+
}
|
|
35
|
+
|
|
36
|
+
function parseExportBullets(exportBody: string): StoreMemoryEntry[] {
|
|
37
|
+
const normalized = normalizeExportMarkdown(exportBody);
|
|
38
|
+
if (!normalized.trim()) return [];
|
|
39
|
+
|
|
40
|
+
const parsed = parseMemoryMarkdown(normalized, "memory-export");
|
|
41
|
+
return parsed.map((entry) => ({
|
|
42
|
+
id: "",
|
|
43
|
+
section: entry.section,
|
|
44
|
+
content: entry.content,
|
|
45
|
+
timestamp: "",
|
|
46
|
+
}));
|
|
47
|
+
}
|
|
48
|
+
|
|
49
|
+
/** Parse durable facts from the ## Memory Export section of a dual-purpose summary. */
|
|
50
|
+
export function parseMemoryExport(summary: string): StoreMemoryEntry[] {
|
|
51
|
+
const exportBody = extractMemoryExportBlock(summary);
|
|
52
|
+
if (!exportBody) return [];
|
|
53
|
+
|
|
54
|
+
const entries = parseExportBullets(exportBody);
|
|
55
|
+
const seen = new Set<string>();
|
|
56
|
+
|
|
57
|
+
return entries.filter((entry) => {
|
|
58
|
+
const key = `${entry.section}\0${entry.content.trim()}`;
|
|
59
|
+
if (seen.has(key)) return false;
|
|
60
|
+
seen.add(key);
|
|
61
|
+
return entry.content.trim().length > 0;
|
|
62
|
+
});
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
export function hasMemoryExportSection(summary: string): boolean {
|
|
66
|
+
return MEMORY_EXPORT_HEADER_RE.test(summary);
|
|
67
|
+
}
|
|
68
|
+
|
|
69
|
+
export function isKnownExportSection(name: string): name is MemorySection {
|
|
70
|
+
return (MEMORY_SECTIONS as readonly string[]).includes(name);
|
|
71
|
+
}
|
|
@@ -0,0 +1,57 @@
|
|
|
1
|
+
import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
|
|
2
|
+
|
|
3
|
+
import type { LlmClient } from "../adapters/llm/types.js";
|
|
4
|
+
import { isSubagentSession } from "../preflight/session.js";
|
|
5
|
+
import type { MemoryStore } from "../store/memoryStore.js";
|
|
6
|
+
import { runDualPurposeCompactionSummary } from "./runSummary.js";
|
|
7
|
+
|
|
8
|
+
export type CompactHandlerDeps = {
|
|
9
|
+
getMemoryStore(): MemoryStore | null;
|
|
10
|
+
getLlmClient(): LlmClient | null;
|
|
11
|
+
onCompactionIngested?(): Promise<void>;
|
|
12
|
+
};
|
|
13
|
+
|
|
14
|
+
export function registerCompactHandlers(pi: ExtensionAPI, deps: CompactHandlerDeps): void {
|
|
15
|
+
pi.on("session_before_compact", async (event, ctx) => {
|
|
16
|
+
const llm = deps.getLlmClient();
|
|
17
|
+
if (!llm) return;
|
|
18
|
+
|
|
19
|
+
const { preparation, signal } = event;
|
|
20
|
+
const workingUi = ctx.hasUI
|
|
21
|
+
? {
|
|
22
|
+
show: () => ctx.ui.setWorkingMessage("Summarizing for memory…"),
|
|
23
|
+
clear: () => ctx.ui.setWorkingMessage(),
|
|
24
|
+
}
|
|
25
|
+
: null;
|
|
26
|
+
|
|
27
|
+
try {
|
|
28
|
+
workingUi?.show();
|
|
29
|
+
const summary = await runDualPurposeCompactionSummary(preparation, llm, signal);
|
|
30
|
+
if (!summary) return;
|
|
31
|
+
|
|
32
|
+
return {
|
|
33
|
+
compaction: {
|
|
34
|
+
summary,
|
|
35
|
+
firstKeptEntryId: preparation.firstKeptEntryId,
|
|
36
|
+
tokensBefore: preparation.tokensBefore,
|
|
37
|
+
},
|
|
38
|
+
};
|
|
39
|
+
} catch {
|
|
40
|
+
return;
|
|
41
|
+
} finally {
|
|
42
|
+
workingUi?.clear();
|
|
43
|
+
}
|
|
44
|
+
});
|
|
45
|
+
|
|
46
|
+
pi.on("session_compact", (event, ctx) => {
|
|
47
|
+
const store = deps.getMemoryStore();
|
|
48
|
+
if (!store) return;
|
|
49
|
+
|
|
50
|
+
store.appendFromCompaction({
|
|
51
|
+
compactionId: event.compactionEntry.id,
|
|
52
|
+
summary: event.compactionEntry.summary,
|
|
53
|
+
subagent: isSubagentSession(ctx),
|
|
54
|
+
onComplete: () => deps.onCompactionIngested?.(),
|
|
55
|
+
});
|
|
56
|
+
});
|
|
57
|
+
}
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
import type { AgentMessage } from "@earendil-works/pi-agent-core";
|
|
2
|
+
import type { SessionBeforeCompactEvent } from "@earendil-works/pi-coding-agent";
|
|
3
|
+
import { convertToLlm, serializeConversation } from "@earendil-works/pi-coding-agent";
|
|
4
|
+
|
|
5
|
+
import type { LlmClient } from "../adapters/llm/types.js";
|
|
6
|
+
import { stripPrivateMemory } from "../preflight/strip.js";
|
|
7
|
+
import { buildCompactionSummaryPrompt } from "./summaryPrompt.js";
|
|
8
|
+
|
|
9
|
+
type CompactionPreparation = SessionBeforeCompactEvent["preparation"];
|
|
10
|
+
|
|
11
|
+
function stripPrivateMemoryFromMessages(messages: AgentMessage[]): AgentMessage[] {
|
|
12
|
+
return messages.map((message) => {
|
|
13
|
+
if (message.role !== "user") return message;
|
|
14
|
+
if (typeof message.content === "string") {
|
|
15
|
+
return { ...message, content: stripPrivateMemory(message.content) };
|
|
16
|
+
}
|
|
17
|
+
return {
|
|
18
|
+
...message,
|
|
19
|
+
content: message.content.map((block) =>
|
|
20
|
+
block.type === "text"
|
|
21
|
+
? { ...block, text: stripPrivateMemory(block.text) }
|
|
22
|
+
: block,
|
|
23
|
+
),
|
|
24
|
+
} as AgentMessage;
|
|
25
|
+
});
|
|
26
|
+
}
|
|
27
|
+
|
|
28
|
+
export async function runDualPurposeCompactionSummary(
|
|
29
|
+
preparation: CompactionPreparation,
|
|
30
|
+
llm: LlmClient,
|
|
31
|
+
signal?: AbortSignal,
|
|
32
|
+
): Promise<string | null> {
|
|
33
|
+
const allMessages = stripPrivateMemoryFromMessages([
|
|
34
|
+
...preparation.messagesToSummarize,
|
|
35
|
+
...preparation.turnPrefixMessages,
|
|
36
|
+
]);
|
|
37
|
+
|
|
38
|
+
const conversationText = serializeConversation(convertToLlm(allMessages));
|
|
39
|
+
const prompt = buildCompactionSummaryPrompt(conversationText, preparation.previousSummary);
|
|
40
|
+
const summary = await llm.complete(prompt, signal);
|
|
41
|
+
return summary.trim() || null;
|
|
42
|
+
}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { ParsedEntry, StoreMemoryEntry } from "../store/types.js";
|
|
2
|
+
|
|
3
|
+
function entryKey(entry: { section: string; content: string }): string {
|
|
4
|
+
return `${entry.section}\0${entry.content.trim()}`;
|
|
5
|
+
}
|
|
6
|
+
|
|
7
|
+
/** Keep only Memory Export facts not already present in MEMORY (parent/root dedup). */
|
|
8
|
+
export function filterCompactionDelta(
|
|
9
|
+
entries: StoreMemoryEntry[],
|
|
10
|
+
existing: ParsedEntry[],
|
|
11
|
+
): StoreMemoryEntry[] {
|
|
12
|
+
const known = new Set(existing.map(entryKey));
|
|
13
|
+
return entries.filter((entry) => !known.has(entryKey(entry)));
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
/** Subagent clone: summary had export bullets but nothing new vs parent memory. */
|
|
17
|
+
export function shouldSkipSubagentCompactionIngest(
|
|
18
|
+
parsed: StoreMemoryEntry[],
|
|
19
|
+
delta: StoreMemoryEntry[],
|
|
20
|
+
): boolean {
|
|
21
|
+
return parsed.length > 0 && delta.length === 0;
|
|
22
|
+
}
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
export const DUAL_PURPOSE_SUMMARY_TEMPLATE = `## Session Context
|
|
2
|
+
<!-- Goal / Progress / Next Actions / key file changes for continuing THIS session -->
|
|
3
|
+
|
|
4
|
+
## Memory Export
|
|
5
|
+
<!-- Cross-session durable facts only; omit this entire section if nothing worth keeping -->
|
|
6
|
+
|
|
7
|
+
### Preferences
|
|
8
|
+
|
|
9
|
+
### Conventions
|
|
10
|
+
|
|
11
|
+
### Findings
|
|
12
|
+
|
|
13
|
+
### Todos
|
|
14
|
+
`;
|
|
15
|
+
|
|
16
|
+
export function buildCompactionSummaryPrompt(
|
|
17
|
+
conversationText: string,
|
|
18
|
+
previousSummary?: string,
|
|
19
|
+
): string {
|
|
20
|
+
const previousContext = previousSummary
|
|
21
|
+
? `\n\nPrevious compaction summary for context:\n${previousSummary.trim()}`
|
|
22
|
+
: "";
|
|
23
|
+
|
|
24
|
+
return `You are summarizing a coding-agent conversation for two purposes at once.
|
|
25
|
+
|
|
26
|
+
1. **Session Context** — everything needed to continue THIS session after context compaction.
|
|
27
|
+
2. **Memory Export** — only durable facts that should persist across NEW sessions (preferences, conventions, findings, todos).
|
|
28
|
+
|
|
29
|
+
Rules:
|
|
30
|
+
- Output markdown ONLY, following the exact section structure below.
|
|
31
|
+
- Session Context: goals, progress, decisions, blockers, next actions, important file paths.
|
|
32
|
+
- Memory Export: bullet lists under ### Preferences / Conventions / Findings / Todos.
|
|
33
|
+
- Each Memory Export bullet is one standalone fact (no pronouns like "we" without context).
|
|
34
|
+
- Skip Memory Export subsections that have nothing new; omit the whole Memory Export section if empty.
|
|
35
|
+
- Do NOT include <private_memory> blocks or ephemeral tool noise in Memory Export.
|
|
36
|
+
${previousContext}
|
|
37
|
+
|
|
38
|
+
Required format:
|
|
39
|
+
|
|
40
|
+
${DUAL_PURPOSE_SUMMARY_TEMPLATE}
|
|
41
|
+
|
|
42
|
+
<conversation>
|
|
43
|
+
${conversationText}
|
|
44
|
+
</conversation>`;
|
|
45
|
+
}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
import { defaultMemoryAgentDir, expandHomePath } from "../utils/paths.js";
|
|
2
|
+
import { readPiMemoryEnv } from "./env.js";
|
|
3
|
+
|
|
4
|
+
export type ResolveMemoryAgentDirOptions = {
|
|
5
|
+
env?: NodeJS.ProcessEnv;
|
|
6
|
+
};
|
|
7
|
+
|
|
8
|
+
/**
|
|
9
|
+
* Memory data root: `PI_MEMORY_AGENT_DIR` when set, otherwise ~/.pi/pi-memory-data.
|
|
10
|
+
* All MEMORY.md, sidecar socket, and `memory.vec.sqlite` live under this path.
|
|
11
|
+
*/
|
|
12
|
+
export function resolveMemoryAgentDir(options: ResolveMemoryAgentDirOptions = {}): string {
|
|
13
|
+
const env = options.env ?? process.env;
|
|
14
|
+
const fromEnv = readPiMemoryEnv(env).agentDir?.trim();
|
|
15
|
+
if (fromEnv) return expandHomePath(fromEnv);
|
|
16
|
+
return defaultMemoryAgentDir();
|
|
17
|
+
}
|
|
18
|
+
|
|
19
|
+
/** CLI: explicit flag, then `resolveMemoryAgentDir`. */
|
|
20
|
+
export function resolveAgentDirFromEnv(explicit?: string, env = process.env): string {
|
|
21
|
+
const fromArg = explicit?.trim();
|
|
22
|
+
if (fromArg) return expandHomePath(fromArg);
|
|
23
|
+
return resolveMemoryAgentDir({ env });
|
|
24
|
+
}
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
import {
|
|
2
|
+
CHUNKING_DISABLED_MAX_CHARS,
|
|
3
|
+
DEFAULT_CHUNK_MAX_CHARS,
|
|
4
|
+
MAX_CHUNK_MAX_CHARS,
|
|
5
|
+
MIN_CHUNK_MAX_CHARS,
|
|
6
|
+
} from "../constants/chunking.js";
|
|
7
|
+
import { ENV_KEYS } from "../constants/env.js";
|
|
8
|
+
|
|
9
|
+
export type ChunkingConfig = {
|
|
10
|
+
/** Max chars per chunk body; <=0 keeps one chunk per entry. */
|
|
11
|
+
maxChars: number;
|
|
12
|
+
};
|
|
13
|
+
|
|
14
|
+
function parseOptionalInt(raw: string | undefined): number | undefined {
|
|
15
|
+
if (!raw?.trim()) return undefined;
|
|
16
|
+
const value = Number.parseInt(raw, 10);
|
|
17
|
+
return Number.isFinite(value) ? value : undefined;
|
|
18
|
+
}
|
|
19
|
+
|
|
20
|
+
export function readChunkingConfig(env: NodeJS.ProcessEnv = process.env): ChunkingConfig {
|
|
21
|
+
const raw = parseOptionalInt(env[ENV_KEYS.CHUNK_MAX_CHARS]);
|
|
22
|
+
const maxChars = raw ?? DEFAULT_CHUNK_MAX_CHARS;
|
|
23
|
+
|
|
24
|
+
if (maxChars <= CHUNKING_DISABLED_MAX_CHARS) {
|
|
25
|
+
return { maxChars: CHUNKING_DISABLED_MAX_CHARS };
|
|
26
|
+
}
|
|
27
|
+
|
|
28
|
+
return {
|
|
29
|
+
maxChars: Math.min(MAX_CHUNK_MAX_CHARS, Math.max(MIN_CHUNK_MAX_CHARS, maxChars)),
|
|
30
|
+
};
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
export {
|
|
34
|
+
CHUNKING_DISABLED_MAX_CHARS,
|
|
35
|
+
DEFAULT_CHUNK_MAX_CHARS,
|
|
36
|
+
} from "../constants/chunking.js";
|
|
@@ -0,0 +1,97 @@
|
|
|
1
|
+
import type { EmbedderProvider } from "../constants/env.js";
|
|
2
|
+
import {
|
|
3
|
+
DEFAULT_EMBEDDER,
|
|
4
|
+
DEFAULT_EMBED_DIM_FALLBACK,
|
|
5
|
+
DEFAULT_HASH_EMBED_DIM,
|
|
6
|
+
DEFAULT_HELPER_MODEL_SPEC,
|
|
7
|
+
DEFAULT_OLLAMA_BASE_URL,
|
|
8
|
+
DEFAULT_OLLAMA_EMBED_MODEL,
|
|
9
|
+
DEFAULT_OPENAI_EMBED_MODEL,
|
|
10
|
+
ENV_KEYS,
|
|
11
|
+
FALLBACK_OPENAI_API_KEY_ENV,
|
|
12
|
+
KNOWN_EMBED_DIMS,
|
|
13
|
+
} from "../constants/env.js";
|
|
14
|
+
import {
|
|
15
|
+
DEFAULT_CONSOLIDATE_DEBOUNCE_MS,
|
|
16
|
+
DEFAULT_HTTP_TIMEOUT_MS,
|
|
17
|
+
DEFAULT_PREFLIGHT_BUDGET_MS,
|
|
18
|
+
DEFAULT_REINDEX_DEBOUNCE_MS,
|
|
19
|
+
} from "../constants/timing.js";
|
|
20
|
+
import { resolvePreflightBudget } from "./preflightBudget.js";
|
|
21
|
+
|
|
22
|
+
export type PiMemoryEnv = {
|
|
23
|
+
embedder: EmbedderProvider;
|
|
24
|
+
openaiApiKey?: string;
|
|
25
|
+
openaiEmbedModel: string;
|
|
26
|
+
ollamaBaseUrl: string;
|
|
27
|
+
ollamaEmbedModel: string;
|
|
28
|
+
ollamaLlmBaseUrl: string;
|
|
29
|
+
ollamaLlmModel?: string;
|
|
30
|
+
helperModel: string;
|
|
31
|
+
helperApiKey?: string;
|
|
32
|
+
openaiCompatBaseUrl?: string;
|
|
33
|
+
openaiCompatModel?: string;
|
|
34
|
+
openaiCompatApiKey?: string;
|
|
35
|
+
embedDimOverride?: number;
|
|
36
|
+
httpTimeoutMs: number;
|
|
37
|
+
preflightBudgetMs: number;
|
|
38
|
+
reindexDebounceMs: number;
|
|
39
|
+
consolidateDebounceMs: number;
|
|
40
|
+
agentDir?: string;
|
|
41
|
+
envFile?: string;
|
|
42
|
+
};
|
|
43
|
+
|
|
44
|
+
function parseEmbedder(value: string | undefined): EmbedderProvider {
|
|
45
|
+
if (value === "openai" || value === "ollama" || value === "hash") return value;
|
|
46
|
+
return DEFAULT_EMBEDDER;
|
|
47
|
+
}
|
|
48
|
+
|
|
49
|
+
/** Read pi-memory env from process.env (after optional loadEnv). */
|
|
50
|
+
export function readPiMemoryEnv(env: NodeJS.ProcessEnv = process.env): PiMemoryEnv {
|
|
51
|
+
const embedDimRaw = env[ENV_KEYS.EMBED_DIM]?.trim();
|
|
52
|
+
const embedDimOverride = embedDimRaw ? Number.parseInt(embedDimRaw, 10) : undefined;
|
|
53
|
+
|
|
54
|
+
return {
|
|
55
|
+
embedder: parseEmbedder(env[ENV_KEYS.EMBEDDER]),
|
|
56
|
+
openaiApiKey: env[ENV_KEYS.OPENAI_API_KEY] ?? env[FALLBACK_OPENAI_API_KEY_ENV],
|
|
57
|
+
openaiEmbedModel: env[ENV_KEYS.OPENAI_EMBED_MODEL] ?? DEFAULT_OPENAI_EMBED_MODEL,
|
|
58
|
+
ollamaBaseUrl: env[ENV_KEYS.OLLAMA_BASE_URL] ?? DEFAULT_OLLAMA_BASE_URL,
|
|
59
|
+
ollamaEmbedModel: env[ENV_KEYS.OLLAMA_EMBED_MODEL] ?? DEFAULT_OLLAMA_EMBED_MODEL,
|
|
60
|
+
ollamaLlmBaseUrl:
|
|
61
|
+
env[ENV_KEYS.OLLAMA_LLM_BASE_URL] ?? env[ENV_KEYS.OLLAMA_BASE_URL] ?? DEFAULT_OLLAMA_BASE_URL,
|
|
62
|
+
ollamaLlmModel: env[ENV_KEYS.OLLAMA_LLM_MODEL]?.trim() || undefined,
|
|
63
|
+
helperModel: env[ENV_KEYS.HELPER_MODEL] ?? DEFAULT_HELPER_MODEL_SPEC,
|
|
64
|
+
helperApiKey: env[ENV_KEYS.HELPER_API_KEY]?.trim() || undefined,
|
|
65
|
+
openaiCompatBaseUrl: env[ENV_KEYS.OPENAI_COMPAT_BASE_URL]?.trim() || undefined,
|
|
66
|
+
openaiCompatModel: env[ENV_KEYS.OPENAI_COMPAT_MODEL]?.trim() || undefined,
|
|
67
|
+
openaiCompatApiKey: env[ENV_KEYS.OPENAI_COMPAT_API_KEY]?.trim() || undefined,
|
|
68
|
+
embedDimOverride: Number.isFinite(embedDimOverride) ? embedDimOverride : undefined,
|
|
69
|
+
httpTimeoutMs: Number.parseInt(env[ENV_KEYS.HTTP_TIMEOUT_MS] ?? String(DEFAULT_HTTP_TIMEOUT_MS), 10),
|
|
70
|
+
preflightBudgetMs: resolvePreflightBudget(
|
|
71
|
+
Number.parseInt(
|
|
72
|
+
env[ENV_KEYS.PREFLIGHT_BUDGET_MS] ??
|
|
73
|
+
env[ENV_KEYS.PREFLIGHT_TIMEOUT_MS] ??
|
|
74
|
+
String(DEFAULT_PREFLIGHT_BUDGET_MS),
|
|
75
|
+
10,
|
|
76
|
+
),
|
|
77
|
+
).totalMs,
|
|
78
|
+
reindexDebounceMs: Number.parseInt(
|
|
79
|
+
env[ENV_KEYS.REINDEX_DEBOUNCE_MS] ?? String(DEFAULT_REINDEX_DEBOUNCE_MS),
|
|
80
|
+
10,
|
|
81
|
+
),
|
|
82
|
+
consolidateDebounceMs: Number.parseInt(
|
|
83
|
+
env[ENV_KEYS.CONSOLIDATE_DEBOUNCE_MS] ?? String(DEFAULT_CONSOLIDATE_DEBOUNCE_MS),
|
|
84
|
+
10,
|
|
85
|
+
),
|
|
86
|
+
agentDir: env[ENV_KEYS.AGENT_DIR]?.trim() || undefined,
|
|
87
|
+
envFile: env[ENV_KEYS.ENV_FILE],
|
|
88
|
+
};
|
|
89
|
+
}
|
|
90
|
+
|
|
91
|
+
export function resolveEmbedDim(model: string, override?: number): number {
|
|
92
|
+
if (override !== undefined) return override;
|
|
93
|
+
return KNOWN_EMBED_DIMS[model] ?? DEFAULT_EMBED_DIM_FALLBACK;
|
|
94
|
+
}
|
|
95
|
+
|
|
96
|
+
export type { EmbedderProvider } from "../constants/env.js";
|
|
97
|
+
export { KNOWN_EMBED_DIMS, DEFAULT_HASH_EMBED_DIM } from "../constants/env.js";
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
export { KNOWN_EMBED_DIMS, readPiMemoryEnv, resolveEmbedDim, type EmbedderProvider, type PiMemoryEnv } from "./env.js";
|
|
2
|
+
export { readRetrievalConfig, type RetrievalConfig } from "./retrieval.js";
|
|
3
|
+
export { readChunkingConfig, type ChunkingConfig } from "./chunking.js";
|
|
4
|
+
export { resolvePreflightBudget, type PreflightBudget } from "./preflightBudget.js";
|
|
5
|
+
export { resolveAgentDirFromEnv, resolveMemoryAgentDir, type ResolveMemoryAgentDirOptions } from "./agentDir.js";
|
|
6
|
+
export { loadEnv } from "./loadEnv.js";
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
import { config } from "dotenv";
|
|
2
|
+
|
|
3
|
+
import { defaultPiMemoryEnvFile } from "../utils/paths.js";
|
|
4
|
+
import { joinPath, pathExists } from "../utils/fs.js";
|
|
5
|
+
import { readPiMemoryEnv } from "./env.js";
|
|
6
|
+
|
|
7
|
+
/**
|
|
8
|
+
* Load pi-memory env into process.env (does not override existing vars).
|
|
9
|
+
*
|
|
10
|
+
* Search order:
|
|
11
|
+
* 1. PI_MEMORY_ENV_FILE (explicit override)
|
|
12
|
+
* 2. cwd `.env` / `.env.local` (project-local dev)
|
|
13
|
+
* 3. ~/.pi/agent/pi-memory.env (recommended)
|
|
14
|
+
*/
|
|
15
|
+
export function loadEnv(cwd = process.cwd()): void {
|
|
16
|
+
const paths: string[] = [];
|
|
17
|
+
const explicit = readPiMemoryEnv(process.env).envFile;
|
|
18
|
+
if (explicit) paths.push(explicit);
|
|
19
|
+
|
|
20
|
+
paths.push(joinPath(cwd, ".env"), joinPath(cwd, ".env.local"), defaultPiMemoryEnvFile());
|
|
21
|
+
|
|
22
|
+
for (const path of paths) {
|
|
23
|
+
if (!path || !pathExists(path)) continue;
|
|
24
|
+
config({ path, override: false, quiet: true });
|
|
25
|
+
}
|
|
26
|
+
}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
import {
|
|
2
|
+
DEFAULT_PREFLIGHT_BUDGET_MS,
|
|
3
|
+
MAX_PREFLIGHT_BUDGET_MS,
|
|
4
|
+
MIN_PREFLIGHT_BUDGET_MS,
|
|
5
|
+
MIN_PREFLIGHT_SIDECAR_RESERVE_MS,
|
|
6
|
+
} from "../constants/timing.js";
|
|
7
|
+
|
|
8
|
+
export type PreflightBudget = {
|
|
9
|
+
totalMs: number;
|
|
10
|
+
intentMs: number;
|
|
11
|
+
sidecarMs: number;
|
|
12
|
+
};
|
|
13
|
+
|
|
14
|
+
function clamp(value: number, min: number, max: number): number {
|
|
15
|
+
return Math.min(max, Math.max(min, value));
|
|
16
|
+
}
|
|
17
|
+
|
|
18
|
+
/** Split shared preflight deadline between QueryIntent LLM and sidecar query. */
|
|
19
|
+
export function resolvePreflightBudget(totalMs = DEFAULT_PREFLIGHT_BUDGET_MS): PreflightBudget {
|
|
20
|
+
const total = clamp(totalMs, MIN_PREFLIGHT_BUDGET_MS, MAX_PREFLIGHT_BUDGET_MS);
|
|
21
|
+
const intentMs = clamp(
|
|
22
|
+
Math.floor(total * 0.3),
|
|
23
|
+
100,
|
|
24
|
+
total - MIN_PREFLIGHT_SIDECAR_RESERVE_MS,
|
|
25
|
+
);
|
|
26
|
+
return {
|
|
27
|
+
totalMs: total,
|
|
28
|
+
intentMs,
|
|
29
|
+
sidecarMs: total - intentMs,
|
|
30
|
+
};
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
export { DEFAULT_PREFLIGHT_BUDGET_MS } from "../constants/timing.js";
|
|
@@ -0,0 +1,50 @@
|
|
|
1
|
+
import {
|
|
2
|
+
CANDIDATE_POOL_MULTIPLIER,
|
|
3
|
+
DEFAULT_MIN_RELEVANCE,
|
|
4
|
+
DEFAULT_TOP_K,
|
|
5
|
+
MAX_MMR_LAMBDA,
|
|
6
|
+
MAX_TOP_K,
|
|
7
|
+
MIN_MMR_LAMBDA,
|
|
8
|
+
MIN_TOP_K,
|
|
9
|
+
MMR_LAMBDA,
|
|
10
|
+
} from "../constants/retrieval.js";
|
|
11
|
+
import { ENV_KEYS } from "../constants/env.js";
|
|
12
|
+
|
|
13
|
+
export type RetrievalConfig = {
|
|
14
|
+
topK: number;
|
|
15
|
+
mmrLambda: number;
|
|
16
|
+
minRelevance: number;
|
|
17
|
+
candidatePoolMultiplier: number;
|
|
18
|
+
};
|
|
19
|
+
|
|
20
|
+
function clamp(value: number, min: number, max: number): number {
|
|
21
|
+
return Math.min(max, Math.max(min, value));
|
|
22
|
+
}
|
|
23
|
+
|
|
24
|
+
function parseOptionalFloat(raw: string | undefined): number | undefined {
|
|
25
|
+
if (!raw?.trim()) return undefined;
|
|
26
|
+
const value = Number.parseFloat(raw);
|
|
27
|
+
return Number.isFinite(value) ? value : undefined;
|
|
28
|
+
}
|
|
29
|
+
|
|
30
|
+
function parseOptionalInt(raw: string | undefined): number | undefined {
|
|
31
|
+
if (!raw?.trim()) return undefined;
|
|
32
|
+
const value = Number.parseInt(raw, 10);
|
|
33
|
+
return Number.isFinite(value) ? value : undefined;
|
|
34
|
+
}
|
|
35
|
+
|
|
36
|
+
/** Vector retrieval tuning (sidecar query path). */
|
|
37
|
+
export function readRetrievalConfig(env: NodeJS.ProcessEnv = process.env): RetrievalConfig {
|
|
38
|
+
const topKRaw = parseOptionalInt(env[ENV_KEYS.TOP_K]);
|
|
39
|
+
const mmrRaw = parseOptionalFloat(env[ENV_KEYS.MMR_LAMBDA]);
|
|
40
|
+
const minRelRaw = parseOptionalFloat(env[ENV_KEYS.MIN_RELEVANCE]);
|
|
41
|
+
|
|
42
|
+
return {
|
|
43
|
+
topK: clamp(topKRaw ?? DEFAULT_TOP_K, MIN_TOP_K, MAX_TOP_K),
|
|
44
|
+
mmrLambda: clamp(mmrRaw ?? MMR_LAMBDA, MIN_MMR_LAMBDA, MAX_MMR_LAMBDA),
|
|
45
|
+
minRelevance: clamp(minRelRaw ?? DEFAULT_MIN_RELEVANCE, 0, 1),
|
|
46
|
+
candidatePoolMultiplier: CANDIDATE_POOL_MULTIPLIER,
|
|
47
|
+
};
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
export { DEFAULT_MIN_RELEVANCE, DEFAULT_TOP_K, MMR_LAMBDA } from "../constants/retrieval.js";
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
export { dedupeEntries } from "./mergeEntries.js";
|
|
2
|
+
export { mergeEntriesWithLlm } from "./mergeWithLlm.js";
|
|
3
|
+
export { runConsolidateJob, type RunConsolidateJobOptions, type RunConsolidateJobResult } from "./runJob.js";
|
|
4
|
+
export { createConsolidateScheduler, startConsolidateInterval, type ConsolidateScheduler } from "./scheduler.js";
|