vaishakgkumar commited on
Commit
8ee8919
1 Parent(s): 363a6c9

Update main.py

Browse files

Added the json function and import

Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -8,6 +8,7 @@ from src.mapper.scimap import scimap
8
  from src.mapper.parser import MapperParser
9
  from src.datatonic.dataloader import DataLoader
10
  from src.teams.agentteam import codingteam, covid19team, financeteam, debateteam, homeworkteam, consultingteam
 
11
 
12
  title = """# Welcome to 👩🏻‍🔬🧪SciTonic
13
  this is a highly adaptive technical operator that will listen to your query and load datasets and multi-agent teams based on those. Simply describe your problem in detail, ask a question and provide a reasoning method to get started:
@@ -90,7 +91,8 @@ def process_query(oai_key, query, max_auto_reply):
90
  dataset = data_loader.load_and_process(task.lower())
91
 
92
  # Save dataset to a JSON file and get the file path
93
- json_file_path = "./src/datatonic" # Define the JSON file path
 
94
  data_loader.save_to_json(dataset, json_file_path)
95
 
96
  # Initialize AgentsFactory with the path to the JSON file
 
8
  from src.mapper.parser import MapperParser
9
  from src.datatonic.dataloader import DataLoader
10
  from src.teams.agentteam import codingteam, covid19team, financeteam, debateteam, homeworkteam, consultingteam
11
+ from src.agentics.agents import AgentsFactory
12
 
13
  title = """# Welcome to 👩🏻‍🔬🧪SciTonic
14
  this is a highly adaptive technical operator that will listen to your query and load datasets and multi-agent teams based on those. Simply describe your problem in detail, ask a question and provide a reasoning method to get started:
 
91
  dataset = data_loader.load_and_process(task.lower())
92
 
93
  # Save dataset to a JSON file and get the file path
94
+ json_file_name = "dataset.json" # Provide a suitable file name
95
+ json_file_path = os.path.join("./src/datatonic/", json_file_name) # Define the complete file path
96
  data_loader.save_to_json(dataset, json_file_path)
97
 
98
  # Initialize AgentsFactory with the path to the JSON file