@@ -254,6 +254,9 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
254
254
Start by validating the workflow to see its current state.""" ,
255
255
model = WORKFLOW_MODEL_NAME ,
256
256
tools = [run_workflow , analyze_error_type , save_current_workflow ],
257
+ config = {
258
+ "max_tokens" : 8192
259
+ }
257
260
)
258
261
259
262
workflow_bugfix_default_agent = create_agent (
@@ -300,6 +303,9 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
300
303
""" ,
301
304
tools = [get_current_workflow , get_node_info , update_workflow ],
302
305
handoffs = [agent ],
306
+ config = {
307
+ "max_tokens" : 8192
308
+ }
303
309
)
304
310
305
311
link_agent = create_agent (
@@ -388,6 +394,9 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
388
394
tools = [analyze_missing_connections , apply_connection_fixes ,
389
395
get_current_workflow , get_node_info ],
390
396
handoffs = [agent ],
397
+ config = {
398
+ "max_tokens" : 8192
399
+ }
391
400
)
392
401
393
402
parameter_agent = create_agent (
@@ -479,6 +488,9 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
479
488
tools = [find_matching_parameter_value , get_model_files ,
480
489
suggest_model_download , update_workflow_parameter , get_current_workflow ],
481
490
handoffs = [agent ],
491
+ config = {
492
+ "max_tokens" : 8192
493
+ }
482
494
)
483
495
484
496
agent .handoffs = [link_agent , workflow_bugfix_default_agent , parameter_agent ]
@@ -719,7 +731,8 @@ async def test_debug():
719
731
720
732
config = {
721
733
"session_id" : "test_session_123" ,
722
- "model" : WORKFLOW_MODEL_NAME
734
+ "model" : WORKFLOW_MODEL_NAME ,
735
+ "max_tokens" : 8192
723
736
}
724
737
725
738
async for text , ext in debug_workflow_errors (test_workflow_data , config ):
0 commit comments