@lobehub/lobehub 2.0.0-next.110 → 2.0.0-next.112

package/packages/database/src/schemas/file.ts

@@ -1,6 +1,7 @@
 /* eslint-disable sort-keys-fix/sort-keys-fix  */
 import { isNotNull } from 'drizzle-orm';
 import {
+  AnyPgColumn,
   boolean,
   index,
   integer,
@@ -41,7 +42,6 @@ export type GlobalFileItem = typeof globalFiles.$inferSelect;
 /**
  * Documents table - Stores file content or web search results
  */
-// @ts-ignore
 export const documents = pgTable(
   'documents',
   {
@@ -72,15 +72,12 @@ export const documents = pgTable(
     source: text('source').notNull(), // File path or web URL
 
     // Associated file (optional)
-    // Forward reference to files table defined below
+    // forward reference needs AnyPgColumn to avoid circular type inference
     // eslint-disable-next-line @typescript-eslint/no-use-before-define
-    // @ts-expect-error - files is defined later in this file, forward reference is valid at runtime
-    // eslint-disable-next-line @typescript-eslint/no-use-before-define
-    fileId: text('file_id').references(() => files.id, { onDelete: 'set null' }),
+    fileId: text('file_id').references((): AnyPgColumn => files.id, { onDelete: 'set null' }),
 
     // Parent document (for folder hierarchy structure)
-    // @ts-ignore
-    parentId: varchar('parent_id', { length: 255 }).references(() => documents.id, {
+    parentId: varchar('parent_id', { length: 255 }).references((): AnyPgColumn => documents.id, {
       onDelete: 'set null',
     }),
 
@@ -113,7 +110,6 @@ export type NewDocument = typeof documents.$inferInsert;
 export type DocumentItem = typeof documents.$inferSelect;
 export const insertDocumentSchema = createInsertSchema(documents);
 
-// @ts-ignore
 export const files = pgTable(
   'files',
   {
@@ -140,8 +136,7 @@ export const files = pgTable(
     source: text('source').$type<FileSource>(),
 
     // Parent Folder or Document
-    // @ts-ignore
-    parentId: varchar('parent_id', { length: 255 }).references(() => documents.id, {
+    parentId: varchar('parent_id', { length: 255 }).references((): AnyPgColumn => documents.id, {
       onDelete: 'set null',
     }),
 

package/packages/database/src/schemas/userMemories.ts

@@ -17,7 +17,7 @@ export const userMemories = pgTable(
     memoryCategory: varchar255('memory_category'),
     memoryLayer: varchar255('memory_layer'),
     memoryType: varchar255('memory_type'),
-    metadata: jsonb('metadata'),
+    metadata: jsonb('metadata').$type<Record<string, unknown>>(),
     tags: text('tags').array(),
 
     title: varchar255('title'),
@@ -55,7 +55,7 @@ export const userMemoriesContexts = pgTable(
     userId: text('user_id').references(() => users.id, { onDelete: 'cascade' }),
     userMemoryIds: jsonb('user_memory_ids'),
 
-    metadata: jsonb('metadata'),
+    metadata: jsonb('metadata').$type<Record<string, unknown>>(),
     tags: text('tags').array(),
 
     associatedObjects: jsonb('associated_objects'),
@@ -99,7 +99,7 @@ export const userMemoriesPreferences = pgTable(
       onDelete: 'cascade',
     }),
 
-    metadata: jsonb('metadata'),
+    metadata: jsonb('metadata').$type<Record<string, unknown>>(),
     tags: text('tags').array(),
 
     conclusionDirectives: text('conclusion_directives'),
@@ -132,7 +132,7 @@ export const userMemoriesIdentities = pgTable(
       onDelete: 'cascade',
     }),
 
-    metadata: jsonb('metadata'),
+    metadata: jsonb('metadata').$type<Record<string, unknown>>(),
     tags: text('tags').array(),
 
     type: varchar255('type'),
@@ -165,7 +165,7 @@ export const userMemoriesExperiences = pgTable(
       onDelete: 'cascade',
     }),
 
-    metadata: jsonb('metadata'),
+    metadata: jsonb('metadata').$type<Record<string, unknown>>(),
     tags: text('tags').array(),
 
     type: varchar255('type'),
@@ -200,16 +200,33 @@ export const userMemoriesExperiences = pgTable(
 );
 
 export type UserMemoryItem = typeof userMemories.$inferSelect;
+export type UserMemoryItemWithoutVectors = Omit<
+  UserMemoryItem,
+  'summaryVector1024' | 'detailsVector1024'
+>;
 export type NewUserMemory = typeof userMemories.$inferInsert;
 
 export type UserMemoryPreference = typeof userMemoriesPreferences.$inferSelect;
+export type UserMemoryPreferencesWithoutVectors = Omit<
+  UserMemoryPreference,
+  'conclusionDirectivesVector'
+>;
 export type NewUserMemoryPreference = typeof userMemoriesPreferences.$inferInsert;
 
 export type UserMemoryContext = typeof userMemoriesContexts.$inferSelect;
+export type UserMemoryContextsWithoutVectors = Omit<
+  UserMemoryContext,
+  'titleVector' | 'descriptionVector'
+>;
 export type NewUserMemoryContext = typeof userMemoriesContexts.$inferInsert;
 
 export type UserMemoryIdentity = typeof userMemoriesIdentities.$inferSelect;
+export type UserMemoryIdentitiesWithoutVectors = Omit<UserMemoryIdentity, 'descriptionVector'>;
 export type NewUserMemoryIdentity = typeof userMemoriesIdentities.$inferInsert;
 
 export type UserMemoryExperience = typeof userMemoriesExperiences.$inferSelect;
+export type UserMemoryExperiencesWithoutVectors = Omit<
+  UserMemoryExperience,
+  'situationVector' | 'actionVector' | 'keyLearningVector'
+>;
 export type NewUserMemoryExperience = typeof userMemoriesExperiences.$inferInsert;

package/packages/model-bank/src/aiModels/qwen.ts

@@ -3,6 +3,31 @@ import { AIChatModelCard, AIImageModelCard } from '../types/aiModel';
 // https://help.aliyun.com/zh/model-studio/models?spm=a2c4g.11186623
 
 const qwenChatModels: AIChatModelCard[] = [
+  {
+    abilities: {
+      functionCall: true,
+      reasoning: true,
+    },
+    contextWindowTokens: 262_144,
+    description:
+      'kimi-k2-thinking模型是月之暗面提供的具有通用 Agentic能力和推理能力的思考模型，它擅长深度推理，并可通过多步工具调用，帮助解决各类难题。',
+    displayName: 'Kimi K2 Thinking',
+    id: 'kimi-k2-thinking',
+    maxOutput: 16_384,
+    organization: 'Qwen',
+    pricing: {
+      currency: 'CNY',
+      units: [
+        { name: 'textInput', rate: 4, strategy: 'fixed', unit: 'millionTokens' },
+        { name: 'textOutput', rate: 16, strategy: 'fixed', unit: 'millionTokens' },
+      ],
+    },
+    releasedAt: '2025-11-10',
+    settings: {
+      extendParams: ['reasoningBudgetToken'],
+    },
+    type: 'chat',
+  },
   {
     abilities: {
       reasoning: true,
@@ -109,6 +134,7 @@ const qwenChatModels: AIChatModelCard[] = [
   {
     abilities: {
       reasoning: true,
+      search: true,
     },
     contextWindowTokens: 131_072,
     description:
@@ -131,6 +157,7 @@ const qwenChatModels: AIChatModelCard[] = [
   {
     abilities: {
       reasoning: true,
+      search: true,
     },
     contextWindowTokens: 131_072,
     description: 'DeepSeek V3.1 模型为混合推理架构模型，同时支持思考模式与非思考模式。',
@@ -151,6 +178,7 @@ const qwenChatModels: AIChatModelCard[] = [
   },
   {
     abilities: {
+      functionCall: true,
       search: true,
     },
     contextWindowTokens: 131_072,
@@ -1231,6 +1259,7 @@ const qwenChatModels: AIChatModelCard[] = [
   },
   {
     abilities: {
+      reasoning: true,
       vision: true,
     },
     config: {
@@ -1239,7 +1268,7 @@ const qwenChatModels: AIChatModelCard[] = [
     contextWindowTokens: 65_536,
     description:
       'Qwen-Omni 模型能够接收文本、图片、音频、视频等多种模态的组合输入，并生成文本或语音形式的回复， 提供多种拟人音色，支持多语言和方言的语音输出，可应用于文本创作、视觉识别、语音助手等场景。',
-    displayName: 'Qwen Omni Turbo',
+    displayName: 'Qwen3 Omni Flash',
     id: 'qwen3-omni-flash',
     maxOutput: 16_384,
     organization: 'Qwen',
@@ -1432,6 +1461,9 @@ const qwenChatModels: AIChatModelCard[] = [
         { name: 'textOutput', rate: 5, strategy: 'fixed', unit: 'millionTokens' },
       ],
     },
+    settings: {
+      extendParams: ['reasoningBudgetToken'],
+    },
     type: 'chat',
   },
   {
@@ -1473,7 +1505,7 @@ const qwenChatModels: AIChatModelCard[] = [
       ],
     },
     settings: {
-      extendParams: ['enableReasoning', 'reasoningBudgetToken'],
+      extendParams: ['reasoningBudgetToken'],
     },
     type: 'chat',
   },
@@ -1517,7 +1549,7 @@ const qwenChatModels: AIChatModelCard[] = [
       ],
     },
     settings: {
-      extendParams: ['enableReasoning', 'reasoningBudgetToken'],
+      extendParams: ['reasoningBudgetToken'],
     },
     type: 'chat',
   },
@@ -1976,7 +2008,9 @@ const qwenChatModels: AIChatModelCard[] = [
   },
   {
     abilities: {
+      functionCall: true,
       reasoning: true,
+      search: true,
     },
     contextWindowTokens: 131_072,
     description:
@@ -1996,6 +2030,10 @@ const qwenChatModels: AIChatModelCard[] = [
     type: 'chat',
   },
   {
+    abilities: {
+      functionCall: true,
+      search: true,
+    },
     contextWindowTokens: 65_536,
     description:
       'DeepSeek-V3 为自研 MoE 模型，671B 参数，激活 37B，在 14.8T token 上进行了预训练，在长文本、代码、数学、百科、中文能力上表现优秀。',

package/packages/model-runtime/src/providers/qwen/index.ts

@@ -51,9 +51,15 @@ export const LobeQwenAI = createOpenAICompatibleRuntime({
               thinking_budget:
                 thinking?.budget_tokens === 0 ? 0 : thinking?.budget_tokens || undefined,
             }
-          : ['qwen3', 'qwen-turbo', 'qwen-plus', 'deepseek-v3.1'].some((keyword) =>
-                model.toLowerCase().includes(keyword),
-              )
+          : [
+                'qwen3',
+                'qwen-turbo',
+                'qwen-plus',
+                'qwen-flash',
+                'deepseek-v3.1',
+                'deepseek-v3.2',
+                'glm',
+              ].some((keyword) => model.toLowerCase().includes(keyword))
             ? {
                 enable_thinking: thinking !== undefined ? thinking.type === 'enabled' : false,
                 thinking_budget:

package/packages/prompts/src/prompts/files/__snapshots__/knowledgeBase.test.ts.snap

@@ -0,0 +1,103 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`promptAgentKnowledge > should format both files and knowledge bases 1`] = `
+"<agent_knowledge>
+<instruction>The following files and knowledge bases are available. For files, refer to their content directly. For knowledge bases, use the searchKnowledgeBase tool to find relevant information.</instruction>
+<files totalCount="1">
+<file id="file1" name="readme.md">
+File content here
+</file>
+</files>
+<knowledge_bases totalCount="1">
+<knowledge_base id="kb1" name="Internal Docs" description="Company knowledge base" />
+</knowledge_bases>
+</agent_knowledge>"
+`;
+
+exports[`promptAgentKnowledge > should format only files when no knowledge bases 1`] = `
+"<agent_knowledge>
+<instruction>The following files are available. Refer to their content directly to answer questions. No knowledge bases are associated.</instruction>
+<files totalCount="2">
+<file id="file1" name="doc1.txt">
+This is the content of document 1
+</file>
+<file id="file2" name="doc2.md">
+This is the content of document 2
+</file>
+</files>
+</agent_knowledge>"
+`;
+
+exports[`promptAgentKnowledge > should format only knowledge bases when no files 1`] = `
+"<agent_knowledge>
+<instruction>The following knowledge bases are available for semantic search. Use the searchKnowledgeBase tool to find relevant information.</instruction>
+<knowledge_bases totalCount="2">
+<knowledge_base id="kb1" name="Documentation" description="API documentation" />
+<knowledge_base id="kb2" name="FAQs" />
+</knowledge_bases>
+</agent_knowledge>"
+`;
+
+exports[`promptAgentKnowledge > should handle file with error 1`] = `
+"<agent_knowledge>
+<instruction>The following files are available. Refer to their content directly to answer questions. No knowledge bases are associated.</instruction>
+<files totalCount="1">
+<file id="file1" name="missing.txt" error="File not found" />
+</files>
+</agent_knowledge>"
+`;
+
+exports[`promptAgentKnowledge > should handle file with multiline content 1`] = `
+"<agent_knowledge>
+<instruction>The following files are available. Refer to their content directly to answer questions. No knowledge bases are associated.</instruction>
+<files totalCount="1">
+<file id="file1" name="multiline.txt">
+Line 1
+Line 2
+Line 3
+
+Line 5 with gap
+</file>
+</files>
+</agent_knowledge>"
+`;
+
+exports[`promptAgentKnowledge > should handle file with special characters in filename 1`] = `
+"<agent_knowledge>
+<instruction>The following files are available. Refer to their content directly to answer questions. No knowledge bases are associated.</instruction>
+<files totalCount="1">
+<file id="file1" name="file with spaces & special-chars.txt">
+Special content
+</file>
+</files>
+</agent_knowledge>"
+`;
+
+exports[`promptAgentKnowledge > should handle knowledge base without description 1`] = `
+"<agent_knowledge>
+<instruction>The following knowledge bases are available for semantic search. Use the searchKnowledgeBase tool to find relevant information.</instruction>
+<knowledge_bases totalCount="1">
+<knowledge_base id="kb1" name="Simple KB" />
+</knowledge_bases>
+</agent_knowledge>"
+`;
+
+exports[`promptAgentKnowledge > should handle multiple files and multiple knowledge bases 1`] = `
+"<agent_knowledge>
+<instruction>The following files and knowledge bases are available. For files, refer to their content directly. For knowledge bases, use the searchKnowledgeBase tool to find relevant information.</instruction>
+<files totalCount="3">
+<file id="file1" name="first.txt">
+Content of first file
+</file>
+<file id="file2" name="second.md">
+Content of second file
+</file>
+<file id="file3" name="broken.pdf" error="Parse error" />
+</files>
+<knowledge_bases totalCount="3">
+<knowledge_base id="kb1" name="Tech Docs" description="Technical documentation" />
+<knowledge_base id="kb2" name="User Guides" />
+<knowledge_base id="kb3" name="FAQ Database" description="Frequently asked questions" />
+</knowledge_bases>
+</agent_knowledge>"
+`;

package/packages/prompts/src/prompts/files/index.ts

@@ -4,6 +4,9 @@ import { filePrompts } from './file';
 import { imagesPrompts } from './image';
 import { videosPrompts } from './video';
 
+export type { KnowledgeBaseInfo, PromptKnowledgeOptions } from './knowledgeBase';
+export { promptAgentKnowledge } from './knowledgeBase';
+
 export const filesPrompts = ({
   imageList,
   fileList,

package/packages/prompts/src/prompts/files/knowledgeBase.test.ts

@@ -0,0 +1,167 @@
+import { describe, expect, it } from 'vitest';
+
+import type { FileContent } from '../knowledgeBaseQA';
+import { promptAgentKnowledge } from './knowledgeBase';
+import type { KnowledgeBaseInfo } from './knowledgeBase';
+
+describe('promptAgentKnowledge', () => {
+  it('should return empty string when no files and no knowledge bases', () => {
+    const result = promptAgentKnowledge({});
+    expect(result).toBe('');
+  });
+
+  it('should format only files when no knowledge bases', () => {
+    const fileContents: FileContent[] = [
+      {
+        content: 'This is the content of document 1',
+        fileId: 'file1',
+        filename: 'doc1.txt',
+      },
+      {
+        content: 'This is the content of document 2',
+        fileId: 'file2',
+        filename: 'doc2.md',
+      },
+    ];
+
+    const result = promptAgentKnowledge({ fileContents });
+    expect(result).toMatchSnapshot();
+  });
+
+  it('should format only knowledge bases when no files', () => {
+    const knowledgeBases: KnowledgeBaseInfo[] = [
+      {
+        description: 'API documentation',
+        id: 'kb1',
+        name: 'Documentation',
+      },
+      {
+        description: null,
+        id: 'kb2',
+        name: 'FAQs',
+      },
+    ];
+
+    const result = promptAgentKnowledge({ knowledgeBases });
+    expect(result).toMatchSnapshot();
+  });
+
+  it('should format both files and knowledge bases', () => {
+    const fileContents: FileContent[] = [
+      {
+        content: 'File content here',
+        fileId: 'file1',
+        filename: 'readme.md',
+      },
+    ];
+
+    const knowledgeBases: KnowledgeBaseInfo[] = [
+      {
+        description: 'Company knowledge base',
+        id: 'kb1',
+        name: 'Internal Docs',
+      },
+    ];
+
+    const result = promptAgentKnowledge({ fileContents, knowledgeBases });
+    expect(result).toMatchSnapshot();
+  });
+
+  it('should handle file with error', () => {
+    const fileContents: FileContent[] = [
+      {
+        content: '',
+        error: 'File not found',
+        fileId: 'file1',
+        filename: 'missing.txt',
+      },
+    ];
+
+    const result = promptAgentKnowledge({ fileContents });
+    expect(result).toMatchSnapshot();
+  });
+
+  it('should handle multiple files and multiple knowledge bases', () => {
+    const fileContents: FileContent[] = [
+      {
+        content: 'Content of first file',
+        fileId: 'file1',
+        filename: 'first.txt',
+      },
+      {
+        content: 'Content of second file',
+        fileId: 'file2',
+        filename: 'second.md',
+      },
+      {
+        content: '',
+        error: 'Parse error',
+        fileId: 'file3',
+        filename: 'broken.pdf',
+      },
+    ];
+
+    const knowledgeBases: KnowledgeBaseInfo[] = [
+      {
+        description: 'Technical documentation',
+        id: 'kb1',
+        name: 'Tech Docs',
+      },
+      {
+        description: null,
+        id: 'kb2',
+        name: 'User Guides',
+      },
+      {
+        description: 'Frequently asked questions',
+        id: 'kb3',
+        name: 'FAQ Database',
+      },
+    ];
+
+    const result = promptAgentKnowledge({ fileContents, knowledgeBases });
+    expect(result).toMatchSnapshot();
+  });
+
+  it('should handle knowledge base without description', () => {
+    const knowledgeBases: KnowledgeBaseInfo[] = [
+      {
+        id: 'kb1',
+        name: 'Simple KB',
+      },
+    ];
+
+    const result = promptAgentKnowledge({ knowledgeBases });
+    expect(result).toMatchSnapshot();
+  });
+
+  it('should handle file with special characters in filename', () => {
+    const fileContents: FileContent[] = [
+      {
+        content: 'Special content',
+        fileId: 'file1',
+        filename: 'file with spaces & special-chars.txt',
+      },
+    ];
+
+    const result = promptAgentKnowledge({ fileContents });
+    expect(result).toMatchSnapshot();
+  });
+
+  it('should handle file with multiline content', () => {
+    const fileContents: FileContent[] = [
+      {
+        content: `Line 1
+Line 2
+Line 3
+
+Line 5 with gap`,
+        fileId: 'file1',
+        filename: 'multiline.txt',
+      },
+    ];
+
+    const result = promptAgentKnowledge({ fileContents });
+    expect(result).toMatchSnapshot();
+  });
+});

package/packages/prompts/src/prompts/files/knowledgeBase.ts

@@ -0,0 +1,85 @@
+import type { FileContent } from '../knowledgeBaseQA';
+
+export interface KnowledgeBaseInfo {
+  description?: string | null;
+  id: string;
+  name: string;
+}
+
+export interface PromptKnowledgeOptions {
+  /** File contents to inject */
+  fileContents?: FileContent[];
+  /** Knowledge bases to include */
+  knowledgeBases?: KnowledgeBaseInfo[];
+}
+
+/**
+ * Formats a single file content with XML tags
+ */
+const formatFileContent = (file: FileContent): string => {
+  if (file.error) {
+    return `<file id="${file.fileId}" name="${file.filename}" error="${file.error}" />`;
+  }
+
+  return `<file id="${file.fileId}" name="${file.filename}">
+${file.content}
+</file>`;
+};
+
+/**
+ * Format agent knowledge (files + knowledge bases) as unified XML prompt
+ */
+export const promptAgentKnowledge = ({
+  fileContents = [],
+  knowledgeBases = [],
+}: PromptKnowledgeOptions) => {
+  const hasFiles = fileContents.length > 0;
+  const hasKnowledgeBases = knowledgeBases.length > 0;
+
+  // If no knowledge at all, return empty
+  if (!hasFiles && !hasKnowledgeBases) {
+    return '';
+  }
+
+  const contentParts: string[] = [];
+
+  // Add instruction based on what's available
+  if (hasFiles && hasKnowledgeBases) {
+    contentParts.push(
+      '<instruction>The following files and knowledge bases are available. For files, refer to their content directly. For knowledge bases, use the searchKnowledgeBase tool to find relevant information.</instruction>',
+    );
+  } else if (hasFiles) {
+    contentParts.push(
+      '<instruction>The following files are available. Refer to their content directly to answer questions. No knowledge bases are associated.</instruction>',
+    );
+  } else {
+    contentParts.push(
+      '<instruction>The following knowledge bases are available for semantic search. Use the searchKnowledgeBase tool to find relevant information.</instruction>',
+    );
+  }
+
+  // Add files section
+  if (hasFiles) {
+    const filesXml = fileContents.map((file) => formatFileContent(file)).join('\n');
+    contentParts.push(`<files totalCount="${fileContents.length}">
+${filesXml}
+</files>`);
+  }
+
+  // Add knowledge bases section
+  if (hasKnowledgeBases) {
+    const kbItems = knowledgeBases
+      .map(
+        (kb) =>
+          `<knowledge_base id="${kb.id}" name="${kb.name}"${kb.description ? ` description="${kb.description}"` : ''} />`,
+      )
+      .join('\n');
+    contentParts.push(`<knowledge_bases totalCount="${knowledgeBases.length}">
+${kbItems}
+</knowledge_bases>`);
+  }
+
+  return `<agent_knowledge>
+${contentParts.join('\n')}
+</agent_knowledge>`;
+};

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

@@ -14,6 +14,7 @@ export enum FileSource {
 }
 
 export interface FileItem {
+  content?: string;
   createdAt: Date;
   enabled?: boolean;
   id: string;

package/packages/types/src/index.ts

@@ -36,4 +36,5 @@ export * from './openai/chat';
 export * from './openai/plugin';
 export * from './subscription';
 export * from './trace';
+export * from './userMemory';
 export * from './zustand';

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

@@ -38,6 +38,7 @@ export enum KnowledgeType {
 
 export interface KnowledgeItem {
   avatar?: string | null;
+  content?: string;
   description?: string | null;
   enabled?: boolean;
   fileType?: string;

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

@@ -0,0 +1,3 @@
+export * from './layers';
+export * from './shared';
+export * from './tools';