update __init__

Added 
- Tasks 
- max_workers
pull/613/head
Sambhav Dixit 3 months ago committed by GitHub
parent f40fa15392
commit 8d7e07d046
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -112,6 +112,7 @@ class ConcurrentWorkflow(BaseSwarm):
return_str_on: bool = False, return_str_on: bool = False,
agent_responses: list = [], agent_responses: list = [],
auto_generate_prompts: bool = False, auto_generate_prompts: bool = False,
max_workers: int = None,
*args, *args,
**kwargs, **kwargs,
): ):
@ -132,9 +133,12 @@ class ConcurrentWorkflow(BaseSwarm):
self.return_str_on = return_str_on self.return_str_on = return_str_on
self.agent_responses = agent_responses self.agent_responses = agent_responses
self.auto_generate_prompts = auto_generate_prompts self.auto_generate_prompts = auto_generate_prompts
self.max_workers = max_workers or os.cpu_count()
self.tasks = [] # Initialize tasks list
self.reliability_check() self.reliability_check()
def reliability_check(self): def reliability_check(self):
try: try:
logger.info("Starting reliability checks") logger.info("Starting reliability checks")

Loading…
Cancel
Save