@lobehub/chat 0.68.1 → 0.69.1

package/src/pages/api/openai.ts

@@ -1,5 +1,5 @@
-import { OpenAIStream, OpenAIStreamCallbacks, StreamingTextResponse } from 'ai';
-import { Configuration, OpenAIApi } from 'openai-edge';
+import { OpenAIStream, StreamingTextResponse } from 'ai';
+import OpenAI, { ClientOptions } from 'openai';
 
 import { getServerConfig } from '@/config/server';
 import { createErrorResponse } from '@/pages/api/error';
@@ -10,25 +10,29 @@ import { OpenAIStreamPayload } from '@/types/openai';
 export const createOpenAI = (userApiKey: string | null, endpoint?: string | null) => {
   const { OPENAI_API_KEY, OPENAI_PROXY_URL } = getServerConfig();
 
-  const config = new Configuration({
+  const baseURL = endpoint ? endpoint : OPENAI_PROXY_URL ? OPENAI_PROXY_URL : undefined;
+
+  const config: ClientOptions = {
     apiKey: !userApiKey ? OPENAI_API_KEY : userApiKey,
-  });
+  };
 
-  const basePath = endpoint ? endpoint : OPENAI_PROXY_URL ? OPENAI_PROXY_URL : undefined;
+  // a bug with openai: https://github.com/openai/openai-node/issues/283
+  // TODO: should refactor when openai fix the bug
+  if (baseURL) {
+    config.baseURL = baseURL;
+  }
 
-  return new OpenAIApi(config, basePath);
+  return new OpenAI(config);
 };
 
 interface CreateChatCompletionOptions {
   OPENAI_API_KEY: string | null;
-  callbacks?: (payload: OpenAIStreamPayload) => OpenAIStreamCallbacks;
   endpoint?: string | null;
   payload: OpenAIStreamPayload;
 }
 
 export const createChatCompletion = async ({
   payload,
-  callbacks,
   OPENAI_API_KEY,
   endpoint,
 }: CreateChatCompletionOptions) => {
@@ -47,37 +51,30 @@ export const createChatCompletion = async ({
 
   // ============  2. 发送请求   ============ //
 
-  const requestParams = { messages: formatMessages, stream: true, ...params };
-
-  let response: Response;
-
   try {
-    response = await openai.createChatCompletion(requestParams);
+    const response = await openai.chat.completions.create({
+      messages: formatMessages,
+      ...params,
+      stream: true,
+    });
+    const stream = OpenAIStream(response);
+    return new StreamingTextResponse(stream);
   } catch (error) {
-    // 如果 await 超时报错，说明是 OpenAI 服务端的问题
-    return createErrorResponse(ChatErrorType.GatewayTimeout, { message: error });
-  }
-
-  // ============  4. 处理异常响应   ============ //
-  if (!response.ok) {
-    let error;
-
-    try {
-      // 正常情况下应该是 OpenAI 的 JSON
-      error = await response.clone().json();
-    } catch {
-      // 如果不是 JSON，那么可能是其他接口端的响应，读 text 为结果
-      const result = await response.text();
-
-      error = { message: result };
+    // Check if the error is an OpenAI APIError
+    if (error instanceof OpenAI.APIError) {
+      return createErrorResponse(ChatErrorType.OpenAIBizError, {
+        endpoint: !!endpoint ? endpoint : undefined,
+        error: error.error ?? error.cause,
+      });
     }
 
-    return createErrorResponse(ChatErrorType.OpenAIBizError, { ...error, endpoint });
-  }
-
-  // ============  5. 发送正常相应   ============ //
-
-  const stream = OpenAIStream(response, callbacks?.(requestParams));
+    // track the error that not an OpenAI APIError
+    console.error(error);
 
-  return new StreamingTextResponse(stream);
+    // return as a GatewayTimeout error
+    return createErrorResponse(ChatErrorType.InternalServerError, {
+      endpoint,
+      error: JSON.stringify(error),
+    });
+  }
 };

package/src/pages/chat/SessionList/Header.tsx

@@ -63,8 +63,10 @@ const Header = memo<{ mobile?: 'navbar' | 'search' }>(({ mobile }) => {
       </Flexbox>
       <SearchBar
         allowClear
+        enableShortKey
         onChange={(e) => useSessionStore.setState({ searchKeywords: e.target.value })}
         placeholder={t('searchAgentPlaceholder')}
+        shortKey={'k'}
         spotlight
         type={'ghost'}
         value={keywords}

package/src/pages/chat/features/Conversation/ChatList/Plugins/IFrameRender/index.tsx

@@ -1,4 +1,4 @@
-import { PluginRenderProps } from '@lobehub/chat-plugin-sdk';
+import { PluginRenderProps } from '@lobehub/chat-plugin-sdk/client';
 import { Skeleton } from 'antd';
 import { memo, useEffect, useRef, useState } from 'react';
 

package/src/pages/chat/features/Conversation/ChatList/Plugins/IFrameRender/utils.ts

@@ -1,4 +1,4 @@
-import { PluginChannel } from '@lobehub/chat-plugin-sdk';
+import { PluginChannel } from '@lobehub/chat-plugin-sdk/client';
 
 export const onPluginReady = (e: MessageEvent, onReady: () => void) => {
   if (e.data.type === PluginChannel.pluginReadyForRender) {

package/src/pages/chat/features/Conversation/ChatList/Plugins/SystemJsRender/index.tsx

@@ -1,4 +1,4 @@
-import { PluginRender, PluginRenderProps } from '@lobehub/chat-plugin-sdk';
+import { PluginRender, PluginRenderProps } from '@lobehub/chat-plugin-sdk/client';
 import { Skeleton } from 'antd';
 import { memo, useEffect, useState } from 'react';
 

package/src/pages/welcome/Footer/index.tsx

@@ -11,7 +11,9 @@ const Footer = memo(() => {
 
   return (
     <Flexbox align={'center'} horizontal justify={'space-between'} style={{ padding: 16 }}>
-      <span style={{ color: theme.colorTextDescription }}>©{new Date().getFullYear()} LobeHub</span>
+      <span style={{ color: theme.colorTextDescription }}>
+        ©{new Date().getFullYear()} LobeHub
+      </span>
       <Flexbox horizontal>
         <ActionIcon
           icon={DiscordIcon}

package/src/services/chatModel.ts

@@ -1,11 +1,10 @@
 import { merge } from 'lodash-es';
-import { ChatCompletionFunctions } from 'openai-edge/types/api';
 
 import { LOBE_CHAT_ACCESS_CODE, OPENAI_API_KEY_HEADER_KEY, OPENAI_END_POINT } from '@/const/fetch';
 import { useGlobalStore } from '@/store/global';
 import { pluginSelectors, usePluginStore } from '@/store/plugin';
 import { initialLobeAgentConfig } from '@/store/session/initialState';
-import type { OpenAIStreamPayload } from '@/types/openai';
+import type { ChatCompletionFunctions, OpenAIStreamPayload } from '@/types/openai';
 
 import { URLS } from './url';
 

package/src/store/plugin/selectors.ts

@@ -1,8 +1,8 @@
 import { uniqBy } from 'lodash-es';
-import { ChatCompletionFunctions } from 'openai-edge/types/api';
 
 import { PLUGIN_SCHEMA_SEPARATOR } from '@/const/plugin';
 import { pluginHelpers } from '@/store/plugin/helpers';
+import { ChatCompletionFunctions } from '@/types/openai';
 
 import { PluginStoreState } from './initialState';
 

package/src/types/openai.ts

@@ -68,3 +68,26 @@ export interface OpenAIStreamPayload {
    */
   top_p?: number;
 }
+
+export interface ChatCompletionFunctions {
+  /**
+   * The description of what the function does.
+   * @type {string}
+   * @memberof ChatCompletionFunctions
+   */
+  description?: string;
+  /**
+   * The name of the function to be called. Must be a-z, A-Z, 0-9, or contain underscores and dashes, with a maximum length of 64.
+   * @type {string}
+   * @memberof ChatCompletionFunctions
+   */
+  name: string;
+  /**
+   * The parameters the functions accepts, described as a JSON Schema object. See the [guide](/docs/guides/gpt/function-calling) for examples, and the [JSON Schema reference](https://json-schema.org/understanding-json-schema/) for documentation about the format.
+   * @type {{ [key: string]: any }}
+   * @memberof ChatCompletionFunctions
+   */
+  parameters?: {
+    [key: string]: any;
+  };
+}