Skip to content

Commit 8b3fe9a

Browse files
fix: correct callback signature mismatch in multi-agent startup (fixes #1730)
- Fix composed_on_task_start to accept both task and task_id parameters - Fix composed_on_task_complete to accept both task and task_output parameters - Ensures callbacks receive the correct number of arguments as expected - Resolves AgentTeam startup error with function signature mismatch Co-authored-by: praisonai-triage-agent[bot] <praisonai-triage-agent[bot]@users.noreply.github.com>
1 parent 5ff79d3 commit 8b3fe9a

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

  • src/praisonai-agents/praisonaiagents/agents

src/praisonai-agents/praisonaiagents/agents/agents.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1609,25 +1609,25 @@ def verbose_task_complete_callback(task, task_output):
16091609
original_on_task_start = self.on_task_start
16101610
original_on_task_complete = self.on_task_complete
16111611

1612-
def composed_on_task_start(task):
1612+
def composed_on_task_start(task, task_id):
16131613
try:
16141614
verbose_task_start_callback(task)
16151615
except Exception as e:
16161616
logging.debug(f"Error in verbose task start callback: {e}")
16171617
if original_on_task_start:
16181618
try:
1619-
original_on_task_start(task)
1619+
original_on_task_start(task, task_id)
16201620
except Exception as e:
16211621
logging.debug(f"Error in original task start callback: {e}")
16221622

1623-
def composed_on_task_complete(task):
1623+
def composed_on_task_complete(task, task_output):
16241624
try:
16251625
verbose_task_complete_callback(task)
16261626
except Exception as e:
16271627
logging.debug(f"Error in verbose task complete callback: {e}")
16281628
if original_on_task_complete:
16291629
try:
1630-
original_on_task_complete(task)
1630+
original_on_task_complete(task, task_output)
16311631
except Exception as e:
16321632
logging.debug(f"Error in original task complete callback: {e}")
16331633

0 commit comments

Comments
 (0)