fix: remove useAllModels()

Author: Neet-Nestor

app/client/api.ts

@@ -58,7 +58,6 @@ export interface ModelRecord {
   size?: string;
   quantization?: string;
   context_length?: string;
-  is_default?: boolean;
   family?: string;
   vram_required_MB?: number;
   buffer_size_required_bytes?: number;

app/components/chat.tsx

@@ -91,7 +91,6 @@ import { ContextPrompts, TemplateAvatar, TemplateConfig } from "./template";
 import { ChatCommandPrefix, useChatCommand, useCommand } from "../command";
 import { prettyObject } from "../utils/format";
 import { ExportMessageModal } from "./exporter";
-import { useAllModels } from "../utils/hooks";
 import { MultimodalContent } from "../client/api";
 import { WebLLMContext } from "../client/webllm";
 import { Template, useTemplateStore } from "../store/template";
@@ -494,20 +493,7 @@ export function ChatActions(props: {
 
   // switch model
   const currentModel = config.modelConfig.model;
-  const allModels = useAllModels();
-  const models = useMemo(() => {
-    const defaultModel = allModels.find((m) => m.is_default);
-
-    if (defaultModel) {
-      const arr = [
-        defaultModel,
-        ...allModels.filter((m) => m !== defaultModel),
-      ];
-      return arr;
-    } else {
-      return allModels;
-    }
-  }, [allModels]);
+  const models = config.models;
   const [showModelSelector, setShowModelSelector] = useState(false);
   const [showUploadImage, setShowUploadImage] = useState(false);
 

app/components/model-config.tsx

@@ -1,17 +1,22 @@
-import { ModalConfigValidator, ModelConfig, ModelType } from "../store";
+import {
+  ModalConfigValidator,
+  ModelConfig,
+  ModelType,
+  useAppConfig,
+} from "../store";
 
 import Locale from "../locales";
 import { InputRange } from "./input-range";
 import { ListItem, Select } from "./ui-lib";
-import { useAllModels } from "../utils/hooks";
 import React from "react";
 
 export function ModelConfigList(props: {
   modelConfig: ModelConfig;
   selectModel: (model: ModelType) => void;
   updateConfig: (updater: (config: ModelConfig) => void) => void;
 }) {
-  const allModels = useAllModels();
+  const config = useAppConfig();
+  const models = config.models;
 
   return (
     <>
@@ -22,9 +27,9 @@ export function ModelConfigList(props: {
             props.selectModel(e.target.value);
           }}
         >
-          {allModels.map((v, i) => (
+          {models.map((v, i) => (
             <React.Fragment key={i}>
-              {i > 0 && v.family !== allModels[i - 1].family && <hr />}
+              {i > 0 && v.family !== models[i - 1].family && <hr />}
               <option value={v.name}>
                 {v.name}
                 {v.provider ? ` (${v.provider})` : ""}

app/store/config.ts

@@ -47,7 +47,6 @@ export const DEFAULT_CONFIG = {
 
   cacheType: "cache" as CacheType,
   logLevel: "INFO" as LogLevel,
-  customModels: "",
   models: DEFAULT_MODELS as any as ModelRecord[],
 
   modelConfig: {