Hemang Thakur commited on
Commit
981afc2
·
1 Parent(s): e948f71

made changes to graph rag file and some ui elements

Browse files
Files changed (1) hide show
  1. src/rag/graph_rag.py +6 -0
src/rag/graph_rag.py CHANGED
@@ -1302,11 +1302,17 @@ class GraphRAG:
1302
  net.options["layout"] = {"improvedLayout": True}
1303
  net.options["interaction"] = {"dragNodes": True}
1304
 
 
 
 
1305
  net.save_graph("temp_graph.html")
1306
 
1307
  with open("temp_graph.html", "r", encoding="utf-8") as f:
1308
  html_str = f.read()
 
1309
  os.remove("temp_graph.html")
 
 
1310
  return html_str
1311
 
1312
  def verify_graph_integrity(self):
 
1302
  net.options["layout"] = {"improvedLayout": True}
1303
  net.options["interaction"] = {"dragNodes": True}
1304
 
1305
+ original_dir = os.getcwd()
1306
+ os.chdir(os.getenv("WRITABLE_DIR", "/tmp"))
1307
+
1308
  net.save_graph("temp_graph.html")
1309
 
1310
  with open("temp_graph.html", "r", encoding="utf-8") as f:
1311
  html_str = f.read()
1312
+
1313
  os.remove("temp_graph.html")
1314
+ os.chdir(original_dir)
1315
+
1316
  return html_str
1317
 
1318
  def verify_graph_integrity(self):