apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-27 19:25:15 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat: implement memory document fetching and saving functionality in the editor panel, and remove deprecated memory hook

Browse source
This commit is contained in:
Anish Sarkar 2026-05-20 12:50:15 +05:30
parent 61234e125f
commit 78a3c71bb5
4 changed files with 140 additions and 203 deletions
Download patch file Download diff file Expand all files Collapse all files

141
surfsense_web/hooks/use-memory.ts
View file

@ -1,141 +0,0 @@
"use client";
import { useCallback, useEffect, useState } from "react";
import { z } from "zod";
import { baseApiService } from "@/lib/apis/base-api.service";
const MemoryLimitsSchema = z.object({
soft: z.number(),
hard: z.number(),
});
const MemoryReadSchema = z.object({
memory_md: z.string(),
limits: MemoryLimitsSchema,
});
type MemoryScope = "user" | "team";
export type MemoryLimits = z.infer<typeof MemoryLimitsSchema>;
export type MemoryLimitLevel = "ok" | "warning" | "error";
interface UseMemoryOptions {
scope: MemoryScope;
searchSpaceId?: number | null;
autoLoad?: boolean;
}
function getMemoryPath(scope: MemoryScope, searchSpaceId?: number | null) {
if (scope === "user") return "/api/v1/users/me/memory";
if (!searchSpaceId) throw new Error("searchSpaceId is required for team memory");
return `/api/v1/searchspaces/${searchSpaceId}/memory`;
}
export function stripMemoryDisplayPrefixes(memory: string) {
return memory.replace(
/^\s*-\s+(?:\(\d{4}-\d{2}-\d{2}\)\s*\[(?:fact|pref|instr)\]\s*|\d{4}-\d{2}-\d{2}:\s*)/gim,
"- "
);
}
export function getMemoryLimitState(length: number, limits?: MemoryLimits | null) {
if (!limits) {
return {
level: "ok" as MemoryLimitLevel,
label: `${length.toLocaleString()} chars`,
isOverLimit: false,
};
}
const isOverLimit = length > limits.hard;
const isNearLimit = length > limits.soft;
const level: MemoryLimitLevel = isOverLimit ? "error" : isNearLimit ? "warning" : "ok";
const suffix = isOverLimit ? " - Exceeds limit" : isNearLimit ? " - Approaching limit" : "";
return {
level,
label: `${length.toLocaleString()}/${limits.hard.toLocaleString()} chars${suffix}`,
isOverLimit,
};
}
export function useMemory({ scope, searchSpaceId, autoLoad = true }: UseMemoryOptions) {
const [memory, setMemory] = useState("");
const [limits, setLimits] = useState<MemoryLimits | null>(null);
const [loading, setLoading] = useState(autoLoad);
const [saving, setSaving] = useState(false);
const load = useCallback(async () => {
setLoading(true);
try {
const data = await baseApiService.get(getMemoryPath(scope, searchSpaceId), MemoryReadSchema);
setMemory(data.memory_md);
setLimits(data.limits);
return data.memory_md;
} finally {
setLoading(false);
}
}, [scope, searchSpaceId]);
useEffect(() => {
if (!autoLoad) return;
load().catch(() => {
setLoading(false);
});
}, [autoLoad, load]);
const save = useCallback(
async (memoryMd: string) => {
setSaving(true);
try {
const data = await baseApiService.put(
getMemoryPath(scope, searchSpaceId),
MemoryReadSchema,
{
body: { memory_md: memoryMd },
}
);
setMemory(data.memory_md);
setLimits(data.limits);
return data.memory_md;
} finally {
setSaving(false);
}
},
[scope, searchSpaceId]
);
const reset = useCallback(async () => {
setSaving(true);
try {
const data = await baseApiService.post(
`${getMemoryPath(scope, searchSpaceId)}/reset`,
MemoryReadSchema
);
setMemory(data.memory_md);
setLimits(data.limits);
return data.memory_md;
} finally {
setSaving(false);
}
}, [scope, searchSpaceId]);
return {
memory,
setMemory,
limits,
displayMemory: stripMemoryDisplayPrefixes(memory),
loading,
saving,
load,
save,
reset,
};
}
export function useUserMemory(searchSpaceId?: number | null) {
return useMemory({ scope: "user", searchSpaceId });
}
export function useTeamMemory(searchSpaceId?: number | null) {
return useMemory({ scope: "team", searchSpaceId });
}