Hansimov commited on
Commit
3fb1e60
1 Parent(s): 1140b5d

:gem: [Feature] Setup hardcoded available models

Browse files
components/inputs_binder.js CHANGED
@@ -1,12 +1,12 @@
1
  import {
2
- setup_available_models_on_select,
3
  setup_temperature_on_select,
4
  } from "../components/llm_models_loader.js";
5
 
6
  export class InputsBinder {
7
  constructor() {}
8
  bind() {
9
- setup_available_models_on_select();
10
  setup_temperature_on_select();
11
  let user_input_resizer = new UserInputResizer();
12
  user_input_resizer.bind();
 
1
  import {
2
+ setup_hardcoded_available_models_on_select,
3
  setup_temperature_on_select,
4
  } from "../components/llm_models_loader.js";
5
 
6
  export class InputsBinder {
7
  constructor() {}
8
  bind() {
9
+ setup_hardcoded_available_models_on_select();
10
  setup_temperature_on_select();
11
  let user_input_resizer = new UserInputResizer();
12
  user_input_resizer.bind();
components/llm_models_loader.js CHANGED
@@ -3,6 +3,35 @@ import {
3
  available_models,
4
  } from "../networks/llm_requester.js";
5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  export async function setup_available_models_on_select(default_option = null) {
7
  var select = $("#available-models-select");
8
  select.empty();
 
3
  available_models,
4
  } from "../networks/llm_requester.js";
5
 
6
+ export async function setup_hardcoded_available_models_on_select(
7
+ default_option = null
8
+ ) {
9
+ var select = $("#available-models-select");
10
+ select.empty();
11
+ const hardcoded_models = [
12
+ "pplx-7b-chat",
13
+ "pplx-70b-chat",
14
+ "pplx-7b-online",
15
+ "pplx-70b-online",
16
+ "llama-2-70b-chat",
17
+ "codellama-34b-instruct",
18
+ "mistral-7b-instruct",
19
+ ];
20
+ hardcoded_models.forEach((value, index) => {
21
+ const option = new Option(value, value);
22
+ select.append(option);
23
+ });
24
+ let default_model = "pplx-7b-chat";
25
+ let local_default_model = localStorage.getItem("default_model");
26
+ if (local_default_model && hardcoded_models.includes(local_default_model)) {
27
+ default_model = local_default_model;
28
+ } else {
29
+ localStorage.setItem("default_model", default_model);
30
+ }
31
+ select.val(default_model);
32
+ console.log(`Default model: ${select.val()}`);
33
+ }
34
+
35
  export async function setup_available_models_on_select(default_option = null) {
36
  var select = $("#available-models-select");
37
  select.empty();