bstraehle commited on
Commit
764a024
·
verified ·
1 Parent(s): 6e6ea30

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +14 -14
crew.py CHANGED
@@ -318,17 +318,17 @@ class GAIACrew():
318
 
319
  @agent
320
  def web_search_agent(self) -> Agent:
321
- return Agent(
322
  allow_delegation=False,
323
  llm=AGENT_MODEL,
324
  max_iter=2,
325
  tools=[web_search_tool],
326
  verbose=True
327
- )
328
 
329
  @agent
330
  def image_analysis_agent(self) -> Agent:
331
- return Agent(
332
  allow_delegation=False,
333
  llm=AGENT_MODEL,
334
  max_iter=2,
@@ -338,7 +338,7 @@ class GAIACrew():
338
 
339
  @agent
340
  def audio_analysis_agent(self) -> Agent:
341
- return Agent(
342
  allow_delegation=False,
343
  llm=AGENT_MODEL,
344
  max_iter=2,
@@ -348,7 +348,7 @@ class GAIACrew():
348
 
349
  @agent
350
  def video_analysis_agent(self) -> Agent:
351
- return Agent(
352
  allow_delegation=False,
353
  llm=AGENT_MODEL,
354
  max_iter=2,
@@ -358,7 +358,7 @@ class GAIACrew():
358
 
359
  @agent
360
  def youtube_analysis_agent(self) -> Agent:
361
- return Agent(
362
  allow_delegation=False,
363
  llm=AGENT_MODEL,
364
  max_iter=2,
@@ -368,7 +368,7 @@ class GAIACrew():
368
 
369
  @agent
370
  def document_analysis_agent(self) -> Agent:
371
- return Agent(
372
  allow_delegation=False,
373
  llm=AGENT_MODEL,
374
  max_iter=2,
@@ -378,7 +378,7 @@ class GAIACrew():
378
 
379
  @agent
380
  def arithmetic_agent(self) -> Agent:
381
- return Agent(
382
  allow_delegation=False,
383
  llm=AGENT_MODEL,
384
  max_iter=2,
@@ -388,7 +388,7 @@ class GAIACrew():
388
 
389
  @agent
390
  def code_execution_agent(self) -> Agent:
391
- return Agent(
392
  allow_delegation=False,
393
  llm=AGENT_MODEL,
394
  max_iter=3,
@@ -398,7 +398,7 @@ class GAIACrew():
398
 
399
  @agent
400
  def code_generation_agent(self) -> Agent:
401
- return Agent(
402
  allow_delegation=False,
403
  llm=AGENT_MODEL,
404
  max_iter=3,
@@ -408,7 +408,7 @@ class GAIACrew():
408
 
409
  @agent
410
  def image_analysis_agent(self) -> Agent:
411
- return Agent(
412
  allow_delegation=True,
413
  llm=MANAGER_MODEL,
414
  max_iter=5,
@@ -417,9 +417,9 @@ class GAIACrew():
417
 
418
  @task
419
  def manager_task(self) -> Task:
420
- return Task(
421
- config=self.tasks_config['manager_task']
422
- )
423
 
424
  @crew
425
  def crew(self) -> Crew:
 
318
 
319
  @agent
320
  def web_search_agent(self) -> Agent:
321
+ return Agent(
322
  allow_delegation=False,
323
  llm=AGENT_MODEL,
324
  max_iter=2,
325
  tools=[web_search_tool],
326
  verbose=True
327
+ )
328
 
329
  @agent
330
  def image_analysis_agent(self) -> Agent:
331
+ return Agent(
332
  allow_delegation=False,
333
  llm=AGENT_MODEL,
334
  max_iter=2,
 
338
 
339
  @agent
340
  def audio_analysis_agent(self) -> Agent:
341
+ return Agent(
342
  allow_delegation=False,
343
  llm=AGENT_MODEL,
344
  max_iter=2,
 
348
 
349
  @agent
350
  def video_analysis_agent(self) -> Agent:
351
+ return Agent(
352
  allow_delegation=False,
353
  llm=AGENT_MODEL,
354
  max_iter=2,
 
358
 
359
  @agent
360
  def youtube_analysis_agent(self) -> Agent:
361
+ return Agent(
362
  allow_delegation=False,
363
  llm=AGENT_MODEL,
364
  max_iter=2,
 
368
 
369
  @agent
370
  def document_analysis_agent(self) -> Agent:
371
+ return Agent(
372
  allow_delegation=False,
373
  llm=AGENT_MODEL,
374
  max_iter=2,
 
378
 
379
  @agent
380
  def arithmetic_agent(self) -> Agent:
381
+ return Agent(
382
  allow_delegation=False,
383
  llm=AGENT_MODEL,
384
  max_iter=2,
 
388
 
389
  @agent
390
  def code_execution_agent(self) -> Agent:
391
+ return Agent(
392
  allow_delegation=False,
393
  llm=AGENT_MODEL,
394
  max_iter=3,
 
398
 
399
  @agent
400
  def code_generation_agent(self) -> Agent:
401
+ return Agent(
402
  allow_delegation=False,
403
  llm=AGENT_MODEL,
404
  max_iter=3,
 
408
 
409
  @agent
410
  def image_analysis_agent(self) -> Agent:
411
+ return Agent(
412
  allow_delegation=True,
413
  llm=MANAGER_MODEL,
414
  max_iter=5,
 
417
 
418
  @task
419
  def manager_task(self) -> Task:
420
+ return Task(
421
+ config=self.tasks_config['manager_task']
422
+ )
423
 
424
  @crew
425
  def crew(self) -> Crew: