Feat: Add Tavily Api Key to chat configuration modal. #5198 (#5347)

### What problem does this PR solve?

Feat: Add Tavily Api Key to chat configuration modal. #5198

### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu 2025-02-26 10:20:40 +08:00 committed by GitHub
parent 150ab9c6a4
commit e5e9ca0015
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
7 changed files with 47 additions and 9 deletions

View File

@ -176,6 +176,7 @@ const MessageItem = ({
documentId={item.doc_id}
documentName={item.doc_name}
prefix="document"
link={item.url}
>
{item.doc_name}
</NewDocumentLink>

View File

@ -107,6 +107,7 @@ export interface Docagg {
count: number;
doc_id: string;
doc_name: string;
url?: string;
}
// interface Chunk {

View File

@ -531,6 +531,10 @@ This procedure will improve precision of retrieval by adding more information to
reasoning: 'Reasoning',
reasoningTip:
'It will trigger reasoning process like Deepseek-R1/OpenAI o1. Integrates an agentic search process into the reasoning workflow, allowing models itself to dynamically retrieve external knowledge whenever they encounter uncertain information.',
tavilyApiKeyTip:
'If API key is set correctly, it will utilize Tavily to do web search as a supplement to knowledge bases.',
tavilyApiKeyMessage: 'Please enter your Tavily Api Key',
tavilyApiKeyHelp: 'How to get it?',
},
setting: {
profile: 'Profile',

View File

@ -148,7 +148,8 @@ export default {
selectFiles: '選擇文件',
changeSpecificCategory: '更改特定類別',
uploadTitle: '點擊或拖拽文件至此區域即可上傳',
uploadDescription: '支持單次或批量上傳。單個檔案大小不超過10MB最多上傳128份檔案。',
uploadDescription:
'支持單次或批量上傳。單個檔案大小不超過10MB最多上傳128份檔案。',
chunk: '解析塊',
bulk: '批量',
cancel: '取消',
@ -510,6 +511,10 @@ export default {
reasoning: '推理',
reasoningTip:
'它將觸發類似Deepseek-R1/OpenAI o1的推理過程。將代理搜尋過程整合到推理工作流程中使得模型本身能夠在遇到不確定資訊時動態地檢索外部知識。',
tavilyApiKeyTip:
'如果 API 金鑰設定正確,它將利用 Tavily 進行網路搜尋作為知識庫的補充。',
tavilyApiKeyMessage: '請輸入你的 Tavily Api Key',
tavilyApiKeyHelp: '如何獲取?',
},
setting: {
profile: '概述',
@ -703,7 +708,8 @@ export default {
newFolder: '新建文件夾',
uploadFile: '上傳文件',
uploadTitle: '點擊或拖拽文件至此區域即可上傳',
uploadDescription: '支持單次或批量上傳。單個檔案大小不超過10MB最多上傳128份檔案。',
uploadDescription:
'支持單次或批量上傳。單個檔案大小不超過10MB最多上傳128份檔案。',
file: '文件',
directory: '文件夾',
local: '本地上傳',

View File

@ -527,6 +527,10 @@ General实体和关系提取提示来自 GitHub - microsoft/graphrag基于
reasoning: '推理',
reasoningTip:
'它将像Deepseek-R1 / OpenAI o1一样触发推理过程。将代理搜索过程集成到推理工作流中允许模型本身在遇到不确定信息时动态地检索外部知识。',
tavilyApiKeyTip:
'如果 API 密钥设置正确,它将利用 Tavily 进行网络搜索作为知识库的补充。',
tavilyApiKeyMessage: '请输入你的 Tavily Api Key',
tavilyApiKeyHelp: '如何获取?',
},
setting: {
profile: '概要',

View File

@ -2,7 +2,7 @@ import KnowledgeBaseItem from '@/components/knowledge-base-item';
import { useTranslate } from '@/hooks/common-hooks';
import { useFetchTenantInfo } from '@/hooks/user-setting-hooks';
import { PlusOutlined } from '@ant-design/icons';
import { Form, Input, message, Select, Switch, Upload } from 'antd';
import { Form, Input, message, Select, Switch, Typography, Upload } from 'antd';
import classNames from 'classnames';
import { useCallback } from 'react';
import { ISegmentedContentProps } from '../interface';
@ -147,6 +147,16 @@ const AssistantSetting = ({
>
<Switch onChange={handleTtsChange} />
</Form.Item>
<Form.Item label={'Tavily Api Key'} tooltip={t('tavilyApiKeyTip')}>
<div className="flex flex-col gap-1">
<Form.Item name={['prompt_config', 'tavily_api_key']} noStyle>
<Input.Password placeholder={t('tavilyApiKeyMessage')} />
</Form.Item>
<Typography.Link href="https://app.tavily.com/home" target={'_blank'}>
{t('tavilyApiKeyHelp')}
</Typography.Link>
</div>
</Form.Item>
<KnowledgeBaseItem
required={false}
onChange={handleChange}

View File

@ -59,12 +59,22 @@ const MarkdownContent = ({
}, [reference, setDocumentIds]);
const handleDocumentButtonClick = useCallback(
(documentId: string, chunk: IReferenceChunk, isPdf: boolean) => () => {
if (!isPdf) {
return;
}
clickDocumentButton?.(documentId, chunk);
},
(
documentId: string,
chunk: IReferenceChunk,
isPdf: boolean,
documentUrl?: string,
) =>
() => {
if (!isPdf) {
if (!documentUrl) {
return;
}
window.open(documentUrl, '_blank');
} else {
clickDocumentButton?.(documentId, chunk);
}
},
[clickDocumentButton],
);
@ -93,6 +103,7 @@ const MarkdownContent = ({
(x) => x?.doc_id === chunkItem?.document_id,
);
const documentId = document?.doc_id;
const documentUrl = document?.url;
const fileThumbnail = documentId ? fileThumbnails[documentId] : '';
const fileExtension = documentId ? getExtension(document?.doc_name) : '';
const imageId = chunkItem?.image_id;
@ -146,6 +157,7 @@ const MarkdownContent = ({
documentId,
chunkItem,
fileExtension === 'pdf',
documentUrl,
)}
>
{document?.doc_name}