Hemang Thakur commited on
Commit
90633d0
·
1 Parent(s): eb560af

changed .env file path to writable directory in main.py file

Browse files
Files changed (1) hide show
  1. main.py +2 -0
main.py CHANGED
@@ -51,6 +51,7 @@ def initialize_components():
51
 
52
  from src.search.search_engine import SearchEngine
53
  from src.query_processing.query_processor import QueryProcessor
 
54
  from src.rag.graph_rag import GraphRAG
55
  from src.evaluation.evaluator import Evaluator
56
  from src.reasoning.reasoner import Reasoner
@@ -63,6 +64,7 @@ def initialize_components():
63
  SESSION_STORE['search_engine'] = SearchEngine()
64
  SESSION_STORE['query_processor'] = QueryProcessor()
65
  SESSION_STORE['crawler'] = CustomCrawler(max_concurrent_requests=1000)
 
66
  SESSION_STORE['graph_rag'] = GraphRAG(num_workers=os.cpu_count() * 2)
67
  SESSION_STORE['evaluator'] = Evaluator()
68
  SESSION_STORE['reasoner'] = Reasoner()
 
51
 
52
  from src.search.search_engine import SearchEngine
53
  from src.query_processing.query_processor import QueryProcessor
54
+ # from src.rag.neo4j_graphrag import Neo4jGraphRAG
55
  from src.rag.graph_rag import GraphRAG
56
  from src.evaluation.evaluator import Evaluator
57
  from src.reasoning.reasoner import Reasoner
 
64
  SESSION_STORE['search_engine'] = SearchEngine()
65
  SESSION_STORE['query_processor'] = QueryProcessor()
66
  SESSION_STORE['crawler'] = CustomCrawler(max_concurrent_requests=1000)
67
+ # SESSION_STORE['graph_rag'] = Neo4jGraphRAG(num_workers=os.cpu_count() * 2)
68
  SESSION_STORE['graph_rag'] = GraphRAG(num_workers=os.cpu_count() * 2)
69
  SESSION_STORE['evaluator'] = Evaluator()
70
  SESSION_STORE['reasoner'] = Reasoner()