ModelSelector.svelte 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141
  1. <script lang="ts">
  2. import { setDefaultModels } from '$lib/apis/configs';
  3. import { models, showSettings, settings, user } from '$lib/stores';
  4. import { onMount, tick, getContext } from 'svelte';
  5. import { toast } from 'svelte-sonner';
  6. const i18n = getContext('i18n');
  7. export let selectedModels = [''];
  8. export let disabled = false;
  9. const saveDefaultModel = async () => {
  10. const hasEmptyModel = selectedModels.filter((it) => it === '');
  11. if (hasEmptyModel.length) {
  12. toast.error($i18n.t('Choose a model before saving...'));
  13. return;
  14. }
  15. settings.set({ ...$settings, models: selectedModels });
  16. localStorage.setItem('settings', JSON.stringify($settings));
  17. if ($user.role === 'admin') {
  18. console.log('setting default models globally');
  19. await setDefaultModels(localStorage.token, selectedModels.join(','));
  20. }
  21. toast.success($i18n.t('Default model updated'));
  22. };
  23. $: if (selectedModels.length > 0 && $models.length > 0) {
  24. selectedModels = selectedModels.map((model) =>
  25. $models.map((m) => m.id).includes(model) ? model : ''
  26. );
  27. }
  28. </script>
  29. <div class="flex flex-col my-2">
  30. {#each selectedModels as selectedModel, selectedModelIdx}
  31. <div class="flex">
  32. <select
  33. id="models"
  34. class="outline-none bg-transparent text-lg font-semibold rounded-lg block w-full placeholder-gray-400"
  35. bind:value={selectedModel}
  36. {disabled}
  37. >
  38. <option class=" text-gray-700" value="" selected disabled
  39. >{$i18n.t('Select a model')}</option
  40. >
  41. {#each $models as model}
  42. {#if model.name === 'hr'}
  43. <hr />
  44. {:else}
  45. <option value={model.id} class="text-gray-700 text-lg"
  46. >{model.name +
  47. `${model.size ? ` (${(model.size / 1024 ** 3).toFixed(1)}GB)` : ''}`}</option
  48. >
  49. {/if}
  50. {/each}
  51. </select>
  52. {#if selectedModelIdx === 0}
  53. <button
  54. class=" self-center {selectedModelIdx === 0
  55. ? 'mr-3'
  56. : 'mr-7'} disabled:text-gray-600 disabled:hover:text-gray-600"
  57. {disabled}
  58. on:click={() => {
  59. selectedModels = [...selectedModels, ''];
  60. }}
  61. >
  62. <svg
  63. xmlns="http://www.w3.org/2000/svg"
  64. fill="none"
  65. viewBox="0 0 24 24"
  66. stroke-width="1.5"
  67. stroke="currentColor"
  68. class="w-4 h-4"
  69. >
  70. <path stroke-linecap="round" stroke-linejoin="round" d="M12 6v12m6-6H6" />
  71. </svg>
  72. </button>
  73. {:else}
  74. <button
  75. class=" self-center disabled:text-gray-600 disabled:hover:text-gray-600 {selectedModelIdx ===
  76. 0
  77. ? 'mr-3'
  78. : 'mr-7'}"
  79. {disabled}
  80. on:click={() => {
  81. selectedModels.splice(selectedModelIdx, 1);
  82. selectedModels = selectedModels;
  83. }}
  84. >
  85. <svg
  86. xmlns="http://www.w3.org/2000/svg"
  87. fill="none"
  88. viewBox="0 0 24 24"
  89. stroke-width="1.5"
  90. stroke="currentColor"
  91. class="w-4 h-4"
  92. >
  93. <path stroke-linecap="round" stroke-linejoin="round" d="M19.5 12h-15" />
  94. </svg>
  95. </button>
  96. {/if}
  97. {#if selectedModelIdx === 0}
  98. <button
  99. class=" self-center dark:hover:text-gray-300"
  100. id="open-settings-button"
  101. on:click={async () => {
  102. await showSettings.set(!$showSettings);
  103. }}
  104. >
  105. <svg
  106. xmlns="http://www.w3.org/2000/svg"
  107. fill="none"
  108. viewBox="0 0 24 24"
  109. stroke-width="1.5"
  110. stroke="currentColor"
  111. class="w-4 h-4"
  112. >
  113. <path
  114. stroke-linecap="round"
  115. stroke-linejoin="round"
  116. d="M10.343 3.94c.09-.542.56-.94 1.11-.94h1.093c.55 0 1.02.398 1.11.94l.149.894c.07.424.384.764.78.93.398.164.855.142 1.205-.108l.737-.527a1.125 1.125 0 011.45.12l.773.774c.39.389.44 1.002.12 1.45l-.527.737c-.25.35-.272.806-.107 1.204.165.397.505.71.93.78l.893.15c.543.09.94.56.94 1.109v1.094c0 .55-.397 1.02-.94 1.11l-.893.149c-.425.07-.765.383-.93.78-.165.398-.143.854.107 1.204l.527.738c.32.447.269 1.06-.12 1.45l-.774.773a1.125 1.125 0 01-1.449.12l-.738-.527c-.35-.25-.806-.272-1.203-.107-.397.165-.71.505-.781.929l-.149.894c-.09.542-.56.94-1.11.94h-1.094c-.55 0-1.019-.398-1.11-.94l-.148-.894c-.071-.424-.384-.764-.781-.93-.398-.164-.854-.142-1.204.108l-.738.527c-.447.32-1.06.269-1.45-.12l-.773-.774a1.125 1.125 0 01-.12-1.45l.527-.737c.25-.35.273-.806.108-1.204-.165-.397-.505-.71-.93-.78l-.894-.15c-.542-.09-.94-.56-.94-1.109v-1.094c0-.55.398-1.02.94-1.11l.894-.149c.424-.07.765-.383.93-.78.165-.398.143-.854-.107-1.204l-.527-.738a1.125 1.125 0 01.12-1.45l.773-.773a1.125 1.125 0 011.45-.12l.737.527c.35.25.807.272 1.204.107.397-.165.71-.505.78-.929l.15-.894z"
  117. />
  118. <path
  119. stroke-linecap="round"
  120. stroke-linejoin="round"
  121. d="M15 12a3 3 0 11-6 0 3 3 0 016 0z"
  122. />
  123. </svg>
  124. </button>
  125. {/if}
  126. </div>
  127. {/each}
  128. </div>
  129. <div class="text-left mt-1.5 text-xs text-gray-500">
  130. <button on:click={saveDefaultModel}> {$i18n.t('Set as default')}</button>
  131. </div>