Update graph_rag.py

graph_rag.py

@@ -27,20 +27,29 @@ def unzip_folder(zip_path, extract_path):
     return output_dir
 
 def run_graphrag_query(query, ragtest_dir):
+    # Check and log the contents of the ragtest_dir
+    logger.info(f"Contents of {ragtest_dir}:")
+    for file in os.listdir(ragtest_dir):
+        logger.info(os.path.join(ragtest_dir, file))
+
+    # Define the command
     command = [
         "python", "-m", "graphrag.query",
         "--root", ragtest_dir,
-        "--method", "local",
+        "--method", "local",  # or "global"
         query
     ]
     
+    # Run the command
     result = subprocess.run(command, capture_output=True, text=True)
     
+    # Return the output or error message
     if result.returncode == 0:
         return result.stdout
     else:
         return result.stderr
 
+
 def qa_tool_graph_rag(user_question):
     original_dir = os.getcwd()  # Store the original directory
     try: