77 lines
2.1 KiB
Vue
77 lines
2.1 KiB
Vue
<script setup lang="ts">
|
|
import type { ResearchStep } from '~/lib/deep-research'
|
|
|
|
export type TreeNodeStatus = Exclude<ResearchStep['type'], 'complete'>
|
|
|
|
export type TreeNode = {
|
|
id: string
|
|
/** Label, represents the search query */
|
|
label: string
|
|
researchGoal?: string
|
|
learnings?: string[]
|
|
followUpQuestions?: string[]
|
|
status?: TreeNodeStatus
|
|
children: TreeNode[]
|
|
/** Current depth of the node */
|
|
depth: number
|
|
/** Maximum breadth at the current depth */
|
|
breadth: number
|
|
/** Index of the node among its siblings */
|
|
index?: number
|
|
}
|
|
|
|
const props = defineProps<{
|
|
node: TreeNode
|
|
}>()
|
|
|
|
const emit = defineEmits<{
|
|
(e: 'select', value: TreeNode): void
|
|
}>()
|
|
|
|
const icon = computed(() => {
|
|
const result = { name: '', pulse: false }
|
|
if (!props.node.status) return result
|
|
switch (props.node.status) {
|
|
case 'generating_query':
|
|
result.name = 'i-lucide-clipboard-list'
|
|
result.pulse = true
|
|
break
|
|
case 'generated_query':
|
|
result.name = 'i-lucide-pause'
|
|
break
|
|
case 'searching':
|
|
result.name = 'i-lucide-search'
|
|
result.pulse = true
|
|
break
|
|
case 'search_complete':
|
|
result.name = 'i-lucide-search-check'
|
|
break
|
|
case 'processing_serach_result':
|
|
result.name = 'i-lucide-brain'
|
|
result.pulse = true
|
|
break
|
|
case 'processed_search_result':
|
|
result.name = 'i-lucide-circle-check-big'
|
|
break
|
|
case 'error':
|
|
result.name = 'i-lucide-octagon-x'
|
|
break
|
|
}
|
|
return result
|
|
})
|
|
</script>
|
|
|
|
<template>
|
|
<div class="flex items-center gap-1">
|
|
<UIcon name="i-lucide-circle-dot" />
|
|
<UButton :class="icon.pulse && 'animate-pulse'" :icon="icon.name" size="sm" color="info" @click="emit('select', node)">{{
|
|
node.label
|
|
}}</UButton>
|
|
<ol v-if="node.children.length > 0" class="space-y-2">
|
|
<li v-for="node in node.children" :key="node.id">
|
|
<Tree class="ml-2" :node="node" @select="emit('select', $event)" />
|
|
</li>
|
|
</ol>
|
|
</div>
|
|
</template>
|