Merge pull request #421 from topoteretes/add-set-graph-database

feat: Add ability to change graph database configuration through cognee

cognee/api/v1/config/config.py

@@ -131,6 +131,18 @@ def set_relational_db_config(config_dict: dict):
                     message=f"'{key}' is not a valid attribute of the config."
                 )
 
+    @staticmethod
+    def set_graph_db_config(config_dict: dict) -> None:
+        """
+        Updates the graph db config with values from config_dict.
+        """
+        graph_db_config = get_graph_config()
+        for key, value in config_dict.items():
+            if hasattr(graph_db_config, key):
+                object.__setattr__(graph_db_config, key, value)
+            else:
+                raise AttributeError(message=f"'{key}' is not a valid attribute of the config.")
+
     @staticmethod
     def set_vector_db_config(config_dict: dict):
         """