Skip to content

Commit

Permalink
Feat: Add Tavily Api Key to chat configuration modal. #5198 (#5347)
Browse files Browse the repository at this point in the history
### 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)
  • Loading branch information
cike8899 authored Feb 26, 2025
1 parent 150ab9c commit e5e9ca0
Show file tree
Hide file tree
Showing 7 changed files with 47 additions and 9 deletions.
1 change: 1 addition & 0 deletions web/src/components/message-item/index.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -176,6 +176,7 @@ const MessageItem = ({
documentId={item.doc_id}
documentName={item.doc_name}
prefix="document"
link={item.url}
>
{item.doc_name}
</NewDocumentLink>
Expand Down
1 change: 1 addition & 0 deletions web/src/interfaces/database/chat.ts
Original file line number Diff line number Diff line change
Expand Up @@ -107,6 +107,7 @@ export interface Docagg {
count: number;
doc_id: string;
doc_name: string;
url?: string;
}

// interface Chunk {
Expand Down
4 changes: 4 additions & 0 deletions web/src/locales/en.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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',
Expand Down
10 changes: 8 additions & 2 deletions web/src/locales/zh-traditional.ts
Original file line number Diff line number Diff line change
Expand Up @@ -148,7 +148,8 @@ export default {
selectFiles: '選擇文件',
changeSpecificCategory: '更改特定類別',
uploadTitle: '點擊或拖拽文件至此區域即可上傳',
uploadDescription: '支持單次或批量上傳。單個檔案大小不超過10MB,最多上傳128份檔案。',
uploadDescription:
'支持單次或批量上傳。單個檔案大小不超過10MB,最多上傳128份檔案。',
chunk: '解析塊',
bulk: '批量',
cancel: '取消',
Expand Down Expand Up @@ -510,6 +511,10 @@ export default {
reasoning: '推理',
reasoningTip:
'它將觸發類似Deepseek-R1/OpenAI o1的推理過程。將代理搜尋過程整合到推理工作流程中,使得模型本身能夠在遇到不確定資訊時動態地檢索外部知識。',
tavilyApiKeyTip:
'如果 API 金鑰設定正確,它將利用 Tavily 進行網路搜尋作為知識庫的補充。',
tavilyApiKeyMessage: '請輸入你的 Tavily Api Key',
tavilyApiKeyHelp: '如何獲取?',
},
setting: {
profile: '概述',
Expand Down Expand Up @@ -703,7 +708,8 @@ export default {
newFolder: '新建文件夾',
uploadFile: '上傳文件',
uploadTitle: '點擊或拖拽文件至此區域即可上傳',
uploadDescription: '支持單次或批量上傳。單個檔案大小不超過10MB,最多上傳128份檔案。',
uploadDescription:
'支持單次或批量上傳。單個檔案大小不超過10MB,最多上傳128份檔案。',
file: '文件',
directory: '文件夾',
local: '本地上傳',
Expand Down
4 changes: 4 additions & 0 deletions web/src/locales/zh.ts
Original file line number Diff line number Diff line change
Expand Up @@ -527,6 +527,10 @@ General:实体和关系提取提示来自 GitHub - microsoft/graphrag:基于
reasoning: '推理',
reasoningTip:
'它将像Deepseek-R1 / OpenAI o1一样触发推理过程。将代理搜索过程集成到推理工作流中,允许模型本身在遇到不确定信息时动态地检索外部知识。',
tavilyApiKeyTip:
'如果 API 密钥设置正确,它将利用 Tavily 进行网络搜索作为知识库的补充。',
tavilyApiKeyMessage: '请输入你的 Tavily Api Key',
tavilyApiKeyHelp: '如何获取?',
},
setting: {
profile: '概要',
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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';
Expand Down Expand Up @@ -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}
Expand Down
24 changes: 18 additions & 6 deletions web/src/pages/chat/markdown-content/index.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -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],
);

Expand Down Expand Up @@ -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;
Expand Down Expand Up @@ -146,6 +157,7 @@ const MarkdownContent = ({
documentId,
chunkItem,
fileExtension === 'pdf',
documentUrl,
)}
>
{document?.doc_name}
Expand Down

0 comments on commit e5e9ca0

Please sign in to comment.