bstraehle commited on
Commit
6e6ea30
·
verified ·
1 Parent(s): 0c0a0d1

Update crew.py

Browse files
Files changed (1) hide show
  1. crew.py +24 -26
crew.py CHANGED
@@ -316,8 +316,8 @@ class GAIACrew():
316
  except Exception as e:
317
  raise RuntimeError(f"Processing failed: {str(e)}")
318
 
319
- @agent
320
- def web_search_agent(self) -> Agent:
321
  return Agent(
322
  allow_delegation=False,
323
  llm=AGENT_MODEL,
@@ -326,8 +326,8 @@ class GAIACrew():
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,
@@ -336,8 +336,8 @@ class GAIACrew():
336
  verbose=True
337
  )
338
 
339
- @agent
340
- def audio_analysis_agent(self) -> Agent:
341
  return Agent(
342
  allow_delegation=False,
343
  llm=AGENT_MODEL,
@@ -346,8 +346,8 @@ class GAIACrew():
346
  verbose=True
347
  )
348
 
349
- @agent
350
- def video_analysis_agent(self) -> Agent:
351
  return Agent(
352
  allow_delegation=False,
353
  llm=AGENT_MODEL,
@@ -356,8 +356,8 @@ class GAIACrew():
356
  verbose=True
357
  )
358
 
359
- @agent
360
- def youtube_analysis_agent(self) -> Agent:
361
  return Agent(
362
  allow_delegation=False,
363
  llm=AGENT_MODEL,
@@ -366,8 +366,8 @@ class GAIACrew():
366
  verbose=True
367
  )
368
 
369
- @agent
370
- def document_analysis_agent(self) -> Agent:
371
  return Agent(
372
  allow_delegation=False,
373
  llm=AGENT_MODEL,
@@ -376,10 +376,9 @@ class GAIACrew():
376
  verbose=True
377
  )
378
 
379
- @agent
380
- def arithmetic_agent(self) -> Agent:
381
  return Agent(
382
- arithmetic_agent = Agent(
383
  allow_delegation=False,
384
  llm=AGENT_MODEL,
385
  max_iter=2,
@@ -387,8 +386,8 @@ class GAIACrew():
387
  verbose=True
388
  )
389
 
390
- @agent
391
- def code_execution_agent(self) -> Agent:
392
  return Agent(
393
  allow_delegation=False,
394
  llm=AGENT_MODEL,
@@ -397,8 +396,8 @@ class GAIACrew():
397
  verbose=True
398
  )
399
 
400
- @agent
401
- def code_generation_agent(self) -> Agent:
402
  return Agent(
403
  allow_delegation=False,
404
  llm=AGENT_MODEL,
@@ -407,24 +406,23 @@ class GAIACrew():
407
  verbose=True
408
  )
409
 
410
- @agent
411
- def image_analysis_agent(self) -> Agent:
412
  return Agent(
413
- manager_agent = Agent(
414
  allow_delegation=True,
415
  llm=MANAGER_MODEL,
416
  max_iter=5,
417
  verbose=True
418
  )
419
 
420
- @task
421
- def manager_task(self) -> Task:
422
  return Task(
423
  config=self.tasks_config['manager_task']
424
  )
425
 
426
- @crew
427
- def crew(self) -> Crew:
428
  return Crew(
429
  agents=self.agents,
430
  tasks=self.tasks,
 
316
  except Exception as e:
317
  raise RuntimeError(f"Processing failed: {str(e)}")
318
 
319
+ @agent
320
+ def web_search_agent(self) -> Agent:
321
  return Agent(
322
  allow_delegation=False,
323
  llm=AGENT_MODEL,
 
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,
 
336
  verbose=True
337
  )
338
 
339
+ @agent
340
+ def audio_analysis_agent(self) -> Agent:
341
  return Agent(
342
  allow_delegation=False,
343
  llm=AGENT_MODEL,
 
346
  verbose=True
347
  )
348
 
349
+ @agent
350
+ def video_analysis_agent(self) -> Agent:
351
  return Agent(
352
  allow_delegation=False,
353
  llm=AGENT_MODEL,
 
356
  verbose=True
357
  )
358
 
359
+ @agent
360
+ def youtube_analysis_agent(self) -> Agent:
361
  return Agent(
362
  allow_delegation=False,
363
  llm=AGENT_MODEL,
 
366
  verbose=True
367
  )
368
 
369
+ @agent
370
+ def document_analysis_agent(self) -> Agent:
371
  return Agent(
372
  allow_delegation=False,
373
  llm=AGENT_MODEL,
 
376
  verbose=True
377
  )
378
 
379
+ @agent
380
+ def arithmetic_agent(self) -> Agent:
381
  return Agent(
 
382
  allow_delegation=False,
383
  llm=AGENT_MODEL,
384
  max_iter=2,
 
386
  verbose=True
387
  )
388
 
389
+ @agent
390
+ def code_execution_agent(self) -> Agent:
391
  return Agent(
392
  allow_delegation=False,
393
  llm=AGENT_MODEL,
 
396
  verbose=True
397
  )
398
 
399
+ @agent
400
+ def code_generation_agent(self) -> Agent:
401
  return Agent(
402
  allow_delegation=False,
403
  llm=AGENT_MODEL,
 
406
  verbose=True
407
  )
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,
415
  verbose=True
416
  )
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:
426
  return Crew(
427
  agents=self.agents,
428
  tasks=self.tasks,