diff --git a/chatdesk-ui/app/api/retrieval/process/docx/route.ts b/chatdesk-ui/app/api/retrieval/process/docx/route.ts index a0dc393..c299ce3 100644 --- a/chatdesk-ui/app/api/retrieval/process/docx/route.ts +++ b/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) => ({ diff --git a/chatdesk-ui/app/api/retrieval/process/route.ts b/chatdesk-ui/app/api/retrieval/process/route.ts index 563872f..98196eb 100644 --- a/chatdesk-ui/app/api/retrieval/process/route.ts +++ b/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, diff --git a/chatdesk-ui/app/api/retrieval/retrieve/route.ts b/chatdesk-ui/app/api/retrieval/retrieve/route.ts index 83e79ae..adc2ead 100644 --- a/chatdesk-ui/app/api/retrieval/retrieve/route.ts +++ b/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( diff --git a/chatdesk-ui/components/workspace/workspace-settings.tsx b/chatdesk-ui/components/workspace/workspace-settings.tsx index 458419d..1a8c2f9 100644 --- a/chatdesk-ui/components/workspace/workspace-settings.tsx +++ b/chatdesk-ui/components/workspace/workspace-settings.tsx @@ -143,6 +143,7 @@ export const WorkspaceSettings: FC = ({}) => { embeddingsProvider: defaultChatSettings.embeddingsProvider as | "openai" | "local" + | "bge-m3" }) } diff --git a/chatdesk-ui/types/chat.ts b/chatdesk-ui/types/chat.ts index cbf6fff..6f7b1fe 100644 --- a/chatdesk-ui/types/chat.ts +++ b/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 {