From 940d4797bc0d56ef855b85ce76dd8758cfc24135 Mon Sep 17 00:00:00 2001 From: vasilije Date: Mon, 1 Sep 2025 17:17:48 +0200 Subject: [PATCH] added fix to embeddings format --- .../databases/vector/embeddings/get_embedding_engine.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/cognee/infrastructure/databases/vector/embeddings/get_embedding_engine.py b/cognee/infrastructure/databases/vector/embeddings/get_embedding_engine.py index 192f1958c..e7fcf4e94 100644 --- a/cognee/infrastructure/databases/vector/embeddings/get_embedding_engine.py +++ b/cognee/infrastructure/databases/vector/embeddings/get_embedding_engine.py @@ -33,7 +33,7 @@ def get_embedding_engine() -> EmbeddingEngine: config.embedding_api_version, config.huggingface_tokenizer, llm_config.llm_api_key, - llm_config.llm_provider + llm_config.llm_provider, ) @@ -48,7 +48,7 @@ def create_embedding_engine( embedding_api_version, huggingface_tokenizer, llm_api_key, - llm_provider + llm_provider, ): """ Create and return an embedding engine based on the specified provider. @@ -101,7 +101,8 @@ def create_embedding_engine( return LiteLLMEmbeddingEngine( provider=embedding_provider, - api_key=embedding_api_key or (embedding_api_key if llm_provider == 'custom' else 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,