feat: external model info

This commit is contained in:
Timothy J. Baek 2024-03-25 13:46:06 -07:00
parent eb004ccfc2
commit 32cb5f3ceb
3 changed files with 27 additions and 21 deletions

View file

@ -33,7 +33,7 @@ export const getLiteLLMModels = async (token: string = '') => {
id: model.id, id: model.id,
name: model.name ?? model.id, name: model.name ?? model.id,
external: true, external: true,
source: 'litellm' source: 'LiteLLM'
})) }))
.sort((a, b) => { .sort((a, b) => {
return a.name.localeCompare(b.name); return a.name.localeCompare(b.name);

View file

@ -12,7 +12,7 @@
import { user, MODEL_DOWNLOAD_POOL, models } from '$lib/stores'; import { user, MODEL_DOWNLOAD_POOL, models } from '$lib/stores';
import { toast } from 'svelte-sonner'; import { toast } from 'svelte-sonner';
import { getModels, splitStream } from '$lib/utils'; import { capitalizeFirstLetter, getModels, splitStream } from '$lib/utils';
import Tooltip from '$lib/components/common/Tooltip.svelte'; import Tooltip from '$lib/components/common/Tooltip.svelte';
const i18n = getContext('i18n'); const i18n = getContext('i18n');
@ -225,6 +225,7 @@
</div> </div>
{#if item.info.external} {#if item.info.external}
<Tooltip content={item.info?.source ?? 'External'}>
<div class=" mr-2"> <div class=" mr-2">
<svg <svg
xmlns="http://www.w3.org/2000/svg" xmlns="http://www.w3.org/2000/svg"
@ -244,6 +245,7 @@
/> />
</svg> </svg>
</div> </div>
</Tooltip>
{/if} {/if}
</div> </div>

View file

@ -31,6 +31,10 @@ export const getModels = async (token: string) => {
// Helper functions // Helper functions
////////////////////////// //////////////////////////
export const capitalizeFirstLetter = (string) => {
return string.charAt(0).toUpperCase() + string.slice(1);
};
export const splitStream = (splitOn) => { export const splitStream = (splitOn) => {
let buffer = ''; let buffer = '';
return new TransformStream({ return new TransformStream({