from langchain_ollama import OllamaEmbeddings
from src.config import settings

# Instantiate the Ollama embeddings once for reuse
ollama_embeddings = OllamaEmbeddings(
    model=settings.ollama_embed_model,
    base_url=f"{settings.ollama_host}:{settings.ollama_port}"
)

def get_embedding(text: str):
    """
    Return the embedding vector for a single text string.
    """
    return ollama_embeddings.embed_query(text)