centersl/lobehub
1
0
Fork 0
mirror of https://github.com/lobehub/lobehub.git synced 2026-03-27 13:29:15 +07:00
Code Issues Packages Projects Releases Wiki Activity

🐛 fix: use configured embedding provider instead of hardcoded OpenAI (#11133)

Browse Source
This commit is contained in:
XYenon
2026-01-04 20:55:41 +08:00
committed by GitHub
parent fe87fa8fbb
commit 503c3eba4e
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/server/routers/lambda/userMemories.ts
View File

@@ -15,7 +15,6 @@ import {
} from '@lobechat/memory-user-memory';
import { LayersEnum, type SearchMemoryResult, searchMemorySchema } from '@lobechat/types';
import { type SQL, and, asc, eq, gte, lte } from 'drizzle-orm';
import { ModelProvider } from 'model-bank';
import pMap from 'p-map';
import { z } from 'zod';
@@ -135,11 +134,14 @@ const searchUserMemories = async (
};
const getEmbeddingRuntime = async (serverDB: LobeChatDatabase, userId: string) => {
const provider = ENABLE_BUSINESS_FEATURES ? BRANDING_PROVIDER : ModelProvider.OpenAI;
// Read user's provider config from database
const agentRuntime = await initModelRuntimeFromDB(serverDB, userId, provider);
const { model: embeddingModel } =
const { provider, model: embeddingModel } =
getServerDefaultFilesConfig().embeddingModel || DEFAULT_USER_MEMORY_EMBEDDING_MODEL_ITEM;
// Read user's provider config from database
const agentRuntime = await initModelRuntimeFromDB(
serverDB,
userId,
ENABLE_BUSINESS_FEATURES ? BRANDING_PROVIDER : provider,
);
return { agentRuntime, embeddingModel };
};