aljaz/enclava
1
0
Fork 0
mirror of https://github.com/aljazceru/enclava.git synced 2025-12-17 23:44:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

add metadata support to RAG

Browse Source
This commit is contained in:
Aljaz Ceru
2025-12-02 21:22:03 +01:00
parent 841d79f26b
commit e6914ed4a4
18 changed files with 2013 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/app/services/embedding_service.py
View File

@@ -20,9 +20,9 @@ class EmbeddingService:
def __init__(self, model_name: Optional[str] = None):
self.model_name = model_name or getattr(
settings, "RAG_EMBEDDING_MODEL", "BAAI/bge-m3"
settings, "RAG_EMBEDDING_MODEL", "BAAI/bge-small-en-v1.5"
)
self.dimension = 1024 # bge-m3 produces 1024-d vectors
self.dimension = 384 # bge-small-en produces 384-d vectors
self.initialized = False
self.local_model = None
self.backend = "uninitialized"
@@ -139,7 +139,7 @@ class EmbeddingService:
def _generate_fallback_embedding(self, text: str) -> List[float]:
"""Generate a single fallback embedding"""
dimension = self.dimension or 1024
dimension = self.dimension or 384
# Use hash for reproducible random embeddings
np.random.seed(hash(text) % 2**32)
return np.random.random(dimension).tolist()