fix: fix ollama embeddings (#2264)

This commit is contained in:
Sarah Wooders
2024-12-17 15:21:33 -08:00
committed by GitHub
parent 91995ae6ff
commit 8cd11696d4

View File

@@ -234,16 +234,10 @@ def embedding_model(config: EmbeddingConfig, user_id: Optional[uuid.UUID] = None
)
elif endpoint_type == "ollama":
from llama_index.embeddings.ollama import OllamaEmbedding
ollama_additional_kwargs = {}
callback_manager = None
model = OllamaEmbedding(
model_name=config.embedding_model,
model = OllamaEmbeddings(
model=config.embedding_model,
base_url=config.embedding_endpoint,
ollama_additional_kwargs=ollama_additional_kwargs or {},
callback_manager=callback_manager or None,
ollama_additional_kwargs={},
)
return model