Commit 0759b29c authored by Joel's avatar Joel

feat: retrial node

parent db6074e0
import type { KnowledgeRetrievalNodeType } from './types'
import { RETRIEVE_TYPE } from '@/types/app'
export const mockData: KnowledgeRetrievalNodeType = {
type: 'KnowledgeRetrieval',
desc: 'xxx',
title: 'KnowledgeRetrieval',
query_variable_selector: ['aaa', 'name'],
dataset_ids: ['1'],
retrieval_mode: RETRIEVE_TYPE.oneWay,
multiple_retrieval_config: {
top_k: 10,
score_threshold: 0.5,
reranking_model: {
provider: '',
model: '',
},
},
}
import type { FC } from 'react'
import { Folder } from '@/app/components/base/icons/src/vender/solid/files'
const Node: FC = () => {
return (
<div>knowledge-retrieval</div>
<div className='px-3'>
<div className='space-y-0.5'>
{['product Doc', 'Text completion'].map(name => (
<div key={name} className='flex items-center h-[26px] bg-gray-100 rounded-md px-1 text-xs font-normal text-gray-700'>
<div className='mr-1 shrink-0 p-1 bg-[#F5F8FF] rounded-md border-[0.5px] border-[#E0EAFF]'>
<Folder className='w-3 h-3 text-[#444CE7]' />
</div>
<div className='text-xs font-normal text-gray-700'>
{name}
</div>
</div>
))}
</div>
</div>
)
}
......
import type { CommonNodeType, ValueSelector } from '@/app/components/workflow/types'
import type { RETRIEVE_TYPE } from '@/types/app'
export type KnowledgeRetrievalNodeType = CommonNodeType & {
query_variable_selector: ValueSelector
dataset_ids: string[]
retrieval_mode: RETRIEVE_TYPE
multiple_retrieval_config?: {
top_k: number
score_threshold: number
reranking_model: {
provider: string
model: string
}
}
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment