Spaces:
Paused
Paused
Update crew.py
Browse files
crew.py
CHANGED
|
@@ -316,8 +316,8 @@ class GAIACrew():
|
|
| 316 |
except Exception as e:
|
| 317 |
raise RuntimeError(f"Processing failed: {str(e)}")
|
| 318 |
|
| 319 |
-
|
| 320 |
-
|
| 321 |
return Agent(
|
| 322 |
allow_delegation=False,
|
| 323 |
llm=AGENT_MODEL,
|
|
@@ -326,8 +326,8 @@ class GAIACrew():
|
|
| 326 |
verbose=True
|
| 327 |
)
|
| 328 |
|
| 329 |
-
|
| 330 |
-
|
| 331 |
return Agent(
|
| 332 |
allow_delegation=False,
|
| 333 |
llm=AGENT_MODEL,
|
|
@@ -336,8 +336,8 @@ class GAIACrew():
|
|
| 336 |
verbose=True
|
| 337 |
)
|
| 338 |
|
| 339 |
-
|
| 340 |
-
|
| 341 |
return Agent(
|
| 342 |
allow_delegation=False,
|
| 343 |
llm=AGENT_MODEL,
|
|
@@ -346,8 +346,8 @@ class GAIACrew():
|
|
| 346 |
verbose=True
|
| 347 |
)
|
| 348 |
|
| 349 |
-
|
| 350 |
-
|
| 351 |
return Agent(
|
| 352 |
allow_delegation=False,
|
| 353 |
llm=AGENT_MODEL,
|
|
@@ -356,8 +356,8 @@ class GAIACrew():
|
|
| 356 |
verbose=True
|
| 357 |
)
|
| 358 |
|
| 359 |
-
|
| 360 |
-
|
| 361 |
return Agent(
|
| 362 |
allow_delegation=False,
|
| 363 |
llm=AGENT_MODEL,
|
|
@@ -366,8 +366,8 @@ class GAIACrew():
|
|
| 366 |
verbose=True
|
| 367 |
)
|
| 368 |
|
| 369 |
-
|
| 370 |
-
|
| 371 |
return Agent(
|
| 372 |
allow_delegation=False,
|
| 373 |
llm=AGENT_MODEL,
|
|
@@ -376,10 +376,9 @@ class GAIACrew():
|
|
| 376 |
verbose=True
|
| 377 |
)
|
| 378 |
|
| 379 |
-
|
| 380 |
-
|
| 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 |
-
|
| 391 |
-
|
| 392 |
return Agent(
|
| 393 |
allow_delegation=False,
|
| 394 |
llm=AGENT_MODEL,
|
|
@@ -397,8 +396,8 @@ class GAIACrew():
|
|
| 397 |
verbose=True
|
| 398 |
)
|
| 399 |
|
| 400 |
-
|
| 401 |
-
|
| 402 |
return Agent(
|
| 403 |
allow_delegation=False,
|
| 404 |
llm=AGENT_MODEL,
|
|
@@ -407,24 +406,23 @@ class GAIACrew():
|
|
| 407 |
verbose=True
|
| 408 |
)
|
| 409 |
|
| 410 |
-
|
| 411 |
-
|
| 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 |
-
|
| 421 |
-
|
| 422 |
return Task(
|
| 423 |
config=self.tasks_config['manager_task']
|
| 424 |
)
|
| 425 |
|
| 426 |
-
|
| 427 |
-
|
| 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,
|