@lobehub/chat 0.161.16 → 0.161.17

package/src/types/agent/index.ts

@@ -1,3 +1,5 @@
+import { z } from 'zod';
+
 import { FewShots, LLMParams } from '@/types/llm';
 
 export type TTSServer = 'openai' | 'edge' | 'microsoft';
@@ -12,27 +14,10 @@ export interface LobeAgentTTSConfig {
     openai: string;
   };
 }
-export interface LobeAgentConfig {
-  autoCreateTopicThreshold: number;
-  compressThreshold?: number;
-  displayMode?: 'chat' | 'docs';
-  enableAutoCreateTopic: boolean;
-  /**
-   * 历史消息长度压缩阈值
-   */
-  enableCompressThreshold?: boolean;
-  /**
-   * 开启历史记录条数
-   */
-  enableHistoryCount?: boolean;
-  enableMaxTokens?: boolean;
 
+export interface LobeAgentConfig {
+  chatConfig: LobeAgentChatConfig;
   fewShots?: FewShots;
-  /**
-   * 历史消息条数
-   */
-  historyCount?: number;
-  inputTemplate?: string;
   /**
    * 角色所使用的语言模型
    * @default gpt-3.5-turbo
@@ -60,6 +45,39 @@ export interface LobeAgentConfig {
   tts: LobeAgentTTSConfig;
 }
 
+export interface LobeAgentChatConfig {
+  autoCreateTopicThreshold: number;
+  compressThreshold?: number;
+  displayMode?: 'chat' | 'docs';
+  enableAutoCreateTopic?: boolean;
+  /**
+   * 历史消息长度压缩阈值
+   */
+  enableCompressThreshold?: boolean;
+  /**
+   * 开启历史记录条数
+   */
+  enableHistoryCount?: boolean;
+  enableMaxTokens?: boolean;
+
+  /**
+   * 历史消息条数
+   */
+  historyCount?: number;
+  inputTemplate?: string;
+}
+
+export const AgentChatConfigSchema = z.object({
+  autoCreateTopicThreshold: z.number().default(2),
+  compressThreshold: z.number().optional(),
+  displayMode: z.enum(['chat', 'docs']).optional(),
+  enableAutoCreateTopic: z.boolean().optional(),
+  enableCompressThreshold: z.boolean().optional(),
+  enableHistoryCount: z.boolean().optional(),
+  enableMaxTokens: z.boolean().optional(),
+  historyCount: z.number().optional(),
+});
+
 export type LobeAgentConfigKeys =
   | keyof LobeAgentConfig
   | ['params', keyof LobeAgentConfig['params']];

package/src/types/session/agentSession.ts

@@ -0,0 +1,35 @@
+import { LobeAgentConfig } from '@/types/agent';
+
+import { MetaData } from '../meta';
+import { SessionGroupId } from './sessionGroup';
+
+export enum LobeSessionType {
+  Agent = 'agent',
+  Group = 'group',
+}
+
+/**
+ * Lobe Agent
+ */
+export interface LobeAgentSession {
+  config: LobeAgentConfig;
+  createdAt: Date;
+  group?: SessionGroupId;
+  id: string;
+  meta: MetaData;
+  model: string;
+  pinned?: boolean;
+  tags?: string[];
+  type: LobeSessionType.Agent;
+  updatedAt: Date;
+}
+
+export interface LobeAgentSettings {
+  /**
+   * 语言模型角色设定
+   */
+  config: LobeAgentConfig;
+  meta: MetaData;
+}
+
+export type LobeSessions = LobeAgentSession[];

package/src/types/session/index.ts

@@ -0,0 +1,10 @@
+import { LobeSessions } from '@/types/session/agentSession';
+import { LobeSessionGroups } from '@/types/session/sessionGroup';
+
+export * from './agentSession';
+export * from './sessionGroup';
+
+export interface ChatSessionList {
+  sessionGroups: LobeSessionGroups;
+  sessions: LobeSessions;
+}

package/src/types/session/sessionGroup.ts

@@ -0,0 +1,24 @@
+import { LobeSessions } from './agentSession';
+
+export enum SessionDefaultGroup {
+  Default = 'default',
+  Pinned = 'pinned',
+}
+
+export type SessionGroupId = SessionDefaultGroup | string;
+
+export interface SessionGroupItem {
+  createdAt: number;
+  id: string;
+  name: string;
+  sort?: number;
+  updatedAt: number;
+}
+
+export type SessionGroups = SessionGroupItem[];
+
+export interface CustomSessionGroup extends SessionGroupItem {
+  children: LobeSessions;
+}
+
+export type LobeSessionGroups = SessionGroupItem[];

package/vitest.config.ts

@@ -13,7 +13,13 @@ export default defineConfig({
     },
     coverage: {
       all: false,
-      exclude: ['__mocks__/**'],
+      exclude: [
+        '__mocks__/**',
+        // just ignore the migration code
+        // we will use pglite in the future
+        // so the coverage of this file is not important
+        'src/database/client/core/db.ts',
+      ],
       provider: 'v8',
       reporter: ['text', 'json', 'lcov', 'text-summary'],
     },

package/src/types/session.ts

@@ -1,57 +0,0 @@
-import { LobeAgentConfig } from '@/types/agent';
-
-import { BaseDataModel, MetaData } from './meta';
-
-export enum LobeSessionType {
-  Agent = 'agent',
-  Group = 'group',
-}
-
-export type SessionGroupId = SessionDefaultGroup | string;
-
-export enum SessionDefaultGroup {
-  Default = 'default',
-  Pinned = 'pinned',
-}
-
-export interface SessionGroupItem {
-  createdAt: number;
-  id: string;
-  name: string;
-  sort?: number;
-  updatedAt: number;
-}
-
-export type SessionGroups = SessionGroupItem[];
-
-/**
- * Lobe Agent
- */
-export interface LobeAgentSession extends BaseDataModel {
-  config: LobeAgentConfig;
-  group?: SessionGroupId;
-  model: string;
-  pinned?: boolean;
-  type: LobeSessionType.Agent;
-}
-
-export interface LobeAgentSettings {
-  /**
-   * 语言模型角色设定
-   */
-  config: LobeAgentConfig;
-  meta: MetaData;
-}
-
-export type LobeSessions = LobeAgentSession[];
-
-export interface CustomSessionGroup extends SessionGroupItem {
-  children: LobeSessions;
-}
-
-export type LobeSessionGroups = SessionGroupItem[];
-
-export interface ChatSessionList {
-  sessionGroups: LobeSessionGroups;
-  sessions: LobeSessions;
-}