@lobehub/chat 0.105.2 → 0.107.0

package/src/types/llm.ts

@@ -52,4 +52,4 @@ export interface LLMMessage {
   role: LLMRoleType;
 }
 
-export type LLMExample = LLMMessage[];
+export type FewShots = LLMMessage[];

package/src/types/meta.ts

@@ -1,26 +1,38 @@
-export interface MetaData {
+import { z } from 'zod';
+
+export const LobeMetaDataSchema = z.object({
   /**
    * 角色头像
-   * @description 可选参数，如果不传则使用默认头像
    */
-  avatar?: string;
+  avatar: z.string().optional(),
   /**
    *  背景色
-   * @description 可选参数，如果不传则使用默认背景色
    */
-  backgroundColor?: string;
-  description?: string;
-  tags?: string[];
+  backgroundColor: z.string().optional(),
+  description: z.string().optional(),
+  tags: z.array(z.string()).optional(),
   /**
    * 名称
-   * @description 可选参数，如果不传则使用默认名称
    */
-  title?: string;
-}
+  title: z.string().optional(),
+});
+
+export type MetaData = z.infer<typeof LobeMetaDataSchema>;
 
 export interface BaseDataModel {
-  createAt: number;
+  /**
+   * @deprecated
+   */
+  createAt?: number;
+
+  createdAt: number;
+
   id: string;
   meta: MetaData;
-  updateAt: number;
+
+  /**
+   * @deprecated
+   */
+  updateAt?: number;
+  updatedAt: number;
 }

package/src/types/openai/chat.ts

@@ -1,6 +1,8 @@
-import { OpenAIFunctionCall } from '@/types/chatMessage';
+
 import { LLMRoleType } from '@/types/llm';
 
+import { OpenAIFunctionCall } from './functionCall';
+
 interface UserMessageContentPartText {
   text: string;
   type: 'text';
@@ -31,6 +33,11 @@ export interface OpenAIChatMessage {
   role: LLMRoleType;
 }
 
+export interface OpenAIChatStringMessage {
+  content: string;
+  role: LLMRoleType;
+}
+
 /**
  * @title OpenAI Stream Payload
  */
@@ -40,6 +47,7 @@ export interface OpenAIChatStreamPayload {
    * @default 0
    */
   frequency_penalty?: number;
+  functions?: ChatCompletionFunctions[];
   /**
    * @title 生成文本的最大长度
    */

package/src/types/openai/functionCall.ts

@@ -0,0 +1,4 @@
+export interface OpenAIFunctionCall {
+  arguments: string;
+  name: string;
+}

package/src/types/session.ts

@@ -1,106 +1,25 @@
-import { ChatMessageMap } from './chatMessage';
-import { LLMExample, LLMParams, LanguageModel } from './llm';
+import { LobeAgentConfig } from '@/types/agent';
+
 import { BaseDataModel, MetaData } from './meta';
-import { ChatTopicMap } from './topic';
 
 export enum LobeSessionType {
-  /**
-   * 角色
-   */
   Agent = 'agent',
-  /**
-   * 群聊
-   */
   Group = 'group',
 }
 
-interface LobeSessionBase extends BaseDataModel {
-  /**
-   * 聊天记录
-   */
-  chats: ChatMessageMap;
-  files?: string[];
-  /**
-   * 置顶
-   */
-  pinned?: boolean;
-  /**
-   * 主题
-   */
-  topics?: ChatTopicMap;
-  /**
-   * 每个会话的类别
-   */
-  type: LobeSessionType;
-}
-
-export type TTSServer = 'openai' | 'edge' | 'microsoft';
+export type SessionGroupKey = 'pinned' | 'default' | string;
 
-export interface LobeAgentTTSConfig {
-  showAllLocaleVoice?: boolean;
-  sttLocale: 'auto' | string;
-  ttsService: TTSServer;
-  voice: {
-    edge?: string;
-    microsoft?: string;
-    openai: string;
-  };
-}
-
-export interface LobeAgentConfig {
-  autoCreateTopicThreshold: number;
-  compressThreshold?: number;
-  displayMode?: 'chat' | 'docs';
-  enableAutoCreateTopic: boolean;
-  /**
-   * 历史消息长度压缩阈值
-   */
-  enableCompressThreshold?: boolean;
-  /**
-   * 开启历史记录条数
-   */
-  enableHistoryCount?: boolean;
-  enableMaxTokens?: boolean;
-  /**
-   * 语言模型示例
-   */
-  example?: LLMExample;
-  /**
-   * 历史消息条数
-   */
-  historyCount?: number;
-  inputTemplate?: string;
-  /**
-   * 角色所使用的语言模型
-   * @default gpt-3.5-turbo
-   */
-  model: LanguageModel | string;
-  /**
-   * 语言模型参数
-   */
-  params: LLMParams;
-  /**
-   * 启用的插件
-   */
-  plugins?: string[];
-  /**
-   * 系统角色
-   */
-  systemRole: string;
-  /**
-   * 语音服务
-   */
-  tts: LobeAgentTTSConfig;
+export enum SessionGroupDefaultKeys {
+  Default = 'default',
+  Pinned = 'pinned',
 }
 
 /**
- * Lobe Agent会话
+ * Lobe Agent
  */
-export interface LobeAgentSession extends LobeSessionBase {
-  /**
-   * 语言模型角色设定
-   */
+export interface LobeAgentSession extends BaseDataModel {
   config: LobeAgentConfig;
+  group?: SessionGroupKey;
   type: LobeSessionType.Agent;
 }
 
@@ -111,8 +30,5 @@ export interface LobeAgentSettings {
   config: LobeAgentConfig;
   meta: MetaData;
 }
-export type LobeSessions = Record<string, LobeAgentSession>;
 
-export type LobeAgentConfigKeys =
-  | keyof LobeAgentConfig
-  | ['params', keyof LobeAgentConfig['params']];
+export type LobeSessions = LobeAgentSession[];

package/src/types/settings.ts

@@ -45,6 +45,8 @@ export interface GlobalBaseSettings {
 
 export type GlobalDefaultAgent = Pick<LobeAgentSession, 'config' | 'meta'>;
 
+export type CustomModels = { displayName: string; name: string }[];
+
 export interface OpenAIConfig {
   OPENAI_API_KEY: string;
   azureApiVersion?: string;

package/src/types/translate.ts

@@ -1,4 +1,5 @@
 export interface Translate {
-  from: string;
+  from?: string;
   to: string;
 }
+

package/src/utils/config.ts

@@ -35,27 +35,21 @@ export const exportConfigFile = (config: object, fileName?: string) => {
   a.remove();
 };
 
-export const importConfigFile = (info: any, onConfigImport: (config: ConfigFile) => void) => {
-  const reader = new FileReader();
-  //读取完文件之后的回调函数
-  reader.onloadend = function (evt) {
-    const fileString = evt.target?.result;
-    const fileJson = fileString as string;
-
+export const importConfigFile = (file: File, onConfigImport: (config: ConfigFile) => void) => {
+  file.text().then((text) => {
     try {
-      const config = JSON.parse(fileJson);
+      const config = JSON.parse(text);
       const { state } = Migration.migrate(config);
 
       onConfigImport({ ...config, state });
     } catch (error) {
+      console.error(error);
       notification.error({
         description: `出错原因: ${(error as Error).message}`,
         message: '导入失败',
       });
     }
-  };
-  //@ts-ignore file 类型不明确
-  reader.readAsText(info.file.originFileObj, 'utf8');
+  });
 };
 
 type CreateConfigFileState<T extends ExportType> = ConfigModelMap[T]['state'];

package/src/utils/fetch.ts

@@ -27,6 +27,7 @@ export const getMessageError = async (response: Response) => {
 
 export interface FetchSSEOptions {
   onErrorHandle?: (error: ChatMessageError) => void;
+  onFinish?: (text: string) => void;
   onMessageHandle?: (text: string) => void;
 }
 
@@ -51,7 +52,7 @@ export const fetchSSE = async (fetchFn: () => Promise<Response>, options: FetchS
   const data = response.body;
 
   if (!data) return;
-
+  let output = '';
   const reader = data.getReader();
   const decoder = new TextDecoder();
 
@@ -62,9 +63,12 @@ export const fetchSSE = async (fetchFn: () => Promise<Response>, options: FetchS
     done = doneReading;
     const chunkValue = decoder.decode(value, { stream: true });
 
+    output += chunkValue;
     options.onMessageHandle?.(chunkValue);
   }
 
+  options?.onFinish?.(output);
+
   return returnRes;
 };
 
@@ -74,6 +78,7 @@ interface FetchAITaskResultParams<T> {
    * 错误处理函数
    */
   onError?: (e: Error, rawError?: any) => void;
+  onFinish?: (text: string) => void;
   /**
    * 加载状态变化处理函数
    * @param loading - 是否处于加载状态
@@ -84,7 +89,6 @@ interface FetchAITaskResultParams<T> {
    * @param text - 消息内容
    */
   onMessageHandle?: (text: string) => void;
-
   /**
    * 请求对象
    */
@@ -96,6 +100,7 @@ export const fetchAIFactory =
   async ({
     params,
     onMessageHandle,
+    onFinish,
     onError,
     onLoadingChange,
     abortController,
@@ -114,6 +119,7 @@ export const fetchAIFactory =
       onErrorHandle: (error) => {
         errorHandle(new Error(error.message), error);
       },
+      onFinish,
       onMessageHandle,
     }).catch(errorHandle);
 

package/src/app/chat/features/Conversation/usePluginsInit.ts

@@ -1,14 +0,0 @@
-import { usePluginStore } from '@/store/plugin';
-import { useSessionStore } from '@/store/session';
-import { agentSelectors } from '@/store/session/selectors';
-
-export const usePluginsInit = () => {
-  const [plugins] = useSessionStore((s) => [agentSelectors.currentAgentPlugins(s)]);
-
-  const useFetchPluginStore = usePluginStore((s) => s.useFetchPluginStore);
-  useFetchPluginStore();
-
-  const checkPluginsIsInstalled = usePluginStore((s) => s.useCheckPluginsIsInstalled);
-
-  checkPluginsIsInstalled(plugins);
-};

package/src/database/core/schema.ts

@@ -1,9 +0,0 @@
-import { LocalFile } from '@/types/database/files';
-
-export const lobeDBSchema = {
-  files: '&id, name, fileType, saveMode',
-};
-
-export interface LobeDBSchemaMap {
-  files: LocalFile;
-}

package/src/helpers/export.ts

@@ -1,69 +0,0 @@
-import { settingsSelectors, useGlobalStore } from '@/store/global';
-import { useSessionStore } from '@/store/session';
-import { sessionSelectors } from '@/store/session/selectors';
-import { createConfigFile, exportConfigFile } from '@/utils/config';
-
-const getSessions = () => sessionSelectors.exportSessions(useSessionStore.getState());
-const getSession = (id: string) => sessionSelectors.getSessionById(id)(useSessionStore.getState());
-
-const getSettings = () => settingsSelectors.exportSettings(useGlobalStore.getState());
-
-// =============   导出所有角色   ============= //
-
-export const exportAgents = () => {
-  const agents = sessionSelectors.exportAgents(useSessionStore.getState());
-
-  const config = createConfigFile('agents', agents);
-
-  exportConfigFile(config, 'agents');
-};
-
-// =============   导出单个角色   ============= //
-
-const getAgent = (id: string) => sessionSelectors.getExportAgent(id)(useSessionStore.getState());
-
-export const exportSingleAgent = (id: string) => {
-  const agent = getAgent(id);
-  if (!agent) return;
-
-  const config = createConfigFile('agents', { sessions: { [id]: agent } });
-
-  exportConfigFile(config, agent.meta?.title || 'agent');
-};
-
-// =============   导出所有会话   ============= //
-
-export const exportSessions = () => {
-  const config = createConfigFile('sessions', getSessions());
-
-  exportConfigFile(config, 'sessions');
-};
-
-// =============   导出单个会话   ============= //
-export const exportSingleSession = (id: string) => {
-  const session = getSession(id);
-  if (!session) return;
-  const sessions = { [id]: session };
-
-  const config = createConfigFile('singleSession', { sessions });
-
-  exportConfigFile(config, `${session.meta?.title}-session`);
-};
-
-// =============   导出设置会话   ============= //
-export const exportSettings = () => {
-  const settings = getSettings();
-
-  const config = createConfigFile('settings', { settings });
-
-  exportConfigFile(config, 'settings');
-};
-
-export const exportAll = () => {
-  const sessions = getSessions();
-  const settings = getSettings();
-
-  const config = createConfigFile('all', { ...sessions, settings });
-
-  exportConfigFile(config, 'config');
-};

package/src/hooks/useExportConfig.ts

@@ -1,6 +0,0 @@
-import { useMemo } from 'react';
-
-import { exportAgents, exportAll, exportSessions, exportSettings } from '@/helpers/export';
-
-export const useExportConfig = () =>
-  useMemo(() => ({ exportAgents, exportAll, exportSessions, exportSettings }), []);

package/src/store/session/slices/chat/actions/index.ts

@@ -1,32 +0,0 @@
-import { StateCreator } from 'zustand/vanilla';
-
-import { SessionStore } from '@/store/session';
-
-import { ChatMessageAction, chatMessage } from './message';
-import { ChatPluginAction, chatPlugin } from './plugin';
-import { ShareAction, chatShare } from './share';
-import { ChatTopicAction, chatTopic } from './topic';
-import { ChatTranslateAction, chatTranslate } from './translate';
-
-/**
- * 聊天操作
- */
-export interface ChatAction
-  extends ChatTopicAction,
-    ChatMessageAction,
-    ShareAction,
-    ChatTranslateAction,
-    ChatPluginAction {}
-
-export const createChatSlice: StateCreator<
-  SessionStore,
-  [['zustand/devtools', never]],
-  [],
-  ChatAction
-> = (...params) => ({
-  ...chatTopic(...params),
-  ...chatMessage(...params),
-  ...chatShare(...params),
-  ...chatTranslate(...params),
-  ...chatPlugin(...params),
-});

package/src/store/session/slices/chat/actions/topic.ts

@@ -1,171 +0,0 @@
-import { StateCreator } from 'zustand/vanilla';
-
-import { chainSummaryTitle } from '@/chains/summaryTitle';
-import { LOADING_FLAT } from '@/const/message';
-import { chatService } from '@/services/chat';
-import { SessionStore } from '@/store/session';
-import { setNamespace } from '@/utils/storeDebug';
-import { nanoid } from '@/utils/uuid';
-
-import { sessionSelectors } from '../../session/selectors';
-import { ChatTopicDispatch, topicReducer } from '../reducers/topic';
-import { chatSelectors, topicSelectors } from '../selectors';
-
-const t = setNamespace('chat/topic');
-export interface ChatTopicAction {
-  /**
-   * 分发主题
-   * @param payload - 要分发的主题
-   */
-  dispatchTopic: (payload: ChatTopicDispatch) => void;
-  openNewTopicOrSaveTopic: () => void;
-  /**
-   * 移出所有话题
-   */
-  removeAllTopic: () => void;
-  /**
-   * 移除话题
-   * @param id
-   */
-  removeTopic: (id: string) => void;
-  /**
-   * 移出所有未标记的话题
-   */
-  removeUnstarredTopic: () => void;
-  /**
-   * 将当前消息保存为主题
-   */
-  saveToTopic: () => Promise<string | undefined>;
-  /**
-   * 切换主题
-   * @param id - 要切换的主题的 ID
-   */
-  toggleTopic: (id?: string) => void;
-  /**
-   * 更新主题加载状态
-   * @param id - 要更新的主题的 ID
-   */
-  updateTopicLoading: (id?: string) => void;
-}
-
-export const chatTopic: StateCreator<
-  SessionStore,
-  [['zustand/devtools', never]],
-  [],
-  ChatTopicAction
-> = (set, get) => ({
-  dispatchTopic: (payload) => {
-    const { activeId } = get();
-    const session = sessionSelectors.currentSession(get());
-    if (!activeId || !session) return;
-
-    const topics = topicReducer(session.topics || {}, payload);
-
-    get().dispatchSession({ id: activeId, topics, type: 'updateSessionTopic' });
-  },
-  openNewTopicOrSaveTopic: () => {
-    const { toggleTopic, saveToTopic, activeTopicId } = get();
-    const hasTopic = !!activeTopicId;
-
-    if (hasTopic) toggleTopic();
-    else {
-      saveToTopic();
-    }
-  },
-  removeAllTopic: () => {
-    const { removeTopic, toggleTopic } = get();
-    const topics = topicSelectors.currentTopics(get());
-
-    for (const { id } of topics) {
-      removeTopic(id);
-    }
-
-    // 切换到默认 topic
-    toggleTopic();
-  },
-  removeTopic: (id) => {
-    const { dispatchTopic, dispatchMessage, toggleTopic } = get();
-
-    // 移除关联的 message
-    const messages = topicSelectors.getTopicMessages(id)(get());
-    for (const m of messages) {
-      dispatchMessage({ id: m.id, type: 'deleteMessage' });
-    }
-
-    // 最后移除 topic
-    dispatchTopic({ id, type: 'deleteChatTopic' });
-
-    // 切换到默认 topic
-    toggleTopic();
-  },
-  removeUnstarredTopic: () => {
-    const { removeTopic, toggleTopic } = get();
-    const topics = topicSelectors.currentTopics(get());
-
-    for (const { id, favorite } of topics) {
-      if (!favorite) removeTopic(id);
-    }
-
-    // 切换到默认 topic
-    toggleTopic();
-  },
-  saveToTopic: async () => {
-    const session = sessionSelectors.currentSession(get());
-    if (!session) return;
-
-    const { dispatchTopic, dispatchMessage, updateTopicLoading } = get();
-    // get current messages
-    const messages = chatSelectors.currentChats(get());
-
-    // if there is no message, stop saving
-    if (messages.length === 0) return;
-
-    const topicId = nanoid();
-
-    const defaultTitle = '默认话题';
-    const newTopic = {
-      createAt: Date.now(),
-      id: topicId,
-      title: defaultTitle,
-      updateAt: Date.now(),
-    };
-
-    dispatchTopic({
-      topic: newTopic,
-      type: 'addChatTopic',
-    });
-
-    // 为所有 message 添加 topicId
-    for (const m of messages) {
-      dispatchMessage({ id: m.id, key: 'topicId', type: 'updateMessage', value: topicId });
-    }
-
-    dispatchTopic({ id: topicId, key: 'title', type: 'updateChatTopic', value: LOADING_FLAT });
-
-    let output = '';
-
-    // 自动总结话题标题
-    chatService.fetchPresetTaskResult({
-      onError: () => {
-        dispatchTopic({ id: topicId, key: 'title', type: 'updateChatTopic', value: defaultTitle });
-      },
-      onLoadingChange: (loading) => {
-        updateTopicLoading(loading ? topicId : undefined);
-      },
-      onMessageHandle: (x) => {
-        output += x;
-        dispatchTopic({ id: topicId, key: 'title', type: 'updateChatTopic', value: output });
-      },
-      params: await chainSummaryTitle(messages),
-    });
-
-    return topicId;
-  },
-  toggleTopic: (id) => {
-    set({ activeTopicId: id }, false, t('toggleTopic'));
-  },
-
-  updateTopicLoading: (id) => {
-    set({ topicLoadingId: id }, false, t('updateTopicLoading'));
-  },
-});