gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added fix to embeddings format

Browse source
This commit is contained in:
vasilije 2025-09-01 17:17:48 +02:00
parent 7b21170e8e
commit 940d4797bc
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cognee/infrastructure/databases/vector/embeddings/get_embedding_engine.py
View file

@ -33,7 +33,7 @@ def get_embedding_engine() -> EmbeddingEngine:
config.embedding_api_version, config.embedding_api_version,
config.huggingface_tokenizer, config.huggingface_tokenizer,
llm_config.llm_api_key, llm_config.llm_api_key,
llm_config.llm_provider llm_config.llm_provider,
) )
@ -48,7 +48,7 @@ def create_embedding_engine(
embedding_api_version, embedding_api_version,
huggingface_tokenizer, huggingface_tokenizer,
llm_api_key, llm_api_key,
llm_provider llm_provider,
): ):
""" """
Create and return an embedding engine based on the specified provider. Create and return an embedding engine based on the specified provider.
@ -101,7 +101,8 @@ def create_embedding_engine(
return LiteLLMEmbeddingEngine( return LiteLLMEmbeddingEngine(
provider=embedding_provider, 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, endpoint=embedding_endpoint,
api_version=embedding_api_version, api_version=embedding_api_version,
model=embedding_model, model=embedding_model,