Merge branch 'feature/mcp-ui' into bot_manager
This commit is contained in:
commit
8482ec35d9
@ -92,6 +92,7 @@ async def enhanced_generate_stream_response(
|
|||||||
logger.info(f"Starting agent stream response")
|
logger.info(f"Starting agent stream response")
|
||||||
chunk_id = 0
|
chunk_id = 0
|
||||||
message_tag = ""
|
message_tag = ""
|
||||||
|
current_tool_name = ""
|
||||||
last_answer_first_char_duration_ms = None
|
last_answer_first_char_duration_ms = None
|
||||||
waiting_for_answer_first_char = False
|
waiting_for_answer_first_char = False
|
||||||
agent, checkpointer, sandbox = await init_agent(config)
|
agent, checkpointer, sandbox = await init_agent(config)
|
||||||
@ -108,10 +109,13 @@ async def enhanced_generate_stream_response(
|
|||||||
if msg.tool_call_chunks:
|
if msg.tool_call_chunks:
|
||||||
message_tag = "TOOL_CALL"
|
message_tag = "TOOL_CALL"
|
||||||
waiting_for_answer_first_char = False
|
waiting_for_answer_first_char = False
|
||||||
if config.tool_response:
|
for tool_call_chunk in msg.tool_call_chunks:
|
||||||
for tool_call_chunk in msg.tool_call_chunks:
|
chunk_name = tool_call_chunk.get("name") if isinstance(tool_call_chunk, dict) else getattr(tool_call_chunk, "name", None)
|
||||||
chunk_name = tool_call_chunk.get("name") if isinstance(tool_call_chunk, dict) else getattr(tool_call_chunk, "name", None)
|
chunk_args = tool_call_chunk.get("args") if isinstance(tool_call_chunk, dict) else getattr(tool_call_chunk, "args", None)
|
||||||
chunk_args = tool_call_chunk.get("args") if isinstance(tool_call_chunk, dict) else getattr(tool_call_chunk, "args", None)
|
if chunk_name:
|
||||||
|
current_tool_name = chunk_name
|
||||||
|
# Always output ask_user tool calls even when tool_response is disabled
|
||||||
|
if config.tool_response or current_tool_name == 'ask_user':
|
||||||
if chunk_name:
|
if chunk_name:
|
||||||
new_content = f"[{message_tag}] {chunk_name}\n"
|
new_content = f"[{message_tag}] {chunk_name}\n"
|
||||||
if chunk_args:
|
if chunk_args:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user