deerflow-code/frontend-web/src/core/memory/hooks.ts

// Memory V2 React Query hooks。

import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";

import {
  addMemoryEntry,
  deleteMemoryEntry,
  loadMemoryConfig,
  loadMemoryV2,
  patchMemoryConfig,
  replaceMemoryEntry,
} from "./api";
import type {
  MemoryTarget,
  MemoryV2ConfigResponse,
  MemoryV2Data,
} from "./types";

const MEMORY_KEY = ["memory", "v2"] as const;
const MEMORY_CONFIG_KEY = ["memory", "v2", "config"] as const;

/** 读取 builtin 本地记忆 (USER.md / MEMORY.md)。 */
export function useMemory(agent?: string) {
  const { data, isLoading, error } = useQuery({
    queryKey: [...MEMORY_KEY, agent ?? ""],
    queryFn: () => loadMemoryV2(agent),
  });
  return { memory: data ?? null, isLoading, error };
}

/** 新增一条记忆条目。 */
export function useAddMemoryEntry(agent?: string) {
  const queryClient = useQueryClient();
  return useMutation({
    mutationFn: ({ target, content }: { target: MemoryTarget; content: string }) =>
      addMemoryEntry(target, content, agent),
    onSuccess: (data) => {
      queryClient.setQueryData<MemoryV2Data>([...MEMORY_KEY, agent ?? ""], data);
    },
  });
}

/** 替换一条记忆条目。 */
export function useReplaceMemoryEntry(agent?: string) {
  const queryClient = useQueryClient();
  return useMutation({
    mutationFn: ({
      target,
      oldText,
      content,
    }: {
      target: MemoryTarget;
      oldText: string;
      content: string;
    }) => replaceMemoryEntry(target, oldText, content, agent),
    onSuccess: (data) => {
      queryClient.setQueryData<MemoryV2Data>([...MEMORY_KEY, agent ?? ""], data);
    },
  });
}

/** 删除一条记忆条目。 */
export function useDeleteMemoryEntry(agent?: string) {
  const queryClient = useQueryClient();
  return useMutation({
    mutationFn: ({ target, oldText }: { target: MemoryTarget; oldText: string }) =>
      deleteMemoryEntry(target, oldText, agent),
    onSuccess: (data) => {
      queryClient.setQueryData<MemoryV2Data>([...MEMORY_KEY, agent ?? ""], data);
    },
  });
}

/** 读取生效记忆配置 + 白名单。 */
export function useMemoryConfig() {
  const { data, isLoading, error } = useQuery({
    queryKey: MEMORY_CONFIG_KEY,
    queryFn: () => loadMemoryConfig(),
  });
  return { configResponse: data ?? null, isLoading, error };
}

/** 写入 per-user 配置覆盖。 */
export function usePatchMemoryConfig() {
  const queryClient = useQueryClient();
  return useMutation({
    mutationFn: (overrides: Record<string, unknown>) => patchMemoryConfig(overrides),
    onSuccess: (data) => {
      queryClient.setQueryData<MemoryV2ConfigResponse>(MEMORY_CONFIG_KEY, data);
    },
  });
}