70 lines
2.5 KiB
Vue
70 lines
2.5 KiB
Vue
<script setup lang="ts">
|
|
import Error from './Error.vue';
|
|
import Reasoning from './Reasoning.vue';
|
|
import Text from './Text.vue';
|
|
import Tool from './Tool/index.vue';
|
|
import type { MessageEntity } from '~/composables/useChat';
|
|
|
|
const props = defineProps<{
|
|
message: MessageEntity
|
|
}>();
|
|
|
|
const triggerRef = ref<HTMLElement | null>(null);
|
|
|
|
const { open, close, isOpen } = useTokenDropdown();
|
|
|
|
const toggleTokenDropdown = () => {
|
|
if (isOpen.value) {
|
|
close();
|
|
return;
|
|
}
|
|
|
|
open(props.message.generation!, triggerRef.value!);
|
|
};
|
|
</script>
|
|
|
|
<template>
|
|
<div class="flex flex-col w-full gap-2">
|
|
<div v-for="part in message.parts" :key="part.id">
|
|
<Reasoning v-if="part.type === 'reasoning'" :part="part" />
|
|
<Text v-else-if="part.type === 'text'" :part="part" />
|
|
<Tool v-else-if="part.type === 'tool-call'" :toolCall="part.toolCall!" />
|
|
<div v-else>
|
|
Unhandled part type: {{ part.type }} {{ part }}
|
|
</div>
|
|
</div>
|
|
|
|
<span class="flex items-center gap-1"
|
|
v-if="message.generation?.status === 'pending' && (message.parts || []).length === 0">
|
|
<span class="i-svg-spinners-pulse-2 text-4"></span>
|
|
<span class="text-sm text-[var(--text-secondary)]">
|
|
Preparing Generation...
|
|
</span>
|
|
</span>
|
|
|
|
<div v-else-if="message.generation?.status === 'failed'">
|
|
<Error :error="message.generation.error" />
|
|
</div>
|
|
|
|
<div class="flex flex-row justify-between text-[var(--text-tertiary)] text-xs"
|
|
v-if="message.generation && message.generation.status !== 'pending'">
|
|
<div class="flex items-center gap-1">
|
|
<ModelIcon :size="12" :model-id="message.generation.modelId" />
|
|
{{ message.generation.modelId }}
|
|
</div>
|
|
|
|
<button ref="triggerRef" @click="toggleTokenDropdown"
|
|
class="flex gap-2 hover:bg-[var(--color-hover)] px-1 rounded transition-colors duration-150 ease-[cubic-bezier(0.5,_1,_0.89,_1)]">
|
|
<span class="flex gap-1 items-center" v-if="message.generation.tokens?.output">
|
|
<span class="i-tabler-coins"></span>
|
|
{{ message.generation?.tokens?.output }}
|
|
</span>
|
|
|
|
<span v-if="message.generation.tokens?.tps">
|
|
{{ message.generation.tokens.tps.toFixed(1) }} tps
|
|
</span>
|
|
</button>
|
|
</div>
|
|
</div>
|
|
</template>
|