新增用户意图确认功能
This commit is contained in:
likun committed
1 parent
cf2e4bdd9c
commit
b8926f00d7
574 files changed
+97664
-97208
No files matched your search
@@ -1,150 +1,150 @@
|
||||
import {
|
||||
DEFAULT_LOCAL_SETTINGS,
|
||||
LOCAL_SETTINGS_KEY,
|
||||
THREAD_MODEL_KEY_PREFIX,
|
||||
getLocalSettings,
|
||||
getThreadModelName,
|
||||
saveLocalSettings,
|
||||
saveThreadModelName,
|
||||
type LocalSettings,
|
||||
} from "./local";
|
||||
|
||||
type Listener = () => void;
|
||||
|
||||
export type LocalSettingsSetter = <K extends keyof LocalSettings>(
|
||||
key: K,
|
||||
value: Partial<LocalSettings[K]>,
|
||||
) => void;
|
||||
|
||||
const listeners = new Set<Listener>();
|
||||
const threadModelNames = new Map<string, string | undefined>();
|
||||
|
||||
let baseSettings: LocalSettings = DEFAULT_LOCAL_SETTINGS;
|
||||
let baseSettingsLoaded = false;
|
||||
let storageListenerRegistered = false;
|
||||
|
||||
function emitChange() {
|
||||
for (const listener of listeners) {
|
||||
listener();
|
||||
}
|
||||
}
|
||||
|
||||
function ensureBaseSettingsLoaded() {
|
||||
if (baseSettingsLoaded || typeof window === "undefined") {
|
||||
return;
|
||||
}
|
||||
|
||||
baseSettings = getLocalSettings();
|
||||
baseSettingsLoaded = true;
|
||||
}
|
||||
|
||||
function ensureStorageListenerRegistered() {
|
||||
if (storageListenerRegistered || typeof window === "undefined") {
|
||||
return;
|
||||
}
|
||||
|
||||
window.addEventListener("storage", handleStorage);
|
||||
storageListenerRegistered = true;
|
||||
}
|
||||
|
||||
function mergeSettingsSection<K extends keyof LocalSettings>(
|
||||
settings: LocalSettings,
|
||||
key: K,
|
||||
value: Partial<LocalSettings[K]>,
|
||||
): LocalSettings {
|
||||
return {
|
||||
...settings,
|
||||
[key]: {
|
||||
...settings[key],
|
||||
...value,
|
||||
},
|
||||
} as LocalSettings;
|
||||
}
|
||||
|
||||
function handleStorage(event: StorageEvent) {
|
||||
if (event.storageArea && event.storageArea !== localStorage) {
|
||||
return;
|
||||
}
|
||||
|
||||
ensureBaseSettingsLoaded();
|
||||
|
||||
if (event.key === null) {
|
||||
baseSettings = getLocalSettings();
|
||||
threadModelNames.clear();
|
||||
emitChange();
|
||||
return;
|
||||
}
|
||||
|
||||
if (event.key === LOCAL_SETTINGS_KEY) {
|
||||
baseSettings = getLocalSettings();
|
||||
emitChange();
|
||||
return;
|
||||
}
|
||||
|
||||
if (!event.key.startsWith(THREAD_MODEL_KEY_PREFIX)) {
|
||||
return;
|
||||
}
|
||||
|
||||
const threadId = event.key.slice(THREAD_MODEL_KEY_PREFIX.length);
|
||||
threadModelNames.set(threadId, getThreadModelName(threadId));
|
||||
emitChange();
|
||||
}
|
||||
|
||||
export function subscribe(listener: Listener): () => void {
|
||||
ensureBaseSettingsLoaded();
|
||||
ensureStorageListenerRegistered();
|
||||
listeners.add(listener);
|
||||
|
||||
return () => {
|
||||
listeners.delete(listener);
|
||||
};
|
||||
}
|
||||
|
||||
export function getBaseSettingsSnapshot(): LocalSettings {
|
||||
ensureBaseSettingsLoaded();
|
||||
return baseSettings;
|
||||
}
|
||||
|
||||
export function getThreadModelSnapshot(threadId: string): string | undefined {
|
||||
ensureBaseSettingsLoaded();
|
||||
|
||||
if (!threadModelNames.has(threadId)) {
|
||||
threadModelNames.set(threadId, getThreadModelName(threadId));
|
||||
}
|
||||
|
||||
return threadModelNames.get(threadId);
|
||||
}
|
||||
|
||||
export const updateLocalSettings: LocalSettingsSetter = (key, value) => {
|
||||
ensureBaseSettingsLoaded();
|
||||
ensureStorageListenerRegistered();
|
||||
|
||||
baseSettings = mergeSettingsSection(baseSettings, key, value);
|
||||
saveLocalSettings(baseSettings);
|
||||
emitChange();
|
||||
};
|
||||
|
||||
export function updateThreadSettings<K extends keyof LocalSettings>(
|
||||
threadId: string,
|
||||
key: K,
|
||||
value: Partial<LocalSettings[K]>,
|
||||
) {
|
||||
ensureBaseSettingsLoaded();
|
||||
ensureStorageListenerRegistered();
|
||||
|
||||
const nextBaseSettings = mergeSettingsSection(baseSettings, key, value);
|
||||
baseSettings = nextBaseSettings;
|
||||
saveLocalSettings(baseSettings);
|
||||
|
||||
if (
|
||||
key === "context" &&
|
||||
Object.prototype.hasOwnProperty.call(value, "model_name")
|
||||
) {
|
||||
const contextValue = value as Partial<LocalSettings["context"]>;
|
||||
const threadModelName = contextValue.model_name;
|
||||
threadModelNames.set(threadId, threadModelName);
|
||||
saveThreadModelName(threadId, threadModelName);
|
||||
}
|
||||
|
||||
emitChange();
|
||||
}
|
||||
import {
|
||||
DEFAULT_LOCAL_SETTINGS,
|
||||
LOCAL_SETTINGS_KEY,
|
||||
THREAD_MODEL_KEY_PREFIX,
|
||||
getLocalSettings,
|
||||
getThreadModelName,
|
||||
saveLocalSettings,
|
||||
saveThreadModelName,
|
||||
type LocalSettings,
|
||||
} from "./local";
|
||||
|
||||
type Listener = () => void;
|
||||
|
||||
export type LocalSettingsSetter = <K extends keyof LocalSettings>(
|
||||
key: K,
|
||||
value: Partial<LocalSettings[K]>,
|
||||
) => void;
|
||||
|
||||
const listeners = new Set<Listener>();
|
||||
const threadModelNames = new Map<string, string | undefined>();
|
||||
|
||||
let baseSettings: LocalSettings = DEFAULT_LOCAL_SETTINGS;
|
||||
let baseSettingsLoaded = false;
|
||||
let storageListenerRegistered = false;
|
||||
|
||||
function emitChange() {
|
||||
for (const listener of listeners) {
|
||||
listener();
|
||||
}
|
||||
}
|
||||
|
||||
function ensureBaseSettingsLoaded() {
|
||||
if (baseSettingsLoaded || typeof window === "undefined") {
|
||||
return;
|
||||
}
|
||||
|
||||
baseSettings = getLocalSettings();
|
||||
baseSettingsLoaded = true;
|
||||
}
|
||||
|
||||
function ensureStorageListenerRegistered() {
|
||||
if (storageListenerRegistered || typeof window === "undefined") {
|
||||
return;
|
||||
}
|
||||
|
||||
window.addEventListener("storage", handleStorage);
|
||||
storageListenerRegistered = true;
|
||||
}
|
||||
|
||||
function mergeSettingsSection<K extends keyof LocalSettings>(
|
||||
settings: LocalSettings,
|
||||
key: K,
|
||||
value: Partial<LocalSettings[K]>,
|
||||
): LocalSettings {
|
||||
return {
|
||||
...settings,
|
||||
[key]: {
|
||||
...settings[key],
|
||||
...value,
|
||||
},
|
||||
} as LocalSettings;
|
||||
}
|
||||
|
||||
function handleStorage(event: StorageEvent) {
|
||||
if (event.storageArea && event.storageArea !== localStorage) {
|
||||
return;
|
||||
}
|
||||
|
||||
ensureBaseSettingsLoaded();
|
||||
|
||||
if (event.key === null) {
|
||||
baseSettings = getLocalSettings();
|
||||
threadModelNames.clear();
|
||||
emitChange();
|
||||
return;
|
||||
}
|
||||
|
||||
if (event.key === LOCAL_SETTINGS_KEY) {
|
||||
baseSettings = getLocalSettings();
|
||||
emitChange();
|
||||
return;
|
||||
}
|
||||
|
||||
if (!event.key.startsWith(THREAD_MODEL_KEY_PREFIX)) {
|
||||
return;
|
||||
}
|
||||
|
||||
const threadId = event.key.slice(THREAD_MODEL_KEY_PREFIX.length);
|
||||
threadModelNames.set(threadId, getThreadModelName(threadId));
|
||||
emitChange();
|
||||
}
|
||||
|
||||
export function subscribe(listener: Listener): () => void {
|
||||
ensureBaseSettingsLoaded();
|
||||
ensureStorageListenerRegistered();
|
||||
listeners.add(listener);
|
||||
|
||||
return () => {
|
||||
listeners.delete(listener);
|
||||
};
|
||||
}
|
||||
|
||||
export function getBaseSettingsSnapshot(): LocalSettings {
|
||||
ensureBaseSettingsLoaded();
|
||||
return baseSettings;
|
||||
}
|
||||
|
||||
export function getThreadModelSnapshot(threadId: string): string | undefined {
|
||||
ensureBaseSettingsLoaded();
|
||||
|
||||
if (!threadModelNames.has(threadId)) {
|
||||
threadModelNames.set(threadId, getThreadModelName(threadId));
|
||||
}
|
||||
|
||||
return threadModelNames.get(threadId);
|
||||
}
|
||||
|
||||
export const updateLocalSettings: LocalSettingsSetter = (key, value) => {
|
||||
ensureBaseSettingsLoaded();
|
||||
ensureStorageListenerRegistered();
|
||||
|
||||
baseSettings = mergeSettingsSection(baseSettings, key, value);
|
||||
saveLocalSettings(baseSettings);
|
||||
emitChange();
|
||||
};
|
||||
|
||||
export function updateThreadSettings<K extends keyof LocalSettings>(
|
||||
threadId: string,
|
||||
key: K,
|
||||
value: Partial<LocalSettings[K]>,
|
||||
) {
|
||||
ensureBaseSettingsLoaded();
|
||||
ensureStorageListenerRegistered();
|
||||
|
||||
const nextBaseSettings = mergeSettingsSection(baseSettings, key, value);
|
||||
baseSettings = nextBaseSettings;
|
||||
saveLocalSettings(baseSettings);
|
||||
|
||||
if (
|
||||
key === "context" &&
|
||||
Object.prototype.hasOwnProperty.call(value, "model_name")
|
||||
) {
|
||||
const contextValue = value as Partial<LocalSettings["context"]>;
|
||||
const threadModelName = contextValue.model_name;
|
||||
threadModelNames.set(threadId, threadModelName);
|
||||
saveThreadModelName(threadId, threadModelName);
|
||||
}
|
||||
|
||||
emitChange();
|
||||
}
|
||||
Reference in new issue
Block a user