Vaishak G Kumar commited on
Commit
76d73a3
·
verified ·
1 Parent(s): 71ab0ea

Update src/teams/agentteam.py

Browse files
Files changed (1) hide show
  1. src/teams/agentteam.py +13 -1
src/teams/agentteam.py CHANGED
@@ -1,9 +1,21 @@
1
  import autogen
2
  from src.agentics.agents import AgentsFactory
3
- from main import boss_aid
4
  # agents_factory = AgentsFactory()
5
 
 
 
6
 
 
 
 
 
 
 
 
 
 
 
7
  # Function to reset agents
8
  def _reset_agents():
9
  boss_aid.reset()
 
1
  import autogen
2
  from src.agentics.agents import AgentsFactory
3
+ from src.datatonic.dataloader import DataLoader
4
  # agents_factory = AgentsFactory()
5
 
6
+ data_loader = DataLoader()
7
+ dataset = data_loader.load_and_process(task.lower())
8
 
9
+ # Save dataset to a JSON file and get the file path
10
+ json_file_name = "dataset.json" # Provide a suitable file name
11
+ json_file_path = os.path.join("./src/datatonic/", json_file_name) # Define the complete file path
12
+ data_loader.save_to_json(dataset, json_file_path)
13
+
14
+ # Initialize AgentsFactory with the path to the JSON file
15
+ agents_factory = AgentsFactory(llm_config, json_file_path)
16
+
17
+ # Retrieve the Boss Assistant agent
18
+ boss_aid = agents_factory.scitonic()
19
  # Function to reset agents
20
  def _reset_agents():
21
  boss_aid.reset()