big-agi/lib/store.ts

import { create } from 'zustand';
import { persist } from 'zustand/middleware';


/// Settings Store

interface SettingsStore {
  apiKey: string;
  setApiKey: (apiKey: string) => void;

  wideMode: boolean;
  setWideMode: (wideMode: boolean) => void;

  freeScroll: boolean;
  setFreeScroll: (freeScroll: boolean) => void;

  modelTemperature: number;
  setModelTemperature: (modelTemperature: number) => void;

  modelMaxTokens: number;
  setModelMaxTokens: (modelMaxTokens: number) => void;
}

export const useSettingsStore = create<SettingsStore>()(
  persist((set) => ({
      apiKey: (function() {
        // backwards compatibility from the former localStorage key
        if (typeof localStorage === 'undefined') return '';
        return localStorage.getItem('app-settings-openai-api-key') || '';
      })(),
      setApiKey: (apiKey: string) => set({ apiKey }),

      wideMode: false,
      setWideMode: (wideMode: boolean) => set({ wideMode }),

      freeScroll: false,
      setFreeScroll: (freeScroll: boolean) => set({ freeScroll }),

      modelTemperature: 0.5,
      setModelTemperature: (modelTemperature: number) => set({ modelTemperature }),

      modelMaxTokens: 2048,
      setModelMaxTokens: (modelMaxTokens: number) => set({ modelMaxTokens }),
    }),
    {
      name: 'app-settings',
    }),
);


/// Composer Store

interface ComposerStore {
  history: {
    date: number,
    text: string,
    count: number,
  }[];

  appendMessageToHistory: (text: string) => void;
}

export const useComposerStore = create<ComposerStore>()(
  persist((set, get) => ({
      history: [],

      appendMessageToHistory: (text: string) => {
        const date = Date.now();
        const history = [...(get().history || [])];

        // take the item from the array, matching by text
        let item = history.find((item) => item.text === text);
        if (item) {
          history.splice(history.indexOf(item), 1);
          item.date = date;
          item.count++;
        } else
          item = { date, text, count: 1 };

        // prepend the item to the history array
        history.unshift(item);

        // update the store (limiting max items)
        set({ history: history.slice(0, 20) });
      },
    }),
    {
      name: 'app-composer',
    }),
);