:boom: [Fix] auto switch endpoint for selected model

networks/endpoint_storage.js

@@ -9,6 +9,7 @@ class EndpointStorage {
     constructor() {
         this.init_database();
         this.create_endpoint_and_api_key_items();
+        this.fill_available_models_select("user-customized");
     }
     init_database() {
         this.db = new Dexie("endpoints");
@@ -82,7 +83,7 @@ class EndpointStorage {
     }
     bind_endpoint_and_api_key_buttons(endpoint_and_api_key_item) {
         let self = this;
-        console.log("endpoint_and_api_key_item:", endpoint_and_api_key_item);
+        // console.log("endpoint_and_api_key_item:", endpoint_and_api_key_item);
         let endpoint_submit_button = endpoint_and_api_key_item.find(
             ".submit-endpoint-button"
         );
@@ -132,23 +133,29 @@ class EndpointStorage {
         console.log("fetch available models for endpoint:", endpoint);
         let available_models_requester = new AvailableModelsRequester(endpoint);
         await available_models_requester.get();
-        select.empty();
-        available_models.forEach((value, index) => {
+        available_models[endpoint].forEach((value, index) => {
             const option = new Option(value, value);
             select.append(option);
         });
 
+        let flatten_available_models = [];
+        Object.entries(available_models).forEach(([key, value]) => {
+            flatten_available_models.push(...value);
+        });
+        flatten_available_models = [...new Set(flatten_available_models)];
+        // console.log("flatten_available_models:", flatten_available_models);
+
         // set default model
         let default_model = "";
         let storage_default_model = localStorage.getItem("default_model");
         console.log("storage_default_model:", storage_default_model);
         if (
             storage_default_model &&
-            available_models.includes(storage_default_model)
+            flatten_available_models.includes(storage_default_model)
         ) {
             default_model = storage_default_model;
-        } else if (available_models) {
-            default_model = available_models[0];
+        } else if (flatten_available_models) {
+            default_model = flatten_available_models[0];
             localStorage.setItem("default_model", default_model);
         } else {
             default_model = "";

networks/llm_requester.js

@@ -32,7 +32,7 @@ export class ChatCompletionsRequester {
             temperature !== null ? temperature : get_selected_temperature();
 
         this.openai_endpoint =
-            openai_endpoint || localStorage.getItem("openai_endpoint");
+            openai_endpoint || get_endpoint_by_model(this.model);
         this.backend_request_endpoint = "/chat/completions";
         this.controller = new AbortController();
     }
@@ -98,7 +98,7 @@ export class ChatCompletionsRequester {
     }
 }
 
-export var available_models = ["notes"];
+export var available_models = { "user-customized": ["notes"] };
 export class AvailableModelsRequester {
     constructor(openai_endpoint) {
         this.openai_endpoint = openai_endpoint;
@@ -136,13 +136,18 @@ export class AvailableModelsRequester {
             .then((response) => response.json())
             .then((response_json) => {
                 let data = response_json.data;
+                if (!(this.openai_endpoint in available_models)) {
+                    available_models[this.openai_endpoint] = [];
+                }
                 data.forEach((item) => {
-                    if (!(item.id in available_models)) {
-                        available_models.push(item.id);
+                    if (
+                        !available_models[this.openai_endpoint].includes(
+                            item.id
+                        )
+                    ) {
+                        available_models[this.openai_endpoint].push(item.id);
                     }
                 });
-                available_models.sort();
-                available_models = [...new Set(available_models)];
                 console.log("available_models:", available_models);
             })
             .catch((error) => {
@@ -153,3 +158,14 @@ export class AvailableModelsRequester {
         this.controller.abort();
     }
 }
+
+export function get_endpoint_by_model(model) {
+    let endpoint = "";
+    Object.entries(available_models).forEach(([key, value]) => {
+        if (value.includes(model)) {
+            endpoint = key;
+            return endpoint;
+        }
+    });
+    return endpoint;
+}