Skip to content

Commit 14cfea5

Browse files
Merge branch 'mcp'
2 parents d1ee6ab + c478711 commit 14cfea5

File tree

2 files changed

+15
-2
lines changed

2 files changed

+15
-2
lines changed

backend/service/debug_agent.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -254,6 +254,9 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
254254
Start by validating the workflow to see its current state.""",
255255
model=WORKFLOW_MODEL_NAME,
256256
tools=[run_workflow, analyze_error_type, save_current_workflow],
257+
config={
258+
"max_tokens": 8192
259+
}
257260
)
258261

259262
workflow_bugfix_default_agent = create_agent(
@@ -300,6 +303,9 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
300303
""",
301304
tools=[get_current_workflow, get_node_info, update_workflow],
302305
handoffs=[agent],
306+
config={
307+
"max_tokens": 8192
308+
}
303309
)
304310

305311
link_agent = create_agent(
@@ -388,6 +394,9 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
388394
tools=[analyze_missing_connections, apply_connection_fixes,
389395
get_current_workflow, get_node_info],
390396
handoffs=[agent],
397+
config={
398+
"max_tokens": 8192
399+
}
391400
)
392401

393402
parameter_agent = create_agent(
@@ -479,6 +488,9 @@ async def debug_workflow_errors(workflow_data: Dict[str, Any]):
479488
tools=[find_matching_parameter_value, get_model_files,
480489
suggest_model_download, update_workflow_parameter, get_current_workflow],
481490
handoffs=[agent],
491+
config={
492+
"max_tokens": 8192
493+
}
482494
)
483495

484496
agent.handoffs = [link_agent, workflow_bugfix_default_agent, parameter_agent]
@@ -719,7 +731,8 @@ async def test_debug():
719731

720732
config = {
721733
"session_id": "test_session_123",
722-
"model": WORKFLOW_MODEL_NAME
734+
"model": WORKFLOW_MODEL_NAME,
735+
"max_tokens": 8192
723736
}
724737

725738
async for text, ext in debug_workflow_errors(test_workflow_data, config):

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
[project]
22
name = "ComfyUI-Copilot"
33
description = "Your Intelligent Assistant for Comfy-UI."
4-
version = "2.0.5"
4+
version = "2.0.6"
55
license = {file = "LICENSE"}
66

77
[project.urls]

0 commit comments

Comments
 (0)