feat: external knowledge base
Yi
2 years ago
parent
ff0260e564
commit
1c7cb3fbc0
@ -1,20 +1,30 @@
|
|||||||
'use client'
|
'use client'
|
||||||
|
|
||||||
import React from 'react'
|
import React, { useState } from 'react'
|
||||||
|
import { useToastContext } from '@/app/components/base/toast'
|
||||||
import ExternalKnowledgeBaseCreate from '@/app/components/datasets/external-knowledge-base/create'
|
import ExternalKnowledgeBaseCreate from '@/app/components/datasets/external-knowledge-base/create'
|
||||||
import type { CreateKnowledgeBaseReq } from '@/app/components/datasets/external-knowledge-base/create/declarations'
|
import type { CreateKnowledgeBaseReq } from '@/app/components/datasets/external-knowledge-base/create/declarations'
|
||||||
import { createExternalKnowledgeBase } from '@/service/datasets'
|
import { createExternalKnowledgeBase } from '@/service/datasets'
|
||||||
|
|
||||||
const ExternalKnowledgeBaseConnector = () => {
|
const ExternalKnowledgeBaseConnector = () => {
|
||||||
|
const { notify } = useToastContext()
|
||||||
|
const [loading, setLoading] = useState(false)
|
||||||
|
|
||||||
const handleConnect = async (formValue: CreateKnowledgeBaseReq) => {
|
const handleConnect = async (formValue: CreateKnowledgeBaseReq) => {
|
||||||
try {
|
try {
|
||||||
|
setLoading(true)
|
||||||
const result = await createExternalKnowledgeBase({ body: formValue })
|
const result = await createExternalKnowledgeBase({ body: formValue })
|
||||||
|
if (result && result.id)
|
||||||
|
notify({ type: 'success', message: 'External Knowledge Base Connected Successfully' })
|
||||||
|
else
|
||||||
|
throw new Error('Failed to create external knowledge base')
|
||||||
}
|
}
|
||||||
catch (error) {
|
catch (error) {
|
||||||
console.error('Error creating external knowledge base:', error)
|
console.error('Error creating external knowledge base:', error)
|
||||||
}
|
}
|
||||||
|
setLoading(false)
|
||||||
}
|
}
|
||||||
return <ExternalKnowledgeBaseCreate onConnect={handleConnect} />
|
return <ExternalKnowledgeBaseCreate onConnect={handleConnect} loading={loading} />
|
||||||
}
|
}
|
||||||
|
|
||||||
export default ExternalKnowledgeBaseConnector
|
export default ExternalKnowledgeBaseConnector
|
||||||
|
|||||||
@ -0,0 +1,65 @@
|
|||||||
|
import { useState } from 'react'
|
||||||
|
import {
|
||||||
|
RiCloseLine,
|
||||||
|
} from '@remixicon/react'
|
||||||
|
import { useTranslation } from 'react-i18next'
|
||||||
|
import RetrievalSettings from '../external-knowledge-base/create/RetrievalSettings'
|
||||||
|
import Button from '@/app/components/base/button'
|
||||||
|
import ActionButton from '@/app/components/base/action-button'
|
||||||
|
|
||||||
|
type ModifyExternalRetrievalModalProps = {
|
||||||
|
onClose: () => void
|
||||||
|
onSave: (data: { top_k: number; score_threshold: number }) => void
|
||||||
|
initialTopK: number
|
||||||
|
initialScoreThreshold: number
|
||||||
|
}
|
||||||
|
|
||||||
|
const ModifyExternalRetrievalModal: React.FC<ModifyExternalRetrievalModalProps> = ({
|
||||||
|
onClose,
|
||||||
|
onSave,
|
||||||
|
initialTopK,
|
||||||
|
initialScoreThreshold,
|
||||||
|
}) => {
|
||||||
|
const { t } = useTranslation()
|
||||||
|
const [topK, setTopK] = useState(initialTopK)
|
||||||
|
const [scoreThreshold, setScoreThreshold] = useState(initialScoreThreshold)
|
||||||
|
|
||||||
|
const handleSettingsChange = (data: { top_k?: number; score_threshold?: number }) => {
|
||||||
|
if (data.top_k !== undefined)
|
||||||
|
setTopK(data.top_k)
|
||||||
|
if (data.score_threshold !== undefined)
|
||||||
|
setScoreThreshold(data.score_threshold)
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleSave = () => {
|
||||||
|
onSave({ top_k: topK, score_threshold: scoreThreshold })
|
||||||
|
onClose()
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className='absolute z-10 top-[36px] right-[14px] flex w-[320px] flex-col items-start rounded-2xl border-[0.5px]
|
||||||
|
border-components-panel-border bg-components-panel-bg shadows-shadow-2xl'
|
||||||
|
>
|
||||||
|
<div className='flex p-4 pb-2 items-center justify-between self-stretch'>
|
||||||
|
<div className='text-text-primary system-xl-semibold flex-grow'>{t('datasetHitTesting.settingTitle')}</div>
|
||||||
|
<ActionButton className='ml-auto' onClick={onClose}>
|
||||||
|
<RiCloseLine className='w-4 h-4 flex-shrink-0' />
|
||||||
|
</ActionButton>
|
||||||
|
</div>
|
||||||
|
<div className='flex p-4 pt-2 flex-col justify-center items-start gap-4 self-stretch'>
|
||||||
|
<RetrievalSettings
|
||||||
|
topK={topK}
|
||||||
|
scoreThreshold={scoreThreshold}
|
||||||
|
onChange={handleSettingsChange}
|
||||||
|
isInHitTesting={true}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
<div className='flex p-4 pt-2 justify-end items-end gap-1 w-full'>
|
||||||
|
<Button className='flex-shrink-0 min-w-[72px]' onClick={onClose}>{t('common.operation.cancel')}</Button>
|
||||||
|
<Button variant='primary' className='flex-shrink-0 min-w-[72px]' onClick={handleSave}>{t('common.operation.save')}</Button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
export default ModifyExternalRetrievalModal
|
||||||
Loading…
Reference in New Issue