Function useMemory

frontend/hooks/useMemory.ts:24–485

({ visible, currentUserId, currentTenantId, message }: UseMemoryOptions)

import log from "@/lib/logger";

export function useMemory({ visible, currentUserId, currentTenantId, message }: UseMemoryOptions) {
  const { t } = useTranslation()
  /* ----------------------- Basic Settings State ----------------------- */
  const [memoryEnabled, setMemoryEnabledState] = useState<boolean>(true)
  const [shareOption, setShareOptionState] = useState<"always" | "ask" | "never">("always")

  /* ------------------------- Original Logic State ------------------------- */
  // Group disabled state (only effective for Agent shared, user Agent tabs)
  const [disabledGroups, setDisabledGroups] = useState<Record<string, boolean>>({})

  const disableAgentIdSet = useRef<Set<string>>(new Set())
  const disableUserAgentIdSet = useRef<Set<string>>(new Set())

  const [openKey, setOpenKey] = useState<string>()

  // Currently active Tab
  const [activeTabKey, setActiveTabKey] = useState<string>("base")

  // Pagination state
  const [pageMap, setPageMap] = useState<Record<string, number>>({ agentShared: 1, userAgent: 1 })

  /* ------------------------------ Data Groups ------------------------------ */
  const [tenantSharedGroup, setTenantSharedGroup] = useState<MemoryGroup>({ title: "", key: "tenant", items: [] })
  const [agentSharedGroups, setAgentSharedGroups] = useState<MemoryGroup[]>([])
  const [userPersonalGroup, setUserPersonalGroup] = useState<MemoryGroup>({ title: "", key: "user-personal", items: [] })
  const [userAgentGroups, setUserAgentGroups] = useState<MemoryGroup[]>([])

  /* ------------------------------ New Memory State ------------------------------ */
  const [addingMemoryKey, setAddingMemoryKey] = useState<string | null>(null)
  const [newMemoryContent, setNewMemoryContent] = useState<string>("")
  const [isAddingMemory, setIsAddingMemory] = useState<boolean>(false)

  /* --------------------------- Initialization Loading --------------------------- */
  useEffect(() => {
    if (!visible) return

    // 1. Load configuration
    loadMemoryConfig().then((cfg) => {
      setMemoryEnabledState(cfg.memoryEnabled)
      setShareOptionState(cfg.shareOption)
      disableAgentIdSet.current = new Set(cfg.disableAgentIds)
      disableUserAgentIdSet.current = new Set(cfg.disableUserAgentIds)
    }).catch((e) => {
      log.error("Failed to load memory config:", e)
      message.error(t('useMemory.loadConfigError'))
    })
  }, [visible, message])

  /* --------------------------- Load Group Data --------------------------- */
  useEffect(() => {
    if (!visible || !memoryEnabled) return

    const loadGroupsForActiveTab = async () => {
      try {
        if (activeTabKey === "tenant") {
          const tenantGrp = await fetchTenantSharedGroup()
          setTenantSharedGroup(tenantGrp)
        } else if (activeTabKey === "agentShared") {