ailine/backend/rag_core/embedders.py

"""
嵌入模型包装器 - 直接使用统一嵌入服务
"""

import logging
from typing import List, Optional

from langchain_core.embeddings import Embeddings

logger = logging.getLogger(__name__)


def get_embeddings() -> Embeddings:
    """
    获取统一的嵌入服务实例。
    
    Returns:
        LangChain 兼容的 Embeddings 实例
    """
    from backend.app.model_services import get_embedding_service
    return get_embedding_service()


def get_embedding_dimension(embeddings: Optional[Embeddings] = None) -> int:
    """
    获取嵌入维度。
    
    Args:
        embeddings: 可选的嵌入实例，如果不提供则自动获取
    
    Returns:
        嵌入维度大小
    """
    if embeddings is None:
        embeddings = get_embeddings()
    test_embedding = embeddings.embed_query("test")
    return len(test_embedding)