Update graph_rag.py

graph_rag.py

@@ -72,18 +72,6 @@ def clean_response(response):
     else:
         # If the search string is not found, return the original response
         return response
-
-def remove_data_reports(response):
-    # Regular expression pattern to find "[Data: Reports (XXX)]" and similar patterns
-    pattern = r'\[Data: Reports \(\d+\)\]'
-    
-    # Substitute the pattern with an empty string
-    cleaned_response = re.sub(pattern, '', response)
-    
-    # Remove extra spaces that might result from the substitution
-    cleaned_response = re.sub(r'\s+', ' ', cleaned_response).strip()
-    
-    return cleaned_response
         
 def qa_tool_graph_rag(user_question):
     original_dir = os.getcwd()  # Store the original directory
@@ -103,8 +91,6 @@ def qa_tool_graph_rag(user_question):
         
         # Clean the response to remove everything before "SUCCESS: Global Search Response:"
         answer = clean_response(raw_answer)
-
-        answer = remove_data_reports(answer)
         
         logger.info(f"GraphRAG answer generated: {answer}")