diff --git a/src/renderer/src/components/Icons/SVGIcon.tsx b/src/renderer/src/components/Icons/SVGIcon.tsx index b9a3eff899..88598bb02e 100644 --- a/src/renderer/src/components/Icons/SVGIcon.tsx +++ b/src/renderer/src/components/Icons/SVGIcon.tsx @@ -112,3 +112,129 @@ export function MdiLightbulbOn(props: SVGProps) { ) } + +export function BingLogo(props: SVGProps) { + return ( + + + + ) +} + +export function SearXNGLogo(props: SVGProps) { + return ( + + + + + + + + ) +} + +export function TavilyLogo(props: SVGProps) { + return ( + + + + + + + + + ) +} + +export function ExaLogo(props: SVGProps) { + return ( + + Exa + + + ) +} + +export function BochaLogo(props: SVGProps) { + return ( + + + + + + + ) +} diff --git a/src/renderer/src/components/Popups/ApiKeyListPopup/hook.ts b/src/renderer/src/components/Popups/ApiKeyListPopup/hook.ts index 541dd2f156..c4c9459ed0 100644 --- a/src/renderer/src/components/Popups/ApiKeyListPopup/hook.ts +++ b/src/renderer/src/components/Popups/ApiKeyListPopup/hook.ts @@ -3,7 +3,14 @@ import { isEmbeddingModel, isRerankModel } from '@renderer/config/models' import SelectProviderModelPopup from '@renderer/pages/settings/ProviderSettings/SelectProviderModelPopup' import { checkApi } from '@renderer/services/ApiService' import WebSearchService from '@renderer/services/WebSearchService' -import { Model, PreprocessProvider, Provider, WebSearchProvider } from '@renderer/types' +import { + isPreprocessProviderId, + isWebSearchProviderId, + Model, + PreprocessProvider, + Provider, + WebSearchProvider +} from '@renderer/types' import { ApiKeyConnectivity, ApiKeyWithStatus, HealthStatus } from '@renderer/types/healthCheck' import { formatApiKeys, splitApiKeyString } from '@renderer/utils/api' import { formatErrorMessage } from '@renderer/utils/error' @@ -12,12 +19,11 @@ import { isEmpty } from 'lodash' import { useCallback, useMemo, useState } from 'react' import { useTranslation } from 'react-i18next' -import { ApiKeyValidity, ApiProviderKind, ApiProviderUnion } from './types' +import { ApiKeyValidity, ApiProvider, UpdateApiProviderFunc } from './types' interface UseApiKeysProps { - provider: ApiProviderUnion - updateProvider: (provider: Partial) => void - providerKind: ApiProviderKind + provider: ApiProvider + updateProvider: UpdateApiProviderFunc } const logger = loggerService.withContext('ApiKeyListPopup') @@ -25,7 +31,7 @@ const logger = loggerService.withContext('ApiKeyListPopup') /** * API Keys 管理 hook */ -export function useApiKeys({ provider, updateProvider, providerKind }: UseApiKeysProps) { +export function useApiKeys({ provider, updateProvider }: UseApiKeysProps) { const { t } = useTranslation() // 连通性检查的 UI 状态管理 @@ -199,11 +205,13 @@ export function useApiKeys({ provider, updateProvider, providerKind }: UseApiKey try { const startTime = Date.now() - if (isLlmProvider(provider, providerKind) && model) { + if (isLlmProvider(provider) && model) { await checkApi({ ...provider, apiKey: keyToCheck }, model) - } else { + } else if (isWebSearchProvider(provider)) { const result = await WebSearchService.checkSearch({ ...provider, apiKey: keyToCheck }) if (!result.valid) throw new Error(result.error) + } else { + // 不处理预处理供应商 } const latency = Date.now() - startTime @@ -228,7 +236,7 @@ export function useApiKeys({ provider, updateProvider, providerKind }: UseApiKey logger.error('failed to validate the connectivity of the api key', error) } }, - [keys, connectivityStates, updateConnectivityState, provider, providerKind] + [keys, connectivityStates, updateConnectivityState, provider] ) // 检查单个 key 的连通性 @@ -240,23 +248,23 @@ export function useApiKeys({ provider, updateProvider, providerKind }: UseApiKey const currentState = connectivityStates.get(keyToCheck) if (currentState?.checking) return - const model = isLlmProvider(provider, providerKind) ? await getModelForCheck(provider, t) : undefined + const model = isLlmProvider(provider) ? await getModelForCheck(provider, t) : undefined if (model === null) return await runConnectivityCheck(index, model) }, - [provider, keys, connectivityStates, providerKind, t, runConnectivityCheck] + [provider, keys, connectivityStates, t, runConnectivityCheck] ) // 检查所有 keys 的连通性 const checkAllKeysConnectivity = useCallback(async () => { if (!provider || keys.length === 0) return - const model = isLlmProvider(provider, providerKind) ? await getModelForCheck(provider, t) : undefined + const model = isLlmProvider(provider) ? await getModelForCheck(provider, t) : undefined if (model === null) return await Promise.allSettled(keys.map((_, index) => runConnectivityCheck(index, model))) - }, [provider, keys, providerKind, t, runConnectivityCheck]) + }, [provider, keys, t, runConnectivityCheck]) // 计算是否有 key 正在检查 const isChecking = useMemo(() => { @@ -275,16 +283,18 @@ export function useApiKeys({ provider, updateProvider, providerKind }: UseApiKey } } -export function isLlmProvider(obj: any, kind: ApiProviderKind): obj is Provider { - return kind === 'llm' && 'type' in obj && 'models' in obj +export function isLlmProvider(provider: ApiProvider): provider is Provider { + return 'models' in provider } -export function isWebSearchProvider(obj: any, kind: ApiProviderKind): obj is WebSearchProvider { - return kind === 'websearch' && ('url' in obj || 'engines' in obj) +export function isWebSearchProvider(provider: ApiProvider): provider is WebSearchProvider { + return isWebSearchProviderId(provider.id) } -export function isPreprocessProvider(obj: any, kind: ApiProviderKind): obj is PreprocessProvider { - return kind === 'doc-preprocess' && ('quota' in obj || 'options' in obj) +export function isPreprocessProvider(provider: ApiProvider): provider is PreprocessProvider { + // NOTE: mistral 同时提供预处理和llm服务，所以其llm provier可能被误判为预处理provider + // 后面需要使用更严格的判断方式 + return isPreprocessProviderId(provider.id) && !isLlmProvider(provider) } // 获取模型用于检查 diff --git a/src/renderer/src/components/Popups/ApiKeyListPopup/list.tsx b/src/renderer/src/components/Popups/ApiKeyListPopup/list.tsx index 03e9796886..86076b4ca8 100644 --- a/src/renderer/src/components/Popups/ApiKeyListPopup/list.tsx +++ b/src/renderer/src/components/Popups/ApiKeyListPopup/list.tsx @@ -6,6 +6,7 @@ import { useProvider } from '@renderer/hooks/useProvider' import { useWebSearchProvider } from '@renderer/hooks/useWebSearchProviders' import { SettingHelpText } from '@renderer/pages/settings' import { isProviderSupportAuth } from '@renderer/services/ProviderService' +import { PreprocessProviderId, WebSearchProviderId } from '@renderer/types' import { ApiKeyWithStatus, HealthStatus } from '@renderer/types/healthCheck' import { Button, Card, Flex, List, Popconfirm, Space, Tooltip, Typography } from 'antd' import { Plus } from 'lucide-react' @@ -15,19 +16,18 @@ import styled from 'styled-components' import { isLlmProvider, useApiKeys } from './hook' import ApiKeyItem from './item' -import { ApiProviderKind, ApiProviderUnion } from './types' +import { ApiProvider, UpdateApiProviderFunc } from './types' interface ApiKeyListProps { - provider: ApiProviderUnion - updateProvider: (provider: Partial) => void - providerKind: ApiProviderKind + provider: ApiProvider + updateProvider: UpdateApiProviderFunc showHealthCheck?: boolean } /** * Api key 列表，管理 CRUD 操作、连接检查 */ -export const ApiKeyList: FC = ({ provider, updateProvider, providerKind, showHealthCheck = true }) => { +export const ApiKeyList: FC = ({ provider, updateProvider, showHealthCheck = true }) => { const { t } = useTranslation() // 临时新项状态 @@ -42,7 +42,7 @@ export const ApiKeyList: FC = ({ provider, updateProvider, prov checkKeyConnectivity, checkAllKeysConnectivity, isChecking - } = useApiKeys({ provider, updateProvider, providerKind: providerKind }) + } = useApiKeys({ provider, updateProvider }) // 创建一个临时新项 const handleAddNew = () => { @@ -73,7 +73,7 @@ export const ApiKeyList: FC = ({ provider, updateProvider, prov const shouldAutoFocus = () => { if (provider.apiKey) return false - return isLlmProvider(provider, providerKind) && provider.enabled && !isProviderSupportAuth(provider) + return isLlmProvider(provider) && provider.enabled && !isProviderSupportAuth(provider) } // 合并真实 keys 和临时新项 @@ -179,55 +179,33 @@ export const ApiKeyList: FC = ({ provider, updateProvider, prov interface SpecificApiKeyListProps { providerId: string - providerKind: ApiProviderKind showHealthCheck?: boolean } -export const LlmApiKeyList: FC = ({ providerId, providerKind, showHealthCheck = true }) => { +type WebSearchApiKeyList = SpecificApiKeyListProps & { + providerId: WebSearchProviderId +} + +type DocPreprocessApiKeyListProps = SpecificApiKeyListProps & { + providerId: PreprocessProviderId +} + +export const LlmApiKeyList: FC = ({ providerId, showHealthCheck = true }) => { const { provider, updateProvider } = useProvider(providerId) - return ( - - ) + return } -export const WebSearchApiKeyList: FC = ({ - providerId, - providerKind, - showHealthCheck = true -}) => { +export const WebSearchApiKeyList: FC = ({ providerId, showHealthCheck = true }) => { const { provider, updateProvider } = useWebSearchProvider(providerId) - return ( - - ) + return } -export const DocPreprocessApiKeyList: FC = ({ - providerId, - providerKind, - showHealthCheck = true -}) => { +export const DocPreprocessApiKeyList: FC = ({ providerId, showHealthCheck = true }) => { const { provider, updateProvider } = usePreprocessProvider(providerId) - return ( - - ) + return } const ListContainer = styled.div` diff --git a/src/renderer/src/components/Popups/ApiKeyListPopup/popup.tsx b/src/renderer/src/components/Popups/ApiKeyListPopup/popup.tsx index 096e00ca58..b4ca91186b 100644 --- a/src/renderer/src/components/Popups/ApiKeyListPopup/popup.tsx +++ b/src/renderer/src/components/Popups/ApiKeyListPopup/popup.tsx @@ -1,14 +1,13 @@ import { TopView } from '@renderer/components/TopView' +import { isPreprocessProviderId, isWebSearchProviderId } from '@renderer/types' import { Modal } from 'antd' import { useMemo, useState } from 'react' import { useTranslation } from 'react-i18next' import { DocPreprocessApiKeyList, LlmApiKeyList, WebSearchApiKeyList } from './list' -import { ApiProviderKind } from './types' interface ShowParams { providerId: string - providerKind: ApiProviderKind title?: string showHealthCheck?: boolean } @@ -20,7 +19,7 @@ interface Props extends ShowParams { /** * API Key 列表弹窗容器组件 */ -const PopupContainer: React.FC = ({ providerId, providerKind, title, resolve, showHealthCheck = true }) => { +const PopupContainer: React.FC = ({ providerId, title, resolve, showHealthCheck = true }) => { const [open, setOpen] = useState(true) const { t } = useTranslation() @@ -33,17 +32,14 @@ const PopupContainer: React.FC = ({ providerId, providerKind, title, reso } const ListComponent = useMemo(() => { - switch (providerKind) { - case 'llm': - return LlmApiKeyList - case 'websearch': - return WebSearchApiKeyList - case 'doc-preprocess': - return DocPreprocessApiKeyList - default: - return null + if (isWebSearchProviderId(providerId)) { + return } - }, [providerKind]) + if (isPreprocessProviderId(providerId)) { + return + } + return + }, [providerId, showHealthCheck]) return ( = ({ providerId, providerKind, title, reso centered width={600} footer={null}> - {ListComponent && ( - - )} + {ListComponent} ) } diff --git a/src/renderer/src/components/Popups/ApiKeyListPopup/types.ts b/src/renderer/src/components/Popups/ApiKeyListPopup/types.ts index 4663e70715..bc230c577d 100644 --- a/src/renderer/src/components/Popups/ApiKeyListPopup/types.ts +++ b/src/renderer/src/components/Popups/ApiKeyListPopup/types.ts @@ -8,6 +8,12 @@ export type ApiKeyValidity = { error?: string } -export type ApiProviderUnion = Provider | WebSearchProvider | PreprocessProvider +export type ApiProvider = Provider | WebSearchProvider | PreprocessProvider -export type ApiProviderKind = 'llm' | 'websearch' | 'doc-preprocess' +export type UpdateProviderFunc = (p: Partial) => void + +export type UpdateWebSearchProviderFunc = (p: Partial) => void + +export type UpdatePreprocessProviderFunc = (p: Partial) => void + +export type UpdateApiProviderFunc = UpdateProviderFunc | UpdateWebSearchProviderFunc | UpdatePreprocessProviderFunc diff --git a/src/renderer/src/config/preprocessProviders.ts b/src/renderer/src/config/preprocessProviders.ts index 587e6ea7f9..88215b328d 100644 --- a/src/renderer/src/config/preprocessProviders.ts +++ b/src/renderer/src/config/preprocessProviders.ts @@ -1,8 +1,9 @@ import Doc2xLogo from '@renderer/assets/images/ocr/doc2x.png' import MinerULogo from '@renderer/assets/images/ocr/mineru.jpg' import MistralLogo from '@renderer/assets/images/providers/mistral.png' +import { PreprocessProviderId } from '@renderer/types' -export function getPreprocessProviderLogo(providerId: string) { +export function getPreprocessProviderLogo(providerId: PreprocessProviderId) { switch (providerId) { case 'doc2x': return Doc2xLogo @@ -15,7 +16,9 @@ export function getPreprocessProviderLogo(providerId: string) { } } -export const PREPROCESS_PROVIDER_CONFIG = { +type PreprocessProviderConfig = { websites: { official: string; apiKey: string } } + +export const PREPROCESS_PROVIDER_CONFIG: Record = { doc2x: { websites: { official: 'https://doc2x.noedgeai.com', diff --git a/src/renderer/src/config/webSearchProviders.ts b/src/renderer/src/config/webSearchProviders.ts index d33e9cba35..62c0536f4d 100644 --- a/src/renderer/src/config/webSearchProviders.ts +++ b/src/renderer/src/config/webSearchProviders.ts @@ -1,24 +1,13 @@ -import BochaLogo from '@renderer/assets/images/search/bocha.webp' -import ExaLogo from '@renderer/assets/images/search/exa.png' -import SearxngLogo from '@renderer/assets/images/search/searxng.svg' -import TavilyLogo from '@renderer/assets/images/search/tavily.png' +import { WebSearchProvider, WebSearchProviderId } from '@renderer/types' -export function getWebSearchProviderLogo(providerId: string) { - switch (providerId) { - case 'tavily': - return TavilyLogo - case 'searxng': - return SearxngLogo - case 'exa': - return ExaLogo - case 'bocha': - return BochaLogo - default: - return undefined +type WebSearchProviderConfig = { + websites: { + official: string + apiKey?: string } } -export const WEB_SEARCH_PROVIDER_CONFIG = { +export const WEB_SEARCH_PROVIDER_CONFIG: Record = { tavily: { websites: { official: 'https://tavily.com', @@ -58,3 +47,46 @@ export const WEB_SEARCH_PROVIDER_CONFIG = { } } } + +export const WEB_SEARCH_PROVIDERS: WebSearchProvider[] = [ + { + id: 'tavily', + name: 'Tavily', + apiHost: 'https://api.tavily.com', + apiKey: '' + }, + { + id: 'searxng', + name: 'Searxng', + apiHost: '', + basicAuthUsername: '', + basicAuthPassword: '' + }, + { + id: 'exa', + name: 'Exa', + apiHost: 'https://api.exa.ai', + apiKey: '' + }, + { + id: 'bocha', + name: 'Bocha', + apiHost: 'https://api.bochaai.com', + apiKey: '' + }, + { + id: 'local-google', + name: 'Google', + url: 'https://www.google.com/search?q=%s' + }, + { + id: 'local-bing', + name: 'Bing', + url: 'https://cn.bing.com/search?q=%s&ensearch=1' + }, + { + id: 'local-baidu', + name: 'Baidu', + url: 'https://www.baidu.com/s?wd=%s' + } +] as const diff --git a/src/renderer/src/hooks/usePreprocess.ts b/src/renderer/src/hooks/usePreprocess.ts index 41463227ad..5172e2c68a 100644 --- a/src/renderer/src/hooks/usePreprocess.ts +++ b/src/renderer/src/hooks/usePreprocess.ts @@ -4,10 +4,10 @@ import { updatePreprocessProvider as _updatePreprocessProvider, updatePreprocessProviders as _updatePreprocessProviders } from '@renderer/store/preprocess' -import { PreprocessProvider } from '@renderer/types' +import { PreprocessProvider, PreprocessProviderId } from '@renderer/types' import { useDispatch, useSelector } from 'react-redux' -export const usePreprocessProvider = (id: string) => { +export const usePreprocessProvider = (id: PreprocessProviderId) => { const dispatch = useDispatch() const preprocessProviders = useSelector((state: RootState) => state.preprocess.providers) const provider = preprocessProviders.find((provider) => provider.id === id) diff --git a/src/renderer/src/hooks/useWebSearchProviders.ts b/src/renderer/src/hooks/useWebSearchProviders.ts index 32f9238abf..34ee07403e 100644 --- a/src/renderer/src/hooks/useWebSearchProviders.ts +++ b/src/renderer/src/hooks/useWebSearchProviders.ts @@ -11,7 +11,7 @@ import { updateWebSearchProvider, updateWebSearchProviders } from '@renderer/store/websearch' -import { WebSearchProvider } from '@renderer/types' +import { WebSearchProvider, WebSearchProviderId } from '@renderer/types' export const useDefaultWebSearchProvider = () => { const defaultProvider = useAppSelector((state) => state.websearch.defaultProvider) @@ -49,7 +49,7 @@ export const useWebSearchProviders = () => { } } -export const useWebSearchProvider = (id: string) => { +export const useWebSearchProvider = (id: WebSearchProviderId) => { const providers = useAppSelector((state) => state.websearch.providers) const provider = providers.find((provider) => provider.id === id) const dispatch = useAppDispatch() @@ -60,7 +60,9 @@ export const useWebSearchProvider = (id: string) => { return { provider, - updateProvider: (updates: Partial) => dispatch(updateWebSearchProvider({ id, ...updates })) + updateProvider: (updates: Partial) => { + dispatch(updateWebSearchProvider({ id, ...updates })) + } } } diff --git a/src/renderer/src/pages/home/Inputbar/WebSearchButton.tsx b/src/renderer/src/pages/home/Inputbar/WebSearchButton.tsx index 7a6a6e6334..70b25084d5 100644 --- a/src/renderer/src/pages/home/Inputbar/WebSearchButton.tsx +++ b/src/renderer/src/pages/home/Inputbar/WebSearchButton.tsx @@ -1,9 +1,11 @@ +import { BaiduOutlined, GoogleOutlined } from '@ant-design/icons' +import { BingLogo, BochaLogo, ExaLogo, SearXNGLogo, TavilyLogo } from '@renderer/components/Icons' import { QuickPanelListItem, useQuickPanel } from '@renderer/components/QuickPanel' import { isWebSearchModel } from '@renderer/config/models' import { useAssistant } from '@renderer/hooks/useAssistant' import { useWebSearchProviders } from '@renderer/hooks/useWebSearchProviders' import WebSearchService from '@renderer/services/WebSearchService' -import { Assistant, WebSearchProvider } from '@renderer/types' +import { Assistant, WebSearchProvider, WebSearchProviderId } from '@renderer/types' import { hasObjectKey } from '@renderer/utils' import { Tooltip } from 'antd' import { Globe } from 'lucide-react' @@ -28,6 +30,33 @@ const WebSearchButton: FC = ({ ref, assistant, ToolbarButton }) => { const enableWebSearch = assistant?.webSearchProviderId || assistant.enableWebSearch + const WebSearchIcon = useCallback( + ({ pid, size = 18 }: { pid?: WebSearchProviderId; size?: number }) => { + const iconColor = enableWebSearch ? 'var(--color-primary)' : 'var(--color-icon)' + + switch (pid) { + case 'bocha': + return + case 'exa': + // size微调，视觉上和其他图标平衡一些 + return + case 'tavily': + return + case 'searxng': + return + case 'local-baidu': + return + case 'local-bing': + return + case 'local-google': + return + default: + return + } + }, + [enableWebSearch] + ) + const updateSelectedWebSearchProvider = useCallback( async (providerId?: WebSearchProvider['id']) => { // TODO: updateAssistant有性能问题，会导致关闭快捷面板卡顿 @@ -58,7 +87,7 @@ const WebSearchButton: FC = ({ ref, assistant, ToolbarButton }) => { ? t('settings.tool.websearch.apikey') : t('settings.tool.websearch.free') : t('chat.input.web_search.enable_content'), - icon: , + icon: , isSelected: p.id === assistant?.webSearchProviderId, disabled: !WebSearchService.isWebSearchEnabled(p.id), action: () => updateSelectedWebSearchProvider(p.id) @@ -80,6 +109,7 @@ const WebSearchButton: FC = ({ ref, assistant, ToolbarButton }) => { return items }, [ + WebSearchIcon, assistant.enableWebSearch, assistant.model, assistant?.webSearchProviderId, @@ -135,12 +165,7 @@ const WebSearchButton: FC = ({ ref, assistant, ToolbarButton }) => { mouseLeaveDelay={0} arrow> - + ) diff --git a/src/renderer/src/pages/knowledge/components/QuotaTag.tsx b/src/renderer/src/pages/knowledge/components/QuotaTag.tsx index 8eb8868e98..b6f29ce5d1 100644 --- a/src/renderer/src/pages/knowledge/components/QuotaTag.tsx +++ b/src/renderer/src/pages/knowledge/components/QuotaTag.tsx @@ -2,14 +2,14 @@ import { loggerService } from '@logger' import { usePreprocessProvider } from '@renderer/hooks/usePreprocess' import { getStoreSetting } from '@renderer/hooks/useSettings' import { getKnowledgeBaseParams } from '@renderer/services/KnowledgeService' -import { KnowledgeBase } from '@renderer/types' +import { KnowledgeBase, PreprocessProviderId } from '@renderer/types' import { Tag } from 'antd' import { FC, useEffect, useState } from 'react' import { useTranslation } from 'react-i18next' const logger = loggerService.withContext('QuotaTag') -const QuotaTag: FC<{ base: KnowledgeBase; providerId: string; quota?: number }> = ({ +const QuotaTag: FC<{ base: KnowledgeBase; providerId: PreprocessProviderId; quota?: number }> = ({ base, providerId, quota: _quota diff --git a/src/renderer/src/pages/settings/PreprocessSettings/PreprocessSettings.tsx b/src/renderer/src/pages/settings/PreprocessSettings/PreprocessSettings.tsx index c1592ac73f..d19eec4d6d 100644 --- a/src/renderer/src/pages/settings/PreprocessSettings/PreprocessSettings.tsx +++ b/src/renderer/src/pages/settings/PreprocessSettings/PreprocessSettings.tsx @@ -4,23 +4,14 @@ import { getPreprocessProviderLogo, PREPROCESS_PROVIDER_CONFIG } from '@renderer import { usePreprocessProvider } from '@renderer/hooks/usePreprocess' import { PreprocessProvider } from '@renderer/types' import { formatApiKeys, hasObjectKey } from '@renderer/utils' -import { Avatar, Button, Divider, Flex, Input, InputNumber, Segmented, Tooltip } from 'antd' +import { Avatar, Button, Divider, Flex, Input, Tooltip } from 'antd' import Link from 'antd/es/typography/Link' import { List } from 'lucide-react' import { FC, useEffect, useState } from 'react' import { useTranslation } from 'react-i18next' import styled from 'styled-components' -import { - SettingDivider, - SettingHelpLink, - SettingHelpText, - SettingHelpTextRow, - SettingRow, - SettingRowTitle, - SettingSubtitle, - SettingTitle -} from '..' +import { SettingHelpLink, SettingHelpText, SettingHelpTextRow, SettingSubtitle, SettingTitle } from '..' interface Props { provider: PreprocessProvider @@ -31,7 +22,7 @@ const PreprocessProviderSettings: FC = ({ provider: _provider }) => { const { t } = useTranslation() const [apiKey, setApiKey] = useState(preprocessProvider.apiKey || '') const [apiHost, setApiHost] = useState(preprocessProvider.apiHost || '') - const [options, setOptions] = useState(preprocessProvider.options || {}) + // const [options, setOptions] = useState(preprocessProvider.options || {}) const preprocessProviderConfig = PREPROCESS_PROVIDER_CONFIG[preprocessProvider.id] const apiKeyWebsite = preprocessProviderConfig?.websites?.apiKey @@ -40,7 +31,7 @@ const PreprocessProviderSettings: FC = ({ provider: _provider }) => { useEffect(() => { setApiKey(preprocessProvider.apiKey ?? '') setApiHost(preprocessProvider.apiHost ?? '') - setOptions(preprocessProvider.options ?? {}) + // setOptions(preprocessProvider.options ?? {}) }, [preprocessProvider.apiKey, preprocessProvider.apiHost, preprocessProvider.options]) const onUpdateApiKey = () => { @@ -52,7 +43,6 @@ const PreprocessProviderSettings: FC = ({ provider: _provider }) => { const openApiKeyList = async () => { await ApiKeyListPopup.show({ providerId: preprocessProvider.id, - providerKind: 'doc-preprocess', title: `${preprocessProvider.name} ${t('settings.provider.api.key.list.title')}`, showHealthCheck: false // FIXME: 目前还没有检查功能 }) @@ -70,11 +60,11 @@ const PreprocessProviderSettings: FC = ({ provider: _provider }) => { } } - const onUpdateOptions = (key: string, value: any) => { - const newOptions = { ...options, [key]: value } - setOptions(newOptions) - updateProvider({ options: newOptions }) - } + // const onUpdateOptions = (key: string, value: any) => { + // const newOptions = { ...options, [key]: value } + // setOptions(newOptions) + // updateProvider({ options: newOptions }) + // } return ( <> @@ -145,7 +135,7 @@ const PreprocessProviderSettings: FC = ({ provider: _provider }) => { )} {/* 这部分看起来暂时用不上了 */} - {hasObjectKey(preprocessProvider, 'options') && preprocessProvider.id === 'system' && ( + {/* {hasObjectKey(preprocessProvider, 'options') && preprocessProvider.id === 'system' && ( <> @@ -177,7 +167,7 @@ const PreprocessProviderSettings: FC = ({ provider: _provider }) => { /> - )} + )} */} ) } diff --git a/src/renderer/src/pages/settings/PreprocessSettings/index.tsx b/src/renderer/src/pages/settings/PreprocessSettings/index.tsx index daa76c042c..f80c0cd679 100644 --- a/src/renderer/src/pages/settings/PreprocessSettings/index.tsx +++ b/src/renderer/src/pages/settings/PreprocessSettings/index.tsx @@ -1,4 +1,3 @@ -import { isMac } from '@renderer/config/constant' import { useTheme } from '@renderer/context/ThemeProvider' import { useDefaultPreprocessProvider, usePreprocessProviders } from '@renderer/hooks/usePreprocess' import { PreprocessProvider } from '@renderer/types' @@ -40,8 +39,9 @@ const PreprocessSettings: FC = () => { placeholder={t('settings.tool.preprocess.provider_placeholder')} options={preprocessProviders.map((p) => ({ value: p.id, - label: p.name, - disabled: !isMac && p.id === 'system' // 在非 Mac 系统下禁用 system 选项 + label: p.name + // 由于system字段实际未使用，先注释掉 + // disabled: !isMac && p.id === 'system' // 在非 Mac 系统下禁用 system 选项 }))} /> diff --git a/src/renderer/src/pages/settings/ProviderSettings/ProviderSetting.tsx b/src/renderer/src/pages/settings/ProviderSettings/ProviderSetting.tsx index b82acaa511..191cce4778 100644 --- a/src/renderer/src/pages/settings/ProviderSettings/ProviderSetting.tsx +++ b/src/renderer/src/pages/settings/ProviderSettings/ProviderSetting.tsx @@ -128,7 +128,6 @@ const ProviderSetting: FC = ({ providerId }) => { const openApiKeyList = async () => { await ApiKeyListPopup.show({ providerId: provider.id, - providerKind: 'llm', title: `${fancyProviderName} ${t('settings.provider.api.key.list.title')}` }) } diff --git a/src/renderer/src/pages/settings/WebSearchSettings/WebSearchProviderSetting.tsx b/src/renderer/src/pages/settings/WebSearchSettings/WebSearchProviderSetting.tsx index 68bb689bd6..5c6faa7b78 100644 --- a/src/renderer/src/pages/settings/WebSearchSettings/WebSearchProviderSetting.tsx +++ b/src/renderer/src/pages/settings/WebSearchSettings/WebSearchProviderSetting.tsx @@ -1,9 +1,14 @@ import { CheckOutlined, ExportOutlined, LoadingOutlined } from '@ant-design/icons' import { loggerService } from '@logger' +import BochaLogo from '@renderer/assets/images/search/bocha.webp' +import ExaLogo from '@renderer/assets/images/search/exa.png' +import SearxngLogo from '@renderer/assets/images/search/searxng.svg' +import TavilyLogo from '@renderer/assets/images/search/tavily.png' import ApiKeyListPopup from '@renderer/components/Popups/ApiKeyListPopup/popup' -import { getWebSearchProviderLogo, WEB_SEARCH_PROVIDER_CONFIG } from '@renderer/config/webSearchProviders' +import { WEB_SEARCH_PROVIDER_CONFIG } from '@renderer/config/webSearchProviders' import { useWebSearchProvider } from '@renderer/hooks/useWebSearchProviders' import WebSearchService from '@renderer/services/WebSearchService' +import { WebSearchProviderId } from '@renderer/types' import { formatApiKeys, hasObjectKey } from '@renderer/utils' import { Button, Divider, Flex, Form, Input, Space, Tooltip } from 'antd' import Link from 'antd/es/typography/Link' @@ -16,7 +21,7 @@ import { SettingDivider, SettingHelpLink, SettingHelpText, SettingHelpTextRow, S const logger = loggerService.withContext('WebSearchProviderSetting') interface Props { - providerId: string + providerId: WebSearchProviderId } const WebSearchProviderSetting: FC = ({ providerId }) => { @@ -74,7 +79,6 @@ const WebSearchProviderSetting: FC = ({ providerId }) => { const openApiKeyList = async () => { await ApiKeyListPopup.show({ providerId: provider.id, - providerKind: 'websearch', title: `${provider.name} ${t('settings.provider.api.key.list.title')}` }) } @@ -132,6 +136,21 @@ const WebSearchProviderSetting: FC = ({ providerId }) => { setBasicAuthPassword(provider.basicAuthPassword ?? '') }, [provider.apiKey, provider.apiHost, provider.basicAuthUsername, provider.basicAuthPassword]) + const getWebSearchProviderLogo = (providerId: WebSearchProviderId) => { + switch (providerId) { + case 'tavily': + return TavilyLogo + case 'searxng': + return SearxngLogo + case 'exa': + return ExaLogo + case 'bocha': + return BochaLogo + default: + return undefined + } + } + return ( <> diff --git a/src/renderer/src/store/websearch.ts b/src/renderer/src/store/websearch.ts index d7f0799814..1e3fe2a25b 100644 --- a/src/renderer/src/store/websearch.ts +++ b/src/renderer/src/store/websearch.ts @@ -1,4 +1,5 @@ import { createSlice, PayloadAction } from '@reduxjs/toolkit' +import { WEB_SEARCH_PROVIDERS } from '@renderer/config/webSearchProviders' import type { Model, WebSearchProvider } from '@renderer/types' export interface SubscribeSource { key: number @@ -42,48 +43,7 @@ export interface WebSearchState { export const initialState: WebSearchState = { defaultProvider: 'local-bing', - providers: [ - { - id: 'tavily', - name: 'Tavily', - apiHost: 'https://api.tavily.com', - apiKey: '' - }, - { - id: 'searxng', - name: 'Searxng', - apiHost: '', - basicAuthUsername: '', - basicAuthPassword: '' - }, - { - id: 'exa', - name: 'Exa', - apiHost: 'https://api.exa.ai', - apiKey: '' - }, - { - id: 'bocha', - name: 'Bocha', - apiHost: 'https://api.bochaai.com', - apiKey: '' - }, - { - id: 'local-google', - name: 'Google', - url: 'https://www.google.com/search?q=%s' - }, - { - id: 'local-bing', - name: 'Bing', - url: 'https://cn.bing.com/search?q=%s&ensearch=1' - }, - { - id: 'local-baidu', - name: 'Baidu', - url: 'https://www.baidu.com/s?wd=%s' - } - ], + providers: WEB_SEARCH_PROVIDERS, searchWithTime: true, maxResults: 5, excludeDomains: [], @@ -111,7 +71,7 @@ const websearchSlice = createSlice({ updateWebSearchProviders: (state, action: PayloadAction) => { state.providers = action.payload }, - updateWebSearchProvider: (state, action: PayloadAction & { id: string }>) => { + updateWebSearchProvider: (state, action: PayloadAction>) => { const index = state.providers.findIndex((provider) => provider.id === action.payload.id) if (index !== -1) { Object.assign(state.providers[index], action.payload) diff --git a/src/renderer/src/types/index.ts b/src/renderer/src/types/index.ts index 87e5843449..ca4a1fe4fd 100644 --- a/src/renderer/src/types/index.ts +++ b/src/renderer/src/types/index.ts @@ -609,8 +609,20 @@ export type KnowledgeBaseParams = { } } +export const PreprocessProviderIds = { + doc2x: 'doc2x', + mistral: 'mistral', + mineru: 'mineru' +} as const + +export type PreprocessProviderId = keyof typeof PreprocessProviderIds + +export const isPreprocessProviderId = (id: string): id is PreprocessProviderId => { + return Object.hasOwn(PreprocessProviderIds, id) +} + export interface PreprocessProvider { - id: string + id: PreprocessProviderId name: string apiKey?: string apiHost?: string @@ -675,8 +687,24 @@ export type ExternalToolResult = { memories?: MemoryItem[] } +export const WebSearchProviderIds = { + tavily: 'tavily', + searxng: 'searxng', + exa: 'exa', + bocha: 'bocha', + 'local-google': 'local-google', + 'local-bing': 'local-bing', + 'local-baidu': 'local-baidu' +} as const + +export type WebSearchProviderId = keyof typeof WebSearchProviderIds + +export const isWebSearchProviderId = (id: string): id is WebSearchProviderId => { + return Object.hasOwn(WebSearchProviderIds, id) +} + export type WebSearchProvider = { - id: string + id: WebSearchProviderId name: string apiKey?: string apiHost?: string