@lobehub/chat 0.128.9 → 0.129.0

package/src/features/Conversation/Error/APIKeyForm/Perplexity.tsx

@@ -0,0 +1,60 @@
+import { Perplexity } from '@lobehub/icons';
+import { Input } from 'antd';
+import { memo } from 'react';
+import { useTranslation } from 'react-i18next';
+
+import { ModelProvider } from '@/libs/agent-runtime';
+import { useGlobalStore } from '@/store/global';
+import { modelProviderSelectors } from '@/store/global/selectors';
+
+import { FormAction } from '../style';
+
+const PerplexityForm = memo(() => {
+  const { t } = useTranslation('error');
+  // const [showProxy, setShow] = useState(false);
+
+  const [apiKey, setConfig] = useGlobalStore((s) => [
+    modelProviderSelectors.perplexityAPIKey(s),
+    s.setModelProviderConfig,
+  ]);
+
+  return (
+    <FormAction
+      avatar={<Perplexity size={56} />}
+      description={t('unlock.apikey.Perplexity.description')}
+      title={t('unlock.apikey.Perplexity.title')}
+    >
+      <Input.Password
+        autoComplete={'new-password'}
+        onChange={(e) => {
+          setConfig(ModelProvider.Perplexity, { apiKey: e.target.value });
+        }}
+        placeholder={'*********************************'}
+        type={'block'}
+        value={apiKey}
+      />
+      {/*{showProxy ? (*/}
+      {/*  <Input*/}
+      {/*    onChange={(e) => {*/}
+      {/*      setConfig({ endpoint: e.target.value });*/}
+      {/*    }}*/}
+      {/*    placeholder={'https://api.openai.com/v1'}*/}
+      {/*    type={'block'}*/}
+      {/*    value={proxyUrl}*/}
+      {/*  />*/}
+      {/*) : (*/}
+      {/*  <Button*/}
+      {/*    icon={<Icon icon={Network} />}*/}
+      {/*    onClick={() => {*/}
+      {/*      setShow(true);*/}
+      {/*    }}*/}
+      {/*    type={'text'}*/}
+      {/*  >*/}
+      {/*    {t('unlock.apikey.addProxyUrl')}*/}
+      {/*  </Button>*/}
+      {/*)}*/}
+    </FormAction>
+  );
+});
+
+export default PerplexityForm;

package/src/features/Conversation/Error/APIKeyForm/index.tsx

@@ -10,6 +10,7 @@ import BedrockForm from './Bedrock';
 import GoogleForm from './Google';
 import MoonshotForm from './Moonshot';
 import OpenAIForm from './OpenAI';
+import PerplexityForm from './Perplexity';
 import ZhipuForm from './Zhipu';
 
 interface APIKeyFormProps {
@@ -40,6 +41,10 @@ const APIKeyForm = memo<APIKeyFormProps>(({ id, provider }) => {
         return <MoonshotForm />;
       }
 
+      case ModelProvider.Perplexity: {
+        return <PerplexityForm />;
+      }
+
       default:
       case ModelProvider.OpenAI: {
         return <OpenAIForm />;

package/src/features/Conversation/Error/index.tsx

@@ -66,6 +66,7 @@ const ErrorMessageExtra = memo<{ data: ChatMessage }>(({ data }) => {
     case AgentRuntimeErrorType.InvalidZhipuAPIKey:
     case AgentRuntimeErrorType.InvalidMoonshotAPIKey:
     case AgentRuntimeErrorType.InvalidGoogleAPIKey:
+    case AgentRuntimeErrorType.InvalidPerplexityAPIKey:
     case AgentRuntimeErrorType.NoOpenAIAPIKey: {
       return <InvalidAPIKey id={data.id} provider={data.error?.body?.provider} />;
     }

package/src/libs/agent-runtime/error.ts

@@ -25,6 +25,9 @@ export const AgentRuntimeErrorType = {
 
   InvalidOllamaArgs: 'InvalidOllamaArgs',
   OllamaBizError: 'OllamaBizError',
+
+  InvalidPerplexityAPIKey: 'InvalidPerplexityAPIKey',
+  PerplexityBizError: 'PerplexityBizError',
 } as const;
 
 export type ILobeAgentRuntimeErrorType =

package/src/libs/agent-runtime/index.ts

@@ -6,6 +6,7 @@ export { LobeGoogleAI } from './google';
 export { LobeMoonshotAI } from './moonshot';
 export { LobeOllamaAI } from './ollama';
 export { LobeOpenAI } from './openai';
+export { LobePerplexityAI } from './perplexity';
 export * from './types';
 export { AgentRuntimeError } from './utils/createError';
 export { LobeZhipuAI } from './zhipu';

package/src/libs/agent-runtime/perplexity/index.ts

@@ -0,0 +1,86 @@
+import { OpenAIStream, StreamingTextResponse } from 'ai';
+import OpenAI from 'openai';
+
+import { LobeRuntimeAI } from '../BaseAI';
+import { AgentRuntimeErrorType } from '../error';
+import { ChatStreamPayload, ModelProvider } from '../types';
+import { AgentRuntimeError } from '../utils/createError';
+import { debugStream } from '../utils/debugStream';
+import { desensitizeUrl } from '../utils/desensitizeUrl';
+import { DEBUG_CHAT_COMPLETION } from '../utils/env';
+import { handleOpenAIError } from '../utils/handleOpenAIError';
+
+const DEFAULT_BASE_URL = 'https://api.perplexity.ai';
+
+export class LobePerplexityAI implements LobeRuntimeAI {
+  private _llm: OpenAI;
+
+  baseURL: string;
+
+  constructor(apiKey?: string, baseURL: string = DEFAULT_BASE_URL) {
+    if (!apiKey) throw AgentRuntimeError.createError(AgentRuntimeErrorType.InvalidPerplexityAPIKey);
+
+    this._llm = new OpenAI({ apiKey, baseURL });
+    this.baseURL = this._llm.baseURL;
+  }
+
+  async chat(payload: ChatStreamPayload) {
+    try {
+      // Set a default frequency penalty value greater than 0
+      const defaultFrequencyPenalty = 0.1;
+      const chatPayload = {
+        ...payload,
+        frequency_penalty: payload.frequency_penalty || defaultFrequencyPenalty,
+      };
+      const response = await this._llm.chat.completions.create(
+        chatPayload as unknown as OpenAI.ChatCompletionCreateParamsStreaming,
+      );
+
+      const stream = OpenAIStream(response);
+
+      const [debug, returnStream] = stream.tee();
+
+      if (DEBUG_CHAT_COMPLETION) {
+        debugStream(debug).catch(console.error);
+      }
+
+      return new StreamingTextResponse(returnStream);
+    } catch (error) {
+      let desensitizedEndpoint = this.baseURL;
+
+      if (this.baseURL !== DEFAULT_BASE_URL) {
+        desensitizedEndpoint = desensitizeUrl(this.baseURL);
+      }
+
+      if ('status' in (error as any)) {
+        switch ((error as Response).status) {
+          case 401: {
+            throw AgentRuntimeError.chat({
+              endpoint: desensitizedEndpoint,
+              error: error as any,
+              errorType: AgentRuntimeErrorType.InvalidPerplexityAPIKey,
+              provider: ModelProvider.Perplexity,
+            });
+          }
+
+          default: {
+            break;
+          }
+        }
+      }
+
+      const { errorResult, RuntimeError } = handleOpenAIError(error);
+
+      const errorType = RuntimeError || AgentRuntimeErrorType.PerplexityBizError;
+
+      throw AgentRuntimeError.chat({
+        endpoint: desensitizedEndpoint,
+        error: errorResult,
+        errorType,
+        provider: ModelProvider.Perplexity,
+      });
+    }
+  }
+}
+
+export default LobePerplexityAI;

package/src/libs/agent-runtime/types/type.ts

@@ -31,6 +31,7 @@ export enum ModelProvider {
   Moonshot = 'moonshot',
   Ollama = 'ollama',
   OpenAI = 'openai',
+  Perplexity = 'perplexity',
   Tongyi = 'tongyi',
   ZhiPu = 'zhipu',
 }

package/src/locales/default/common.ts

@@ -107,6 +107,7 @@ export default {
     ollama: 'Ollama',
     oneapi: 'One API',
     openai: 'OpenAI',
+    perplexity: 'Perplexity',
     zhipu: '智谱AI',
   },
   noDescription: '暂无描述',

package/src/locales/default/error.ts

@@ -73,6 +73,9 @@ export default {
     InvalidAzureAPIKey: 'Azure API Key 不正确或为空，请检查 Azure API Key 后重试',
     AzureBizError: '请求 Azure AI 服务出错，请根据以下信息排查或重试',
 
+    InvalidPerplexityAPIKey: 'Perplexity API Key 不正确或为空，请检查 Perplexity API Key 后重试',
+    PerplexityBizError: '请求 Perplexity AI 服务出错，请根据以下信息排查或重试',
+
     InvalidOllamaArgs: 'Ollama 配置不正确，请检查 Ollama 配置后重试',
     OllamaBizError: '请求 Ollama 服务出错，请根据以下信息排查或重试',
 
@@ -106,6 +109,10 @@ export default {
         description: '输入你的 OpenAI API Key 即可开始会话。应用不会记录你的 API Key',
         title: '使用自定义 OpenAI API Key',
       },
+      Perplexity: {
+        description: '输入你的 Perplexity API Key 即可开始会话。应用不会记录你的 API Key',
+        title: '使用自定义 Perplexity API Key',
+      },
       Zhipu: {
         description: '输入你的 Zhipu API Key 即可开始会话。应用不会记录你的 API Key',
         title: '使用自定义 Zhipu API Key',

package/src/locales/default/setting.ts

@@ -139,6 +139,14 @@ export default {
         title: 'Azure OpenAI',
       },
     },
+    Perplexity: {
+      title: 'Perplexity',
+      token: {
+        desc: '填入来自 Perplexity AI 的 API Key',
+        placeholder: 'Perplexity AI API Key',
+        title: 'API Key',
+      },
+    },
     Zhipu: {
       title: '智谱',
       token: {

package/src/services/_auth.ts

@@ -48,6 +48,10 @@ export const getProviderAuthPayload = (provider: string) => {
       };
     }
 
+    case ModelProvider.Perplexity: {
+      return { apiKey: modelProviderSelectors.perplexityAPIKey(useGlobalStore.getState()) };
+    }
+
     default:
     case ModelProvider.OpenAI: {
       const openai = modelProviderSelectors.openAIConfig(useGlobalStore.getState());

package/src/store/global/slices/settings/selectors/modelProvider.ts

@@ -7,6 +7,7 @@ import {
   MoonshotProvider,
   OllamaProvider,
   OpenAIProvider,
+  PerplexityProvider,
   ZhiPuProvider,
 } from '@/config/modelProviders';
 import { ChatModelCard, ModelProviderCard } from '@/types/llm';
@@ -48,6 +49,9 @@ const enableOllamaConfigInSettings = (s: GlobalStore) =>
 const enableOllama = (s: GlobalStore) => modelProvider(s).ollama.enabled;
 const ollamaProxyUrl = (s: GlobalStore) => modelProvider(s).ollama.endpoint;
 
+const enablePerplexity = (s: GlobalStore) => modelProvider(s).perplexity.enabled;
+const perplexityAPIKey = (s: GlobalStore) => modelProvider(s).perplexity.apiKey;
+
 // const azureModelList = (s: GlobalStore): ModelProviderCard => {
 //   const azure = azureConfig(s);
 //   return {
@@ -133,6 +137,7 @@ const modelSelectList = (s: GlobalStore): ModelProviderCard[] => {
     { ...GoogleProvider, enabled: enableGoogle(s) },
     { ...BedrockProvider, enabled: enableBedrock(s) },
     { ...OllamaProvider, chatModels: ollamaChatModels, enabled: enableOllama(s) },
+    { ...PerplexityProvider, enabled: enablePerplexity(s) },
   ];
 };
 
@@ -203,4 +208,8 @@ export const modelProviderSelectors = {
   enableOllamaConfigInSettings,
   enableOllama,
   ollamaProxyUrl,
+
+  // Perplexity
+  enablePerplexity,
+  perplexityAPIKey,
 };

package/src/types/settings/modelProvider.ts

@@ -54,6 +54,12 @@ export interface OllamaConfig {
   endpoint?: string;
 }
 
+export interface PerplexityConfig {
+  apiKey?: string;
+  enabled: boolean;
+  endpoint?: string;
+}
+
 export interface GlobalLLMConfig {
   azure: AzureOpenAIConfig;
   bedrock: AWSBedrockConfig;
@@ -61,6 +67,7 @@ export interface GlobalLLMConfig {
   moonshot: MoonshotConfig;
   ollama: OllamaConfig;
   openAI: OpenAIConfig;
+  perplexity: PerplexityConfig;
   zhipu: ZhiPuConfig;
 }