.

chatdesk-ui/app/api/retrieval/process/docx/route.ts

@@ -13,7 +13,7 @@ export async function POST(req: Request) {
   const { text, fileId, embeddingsProvider, fileExtension } = json as {
     text: string
     fileId: string
-    embeddingsProvider: "openai" | "local"
+    embeddingsProvider: "openai" | "local" | "bge-m3"
     fileExtension: string
   }
 
@@ -82,6 +82,8 @@ export async function POST(req: Request) {
       })
 
       embeddings = await Promise.all(embeddingPromises)
+    } else if (embeddingsProvider === "bge-m3"){
+      // 示例：调用你自己的 BGE-M3 API 或本地函数
     }
 
     const file_items = chunks.map((chunk, index) => ({

chatdesk-ui/app/api/retrieval/process/route.ts

@@ -136,8 +136,11 @@ export async function POST(req: Request) {
       })
 
       embeddings = await Promise.all(embeddingPromises)
+    } else if (embeddingsProvider === "bge-m3"){
+      // 示例：调用你自己的 BGE-M3 API 或本地函数
     }
 
+
     const file_items = chunks.map((chunk, index) => ({
       file_id,
       user_id: profile.user_id,

chatdesk-ui/app/api/retrieval/retrieve/route.ts

@@ -10,7 +10,7 @@ export async function POST(request: Request) {
   const { userInput, fileIds, embeddingsProvider, sourceCount } = json as {
     userInput: string
     fileIds: string[]
-    embeddingsProvider: "openai" | "local"
+    embeddingsProvider: "openai" | "local" | "bge-m3"
     sourceCount: number
   }
 
@@ -84,6 +84,8 @@ export async function POST(request: Request) {
       }
 
       chunks = localFileItems
+    } else if (embeddingsProvider === "bge-m3"){
+      // 示例：调用你自己的 BGE-M3 API 或本地函数
     }
 
     const mostSimilarChunks = chunks?.sort(

chatdesk-ui/components/workspace/workspace-settings.tsx

@@ -143,6 +143,7 @@ export const WorkspaceSettings: FC<WorkspaceSettingsProps> = ({}) => {
         embeddingsProvider: defaultChatSettings.embeddingsProvider as
           | "openai"
           | "local"
+          | "bge-m3"
       })
     }
 

chatdesk-ui/types/chat.ts

@@ -8,7 +8,7 @@ export interface ChatSettings {
   contextLength: number
   includeProfileContext: boolean
   includeWorkspaceInstructions: boolean
-  embeddingsProvider: "openai" | "local"
+  embeddingsProvider: "openai" | "local" | "bge-m3"
 }
 
 export interface ChatPayload {