14 lines
422 B
Python
14 lines
422 B
Python
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) |