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

Updates to the configs

Browse source
This commit is contained in:
Vasilije 2024-05-26 09:49:39 +02:00
parent 1a135bfdf8
commit 41ef5cf194
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cognee/infrastructure/InfrastructureConfig.py
View file

@ -5,7 +5,7 @@ from cognee.config import Config
from .data.chunking.config import get_chunk_config from .data.chunking.config import get_chunk_config
from .databases.relational import DuckDBAdapter, DatabaseEngine from .databases.relational import DuckDBAdapter, DatabaseEngine
from .databases.vector.vector_db_interface import VectorDBInterface from .databases.vector.vector_db_interface import VectorDBInterface
from .databases.vector.embeddings.DefaultEmbeddingEngine import DefaultEmbeddingEngine # from .databases.vector.embeddings.DefaultEmbeddingEngine import DefaultEmbeddingEngine
from .llm.llm_interface import LLMInterface from .llm.llm_interface import LLMInterface
from .llm.get_llm_client import get_llm_client from .llm.get_llm_client import get_llm_client
from .files.storage import LocalStorage from .files.storage import LocalStorage
@ -81,8 +81,8 @@ class InfrastructureConfig():
if self.intra_layer_score_treshold is None: if self.intra_layer_score_treshold is None:
self.intra_layer_score_treshold = config.intra_layer_score_treshold self.intra_layer_score_treshold = config.intra_layer_score_treshold
if self.embedding_engine is None: # if self.embedding_engine is None:
self.embedding_engine = DefaultEmbeddingEngine() # self.embedding_engine = DefaultEmbeddingEngine()
if self.connect_documents is None: if self.connect_documents is None:
self.connect_documents = config.connect_documents self.connect_documents = config.connect_documents