Spaces:
Sleeping
Sleeping
Vaishak G Kumar
commited on
Update src/teams/agentteam.py
Browse files- 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
|
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()
|