66 lines
2.6 KiB
TypeScript
66 lines
2.6 KiB
TypeScript
|
/**
|
||
|
|
* Keeps agent memory loaded for whatever project the session belongs to.
|
||
|
|
*
|
||
|
|
* This does not belong to the Memory tab. The session index is built from the
|
||
|
|
* loaded snapshot, so leaving the load to the panel meant a user who never
|
||
|
|
* opened Project notes sent every message with no memory index at all — the
|
||
|
|
* agent had memories it was never told about.
|
||
|
|
*
|
||
|
|
* The session directory is resolved to its project first. A session in a
|
||
|
|
* worktree has the worktree's path, and loading by that path reads a store the
|
||
|
|
* agent does not write to, which is the same mismatch in the other direction.
|
||
|
|
*/
|
||
|
|
|
||
|
|
import React from 'react';
|
||
|
|
|
||
|
|
import { resolveProjectForSessionDirectory } from '@/lib/projectResolution';
|
||
|
|
import { subscribeOpenchamberEvents } from '@/lib/openchamberEvents';
|
||
|
|
import { useAgentMemoryStore } from '@/stores/useAgentMemoryStore';
|
||
|
|
import { useProjectsStore } from '@/stores/useProjectsStore';
|
||
|
|
import { useSessionUIStore } from '@/sync/session-ui-store';
|
||
|
|
import { useUIStore } from '@/stores/useUIStore';
|
||
|
|
|
||
|
|
/**
|
||
|
|
* The directory is a parameter rather than read from `useEffectiveDirectory`,
|
||
|
|
* because this runs above `SyncProvider` — that hook reads the sync context and
|
||
|
|
* throws outside it, which took the whole app down with a blank window.
|
||
|
|
*/
|
||
|
|
export const useAgentMemorySync = (directory: string | null): void => {
|
||
|
|
const enabled = useUIStore((state) => (
|
||
|
|
state.agentMemoryFeatureAvailable && state.agentMemoryToolEnabled
|
||
|
|
));
|
||
|
|
const projects = useProjectsStore((state) => state.projects);
|
||
|
|
const availableWorktreesByProject = useSessionUIStore((state) => state.availableWorktreesByProject);
|
||
|
|
const effectiveDirectory = directory ?? '';
|
||
|
|
const load = useAgentMemoryStore((state) => state.load);
|
||
|
|
|
||
|
|
const projectPath = React.useMemo(() => {
|
||
|
|
if (!effectiveDirectory) {
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
const resolved = resolveProjectForSessionDirectory(projects, availableWorktreesByProject, effectiveDirectory);
|
||
|
|
return resolved?.path ?? null;
|
||
|
|
}, [availableWorktreesByProject, effectiveDirectory, projects]);
|
||
|
|
|
||
|
|
React.useEffect(() => {
|
||
|
|
if (!enabled) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
void load(projectPath);
|
||
|
|
}, [enabled, load, projectPath]);
|
||
|
|
|
||
|
|
// The agent writes memory mid-turn through its own tool, so the index for the
|
||
|
|
// next message has to come from a fresh read rather than the snapshot taken
|
||
|
|
// before the turn started.
|
||
|
|
React.useEffect(() => {
|
||
|
|
if (!enabled) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
return subscribeOpenchamberEvents((event) => {
|
||
|
|
if (event.type === 'agent-memory-changed') {
|
||
|
|
void load(projectPath);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}, [enabled, load, projectPath]);
|
||
|
|
};
|