56 lines
2.4 KiB
Vue
56 lines
2.4 KiB
Vue
<script setup lang="ts">
|
|
import type { Entity } from '@triplit/client';
|
|
import type schema from '#triplit/schema';
|
|
import ShikiHighlight from '~/components/Markdown/ShikiHighlight.vue';
|
|
import Reasoning from './Reasoning.vue';
|
|
import Text from './Text.vue';
|
|
import Tool from './Tool/index.vue';
|
|
|
|
defineProps<{
|
|
message: Readonly<
|
|
Entity<typeof schema, 'messages'> & {
|
|
parts: (Entity<typeof schema, 'message_parts'> & {
|
|
toolCall: Entity<typeof schema, 'tool_calls'> | null;
|
|
})[];
|
|
} & { generation: Entity<typeof schema, 'generations'> | null }
|
|
>;
|
|
}>();
|
|
</script>
|
|
|
|
<template>
|
|
<!-- <pre class="max-w-full overflow-x-auto">{{ JSON.stringify(message, null, 2) }}</pre> -->
|
|
<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>
|
|
<div class="flex flex-row justify-between text-zinc-400 dark:text-zinc-600 text-xs"
|
|
v-if="message.generation && message.generation.status !== 'pending'">
|
|
<span class="flex items-center gap-1">
|
|
<ModelIcon :size="12" :model-id="message.generation.modelId" />
|
|
{{ message.generation.modelId }}
|
|
</span>
|
|
<span class="flex gap-1 items-center" v-if="message.generation.tokens?.output">
|
|
<Icon name="tabler:coins" />
|
|
{{ message.generation?.tokens?.output }}
|
|
</span>
|
|
</div>
|
|
</div>
|
|
<span v-if="message.generation?.status === 'pending' && message.parts.length === 0">
|
|
<span class="text-sm text-[var(--color-muted)] flex flex-row items-center">
|
|
<Icon name="svg-spinners:pulse-2" class="text-4" />
|
|
Preparing generating...
|
|
</span>
|
|
</span>
|
|
<div v-else-if="message.generation?.status === 'failed'">
|
|
<span class="text-sm text-[var(--color-error)]">Generation failed</span>
|
|
<div class="text-sm">
|
|
<ShikiHighlight :code="message.generation.error ?? 'An unknown error occurred'" lang="json" />
|
|
</div>
|
|
</div>
|
|
</template>
|