From 8cd11696d435e7e49b1a8282f0db725387dbaa85 Mon Sep 17 00:00:00 2001 From: Sarah Wooders Date: Tue, 17 Dec 2024 15:21:33 -0800 Subject: [PATCH] fix: fix ollama embeddings (#2264) --- letta/embeddings.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/letta/embeddings.py b/letta/embeddings.py index 5b521032..0d82d158 100644 --- a/letta/embeddings.py +++ b/letta/embeddings.py @@ -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