@lobehub/lobehub 2.0.0-next.4 → 2.0.0-next.40

package/packages/types/src/message/db/params.ts

@@ -1,11 +1,20 @@
-import { GroundingSearch } from '../../search';
+/* eslint-disable sort-keys-fix/sort-keys-fix */
+import { z } from 'zod';
+
+import { GroundingSearch, GroundingSearchSchema } from '../../search';
 import {
   ChatImageItem,
+  ChatImageItemSchema,
   ChatMessageError,
+  ChatMessageErrorSchema,
   ChatToolPayload,
+  ChatToolPayloadSchema,
   MessageMetadata,
+  MessageMetadataSchema,
   MessageToolCall,
+  MessageToolCallSchema,
   ModelReasoning,
+  ModelReasoningSchema,
 } from '../common';
 import { UIChatMessage } from '../ui';
 
@@ -34,6 +43,21 @@ export interface CreateMessageResult {
   messages: UIChatMessage[];
 }
 
+/**
+ * Result type for updateMessage
+ * Contains success status and optional message list
+ */
+export interface UpdateMessageResult {
+  /**
+   * Updated message list (only present when success is true and sessionId/topicId provided)
+   */
+  messages?: UIChatMessage[];
+  /**
+   * Whether the update was successful
+   */
+  success: boolean;
+}
+
 export interface NewMessage {
   agentId?: string | null;
   clientId?: string | null;
@@ -67,12 +91,14 @@ export interface UpdateMessageParams {
   imageList?: ChatImageItem[];
   metadata?: MessageMetadata;
   model?: string;
+  observationId?: string;
   provider?: string;
   reasoning?: ModelReasoning;
   role?: string;
   search?: GroundingSearch;
   toolCalls?: MessageToolCall[];
   tools?: ChatToolPayload[] | null;
+  traceId?: string;
 }
 
 export interface NewMessageQueryParams {
@@ -81,3 +107,23 @@ export interface NewMessageQueryParams {
   rewriteQuery: string;
   userQuery: string;
 }
+
+// ========== Zod Schemas ========== //
+
+export const UpdateMessageParamsSchema = z
+  .object({
+    content: z.string().optional(),
+    error: ChatMessageErrorSchema.nullable().optional(),
+    imageList: z.array(ChatImageItemSchema).optional(),
+    metadata: MessageMetadataSchema.optional(),
+    model: z.string().optional(),
+    observationId: z.string().optional(),
+    provider: z.string().optional(),
+    reasoning: ModelReasoningSchema.optional(),
+    role: z.string().optional(),
+    search: GroundingSearchSchema.optional(),
+    toolCalls: z.array(MessageToolCallSchema).optional(),
+    tools: z.array(ChatToolPayloadSchema).nullable().optional(),
+    traceId: z.string().optional(),
+  })
+  .passthrough();

package/packages/types/src/message/ui/chat.ts

@@ -13,7 +13,13 @@ import { ChatMessageExtra } from './extra';
 import { ChatFileChunk } from './rag';
 import { ChatVideoItem } from './video';
 
-export type UIMessageRoleType = 'user' | 'system' | 'assistant' | 'tool' | 'supervisor' | 'group';
+export type UIMessageRoleType =
+  | 'user'
+  | 'system'
+  | 'assistant'
+  | 'tool'
+  | 'supervisor'
+  | 'assistantGroup';
 
 export interface ChatFileItem {
   content?: string;
@@ -26,17 +32,28 @@ export interface ChatFileItem {
 
 export interface AssistantContentBlock {
   content: string;
-  fileList?: ChatFileItem[];
+  error?: ChatMessageError | null;
   id: string;
   imageList?: ChatImageItem[];
   performance?: ModelPerformance;
+  reasoning?: ModelReasoning;
   tools?: ChatToolPayloadWithResult[];
   usage?: ModelUsage;
 }
+interface UIMessageBranch {
+  /** Index of the active branch (0-based) */
+  activeBranchIndex: number;
+  /** Total number of branches */
+  count: number;
+}
 
 export interface UIChatMessage {
   // Group chat fields (alphabetically before other fields)
   agentId?: string | 'supervisor';
+  /**
+   * Branch information for user messages with multiple children
+   */
+  branch?: UIMessageBranch;
   /**
    * children messages for grouped display
    * Used to group tool messages under their parent assistant message
@@ -62,6 +79,7 @@ export interface UIChatMessage {
   imageList?: ChatImageItem[];
   meta: MetaData;
   metadata?: MessageMetadata | null;
+  model?: string | null;
   /**
    * observation id
    */
@@ -78,6 +96,7 @@ export interface UIChatMessage {
   plugin?: ChatPluginPayload;
   pluginError?: any;
   pluginState?: any;
+  provider?: string | null;
   /**
    * quoted other message's id
    */

package/packages/types/src/message/ui/extra.ts

@@ -7,8 +7,8 @@ export interface ChatTTS {
 }
 
 export interface ChatMessageExtra {
-  fromModel?: string;
-  fromProvider?: string;
+  model?: string;
+  provider?: string;
   // 翻译
   translate?: ChatTranslate | false | null;
   // TTS

package/packages/types/src/message/ui/params.ts

@@ -1,9 +1,14 @@
 /* eslint-disable sort-keys-fix/sort-keys-fix , typescript-sort-keys/interface */
+import { z } from 'zod';
+
 import { UploadFileItem } from '../../files';
 import { MessageSemanticSearchChunk } from '../../rag';
-import { ChatMessageError } from '../common/base';
+import { ChatMessageError, ChatMessageErrorSchema } from '../common/base';
 import { ChatPluginPayload } from '../common/tools';
-import { UIChatMessage, UIMessageRoleType } from './chat';
+import { UIChatMessage } from './chat';
+import { SemanticSearchChunkSchema } from './rag';
+
+export type CreateMessageRoleType = 'user' | 'assistant' | 'tool' | 'supervisor';
 
 export interface CreateMessageParams
   extends Partial<Omit<UIChatMessage, 'content' | 'role' | 'topicId' | 'chunksList'>> {
@@ -11,10 +16,10 @@ export interface CreateMessageParams
   error?: ChatMessageError | null;
   fileChunks?: MessageSemanticSearchChunk[];
   files?: string[];
-  fromModel?: string;
-  fromProvider?: string;
+  model?: string;
+  provider?: string;
   groupId?: string;
-  role: UIMessageRoleType;
+  role: CreateMessageRoleType;
   sessionId: string;
   targetId?: string | null;
   threadId?: string | null;
@@ -28,7 +33,7 @@ export interface CreateMessageParams
  */
 export interface CreateNewMessageParams {
   // ========== Required fields ==========
-  role: UIMessageRoleType;
+  role: CreateMessageRoleType;
   content: string;
   sessionId: string;
 
@@ -103,3 +108,43 @@ export interface SendGroupMessageParams {
    */
   targetMemberId?: string | null;
 }
+
+// ========== Zod Schemas ========== //
+
+const UIMessageRoleTypeSchema = z.enum(['user', 'assistant', 'tool', 'supervisor']);
+
+const ChatPluginPayloadSchema = z.object({
+  apiName: z.string(),
+  arguments: z.string(),
+  identifier: z.string(),
+  type: z.string(),
+});
+
+export const CreateNewMessageParamsSchema = z
+  .object({
+    // Required fields
+    role: UIMessageRoleTypeSchema,
+    content: z.string(),
+    sessionId: z.string().nullable().optional(),
+    // Tool related
+    tool_call_id: z.string().optional(),
+    plugin: ChatPluginPayloadSchema.optional(),
+    // Grouping
+    parentId: z.string().optional(),
+    groupId: z.string().nullable().optional(),
+    // Context
+    topicId: z.string().nullable().optional(),
+    threadId: z.string().nullable().optional(),
+    targetId: z.string().nullable().optional(),
+    // Model info
+    model: z.string().nullable().optional(),
+    provider: z.string().nullable().optional(),
+    // Content
+    files: z.array(z.string()).optional(),
+    // Error handling
+    error: ChatMessageErrorSchema.nullable().optional(),
+    // Metadata
+    traceId: z.string().optional(),
+    fileChunks: z.array(SemanticSearchChunkSchema).optional(),
+  })
+  .passthrough();

package/packages/types/src/plugins/mcp.ts

@@ -3,6 +3,7 @@ import { z } from 'zod';
 
 import { MCPErrorType } from '@/libs/mcp';
 
+import { McpConnectionType } from '../discover/mcp';
 import { CustomPluginMetadata } from '../tool/plugin';
 
 /* eslint-disable typescript-sort-keys/string-enum */
@@ -110,7 +111,9 @@ export interface CheckMcpInstallResult {
   }>;
 }
 
-export type MCPPluginListParams = Pick<PluginQueryParams, 'locale' | 'pageSize' | 'page' | 'q'>;
+export type MCPPluginListParams = Pick<PluginQueryParams, 'locale' | 'pageSize' | 'page' | 'q'> & {
+  connectionType?: McpConnectionType;
+};
 
 export interface MCPErrorInfoMetadata {
   errorLog?: string;

package/packages/types/src/search.ts

@@ -1,3 +1,5 @@
+import { z } from 'zod';
+
 export type SearchMode = 'off' | 'auto' | 'on';
 
 export enum ModelSearchImplement {
@@ -27,3 +29,17 @@ export interface GroundingSearch {
   citations?: CitationItem[];
   searchQueries?: string[];
 }
+
+export const GroundingSearchSchema = z.object({
+  citations: z
+    .array(
+      z.object({
+        favicon: z.string().optional(),
+        id: z.string().optional(),
+        title: z.string().optional(),
+        url: z.string(),
+      }),
+    )
+    .optional(),
+  searchQueries: z.array(z.string()).optional(),
+});

package/packages/types/src/serverConfig.ts

@@ -15,9 +15,9 @@ export interface ServerModelProviderConfig {
   enabledModels?: string[];
   fetchOnClient?: boolean;
   /**
-   * the model cards defined in server
+   * the model lists defined in server
    */
-  serverModelCards?: ChatModelCard[];
+  serverModelLists?: ChatModelCard[];
 }
 
 export type ServerLanguageModel = Partial<Record<GlobalLLMProviderKey, ServerModelProviderConfig>>;
@@ -32,10 +32,6 @@ export interface GlobalServerConfig {
    */
   enabledOAuthSSO?: boolean;
   image?: PartialDeep<UserImageConfig>;
-  /**
-   * @deprecated
-   */
-  languageModel?: ServerLanguageModel;
   oAuthSSOProviders?: string[];
   systemAgent?: PartialDeep<UserSystemAgentConfig>;
   telemetry: {

package/packages/types/src/topic/topic.ts

@@ -51,3 +51,17 @@ export interface TopicRankItem {
   sessionId: string | null;
   title: string | null;
 }
+
+export interface CreateTopicParams {
+  favorite?: boolean;
+  groupId?: string | null;
+  messages?: string[];
+  sessionId?: string | null;
+  title: string;
+}
+
+export interface QueryTopicParams {
+  containerId?: string | null; // sessionId or groupId
+  current?: number;
+  pageSize?: number;
+}

package/packages/types/src/usage/usageRecord.ts

@@ -0,0 +1,54 @@
+import { MessageMetadata } from '../message';
+
+export interface UsageRecordItem {
+  createdAt: Date;
+  /**
+   * ID
+   **/
+  id: string;
+  inputStartAt?: Date | null;
+  /**
+   * Meta information
+   **/
+  metadata?: MessageMetadata | null;
+  /**
+   * Model id
+   */
+  model: string;
+  outputFinishAt?: Date | null;
+  outputStartAt?: Date | null;
+  /**
+   * Provider id
+   */
+  provider: string;
+  /**
+   * Spend
+   **/
+  spend: number;
+  /**
+   * Usage details
+   **/
+  totalInputTokens?: number | null;
+  totalOutputTokens?: number | null;
+  totalTokens?: number | null;
+  /**
+   * Performance details
+   **/
+  tps?: number | null;
+  ttft?: number | null;
+  /**
+   * Call types
+   **/
+  type: string;
+  updatedAt: Date;
+  userId: string;
+}
+
+export type UsageLog = {
+  date: number;
+  day: string;
+  records: UsageRecordItem[];
+  totalRequests: number;
+  totalSpend: number;
+  totalTokens: number;
+};

package/packages/types/src/user/index.ts

@@ -1,76 +1,2 @@
-import type { PartialDeep } from 'type-fest';
-import { z } from 'zod';
-
-import { Plans } from '../subscription';
-import { TopicDisplayMode } from '../topic';
-import { UserSettings } from '../user/settings';
-
-export interface LobeUser {
-  avatar?: string;
-  email?: string | null;
-  firstName?: string | null;
-  fullName?: string | null;
-  id: string;
-  latestName?: string | null;
-  username?: string | null;
-}
-
-export const UserGuideSchema = z.object({
-  /**
-   * Move the settings button to the avatar dropdown
-   */
-  moveSettingsToAvatar: z.boolean().optional(),
-
-  /**
-   * Topic Guide
-   */
-  topic: z.boolean().optional(),
-
-  /**
-   * tell user that uploaded files can be found in knowledge base
-   */
-  uploadFileInKnowledgeBase: z.boolean().optional(),
-});
-
-export type UserGuide = z.infer<typeof UserGuideSchema>;
-
-export interface UserPreference {
-  /**
-   * disable markdown rendering in chat input editor
-   */
-  disableInputMarkdownRender?: boolean;
-  /**
-   * enable multi-agent group chat mode
-   */
-  enableGroupChat?: boolean;
-  guide?: UserGuide;
-  hideSyncAlert?: boolean;
-  telemetry: boolean | null;
-  topicDisplayMode?: TopicDisplayMode;
-  /**
-   * whether to use cmd + enter to send message
-   */
-  useCmdEnterToSend?: boolean;
-}
-
-export interface UserInitializationState {
-  avatar?: string;
-  canEnablePWAGuide?: boolean;
-  canEnableTrace?: boolean;
-  email?: string;
-  firstName?: string;
-  fullName?: string;
-  hasConversation?: boolean;
-  isOnboard?: boolean;
-  lastName?: string;
-  preference: UserPreference;
-  settings: PartialDeep<UserSettings>;
-  subscriptionPlan?: Plans;
-  userId?: string;
-  username?: string;
-}
-
-export const NextAuthAccountSchame = z.object({
-  provider: z.string(),
-  providerAccountId: z.string(),
-});
+export * from './preference';
+export * from './settings';

package/packages/types/src/user/preference.ts

@@ -0,0 +1,101 @@
+import type { PartialDeep } from 'type-fest';
+import { z } from 'zod';
+
+import { Plans } from '../subscription';
+import { TopicDisplayMode } from '../topic';
+import { UserSettings } from './settings';
+
+export interface LobeUser {
+  avatar?: string;
+  email?: string | null;
+  firstName?: string | null;
+  fullName?: string | null;
+  id: string;
+  latestName?: string | null;
+  username?: string | null;
+}
+
+export const UserGuideSchema = z.object({
+  /**
+   * Move the settings button to the avatar dropdown
+   */
+  moveSettingsToAvatar: z.boolean().optional(),
+
+  /**
+   * Topic Guide
+   */
+  topic: z.boolean().optional(),
+
+  /**
+   * tell user that uploaded files can be found in knowledge base
+   */
+  uploadFileInKnowledgeBase: z.boolean().optional(),
+});
+
+export type UserGuide = z.infer<typeof UserGuideSchema>;
+
+export const UserLabSchema = z.object({
+  /**
+   * enable multi-agent group chat mode
+   */
+  enableGroupChat: z.boolean().optional(),
+  /**
+   * enable markdown rendering in chat input editor
+   */
+  enableInputMarkdown: z.boolean().optional(),
+});
+
+export type UserLab = z.infer<typeof UserLabSchema>;
+
+export interface UserPreference {
+  /**
+   * disable markdown rendering in chat input editor
+   * @deprecated Use lab.enableInputMarkdown instead
+   */
+  disableInputMarkdownRender?: boolean;
+  guide?: UserGuide;
+  hideSyncAlert?: boolean;
+  /**
+   * lab experimental features
+   */
+  lab?: UserLab;
+  telemetry: boolean | null;
+  topicDisplayMode?: TopicDisplayMode;
+  /**
+   * whether to use cmd + enter to send message
+   */
+  useCmdEnterToSend?: boolean;
+}
+
+export interface UserInitializationState {
+  avatar?: string;
+  canEnablePWAGuide?: boolean;
+  canEnableTrace?: boolean;
+  email?: string;
+  firstName?: string;
+  fullName?: string;
+  hasConversation?: boolean;
+  isOnboard?: boolean;
+  lastName?: string;
+  preference: UserPreference;
+  settings: PartialDeep<UserSettings>;
+  subscriptionPlan?: Plans;
+  userId?: string;
+  username?: string;
+}
+
+export const NextAuthAccountSchame = z.object({
+  provider: z.string(),
+  providerAccountId: z.string(),
+});
+
+export const UserPreferenceSchema = z
+  .object({
+    guide: UserGuideSchema.optional(),
+    hideSyncAlert: z.boolean().optional(),
+    lab: UserLabSchema.optional(),
+    telemetry: z.boolean().nullable(),
+    topicDisplayMode: z.nativeEnum(TopicDisplayMode).optional(),
+    useCmdEnterToSend: z.boolean().optional(),
+  })
+  .partial();

package/packages/types/src/user/settings/index.ts

@@ -1,3 +1,5 @@
+import { z } from 'zod';
+
 import type { LobeAgentSettings } from '../../session';
 import { UserGeneralConfig } from './general';
 import { UserHotkeyConfig } from './hotkey';
@@ -18,6 +20,7 @@ export * from './keyVaults';
 export * from './modelProvider';
 export * from './sync';
 export * from './systemAgent';
+export * from './tool';
 export * from './tts';
 
 /**
@@ -34,3 +37,22 @@ export interface UserSettings {
   tool: UserToolConfig;
   tts: UserTTSConfig;
 }
+
+/**
+ * Zod schema for partial UserSettings updates
+ * Uses passthrough to allow any nested settings fields
+ */
+export const UserSettingsSchema = z
+  .object({
+    defaultAgent: z.any().optional(),
+    general: z.any().optional(),
+    hotkey: z.any().optional(),
+    image: z.any().optional(),
+    keyVaults: z.any().optional(),
+    languageModel: z.any().optional(),
+    systemAgent: z.any().optional(),
+    tool: z.any().optional(),
+    tts: z.any().optional(),
+  })
+  .passthrough()
+  .partial();