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

added fix to embedding engine

Browse source
This commit is contained in:
vasilije 2025-08-30 10:38:46 +02:00
parent a3da74a01d
commit 377c0d3973
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

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