From 377c0d3973a760dc9163d4f6a0a6bbfeccb7dbef Mon Sep 17 00:00:00 2001 From: vasilije Date: Sat, 30 Aug 2025 10:38:46 +0200 Subject: [PATCH] added fix to embedding engine --- .../databases/vector/embeddings/get_embedding_engine.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cognee/infrastructure/databases/vector/embeddings/get_embedding_engine.py b/cognee/infrastructure/databases/vector/embeddings/get_embedding_engine.py index ae15b6c6e..192f1958c 100644 --- a/cognee/infrastructure/databases/vector/embeddings/get_embedding_engine.py +++ b/cognee/infrastructure/databases/vector/embeddings/get_embedding_engine.py @@ -33,6 +33,7 @@ def get_embedding_engine() -> EmbeddingEngine: config.embedding_api_version, config.huggingface_tokenizer, llm_config.llm_api_key, + llm_config.llm_provider ) @@ -47,6 +48,7 @@ def create_embedding_engine( embedding_api_version, huggingface_tokenizer, llm_api_key, + llm_provider ): """ Create and return an embedding engine based on the specified provider. @@ -99,7 +101,7 @@ def create_embedding_engine( return LiteLLMEmbeddingEngine( provider=embedding_provider, - api_key=embedding_api_key or llm_api_key, + api_key=embedding_api_key or (embedding_api_key if llm_provider == 'custom' else llm_api_key), endpoint=embedding_endpoint, api_version=embedding_api_version, model=embedding_model,