Add axios request to fetch models template

Author: small-stone

src/views/App.tsx

@@ -8,6 +8,7 @@ import { useMst } from "./stores/RootStore";
 import MessageUtil from "@/util/MessageUtil";
 import "./App.css";
 import "./i18n";
+import axios from "axios";
 
 export default function App() {
   const [ready, setReady] = useState(false);
@@ -25,11 +26,17 @@ export default function App() {
   };
 
   const getConfig = () => {
-    MessageUtil.sendMessage({ command: "readConfig", key: "" });
     MessageUtil.registerHandler("readConfig", (data: { value: any }) => {
       config.setConfig(data.value);
       setReady(true);
     });
+    // 获取 models 模版列表
+    axios.get("https://api-test.devchat.ai/v1/models").then((res) => {
+      if (res?.data?.data && Array.isArray(res?.data?.data)) {
+        config.setTemplate(res.data.data);
+        MessageUtil.sendMessage({ command: "readConfig", key: "" });
+      }
+    });
   };
 
   useEffect(() => {

src/views/stores/ConfigStore.ts

@@ -4,17 +4,32 @@ import modelsTemplate from "@/models";
 import cloneDeep from "lodash.clonedeep";
 
 const defaultAPIBase = [
-  "https://api.devchat.ai",
+  "https://api.devchat.ai/v1",
   "https://api.devchat-ai.cn/v1",
 ];
 
 export const ConfigStore = types
   .model("Config", {
     config: types.optional(types.frozen(), {}),
+    modelsTemplate: types.optional(types.frozen(), {}),
     settle: types.optional(types.boolean, false),
     defaultModel: types.optional(types.string, ""),
   })
   .actions((self) => ({
+    setTemplate: (value: any) => {
+      const models = value
+        .filter((item) => item.category === "chat")
+        .map((item) => {
+          return {
+            name: item.model,
+            max_input_tokens: item.max_input_tokens,
+            provider: "devchat",
+            stream: true,
+          };
+        });
+      console.log("models: ", models);
+      self.modelsTemplate = models;
+    },
     updateSettle: (value: boolean) => {
       self.settle = value;
     },
@@ -36,6 +51,12 @@ export const ConfigStore = types
     },
     getAPIBase: () => {
       if (self.config?.providers?.devchat?.api_base) {
+        if (
+          self.config.providers.devchat.api_base === "custom" &&
+          self.config.providers.devchat.cumstom_api_base
+        ) {
+          return self.config.providers.devchat.cumstom_api_base;
+        }
         return self.config.providers.devchat.api_base;
       }
       if (self.config?.providers?.openai?.api_base) {
@@ -49,7 +70,7 @@ export const ConfigStore = types
       if (!data.models) {
         newConfig.models = {};
       }
-      modelsTemplate.forEach((item) => {
+      self.modelsTemplate.forEach((item) => {
         const currentModel: any = {
           ...item,
         };
@@ -96,7 +117,7 @@ export const ConfigStore = types
       self.defaultModel = newConfig.default_model;
     },
     getModelList: () => {
-      const modelsArray = modelsTemplate.map((item) => {
+      const modelsArray = self.modelsTemplate.map((item) => {
         return item.name;
       });
       return modelsArray;