Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2025-12-19 14:41:24 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix/cannot get dimension (#11879)

Browse Source 
* fix: use ModernAiProvider for embedding dimensions

* fix(ollama)

* Update src/renderer/src/aiCore/legacy/clients/openai/OpenAIBaseClient.ts

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

---------

Co-authored-by: defi-failure <159208748+defi-failure@users.noreply.github.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
SuYao 2025-12-13 21:09:38 +08:00 committed by GitHub
parent dc0c47c64d
commit 5bd550bfb4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/renderer/src/aiCore/legacy/clients/openai/OpenAIBaseClient.ts
View File

@ -88,7 +88,11 @@ export abstract class OpenAIBaseClient<
} }
override async getEmbeddingDimensions(model: Model): Promise<number> { override async getEmbeddingDimensions(model: Model): Promise<number> {
const sdk = await this.getSdkInstance() let sdk: OpenAI = await this.getSdkInstance()
if (isOllamaProvider(this.provider)) {
const embedBaseUrl = `${this.provider.apiHost.replace(/(\/(api|v1))\/?$/, '')}/v1`
sdk = sdk.withOptions({ baseURL: embedBaseUrl })
}
const data = await sdk.embeddings.create({ const data = await sdk.embeddings.create({
model: model.id, model: model.id,