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

try few things

Browse source
This commit is contained in:
Vasilije 2024-05-25 17:35:36 +02:00
parent 423a7c2e20
commit ad8ba24e2a
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
cognee/infrastructure/InfrastructureConfig.py
View file

@ -11,10 +11,12 @@ from .data.chunking.DefaultChunkEngine import DefaultChunkEngine
from ..shared.data_models import GraphDBType, DefaultContentPrediction, KnowledgeGraph, SummarizedContent, \ from ..shared.data_models import GraphDBType, DefaultContentPrediction, KnowledgeGraph, SummarizedContent, \
LabeledContent, DefaultCognitiveLayer LabeledContent, DefaultCognitiveLayer
config = Config()
config.load()
class InfrastructureConfig(): class InfrastructureConfig():
config = Config()
config.load()
logging.info("cf path: %s", config.db_path)
system_root_directory: str = config.system_root_directory system_root_directory: str = config.system_root_directory
data_root_directory: str = config.data_root_directory data_root_directory: str = config.data_root_directory
llm_provider: str = config.llm_provider llm_provider: str = config.llm_provider
@ -43,6 +45,9 @@ class InfrastructureConfig():
llm_api_key: str = None llm_api_key: str = None
def get_config(self, config_entity: str = None) -> dict: def get_config(self, config_entity: str = None) -> dict:
config = Config()
config.load()
logging.info("cf path: %s", config.db_path)
if (config_entity is None or config_entity == "database_engine") and self.database_engine is None: if (config_entity is None or config_entity == "database_engine") and self.database_engine is None: