From 75a8d3414e687cdb1eeb3b0c0cc759e9bff2ae71 Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Sun, 26 May 2024 14:42:38 +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 176c1fec8..8a863b05d 100644
--- a/cognee/modules/cognify/config.py
+++ b/cognee/modules/cognify/config.py
@@ -35,7 +35,8 @@ class CognifyConfig(BaseSettings):
             "intra_layer_score_treshold": self.intra_layer_score_treshold,
             "connect_documents": self.connect_documents,
             "graph_topology": self.graph_topology,
-            "cognitive_layers_limit": self.cognitive_layers_limit
+            "cognitive_layers_limit": self.cognitive_layers_limit,
+            "graph_model": self.graph_model
         }
 
 @lru_cache