@lobehub/chat 0.162.12 → 0.162.13

package/src/types/importer.ts

@@ -0,0 +1,128 @@
+import { LobeAgentConfig } from '@/types/agent';
+import {
+  ChatMessageError,
+  ChatPluginPayload,
+  ChatTTS,
+  ChatToolPayload,
+  ChatTranslate,
+  MessageRoleType,
+} from '@/types/message';
+import { MetaData } from '@/types/meta';
+import { SessionGroupId, SessionGroupItem } from '@/types/session';
+import { ChatTopic } from '@/types/topic';
+
+interface ImportSession {
+  config: LobeAgentConfig;
+  createdAt: string;
+  group?: SessionGroupId;
+  id: string;
+  meta: MetaData;
+  pinned?: boolean;
+  type: 'agent' | 'group';
+  updatedAt: string;
+}
+
+interface ImportMessage {
+  content: string;
+  createdAt: number;
+  error?: ChatMessageError;
+
+  // 扩展字段
+  extra?: {
+    fromModel?: string;
+    fromProvider?: string;
+    // 翻译
+    translate?: ChatTranslate | false | null;
+    // TTS
+    tts?: ChatTTS;
+  } & Record<string, any>;
+  files?: string[];
+  id: string;
+
+  /**
+   * observation id
+   */
+  observationId?: string;
+
+  /**
+   * parent message id
+   */
+  parentId?: string;
+  plugin?: ChatPluginPayload;
+  pluginState?: any;
+
+  quotaId?: string;
+  role: MessageRoleType;
+
+  sessionId?: string;
+  tool_call_id?: string;
+  tools?: ChatToolPayload[];
+
+  topicId?: string;
+  traceId?: string;
+
+  updatedAt: number;
+}
+
+export interface ImporterEntryData {
+  messages?: ImportMessage[];
+  sessionGroups?: SessionGroupItem[];
+  sessions?: ImportSession[];
+  topics?: ChatTopic[];
+  version: number;
+}
+
+export interface ImportResult {
+  added: number;
+  errors: number;
+  skips: number;
+}
+
+export interface ImportResults {
+  messages?: ImportResult;
+  sessionGroups?: ImportResult;
+  sessions?: ImportResult;
+  topics?: ImportResult;
+  type?: string;
+}
+
+export enum ImportStage {
+  Start,
+  Preparing,
+  Uploading,
+  Importing,
+  Success,
+  Error,
+  Finished,
+}
+
+export interface FileUploadState {
+  progress: number;
+  /**
+   * rest time in ms
+   */
+  restTime: number;
+  /**
+   * upload speed in KB/s
+   */
+  speed: number;
+}
+
+export interface ErrorShape {
+  code: string;
+  httpStatus: number;
+  message: string;
+  path?: string;
+}
+
+export interface OnImportCallbacks {
+  onError?: (error: ErrorShape) => void;
+  onFileUploading?: (state: FileUploadState) => void;
+  onStageChange?: (stage: ImportStage) => void;
+  /**
+   *
+   * @param results
+   * @param duration in ms
+   */
+  onSuccess?: (results: ImportResults, duration: number) => void;
+}

package/src/utils/config.ts

@@ -34,21 +34,24 @@ export const exportConfigFile = (config: object, fileName?: string) => {
   a.remove();
 };
 
-export const importConfigFile = (file: File, onConfigImport: (config: ConfigFile) => void) => {
-  file.text().then((text) => {
-    try {
-      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: '导入失败',
-      });
-    }
-  });
+export const importConfigFile = async (
+  file: File,
+  onConfigImport: (config: ConfigFile) => Promise<void>,
+) => {
+  const text = await file.text();
+
+  try {
+    const config = JSON.parse(text);
+    const { state, version } = Migration.migrate(config);
+
+    await onConfigImport({ ...config, state, version });
+  } catch (error) {
+    console.error(error);
+    notification.error({
+      description: `出错原因: ${(error as Error).message}`,
+      message: '导入失败',
+    });
+  }
 };
 
 type CreateConfigFileState<T extends ExportType> = ConfigModelMap[T]['state'];

package/src/utils/speed.ts

@@ -0,0 +1,32 @@
+/**
+ * format speed from KB number to string like KB/s, MB/s or GB/s
+ * @param speed
+ */
+export const formatSpeed = (speed: number) => {
+  let word = '';
+
+  // KB
+  if (speed <= 1000) {
+    word = speed.toFixed(2) + 'KB/s';
+  }
+  // MB
+  else if (speed / 1024 <= 1000) {
+    word = (speed / 1024).toFixed(2) + 'MB/s';
+  }
+  // GB
+  else {
+    word = (speed / 1024 / 1024).toFixed(2) + 'GB/s';
+  }
+
+  return word;
+};
+
+export const formatTime = (timeInSeconds: number): string => {
+  if (timeInSeconds < 60) {
+    return `${timeInSeconds.toFixed(1)} s`;
+  } else if (timeInSeconds < 3600) {
+    return `${(timeInSeconds / 60).toFixed(1)} min`;
+  } else {
+    return `${(timeInSeconds / 3600).toFixed(2)} h`;
+  }
+};

package/src/features/DataImporter/useImportConfig.ts

@@ -1,27 +0,0 @@
-import { useCallback } from 'react';
-
-import { ImportResults, configService } from '@/services/config';
-import { useChatStore } from '@/store/chat';
-import { useSessionStore } from '@/store/session';
-import { importConfigFile } from '@/utils/config';
-
-export const useImportConfig = () => {
-  const refreshSessions = useSessionStore((s) => s.refreshSessions);
-  const [refreshMessages, refreshTopics] = useChatStore((s) => [s.refreshMessages, s.refreshTopic]);
-
-  return useCallback(
-    async (file: File) =>
-      new Promise<ImportResults | undefined>((resolve) => {
-        importConfigFile(file, async (config) => {
-          const data = await configService.importConfigState(config);
-
-          await refreshSessions();
-          await refreshMessages();
-          await refreshTopics();
-
-          resolve(data);
-        });
-      }),
-    [],
-  );
-};