forked from DataBassGit/AgentForge
-
Notifications
You must be signed in to change notification settings - Fork 0
/
salience.py
59 lines (44 loc) · 1.61 KB
/
salience.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
from Agents.execution_agent import ExecutionAgent
from Agents.task_creation_agent import TaskCreationAgent
from Agents.prioritization_agent import PrioritizationAgent
from Agents.salience_agent import SalienceAgent
from Agents.status_agent import StatusAgent
from Utilities.function_utils import Functions
from Utilities.storage_interface import StorageInterface
# Load Agents
storage = StorageInterface()
taskCreationAgent = TaskCreationAgent()
prioritizationAgent = PrioritizationAgent()
executionAgent = ExecutionAgent()
salienceAgent = SalienceAgent()
statusAgent = StatusAgent()
# Add a variable to set the mode
functions = Functions()
functions.set_auto_mode()
# Salience loop
while True:
# collection_list = storage.storage_utils.collection_list()
# print(f"\nList: {collection_list}")
#
# peek = storage.storage_utils.peek("results")['documents']
# print(f"\nPeak Results: {peek}")
#
# peek = storage.storage_utils.peek("tasks")['documents']
# print(f"\nPeak Tasks: {peek}")
#
# text = "As an AI tasked with developing"
# res = storage.storage_utils.query_db("results", text)['documents']
# print(f"\nres:{res}")
# # Create task list
# taskCreationAgent.run_task_creation_agent()
#
# # Prioritize task list
# prioritizationAgent.run_prioritization_agent()
collection_list = storage.storage_utils.collection_list()
print(f"\nList: {collection_list}")
# quit()
# Allow for feedback if auto mode is disabled
feedback = functions.check_auto_mode()
data = salienceAgent.run_salience_agent()
statusAgent.run_status_agent(data)
# quit()