Function useChatModel

ChatGPT/src/hooks/useChatModel.ts:8–26

(key: string, file = CHAT_MODEL_JSON)

import useInit from '@/hooks/useInit';

export default function useChatModel(key: string, file = CHAT_MODEL_JSON) {
  const [modelJson, setModelJson] = useState<Record<string, any>>({});

  useInit(async () => {
    const data = await readJSON(file, {
      defaultVal: { name: 'ChatGPT Model', [key]: null },
    });
    setModelJson(data);
  });

  const modelSet = async (data: Record<string, any>[] | Record<string, any>) => {
    const oData = clone(modelJson);
    oData[key] = data;
    await writeJSON(file, oData);
    setModelJson(oData);
  };

  return { modelJson, modelSet, modelData: modelJson?.[key] || [] };
}

export function useCacheModel(file = '') {
  const [modelCacheJson, setModelCacheJson] = useState<Record<string, any>[]>([]);