Update agents/crew.py
Browse files- agents/crew.py +6 -9
agents/crew.py
CHANGED
|
@@ -7,8 +7,6 @@ import os
|
|
| 7 |
from agents.models.llms import (
|
| 8 |
LLM_CREW_MANAGER,
|
| 9 |
LLM_CREW_PLANNING,
|
| 10 |
-
LLM_CREW_FUNCTION_CALLING,
|
| 11 |
-
LLM_CREW_EMBEDDER,
|
| 12 |
LLM_MANAGER,
|
| 13 |
LLM_AGENT
|
| 14 |
)
|
|
@@ -26,18 +24,18 @@ from utils.utils import read_file_json, is_ext
|
|
| 26 |
|
| 27 |
# Configuration
|
| 28 |
|
| 29 |
-
MEMORY_CREW = True
|
| 30 |
PLANNING_CREW = True
|
|
|
|
| 31 |
VERBOSE_CREW = False
|
| 32 |
|
| 33 |
MAX_ITER_MANAGER = 3
|
| 34 |
-
MAX_REASONING_ATTEMPTS_MANAGER = 3
|
| 35 |
REASONING_MANAGER = True
|
|
|
|
| 36 |
VERBOSE_MANAGER = False
|
| 37 |
|
| 38 |
MAX_ITER_AGENT = 2
|
| 39 |
-
MAX_REASONING_ATTEMPTS_AGENT = 2
|
| 40 |
REASONING_AGENT = True
|
|
|
|
| 41 |
VERBOSE_AGENT = False
|
| 42 |
|
| 43 |
os.environ["CHROMA_OPENAI_API_KEY"] = os.getenv("CHROMA_OPENAI_API_KEY") # Memory
|
|
@@ -231,12 +229,11 @@ class GAIACrew():
|
|
| 231 |
return Crew(
|
| 232 |
agents=self.agents,
|
| 233 |
tasks=self.tasks,
|
| 234 |
-
|
|
|
|
| 235 |
planning_llm=LLM_CREW_PLANNING,
|
| 236 |
-
function_calling_llm=LLM_CREW_FUNCTION_CALLING,
|
| 237 |
-
embedder=LLM_CREW_EMBEDDER,
|
| 238 |
-
memory=MEMORY_CREW,
|
| 239 |
planning=PLANNING_CREW,
|
|
|
|
| 240 |
verbose=VERBOSE_CREW
|
| 241 |
)
|
| 242 |
|
|
|
|
| 7 |
from agents.models.llms import (
|
| 8 |
LLM_CREW_MANAGER,
|
| 9 |
LLM_CREW_PLANNING,
|
|
|
|
|
|
|
| 10 |
LLM_MANAGER,
|
| 11 |
LLM_AGENT
|
| 12 |
)
|
|
|
|
| 24 |
|
| 25 |
# Configuration
|
| 26 |
|
|
|
|
| 27 |
PLANNING_CREW = True
|
| 28 |
+
MEMORY_CREW = True
|
| 29 |
VERBOSE_CREW = False
|
| 30 |
|
| 31 |
MAX_ITER_MANAGER = 3
|
|
|
|
| 32 |
REASONING_MANAGER = True
|
| 33 |
+
MAX_REASONING_ATTEMPTS_MANAGER = 3
|
| 34 |
VERBOSE_MANAGER = False
|
| 35 |
|
| 36 |
MAX_ITER_AGENT = 2
|
|
|
|
| 37 |
REASONING_AGENT = True
|
| 38 |
+
MAX_REASONING_ATTEMPTS_AGENT = 2
|
| 39 |
VERBOSE_AGENT = False
|
| 40 |
|
| 41 |
os.environ["CHROMA_OPENAI_API_KEY"] = os.getenv("CHROMA_OPENAI_API_KEY") # Memory
|
|
|
|
| 229 |
return Crew(
|
| 230 |
agents=self.agents,
|
| 231 |
tasks=self.tasks,
|
| 232 |
+
process="hierarchical", # TODO
|
| 233 |
+
manager_llm=LLM_CREW_MANAGER, # TODO
|
| 234 |
planning_llm=LLM_CREW_PLANNING,
|
|
|
|
|
|
|
|
|
|
| 235 |
planning=PLANNING_CREW,
|
| 236 |
+
memory=MEMORY_CREW,
|
| 237 |
verbose=VERBOSE_CREW
|
| 238 |
)
|
| 239 |
|