scriptoriumadmin/Roo-Code
1
0
Fork 0
mirror of https://github.com/RooVetGit/Roo-Code.git synced 2026-09-06 08:18:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add Amazon Bedrock embedding models support for OpenAI-compatible API

Browse source 
- Add Amazon Titan Embed v1, v2, and Image models
- Add Cohere embed-english-v3 and embed-multilingual-v3 models
- Support custom models via manual dimension configuration
- Tests pass with no regression

Fixes #9404
This commit is contained in:
Roo Code 2025-11-19 20:36:13 +00:00
parent ee19904b29
commit 895abf6eab
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/shared/embeddingModels.ts
View file

@ -44,14 +44,24 @@ export const EMBEDDING_MODEL_PROFILES: EmbeddingModelProfiles = {
// 'default': { dimension: 768 } // Assuming a default dimension
},
"openai-compatible": {
// OpenAI models
"text-embedding-3-small": { dimension: 1536, scoreThreshold: 0.4 },
"text-embedding-3-large": { dimension: 3072, scoreThreshold: 0.4 },
"text-embedding-ada-002": { dimension: 1536, scoreThreshold: 0.4 },
// Nomic models
"nomic-embed-code": {
dimension: 3584,
scoreThreshold: 0.15,
queryPrefix: "Represent this query for searching relevant code: ",
},
// Amazon Bedrock models
"amazon.titan-embed-text-v1": { dimension: 1536, scoreThreshold: 0.4 },
"amazon.titan-embed-text-v2:0": { dimension: 1024, scoreThreshold: 0.4 },
"amazon.titan-embed-image-v1": { dimension: 1024, scoreThreshold: 0.4 },
"cohere.embed-english-v3": { dimension: 1024, scoreThreshold: 0.4 },
"cohere.embed-multilingual-v3": { dimension: 1024, scoreThreshold: 0.4 },
// Custom model placeholder - users can use any model with manual dimension
// When model is not in this list, the system will use manual modelDimension if provided
},
gemini: {
"text-embedding-004": { dimension: 768 },