LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-03-30 01:06:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

perf(reply): lazy-load session store writes

Browse Source
This commit is contained in:
Vincent Koc
2026-03-22 19:32:06 -07:00
parent 009980465f
commit c0933e2fc8
2 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/auto-reply/reply/agent-runner.ts
View File

@@ -9,7 +9,6 @@ import {
resolveSessionFilePathOptions,
resolveSessionTranscriptPath,
} from "../../config/sessions/paths.js";
import { updateSessionStore, updateSessionStoreEntry } from "../../config/sessions/store.js";
import type { SessionEntry } from "../../config/sessions/types.js";
import type { TypingMode } from "../../config/types.js";
import { emitAgentEvent } from "../../infra/agent-events.js";
@@ -70,6 +69,8 @@ let contextTokensRuntimePromise: Promise<
let replyMediaPathsRuntimePromise: Promise<
typeof import("./reply-media-paths.runtime.js")
> | null = null;
let sessionStoreRuntimePromise: Promise<typeof import("../../config/sessions/store.runtime.js")> | null =
null;
function loadPiEmbeddedQueueRuntime() {
piEmbeddedQueueRuntimePromise ??= import("../../agents/pi-embedded-queue.runtime.js");
@@ -101,6 +102,11 @@ function loadReplyMediaPathsRuntime() {
return replyMediaPathsRuntimePromise;
}
function loadSessionStoreRuntime() {
sessionStoreRuntimePromise ??= import("../../config/sessions/store.runtime.js");
return sessionStoreRuntimePromise;
}
export async function runReplyAgent(params: {
commandBody: string;
followupRun: FollowupRun;
@@ -228,6 +234,7 @@ export async function runReplyAgent(params: {
activeSessionEntry.updatedAt = updatedAt;
activeSessionStore[sessionKey] = activeSessionEntry;
if (storePath) {
const { updateSessionStoreEntry } = await loadSessionStoreRuntime();
await updateSessionStoreEntry({
storePath,
sessionKey,
@@ -347,6 +354,7 @@ export async function runReplyAgent(params: {
nextEntry.sessionFile = nextSessionFile;
activeSessionStore[sessionKey] = nextEntry;
try {
const { updateSessionStore } = await loadSessionStoreRuntime();
await updateSessionStore(storePath, (store) => {
store[sessionKey] = nextEntry;
});
@@ -447,6 +455,7 @@ export async function runReplyAgent(params: {
activeSessionEntry.updatedAt = updatedAt;
activeSessionStore[sessionKey] = activeSessionEntry;
if (storePath) {
const { updateSessionStoreEntry } = await loadSessionStoreRuntime();
await updateSessionStoreEntry({
storePath,
sessionKey,
@@ -503,6 +512,7 @@ export async function runReplyAgent(params: {
activeSessionStore[sessionKey] = fallbackStateEntry;
}
if (sessionKey && storePath) {
const { updateSessionStoreEntry } = await loadSessionStoreRuntime();
await updateSessionStoreEntry({
storePath,
sessionKey,

2
src/config/sessions/store.runtime.ts
View File

@@ -1 +1 @@
export { updateSessionStore } from "./store.js";
export { updateSessionStore, updateSessionStoreEntry } from "./store.js";