mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-05-21 23:46:50 +00:00
feat: store the local settings
This commit is contained in:
@@ -0,0 +1,34 @@
|
||||
import { useCallback, useState } from "react";
|
||||
import { useEffect } from "react";
|
||||
|
||||
import type { AgentThreadContext } from "../threads";
|
||||
|
||||
import {
|
||||
DEFAULT_LOCAL_SETTINGS,
|
||||
getLocalSettings,
|
||||
updateContextOfLocalSettings,
|
||||
} from "./local";
|
||||
|
||||
export function useLocalSettings() {
|
||||
const [mounted, setMounted] = useState(false);
|
||||
const [threadContextState, setThreadContextState] = useState<
|
||||
Omit<AgentThreadContext, "thread_id">
|
||||
>(DEFAULT_LOCAL_SETTINGS.context);
|
||||
useEffect(() => {
|
||||
if (!mounted) {
|
||||
setThreadContextState(getLocalSettings().context);
|
||||
}
|
||||
setMounted(true);
|
||||
}, [mounted]);
|
||||
const setThreadContext = useCallback(
|
||||
(context: Omit<AgentThreadContext, "thread_id">) => {
|
||||
setThreadContextState(context);
|
||||
updateContextOfLocalSettings(context);
|
||||
},
|
||||
[],
|
||||
);
|
||||
return {
|
||||
threadContext: threadContextState,
|
||||
setThreadContext,
|
||||
};
|
||||
}
|
||||
@@ -0,0 +1,2 @@
|
||||
export * from "./hooks";
|
||||
export * from "./local";
|
||||
@@ -0,0 +1,44 @@
|
||||
import type { AgentThreadContext } from "../threads";
|
||||
|
||||
export const DEFAULT_LOCAL_SETTINGS: LocalSettings = {
|
||||
context: {
|
||||
model: "deepseek-v3.2",
|
||||
thinking_enabled: true,
|
||||
},
|
||||
};
|
||||
|
||||
const LOCAL_SETTINGS_KEY = "deerflow.local-settings";
|
||||
|
||||
export interface LocalSettings {
|
||||
context: Omit<AgentThreadContext, "thread_id">;
|
||||
}
|
||||
|
||||
export function getLocalSettings(): LocalSettings {
|
||||
if (typeof window === "undefined") {
|
||||
return DEFAULT_LOCAL_SETTINGS;
|
||||
}
|
||||
const json = localStorage.getItem(LOCAL_SETTINGS_KEY);
|
||||
try {
|
||||
if (json) {
|
||||
return JSON.parse(json);
|
||||
}
|
||||
} catch {}
|
||||
return DEFAULT_LOCAL_SETTINGS;
|
||||
}
|
||||
|
||||
export function saveLocalSettings(settings: LocalSettings) {
|
||||
localStorage.setItem(LOCAL_SETTINGS_KEY, JSON.stringify(settings));
|
||||
}
|
||||
|
||||
export function updateContextOfLocalSettings(
|
||||
context: LocalSettings["context"],
|
||||
) {
|
||||
const settings = getLocalSettings();
|
||||
saveLocalSettings({
|
||||
...settings,
|
||||
context: {
|
||||
...settings.context,
|
||||
...context,
|
||||
},
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user