Function useModels

src/hooks/useModels.ts:7–112

()

import { updateModel } from "../lib/models";

export function useModels() {
  const [models, setModels] = useState<Model[]>([]);
  const [isLoading, setIsLoading] = useState<boolean>(true);
  const [error, setError] = useState<string>();

  const loadModels = async () => {
    // Get the model settings
    setIsLoading(true);
    const items = await LocalStorage.allItems();
    const modelObjs = Object.entries(items)
      .filter(([key]) => key.startsWith("--model-"))
      .map(([, value]) => JSON.parse(value));
    setModels(modelObjs);
    setIsLoading(false);
  };

  useEffect(() => {
    Promise.resolve(installDefaults()).then(() => {
      Promise.resolve(loadModels());
    });
  }, []);

  const revalidate = async () => {
    return loadModels();
  };

  const dummyModel = (): Model => {
    return {
      name: "",
      description: "",
      icon: Icon.Gear,
      iconColor: Color.Red,
      favorited: false,
      endpoint: "",
      authType: "",
      apiKey: "",
      inputSchema: "",
      outputKeyPath: "",
      outputTiming: "sync",
      lengthLimit: "2500",
      id: "",
      notes: "",
      isDefault: false,
      temperature: "1.0",
    };
  };

  const createModel = async (newData: Model & { [key: string]: string | boolean }) => {
    // Check if the name is empty
    if (!newData.name) {
      setError("Name cannot be empty.");
      return false;
    }

    // Check if a model with that name already exists
    if (models.find((model) => model.name == newData.name)) {
      setError("A model with that name already exists.");
      return false;