From 49fb408187e48ef6d27fd3ccf8c9058e20b4073e Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Sun, 26 May 2024 14:41:49 +0200
Subject: [PATCH] Updates to the configs

---
 cognee/modules/cognify/config.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/cognee/modules/cognify/config.py b/cognee/modules/cognify/config.py
index d0470cdd3..176c1fec8 100644
--- a/cognee/modules/cognify/config.py
+++ b/cognee/modules/cognify/config.py
@@ -4,7 +4,7 @@ from pydantic_settings import BaseSettings, SettingsConfigDict
 from cognee.root_dir import get_absolute_path
 
 from cognee.shared.data_models import MonitoringTool, DefaultContentPrediction, LabeledContent, SummarizedContent, \
-    DefaultCognitiveLayer, DefaultGraphModel
+    DefaultCognitiveLayer, DefaultGraphModel, KnowledgeGraph
 
 
 # Monitoring tool
@@ -20,6 +20,7 @@ class CognifyConfig(BaseSettings):
     connect_documents: bool = False
     graph_topology: object = DefaultGraphModel
     cognitive_layers_limit: int = 2
+    graph_model:object = KnowledgeGraph