bstraehle commited on
Commit
ecf5d02
·
verified ·
1 Parent(s): 38fa2b9

Update agents/crew.py

Browse files
Files changed (1) hide show
  1. 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
- manager_llm=LLM_CREW_MANAGER,
 
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