Create context_graph.py

context_graph.py

@@ -2,6 +2,11 @@ import json
 import os
 
 class ContextGraph:
+    """
+    Stores relationships between prompts and responses.
+    Think of it as a 'map' of what the agent has learned.
+    """
+
     def __init__(self, file_path="context_graph.json"):
         self.file_path = file_path
         self._init_graph()
@@ -19,6 +24,7 @@ class ContextGraph:
         graph["links"][agent_id][key] = value
         with open(self.file_path, "w") as f:
             json.dump(graph, f, indent=2)
+        print(f"[ContextGraph] Linked ({key} → {value}) for {agent_id}")
 
     def get_context(self, agent_id):
         with open(self.file_path, "r") as f: