Update context_graph.py

context_graph.py

@@ -25,6 +25,7 @@ class ContextGraph:
         # Save the updated graph
         with open(self.file_path, "w") as f:
             json.dump(graph, f, indent=2)
+
         os.sync()  # Force Hugging Face to write to disk
 
     def query_context(self, agent_id, keyword):
@@ -38,21 +39,10 @@ class ContextGraph:
             if keyword.lower() in k.lower() or keyword.lower() in v.lower()
         ]
 
-        return results if results else ["No matching context found."]
-                    def query_context(self, agent_id, keyword):
-        # Search for any stored memory entries that contain the keyword
-        with open(self.file_path, "r") as f:
-            graph = json.load(f)
-
-        agent_data = graph["context_links"].get(agent_id, {})
-        results = [
-            f"{k}: {v}" for k, v in agent_data.items()
-            if keyword.lower() in k.lower() or keyword.lower() in v.lower()
-        ]
         return results if results else ["No matching context found."]
 
     def get_context(self, agent_id):
         # Simple alias to maintain backward compatibility with AgentCore
         with open(self.file_path, "r") as f:
             graph = json.load(f)
-        return graph["context_links"].get(agent_id, {}) {})
+        return graph["context_links"].get(agent_id, {})