fix: fix ollama embeddings (#2264)
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user