@lobehub/lobehub 2.0.0-next.97 → 2.0.0-next.99

package/packages/database/migrations/meta/_journal.json

@@ -329,6 +329,13 @@
       "when": 1763453175961,
       "tag": "0046_add_parent_id",
       "breakpoints": true
+    },
+    {
+      "idx": 47,
+      "version": "7",
+      "when": 1763535087148,
+      "tag": "0047_add_slug_document",
+      "breakpoints": true
     }
   ],
   "version": "6"

package/packages/database/src/core/migrations.json

@@ -778,15 +778,24 @@
   {
     "sql": [
       "ALTER TABLE \"documents\" ALTER COLUMN \"id\" SET DATA TYPE varchar(255);",
-      "\nALTER TABLE \"documents\" ADD COLUMN \"parent_id\" varchar(255);",
-      "\nALTER TABLE \"files\" ADD COLUMN \"parent_id\" varchar(255);",
-      "\nALTER TABLE \"documents\" ADD CONSTRAINT \"documents_parent_id_documents_id_fk\" FOREIGN KEY (\"parent_id\") REFERENCES \"public\".\"documents\"(\"id\") ON DELETE set null ON UPDATE no action;",
-      "\nALTER TABLE \"files\" ADD CONSTRAINT \"files_parent_id_documents_id_fk\" FOREIGN KEY (\"parent_id\") REFERENCES \"public\".\"documents\"(\"id\") ON DELETE set null ON UPDATE no action;",
-      "\nCREATE INDEX \"documents_parent_id_idx\" ON \"documents\" USING btree (\"parent_id\");",
-      "\nCREATE INDEX \"files_parent_id_idx\" ON \"files\" USING btree (\"parent_id\");"
+      "\nALTER TABLE \"documents\" ADD COLUMN IF NOT EXISTS \"parent_id\" varchar(255);",
+      "\nALTER TABLE \"files\" ADD COLUMN IF NOT EXISTS \"parent_id\" varchar(255);",
+      "\nDO $$ BEGIN\n ALTER TABLE \"documents\" ADD CONSTRAINT \"documents_parent_id_documents_id_fk\" FOREIGN KEY (\"parent_id\") REFERENCES \"public\".\"documents\"(\"id\") ON DELETE set null ON UPDATE no action;\nEXCEPTION\n WHEN duplicate_object THEN null;\nEND $$;",
+      "\nDO $$ BEGIN\n ALTER TABLE \"files\" ADD CONSTRAINT \"files_parent_id_documents_id_fk\" FOREIGN KEY (\"parent_id\") REFERENCES \"public\".\"documents\"(\"id\") ON DELETE set null ON UPDATE no action;\nEXCEPTION\n WHEN duplicate_object THEN null;\nEND $$;",
+      "\nCREATE INDEX IF NOT EXISTS \"documents_parent_id_idx\" ON \"documents\" USING btree (\"parent_id\");",
+      "\nCREATE INDEX IF NOT EXISTS \"files_parent_id_idx\" ON \"files\" USING btree (\"parent_id\");"
     ],
     "bps": true,
     "folderMillis": 1763453175961,
-    "hash": "6cfc00744de6a8f4d60b793673911bb740f9a50661663e28b843e5adae08f94a"
+    "hash": "6c2081c8ac22772a8276052c9ae0852a4c03463cbe998df26f85683ab356b914"
+  },
+  {
+    "sql": [
+      "ALTER TABLE \"documents\" ADD COLUMN IF NOT EXISTS \"slug\" varchar(255);",
+      "\nDO $$ BEGIN\n CREATE UNIQUE INDEX IF NOT EXISTS \"documents_slug_user_id_unique\" ON \"documents\" (\"slug\",\"user_id\") WHERE \"slug\" IS NOT NULL;\nEXCEPTION\n WHEN duplicate_object THEN null;\nEND $$;"
+    ],
+    "bps": true,
+    "folderMillis": 1763535087148,
+    "hash": "57a82ce14d96ffa9f140ff63f00af994e91a74703f4d2378286e36be259f117b"
   }
 ]

package/packages/database/src/models/__tests__/document.test.ts

@@ -0,0 +1,149 @@
+// @vitest-environment node
+import { afterEach, beforeEach, describe, expect, it } from 'vitest';
+
+import { documents, files, users } from '../../schemas';
+import { LobeChatDatabase } from '../../type';
+import { DocumentModel } from '../document';
+import { FileModel } from '../file';
+import { getTestDB } from './_util';
+
+const serverDB: LobeChatDatabase = await getTestDB();
+
+const userId = 'document-model-test-user-id';
+const userId2 = 'document-model-test-user-id-2';
+const documentModel = new DocumentModel(serverDB, userId);
+const documentModel2 = new DocumentModel(serverDB, userId2);
+const fileModel = new FileModel(serverDB, userId);
+const fileModel2 = new FileModel(serverDB, userId2);
+
+beforeEach(async () => {
+  await serverDB.delete(users);
+  await serverDB.insert(users).values([{ id: userId }, { id: userId2 }]);
+});
+
+afterEach(async () => {
+  await serverDB.delete(users);
+  await serverDB.delete(files);
+  await serverDB.delete(documents);
+});
+
+// Helper to create a minimal valid document
+const createTestDocument = async (model: DocumentModel, fModel: FileModel, content: string) => {
+  const { id: fileId } = await fModel.create({
+    fileType: 'text/plain',
+    name: 'test.txt',
+    size: 100,
+    url: 'https://example.com/test.txt',
+  });
+
+  // Fetch the file to get complete data
+  const file = await fModel.findById(fileId);
+  if (!file) throw new Error('File not found after creation');
+
+  const { id } = await model.create({
+    content,
+    fileId: file.id,
+    fileType: 'text/plain',
+    source: file.url,
+    sourceType: 'file',
+    totalCharCount: content.length,
+    totalLineCount: content.split('\n').length,
+  });
+
+  return { documentId: id, file };
+};
+
+describe('DocumentModel', () => {
+  describe('findByFileId', () => {
+    it('should find document by fileId', async () => {
+      const { documentId, file } = await createTestDocument(
+        documentModel,
+        fileModel,
+        'Test content for file',
+      );
+
+      const found = await documentModel.findByFileId(file.id);
+      expect(found).toBeDefined();
+      expect(found?.id).toBe(documentId);
+      expect(found?.fileId).toBe(file.id);
+      expect(found?.content).toBe('Test content for file');
+    });
+
+    it('should not find document from another user', async () => {
+      const { file } = await createTestDocument(documentModel, fileModel, 'Test content');
+
+      // Try to find with another user's model
+      const found = await documentModel2.findByFileId(file.id);
+      expect(found).toBeUndefined();
+    });
+
+    it('should return undefined for non-existent fileId', async () => {
+      const found = await documentModel.findByFileId('non-existent-file-id');
+      expect(found).toBeUndefined();
+    });
+
+    it('should return the first document when multiple documents exist for same file', async () => {
+      const { id: fileId } = await fileModel.create({
+        fileType: 'text/plain',
+        name: 'test.txt',
+        size: 100,
+        url: 'https://example.com/test.txt',
+      });
+
+      const file = await fileModel.findById(fileId);
+      if (!file) throw new Error('File not found after creation');
+
+      const { id: firstId } = await documentModel.create({
+        content: 'First document',
+        fileId: file.id,
+        fileType: 'text/plain',
+        source: file.url,
+        sourceType: 'file',
+        totalCharCount: 14,
+        totalLineCount: 1,
+      });
+
+      await documentModel.create({
+        content: 'Second document',
+        fileId: file.id,
+        fileType: 'text/plain',
+        source: file.url,
+        sourceType: 'file',
+        totalCharCount: 15,
+        totalLineCount: 1,
+      });
+
+      const found = await documentModel.findByFileId(file.id);
+      expect(found).toBeDefined();
+      // Should return the first created document
+      expect(found?.id).toBe(firstId);
+    });
+
+    it('should handle different file types', async () => {
+      const { id: pdfFileId } = await fileModel.create({
+        fileType: 'application/pdf',
+        name: 'document.pdf',
+        size: 5000,
+        url: 'https://example.com/document.pdf',
+      });
+
+      const pdfFile = await fileModel.findById(pdfFileId);
+      if (!pdfFile) throw new Error('File not found after creation');
+
+      await documentModel.create({
+        content: 'PDF content',
+        fileId: pdfFile.id,
+        fileType: 'application/pdf',
+        source: pdfFile.url,
+        sourceType: 'file',
+        totalCharCount: 11,
+        totalLineCount: 1,
+      });
+
+      const found = await documentModel.findByFileId(pdfFile.id);
+      expect(found).toBeDefined();
+      expect(found?.fileType).toBe('application/pdf');
+      expect(found?.content).toBe('PDF content');
+    });
+  });
+});

package/packages/database/src/models/chunk.ts

@@ -181,10 +181,12 @@ export class ChunkModel {
   semanticSearchForChat = async ({
     embedding,
     fileIds,
+    topK = 15,
   }: {
     embedding: number[];
     fileIds: string[] | undefined;
     query: string;
+    topK?: number;
   }) => {
     const similarity = sql<number>`1 - (${cosineDistance(embeddings.embeddings, embedding)})`;
 
@@ -210,7 +212,7 @@ export class ChunkModel {
       .where(inArray(fileChunks.fileId, fileIds))
       .orderBy((t) => desc(t.similarity))
       // Relaxed to 15 for now
-      .limit(15);
+      .limit(topK);
 
     return result.map((item) => {
       return {

package/packages/database/src/models/document.ts

@@ -31,16 +31,22 @@ export class DocumentModel {
     return this.db.delete(documents).where(eq(documents.userId, this.userId));
   };
 
-  query = async () => {
+  query = async (): Promise<DocumentItem[]> => {
     return this.db.query.documents.findMany({
       orderBy: [desc(documents.updatedAt)],
       where: eq(documents.userId, this.userId),
     });
   };
 
-  findById = async (id: string) => {
+  findById = async (id: string): Promise<DocumentItem | undefined> => {
     return this.db.query.documents.findFirst({
-      where: and(eq(documents.id, id), eq(documents.userId, this.userId)),
+      where: and(eq(documents.userId, this.userId), eq(documents.id, id)),
+    });
+  };
+
+  findByFileId = async (fileId: string) => {
+    return this.db.query.documents.findFirst({
+      where: and(eq(documents.userId, this.userId), eq(documents.fileId, fileId)),
     });
   };
 
@@ -48,6 +54,6 @@ export class DocumentModel {
     return this.db
       .update(documents)
       .set({ ...value, updatedAt: new Date() })
-      .where(and(eq(documents.id, id), eq(documents.userId, this.userId)));
+      .where(and(eq(documents.userId, this.userId), eq(documents.id, id)));
   };
 }

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

@@ -1,4 +1,5 @@
 /* eslint-disable sort-keys-fix/sort-keys-fix  */
+import { isNotNull } from 'drizzle-orm';
 import {
   boolean,
   index,
@@ -91,6 +92,8 @@ export const documents = pgTable(
 
     editorData: jsonb('editor_data').$type<Record<string, any>>(),
 
+    slug: varchar('slug', { length: 255 }),
+
     // Timestamps
     ...timestamps,
   },
@@ -100,6 +103,9 @@ export const documents = pgTable(
     index('documents_file_id_idx').on(table.fileId),
     index('documents_parent_id_idx').on(table.parentId),
     uniqueIndex('documents_client_id_user_id_unique').on(table.clientId, table.userId),
+    uniqueIndex('documents_slug_user_id_unique')
+      .on(table.slug, table.userId)
+      .where(isNotNull(table.slug)),
   ],
 );
 
@@ -133,7 +139,7 @@ export const files = pgTable(
     url: text('url').notNull(),
     source: text('source').$type<FileSource>(),
 
-    // Parent document (for folder hierarchy structure)
+    // Parent Folder or Document
     // @ts-ignore
     parentId: varchar('parent_id', { length: 255 }).references(() => documents.id, {
       onDelete: 'set null',

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

@@ -1499,11 +1499,12 @@ const qwenChatModels: AIChatModelCard[] = [
   },
   {
     abilities: {
-      vision: true,
       reasoning: true,
+      vision: true,
     },
     contextWindowTokens: 131_072,
-    description: 'Qwen3 VL 32B 思考模式（开源版），针对高难度强推理与长视频理解场景，提供顶尖的视觉+文本推理能力。',
+    description:
+      'Qwen3 VL 32B 思考模式（开源版），针对高难度强推理与长视频理解场景，提供顶尖的视觉+文本推理能力。',
     displayName: 'Qwen3 VL 32B Thinking',
     id: 'qwen3-vl-32b-thinking',
     maxOutput: 32_768,
@@ -1525,7 +1526,8 @@ const qwenChatModels: AIChatModelCard[] = [
       vision: true,
     },
     contextWindowTokens: 131_072,
-    description: 'Qwen3 VL 32B 非思考模式（Instruct），适用于非思考指令场景，保持强大的视觉理解能力。',
+    description:
+      'Qwen3 VL 32B 非思考模式（Instruct），适用于非思考指令场景，保持强大的视觉理解能力。',
     displayName: 'Qwen3 VL 32B Instruct',
     id: 'qwen3-vl-32b-instruct',
     maxOutput: 32_768,

package/packages/model-runtime/src/core/openaiCompatibleFactory/index.ts

@@ -138,10 +138,10 @@ export interface OpenAICompatibleFactoryOptions<T extends Record<string, any> =
     useToolsCalling?: boolean;
   };
   models?:
-  | ((params: { client: OpenAI }) => Promise<ChatModelCard[]>)
-  | {
-    transformModel?: (model: OpenAI.Model) => ChatModelCard;
-  };
+    | ((params: { client: OpenAI }) => Promise<ChatModelCard[]>)
+    | {
+        transformModel?: (model: OpenAI.Model) => ChatModelCard;
+      };
   provider: string;
   responses?: {
     handlePayload?: (
@@ -317,9 +317,9 @@ export const createOpenAICompatibleRuntime = <T extends Record<string, any> = an
         const postPayload = chatCompletion?.handlePayload
           ? chatCompletion.handlePayload(processedPayload, this._options)
           : ({
-            ...processedPayload,
-            stream: processedPayload.stream ?? true,
-          } as OpenAI.ChatCompletionCreateParamsStreaming);
+              ...processedPayload,
+              stream: processedPayload.stream ?? true,
+            } as OpenAI.ChatCompletionCreateParamsStreaming);
 
         if ((postPayload as any).apiMode === 'responses') {
           return this.handleResponseAPIMode(processedPayload, options);
@@ -385,13 +385,13 @@ export const createOpenAICompatibleRuntime = <T extends Record<string, any> = an
           return StreamingResponse(
             chatCompletion?.handleStream
               ? chatCompletion.handleStream(prod, {
-                callbacks: streamOptions.callbacks,
-                inputStartAt,
-              })
+                  callbacks: streamOptions.callbacks,
+                  inputStartAt,
+                })
               : OpenAIStream(prod, {
-                ...streamOptions,
-                inputStartAt,
-              }),
+                  ...streamOptions,
+                  inputStartAt,
+                }),
             {
               headers: options?.headers,
             },
@@ -415,9 +415,9 @@ export const createOpenAICompatibleRuntime = <T extends Record<string, any> = an
         return StreamingResponse(
           chatCompletion?.handleStream
             ? chatCompletion.handleStream(stream, {
-              callbacks: streamOptions.callbacks,
-              inputStartAt,
-            })
+                callbacks: streamOptions.callbacks,
+                inputStartAt,
+              })
             : OpenAIStream(stream, { ...streamOptions, enableStreaming: false, inputStartAt }),
           {
             headers: options?.headers,
@@ -839,11 +839,11 @@ export const createOpenAICompatibleRuntime = <T extends Record<string, any> = an
         ...res,
         ...(reasoning || reasoning_effort
           ? {
-            reasoning: {
-              ...reasoning,
-              ...(reasoning_effort && { effort: reasoning_effort }),
-            },
-          }
+              reasoning: {
+                ...reasoning,
+                ...(reasoning_effort && { effort: reasoning_effort }),
+              },
+            }
           : {}),
         input,
         ...(max_tokens && { max_output_tokens: max_tokens }),

package/packages/obervability-otel/package.json

@@ -9,12 +9,12 @@
   },
   "dependencies": {
     "@opentelemetry/api": "^1.9.0",
-    "@opentelemetry/auto-instrumentations-node": "^0.66.0",
+    "@opentelemetry/auto-instrumentations-node": "^0.67.0",
     "@opentelemetry/exporter-metrics-otlp-http": "^0.208.0",
     "@opentelemetry/exporter-trace-otlp-http": "^0.208.0",
     "@opentelemetry/instrumentation": "^0.208.0",
     "@opentelemetry/instrumentation-http": "^0.208.0",
-    "@opentelemetry/instrumentation-pg": "^0.60.0",
+    "@opentelemetry/instrumentation-pg": "^0.61.0",
     "@opentelemetry/resources": "^2.2.0",
     "@opentelemetry/sdk-metrics": "^2.2.0",
     "@opentelemetry/sdk-node": "^0.208.0",

package/packages/prompts/src/prompts/knowledgeBaseQA/__snapshots__/formatFileContents.test.ts.snap

@@ -0,0 +1,75 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`promptFileContents > should format multiple file contents 1`] = `
+"<knowledge_base_files totalCount="3">
+<instruction>Use the information from these files to answer the user's question. Always cite the source files.</instruction>
+<file id="file-001" name="auth-guide.md">
+First file content about API authentication.
+</file>
+<file id="file-002" name="db-setup.md">
+Second file content about database setup and configuration.
+</file>
+<file id="file-003" name="deployment.md">
+Third file content with deployment instructions.
+</file>
+</knowledge_base_files>"
+`;
+
+exports[`promptFileContents > should format single file content 1`] = `
+"<knowledge_base_files totalCount="1">
+<instruction>Use the information from these files to answer the user's question. Always cite the source files.</instruction>
+<file id="file-001" name="document.md">
+This is the file content with some important information.
+</file>
+</knowledge_base_files>"
+`;
+
+exports[`promptFileContents > should handle file with error 1`] = `
+"<knowledge_base_files totalCount="1">
+<instruction>Use the information from these files to answer the user's question. Always cite the source files.</instruction>
+<file id="file-404" name="missing.md" error="File not found" />
+</knowledge_base_files>"
+`;
+
+exports[`promptFileContents > should handle file with long content 1`] = `
+"<knowledge_base_files totalCount="1">
+<instruction>Use the information from these files to answer the user's question. Always cite the source files.</instruction>
+<file id="file-long" name="comprehensive-guide.md">
+# Comprehensive Guide
+
+## Introduction
+This is a very long document with multiple sections and detailed information.
+
+## Section 1
+Lorem ipsum dolor sit amet, consectetur adipiscing elit. Sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.
+
+## Section 2
+Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.
+
+## Conclusion
+Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur.
+</file>
+</knowledge_base_files>"
+`;
+
+exports[`promptFileContents > should handle file with special characters in name 1`] = `
+"<knowledge_base_files totalCount="1">
+<instruction>Use the information from these files to answer the user's question. Always cite the source files.</instruction>
+<file id="file-special" name="FAQ: Q&A (2024).md">
+Content from a file with special characters in the name.
+</file>
+</knowledge_base_files>"
+`;
+
+exports[`promptFileContents > should handle mixed successful and error files 1`] = `
+"<knowledge_base_files totalCount="3">
+<instruction>Use the information from these files to answer the user's question. Always cite the source files.</instruction>
+<file id="file-001" name="success.md">
+Successfully loaded content from this file.
+</file>
+<file id="file-002" name="restricted.md" error="Permission denied" />
+<file id="file-003" name="another-success.md">
+Another successfully loaded file with more content.
+</file>
+</knowledge_base_files>"
+`;

package/packages/prompts/src/prompts/knowledgeBaseQA/__snapshots__/formatNoSearchResults.test.ts.snap

@@ -0,0 +1,45 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`promptNoSearchResults > should format no search results with complex query 1`] = `
+"<knowledge_base_search_results query="API authentication with OAuth 2.0" totalCount="0">
+<instruction>No relevant files found in the knowledge base for this query.</instruction>
+<suggestions>
+<suggestion>Try rephrasing your question with different keywords</suggestion>
+<suggestion>Check if the information exists in the uploaded documents</suggestion>
+<suggestion>Ask the user to provide more context or upload relevant documents</suggestion>
+</suggestions>
+</knowledge_base_search_results>"
+`;
+
+exports[`promptNoSearchResults > should format no search results with non-English query 1`] = `
+"<knowledge_base_search_results query="如何配置数据库连接" totalCount="0">
+<instruction>No relevant files found in the knowledge base for this query.</instruction>
+<suggestions>
+<suggestion>Try rephrasing your question with different keywords</suggestion>
+<suggestion>Check if the information exists in the uploaded documents</suggestion>
+<suggestion>Ask the user to provide more context or upload relevant documents</suggestion>
+</suggestions>
+</knowledge_base_search_results>"
+`;
+
+exports[`promptNoSearchResults > should format no search results with simple query 1`] = `
+"<knowledge_base_search_results query="how to install" totalCount="0">
+<instruction>No relevant files found in the knowledge base for this query.</instruction>
+<suggestions>
+<suggestion>Try rephrasing your question with different keywords</suggestion>
+<suggestion>Check if the information exists in the uploaded documents</suggestion>
+<suggestion>Ask the user to provide more context or upload relevant documents</suggestion>
+</suggestions>
+</knowledge_base_search_results>"
+`;
+
+exports[`promptNoSearchResults > should format no search results with special characters 1`] = `
+"<knowledge_base_search_results query="How to use fetchData<T> with async/await?" totalCount="0">
+<instruction>No relevant files found in the knowledge base for this query.</instruction>
+<suggestions>
+<suggestion>Try rephrasing your question with different keywords</suggestion>
+<suggestion>Check if the information exists in the uploaded documents</suggestion>
+<suggestion>Ask the user to provide more context or upload relevant documents</suggestion>
+</suggestions>
+</knowledge_base_search_results>"
+`;

package/packages/prompts/src/prompts/knowledgeBaseQA/__snapshots__/formatSearchResults.test.ts.snap

@@ -0,0 +1,82 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`formatSearchResults > should format file with special characters in name 1`] = `
+"<knowledge_base_search_results query="password reset" totalCount="1">
+<instruction>Here are the search results from the knowledge base. Use the readKnowledge tool with file IDs to get complete content.</instruction>
+<file id="special-123" name="FAQ: Common Questions & Answers (2024).md" relevanceScore="0.88">
+<chunk fileId="special-123" fileName="FAQ: Common Questions & Answers (2024).md" similarity="0.9">Q: How do I reset my password? A: Click on the "Forgot Password" link on the login page.</chunk>
+</file>
+</knowledge_base_search_results>"
+`;
+
+exports[`formatSearchResults > should format long text chunks without truncation 1`] = `
+"<knowledge_base_search_results query="documentation details" totalCount="1">
+<instruction>Here are the search results from the knowledge base. Use the readKnowledge tool with file IDs to get complete content.</instruction>
+<file id="long-doc" name="Detailed Documentation.md" relevanceScore="0.91">
+<chunk fileId="long-doc" fileName="Detailed Documentation.md" similarity="0.93">Lorem ipsum dolor sit amet, consectetur adipiscing elit. Sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.</chunk>
+</file>
+</knowledge_base_search_results>"
+`;
+
+exports[`formatSearchResults > should format multiple files with varying relevance scores 1`] = `
+"<knowledge_base_search_results query="API authentication security" totalCount="3">
+<instruction>Here are the search results from the knowledge base. Use the readKnowledge tool with file IDs to get complete content.</instruction>
+<file id="doc-001" name="API Reference.md" relevanceScore="0.95">
+<chunk fileId="doc-001" fileName="API Reference.md" similarity="0.98">The Authentication API provides endpoints for user login, logout, and token refresh. All endpoints require HTTPS and proper API keys.</chunk>
+<chunk fileId="doc-001" fileName="API Reference.md" similarity="0.92">To authenticate a user, send a POST request to /api/auth/login with username and password in the request body.</chunk>
+</file>
+<file id="doc-002" name="Security Best Practices.md" relevanceScore="0.87">
+<chunk fileId="doc-002" fileName="Security Best Practices.md" similarity="0.89">Always use environment variables to store sensitive information like API keys and database credentials. Never commit these to version control.</chunk>
+<chunk fileId="doc-002" fileName="Security Best Practices.md" similarity="0.85">Implement rate limiting on all public endpoints to prevent abuse and DDoS attacks.</chunk>
+</file>
+<file id="doc-003" name="Troubleshooting Guide.md" relevanceScore="0.73">
+<chunk fileId="doc-003" fileName="Troubleshooting Guide.md" similarity="0.75">If you encounter authentication errors, first verify that your API key is valid and has not expired.</chunk>
+</file>
+</knowledge_base_search_results>"
+`;
+
+exports[`formatSearchResults > should format query with special characters 1`] = `
+"<knowledge_base_search_results query="How to use fetchData<T> with async/await?" totalCount="1">
+<instruction>Here are the search results from the knowledge base. Use the readKnowledge tool with file IDs to get complete content.</instruction>
+<file id="code-sample" name="Code Examples.md" relevanceScore="0.94">
+<chunk fileId="code-sample" fileName="Code Examples.md" similarity="0.96">Use the following TypeScript code: const result = await fetchData<T>({ url, params });</chunk>
+</file>
+</knowledge_base_search_results>"
+`;
+
+exports[`formatSearchResults > should format single file with multiple chunks 1`] = `
+"<knowledge_base_search_results query="how to install the application" totalCount="1">
+<instruction>Here are the search results from the knowledge base. Use the readKnowledge tool with file IDs to get complete content.</instruction>
+<file id="file-123" name="Getting Started Guide.md" relevanceScore="0.92">
+<chunk fileId="file-123" fileName="Getting Started Guide.md" similarity="0.95">To get started with the application, first install all dependencies using npm install or yarn install. Make sure you have Node.js version 16 or higher installed on your system.</chunk>
+<chunk fileId="file-123" fileName="Getting Started Guide.md" similarity="0.88">After installation, you can run the development server with npm run dev. The application will be available at http://localhost:3000 by default.</chunk>
+</file>
+</knowledge_base_search_results>"
+`;
+
+exports[`formatSearchResults > should handle empty file results 1`] = `
+"<knowledge_base_search_results query="no results query" totalCount="0">
+<instruction>No relevant files found in the knowledge base for this query.</instruction>
+</knowledge_base_search_results>"
+`;
+
+exports[`formatSearchResults > should handle file with multiple high-similarity chunks 1`] = `
+"<knowledge_base_search_results query="project setup steps" totalCount="1">
+<instruction>Here are the search results from the knowledge base. Use the readKnowledge tool with file IDs to get complete content.</instruction>
+<file id="comprehensive-guide" name="Complete Tutorial.md" relevanceScore="0.96">
+<chunk fileId="comprehensive-guide" fileName="Complete Tutorial.md" similarity="0.98">Step 1: Initialize the project with the configuration file.</chunk>
+<chunk fileId="comprehensive-guide" fileName="Complete Tutorial.md" similarity="0.97">Step 2: Configure the environment variables for your deployment.</chunk>
+<chunk fileId="comprehensive-guide" fileName="Complete Tutorial.md" similarity="0.96">Step 3: Run the migration scripts to set up the database schema.</chunk>
+<chunk fileId="comprehensive-guide" fileName="Complete Tutorial.md" similarity="0.95">Step 4: Start the development server and verify everything is working.</chunk>
+</file>
+</knowledge_base_search_results>"
+`;
+
+exports[`formatSearchResults > should handle files with low relevance scores 1`] = `
+"<knowledge_base_search_results query="specific technical query" totalCount="1">
+<instruction>Here are the search results from the knowledge base. Use the readKnowledge tool with file IDs to get complete content.</instruction>
+<file id="low-relevance" name="Tangentially Related.md" relevanceScore="0.32">
+<chunk fileId="low-relevance" fileName="Tangentially Related.md" similarity="0.35">This document contains some loosely related information that might be helpful.</chunk>
+</file>
+</knowledge_base_search_results>"
+`;