open-webui/src/lib/components/chat/Overview/Node.svelte
Timothy Jaeryang Baek 6e468c5b95 refac
2026-08-17 00:50:54 -07:00

102 lines
3.2 KiB
Svelte

<script lang="ts">
import { WEBUI_API_BASE_URL } from '$lib/constants';
import { Handle, Position, type NodeProps } from '@xyflow/svelte';
import { getContext } from 'svelte';
import ProfileImage from '../Messages/ProfileImage.svelte';
import Tooltip from '$lib/components/common/Tooltip.svelte';
import Heart from '$lib/components/icons/Heart.svelte';
import { getOutputText } from '../Messages/structuredOutput';
const i18n = getContext('i18n');
type $$Props = NodeProps;
export let data: $$Props['data'];
const getMessageContent = (nodeData: any) =>
getOutputText(nodeData?.message?.output) || nodeData?.message?.content || '';
$: messageContent = getMessageContent(data);
</script>
<div
class="px-4 py-3 shadow-md rounded-xl dark:bg-black bg-white border border-gray-100 dark:border-gray-900 w-60 h-20 group"
>
<Tooltip
content={data?.message?.error ? data.message.error.content : messageContent}
class="w-full"
allowHTML={false}
>
{#if data.message.role === 'user'}
<div class="flex w-full">
<ProfileImage
src={`${WEBUI_API_BASE_URL}/users/${data.user.id}/profile/image`}
className={'size-5 -translate-y-[1px] flex-shrink-0'}
/>
<div class="ml-2 flex-1 min-w-0">
<div class=" flex justify-between items-center">
<div class="text-xs text-black dark:text-white font-normal line-clamp-1">
{data?.user?.name ?? 'User'}
</div>
</div>
{#if data?.message?.error}
<div class="text-red-500 line-clamp-2 text-xs mt-0.5">{data.message.error.content}</div>
{:else}
<div class="text-gray-500 line-clamp-2 text-xs mt-0.5">{messageContent}</div>
{/if}
</div>
</div>
{:else}
<div class="flex w-full">
<ProfileImage
src={`${WEBUI_API_BASE_URL}/models/model/profile/image?id=${data.model?.id ?? data.message.model}&lang=${$i18n.language}`}
className={'size-5 -translate-y-[1px] flex-shrink-0'}
/>
<div class="ml-2 flex-1 min-w-0">
<div class=" flex justify-between items-center">
<div class="text-xs text-black dark:text-white font-normal line-clamp-1">
{data?.model?.name ?? data?.message?.model ?? 'Assistant'}
</div>
<button
class={data?.message?.favorite ? '' : 'hover-reveal'}
aria-label={data?.message?.favorite
? $i18n.t('Remove from favorites')
: $i18n.t('Add to favorites')}
on:click={() => {
data.message.favorite = !(data?.message?.favorite ?? false);
}}
>
<Heart
className="size-3 {data?.message?.favorite
? 'fill-red-500 stroke-red-500'
: 'hover:fill-red-500 hover:stroke-red-500'} "
strokeWidth="2.5"
/>
</button>
</div>
{#if data?.message?.error}
<div class="text-red-500 line-clamp-2 text-xs mt-0.5">
{data.message.error.content}
</div>
{:else}
<div class="text-gray-500 line-clamp-2 text-xs mt-0.5">{messageContent}</div>
{/if}
</div>
</div>
{/if}
</Tooltip>
<Handle
type="target"
position={data?.direction === 'horizontal' ? Position.Left : Position.Top}
class="w-2 rounded-full dark:bg-gray-900"
/>
<Handle
type="source"
position={data?.direction === 'horizontal' ? Position.Right : Position.Bottom}
class="w-2 rounded-full dark:bg-gray-900"
/>
</div>